{
  "best_global_step": null,
  "best_metric": null,
  "best_model_checkpoint": null,
  "epoch": 0.6,
  "eval_steps": 2000,
  "global_step": 30000,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 4e-05,
      "grad_norm": 456.0,
      "learning_rate": 1.18e-05,
      "loss": 85.4554,
      "loss/crossentropy": 9.650346755981445,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 8.066818237304688,
      "step": 2
    },
    {
      "epoch": 8e-05,
      "grad_norm": 416.0,
      "learning_rate": 1.3600000000000002e-05,
      "loss": 84.3418,
      "loss/crossentropy": 9.544375896453857,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 7.628942489624023,
      "step": 4
    },
    {
      "epoch": 0.00012,
      "grad_norm": 466.0,
      "learning_rate": 1.54e-05,
      "loss": 87.2187,
      "loss/crossentropy": 9.569977283477783,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 7.7909746170043945,
      "step": 6
    },
    {
      "epoch": 0.00016,
      "grad_norm": 247.0,
      "learning_rate": 1.72e-05,
      "loss": 82.5078,
      "loss/crossentropy": 9.06786823272705,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 7.3673131465911865,
      "step": 8
    },
    {
      "epoch": 0.0002,
      "grad_norm": 179.0,
      "learning_rate": 1.9e-05,
      "loss": 78.2757,
      "loss/crossentropy": 8.918366432189941,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 6.912693023681641,
      "step": 10
    },
    {
      "epoch": 0.00024,
      "grad_norm": 148.0,
      "learning_rate": 2.0800000000000004e-05,
      "loss": 74.4248,
      "loss/crossentropy": 8.443636417388916,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 6.567321538925171,
      "step": 12
    },
    {
      "epoch": 0.00028,
      "grad_norm": 131.0,
      "learning_rate": 2.2600000000000004e-05,
      "loss": 73.0003,
      "loss/crossentropy": 8.428278923034668,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 6.706400156021118,
      "step": 14
    },
    {
      "epoch": 0.00032,
      "grad_norm": 181.0,
      "grad_norm_var": 16279.8625,
      "learning_rate": 2.4400000000000004e-05,
      "loss": 70.0047,
      "loss/crossentropy": 8.216889381408691,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 6.056080102920532,
      "step": 16
    },
    {
      "epoch": 0.00036,
      "grad_norm": 90.5,
      "grad_norm_var": 14154.148958333333,
      "learning_rate": 2.6200000000000003e-05,
      "loss": 69.9766,
      "loss/crossentropy": 8.191599607467651,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 6.429446697235107,
      "step": 18
    },
    {
      "epoch": 0.0004,
      "grad_norm": 52.25,
      "grad_norm_var": 12194.27890625,
      "learning_rate": 2.8000000000000003e-05,
      "loss": 64.3807,
      "loss/crossentropy": 7.506032228469849,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 5.794633388519287,
      "step": 20
    },
    {
      "epoch": 0.00044,
      "grad_norm": 39.25,
      "grad_norm_var": 6249.4875,
      "learning_rate": 2.9800000000000006e-05,
      "loss": 61.2802,
      "loss/crossentropy": 7.152851343154907,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 5.261489152908325,
      "step": 22
    },
    {
      "epoch": 0.00048,
      "grad_norm": 57.0,
      "grad_norm_var": 4626.8875,
      "learning_rate": 3.16e-05,
      "loss": 58.3454,
      "loss/crossentropy": 6.956738471984863,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 5.020895004272461,
      "step": 24
    },
    {
      "epoch": 0.00052,
      "grad_norm": 86.0,
      "grad_norm_var": 4244.565625,
      "learning_rate": 3.3400000000000005e-05,
      "loss": 54.2703,
      "loss/crossentropy": 6.686542987823486,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 4.801911354064941,
      "step": 26
    },
    {
      "epoch": 0.00056,
      "grad_norm": 110.5,
      "grad_norm_var": 3868.5875,
      "learning_rate": 3.520000000000001e-05,
      "loss": 51.7343,
      "loss/crossentropy": 6.4867262840271,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 4.5746169090271,
      "step": 28
    },
    {
      "epoch": 0.0006,
      "grad_norm": 50.0,
      "grad_norm_var": 3953.82890625,
      "learning_rate": 3.7e-05,
      "loss": 49.6807,
      "loss/crossentropy": 6.364065408706665,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 4.230688810348511,
      "step": 30
    },
    {
      "epoch": 0.00064,
      "grad_norm": 68.0,
      "grad_norm_var": 3157.4958333333334,
      "learning_rate": 3.88e-05,
      "loss": 44.7112,
      "loss/crossentropy": 5.731794834136963,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 3.6969637870788574,
      "step": 32
    },
    {
      "epoch": 0.00068,
      "grad_norm": 50.75,
      "grad_norm_var": 500.59973958333336,
      "learning_rate": 4.0600000000000004e-05,
      "loss": 42.2923,
      "loss/crossentropy": 5.553718328475952,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 3.7271878719329834,
      "step": 34
    },
    {
      "epoch": 0.00072,
      "grad_norm": 55.5,
      "grad_norm_var": 342.2122395833333,
      "learning_rate": 4.240000000000001e-05,
      "loss": 37.7465,
      "loss/crossentropy": 5.023651361465454,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 3.2670111656188965,
      "step": 36
    },
    {
      "epoch": 0.00076,
      "grad_norm": 75.5,
      "grad_norm_var": 283.37395833333335,
      "learning_rate": 4.420000000000001e-05,
      "loss": 35.1313,
      "loss/crossentropy": 4.921839237213135,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 2.950661063194275,
      "step": 38
    },
    {
      "epoch": 0.0008,
      "grad_norm": 44.5,
      "grad_norm_var": 299.1958333333333,
      "learning_rate": 4.600000000000001e-05,
      "loss": 32.3316,
      "loss/crossentropy": 4.782621145248413,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 2.7473502159118652,
      "step": 40
    },
    {
      "epoch": 0.00084,
      "grad_norm": 36.5,
      "grad_norm_var": 361.1372395833333,
      "learning_rate": 4.78e-05,
      "loss": 28.4104,
      "loss/crossentropy": 3.8754972219467163,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 2.410745143890381,
      "step": 42
    },
    {
      "epoch": 0.00088,
      "grad_norm": 36.25,
      "grad_norm_var": 225.80598958333334,
      "learning_rate": 4.96e-05,
      "loss": 26.1806,
      "loss/crossentropy": 3.9885865449905396,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 2.1414425373077393,
      "step": 44
    },
    {
      "epoch": 0.00092,
      "grad_norm": 46.0,
      "grad_norm_var": 249.475,
      "learning_rate": 5.14e-05,
      "loss": 24.4012,
      "loss/crossentropy": 3.750515580177307,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 2.063339352607727,
      "step": 46
    },
    {
      "epoch": 0.00096,
      "grad_norm": 20.875,
      "grad_norm_var": 315.8291015625,
      "learning_rate": 5.3200000000000006e-05,
      "loss": 22.822,
      "loss/crossentropy": 3.7912577390670776,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 1.9196046590805054,
      "step": 48
    },
    {
      "epoch": 0.001,
      "grad_norm": 35.5,
      "grad_norm_var": 279.284375,
      "learning_rate": 5.500000000000001e-05,
      "loss": 21.036,
      "loss/crossentropy": 3.777758002281189,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 1.7479944229125977,
      "step": 50
    },
    {
      "epoch": 0.00104,
      "grad_norm": 20.75,
      "grad_norm_var": 306.15149739583336,
      "learning_rate": 5.680000000000001e-05,
      "loss": 20.3608,
      "loss/crossentropy": 3.5903185606002808,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 1.5533717274665833,
      "step": 52
    },
    {
      "epoch": 0.00108,
      "grad_norm": 43.75,
      "grad_norm_var": 213.07395833333334,
      "learning_rate": 5.860000000000001e-05,
      "loss": 18.813,
      "loss/crossentropy": 3.691780686378479,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 1.4755533933639526,
      "step": 54
    },
    {
      "epoch": 0.00112,
      "grad_norm": 21.25,
      "grad_norm_var": 70.690625,
      "learning_rate": 6.040000000000001e-05,
      "loss": 19.1421,
      "loss/crossentropy": 3.557003617286682,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 1.5198156833648682,
      "step": 56
    },
    {
      "epoch": 0.00116,
      "grad_norm": 21.5,
      "grad_norm_var": 76.30390625,
      "learning_rate": 6.220000000000001e-05,
      "loss": 17.2705,
      "loss/crossentropy": 3.2730292081832886,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 1.4131997227668762,
      "step": 58
    },
    {
      "epoch": 0.0012,
      "grad_norm": 19.875,
      "grad_norm_var": 77.11399739583334,
      "learning_rate": 6.400000000000001e-05,
      "loss": 16.4712,
      "loss/crossentropy": 3.419156074523926,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 1.3277101516723633,
      "step": 60
    },
    {
      "epoch": 0.00124,
      "grad_norm": 25.75,
      "grad_norm_var": 48.60520833333333,
      "learning_rate": 6.58e-05,
      "loss": 16.6219,
      "loss/crossentropy": 2.973878502845764,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 1.3438007831573486,
      "step": 62
    },
    {
      "epoch": 0.00128,
      "grad_norm": 34.5,
      "grad_norm_var": 53.18020833333333,
      "learning_rate": 6.76e-05,
      "loss": 15.0929,
      "loss/crossentropy": 2.892021059989929,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 1.1624282002449036,
      "step": 64
    },
    {
      "epoch": 0.00132,
      "grad_norm": 15.4375,
      "grad_norm_var": 51.195947265625,
      "learning_rate": 6.94e-05,
      "loss": 15.1967,
      "loss/crossentropy": 2.954660177230835,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 1.116301715373993,
      "step": 66
    },
    {
      "epoch": 0.00136,
      "grad_norm": 32.0,
      "grad_norm_var": 51.064306640625,
      "learning_rate": 7.120000000000001e-05,
      "loss": 14.9397,
      "loss/crossentropy": 3.2686156034469604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 1.1978037357330322,
      "step": 68
    },
    {
      "epoch": 0.0014,
      "grad_norm": 29.25,
      "grad_norm_var": 32.794270833333336,
      "learning_rate": 7.3e-05,
      "loss": 14.4846,
      "loss/crossentropy": 2.7956581115722656,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 1.1973016262054443,
      "step": 70
    },
    {
      "epoch": 0.00144,
      "grad_norm": 15.25,
      "grad_norm_var": 37.80149739583333,
      "learning_rate": 7.48e-05,
      "loss": 14.1296,
      "loss/crossentropy": 3.08966863155365,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 1.151496708393097,
      "step": 72
    },
    {
      "epoch": 0.00148,
      "grad_norm": 18.625,
      "grad_norm_var": 41.88984375,
      "learning_rate": 7.66e-05,
      "loss": 13.6812,
      "loss/crossentropy": 2.949987292289734,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.9787414371967316,
      "step": 74
    },
    {
      "epoch": 0.00152,
      "grad_norm": 15.1875,
      "grad_norm_var": 49.143489583333334,
      "learning_rate": 7.840000000000001e-05,
      "loss": 12.8901,
      "loss/crossentropy": 3.1161292791366577,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 1.0640113949775696,
      "step": 76
    },
    {
      "epoch": 0.00156,
      "grad_norm": 22.125,
      "grad_norm_var": 47.63170572916667,
      "learning_rate": 8.020000000000001e-05,
      "loss": 13.157,
      "loss/crossentropy": 3.3661664724349976,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 1.1353825330734253,
      "step": 78
    },
    {
      "epoch": 0.0016,
      "grad_norm": 16.25,
      "grad_norm_var": 35.33274739583333,
      "learning_rate": 8.200000000000001e-05,
      "loss": 12.9372,
      "loss/crossentropy": 2.927241563796997,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.9984134435653687,
      "step": 80
    },
    {
      "epoch": 0.00164,
      "grad_norm": 13.625,
      "grad_norm_var": 37.53951822916667,
      "learning_rate": 8.38e-05,
      "loss": 12.0477,
      "loss/crossentropy": 3.1273285150527954,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.9241160154342651,
      "step": 82
    },
    {
      "epoch": 0.00168,
      "grad_norm": 19.125,
      "grad_norm_var": 19.602718098958334,
      "learning_rate": 8.560000000000001e-05,
      "loss": 11.9084,
      "loss/crossentropy": 2.737278938293457,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.8822851181030273,
      "step": 84
    },
    {
      "epoch": 0.00172,
      "grad_norm": 13.0625,
      "grad_norm_var": 11.107014973958334,
      "learning_rate": 8.740000000000001e-05,
      "loss": 11.8594,
      "loss/crossentropy": 2.4452388286590576,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.8693483769893646,
      "step": 86
    },
    {
      "epoch": 0.00176,
      "grad_norm": 18.375,
      "grad_norm_var": 11.328369140625,
      "learning_rate": 8.92e-05,
      "loss": 11.5058,
      "loss/crossentropy": 2.89771831035614,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.8749278783798218,
      "step": 88
    },
    {
      "epoch": 0.0018,
      "grad_norm": 12.1875,
      "grad_norm_var": 13.811572265625,
      "learning_rate": 9.1e-05,
      "loss": 11.9281,
      "loss/crossentropy": 3.0173208713531494,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.8714744746685028,
      "step": 90
    },
    {
      "epoch": 0.00184,
      "grad_norm": 20.375,
      "grad_norm_var": 16.114306640625,
      "learning_rate": 9.28e-05,
      "loss": 11.4244,
      "loss/crossentropy": 2.588515043258667,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.9651442170143127,
      "step": 92
    },
    {
      "epoch": 0.00188,
      "grad_norm": 13.375,
      "grad_norm_var": 12.784309895833333,
      "learning_rate": 9.46e-05,
      "loss": 11.6092,
      "loss/crossentropy": 2.8116774559020996,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.9143906235694885,
      "step": 94
    },
    {
      "epoch": 0.00192,
      "grad_norm": 16.375,
      "grad_norm_var": 13.225764973958333,
      "learning_rate": 9.64e-05,
      "loss": 10.6723,
      "loss/crossentropy": 2.8734441995620728,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.814399927854538,
      "step": 96
    },
    {
      "epoch": 0.00196,
      "grad_norm": 10.3125,
      "grad_norm_var": 17.228238932291667,
      "learning_rate": 9.82e-05,
      "loss": 10.6577,
      "loss/crossentropy": 2.664194703102112,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.7905783653259277,
      "step": 98
    },
    {
      "epoch": 0.002,
      "grad_norm": 12.3125,
      "grad_norm_var": 17.113785807291666,
      "learning_rate": 0.0001,
      "loss": 10.6847,
      "loss/crossentropy": 2.4851003885269165,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.769305944442749,
      "step": 100
    },
    {
      "epoch": 0.00204,
      "grad_norm": 10.75,
      "grad_norm_var": 13.9140625,
      "learning_rate": 0.0001,
      "loss": 10.8119,
      "loss/crossentropy": 2.2757182121276855,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.8241511881351471,
      "step": 102
    },
    {
      "epoch": 0.00208,
      "grad_norm": 11.375,
      "grad_norm_var": 13.615738932291666,
      "learning_rate": 0.0001,
      "loss": 10.6244,
      "loss/crossentropy": 2.7211785316467285,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.8518709540367126,
      "step": 104
    },
    {
      "epoch": 0.00212,
      "grad_norm": 11.625,
      "grad_norm_var": 14.163395182291667,
      "learning_rate": 0.0001,
      "loss": 10.5629,
      "loss/crossentropy": 2.387019991874695,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.7568954229354858,
      "step": 106
    },
    {
      "epoch": 0.00216,
      "grad_norm": 9.9375,
      "grad_norm_var": 10.788525390625,
      "learning_rate": 0.0001,
      "loss": 10.1364,
      "loss/crossentropy": 2.5363346338272095,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.7648341059684753,
      "step": 108
    },
    {
      "epoch": 0.0022,
      "grad_norm": 20.0,
      "grad_norm_var": 14.898893229166667,
      "learning_rate": 0.0001,
      "loss": 10.8773,
      "loss/crossentropy": 2.8450236320495605,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.737193763256073,
      "step": 110
    },
    {
      "epoch": 0.00224,
      "grad_norm": 10.25,
      "grad_norm_var": 15.364518229166666,
      "learning_rate": 0.0001,
      "loss": 9.4554,
      "loss/crossentropy": 2.4827451705932617,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.6924614012241364,
      "step": 112
    },
    {
      "epoch": 0.00228,
      "grad_norm": 10.5625,
      "grad_norm_var": 7.461832682291667,
      "learning_rate": 0.0001,
      "loss": 9.9651,
      "loss/crossentropy": 2.093318462371826,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.7038464546203613,
      "step": 114
    },
    {
      "epoch": 0.00232,
      "grad_norm": 9.6875,
      "grad_norm_var": 7.597330729166667,
      "learning_rate": 0.0001,
      "loss": 10.0297,
      "loss/crossentropy": 2.5149790048599243,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.6980189085006714,
      "step": 116
    },
    {
      "epoch": 0.00236,
      "grad_norm": 11.625,
      "grad_norm_var": 6.672509765625,
      "learning_rate": 0.0001,
      "loss": 9.8176,
      "loss/crossentropy": 2.6022276878356934,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.7005251348018646,
      "step": 118
    },
    {
      "epoch": 0.0024,
      "grad_norm": 7.625,
      "grad_norm_var": 7.298160807291667,
      "learning_rate": 0.0001,
      "loss": 9.5658,
      "loss/crossentropy": 2.6836462020874023,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.7237774729728699,
      "step": 120
    },
    {
      "epoch": 0.00244,
      "grad_norm": 9.5625,
      "grad_norm_var": 7.402018229166667,
      "learning_rate": 0.0001,
      "loss": 9.7376,
      "loss/crossentropy": 2.6823805570602417,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.7570162117481232,
      "step": 122
    },
    {
      "epoch": 0.00248,
      "grad_norm": 11.25,
      "grad_norm_var": 7.391259765625,
      "learning_rate": 0.0001,
      "loss": 9.4713,
      "loss/crossentropy": 2.6233514547348022,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.7183247208595276,
      "step": 124
    },
    {
      "epoch": 0.00252,
      "grad_norm": 9.9375,
      "grad_norm_var": 1.0839680989583333,
      "learning_rate": 0.0001,
      "loss": 9.2243,
      "loss/crossentropy": 2.331676959991455,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.6982125043869019,
      "step": 126
    },
    {
      "epoch": 0.00256,
      "grad_norm": 9.1875,
      "grad_norm_var": 0.9687337239583333,
      "learning_rate": 0.0001,
      "loss": 9.4777,
      "loss/crossentropy": 2.429046392440796,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.696417510509491,
      "step": 128
    },
    {
      "epoch": 0.0026,
      "grad_norm": 14.8125,
      "grad_norm_var": 2.448291015625,
      "learning_rate": 0.0001,
      "loss": 9.9024,
      "loss/crossentropy": 2.5262571573257446,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.716008871793747,
      "step": 130
    },
    {
      "epoch": 0.00264,
      "grad_norm": 8.8125,
      "grad_norm_var": 2.7044270833333335,
      "learning_rate": 0.0001,
      "loss": 9.2836,
      "loss/crossentropy": 2.187526524066925,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.666053056716919,
      "step": 132
    },
    {
      "epoch": 0.00268,
      "grad_norm": 8.9375,
      "grad_norm_var": 3.285270182291667,
      "learning_rate": 0.0001,
      "loss": 9.8338,
      "loss/crossentropy": 2.4199057817459106,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.6709816455841064,
      "step": 134
    },
    {
      "epoch": 0.00272,
      "grad_norm": 9.6875,
      "grad_norm_var": 3.4072916666666666,
      "learning_rate": 0.0001,
      "loss": 9.4225,
      "loss/crossentropy": 2.1963008642196655,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5963725447654724,
      "step": 136
    },
    {
      "epoch": 0.00276,
      "grad_norm": 8.0625,
      "grad_norm_var": 3.556884765625,
      "learning_rate": 0.0001,
      "loss": 9.44,
      "loss/crossentropy": 2.5878132581710815,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.6362220048904419,
      "step": 138
    },
    {
      "epoch": 0.0028,
      "grad_norm": 9.25,
      "grad_norm_var": 3.4852701822916665,
      "learning_rate": 0.0001,
      "loss": 9.4314,
      "loss/crossentropy": 2.7800480127334595,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.6819457113742828,
      "step": 140
    },
    {
      "epoch": 0.00284,
      "grad_norm": 8.4375,
      "grad_norm_var": 3.838997395833333,
      "learning_rate": 0.0001,
      "loss": 9.1047,
      "loss/crossentropy": 2.5055110454559326,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.674308180809021,
      "step": 142
    },
    {
      "epoch": 0.00288,
      "grad_norm": 8.25,
      "grad_norm_var": 4.078499348958333,
      "learning_rate": 0.0001,
      "loss": 9.1578,
      "loss/crossentropy": 2.8532944917678833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.7083481848239899,
      "step": 144
    },
    {
      "epoch": 0.00292,
      "grad_norm": 7.8125,
      "grad_norm_var": 2.2759765625,
      "learning_rate": 0.0001,
      "loss": 8.8023,
      "loss/crossentropy": 2.442527174949646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.6510869562625885,
      "step": 146
    },
    {
      "epoch": 0.00296,
      "grad_norm": 10.9375,
      "grad_norm_var": 2.8544881184895834,
      "learning_rate": 0.0001,
      "loss": 8.7238,
      "loss/crossentropy": 2.516597867012024,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.6362285614013672,
      "step": 148
    },
    {
      "epoch": 0.003,
      "grad_norm": 7.40625,
      "grad_norm_var": 1.8615885416666667,
      "learning_rate": 0.0001,
      "loss": 8.5543,
      "loss/crossentropy": 2.8672900199890137,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.6459421515464783,
      "step": 150
    },
    {
      "epoch": 0.00304,
      "grad_norm": 8.1875,
      "grad_norm_var": 1.7195963541666666,
      "learning_rate": 0.0001,
      "loss": 8.6403,
      "loss/crossentropy": 2.2042795419692993,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5720505118370056,
      "step": 152
    },
    {
      "epoch": 0.00308,
      "grad_norm": 7.8125,
      "grad_norm_var": 1.876025390625,
      "learning_rate": 0.0001,
      "loss": 8.768,
      "loss/crossentropy": 2.225563883781433,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.6328675150871277,
      "step": 154
    },
    {
      "epoch": 0.00312,
      "grad_norm": 8.0625,
      "grad_norm_var": 1.5925618489583333,
      "learning_rate": 0.0001,
      "loss": 8.5743,
      "loss/crossentropy": 2.3541462421417236,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5927431881427765,
      "step": 156
    },
    {
      "epoch": 0.00316,
      "grad_norm": 7.34375,
      "grad_norm_var": 1.834619140625,
      "learning_rate": 0.0001,
      "loss": 8.7329,
      "loss/crossentropy": 2.4685616493225098,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.6299368739128113,
      "step": 158
    },
    {
      "epoch": 0.0032,
      "grad_norm": 9.25,
      "grad_norm_var": 1.4429646809895833,
      "learning_rate": 0.0001,
      "loss": 8.4796,
      "loss/crossentropy": 2.4637919664382935,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.6084515154361725,
      "step": 160
    },
    {
      "epoch": 0.00324,
      "grad_norm": 9.0625,
      "grad_norm_var": 1.3692545572916666,
      "learning_rate": 0.0001,
      "loss": 9.0446,
      "loss/crossentropy": 2.598397374153137,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.574739396572113,
      "step": 162
    },
    {
      "epoch": 0.00328,
      "grad_norm": 7.15625,
      "grad_norm_var": 0.9171834309895833,
      "learning_rate": 0.0001,
      "loss": 8.1508,
      "loss/crossentropy": 2.5183030366897583,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5985860526561737,
      "step": 164
    },
    {
      "epoch": 0.00332,
      "grad_norm": 9.125,
      "grad_norm_var": 0.9571614583333333,
      "learning_rate": 0.0001,
      "loss": 8.4296,
      "loss/crossentropy": 2.252183437347412,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5627379417419434,
      "step": 166
    },
    {
      "epoch": 0.00336,
      "grad_norm": 7.875,
      "grad_norm_var": 0.70533447265625,
      "learning_rate": 0.0001,
      "loss": 8.4549,
      "loss/crossentropy": 2.5720516443252563,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5945309698581696,
      "step": 168
    },
    {
      "epoch": 0.0034,
      "grad_norm": 9.1875,
      "grad_norm_var": 0.8844034830729167,
      "learning_rate": 0.0001,
      "loss": 8.6096,
      "loss/crossentropy": 2.3004332184791565,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5401738286018372,
      "step": 170
    },
    {
      "epoch": 0.00344,
      "grad_norm": 7.71875,
      "grad_norm_var": 0.948681640625,
      "learning_rate": 0.0001,
      "loss": 8.5484,
      "loss/crossentropy": 2.689734935760498,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.6384358406066895,
      "step": 172
    },
    {
      "epoch": 0.00348,
      "grad_norm": 6.84375,
      "grad_norm_var": 0.9418253580729167,
      "learning_rate": 0.0001,
      "loss": 8.1888,
      "loss/crossentropy": 1.944397747516632,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5828913450241089,
      "step": 174
    },
    {
      "epoch": 0.00352,
      "grad_norm": 6.53125,
      "grad_norm_var": 0.8817342122395834,
      "learning_rate": 0.0001,
      "loss": 8.0577,
      "loss/crossentropy": 2.8166507482528687,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5916908979415894,
      "step": 176
    },
    {
      "epoch": 0.00356,
      "grad_norm": 7.09375,
      "grad_norm_var": 0.7456990559895833,
      "learning_rate": 0.0001,
      "loss": 8.7701,
      "loss/crossentropy": 2.3208402395248413,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5968939661979675,
      "step": 178
    },
    {
      "epoch": 0.0036,
      "grad_norm": 7.25,
      "grad_norm_var": 0.7425130208333334,
      "learning_rate": 0.0001,
      "loss": 8.2615,
      "loss/crossentropy": 2.817763566970825,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5800873041152954,
      "step": 180
    },
    {
      "epoch": 0.00364,
      "grad_norm": 6.78125,
      "grad_norm_var": 0.6126912434895834,
      "learning_rate": 0.0001,
      "loss": 8.3053,
      "loss/crossentropy": 2.250023365020752,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5279016494750977,
      "step": 182
    },
    {
      "epoch": 0.00368,
      "grad_norm": 6.09375,
      "grad_norm_var": 0.6917805989583333,
      "learning_rate": 0.0001,
      "loss": 7.7974,
      "loss/crossentropy": 2.1337096095085144,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5393811166286469,
      "step": 184
    },
    {
      "epoch": 0.00372,
      "grad_norm": 6.34375,
      "grad_norm_var": 0.5962198893229167,
      "learning_rate": 0.0001,
      "loss": 7.7258,
      "loss/crossentropy": 2.6338934898376465,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.6444519460201263,
      "step": 186
    },
    {
      "epoch": 0.00376,
      "grad_norm": 10.4375,
      "grad_norm_var": 1.2786458333333333,
      "learning_rate": 0.0001,
      "loss": 8.0762,
      "loss/crossentropy": 2.66677463054657,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.6012931764125824,
      "step": 188
    },
    {
      "epoch": 0.0038,
      "grad_norm": 6.375,
      "grad_norm_var": 1.6642537434895834,
      "learning_rate": 0.0001,
      "loss": 8.3177,
      "loss/crossentropy": 2.3731196522712708,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.6299596428871155,
      "step": 190
    },
    {
      "epoch": 0.00384,
      "grad_norm": 7.4375,
      "grad_norm_var": 1.7719889322916667,
      "learning_rate": 0.0001,
      "loss": 8.214,
      "loss/crossentropy": 2.411492705345154,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5328834354877472,
      "step": 192
    },
    {
      "epoch": 0.00388,
      "grad_norm": 7.15625,
      "grad_norm_var": 1.9489420572916667,
      "learning_rate": 0.0001,
      "loss": 7.9763,
      "loss/crossentropy": 2.2402734756469727,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.539243072271347,
      "step": 194
    },
    {
      "epoch": 0.00392,
      "grad_norm": 7.53125,
      "grad_norm_var": 1.895556640625,
      "learning_rate": 0.0001,
      "loss": 7.9292,
      "loss/crossentropy": 2.3250681161880493,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5374342203140259,
      "step": 196
    },
    {
      "epoch": 0.00396,
      "grad_norm": 7.75,
      "grad_norm_var": 1.8979451497395834,
      "learning_rate": 0.0001,
      "loss": 7.9201,
      "loss/crossentropy": 2.42138135433197,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5410921573638916,
      "step": 198
    },
    {
      "epoch": 0.004,
      "grad_norm": 6.78125,
      "grad_norm_var": 1.8954386393229166,
      "learning_rate": 0.0001,
      "loss": 7.7597,
      "loss/crossentropy": 2.1954251527786255,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.49367184937000275,
      "step": 200
    },
    {
      "epoch": 0.00404,
      "grad_norm": 6.75,
      "grad_norm_var": 1.6848307291666667,
      "learning_rate": 0.0001,
      "loss": 7.9033,
      "loss/crossentropy": 2.81479811668396,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5567455887794495,
      "step": 202
    },
    {
      "epoch": 0.00408,
      "grad_norm": 6.5,
      "grad_norm_var": 1.08228759765625,
      "learning_rate": 0.0001,
      "loss": 7.9812,
      "loss/crossentropy": 2.611761450767517,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.544409453868866,
      "step": 204
    },
    {
      "epoch": 0.00412,
      "grad_norm": 6.125,
      "grad_norm_var": 0.650244140625,
      "learning_rate": 0.0001,
      "loss": 7.7921,
      "loss/crossentropy": 2.1369245052337646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5181446373462677,
      "step": 206
    },
    {
      "epoch": 0.00416,
      "grad_norm": 7.1875,
      "grad_norm_var": 0.30745035807291665,
      "learning_rate": 0.0001,
      "loss": 8.3375,
      "loss/crossentropy": 2.435856580734253,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5506252646446228,
      "step": 208
    },
    {
      "epoch": 0.0042,
      "grad_norm": 6.21875,
      "grad_norm_var": 0.26848958333333334,
      "learning_rate": 0.0001,
      "loss": 7.7599,
      "loss/crossentropy": 2.2404768466949463,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.46975645422935486,
      "step": 210
    },
    {
      "epoch": 0.00424,
      "grad_norm": 6.71875,
      "grad_norm_var": 0.19713541666666667,
      "learning_rate": 0.0001,
      "loss": 7.7083,
      "loss/crossentropy": 2.4866777658462524,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5217231214046478,
      "step": 212
    },
    {
      "epoch": 0.00428,
      "grad_norm": 6.15625,
      "grad_norm_var": 0.121484375,
      "learning_rate": 0.0001,
      "loss": 7.6519,
      "loss/crossentropy": 2.074867010116577,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4683506190776825,
      "step": 214
    },
    {
      "epoch": 0.00432,
      "grad_norm": 7.59375,
      "grad_norm_var": 0.21679280598958334,
      "learning_rate": 0.0001,
      "loss": 7.6062,
      "loss/crossentropy": 2.2040151357650757,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5235520601272583,
      "step": 216
    },
    {
      "epoch": 0.00436,
      "grad_norm": 7.25,
      "grad_norm_var": 0.25690104166666666,
      "learning_rate": 0.0001,
      "loss": 7.886,
      "loss/crossentropy": 2.174479365348816,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5243187248706818,
      "step": 218
    },
    {
      "epoch": 0.0044,
      "grad_norm": 6.28125,
      "grad_norm_var": 0.26523030598958336,
      "learning_rate": 0.0001,
      "loss": 7.7535,
      "loss/crossentropy": 2.5678584575653076,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5818615555763245,
      "step": 220
    },
    {
      "epoch": 0.00444,
      "grad_norm": 6.0625,
      "grad_norm_var": 0.2814412434895833,
      "learning_rate": 0.0001,
      "loss": 7.859,
      "loss/crossentropy": 2.4551891088485718,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5400412082672119,
      "step": 222
    },
    {
      "epoch": 0.00448,
      "grad_norm": 6.15625,
      "grad_norm_var": 0.25310872395833334,
      "learning_rate": 0.0001,
      "loss": 7.7341,
      "loss/crossentropy": 2.0638335943222046,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.49279752373695374,
      "step": 224
    },
    {
      "epoch": 0.00452,
      "grad_norm": 6.6875,
      "grad_norm_var": 0.38235677083333336,
      "learning_rate": 0.0001,
      "loss": 8.1064,
      "loss/crossentropy": 2.553247332572937,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5566797852516174,
      "step": 226
    },
    {
      "epoch": 0.00456,
      "grad_norm": 6.0625,
      "grad_norm_var": 0.39451497395833335,
      "learning_rate": 0.0001,
      "loss": 7.7812,
      "loss/crossentropy": 2.544332265853882,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5118530094623566,
      "step": 228
    },
    {
      "epoch": 0.0046,
      "grad_norm": 5.875,
      "grad_norm_var": 0.4054036458333333,
      "learning_rate": 0.0001,
      "loss": 7.064,
      "loss/crossentropy": 2.191234052181244,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5157675743103027,
      "step": 230
    },
    {
      "epoch": 0.00464,
      "grad_norm": 5.4375,
      "grad_norm_var": 0.360791015625,
      "learning_rate": 0.0001,
      "loss": 7.6611,
      "loss/crossentropy": 2.3671151399612427,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5233491659164429,
      "step": 232
    },
    {
      "epoch": 0.00468,
      "grad_norm": 7.125,
      "grad_norm_var": 0.4180826822916667,
      "learning_rate": 0.0001,
      "loss": 7.4509,
      "loss/crossentropy": 2.3003920316696167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5037694871425629,
      "step": 234
    },
    {
      "epoch": 0.00472,
      "grad_norm": 5.625,
      "grad_norm_var": 0.4305338541666667,
      "learning_rate": 0.0001,
      "loss": 7.8236,
      "loss/crossentropy": 2.4672670364379883,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5187652707099915,
      "step": 236
    },
    {
      "epoch": 0.00476,
      "grad_norm": 6.0625,
      "grad_norm_var": 0.4493326822916667,
      "learning_rate": 0.0001,
      "loss": 7.3246,
      "loss/crossentropy": 2.179289937019348,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5111505687236786,
      "step": 238
    },
    {
      "epoch": 0.0048,
      "grad_norm": 6.34375,
      "grad_norm_var": 0.5123697916666666,
      "learning_rate": 0.0001,
      "loss": 7.6064,
      "loss/crossentropy": 2.2424627542495728,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5187103897333145,
      "step": 240
    },
    {
      "epoch": 0.00484,
      "grad_norm": 5.96875,
      "grad_norm_var": 0.2986979166666667,
      "learning_rate": 0.0001,
      "loss": 7.8108,
      "loss/crossentropy": 2.8024520874023438,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5700482130050659,
      "step": 242
    },
    {
      "epoch": 0.00488,
      "grad_norm": 6.25,
      "grad_norm_var": 0.4554524739583333,
      "learning_rate": 0.0001,
      "loss": 7.6644,
      "loss/crossentropy": 2.3653491735458374,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5232449471950531,
      "step": 244
    },
    {
      "epoch": 0.00492,
      "grad_norm": 6.90625,
      "grad_norm_var": 0.48513997395833336,
      "learning_rate": 0.0001,
      "loss": 7.5028,
      "loss/crossentropy": 2.6201778650283813,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5684538185596466,
      "step": 246
    },
    {
      "epoch": 0.00496,
      "grad_norm": 7.3125,
      "grad_norm_var": 0.49654541015625,
      "learning_rate": 0.0001,
      "loss": 7.6631,
      "loss/crossentropy": 2.2811471819877625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5543638169765472,
      "step": 248
    },
    {
      "epoch": 0.005,
      "grad_norm": 5.625,
      "grad_norm_var": 0.48544514973958336,
      "learning_rate": 0.0001,
      "loss": 7.6914,
      "loss/crossentropy": 2.4381459951400757,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5594777166843414,
      "step": 250
    },
    {
      "epoch": 0.00504,
      "grad_norm": 9.375,
      "grad_norm_var": 1.04000244140625,
      "learning_rate": 0.0001,
      "loss": 7.7346,
      "loss/crossentropy": 2.435782313346863,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5159732699394226,
      "step": 252
    },
    {
      "epoch": 0.00508,
      "grad_norm": 5.46875,
      "grad_norm_var": 1.0892862955729166,
      "learning_rate": 0.0001,
      "loss": 8.0282,
      "loss/crossentropy": 2.7867215871810913,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.551640123128891,
      "step": 254
    },
    {
      "epoch": 0.00512,
      "grad_norm": 5.46875,
      "grad_norm_var": 1.2432902018229166,
      "learning_rate": 0.0001,
      "loss": 7.0744,
      "loss/crossentropy": 1.9328945875167847,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.45153285562992096,
      "step": 256
    },
    {
      "epoch": 0.00516,
      "grad_norm": 6.21875,
      "grad_norm_var": 1.2460286458333334,
      "learning_rate": 0.0001,
      "loss": 7.292,
      "loss/crossentropy": 2.552613139152527,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5215992629528046,
      "step": 258
    },
    {
      "epoch": 0.0052,
      "grad_norm": 5.40625,
      "grad_norm_var": 1.1848958333333333,
      "learning_rate": 0.0001,
      "loss": 7.4992,
      "loss/crossentropy": 2.3720492124557495,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5024219453334808,
      "step": 260
    },
    {
      "epoch": 0.00524,
      "grad_norm": 8.1875,
      "grad_norm_var": 1.4898274739583333,
      "learning_rate": 0.0001,
      "loss": 7.4676,
      "loss/crossentropy": 2.465815782546997,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5383751839399338,
      "step": 262
    },
    {
      "epoch": 0.00528,
      "grad_norm": 6.0625,
      "grad_norm_var": 1.5113240559895833,
      "learning_rate": 0.0001,
      "loss": 7.3163,
      "loss/crossentropy": 2.2791935205459595,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.534159854054451,
      "step": 264
    },
    {
      "epoch": 0.00532,
      "grad_norm": 6.28125,
      "grad_norm_var": 1.3855305989583333,
      "learning_rate": 0.0001,
      "loss": 7.9279,
      "loss/crossentropy": 2.48906409740448,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5344790518283844,
      "step": 266
    },
    {
      "epoch": 0.00536,
      "grad_norm": 4.90625,
      "grad_norm_var": 1.0180826822916667,
      "learning_rate": 0.0001,
      "loss": 7.3178,
      "loss/crossentropy": 2.0858306884765625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.47646892070770264,
      "step": 268
    },
    {
      "epoch": 0.0054,
      "grad_norm": 8.375,
      "grad_norm_var": 1.18258056640625,
      "learning_rate": 0.0001,
      "loss": 7.383,
      "loss/crossentropy": 2.159322738647461,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5543113648891449,
      "step": 270
    },
    {
      "epoch": 0.00544,
      "grad_norm": 4.9375,
      "grad_norm_var": 1.204931640625,
      "learning_rate": 0.0001,
      "loss": 7.1635,
      "loss/crossentropy": 2.249913454055786,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4896702766418457,
      "step": 272
    },
    {
      "epoch": 0.00548,
      "grad_norm": 8.125,
      "grad_norm_var": 1.388916015625,
      "learning_rate": 0.0001,
      "loss": 7.3565,
      "loss/crossentropy": 1.998712420463562,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4770403504371643,
      "step": 274
    },
    {
      "epoch": 0.00552,
      "grad_norm": 5.53125,
      "grad_norm_var": 1.9266764322916667,
      "learning_rate": 0.0001,
      "loss": 7.6522,
      "loss/crossentropy": 2.391260862350464,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5259605348110199,
      "step": 276
    },
    {
      "epoch": 0.00556,
      "grad_norm": 6.1875,
      "grad_norm_var": 1.6884765625,
      "learning_rate": 0.0001,
      "loss": 7.2935,
      "loss/crossentropy": 2.523361325263977,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4724537879228592,
      "step": 278
    },
    {
      "epoch": 0.0056,
      "grad_norm": 5.34375,
      "grad_norm_var": 1.7386555989583334,
      "learning_rate": 0.0001,
      "loss": 7.3505,
      "loss/crossentropy": 2.281963586807251,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.49724647402763367,
      "step": 280
    },
    {
      "epoch": 0.00564,
      "grad_norm": 5.0625,
      "grad_norm_var": 1.8446451822916667,
      "learning_rate": 0.0001,
      "loss": 7.1079,
      "loss/crossentropy": 2.2403814792633057,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.45384156703948975,
      "step": 282
    },
    {
      "epoch": 0.00568,
      "grad_norm": 5.6875,
      "grad_norm_var": 1.670166015625,
      "learning_rate": 0.0001,
      "loss": 7.4318,
      "loss/crossentropy": 2.2687963247299194,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.46648281812667847,
      "step": 284
    },
    {
      "epoch": 0.00572,
      "grad_norm": 5.75,
      "grad_norm_var": 1.38541259765625,
      "learning_rate": 0.0001,
      "loss": 7.3035,
      "loss/crossentropy": 2.3336217403411865,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4792183041572571,
      "step": 286
    },
    {
      "epoch": 0.00576,
      "grad_norm": 6.4375,
      "grad_norm_var": 1.292431640625,
      "learning_rate": 0.0001,
      "loss": 7.0031,
      "loss/crossentropy": 2.4006571769714355,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.473391056060791,
      "step": 288
    },
    {
      "epoch": 0.0058,
      "grad_norm": 5.09375,
      "grad_norm_var": 1.18671875,
      "learning_rate": 0.0001,
      "loss": 6.8037,
      "loss/crossentropy": 2.0306124687194824,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4722501188516617,
      "step": 290
    },
    {
      "epoch": 0.00584,
      "grad_norm": 5.34375,
      "grad_norm_var": 0.38865559895833335,
      "learning_rate": 0.0001,
      "loss": 7.1378,
      "loss/crossentropy": 2.412277102470398,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.48340730369091034,
      "step": 292
    },
    {
      "epoch": 0.00588,
      "grad_norm": 6.21875,
      "grad_norm_var": 0.3952433268229167,
      "learning_rate": 0.0001,
      "loss": 7.3589,
      "loss/crossentropy": 2.3195769786834717,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5074818134307861,
      "step": 294
    },
    {
      "epoch": 0.00592,
      "grad_norm": 6.71875,
      "grad_norm_var": 0.2867024739583333,
      "learning_rate": 0.0001,
      "loss": 7.2826,
      "loss/crossentropy": 2.3265275955200195,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5351093411445618,
      "step": 296
    },
    {
      "epoch": 0.00596,
      "grad_norm": 5.625,
      "grad_norm_var": 0.26171468098958334,
      "learning_rate": 0.0001,
      "loss": 7.0106,
      "loss/crossentropy": 2.210574746131897,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.45469868183135986,
      "step": 298
    },
    {
      "epoch": 0.006,
      "grad_norm": 6.3125,
      "grad_norm_var": 0.3034138997395833,
      "learning_rate": 0.0001,
      "loss": 7.4741,
      "loss/crossentropy": 2.347964644432068,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.45269395411014557,
      "step": 300
    },
    {
      "epoch": 0.00604,
      "grad_norm": 5.0,
      "grad_norm_var": 0.34308268229166666,
      "learning_rate": 0.0001,
      "loss": 6.8901,
      "loss/crossentropy": 2.197494626045227,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.45010584592819214,
      "step": 302
    },
    {
      "epoch": 0.00608,
      "grad_norm": 5.8125,
      "grad_norm_var": 0.32224934895833335,
      "learning_rate": 0.0001,
      "loss": 6.909,
      "loss/crossentropy": 2.3295196890830994,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.48989084362983704,
      "step": 304
    },
    {
      "epoch": 0.00612,
      "grad_norm": 6.3125,
      "grad_norm_var": 0.26347249348958335,
      "learning_rate": 0.0001,
      "loss": 7.6606,
      "loss/crossentropy": 2.5208678245544434,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.46991507709026337,
      "step": 306
    },
    {
      "epoch": 0.00616,
      "grad_norm": 5.40625,
      "grad_norm_var": 0.22849934895833332,
      "learning_rate": 0.0001,
      "loss": 7.2881,
      "loss/crossentropy": 2.6091307401657104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5090319812297821,
      "step": 308
    },
    {
      "epoch": 0.0062,
      "grad_norm": 5.4375,
      "grad_norm_var": 0.3083984375,
      "learning_rate": 0.0001,
      "loss": 7.0888,
      "loss/crossentropy": 2.4142966270446777,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.45959460735321045,
      "step": 310
    },
    {
      "epoch": 0.00624,
      "grad_norm": 6.0625,
      "grad_norm_var": 0.25006103515625,
      "learning_rate": 0.0001,
      "loss": 7.3054,
      "loss/crossentropy": 2.3062673807144165,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4381408095359802,
      "step": 312
    },
    {
      "epoch": 0.00628,
      "grad_norm": 4.875,
      "grad_norm_var": 0.29498697916666666,
      "learning_rate": 0.0001,
      "loss": 6.5202,
      "loss/crossentropy": 2.1124885082244873,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4392775595188141,
      "step": 314
    },
    {
      "epoch": 0.00632,
      "grad_norm": 5.09375,
      "grad_norm_var": 0.3001302083333333,
      "learning_rate": 0.0001,
      "loss": 6.3297,
      "loss/crossentropy": 2.0250568985939026,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4133095294237137,
      "step": 316
    },
    {
      "epoch": 0.00636,
      "grad_norm": 5.625,
      "grad_norm_var": 0.31021728515625,
      "learning_rate": 0.0001,
      "loss": 6.9903,
      "loss/crossentropy": 2.4011316299438477,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.46330246329307556,
      "step": 318
    },
    {
      "epoch": 0.0064,
      "grad_norm": 5.65625,
      "grad_norm_var": 0.30305582682291665,
      "learning_rate": 0.0001,
      "loss": 7.2114,
      "loss/crossentropy": 2.487559676170349,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.47237157821655273,
      "step": 320
    },
    {
      "epoch": 0.00644,
      "grad_norm": 5.1875,
      "grad_norm_var": 0.2775349934895833,
      "learning_rate": 0.0001,
      "loss": 6.5935,
      "loss/crossentropy": 1.999566912651062,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4165455400943756,
      "step": 322
    },
    {
      "epoch": 0.00648,
      "grad_norm": 6.03125,
      "grad_norm_var": 0.27496337890625,
      "learning_rate": 0.0001,
      "loss": 7.0573,
      "loss/crossentropy": 2.545841693878174,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4725654572248459,
      "step": 324
    },
    {
      "epoch": 0.00652,
      "grad_norm": 5.0625,
      "grad_norm_var": 0.2528483072916667,
      "learning_rate": 0.0001,
      "loss": 7.2351,
      "loss/crossentropy": 2.119086444377899,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4649319499731064,
      "step": 326
    },
    {
      "epoch": 0.00656,
      "grad_norm": 5.21875,
      "grad_norm_var": 0.221728515625,
      "learning_rate": 0.0001,
      "loss": 6.8367,
      "loss/crossentropy": 2.365525245666504,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5121739506721497,
      "step": 328
    },
    {
      "epoch": 0.0066,
      "grad_norm": 5.25,
      "grad_norm_var": 0.17467041015625,
      "learning_rate": 0.0001,
      "loss": 6.8384,
      "loss/crossentropy": 2.2604740858078003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4704676419496536,
      "step": 330
    },
    {
      "epoch": 0.00664,
      "grad_norm": 6.15625,
      "grad_norm_var": 0.18977864583333334,
      "learning_rate": 0.0001,
      "loss": 7.5125,
      "loss/crossentropy": 2.4891955852508545,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5196183770895004,
      "step": 332
    },
    {
      "epoch": 0.00668,
      "grad_norm": 5.46875,
      "grad_norm_var": 0.1767578125,
      "learning_rate": 0.0001,
      "loss": 7.3139,
      "loss/crossentropy": 2.430082321166992,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.47671228647232056,
      "step": 334
    },
    {
      "epoch": 0.00672,
      "grad_norm": 5.53125,
      "grad_norm_var": 0.187353515625,
      "learning_rate": 0.0001,
      "loss": 6.6969,
      "loss/crossentropy": 2.2450510263442993,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5124029517173767,
      "step": 336
    },
    {
      "epoch": 0.00676,
      "grad_norm": 5.375,
      "grad_norm_var": 0.18251546223958334,
      "learning_rate": 0.0001,
      "loss": 6.8537,
      "loss/crossentropy": 2.225212812423706,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4358299970626831,
      "step": 338
    },
    {
      "epoch": 0.0068,
      "grad_norm": 6.71875,
      "grad_norm_var": 26.47734375,
      "learning_rate": 0.0001,
      "loss": 6.8775,
      "loss/crossentropy": 2.320846140384674,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.42757023870944977,
      "step": 340
    },
    {
      "epoch": 0.00684,
      "grad_norm": 4.84375,
      "grad_norm_var": 26.739453125,
      "learning_rate": 0.0001,
      "loss": 6.7394,
      "loss/crossentropy": 2.419093132019043,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4633703678846359,
      "step": 342
    },
    {
      "epoch": 0.00688,
      "grad_norm": 5.3125,
      "grad_norm_var": 26.632405598958332,
      "learning_rate": 0.0001,
      "loss": 6.7304,
      "loss/crossentropy": 1.939517080783844,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4387335330247879,
      "step": 344
    },
    {
      "epoch": 0.00692,
      "grad_norm": 5.75,
      "grad_norm_var": 26.504410807291666,
      "learning_rate": 0.0001,
      "loss": 7.0914,
      "loss/crossentropy": 2.695888638496399,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.48031529784202576,
      "step": 346
    },
    {
      "epoch": 0.00696,
      "grad_norm": 6.625,
      "grad_norm_var": 26.468094889322916,
      "learning_rate": 0.0001,
      "loss": 6.8381,
      "loss/crossentropy": 2.245330333709717,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5032568573951721,
      "step": 348
    },
    {
      "epoch": 0.007,
      "grad_norm": 4.28125,
      "grad_norm_var": 26.707421875,
      "learning_rate": 0.0001,
      "loss": 6.4774,
      "loss/crossentropy": 1.9668607115745544,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4390462785959244,
      "step": 350
    },
    {
      "epoch": 0.00704,
      "grad_norm": 5.25,
      "grad_norm_var": 26.92008056640625,
      "learning_rate": 0.0001,
      "loss": 6.7795,
      "loss/crossentropy": 2.4035123586654663,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.43734824657440186,
      "step": 352
    },
    {
      "epoch": 0.00708,
      "grad_norm": 6.5,
      "grad_norm_var": 26.851460774739582,
      "learning_rate": 0.0001,
      "loss": 7.3932,
      "loss/crossentropy": 2.4636529684066772,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5208825469017029,
      "step": 354
    },
    {
      "epoch": 0.00712,
      "grad_norm": 6.4375,
      "grad_norm_var": 0.5743123372395833,
      "learning_rate": 0.0001,
      "loss": 6.8482,
      "loss/crossentropy": 2.085066556930542,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3971068561077118,
      "step": 356
    },
    {
      "epoch": 0.00716,
      "grad_norm": 4.59375,
      "grad_norm_var": 0.57847900390625,
      "learning_rate": 0.0001,
      "loss": 6.8962,
      "loss/crossentropy": 2.194266200065613,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.41747787594795227,
      "step": 358
    },
    {
      "epoch": 0.0072,
      "grad_norm": 4.59375,
      "grad_norm_var": 0.649462890625,
      "learning_rate": 0.0001,
      "loss": 6.7381,
      "loss/crossentropy": 2.4678618907928467,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4633233994245529,
      "step": 360
    },
    {
      "epoch": 0.00724,
      "grad_norm": 5.59375,
      "grad_norm_var": 0.6476847330729166,
      "learning_rate": 0.0001,
      "loss": 6.5748,
      "loss/crossentropy": 2.362962484359741,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4360974431037903,
      "step": 362
    },
    {
      "epoch": 0.00728,
      "grad_norm": 5.6875,
      "grad_norm_var": 0.5360514322916666,
      "learning_rate": 0.0001,
      "loss": 7.3497,
      "loss/crossentropy": 2.3162096738815308,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4641396403312683,
      "step": 364
    },
    {
      "epoch": 0.00732,
      "grad_norm": 6.0625,
      "grad_norm_var": 0.4325358072916667,
      "learning_rate": 0.0001,
      "loss": 7.0856,
      "loss/crossentropy": 2.279396176338196,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4331911951303482,
      "step": 366
    },
    {
      "epoch": 0.00736,
      "grad_norm": 4.96875,
      "grad_norm_var": 0.377587890625,
      "learning_rate": 0.0001,
      "loss": 6.8288,
      "loss/crossentropy": 2.333961606025696,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.43770918250083923,
      "step": 368
    },
    {
      "epoch": 0.0074,
      "grad_norm": 5.1875,
      "grad_norm_var": 0.27955322265625,
      "learning_rate": 0.0001,
      "loss": 6.9245,
      "loss/crossentropy": 2.130259871482849,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4134685546159744,
      "step": 370
    },
    {
      "epoch": 0.00744,
      "grad_norm": 5.59375,
      "grad_norm_var": 0.19478759765625,
      "learning_rate": 0.0001,
      "loss": 6.4884,
      "loss/crossentropy": 2.3000282049179077,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.46207693219184875,
      "step": 372
    },
    {
      "epoch": 0.00748,
      "grad_norm": 5.59375,
      "grad_norm_var": 0.20399983723958334,
      "learning_rate": 0.0001,
      "loss": 7.3714,
      "loss/crossentropy": 2.687412142753601,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.46891947090625763,
      "step": 374
    },
    {
      "epoch": 0.00752,
      "grad_norm": 4.28125,
      "grad_norm_var": 0.21962483723958334,
      "learning_rate": 0.0001,
      "loss": 6.4194,
      "loss/crossentropy": 2.2366563081741333,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.42876073718070984,
      "step": 376
    },
    {
      "epoch": 0.00756,
      "grad_norm": 5.71875,
      "grad_norm_var": 0.23108317057291666,
      "learning_rate": 0.0001,
      "loss": 7.0141,
      "loss/crossentropy": 2.5960274934768677,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4519210159778595,
      "step": 378
    },
    {
      "epoch": 0.0076,
      "grad_norm": 5.71875,
      "grad_norm_var": 0.23435872395833332,
      "learning_rate": 0.0001,
      "loss": 6.9654,
      "loss/crossentropy": 2.4690704345703125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5009289383888245,
      "step": 380
    },
    {
      "epoch": 0.00764,
      "grad_norm": 5.25,
      "grad_norm_var": 0.17952067057291668,
      "learning_rate": 0.0001,
      "loss": 6.6068,
      "loss/crossentropy": 2.188890814781189,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4912077784538269,
      "step": 382
    },
    {
      "epoch": 0.00768,
      "grad_norm": 5.15625,
      "grad_norm_var": 0.19055582682291666,
      "learning_rate": 0.0001,
      "loss": 6.5789,
      "loss/crossentropy": 2.2374125719070435,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4437579810619354,
      "step": 384
    },
    {
      "epoch": 0.00772,
      "grad_norm": 5.1875,
      "grad_norm_var": 0.19308268229166667,
      "learning_rate": 0.0001,
      "loss": 6.8081,
      "loss/crossentropy": 2.101546287536621,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.39717453718185425,
      "step": 386
    },
    {
      "epoch": 0.00776,
      "grad_norm": 5.1875,
      "grad_norm_var": 0.182666015625,
      "learning_rate": 0.0001,
      "loss": 6.5688,
      "loss/crossentropy": 2.2907408475875854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4657403528690338,
      "step": 388
    },
    {
      "epoch": 0.0078,
      "grad_norm": 4.53125,
      "grad_norm_var": 0.18203125,
      "learning_rate": 0.0001,
      "loss": 6.4664,
      "loss/crossentropy": 1.9909976720809937,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.39810416102409363,
      "step": 390
    },
    {
      "epoch": 0.00784,
      "grad_norm": 6.28125,
      "grad_norm_var": 0.23746337890625,
      "learning_rate": 0.0001,
      "loss": 6.6015,
      "loss/crossentropy": 2.109456777572632,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4226441979408264,
      "step": 392
    },
    {
      "epoch": 0.00788,
      "grad_norm": 6.03125,
      "grad_norm_var": 0.26549072265625,
      "learning_rate": 0.0001,
      "loss": 7.2592,
      "loss/crossentropy": 2.330615997314453,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5082881152629852,
      "step": 394
    },
    {
      "epoch": 0.00792,
      "grad_norm": 5.90625,
      "grad_norm_var": 0.49332275390625,
      "learning_rate": 0.0001,
      "loss": 7.301,
      "loss/crossentropy": 2.37632155418396,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5454596877098083,
      "step": 396
    },
    {
      "epoch": 0.00796,
      "grad_norm": 4.9375,
      "grad_norm_var": 0.5669921875,
      "learning_rate": 0.0001,
      "loss": 6.6057,
      "loss/crossentropy": 2.0303866863250732,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3780263066291809,
      "step": 398
    },
    {
      "epoch": 0.008,
      "grad_norm": 4.1875,
      "grad_norm_var": 0.6671712239583333,
      "learning_rate": 0.0001,
      "loss": 6.8129,
      "loss/crossentropy": 2.077945590019226,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4111042767763138,
      "step": 400
    },
    {
      "epoch": 0.00804,
      "grad_norm": 5.5625,
      "grad_norm_var": 0.6867146809895833,
      "learning_rate": 0.0001,
      "loss": 6.919,
      "loss/crossentropy": 2.3042391538619995,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.44224822521209717,
      "step": 402
    },
    {
      "epoch": 0.00808,
      "grad_norm": 4.3125,
      "grad_norm_var": 0.74537353515625,
      "learning_rate": 0.0001,
      "loss": 6.4785,
      "loss/crossentropy": 2.15978467464447,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.44845885038375854,
      "step": 404
    },
    {
      "epoch": 0.00812,
      "grad_norm": 4.90625,
      "grad_norm_var": 0.69576416015625,
      "learning_rate": 0.0001,
      "loss": 6.3875,
      "loss/crossentropy": 2.4571259021759033,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4629521369934082,
      "step": 406
    },
    {
      "epoch": 0.00816,
      "grad_norm": 5.40625,
      "grad_norm_var": 0.6654947916666667,
      "learning_rate": 0.0001,
      "loss": 7.1165,
      "loss/crossentropy": 2.653234601020813,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5556869208812714,
      "step": 408
    },
    {
      "epoch": 0.0082,
      "grad_norm": 4.53125,
      "grad_norm_var": 0.66138916015625,
      "learning_rate": 0.0001,
      "loss": 6.6643,
      "loss/crossentropy": 1.9738067388534546,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.42570993304252625,
      "step": 410
    },
    {
      "epoch": 0.00824,
      "grad_norm": 4.0625,
      "grad_norm_var": 0.46920166015625,
      "learning_rate": 0.0001,
      "loss": 6.2205,
      "loss/crossentropy": 2.093988060951233,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.40544557571411133,
      "step": 412
    },
    {
      "epoch": 0.00828,
      "grad_norm": 4.34375,
      "grad_norm_var": 0.36213785807291665,
      "learning_rate": 0.0001,
      "loss": 6.6356,
      "loss/crossentropy": 2.4798851013183594,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4204765260219574,
      "step": 414
    },
    {
      "epoch": 0.00832,
      "grad_norm": 5.625,
      "grad_norm_var": 0.3986287434895833,
      "learning_rate": 0.0001,
      "loss": 6.5601,
      "loss/crossentropy": 2.4342020750045776,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5148662775754929,
      "step": 416
    },
    {
      "epoch": 0.00836,
      "grad_norm": 5.40625,
      "grad_norm_var": 0.3985514322916667,
      "learning_rate": 0.0001,
      "loss": 6.7757,
      "loss/crossentropy": 2.3637804985046387,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.45475171506404877,
      "step": 418
    },
    {
      "epoch": 0.0084,
      "grad_norm": 4.09375,
      "grad_norm_var": 0.39586181640625,
      "learning_rate": 0.0001,
      "loss": 6.6923,
      "loss/crossentropy": 2.4066261053085327,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.45181581377983093,
      "step": 420
    },
    {
      "epoch": 0.00844,
      "grad_norm": 4.0,
      "grad_norm_var": 0.43176676432291666,
      "learning_rate": 0.0001,
      "loss": 6.2428,
      "loss/crossentropy": 2.1273797750473022,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3976929485797882,
      "step": 422
    },
    {
      "epoch": 0.00848,
      "grad_norm": 4.90625,
      "grad_norm_var": 0.26330973307291666,
      "learning_rate": 0.0001,
      "loss": 6.6524,
      "loss/crossentropy": 2.4227113723754883,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.46154454350471497,
      "step": 424
    },
    {
      "epoch": 0.00852,
      "grad_norm": 4.65625,
      "grad_norm_var": 0.261181640625,
      "learning_rate": 0.0001,
      "loss": 6.6558,
      "loss/crossentropy": 2.3502479791641235,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.43580910563468933,
      "step": 426
    },
    {
      "epoch": 0.00856,
      "grad_norm": 5.3125,
      "grad_norm_var": 0.26236572265625,
      "learning_rate": 0.0001,
      "loss": 6.903,
      "loss/crossentropy": 2.5034282207489014,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4730219095945358,
      "step": 428
    },
    {
      "epoch": 0.0086,
      "grad_norm": 4.40625,
      "grad_norm_var": 0.2557576497395833,
      "learning_rate": 0.0001,
      "loss": 6.2148,
      "loss/crossentropy": 2.0902098417282104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.409458264708519,
      "step": 430
    },
    {
      "epoch": 0.00864,
      "grad_norm": 4.9375,
      "grad_norm_var": 0.19420166015625,
      "learning_rate": 0.0001,
      "loss": 6.4634,
      "loss/crossentropy": 2.2203429341316223,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.41066519916057587,
      "step": 432
    },
    {
      "epoch": 0.00868,
      "grad_norm": 5.03125,
      "grad_norm_var": 0.14855143229166667,
      "learning_rate": 0.0001,
      "loss": 6.6943,
      "loss/crossentropy": 2.568304419517517,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.48846572637557983,
      "step": 434
    },
    {
      "epoch": 0.00872,
      "grad_norm": 5.15625,
      "grad_norm_var": 0.13489176432291666,
      "learning_rate": 0.0001,
      "loss": 6.4829,
      "loss/crossentropy": 2.359646439552307,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.43610572814941406,
      "step": 436
    },
    {
      "epoch": 0.00876,
      "grad_norm": 5.03125,
      "grad_norm_var": 0.08870035807291667,
      "learning_rate": 0.0001,
      "loss": 6.6119,
      "loss/crossentropy": 2.2751121520996094,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.45837563276290894,
      "step": 438
    },
    {
      "epoch": 0.0088,
      "grad_norm": 4.125,
      "grad_norm_var": 0.13665364583333334,
      "learning_rate": 0.0001,
      "loss": 6.5338,
      "loss/crossentropy": 2.334506392478943,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4196038395166397,
      "step": 440
    },
    {
      "epoch": 0.00884,
      "grad_norm": 4.1875,
      "grad_norm_var": 0.15891927083333332,
      "learning_rate": 0.0001,
      "loss": 6.2206,
      "loss/crossentropy": 1.9731069803237915,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.41661541163921356,
      "step": 442
    },
    {
      "epoch": 0.00888,
      "grad_norm": 5.03125,
      "grad_norm_var": 0.13339436848958333,
      "learning_rate": 0.0001,
      "loss": 6.3377,
      "loss/crossentropy": 2.319058418273926,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4656156301498413,
      "step": 444
    },
    {
      "epoch": 0.00892,
      "grad_norm": 4.125,
      "grad_norm_var": 0.15013020833333332,
      "learning_rate": 0.0001,
      "loss": 6.5345,
      "loss/crossentropy": 2.309122085571289,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.41400712728500366,
      "step": 446
    },
    {
      "epoch": 0.00896,
      "grad_norm": 4.53125,
      "grad_norm_var": 0.14468994140625,
      "learning_rate": 0.0001,
      "loss": 6.5385,
      "loss/crossentropy": 1.867617905139923,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.39080700278282166,
      "step": 448
    },
    {
      "epoch": 0.009,
      "grad_norm": 5.21875,
      "grad_norm_var": 0.79830322265625,
      "learning_rate": 0.0001,
      "loss": 6.799,
      "loss/crossentropy": 2.205033838748932,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4887084364891052,
      "step": 450
    },
    {
      "epoch": 0.00904,
      "grad_norm": 4.9375,
      "grad_norm_var": 0.826171875,
      "learning_rate": 0.0001,
      "loss": 6.6476,
      "loss/crossentropy": 2.3054174184799194,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4140937328338623,
      "step": 452
    },
    {
      "epoch": 0.00908,
      "grad_norm": 4.53125,
      "grad_norm_var": 0.8572224934895833,
      "learning_rate": 0.0001,
      "loss": 6.7036,
      "loss/crossentropy": 2.1358219981193542,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3849467635154724,
      "step": 454
    },
    {
      "epoch": 0.00912,
      "grad_norm": 4.9375,
      "grad_norm_var": 0.8132120768229166,
      "learning_rate": 0.0001,
      "loss": 6.4024,
      "loss/crossentropy": 1.9811997413635254,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.42263032495975494,
      "step": 456
    },
    {
      "epoch": 0.00916,
      "grad_norm": 4.8125,
      "grad_norm_var": 0.7787109375,
      "learning_rate": 0.0001,
      "loss": 6.8381,
      "loss/crossentropy": 2.319555103778839,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4539293050765991,
      "step": 458
    },
    {
      "epoch": 0.0092,
      "grad_norm": 4.6875,
      "grad_norm_var": 0.7744425455729167,
      "learning_rate": 0.0001,
      "loss": 6.72,
      "loss/crossentropy": 2.4030569791793823,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.41373930871486664,
      "step": 460
    },
    {
      "epoch": 0.00924,
      "grad_norm": 4.8125,
      "grad_norm_var": 0.7218587239583333,
      "learning_rate": 0.0001,
      "loss": 6.7376,
      "loss/crossentropy": 2.4479328393936157,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.447835311293602,
      "step": 462
    },
    {
      "epoch": 0.00928,
      "grad_norm": 5.5,
      "grad_norm_var": 0.7289021809895834,
      "learning_rate": 0.0001,
      "loss": 7.0562,
      "loss/crossentropy": 2.056324601173401,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4374549984931946,
      "step": 464
    },
    {
      "epoch": 0.00932,
      "grad_norm": 4.28125,
      "grad_norm_var": 0.19256184895833334,
      "learning_rate": 0.0001,
      "loss": 6.6994,
      "loss/crossentropy": 2.4104079008102417,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4148041307926178,
      "step": 466
    },
    {
      "epoch": 0.00936,
      "grad_norm": 4.75,
      "grad_norm_var": 0.13527018229166668,
      "learning_rate": 0.0001,
      "loss": 7.0758,
      "loss/crossentropy": 2.734652876853943,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.511719822883606,
      "step": 468
    },
    {
      "epoch": 0.0094,
      "grad_norm": 5.0,
      "grad_norm_var": 0.13118082682291668,
      "learning_rate": 0.0001,
      "loss": 6.4772,
      "loss/crossentropy": 2.1748571395874023,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3852091133594513,
      "step": 470
    },
    {
      "epoch": 0.00944,
      "grad_norm": 4.625,
      "grad_norm_var": 0.13912353515625,
      "learning_rate": 0.0001,
      "loss": 6.7127,
      "loss/crossentropy": 2.476449966430664,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4194856435060501,
      "step": 472
    },
    {
      "epoch": 0.00948,
      "grad_norm": 4.40625,
      "grad_norm_var": 0.16887613932291667,
      "learning_rate": 0.0001,
      "loss": 6.494,
      "loss/crossentropy": 2.6383973360061646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.44083887338638306,
      "step": 474
    },
    {
      "epoch": 0.00952,
      "grad_norm": 4.375,
      "grad_norm_var": 0.18435872395833333,
      "learning_rate": 0.0001,
      "loss": 6.3184,
      "loss/crossentropy": 2.3149259090423584,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3863615244626999,
      "step": 476
    },
    {
      "epoch": 0.00956,
      "grad_norm": 5.0625,
      "grad_norm_var": 0.18388264973958332,
      "learning_rate": 0.0001,
      "loss": 6.6217,
      "loss/crossentropy": 2.3096635341644287,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.40995509922504425,
      "step": 478
    },
    {
      "epoch": 0.0096,
      "grad_norm": 4.34375,
      "grad_norm_var": 0.10846354166666666,
      "learning_rate": 0.0001,
      "loss": 6.4849,
      "loss/crossentropy": 2.6495853662490845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.412080317735672,
      "step": 480
    },
    {
      "epoch": 0.00964,
      "grad_norm": 5.21875,
      "grad_norm_var": 0.13948160807291668,
      "learning_rate": 0.0001,
      "loss": 6.9293,
      "loss/crossentropy": 2.445754885673523,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.45622071623802185,
      "step": 482
    },
    {
      "epoch": 0.00968,
      "grad_norm": 4.40625,
      "grad_norm_var": 0.13661702473958334,
      "learning_rate": 0.0001,
      "loss": 6.4119,
      "loss/crossentropy": 2.418110966682434,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.40424875915050507,
      "step": 484
    },
    {
      "epoch": 0.00972,
      "grad_norm": 5.0625,
      "grad_norm_var": 0.15959879557291667,
      "learning_rate": 0.0001,
      "loss": 6.6356,
      "loss/crossentropy": 1.9564435482025146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.45834848284721375,
      "step": 486
    },
    {
      "epoch": 0.00976,
      "grad_norm": 5.375,
      "grad_norm_var": 0.16946207682291667,
      "learning_rate": 0.0001,
      "loss": 6.7056,
      "loss/crossentropy": 2.3772581815719604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4418800473213196,
      "step": 488
    },
    {
      "epoch": 0.0098,
      "grad_norm": 4.0625,
      "grad_norm_var": 0.208056640625,
      "learning_rate": 0.0001,
      "loss": 6.3239,
      "loss/crossentropy": 1.9526153802871704,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3661540001630783,
      "step": 490
    },
    {
      "epoch": 0.00984,
      "grad_norm": 4.875,
      "grad_norm_var": 0.19999593098958332,
      "learning_rate": 0.0001,
      "loss": 6.7642,
      "loss/crossentropy": 2.40561842918396,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.44370119273662567,
      "step": 492
    },
    {
      "epoch": 0.00988,
      "grad_norm": 4.53125,
      "grad_norm_var": 0.1943359375,
      "learning_rate": 0.0001,
      "loss": 6.6475,
      "loss/crossentropy": 2.4316108226776123,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4437306672334671,
      "step": 494
    },
    {
      "epoch": 0.00992,
      "grad_norm": 4.53125,
      "grad_norm_var": 0.25276285807291665,
      "learning_rate": 0.0001,
      "loss": 6.4095,
      "loss/crossentropy": 2.2919591665267944,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4170967787504196,
      "step": 496
    },
    {
      "epoch": 0.00996,
      "grad_norm": 5.03125,
      "grad_norm_var": 0.21116129557291666,
      "learning_rate": 0.0001,
      "loss": 6.6291,
      "loss/crossentropy": 2.571357250213623,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4073399156332016,
      "step": 498
    },
    {
      "epoch": 0.01,
      "grad_norm": 4.375,
      "grad_norm_var": 0.21330973307291667,
      "learning_rate": 0.0001,
      "loss": 6.2903,
      "loss/crossentropy": 2.389290928840637,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.41799379885196686,
      "step": 500
    },
    {
      "epoch": 0.01004,
      "grad_norm": 4.90625,
      "grad_norm_var": 0.20245768229166666,
      "learning_rate": 0.0001,
      "loss": 6.4319,
      "loss/crossentropy": 2.0904359221458435,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3914492577314377,
      "step": 502
    },
    {
      "epoch": 0.01008,
      "grad_norm": 3.625,
      "grad_norm_var": 0.218212890625,
      "learning_rate": 0.0001,
      "loss": 6.5581,
      "loss/crossentropy": 2.5435129404067993,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4660491645336151,
      "step": 504
    },
    {
      "epoch": 0.01012,
      "grad_norm": 4.75,
      "grad_norm_var": 0.21458333333333332,
      "learning_rate": 0.0001,
      "loss": 6.3665,
      "loss/crossentropy": 2.039083242416382,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.38188865780830383,
      "step": 506
    },
    {
      "epoch": 0.01016,
      "grad_norm": 5.21875,
      "grad_norm_var": 0.361181640625,
      "learning_rate": 0.0001,
      "loss": 6.4862,
      "loss/crossentropy": 2.056805729866028,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4257620573043823,
      "step": 508
    },
    {
      "epoch": 0.0102,
      "grad_norm": 4.03125,
      "grad_norm_var": 0.38592122395833334,
      "learning_rate": 0.0001,
      "loss": 6.4822,
      "loss/crossentropy": 2.6178410053253174,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.43898941576480865,
      "step": 510
    },
    {
      "epoch": 0.01024,
      "grad_norm": 4.59375,
      "grad_norm_var": 0.3405558268229167,
      "learning_rate": 0.0001,
      "loss": 6.4638,
      "loss/crossentropy": 2.232435703277588,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.37769296765327454,
      "step": 512
    },
    {
      "epoch": 0.01028,
      "grad_norm": 4.40625,
      "grad_norm_var": 0.33033447265625,
      "learning_rate": 0.0001,
      "loss": 6.378,
      "loss/crossentropy": 1.8679735660552979,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3805970698595047,
      "step": 514
    },
    {
      "epoch": 0.01032,
      "grad_norm": 4.3125,
      "grad_norm_var": 0.33175455729166664,
      "learning_rate": 0.0001,
      "loss": 6.7453,
      "loss/crossentropy": 2.6537472009658813,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4450981914997101,
      "step": 516
    },
    {
      "epoch": 0.01036,
      "grad_norm": 4.03125,
      "grad_norm_var": 0.34915262858072915,
      "learning_rate": 0.0001,
      "loss": 6.2745,
      "loss/crossentropy": 2.5199841260910034,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4748214781284332,
      "step": 518
    },
    {
      "epoch": 0.0104,
      "grad_norm": 5.5,
      "grad_norm_var": 0.3482004801432292,
      "learning_rate": 0.0001,
      "loss": 6.6212,
      "loss/crossentropy": 2.6603333950042725,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4144483357667923,
      "step": 520
    },
    {
      "epoch": 0.01044,
      "grad_norm": 4.28125,
      "grad_norm_var": 0.3241119384765625,
      "learning_rate": 0.0001,
      "loss": 6.4073,
      "loss/crossentropy": 2.284039616584778,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.42801250517368317,
      "step": 522
    },
    {
      "epoch": 0.01048,
      "grad_norm": 3.78125,
      "grad_norm_var": 0.1894683837890625,
      "learning_rate": 0.0001,
      "loss": 6.2153,
      "loss/crossentropy": 2.473629951477051,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4107852131128311,
      "step": 524
    },
    {
      "epoch": 0.01052,
      "grad_norm": 4.5,
      "grad_norm_var": 0.1827789306640625,
      "learning_rate": 0.0001,
      "loss": 6.6746,
      "loss/crossentropy": 2.1443774700164795,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3606857359409332,
      "step": 526
    },
    {
      "epoch": 0.01056,
      "grad_norm": 3.96875,
      "grad_norm_var": 0.18889058430989583,
      "learning_rate": 0.0001,
      "loss": 5.8493,
      "loss/crossentropy": 1.8425135016441345,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.37770508229732513,
      "step": 528
    },
    {
      "epoch": 0.0106,
      "grad_norm": 3.859375,
      "grad_norm_var": 0.179052734375,
      "learning_rate": 0.0001,
      "loss": 6.3319,
      "loss/crossentropy": 2.3705164194107056,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4354119151830673,
      "step": 530
    },
    {
      "epoch": 0.01064,
      "grad_norm": 4.53125,
      "grad_norm_var": 0.16344401041666667,
      "learning_rate": 0.0001,
      "loss": 6.4823,
      "loss/crossentropy": 2.1141316294670105,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34213581681251526,
      "step": 532
    },
    {
      "epoch": 0.01068,
      "grad_norm": 4.3125,
      "grad_norm_var": 0.15579325358072918,
      "learning_rate": 0.0001,
      "loss": 6.0454,
      "loss/crossentropy": 2.14878511428833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3756616413593292,
      "step": 534
    },
    {
      "epoch": 0.01072,
      "grad_norm": 4.34375,
      "grad_norm_var": 0.052155558268229166,
      "learning_rate": 0.0001,
      "loss": 6.4363,
      "loss/crossentropy": 2.2513046264648438,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.416190966963768,
      "step": 536
    },
    {
      "epoch": 0.01076,
      "grad_norm": 3.921875,
      "grad_norm_var": 0.049169921875,
      "learning_rate": 0.0001,
      "loss": 6.2674,
      "loss/crossentropy": 2.1337047815322876,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3839789927005768,
      "step": 538
    },
    {
      "epoch": 0.0108,
      "grad_norm": 5.8125,
      "grad_norm_var": 0.20488993326822916,
      "learning_rate": 0.0001,
      "loss": 6.0559,
      "loss/crossentropy": 2.2019962072372437,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3806414008140564,
      "step": 540
    },
    {
      "epoch": 0.01084,
      "grad_norm": 4.125,
      "grad_norm_var": 0.2133697509765625,
      "learning_rate": 0.0001,
      "loss": 5.8434,
      "loss/crossentropy": 2.113224983215332,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3797851800918579,
      "step": 542
    },
    {
      "epoch": 0.01088,
      "grad_norm": 4.0,
      "grad_norm_var": 0.21507059733072917,
      "learning_rate": 0.0001,
      "loss": 6.5131,
      "loss/crossentropy": 2.461037516593933,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.41689516603946686,
      "step": 544
    },
    {
      "epoch": 0.01092,
      "grad_norm": 4.625,
      "grad_norm_var": 0.24849853515625,
      "learning_rate": 0.0001,
      "loss": 6.4191,
      "loss/crossentropy": 2.277098298072815,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4108494818210602,
      "step": 546
    },
    {
      "epoch": 0.01096,
      "grad_norm": 4.34375,
      "grad_norm_var": 0.48818359375,
      "learning_rate": 0.0001,
      "loss": 6.3436,
      "loss/crossentropy": 2.007936477661133,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.36796192824840546,
      "step": 548
    },
    {
      "epoch": 0.011,
      "grad_norm": 4.625,
      "grad_norm_var": 0.5631795247395833,
      "learning_rate": 0.0001,
      "loss": 6.5793,
      "loss/crossentropy": 2.197320520877838,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.38722094893455505,
      "step": 550
    },
    {
      "epoch": 0.01104,
      "grad_norm": 4.6875,
      "grad_norm_var": 0.5419230143229167,
      "learning_rate": 0.0001,
      "loss": 6.3185,
      "loss/crossentropy": 2.225432515144348,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3928917348384857,
      "step": 552
    },
    {
      "epoch": 0.01108,
      "grad_norm": 4.28125,
      "grad_norm_var": 0.5455067952473959,
      "learning_rate": 0.0001,
      "loss": 5.9686,
      "loss/crossentropy": 2.5253326892852783,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.41547106206417084,
      "step": 554
    },
    {
      "epoch": 0.01112,
      "grad_norm": 4.15625,
      "grad_norm_var": 0.44996337890625,
      "learning_rate": 0.0001,
      "loss": 6.2158,
      "loss/crossentropy": 2.488635540008545,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4523312896490097,
      "step": 556
    },
    {
      "epoch": 0.01116,
      "grad_norm": 4.03125,
      "grad_norm_var": 0.46119384765625,
      "learning_rate": 0.0001,
      "loss": 6.3632,
      "loss/crossentropy": 2.395568609237671,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.40695953369140625,
      "step": 558
    },
    {
      "epoch": 0.0112,
      "grad_norm": 4.28125,
      "grad_norm_var": 0.4615234375,
      "learning_rate": 0.0001,
      "loss": 5.998,
      "loss/crossentropy": 2.383823275566101,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4482497274875641,
      "step": 560
    },
    {
      "epoch": 0.01124,
      "grad_norm": 4.46875,
      "grad_norm_var": 0.46280008951822915,
      "learning_rate": 0.0001,
      "loss": 5.968,
      "loss/crossentropy": 2.0261693000793457,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.36933301389217377,
      "step": 562
    },
    {
      "epoch": 0.01128,
      "grad_norm": 4.65625,
      "grad_norm_var": 0.2221588134765625,
      "learning_rate": 0.0001,
      "loss": 6.1467,
      "loss/crossentropy": 2.131769895553589,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.341948002576828,
      "step": 564
    },
    {
      "epoch": 0.01132,
      "grad_norm": 4.8125,
      "grad_norm_var": 0.0982574462890625,
      "learning_rate": 0.0001,
      "loss": 6.5288,
      "loss/crossentropy": 2.3899158239364624,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.417646586894989,
      "step": 566
    },
    {
      "epoch": 0.01136,
      "grad_norm": 4.21875,
      "grad_norm_var": 0.07683817545572917,
      "learning_rate": 0.0001,
      "loss": 6.6198,
      "loss/crossentropy": 2.3139528036117554,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.39010919630527496,
      "step": 568
    },
    {
      "epoch": 0.0114,
      "grad_norm": 4.3125,
      "grad_norm_var": 0.06782124837239584,
      "learning_rate": 0.0001,
      "loss": 6.1209,
      "loss/crossentropy": 1.966201364994049,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3554569333791733,
      "step": 570
    },
    {
      "epoch": 0.01144,
      "grad_norm": 4.21875,
      "grad_norm_var": 0.07773335774739583,
      "learning_rate": 0.0001,
      "loss": 6.1746,
      "loss/crossentropy": 2.2325466871261597,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4008040726184845,
      "step": 572
    },
    {
      "epoch": 0.01148,
      "grad_norm": 4.96875,
      "grad_norm_var": 0.09848531087239583,
      "learning_rate": 0.0001,
      "loss": 6.121,
      "loss/crossentropy": 1.7670194506645203,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3362845778465271,
      "step": 574
    },
    {
      "epoch": 0.01152,
      "grad_norm": 4.125,
      "grad_norm_var": 0.10274149576822916,
      "learning_rate": 0.0001,
      "loss": 6.3956,
      "loss/crossentropy": 2.332284092903137,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4190225303173065,
      "step": 576
    },
    {
      "epoch": 0.01156,
      "grad_norm": 4.71875,
      "grad_norm_var": 0.10549723307291667,
      "learning_rate": 0.0001,
      "loss": 6.4367,
      "loss/crossentropy": 2.265386700630188,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4109695851802826,
      "step": 578
    },
    {
      "epoch": 0.0116,
      "grad_norm": 5.15625,
      "grad_norm_var": 0.13811442057291667,
      "learning_rate": 0.0001,
      "loss": 6.4475,
      "loss/crossentropy": 2.265889286994934,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.42450472712516785,
      "step": 580
    },
    {
      "epoch": 0.01164,
      "grad_norm": 4.5625,
      "grad_norm_var": 0.163134765625,
      "learning_rate": 0.0001,
      "loss": 6.1483,
      "loss/crossentropy": 1.867847204208374,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3627365529537201,
      "step": 582
    },
    {
      "epoch": 0.01168,
      "grad_norm": 3.796875,
      "grad_norm_var": 0.23681538899739582,
      "learning_rate": 0.0001,
      "loss": 6.5909,
      "loss/crossentropy": 2.5827555656433105,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5194894820451736,
      "step": 584
    },
    {
      "epoch": 0.01172,
      "grad_norm": 4.5625,
      "grad_norm_var": 0.23188374837239584,
      "learning_rate": 0.0001,
      "loss": 6.6319,
      "loss/crossentropy": 2.26031893491745,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.412450835108757,
      "step": 586
    },
    {
      "epoch": 0.01176,
      "grad_norm": 5.53125,
      "grad_norm_var": 0.2718739827473958,
      "learning_rate": 0.0001,
      "loss": 6.5782,
      "loss/crossentropy": 2.1885104179382324,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.41636165976524353,
      "step": 588
    },
    {
      "epoch": 0.0118,
      "grad_norm": 4.53125,
      "grad_norm_var": 0.26240132649739584,
      "learning_rate": 0.0001,
      "loss": 6.5247,
      "loss/crossentropy": 2.682767391204834,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4279082715511322,
      "step": 590
    },
    {
      "epoch": 0.01184,
      "grad_norm": 4.3125,
      "grad_norm_var": 0.24807027180989583,
      "learning_rate": 0.0001,
      "loss": 6.5862,
      "loss/crossentropy": 2.185304641723633,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4266812950372696,
      "step": 592
    },
    {
      "epoch": 0.01188,
      "grad_norm": 4.15625,
      "grad_norm_var": 0.2589752197265625,
      "learning_rate": 0.0001,
      "loss": 6.2494,
      "loss/crossentropy": 2.383716344833374,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.38570962846279144,
      "step": 594
    },
    {
      "epoch": 0.01192,
      "grad_norm": 4.3125,
      "grad_norm_var": 0.2337066650390625,
      "learning_rate": 0.0001,
      "loss": 6.3066,
      "loss/crossentropy": 2.2963398694992065,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.43451687693595886,
      "step": 596
    },
    {
      "epoch": 0.01196,
      "grad_norm": 4.96875,
      "grad_norm_var": 0.2128570556640625,
      "learning_rate": 0.0001,
      "loss": 6.3368,
      "loss/crossentropy": 2.2728757858276367,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4280686676502228,
      "step": 598
    },
    {
      "epoch": 0.012,
      "grad_norm": 4.34375,
      "grad_norm_var": 0.141650390625,
      "learning_rate": 0.0001,
      "loss": 6.1147,
      "loss/crossentropy": 2.3486615419387817,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.37224848568439484,
      "step": 600
    },
    {
      "epoch": 0.01204,
      "grad_norm": 3.765625,
      "grad_norm_var": 0.17512613932291668,
      "learning_rate": 0.0001,
      "loss": 6.4017,
      "loss/crossentropy": 2.3265292644500732,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4079990088939667,
      "step": 602
    },
    {
      "epoch": 0.01208,
      "grad_norm": 3.84375,
      "grad_norm_var": 0.08847554524739583,
      "learning_rate": 0.0001,
      "loss": 6.0861,
      "loss/crossentropy": 2.45253849029541,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.42630523443222046,
      "step": 604
    },
    {
      "epoch": 0.01212,
      "grad_norm": 4.03125,
      "grad_norm_var": 0.08964742024739583,
      "learning_rate": 0.0001,
      "loss": 6.4101,
      "loss/crossentropy": 2.4417446851730347,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.40342913568019867,
      "step": 606
    },
    {
      "epoch": 0.01216,
      "grad_norm": 4.25,
      "grad_norm_var": 0.08886617024739583,
      "learning_rate": 0.0001,
      "loss": 6.2513,
      "loss/crossentropy": 2.1483529210090637,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3846609443426132,
      "step": 608
    },
    {
      "epoch": 0.0122,
      "grad_norm": 4.1875,
      "grad_norm_var": 0.09990132649739583,
      "learning_rate": 0.0001,
      "loss": 6.5479,
      "loss/crossentropy": 2.481536865234375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.440000057220459,
      "step": 610
    },
    {
      "epoch": 0.01224,
      "grad_norm": 4.15625,
      "grad_norm_var": 0.09464518229166667,
      "learning_rate": 0.0001,
      "loss": 6.4986,
      "loss/crossentropy": 2.4472655057907104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.41242682933807373,
      "step": 612
    },
    {
      "epoch": 0.01228,
      "grad_norm": 4.09375,
      "grad_norm_var": 0.07828369140625,
      "learning_rate": 0.0001,
      "loss": 6.4348,
      "loss/crossentropy": 2.3511135578155518,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.38262398540973663,
      "step": 614
    },
    {
      "epoch": 0.01232,
      "grad_norm": 3.71875,
      "grad_norm_var": 0.085791015625,
      "learning_rate": 0.0001,
      "loss": 6.4821,
      "loss/crossentropy": 2.5090683698654175,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.44584622979164124,
      "step": 616
    },
    {
      "epoch": 0.01236,
      "grad_norm": 4.75,
      "grad_norm_var": 0.1117340087890625,
      "learning_rate": 0.0001,
      "loss": 6.0395,
      "loss/crossentropy": 2.166012942790985,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.40271493792533875,
      "step": 618
    },
    {
      "epoch": 0.0124,
      "grad_norm": 5.09375,
      "grad_norm_var": 0.15511067708333334,
      "learning_rate": 0.0001,
      "loss": 6.4644,
      "loss/crossentropy": 2.583309531211853,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.43380285799503326,
      "step": 620
    },
    {
      "epoch": 0.01244,
      "grad_norm": 3.984375,
      "grad_norm_var": 0.19877827962239583,
      "learning_rate": 0.0001,
      "loss": 6.3289,
      "loss/crossentropy": 2.125720262527466,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.40389589965343475,
      "step": 622
    },
    {
      "epoch": 0.01248,
      "grad_norm": 6.34375,
      "grad_norm_var": 0.4984283447265625,
      "learning_rate": 0.0001,
      "loss": 5.9651,
      "loss/crossentropy": 1.7034094333648682,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31466029584407806,
      "step": 624
    },
    {
      "epoch": 0.01252,
      "grad_norm": 4.40625,
      "grad_norm_var": 0.4901041666666667,
      "learning_rate": 0.0001,
      "loss": 5.9451,
      "loss/crossentropy": 2.163281202316284,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.38622182607650757,
      "step": 626
    },
    {
      "epoch": 0.01256,
      "grad_norm": 3.875,
      "grad_norm_var": 0.4982818603515625,
      "learning_rate": 0.0001,
      "loss": 5.8931,
      "loss/crossentropy": 1.7754453420639038,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33256995677948,
      "step": 628
    },
    {
      "epoch": 0.0126,
      "grad_norm": 4.09375,
      "grad_norm_var": 0.4894683837890625,
      "learning_rate": 0.0001,
      "loss": 6.8606,
      "loss/crossentropy": 2.273309350013733,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5319447964429855,
      "step": 630
    },
    {
      "epoch": 0.01264,
      "grad_norm": 3.828125,
      "grad_norm_var": 0.480126953125,
      "learning_rate": 0.0001,
      "loss": 6.2103,
      "loss/crossentropy": 2.397401988506317,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4255771040916443,
      "step": 632
    },
    {
      "epoch": 0.01268,
      "grad_norm": 5.28125,
      "grad_norm_var": 0.521875,
      "learning_rate": 0.0001,
      "loss": 6.4179,
      "loss/crossentropy": 2.3898611068725586,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.46314406394958496,
      "step": 634
    },
    {
      "epoch": 0.01272,
      "grad_norm": 7.40625,
      "grad_norm_var": 1.0609212239583334,
      "learning_rate": 0.0001,
      "loss": 6.5634,
      "loss/crossentropy": 2.3740471601486206,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.48123428225517273,
      "step": 636
    },
    {
      "epoch": 0.01276,
      "grad_norm": 3.90625,
      "grad_norm_var": 1.0519846598307292,
      "learning_rate": 0.0001,
      "loss": 6.1136,
      "loss/crossentropy": 2.236217498779297,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.39564159512519836,
      "step": 638
    },
    {
      "epoch": 0.0128,
      "grad_norm": 4.21875,
      "grad_norm_var": 0.7972157796223959,
      "learning_rate": 0.0001,
      "loss": 6.3134,
      "loss/crossentropy": 2.4049174785614014,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.41312308609485626,
      "step": 640
    },
    {
      "epoch": 0.01284,
      "grad_norm": 4.0,
      "grad_norm_var": 0.80142822265625,
      "learning_rate": 0.0001,
      "loss": 6.2726,
      "loss/crossentropy": 2.173800766468048,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3983110189437866,
      "step": 642
    },
    {
      "epoch": 0.01288,
      "grad_norm": 4.65625,
      "grad_norm_var": 0.80084228515625,
      "learning_rate": 0.0001,
      "loss": 6.2779,
      "loss/crossentropy": 2.2124537229537964,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.38358184695243835,
      "step": 644
    },
    {
      "epoch": 0.01292,
      "grad_norm": 3.90625,
      "grad_norm_var": 0.8355377197265625,
      "learning_rate": 0.0001,
      "loss": 6.1209,
      "loss/crossentropy": 2.4939264059066772,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3805152475833893,
      "step": 646
    },
    {
      "epoch": 0.01296,
      "grad_norm": 4.25,
      "grad_norm_var": 0.8250935872395834,
      "learning_rate": 0.0001,
      "loss": 6.0714,
      "loss/crossentropy": 2.4526472091674805,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3958089202642441,
      "step": 648
    },
    {
      "epoch": 0.013,
      "grad_norm": 4.25,
      "grad_norm_var": 0.7562978108723958,
      "learning_rate": 0.0001,
      "loss": 6.3648,
      "loss/crossentropy": 2.5171029567718506,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.446771502494812,
      "step": 650
    },
    {
      "epoch": 0.01304,
      "grad_norm": 4.125,
      "grad_norm_var": 0.1185943603515625,
      "learning_rate": 0.0001,
      "loss": 6.1656,
      "loss/crossentropy": 2.270598888397217,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.38787929713726044,
      "step": 652
    },
    {
      "epoch": 0.01308,
      "grad_norm": 4.15625,
      "grad_norm_var": 0.11404520670572917,
      "learning_rate": 0.0001,
      "loss": 5.7739,
      "loss/crossentropy": 1.8847617506980896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3258303850889206,
      "step": 654
    },
    {
      "epoch": 0.01312,
      "grad_norm": 3.9375,
      "grad_norm_var": 0.11286519368489584,
      "learning_rate": 0.0001,
      "loss": 6.042,
      "loss/crossentropy": 2.2471452951431274,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.38874460756778717,
      "step": 656
    },
    {
      "epoch": 0.01316,
      "grad_norm": 4.125,
      "grad_norm_var": 0.10369364420572917,
      "learning_rate": 0.0001,
      "loss": 6.4383,
      "loss/crossentropy": 2.3776252269744873,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.40553848445415497,
      "step": 658
    },
    {
      "epoch": 0.0132,
      "grad_norm": 3.84375,
      "grad_norm_var": 0.06461588541666667,
      "learning_rate": 0.0001,
      "loss": 5.5389,
      "loss/crossentropy": 2.291012167930603,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3630271404981613,
      "step": 660
    },
    {
      "epoch": 0.01324,
      "grad_norm": 3.90625,
      "grad_norm_var": 0.06116536458333333,
      "learning_rate": 0.0001,
      "loss": 6.298,
      "loss/crossentropy": 2.2029112577438354,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.37700483202934265,
      "step": 662
    },
    {
      "epoch": 0.01328,
      "grad_norm": 3.984375,
      "grad_norm_var": 0.0598541259765625,
      "learning_rate": 0.0001,
      "loss": 6.4093,
      "loss/crossentropy": 2.571411967277527,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.5071892440319061,
      "step": 664
    },
    {
      "epoch": 0.01332,
      "grad_norm": 3.484375,
      "grad_norm_var": 0.045947265625,
      "learning_rate": 0.0001,
      "loss": 5.6839,
      "loss/crossentropy": 2.148792862892151,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3591457009315491,
      "step": 666
    },
    {
      "epoch": 0.01336,
      "grad_norm": 4.09375,
      "grad_norm_var": 0.04755452473958333,
      "learning_rate": 0.0001,
      "loss": 6.4444,
      "loss/crossentropy": 2.5091140270233154,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3882133811712265,
      "step": 668
    },
    {
      "epoch": 0.0134,
      "grad_norm": 4.09375,
      "grad_norm_var": 0.049332682291666666,
      "learning_rate": 0.0001,
      "loss": 6.15,
      "loss/crossentropy": 2.4669524431228638,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3941466957330704,
      "step": 670
    },
    {
      "epoch": 0.01344,
      "grad_norm": 3.9375,
      "grad_norm_var": 0.06620686848958333,
      "learning_rate": 0.0001,
      "loss": 6.5358,
      "loss/crossentropy": 2.4111422300338745,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3973373472690582,
      "step": 672
    },
    {
      "epoch": 0.01348,
      "grad_norm": 4.1875,
      "grad_norm_var": 0.06495768229166667,
      "learning_rate": 0.0001,
      "loss": 5.765,
      "loss/crossentropy": 2.1109864115715027,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.36189010739326477,
      "step": 674
    },
    {
      "epoch": 0.01352,
      "grad_norm": 3.578125,
      "grad_norm_var": 0.0744140625,
      "learning_rate": 0.0001,
      "loss": 6.0289,
      "loss/crossentropy": 2.069494664669037,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3562029302120209,
      "step": 676
    },
    {
      "epoch": 0.01356,
      "grad_norm": 4.0625,
      "grad_norm_var": 0.07224934895833333,
      "learning_rate": 0.0001,
      "loss": 6.4526,
      "loss/crossentropy": 2.1924527883529663,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.38948506116867065,
      "step": 678
    },
    {
      "epoch": 0.0136,
      "grad_norm": 3.6875,
      "grad_norm_var": 0.082421875,
      "learning_rate": 0.0001,
      "loss": 5.7311,
      "loss/crossentropy": 2.1603400707244873,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.40414859354496,
      "step": 680
    },
    {
      "epoch": 0.01364,
      "grad_norm": 3.71875,
      "grad_norm_var": 0.07177632649739583,
      "learning_rate": 0.0001,
      "loss": 6.2459,
      "loss/crossentropy": 2.515262722969055,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4363710880279541,
      "step": 682
    },
    {
      "epoch": 0.01368,
      "grad_norm": 4.09375,
      "grad_norm_var": 0.07869364420572916,
      "learning_rate": 0.0001,
      "loss": 6.1174,
      "loss/crossentropy": 2.3615161180496216,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.35220713913440704,
      "step": 684
    },
    {
      "epoch": 0.01372,
      "grad_norm": 6.03125,
      "grad_norm_var": 0.3293690999348958,
      "learning_rate": 0.0001,
      "loss": 6.1941,
      "loss/crossentropy": 1.920493245124817,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.35293935239315033,
      "step": 686
    },
    {
      "epoch": 0.01376,
      "grad_norm": 4.03125,
      "grad_norm_var": 0.31579488118489585,
      "learning_rate": 0.0001,
      "loss": 6.4064,
      "loss/crossentropy": 2.493665337562561,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4243907481431961,
      "step": 688
    },
    {
      "epoch": 0.0138,
      "grad_norm": 4.03125,
      "grad_norm_var": 0.3451324462890625,
      "learning_rate": 0.0001,
      "loss": 6.1519,
      "loss/crossentropy": 2.1182271242141724,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.39624081552028656,
      "step": 690
    },
    {
      "epoch": 0.01384,
      "grad_norm": 4.09375,
      "grad_norm_var": 0.32034098307291664,
      "learning_rate": 0.0001,
      "loss": 6.2457,
      "loss/crossentropy": 2.146227180957794,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3491668850183487,
      "step": 692
    },
    {
      "epoch": 0.01388,
      "grad_norm": 3.734375,
      "grad_norm_var": 0.3376261393229167,
      "learning_rate": 0.0001,
      "loss": 5.7331,
      "loss/crossentropy": 1.8458876609802246,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34608474373817444,
      "step": 694
    },
    {
      "epoch": 0.01392,
      "grad_norm": 4.03125,
      "grad_norm_var": 0.3294911702473958,
      "learning_rate": 0.0001,
      "loss": 6.3661,
      "loss/crossentropy": 2.270371675491333,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33937984704971313,
      "step": 696
    },
    {
      "epoch": 0.01396,
      "grad_norm": 3.875,
      "grad_norm_var": 0.32066141764322914,
      "learning_rate": 0.0001,
      "loss": 5.8264,
      "loss/crossentropy": 2.040702223777771,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3821101486682892,
      "step": 698
    },
    {
      "epoch": 0.014,
      "grad_norm": 4.09375,
      "grad_norm_var": 0.3197428385416667,
      "learning_rate": 0.0001,
      "loss": 6.1216,
      "loss/crossentropy": 2.1711822152137756,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.35383065044879913,
      "step": 700
    },
    {
      "epoch": 0.01404,
      "grad_norm": 3.890625,
      "grad_norm_var": 0.0597076416015625,
      "learning_rate": 0.0001,
      "loss": 5.8384,
      "loss/crossentropy": 2.3292651176452637,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.37580642104148865,
      "step": 702
    },
    {
      "epoch": 0.01408,
      "grad_norm": 4.34375,
      "grad_norm_var": 0.06575520833333333,
      "learning_rate": 0.0001,
      "loss": 6.0789,
      "loss/crossentropy": 2.243735432624817,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3997037708759308,
      "step": 704
    },
    {
      "epoch": 0.01412,
      "grad_norm": 3.921875,
      "grad_norm_var": 0.0524566650390625,
      "learning_rate": 0.0001,
      "loss": 5.9987,
      "loss/crossentropy": 1.9908145666122437,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33391132950782776,
      "step": 706
    },
    {
      "epoch": 0.01416,
      "grad_norm": 4.375,
      "grad_norm_var": 0.06243082682291667,
      "learning_rate": 0.0001,
      "loss": 5.536,
      "loss/crossentropy": 2.280096471309662,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3899015784263611,
      "step": 708
    },
    {
      "epoch": 0.0142,
      "grad_norm": 3.6875,
      "grad_norm_var": 0.12727762858072916,
      "learning_rate": 0.0001,
      "loss": 5.9373,
      "loss/crossentropy": 2.0714810490608215,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3797626197338104,
      "step": 710
    },
    {
      "epoch": 0.01424,
      "grad_norm": 3.875,
      "grad_norm_var": 0.14480692545572918,
      "learning_rate": 0.0001,
      "loss": 6.2466,
      "loss/crossentropy": 2.250674605369568,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3948116898536682,
      "step": 712
    },
    {
      "epoch": 0.01428,
      "grad_norm": 3.65625,
      "grad_norm_var": 0.15719401041666667,
      "learning_rate": 0.0001,
      "loss": 5.8891,
      "loss/crossentropy": 2.0895228385925293,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34471337497234344,
      "step": 714
    },
    {
      "epoch": 0.01432,
      "grad_norm": 4.71875,
      "grad_norm_var": 0.1826812744140625,
      "learning_rate": 0.0001,
      "loss": 6.3748,
      "loss/crossentropy": 2.337170124053955,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4226381927728653,
      "step": 716
    },
    {
      "epoch": 0.01436,
      "grad_norm": 4.03125,
      "grad_norm_var": 0.174462890625,
      "learning_rate": 0.0001,
      "loss": 6.3996,
      "loss/crossentropy": 2.337436556816101,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3738469183444977,
      "step": 718
    },
    {
      "epoch": 0.0144,
      "grad_norm": 4.15625,
      "grad_norm_var": 0.1669921875,
      "learning_rate": 0.0001,
      "loss": 6.0278,
      "loss/crossentropy": 1.9506489634513855,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.37431904673576355,
      "step": 720
    },
    {
      "epoch": 0.01444,
      "grad_norm": 3.71875,
      "grad_norm_var": 0.17283528645833332,
      "learning_rate": 0.0001,
      "loss": 5.8083,
      "loss/crossentropy": 2.253044009208679,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.36500048637390137,
      "step": 722
    },
    {
      "epoch": 0.01448,
      "grad_norm": 5.15625,
      "grad_norm_var": 0.23413798014322917,
      "learning_rate": 0.0001,
      "loss": 6.2307,
      "loss/crossentropy": 1.9980219006538391,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34674490988254547,
      "step": 724
    },
    {
      "epoch": 0.01452,
      "grad_norm": 3.609375,
      "grad_norm_var": 0.1903472900390625,
      "learning_rate": 0.0001,
      "loss": 5.8283,
      "loss/crossentropy": 1.894662618637085,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34921175241470337,
      "step": 726
    },
    {
      "epoch": 0.01456,
      "grad_norm": 3.609375,
      "grad_norm_var": 0.18310546875,
      "learning_rate": 0.0001,
      "loss": 6.1117,
      "loss/crossentropy": 2.304685115814209,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.36858032643795013,
      "step": 728
    },
    {
      "epoch": 0.0146,
      "grad_norm": 3.6875,
      "grad_norm_var": 0.17696940104166667,
      "learning_rate": 0.0001,
      "loss": 6.1041,
      "loss/crossentropy": 1.9020891189575195,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3854057639837265,
      "step": 730
    },
    {
      "epoch": 0.01464,
      "grad_norm": 3.4375,
      "grad_norm_var": 0.15485026041666666,
      "learning_rate": 0.0001,
      "loss": 5.5327,
      "loss/crossentropy": 1.707019329071045,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2961048036813736,
      "step": 732
    },
    {
      "epoch": 0.01468,
      "grad_norm": 3.828125,
      "grad_norm_var": 0.15344136555989582,
      "learning_rate": 0.0001,
      "loss": 6.0543,
      "loss/crossentropy": 2.423463463783264,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3739102631807327,
      "step": 734
    },
    {
      "epoch": 0.01472,
      "grad_norm": 4.0625,
      "grad_norm_var": 0.15335286458333333,
      "learning_rate": 0.0001,
      "loss": 6.2855,
      "loss/crossentropy": 2.0490055680274963,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4027387350797653,
      "step": 736
    },
    {
      "epoch": 0.01476,
      "grad_norm": 3.984375,
      "grad_norm_var": 0.15038960774739582,
      "learning_rate": 0.0001,
      "loss": 6.1236,
      "loss/crossentropy": 2.3712635040283203,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3602859079837799,
      "step": 738
    },
    {
      "epoch": 0.0148,
      "grad_norm": 3.75,
      "grad_norm_var": 0.058649698893229164,
      "learning_rate": 0.0001,
      "loss": 6.2938,
      "loss/crossentropy": 2.306379556655884,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.38698625564575195,
      "step": 740
    },
    {
      "epoch": 0.01484,
      "grad_norm": 3.984375,
      "grad_norm_var": 0.055939737955729166,
      "learning_rate": 0.0001,
      "loss": 6.3983,
      "loss/crossentropy": 2.6846178770065308,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3881431221961975,
      "step": 742
    },
    {
      "epoch": 0.01488,
      "grad_norm": 3.9375,
      "grad_norm_var": 0.058690388997395836,
      "learning_rate": 0.0001,
      "loss": 5.7984,
      "loss/crossentropy": 2.0555977821350098,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.36078818142414093,
      "step": 744
    },
    {
      "epoch": 0.01492,
      "grad_norm": 4.09375,
      "grad_norm_var": 0.059325154622395834,
      "learning_rate": 0.0001,
      "loss": 5.7834,
      "loss/crossentropy": 2.0597304701805115,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34516778588294983,
      "step": 746
    },
    {
      "epoch": 0.01496,
      "grad_norm": 3.5625,
      "grad_norm_var": 0.049479166666666664,
      "learning_rate": 0.0001,
      "loss": 5.8911,
      "loss/crossentropy": 1.9607329964637756,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33818933367729187,
      "step": 748
    },
    {
      "epoch": 0.015,
      "grad_norm": 3.546875,
      "grad_norm_var": 0.057291666666666664,
      "learning_rate": 0.0001,
      "loss": 5.8484,
      "loss/crossentropy": 1.7854246497154236,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3463115990161896,
      "step": 750
    },
    {
      "epoch": 0.01504,
      "grad_norm": 3.703125,
      "grad_norm_var": 0.0590240478515625,
      "learning_rate": 0.0001,
      "loss": 5.7788,
      "loss/crossentropy": 1.845078468322754,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3714388310909271,
      "step": 752
    },
    {
      "epoch": 0.01508,
      "grad_norm": 3.9375,
      "grad_norm_var": 0.05771077473958333,
      "learning_rate": 0.0001,
      "loss": 6.1677,
      "loss/crossentropy": 2.3951027393341064,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3723580837249756,
      "step": 754
    },
    {
      "epoch": 0.01512,
      "grad_norm": 3.6875,
      "grad_norm_var": 0.032486979166666666,
      "learning_rate": 0.0001,
      "loss": 6.219,
      "loss/crossentropy": 2.500870108604431,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.42450079321861267,
      "step": 756
    },
    {
      "epoch": 0.01516,
      "grad_norm": 3.75,
      "grad_norm_var": 0.030060831705729166,
      "learning_rate": 0.0001,
      "loss": 5.886,
      "loss/crossentropy": 2.1584274768829346,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3798908591270447,
      "step": 758
    },
    {
      "epoch": 0.0152,
      "grad_norm": 3.5625,
      "grad_norm_var": 0.028251139322916667,
      "learning_rate": 0.0001,
      "loss": 5.7937,
      "loss/crossentropy": 2.3126983642578125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33131279051303864,
      "step": 760
    },
    {
      "epoch": 0.01524,
      "grad_norm": 3.5,
      "grad_norm_var": 0.020246378580729165,
      "learning_rate": 0.0001,
      "loss": 5.9182,
      "loss/crossentropy": 2.349764347076416,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3910932093858719,
      "step": 762
    },
    {
      "epoch": 0.01528,
      "grad_norm": 3.640625,
      "grad_norm_var": 0.0224273681640625,
      "learning_rate": 0.0001,
      "loss": 6.0472,
      "loss/crossentropy": 2.2232795357704163,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3483322113752365,
      "step": 764
    },
    {
      "epoch": 0.01532,
      "grad_norm": 3.65625,
      "grad_norm_var": 0.0207427978515625,
      "learning_rate": 0.0001,
      "loss": 6.0312,
      "loss/crossentropy": 2.2273412942886353,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.38545119762420654,
      "step": 766
    },
    {
      "epoch": 0.01536,
      "grad_norm": 4.125,
      "grad_norm_var": 0.03355712890625,
      "learning_rate": 0.0001,
      "loss": 6.0523,
      "loss/crossentropy": 2.5879149436950684,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.38191574811935425,
      "step": 768
    },
    {
      "epoch": 0.0154,
      "grad_norm": 3.953125,
      "grad_norm_var": 0.034032185872395836,
      "learning_rate": 0.0001,
      "loss": 6.027,
      "loss/crossentropy": 2.3305420875549316,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3704134076833725,
      "step": 770
    },
    {
      "epoch": 0.01544,
      "grad_norm": 3.59375,
      "grad_norm_var": 0.03241780598958333,
      "learning_rate": 0.0001,
      "loss": 6.121,
      "loss/crossentropy": 2.0433666706085205,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34387587010860443,
      "step": 772
    },
    {
      "epoch": 0.01548,
      "grad_norm": 3.609375,
      "grad_norm_var": 0.03337300618489583,
      "learning_rate": 0.0001,
      "loss": 5.5837,
      "loss/crossentropy": 2.1127337217330933,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3484109789133072,
      "step": 774
    },
    {
      "epoch": 0.01552,
      "grad_norm": 3.859375,
      "grad_norm_var": 0.029002888997395834,
      "learning_rate": 0.0001,
      "loss": 6.0028,
      "loss/crossentropy": 2.1637459993362427,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3757011145353317,
      "step": 776
    },
    {
      "epoch": 0.01556,
      "grad_norm": 3.734375,
      "grad_norm_var": 0.024446614583333335,
      "learning_rate": 0.0001,
      "loss": 5.9904,
      "loss/crossentropy": 2.4118471145629883,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3797076344490051,
      "step": 778
    },
    {
      "epoch": 0.0156,
      "grad_norm": 3.828125,
      "grad_norm_var": 0.022191365559895832,
      "learning_rate": 0.0001,
      "loss": 6.2649,
      "loss/crossentropy": 1.9410768151283264,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3254295587539673,
      "step": 780
    },
    {
      "epoch": 0.01564,
      "grad_norm": 3.609375,
      "grad_norm_var": 0.023558553059895834,
      "learning_rate": 0.0001,
      "loss": 5.9008,
      "loss/crossentropy": 2.1669737100601196,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3583529591560364,
      "step": 782
    },
    {
      "epoch": 0.01568,
      "grad_norm": 3.5625,
      "grad_norm_var": 0.017606608072916665,
      "learning_rate": 0.0001,
      "loss": 5.9868,
      "loss/crossentropy": 2.217113733291626,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3503105044364929,
      "step": 784
    },
    {
      "epoch": 0.01572,
      "grad_norm": 3.84375,
      "grad_norm_var": 0.017154947916666666,
      "learning_rate": 0.0001,
      "loss": 6.0695,
      "loss/crossentropy": 2.588438868522644,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3941201716661453,
      "step": 786
    },
    {
      "epoch": 0.01576,
      "grad_norm": 3.546875,
      "grad_norm_var": 0.018310546875,
      "learning_rate": 0.0001,
      "loss": 5.9436,
      "loss/crossentropy": 2.3925808668136597,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3755808621644974,
      "step": 788
    },
    {
      "epoch": 0.0158,
      "grad_norm": 3.625,
      "grad_norm_var": 0.018684895833333333,
      "learning_rate": 0.0001,
      "loss": 5.7254,
      "loss/crossentropy": 1.9568504691123962,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3865346759557724,
      "step": 790
    },
    {
      "epoch": 0.01584,
      "grad_norm": 3.96875,
      "grad_norm_var": 0.024104817708333334,
      "learning_rate": 0.0001,
      "loss": 6.0174,
      "loss/crossentropy": 2.336462616920471,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.36937348544597626,
      "step": 792
    },
    {
      "epoch": 0.01588,
      "grad_norm": 3.421875,
      "grad_norm_var": 0.03623046875,
      "learning_rate": 0.0001,
      "loss": 5.7742,
      "loss/crossentropy": 2.1867082715034485,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.35803911089897156,
      "step": 794
    },
    {
      "epoch": 0.01592,
      "grad_norm": 3.765625,
      "grad_norm_var": 0.03877665201822917,
      "learning_rate": 0.0001,
      "loss": 6.2825,
      "loss/crossentropy": 2.070562243461609,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.42781224846839905,
      "step": 796
    },
    {
      "epoch": 0.01596,
      "grad_norm": 4.09375,
      "grad_norm_var": 0.044384765625,
      "learning_rate": 0.0001,
      "loss": 6.401,
      "loss/crossentropy": 2.160820960998535,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34495553374290466,
      "step": 798
    },
    {
      "epoch": 0.016,
      "grad_norm": 4.0,
      "grad_norm_var": 0.045426432291666666,
      "learning_rate": 0.0001,
      "loss": 5.9628,
      "loss/crossentropy": 2.3424230813980103,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4417698383331299,
      "step": 800
    },
    {
      "epoch": 0.01604,
      "grad_norm": 4.59375,
      "grad_norm_var": 0.08385009765625,
      "learning_rate": 0.0001,
      "loss": 6.1984,
      "loss/crossentropy": 2.090175747871399,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.35883304476737976,
      "step": 802
    },
    {
      "epoch": 0.01608,
      "grad_norm": 3.90625,
      "grad_norm_var": 0.0759429931640625,
      "learning_rate": 0.0001,
      "loss": 6.2044,
      "loss/crossentropy": 2.460660457611084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.36723683774471283,
      "step": 804
    },
    {
      "epoch": 0.01612,
      "grad_norm": 3.78125,
      "grad_norm_var": 0.0783203125,
      "learning_rate": 0.0001,
      "loss": 5.8788,
      "loss/crossentropy": 2.2680885791778564,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3925183415412903,
      "step": 806
    },
    {
      "epoch": 0.01616,
      "grad_norm": 3.796875,
      "grad_norm_var": 0.10422261555989583,
      "learning_rate": 0.0001,
      "loss": 6.1179,
      "loss/crossentropy": 2.272566020488739,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3425859659910202,
      "step": 808
    },
    {
      "epoch": 0.0162,
      "grad_norm": 3.546875,
      "grad_norm_var": 0.10061442057291667,
      "learning_rate": 0.0001,
      "loss": 5.8933,
      "loss/crossentropy": 2.2417107820510864,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3634066879749298,
      "step": 810
    },
    {
      "epoch": 0.01624,
      "grad_norm": 4.09375,
      "grad_norm_var": 0.10075581868489583,
      "learning_rate": 0.0001,
      "loss": 5.9907,
      "loss/crossentropy": 2.2117987275123596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3669978231191635,
      "step": 812
    },
    {
      "epoch": 0.01628,
      "grad_norm": 4.53125,
      "grad_norm_var": 0.12078348795572917,
      "learning_rate": 0.0001,
      "loss": 6.1767,
      "loss/crossentropy": 2.3471380472183228,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.39207911491394043,
      "step": 814
    },
    {
      "epoch": 0.01632,
      "grad_norm": 4.1875,
      "grad_norm_var": 0.12200113932291666,
      "learning_rate": 0.0001,
      "loss": 6.005,
      "loss/crossentropy": 2.1516740322113037,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3394138962030411,
      "step": 816
    },
    {
      "epoch": 0.01636,
      "grad_norm": 3.765625,
      "grad_norm_var": 0.10528055826822917,
      "learning_rate": 0.0001,
      "loss": 6.0827,
      "loss/crossentropy": 2.5085272789001465,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.39268872141838074,
      "step": 818
    },
    {
      "epoch": 0.0164,
      "grad_norm": 3.515625,
      "grad_norm_var": 0.1198883056640625,
      "learning_rate": 0.0001,
      "loss": 6.0363,
      "loss/crossentropy": 2.3051916360855103,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3798936903476715,
      "step": 820
    },
    {
      "epoch": 0.01644,
      "grad_norm": 3.5,
      "grad_norm_var": 0.14094136555989584,
      "learning_rate": 0.0001,
      "loss": 5.4403,
      "loss/crossentropy": 2.1685640811920166,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3666132390499115,
      "step": 822
    },
    {
      "epoch": 0.01648,
      "grad_norm": 3.53125,
      "grad_norm_var": 0.109521484375,
      "learning_rate": 0.0001,
      "loss": 5.6981,
      "loss/crossentropy": 2.3374987840652466,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32505205273628235,
      "step": 824
    },
    {
      "epoch": 0.01652,
      "grad_norm": 5.4375,
      "grad_norm_var": 0.280859375,
      "learning_rate": 0.0001,
      "loss": 6.1428,
      "loss/crossentropy": 2.6427528858184814,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4296632409095764,
      "step": 826
    },
    {
      "epoch": 0.01656,
      "grad_norm": 4.0,
      "grad_norm_var": 0.28609619140625,
      "learning_rate": 0.0001,
      "loss": 5.8304,
      "loss/crossentropy": 2.1534847617149353,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.37605202198028564,
      "step": 828
    },
    {
      "epoch": 0.0166,
      "grad_norm": 4.28125,
      "grad_norm_var": 0.2637278238932292,
      "learning_rate": 0.0001,
      "loss": 6.2115,
      "loss/crossentropy": 1.9642478227615356,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.43740569055080414,
      "step": 830
    },
    {
      "epoch": 0.01664,
      "grad_norm": 4.59375,
      "grad_norm_var": 3.3863433837890624,
      "learning_rate": 0.0001,
      "loss": 6.5306,
      "loss/crossentropy": 2.283148407936096,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3901669532060623,
      "step": 832
    },
    {
      "epoch": 0.01668,
      "grad_norm": 3.46875,
      "grad_norm_var": 3.382255045572917,
      "learning_rate": 0.0001,
      "loss": 6.0831,
      "loss/crossentropy": 2.418351888656616,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4011431038379669,
      "step": 834
    },
    {
      "epoch": 0.01672,
      "grad_norm": 4.03125,
      "grad_norm_var": 3.322565714518229,
      "learning_rate": 0.0001,
      "loss": 6.1852,
      "loss/crossentropy": 2.40928852558136,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.40596309304237366,
      "step": 836
    },
    {
      "epoch": 0.01676,
      "grad_norm": 3.34375,
      "grad_norm_var": 3.3059234619140625,
      "learning_rate": 0.0001,
      "loss": 5.791,
      "loss/crossentropy": 2.4211392402648926,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3640473484992981,
      "step": 838
    },
    {
      "epoch": 0.0168,
      "grad_norm": 3.484375,
      "grad_norm_var": 3.2890625,
      "learning_rate": 0.0001,
      "loss": 5.6553,
      "loss/crossentropy": 2.047215461730957,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3346950262784958,
      "step": 840
    },
    {
      "epoch": 0.01684,
      "grad_norm": 3.578125,
      "grad_norm_var": 3.28623046875,
      "learning_rate": 0.0001,
      "loss": 5.7093,
      "loss/crossentropy": 2.020021378993988,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3442958742380142,
      "step": 842
    },
    {
      "epoch": 0.01688,
      "grad_norm": 3.96875,
      "grad_norm_var": 3.27164306640625,
      "learning_rate": 0.0001,
      "loss": 6.1484,
      "loss/crossentropy": 2.1585127115249634,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3768642693758011,
      "step": 844
    },
    {
      "epoch": 0.01692,
      "grad_norm": 3.84375,
      "grad_norm_var": 3.298802693684896,
      "learning_rate": 0.0001,
      "loss": 5.8512,
      "loss/crossentropy": 2.2717286348342896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3644135594367981,
      "step": 846
    },
    {
      "epoch": 0.01696,
      "grad_norm": 3.828125,
      "grad_norm_var": 0.06539713541666667,
      "learning_rate": 0.0001,
      "loss": 5.9203,
      "loss/crossentropy": 2.2683321237564087,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3795373737812042,
      "step": 848
    },
    {
      "epoch": 0.017,
      "grad_norm": 3.1875,
      "grad_norm_var": 0.07066141764322917,
      "learning_rate": 0.0001,
      "loss": 5.7235,
      "loss/crossentropy": 2.1189464330673218,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.35512739419937134,
      "step": 850
    },
    {
      "epoch": 0.01704,
      "grad_norm": 3.65625,
      "grad_norm_var": 0.060384114583333336,
      "learning_rate": 0.0001,
      "loss": 5.6536,
      "loss/crossentropy": 2.260777235031128,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.35152101516723633,
      "step": 852
    },
    {
      "epoch": 0.01708,
      "grad_norm": 3.671875,
      "grad_norm_var": 0.06897379557291666,
      "learning_rate": 0.0001,
      "loss": 5.93,
      "loss/crossentropy": 2.323577642440796,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32252687215805054,
      "step": 854
    },
    {
      "epoch": 0.01712,
      "grad_norm": 3.71875,
      "grad_norm_var": 0.07024637858072917,
      "learning_rate": 0.0001,
      "loss": 6.1006,
      "loss/crossentropy": 2.5965300798416138,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.43394728004932404,
      "step": 856
    },
    {
      "epoch": 0.01716,
      "grad_norm": 3.140625,
      "grad_norm_var": 0.08170572916666667,
      "learning_rate": 0.0001,
      "loss": 5.8612,
      "loss/crossentropy": 2.078580856323242,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3437638282775879,
      "step": 858
    },
    {
      "epoch": 0.0172,
      "grad_norm": 3.359375,
      "grad_norm_var": 0.08163960774739583,
      "learning_rate": 0.0001,
      "loss": 5.8394,
      "loss/crossentropy": 2.425456404685974,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3522993326187134,
      "step": 860
    },
    {
      "epoch": 0.01724,
      "grad_norm": 3.515625,
      "grad_norm_var": 0.08478190104166666,
      "learning_rate": 0.0001,
      "loss": 6.0154,
      "loss/crossentropy": 2.2830835580825806,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.37188920378685,
      "step": 862
    },
    {
      "epoch": 0.01728,
      "grad_norm": 3.8125,
      "grad_norm_var": 0.06896870930989583,
      "learning_rate": 0.0001,
      "loss": 5.9086,
      "loss/crossentropy": 2.090674340724945,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32989686727523804,
      "step": 864
    },
    {
      "epoch": 0.01732,
      "grad_norm": 3.8125,
      "grad_norm_var": 0.0698394775390625,
      "learning_rate": 0.0001,
      "loss": 5.9617,
      "loss/crossentropy": 2.304458498954773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.37342821061611176,
      "step": 866
    },
    {
      "epoch": 0.01736,
      "grad_norm": 3.421875,
      "grad_norm_var": 0.0715484619140625,
      "learning_rate": 0.0001,
      "loss": 5.8593,
      "loss/crossentropy": 2.6545844078063965,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3627774566411972,
      "step": 868
    },
    {
      "epoch": 0.0174,
      "grad_norm": 3.3125,
      "grad_norm_var": 0.059789021809895836,
      "learning_rate": 0.0001,
      "loss": 5.6956,
      "loss/crossentropy": 1.9977945685386658,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.318715900182724,
      "step": 870
    },
    {
      "epoch": 0.01744,
      "grad_norm": 3.203125,
      "grad_norm_var": 0.08033854166666667,
      "learning_rate": 0.0001,
      "loss": 5.7408,
      "loss/crossentropy": 1.9226595759391785,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3187277615070343,
      "step": 872
    },
    {
      "epoch": 0.01748,
      "grad_norm": 3.6875,
      "grad_norm_var": 0.0694732666015625,
      "learning_rate": 0.0001,
      "loss": 5.9863,
      "loss/crossentropy": 2.323302686214447,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.36724327504634857,
      "step": 874
    },
    {
      "epoch": 0.01752,
      "grad_norm": 3.5625,
      "grad_norm_var": 0.07043355305989583,
      "learning_rate": 0.0001,
      "loss": 5.9913,
      "loss/crossentropy": 2.254343032836914,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3444042354822159,
      "step": 876
    },
    {
      "epoch": 0.01756,
      "grad_norm": 3.296875,
      "grad_norm_var": 0.0774078369140625,
      "learning_rate": 0.0001,
      "loss": 5.5038,
      "loss/crossentropy": 2.0819836854934692,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3477388769388199,
      "step": 878
    },
    {
      "epoch": 0.0176,
      "grad_norm": 3.703125,
      "grad_norm_var": 0.10198160807291666,
      "learning_rate": 0.0001,
      "loss": 5.9947,
      "loss/crossentropy": 2.377693295478821,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.36597058176994324,
      "step": 880
    },
    {
      "epoch": 0.01764,
      "grad_norm": 4.21875,
      "grad_norm_var": 0.11789449055989583,
      "learning_rate": 0.0001,
      "loss": 6.3011,
      "loss/crossentropy": 2.5598798990249634,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4082639515399933,
      "step": 882
    },
    {
      "epoch": 0.01768,
      "grad_norm": 3.6875,
      "grad_norm_var": 0.1294830322265625,
      "learning_rate": 0.0001,
      "loss": 5.9966,
      "loss/crossentropy": 2.2847843170166016,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3531750440597534,
      "step": 884
    },
    {
      "epoch": 0.01772,
      "grad_norm": 3.390625,
      "grad_norm_var": 0.12878316243489582,
      "learning_rate": 0.0001,
      "loss": 5.6685,
      "loss/crossentropy": 1.8283140063285828,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.336555078625679,
      "step": 886
    },
    {
      "epoch": 0.01776,
      "grad_norm": 3.65625,
      "grad_norm_var": 0.09971415201822917,
      "learning_rate": 0.0001,
      "loss": 5.9507,
      "loss/crossentropy": 2.1001436710357666,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3380406051874161,
      "step": 888
    },
    {
      "epoch": 0.0178,
      "grad_norm": 4.21875,
      "grad_norm_var": 0.11876627604166666,
      "learning_rate": 0.0001,
      "loss": 5.7552,
      "loss/crossentropy": 2.0079030990600586,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.35362809896469116,
      "step": 890
    },
    {
      "epoch": 0.01784,
      "grad_norm": 4.46875,
      "grad_norm_var": 0.15650634765625,
      "learning_rate": 0.0001,
      "loss": 5.7416,
      "loss/crossentropy": 2.176286220550537,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34369874000549316,
      "step": 892
    },
    {
      "epoch": 0.01788,
      "grad_norm": 3.984375,
      "grad_norm_var": 0.13087565104166668,
      "learning_rate": 0.0001,
      "loss": 5.9236,
      "loss/crossentropy": 2.17675244808197,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3495863378047943,
      "step": 894
    },
    {
      "epoch": 0.01792,
      "grad_norm": 4.0,
      "grad_norm_var": 0.12189127604166666,
      "learning_rate": 0.0001,
      "loss": 5.9109,
      "loss/crossentropy": 2.312318801879883,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.40353919565677643,
      "step": 896
    },
    {
      "epoch": 0.01796,
      "grad_norm": 3.375,
      "grad_norm_var": 0.14010009765625,
      "learning_rate": 0.0001,
      "loss": 6.071,
      "loss/crossentropy": 2.28923499584198,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3770768642425537,
      "step": 898
    },
    {
      "epoch": 0.018,
      "grad_norm": 3.546875,
      "grad_norm_var": 0.1447662353515625,
      "learning_rate": 0.0001,
      "loss": 6.0275,
      "loss/crossentropy": 2.2060720920562744,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.35431359708309174,
      "step": 900
    },
    {
      "epoch": 0.01804,
      "grad_norm": 3.296875,
      "grad_norm_var": 0.17552083333333332,
      "learning_rate": 0.0001,
      "loss": 5.4052,
      "loss/crossentropy": 2.0325206518173218,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3037416934967041,
      "step": 902
    },
    {
      "epoch": 0.01808,
      "grad_norm": 3.546875,
      "grad_norm_var": 0.17635091145833334,
      "learning_rate": 0.0001,
      "loss": 5.9148,
      "loss/crossentropy": 2.1943042278289795,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3509814143180847,
      "step": 904
    },
    {
      "epoch": 0.01812,
      "grad_norm": 3.4375,
      "grad_norm_var": 0.1696197509765625,
      "learning_rate": 0.0001,
      "loss": 5.563,
      "loss/crossentropy": 1.9589214324951172,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31621459126472473,
      "step": 906
    },
    {
      "epoch": 0.01816,
      "grad_norm": 3.375,
      "grad_norm_var": 0.12841389973958334,
      "learning_rate": 0.0001,
      "loss": 5.6511,
      "loss/crossentropy": 2.329489588737488,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3837278485298157,
      "step": 908
    },
    {
      "epoch": 0.0182,
      "grad_norm": 3.5625,
      "grad_norm_var": 0.09648030598958333,
      "learning_rate": 0.0001,
      "loss": 5.8082,
      "loss/crossentropy": 2.1757726669311523,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3659580200910568,
      "step": 910
    },
    {
      "epoch": 0.01824,
      "grad_norm": 3.75,
      "grad_norm_var": 0.08772786458333333,
      "learning_rate": 0.0001,
      "loss": 5.7372,
      "loss/crossentropy": 2.1498661041259766,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3474857211112976,
      "step": 912
    },
    {
      "epoch": 0.01828,
      "grad_norm": 15.8125,
      "grad_norm_var": 9.504325358072917,
      "learning_rate": 0.0001,
      "loss": 5.9297,
      "loss/crossentropy": 2.4722740650177,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.505307987332344,
      "step": 914
    },
    {
      "epoch": 0.01832,
      "grad_norm": 9.0,
      "grad_norm_var": 10.75227762858073,
      "learning_rate": 0.0001,
      "loss": 5.6296,
      "loss/crossentropy": 1.855428695678711,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31347331404685974,
      "step": 916
    },
    {
      "epoch": 0.01836,
      "grad_norm": 3.75,
      "grad_norm_var": 10.50523173014323,
      "learning_rate": 0.0001,
      "loss": 5.9769,
      "loss/crossentropy": 2.326256275177002,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3996751010417938,
      "step": 918
    },
    {
      "epoch": 0.0184,
      "grad_norm": 3.546875,
      "grad_norm_var": 10.518257649739583,
      "learning_rate": 0.0001,
      "loss": 5.8444,
      "loss/crossentropy": 2.3712844848632812,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.37993232905864716,
      "step": 920
    },
    {
      "epoch": 0.01844,
      "grad_norm": 3.5,
      "grad_norm_var": 10.657861328125,
      "learning_rate": 0.0001,
      "loss": 5.5577,
      "loss/crossentropy": 2.0161430835723877,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3590858578681946,
      "step": 922
    },
    {
      "epoch": 0.01848,
      "grad_norm": 5.40625,
      "grad_norm_var": 10.520926920572917,
      "learning_rate": 0.0001,
      "loss": 5.6675,
      "loss/crossentropy": 2.2401121258735657,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33380126953125,
      "step": 924
    },
    {
      "epoch": 0.01852,
      "grad_norm": 3.328125,
      "grad_norm_var": 10.598356119791667,
      "learning_rate": 0.0001,
      "loss": 5.9331,
      "loss/crossentropy": 2.2354423999786377,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34832654893398285,
      "step": 926
    },
    {
      "epoch": 0.01856,
      "grad_norm": 3.46875,
      "grad_norm_var": 10.647850545247396,
      "learning_rate": 0.0001,
      "loss": 5.5212,
      "loss/crossentropy": 2.158566176891327,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3664778769016266,
      "step": 928
    },
    {
      "epoch": 0.0186,
      "grad_norm": 3.703125,
      "grad_norm_var": 2.1230377197265624,
      "learning_rate": 0.0001,
      "loss": 5.9146,
      "loss/crossentropy": 2.270231008529663,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.35586032271385193,
      "step": 930
    },
    {
      "epoch": 0.01864,
      "grad_norm": 3.609375,
      "grad_norm_var": 0.3744303385416667,
      "learning_rate": 0.0001,
      "loss": 6.0013,
      "loss/crossentropy": 2.233540892601013,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3789799362421036,
      "step": 932
    },
    {
      "epoch": 0.01868,
      "grad_norm": 3.65625,
      "grad_norm_var": 0.27898763020833334,
      "learning_rate": 0.0001,
      "loss": 5.5019,
      "loss/crossentropy": 1.9381731152534485,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28776855766773224,
      "step": 934
    },
    {
      "epoch": 0.01872,
      "grad_norm": 3.34375,
      "grad_norm_var": 0.28227437337239586,
      "learning_rate": 0.0001,
      "loss": 5.6759,
      "loss/crossentropy": 2.4225244522094727,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.355392187833786,
      "step": 936
    },
    {
      "epoch": 0.01876,
      "grad_norm": 3.15625,
      "grad_norm_var": 0.28084208170572916,
      "learning_rate": 0.0001,
      "loss": 5.911,
      "loss/crossentropy": 2.58090603351593,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3640855699777603,
      "step": 938
    },
    {
      "epoch": 0.0188,
      "grad_norm": 3.5625,
      "grad_norm_var": 0.03673502604166667,
      "learning_rate": 0.0001,
      "loss": 6.0623,
      "loss/crossentropy": 2.436452269554138,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3797999918460846,
      "step": 940
    },
    {
      "epoch": 0.01884,
      "grad_norm": 3.3125,
      "grad_norm_var": 0.03680013020833333,
      "learning_rate": 0.0001,
      "loss": 5.7428,
      "loss/crossentropy": 2.0378769636154175,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3278265744447708,
      "step": 942
    },
    {
      "epoch": 0.01888,
      "grad_norm": 3.546875,
      "grad_norm_var": 0.03860270182291667,
      "learning_rate": 0.0001,
      "loss": 5.6211,
      "loss/crossentropy": 2.1212962865829468,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34586282074451447,
      "step": 944
    },
    {
      "epoch": 0.01892,
      "grad_norm": 3.515625,
      "grad_norm_var": 0.040087890625,
      "learning_rate": 0.0001,
      "loss": 5.6695,
      "loss/crossentropy": 2.1884353160858154,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.40608666837215424,
      "step": 946
    },
    {
      "epoch": 0.01896,
      "grad_norm": 3.671875,
      "grad_norm_var": 0.046858723958333334,
      "learning_rate": 0.0001,
      "loss": 5.6684,
      "loss/crossentropy": 2.2093260288238525,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3519093841314316,
      "step": 948
    },
    {
      "epoch": 0.019,
      "grad_norm": 3.328125,
      "grad_norm_var": 0.03243815104166667,
      "learning_rate": 0.0001,
      "loss": 6.0842,
      "loss/crossentropy": 2.4246588945388794,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.37247334420681,
      "step": 950
    },
    {
      "epoch": 0.01904,
      "grad_norm": 3.671875,
      "grad_norm_var": 0.031022135416666666,
      "learning_rate": 0.0001,
      "loss": 5.6116,
      "loss/crossentropy": 1.932490050792694,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32985979318618774,
      "step": 952
    },
    {
      "epoch": 0.01908,
      "grad_norm": 3.90625,
      "grad_norm_var": 0.7084920247395833,
      "learning_rate": 0.0001,
      "loss": 5.7393,
      "loss/crossentropy": 2.4439035654067993,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.38164034485816956,
      "step": 954
    },
    {
      "epoch": 0.01912,
      "grad_norm": 3.296875,
      "grad_norm_var": 0.72437744140625,
      "learning_rate": 0.0001,
      "loss": 5.6255,
      "loss/crossentropy": 1.8876591920852661,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3267661929130554,
      "step": 956
    },
    {
      "epoch": 0.01916,
      "grad_norm": 3.578125,
      "grad_norm_var": 0.6990193684895833,
      "learning_rate": 0.0001,
      "loss": 5.7367,
      "loss/crossentropy": 2.284990072250366,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34810060262680054,
      "step": 958
    },
    {
      "epoch": 0.0192,
      "grad_norm": 3.53125,
      "grad_norm_var": 0.6961008707682291,
      "learning_rate": 0.0001,
      "loss": 5.888,
      "loss/crossentropy": 2.333263397216797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.42767176032066345,
      "step": 960
    },
    {
      "epoch": 0.01924,
      "grad_norm": 3.296875,
      "grad_norm_var": 0.7323527018229167,
      "learning_rate": 0.0001,
      "loss": 5.6021,
      "loss/crossentropy": 2.2526148557662964,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3284706473350525,
      "step": 962
    },
    {
      "epoch": 0.01928,
      "grad_norm": 4.25,
      "grad_norm_var": 0.7586252848307292,
      "learning_rate": 0.0001,
      "loss": 5.5479,
      "loss/crossentropy": 2.1782984137535095,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.35957905650138855,
      "step": 964
    },
    {
      "epoch": 0.01932,
      "grad_norm": 4.28125,
      "grad_norm_var": 0.75205078125,
      "learning_rate": 0.0001,
      "loss": 6.2783,
      "loss/crossentropy": 2.292098045349121,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.35862940549850464,
      "step": 966
    },
    {
      "epoch": 0.01936,
      "grad_norm": 3.546875,
      "grad_norm_var": 0.7503214518229167,
      "learning_rate": 0.0001,
      "loss": 5.9252,
      "loss/crossentropy": 2.102781653404236,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3527261018753052,
      "step": 968
    },
    {
      "epoch": 0.0194,
      "grad_norm": 3.453125,
      "grad_norm_var": 0.11111653645833333,
      "learning_rate": 0.0001,
      "loss": 5.8891,
      "loss/crossentropy": 2.223380208015442,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3608042299747467,
      "step": 970
    },
    {
      "epoch": 0.01944,
      "grad_norm": 3.4375,
      "grad_norm_var": 0.10501200358072917,
      "learning_rate": 0.0001,
      "loss": 5.3348,
      "loss/crossentropy": 2.0684096813201904,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31996411085128784,
      "step": 972
    },
    {
      "epoch": 0.01948,
      "grad_norm": 4.15625,
      "grad_norm_var": 0.126708984375,
      "learning_rate": 0.0001,
      "loss": 5.6642,
      "loss/crossentropy": 2.2011090517044067,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34404022991657257,
      "step": 974
    },
    {
      "epoch": 0.01952,
      "grad_norm": 3.140625,
      "grad_norm_var": 0.14937235514322916,
      "learning_rate": 0.0001,
      "loss": 5.5033,
      "loss/crossentropy": 2.027641534805298,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31718479096889496,
      "step": 976
    },
    {
      "epoch": 0.01956,
      "grad_norm": 3.484375,
      "grad_norm_var": 0.13909403483072916,
      "learning_rate": 0.0001,
      "loss": 5.7294,
      "loss/crossentropy": 2.311842203140259,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3904002010822296,
      "step": 978
    },
    {
      "epoch": 0.0196,
      "grad_norm": 3.859375,
      "grad_norm_var": 0.11236572265625,
      "learning_rate": 0.0001,
      "loss": 5.4402,
      "loss/crossentropy": 2.3605271577835083,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.38789358735084534,
      "step": 980
    },
    {
      "epoch": 0.01964,
      "grad_norm": 3.765625,
      "grad_norm_var": 0.095166015625,
      "learning_rate": 0.0001,
      "loss": 6.1094,
      "loss/crossentropy": 2.1687097549438477,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3534909188747406,
      "step": 982
    },
    {
      "epoch": 0.01968,
      "grad_norm": 3.6875,
      "grad_norm_var": 0.0962554931640625,
      "learning_rate": 0.0001,
      "loss": 5.6656,
      "loss/crossentropy": 2.194393038749695,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34674669802188873,
      "step": 984
    },
    {
      "epoch": 0.01972,
      "grad_norm": 3.359375,
      "grad_norm_var": 0.09602762858072916,
      "learning_rate": 0.0001,
      "loss": 5.5513,
      "loss/crossentropy": 2.1355903148651123,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3180558532476425,
      "step": 986
    },
    {
      "epoch": 0.01976,
      "grad_norm": 3.390625,
      "grad_norm_var": 0.08559468587239584,
      "learning_rate": 0.0001,
      "loss": 5.9431,
      "loss/crossentropy": 2.2688111066818237,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.365144744515419,
      "step": 988
    },
    {
      "epoch": 0.0198,
      "grad_norm": 3.65625,
      "grad_norm_var": 0.06303609212239583,
      "learning_rate": 0.0001,
      "loss": 5.5117,
      "loss/crossentropy": 2.423216700553894,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34281550347805023,
      "step": 990
    },
    {
      "epoch": 0.01984,
      "grad_norm": 3.375,
      "grad_norm_var": 0.0419586181640625,
      "learning_rate": 0.0001,
      "loss": 5.4698,
      "loss/crossentropy": 1.9360128045082092,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3343205749988556,
      "step": 992
    },
    {
      "epoch": 0.01988,
      "grad_norm": 3.421875,
      "grad_norm_var": 0.04468994140625,
      "learning_rate": 0.0001,
      "loss": 5.8128,
      "loss/crossentropy": 2.181576132774353,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.35567884147167206,
      "step": 994
    },
    {
      "epoch": 0.01992,
      "grad_norm": 3.578125,
      "grad_norm_var": 0.037398274739583334,
      "learning_rate": 0.0001,
      "loss": 5.9295,
      "loss/crossentropy": 2.166663408279419,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34105008840560913,
      "step": 996
    },
    {
      "epoch": 0.01996,
      "grad_norm": 3.203125,
      "grad_norm_var": 0.027408854166666666,
      "learning_rate": 0.0001,
      "loss": 5.5579,
      "loss/crossentropy": 2.285332202911377,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.35812389850616455,
      "step": 998
    },
    {
      "epoch": 0.02,
      "grad_norm": 4.0,
      "grad_norm_var": 35.396484375,
      "learning_rate": 0.0001,
      "loss": 6.3986,
      "loss/crossentropy": 2.088365077972412,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3717931807041168,
      "step": 1000
    },
    {
      "epoch": 0.02004,
      "grad_norm": 4.25,
      "grad_norm_var": 35.223714192708336,
      "learning_rate": 0.0001,
      "loss": 6.1327,
      "loss/crossentropy": 2.4051828384399414,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.420933797955513,
      "step": 1002
    },
    {
      "epoch": 0.02008,
      "grad_norm": 3.265625,
      "grad_norm_var": 35.27108968098958,
      "learning_rate": 0.0001,
      "loss": 5.6789,
      "loss/crossentropy": 2.3092572689056396,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.37571050226688385,
      "step": 1004
    },
    {
      "epoch": 0.02012,
      "grad_norm": 3.640625,
      "grad_norm_var": 35.29562072753906,
      "learning_rate": 0.0001,
      "loss": 5.6972,
      "loss/crossentropy": 2.147248387336731,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29974566400051117,
      "step": 1006
    },
    {
      "epoch": 0.02016,
      "grad_norm": 3.984375,
      "grad_norm_var": 35.141299438476565,
      "learning_rate": 0.0001,
      "loss": 5.988,
      "loss/crossentropy": 2.3385868668556213,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4254964739084244,
      "step": 1008
    },
    {
      "epoch": 0.0202,
      "grad_norm": 3.625,
      "grad_norm_var": 35.15125223795573,
      "learning_rate": 0.0001,
      "loss": 5.7553,
      "loss/crossentropy": 2.142681658267975,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3764440715312958,
      "step": 1010
    },
    {
      "epoch": 0.02024,
      "grad_norm": 3.34375,
      "grad_norm_var": 35.18043619791667,
      "learning_rate": 0.0001,
      "loss": 5.5947,
      "loss/crossentropy": 2.241790771484375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3385400176048279,
      "step": 1012
    },
    {
      "epoch": 0.02028,
      "grad_norm": 3.859375,
      "grad_norm_var": 34.836360677083334,
      "learning_rate": 0.0001,
      "loss": 6.3228,
      "loss/crossentropy": 2.1563867330551147,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3954617381095886,
      "step": 1014
    },
    {
      "epoch": 0.02032,
      "grad_norm": 3.484375,
      "grad_norm_var": 0.09921468098958333,
      "learning_rate": 0.0001,
      "loss": 5.4613,
      "loss/crossentropy": 1.9462800025939941,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32294341921806335,
      "step": 1016
    },
    {
      "epoch": 0.02036,
      "grad_norm": 3.53125,
      "grad_norm_var": 0.08026936848958334,
      "learning_rate": 0.0001,
      "loss": 5.4993,
      "loss/crossentropy": 1.83676278591156,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3142092078924179,
      "step": 1018
    },
    {
      "epoch": 0.0204,
      "grad_norm": 3.84375,
      "grad_norm_var": 0.09038798014322917,
      "learning_rate": 0.0001,
      "loss": 5.8174,
      "loss/crossentropy": 1.951962649822235,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3889008015394211,
      "step": 1020
    },
    {
      "epoch": 0.02044,
      "grad_norm": 3.578125,
      "grad_norm_var": 0.09976806640625,
      "learning_rate": 0.0001,
      "loss": 5.7173,
      "loss/crossentropy": 2.299771785736084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3214885741472244,
      "step": 1022
    },
    {
      "epoch": 0.02048,
      "grad_norm": 3.1875,
      "grad_norm_var": 0.09135640462239583,
      "learning_rate": 0.0001,
      "loss": 5.2291,
      "loss/crossentropy": 1.9117569327354431,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32827669382095337,
      "step": 1024
    },
    {
      "epoch": 0.02052,
      "grad_norm": 3.1875,
      "grad_norm_var": 0.096533203125,
      "learning_rate": 0.0001,
      "loss": 5.7974,
      "loss/crossentropy": 2.484488010406494,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34361426532268524,
      "step": 1026
    },
    {
      "epoch": 0.02056,
      "grad_norm": 3.59375,
      "grad_norm_var": 0.09973551432291666,
      "learning_rate": 0.0001,
      "loss": 5.7044,
      "loss/crossentropy": 2.3155311346054077,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3734404444694519,
      "step": 1028
    },
    {
      "epoch": 0.0206,
      "grad_norm": 3.125,
      "grad_norm_var": 0.04983317057291667,
      "learning_rate": 0.0001,
      "loss": 5.4202,
      "loss/crossentropy": 2.081188380718231,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3250262886285782,
      "step": 1030
    },
    {
      "epoch": 0.02064,
      "grad_norm": 3.09375,
      "grad_norm_var": 0.0574127197265625,
      "learning_rate": 0.0001,
      "loss": 5.5885,
      "loss/crossentropy": 2.044768512248993,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34609031677246094,
      "step": 1032
    },
    {
      "epoch": 0.02068,
      "grad_norm": 4.0,
      "grad_norm_var": 0.0875640869140625,
      "learning_rate": 0.0001,
      "loss": 6.078,
      "loss/crossentropy": 2.0666560530662537,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.365878626704216,
      "step": 1034
    },
    {
      "epoch": 0.02072,
      "grad_norm": 3.5625,
      "grad_norm_var": 0.08336588541666666,
      "learning_rate": 0.0001,
      "loss": 5.9891,
      "loss/crossentropy": 2.2933902740478516,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3754771202802658,
      "step": 1036
    },
    {
      "epoch": 0.02076,
      "grad_norm": 3.28125,
      "grad_norm_var": 0.08640848795572917,
      "learning_rate": 0.0001,
      "loss": 5.8105,
      "loss/crossentropy": 2.28829288482666,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3676798492670059,
      "step": 1038
    },
    {
      "epoch": 0.0208,
      "grad_norm": 3.5,
      "grad_norm_var": 0.08378499348958333,
      "learning_rate": 0.0001,
      "loss": 5.9305,
      "loss/crossentropy": 2.5891193151474,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.40094244480133057,
      "step": 1040
    },
    {
      "epoch": 0.02084,
      "grad_norm": 3.234375,
      "grad_norm_var": 0.08056233723958334,
      "learning_rate": 0.0001,
      "loss": 5.8579,
      "loss/crossentropy": 2.238967180252075,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3226759731769562,
      "step": 1042
    },
    {
      "epoch": 0.02088,
      "grad_norm": 3.296875,
      "grad_norm_var": 0.07796223958333333,
      "learning_rate": 0.0001,
      "loss": 5.2959,
      "loss/crossentropy": 2.0116711258888245,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3511478453874588,
      "step": 1044
    },
    {
      "epoch": 0.02092,
      "grad_norm": 3.515625,
      "grad_norm_var": 0.07344462076822916,
      "learning_rate": 0.0001,
      "loss": 5.4885,
      "loss/crossentropy": 2.4924051761627197,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3375450670719147,
      "step": 1046
    },
    {
      "epoch": 0.02096,
      "grad_norm": 3.234375,
      "grad_norm_var": 0.059235636393229166,
      "learning_rate": 0.0001,
      "loss": 5.6838,
      "loss/crossentropy": 2.138728439807892,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34410610795021057,
      "step": 1048
    },
    {
      "epoch": 0.021,
      "grad_norm": 3.515625,
      "grad_norm_var": 0.046122233072916664,
      "learning_rate": 0.0001,
      "loss": 5.7371,
      "loss/crossentropy": 2.3748635053634644,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3580169975757599,
      "step": 1050
    },
    {
      "epoch": 0.02104,
      "grad_norm": 3.125,
      "grad_norm_var": 0.044310506184895834,
      "learning_rate": 0.0001,
      "loss": 5.427,
      "loss/crossentropy": 2.061552882194519,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31069953739643097,
      "step": 1052
    },
    {
      "epoch": 0.02108,
      "grad_norm": 3.515625,
      "grad_norm_var": 0.03769124348958333,
      "learning_rate": 0.0001,
      "loss": 5.3469,
      "loss/crossentropy": 2.299555718898773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31592319905757904,
      "step": 1054
    },
    {
      "epoch": 0.02112,
      "grad_norm": 3.53125,
      "grad_norm_var": 0.04571940104166667,
      "learning_rate": 0.0001,
      "loss": 6.1254,
      "loss/crossentropy": 2.4866377115249634,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.37435297667980194,
      "step": 1056
    },
    {
      "epoch": 0.02116,
      "grad_norm": 3.203125,
      "grad_norm_var": 0.047684733072916666,
      "learning_rate": 0.0001,
      "loss": 5.5676,
      "loss/crossentropy": 1.8185054063796997,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2784232199192047,
      "step": 1058
    },
    {
      "epoch": 0.0212,
      "grad_norm": 3.3125,
      "grad_norm_var": 0.04840087890625,
      "learning_rate": 0.0001,
      "loss": 5.6184,
      "loss/crossentropy": 2.215538501739502,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3471361994743347,
      "step": 1060
    },
    {
      "epoch": 0.02124,
      "grad_norm": 3.078125,
      "grad_norm_var": 0.057616170247395834,
      "learning_rate": 0.0001,
      "loss": 5.7615,
      "loss/crossentropy": 2.6912894248962402,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.35911867022514343,
      "step": 1062
    },
    {
      "epoch": 0.02128,
      "grad_norm": 3.125,
      "grad_norm_var": 0.06670633951822917,
      "learning_rate": 0.0001,
      "loss": 5.3436,
      "loss/crossentropy": 1.9757090210914612,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29868973791599274,
      "step": 1064
    },
    {
      "epoch": 0.02132,
      "grad_norm": 3.21875,
      "grad_norm_var": 0.052294921875,
      "learning_rate": 0.0001,
      "loss": 5.5334,
      "loss/crossentropy": 2.3396666049957275,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3561312407255173,
      "step": 1066
    },
    {
      "epoch": 0.02136,
      "grad_norm": 3.359375,
      "grad_norm_var": 0.0477935791015625,
      "learning_rate": 0.0001,
      "loss": 5.7564,
      "loss/crossentropy": 2.3498982191085815,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3842166066169739,
      "step": 1068
    },
    {
      "epoch": 0.0214,
      "grad_norm": 3.234375,
      "grad_norm_var": 0.03717041015625,
      "learning_rate": 0.0001,
      "loss": 5.8936,
      "loss/crossentropy": 2.037585139274597,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3429017663002014,
      "step": 1070
    },
    {
      "epoch": 0.02144,
      "grad_norm": 3.484375,
      "grad_norm_var": 0.0216949462890625,
      "learning_rate": 0.0001,
      "loss": 5.6938,
      "loss/crossentropy": 2.4804376363754272,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.35338981449604034,
      "step": 1072
    },
    {
      "epoch": 0.02148,
      "grad_norm": 3.375,
      "grad_norm_var": 0.027448527018229165,
      "learning_rate": 0.0001,
      "loss": 5.8146,
      "loss/crossentropy": 2.5210201740264893,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3769296407699585,
      "step": 1074
    },
    {
      "epoch": 0.02152,
      "grad_norm": 3.40625,
      "grad_norm_var": 0.03369852701822917,
      "learning_rate": 0.0001,
      "loss": 5.7854,
      "loss/crossentropy": 2.1258187294006348,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3353133201599121,
      "step": 1076
    },
    {
      "epoch": 0.02156,
      "grad_norm": 3.328125,
      "grad_norm_var": 0.026676432291666666,
      "learning_rate": 0.0001,
      "loss": 5.5436,
      "loss/crossentropy": 2.2107938528060913,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34667155146598816,
      "step": 1078
    },
    {
      "epoch": 0.0216,
      "grad_norm": 3.171875,
      "grad_norm_var": 0.0194244384765625,
      "learning_rate": 0.0001,
      "loss": 5.7639,
      "loss/crossentropy": 1.9614633321762085,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2958581745624542,
      "step": 1080
    },
    {
      "epoch": 0.02164,
      "grad_norm": 3.515625,
      "grad_norm_var": 0.0194976806640625,
      "learning_rate": 0.0001,
      "loss": 5.8219,
      "loss/crossentropy": 2.1403249502182007,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3220343589782715,
      "step": 1082
    },
    {
      "epoch": 0.02168,
      "grad_norm": 3.28125,
      "grad_norm_var": 0.022493489583333335,
      "learning_rate": 0.0001,
      "loss": 5.6037,
      "loss/crossentropy": 1.6533048152923584,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2814648747444153,
      "step": 1084
    },
    {
      "epoch": 0.02172,
      "grad_norm": 3.59375,
      "grad_norm_var": 0.0232086181640625,
      "learning_rate": 0.0001,
      "loss": 5.7731,
      "loss/crossentropy": 2.69880211353302,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3835880607366562,
      "step": 1086
    },
    {
      "epoch": 0.02176,
      "grad_norm": 3.5,
      "grad_norm_var": 0.0240631103515625,
      "learning_rate": 0.0001,
      "loss": 5.8119,
      "loss/crossentropy": 2.214504837989807,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31686101853847504,
      "step": 1088
    },
    {
      "epoch": 0.0218,
      "grad_norm": 3.765625,
      "grad_norm_var": 0.027164713541666666,
      "learning_rate": 0.0001,
      "loss": 6.0376,
      "loss/crossentropy": 2.2377456426620483,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3682183176279068,
      "step": 1090
    },
    {
      "epoch": 0.02184,
      "grad_norm": 3.359375,
      "grad_norm_var": 0.027799479166666665,
      "learning_rate": 0.0001,
      "loss": 5.7387,
      "loss/crossentropy": 2.0977545976638794,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32313986122608185,
      "step": 1092
    },
    {
      "epoch": 0.02188,
      "grad_norm": 3.125,
      "grad_norm_var": 0.0371490478515625,
      "learning_rate": 0.0001,
      "loss": 5.6407,
      "loss/crossentropy": 2.1717870235443115,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3619600385427475,
      "step": 1094
    },
    {
      "epoch": 0.02192,
      "grad_norm": 3.546875,
      "grad_norm_var": 0.03674723307291667,
      "learning_rate": 0.0001,
      "loss": 5.554,
      "loss/crossentropy": 2.2805471420288086,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3328556418418884,
      "step": 1096
    },
    {
      "epoch": 0.02196,
      "grad_norm": 3.71875,
      "grad_norm_var": 0.04625244140625,
      "learning_rate": 0.0001,
      "loss": 5.8009,
      "loss/crossentropy": 2.034846782684326,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3610256612300873,
      "step": 1098
    },
    {
      "epoch": 0.022,
      "grad_norm": 3.359375,
      "grad_norm_var": 0.04612528483072917,
      "learning_rate": 0.0001,
      "loss": 5.6121,
      "loss/crossentropy": 2.0208348631858826,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3311958611011505,
      "step": 1100
    },
    {
      "epoch": 0.02204,
      "grad_norm": 3.59375,
      "grad_norm_var": 0.04737040201822917,
      "learning_rate": 0.0001,
      "loss": 5.7135,
      "loss/crossentropy": 2.1020554900169373,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3638540059328079,
      "step": 1102
    },
    {
      "epoch": 0.02208,
      "grad_norm": 3.46875,
      "grad_norm_var": 0.046019490559895834,
      "learning_rate": 0.0001,
      "loss": 5.5099,
      "loss/crossentropy": 2.28346848487854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3434429168701172,
      "step": 1104
    },
    {
      "epoch": 0.02212,
      "grad_norm": 3.21875,
      "grad_norm_var": 0.1152740478515625,
      "learning_rate": 0.0001,
      "loss": 5.6866,
      "loss/crossentropy": 2.103623867034912,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34000229835510254,
      "step": 1106
    },
    {
      "epoch": 0.02216,
      "grad_norm": 4.1875,
      "grad_norm_var": 0.14846089680989583,
      "learning_rate": 0.0001,
      "loss": 5.5196,
      "loss/crossentropy": 2.205894947052002,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3432965874671936,
      "step": 1108
    },
    {
      "epoch": 0.0222,
      "grad_norm": 3.109375,
      "grad_norm_var": 0.16467183430989582,
      "learning_rate": 0.0001,
      "loss": 5.6166,
      "loss/crossentropy": 2.395035982131958,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3834904432296753,
      "step": 1110
    },
    {
      "epoch": 0.02224,
      "grad_norm": 3.296875,
      "grad_norm_var": 0.16505533854166668,
      "learning_rate": 0.0001,
      "loss": 5.9042,
      "loss/crossentropy": 2.3755353689193726,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33256760239601135,
      "step": 1112
    },
    {
      "epoch": 0.02228,
      "grad_norm": 3.28125,
      "grad_norm_var": 0.16402079264322916,
      "learning_rate": 0.0001,
      "loss": 6.0135,
      "loss/crossentropy": 2.6754449605941772,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3524845540523529,
      "step": 1114
    },
    {
      "epoch": 0.02232,
      "grad_norm": 3.515625,
      "grad_norm_var": 0.15730692545572916,
      "learning_rate": 0.0001,
      "loss": 5.6448,
      "loss/crossentropy": 2.2398552894592285,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.343609020113945,
      "step": 1116
    },
    {
      "epoch": 0.02236,
      "grad_norm": 3.3125,
      "grad_norm_var": 0.16770833333333332,
      "learning_rate": 0.0001,
      "loss": 5.3075,
      "loss/crossentropy": 2.399322271347046,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34514427185058594,
      "step": 1118
    },
    {
      "epoch": 0.0224,
      "grad_norm": 3.3125,
      "grad_norm_var": 0.17229410807291667,
      "learning_rate": 0.0001,
      "loss": 5.8886,
      "loss/crossentropy": 2.4002050161361694,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.35784730315208435,
      "step": 1120
    },
    {
      "epoch": 0.02244,
      "grad_norm": 3.21875,
      "grad_norm_var": 0.10064188639322917,
      "learning_rate": 0.0001,
      "loss": 5.6239,
      "loss/crossentropy": 2.293683171272278,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.37052902579307556,
      "step": 1122
    },
    {
      "epoch": 0.02248,
      "grad_norm": 3.234375,
      "grad_norm_var": 0.0532379150390625,
      "learning_rate": 0.0001,
      "loss": 5.8244,
      "loss/crossentropy": 2.2177391052246094,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3482564836740494,
      "step": 1124
    },
    {
      "epoch": 0.02252,
      "grad_norm": 4.1875,
      "grad_norm_var": 0.0637603759765625,
      "learning_rate": 0.0001,
      "loss": 5.8959,
      "loss/crossentropy": 2.288211703300476,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3401540517807007,
      "step": 1126
    },
    {
      "epoch": 0.02256,
      "grad_norm": 3.1875,
      "grad_norm_var": 0.0933258056640625,
      "learning_rate": 0.0001,
      "loss": 5.5054,
      "loss/crossentropy": 2.1786144971847534,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.331302285194397,
      "step": 1128
    },
    {
      "epoch": 0.0226,
      "grad_norm": 3.5,
      "grad_norm_var": 0.09163004557291667,
      "learning_rate": 0.0001,
      "loss": 5.7604,
      "loss/crossentropy": 2.0390175580978394,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32953669130802155,
      "step": 1130
    },
    {
      "epoch": 0.02264,
      "grad_norm": 3.5,
      "grad_norm_var": 0.0921295166015625,
      "learning_rate": 0.0001,
      "loss": 5.6952,
      "loss/crossentropy": 2.188543677330017,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3585694134235382,
      "step": 1132
    },
    {
      "epoch": 0.02268,
      "grad_norm": 3.296875,
      "grad_norm_var": 0.08284098307291667,
      "learning_rate": 0.0001,
      "loss": 5.732,
      "loss/crossentropy": 2.0731694102287292,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3121813088655472,
      "step": 1134
    },
    {
      "epoch": 0.02272,
      "grad_norm": 3.390625,
      "grad_norm_var": 0.0794097900390625,
      "learning_rate": 0.0001,
      "loss": 5.6306,
      "loss/crossentropy": 2.144552707672119,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3303966820240021,
      "step": 1136
    },
    {
      "epoch": 0.02276,
      "grad_norm": 3.890625,
      "grad_norm_var": 0.09900614420572916,
      "learning_rate": 0.0001,
      "loss": 5.3794,
      "loss/crossentropy": 2.1617428064346313,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3420899361371994,
      "step": 1138
    },
    {
      "epoch": 0.0228,
      "grad_norm": 3.109375,
      "grad_norm_var": 0.12813212076822916,
      "learning_rate": 0.0001,
      "loss": 5.2834,
      "loss/crossentropy": 1.8098865747451782,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28376901149749756,
      "step": 1140
    },
    {
      "epoch": 0.02284,
      "grad_norm": 2.984375,
      "grad_norm_var": 0.09795633951822917,
      "learning_rate": 0.0001,
      "loss": 5.3911,
      "loss/crossentropy": 2.133797824382782,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3587050139904022,
      "step": 1142
    },
    {
      "epoch": 0.02288,
      "grad_norm": 4.96875,
      "grad_norm_var": 0.24221903483072918,
      "learning_rate": 0.0001,
      "loss": 5.8787,
      "loss/crossentropy": 2.378090739250183,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4244185537099838,
      "step": 1144
    },
    {
      "epoch": 0.02292,
      "grad_norm": 3.03125,
      "grad_norm_var": 0.2736887613932292,
      "learning_rate": 0.0001,
      "loss": 5.6692,
      "loss/crossentropy": 2.4442414045333862,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3569464683532715,
      "step": 1146
    },
    {
      "epoch": 0.02296,
      "grad_norm": 3.671875,
      "grad_norm_var": 0.29189453125,
      "learning_rate": 0.0001,
      "loss": 5.5468,
      "loss/crossentropy": 2.6446973085403442,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.36859095096588135,
      "step": 1148
    },
    {
      "epoch": 0.023,
      "grad_norm": 4.4375,
      "grad_norm_var": 0.3506988525390625,
      "learning_rate": 0.0001,
      "loss": 5.9727,
      "loss/crossentropy": 2.4100207090377808,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4082309305667877,
      "step": 1150
    },
    {
      "epoch": 0.02304,
      "grad_norm": 3.25,
      "grad_norm_var": 0.3552317301432292,
      "learning_rate": 0.0001,
      "loss": 5.3537,
      "loss/crossentropy": 2.1472485661506653,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3299275189638138,
      "step": 1152
    },
    {
      "epoch": 0.02308,
      "grad_norm": 3.21875,
      "grad_norm_var": 0.3376373291015625,
      "learning_rate": 0.0001,
      "loss": 5.4393,
      "loss/crossentropy": 2.1891872882843018,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.356732040643692,
      "step": 1154
    },
    {
      "epoch": 0.02312,
      "grad_norm": 3.71875,
      "grad_norm_var": 0.30201416015625,
      "learning_rate": 0.0001,
      "loss": 6.0049,
      "loss/crossentropy": 2.1432933807373047,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.360423281788826,
      "step": 1156
    },
    {
      "epoch": 0.02316,
      "grad_norm": 3.1875,
      "grad_norm_var": 0.2923248291015625,
      "learning_rate": 0.0001,
      "loss": 5.5541,
      "loss/crossentropy": 2.176819324493408,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.340317040681839,
      "step": 1158
    },
    {
      "epoch": 0.0232,
      "grad_norm": 2.96875,
      "grad_norm_var": 0.17407938639322917,
      "learning_rate": 0.0001,
      "loss": 5.2312,
      "loss/crossentropy": 2.325207471847534,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3309635668992996,
      "step": 1160
    },
    {
      "epoch": 0.02324,
      "grad_norm": 3.515625,
      "grad_norm_var": 0.14537353515625,
      "learning_rate": 0.0001,
      "loss": 5.7322,
      "loss/crossentropy": 2.2536743879318237,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31422293186187744,
      "step": 1162
    },
    {
      "epoch": 0.02328,
      "grad_norm": 3.328125,
      "grad_norm_var": 0.12657877604166667,
      "learning_rate": 0.0001,
      "loss": 5.4354,
      "loss/crossentropy": 2.180476427078247,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2956361174583435,
      "step": 1164
    },
    {
      "epoch": 0.02332,
      "grad_norm": 3.15625,
      "grad_norm_var": 0.0527008056640625,
      "learning_rate": 0.0001,
      "loss": 5.6067,
      "loss/crossentropy": 1.995088815689087,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2936599552631378,
      "step": 1166
    },
    {
      "epoch": 0.02336,
      "grad_norm": 3.1875,
      "grad_norm_var": 0.05378316243489583,
      "learning_rate": 0.0001,
      "loss": 5.7304,
      "loss/crossentropy": 2.2555994987487793,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3630402684211731,
      "step": 1168
    },
    {
      "epoch": 0.0234,
      "grad_norm": 3.296875,
      "grad_norm_var": 0.05856119791666667,
      "learning_rate": 0.0001,
      "loss": 5.1613,
      "loss/crossentropy": 2.0441415905952454,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2960120141506195,
      "step": 1170
    },
    {
      "epoch": 0.02344,
      "grad_norm": 3.25,
      "grad_norm_var": 0.0204498291015625,
      "learning_rate": 0.0001,
      "loss": 5.2238,
      "loss/crossentropy": 1.8090497255325317,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29750876128673553,
      "step": 1172
    },
    {
      "epoch": 0.02348,
      "grad_norm": 3.234375,
      "grad_norm_var": 0.0281158447265625,
      "learning_rate": 0.0001,
      "loss": 5.2506,
      "loss/crossentropy": 2.2780312299728394,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3376633822917938,
      "step": 1174
    },
    {
      "epoch": 0.02352,
      "grad_norm": 3.4375,
      "grad_norm_var": 0.02506103515625,
      "learning_rate": 0.0001,
      "loss": 5.53,
      "loss/crossentropy": 1.8047232627868652,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29132279753685,
      "step": 1176
    },
    {
      "epoch": 0.02356,
      "grad_norm": 3.265625,
      "grad_norm_var": 0.020417277018229166,
      "learning_rate": 0.0001,
      "loss": 5.4149,
      "loss/crossentropy": 2.203469753265381,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3124798536300659,
      "step": 1178
    },
    {
      "epoch": 0.0236,
      "grad_norm": 3.34375,
      "grad_norm_var": 0.019775390625,
      "learning_rate": 0.0001,
      "loss": 5.4481,
      "loss/crossentropy": 2.078102231025696,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.331977054476738,
      "step": 1180
    },
    {
      "epoch": 0.02364,
      "grad_norm": 3.765625,
      "grad_norm_var": 0.0368316650390625,
      "learning_rate": 0.0001,
      "loss": 5.93,
      "loss/crossentropy": 2.2701854705810547,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3202142268419266,
      "step": 1182
    },
    {
      "epoch": 0.02368,
      "grad_norm": 3.671875,
      "grad_norm_var": 0.047240193684895834,
      "learning_rate": 0.0001,
      "loss": 5.6414,
      "loss/crossentropy": 1.930423617362976,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3192301094532013,
      "step": 1184
    },
    {
      "epoch": 0.02372,
      "grad_norm": 3.546875,
      "grad_norm_var": 0.06542867024739583,
      "learning_rate": 0.0001,
      "loss": 6.0688,
      "loss/crossentropy": 2.291581869125366,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3759836256504059,
      "step": 1186
    },
    {
      "epoch": 0.02376,
      "grad_norm": 3.375,
      "grad_norm_var": 0.06529541015625,
      "learning_rate": 0.0001,
      "loss": 5.6689,
      "loss/crossentropy": 1.9887789487838745,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31285202503204346,
      "step": 1188
    },
    {
      "epoch": 0.0238,
      "grad_norm": 3.109375,
      "grad_norm_var": 0.054671223958333334,
      "learning_rate": 0.0001,
      "loss": 5.7457,
      "loss/crossentropy": 2.290129065513611,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3461494445800781,
      "step": 1190
    },
    {
      "epoch": 0.02384,
      "grad_norm": 3.6875,
      "grad_norm_var": 0.0604888916015625,
      "learning_rate": 0.0001,
      "loss": 6.288,
      "loss/crossentropy": 2.3252567052841187,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3817393332719803,
      "step": 1192
    },
    {
      "epoch": 0.02388,
      "grad_norm": 3.21875,
      "grad_norm_var": 0.06230367024739583,
      "learning_rate": 0.0001,
      "loss": 5.4498,
      "loss/crossentropy": 1.736217737197876,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28928153216838837,
      "step": 1194
    },
    {
      "epoch": 0.02392,
      "grad_norm": 3.09375,
      "grad_norm_var": 0.06886393229166667,
      "learning_rate": 0.0001,
      "loss": 5.6234,
      "loss/crossentropy": 2.112093210220337,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3196643739938736,
      "step": 1196
    },
    {
      "epoch": 0.02396,
      "grad_norm": 3.15625,
      "grad_norm_var": 0.07258707682291667,
      "learning_rate": 0.0001,
      "loss": 5.3552,
      "loss/crossentropy": 2.3334370851516724,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3539857119321823,
      "step": 1198
    },
    {
      "epoch": 0.024,
      "grad_norm": 3.484375,
      "grad_norm_var": 0.09695536295572917,
      "learning_rate": 0.0001,
      "loss": 5.3293,
      "loss/crossentropy": 1.7393567562103271,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3211805671453476,
      "step": 1200
    },
    {
      "epoch": 0.02404,
      "grad_norm": 3.4375,
      "grad_norm_var": 0.07935791015625,
      "learning_rate": 0.0001,
      "loss": 5.6292,
      "loss/crossentropy": 2.314788579940796,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.37599092721939087,
      "step": 1202
    },
    {
      "epoch": 0.02408,
      "grad_norm": 3.1875,
      "grad_norm_var": 0.076318359375,
      "learning_rate": 0.0001,
      "loss": 5.435,
      "loss/crossentropy": 2.2820088863372803,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33955833315849304,
      "step": 1204
    },
    {
      "epoch": 0.02412,
      "grad_norm": 3.46875,
      "grad_norm_var": 0.07183837890625,
      "learning_rate": 0.0001,
      "loss": 5.87,
      "loss/crossentropy": 2.012476146221161,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3048281967639923,
      "step": 1206
    },
    {
      "epoch": 0.02416,
      "grad_norm": 3.1875,
      "grad_norm_var": 0.07356669108072916,
      "learning_rate": 0.0001,
      "loss": 5.6751,
      "loss/crossentropy": 2.315858483314514,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32628118991851807,
      "step": 1208
    },
    {
      "epoch": 0.0242,
      "grad_norm": 3.359375,
      "grad_norm_var": 0.14773763020833333,
      "learning_rate": 0.0001,
      "loss": 5.6813,
      "loss/crossentropy": 2.3260581493377686,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.352965384721756,
      "step": 1210
    },
    {
      "epoch": 0.02424,
      "grad_norm": 4.0625,
      "grad_norm_var": 0.1748687744140625,
      "learning_rate": 0.0001,
      "loss": 5.689,
      "loss/crossentropy": 2.199007749557495,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3089260905981064,
      "step": 1212
    },
    {
      "epoch": 0.02428,
      "grad_norm": 3.703125,
      "grad_norm_var": 1.4942047119140625,
      "learning_rate": 0.0001,
      "loss": 5.798,
      "loss/crossentropy": 2.30281138420105,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3629491478204727,
      "step": 1214
    },
    {
      "epoch": 0.02432,
      "grad_norm": 3.1875,
      "grad_norm_var": 1.5125640869140624,
      "learning_rate": 0.0001,
      "loss": 5.4927,
      "loss/crossentropy": 2.238295316696167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3436104357242584,
      "step": 1216
    },
    {
      "epoch": 0.02436,
      "grad_norm": 3.9375,
      "grad_norm_var": 1.501488240559896,
      "learning_rate": 0.0001,
      "loss": 5.5763,
      "loss/crossentropy": 2.52456271648407,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.40382860600948334,
      "step": 1218
    },
    {
      "epoch": 0.0244,
      "grad_norm": 3.46875,
      "grad_norm_var": 1.47197265625,
      "learning_rate": 0.0001,
      "loss": 5.8796,
      "loss/crossentropy": 2.4516665935516357,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.378818541765213,
      "step": 1220
    },
    {
      "epoch": 0.02444,
      "grad_norm": 3.046875,
      "grad_norm_var": 1.485480753580729,
      "learning_rate": 0.0001,
      "loss": 5.5438,
      "loss/crossentropy": 2.593857169151306,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3743816912174225,
      "step": 1222
    },
    {
      "epoch": 0.02448,
      "grad_norm": 3.375,
      "grad_norm_var": 1.4396799723307292,
      "learning_rate": 0.0001,
      "loss": 5.2807,
      "loss/crossentropy": 1.873874843120575,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2743247449398041,
      "step": 1224
    },
    {
      "epoch": 0.02452,
      "grad_norm": 3.265625,
      "grad_norm_var": 1.4642862955729166,
      "learning_rate": 0.0001,
      "loss": 5.6455,
      "loss/crossentropy": 2.1173152923583984,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.350399985909462,
      "step": 1226
    },
    {
      "epoch": 0.02456,
      "grad_norm": 3.5,
      "grad_norm_var": 1.4721018473307292,
      "learning_rate": 0.0001,
      "loss": 5.7318,
      "loss/crossentropy": 2.3650271892547607,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3241463750600815,
      "step": 1228
    },
    {
      "epoch": 0.0246,
      "grad_norm": 3.1875,
      "grad_norm_var": 0.08748270670572916,
      "learning_rate": 0.0001,
      "loss": 5.5229,
      "loss/crossentropy": 2.180622935295105,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31293927133083344,
      "step": 1230
    },
    {
      "epoch": 0.02464,
      "grad_norm": 3.546875,
      "grad_norm_var": 0.08329976399739583,
      "learning_rate": 0.0001,
      "loss": 5.7443,
      "loss/crossentropy": 2.230265259742737,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3405953049659729,
      "step": 1232
    },
    {
      "epoch": 0.02468,
      "grad_norm": 3.6875,
      "grad_norm_var": 0.07164713541666666,
      "learning_rate": 0.0001,
      "loss": 5.5518,
      "loss/crossentropy": 2.050285518169403,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32737791538238525,
      "step": 1234
    },
    {
      "epoch": 0.02472,
      "grad_norm": 3.140625,
      "grad_norm_var": 0.07604166666666666,
      "learning_rate": 0.0001,
      "loss": 5.4412,
      "loss/crossentropy": 2.2357693910598755,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34366659820079803,
      "step": 1236
    },
    {
      "epoch": 0.02476,
      "grad_norm": 3.09375,
      "grad_norm_var": 0.09081624348958334,
      "learning_rate": 0.0001,
      "loss": 5.6066,
      "loss/crossentropy": 2.308778762817383,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33825138211250305,
      "step": 1238
    },
    {
      "epoch": 0.0248,
      "grad_norm": 3.390625,
      "grad_norm_var": 0.0979888916015625,
      "learning_rate": 0.0001,
      "loss": 5.419,
      "loss/crossentropy": 2.016503393650055,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30135589838027954,
      "step": 1240
    },
    {
      "epoch": 0.02484,
      "grad_norm": 3.359375,
      "grad_norm_var": 0.0870025634765625,
      "learning_rate": 0.0001,
      "loss": 5.2494,
      "loss/crossentropy": 1.8450073599815369,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34388674795627594,
      "step": 1242
    },
    {
      "epoch": 0.02488,
      "grad_norm": 3.515625,
      "grad_norm_var": 0.05263671875,
      "learning_rate": 0.0001,
      "loss": 5.7092,
      "loss/crossentropy": 2.486730694770813,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.35796378552913666,
      "step": 1244
    },
    {
      "epoch": 0.02492,
      "grad_norm": 2.859375,
      "grad_norm_var": 0.0688385009765625,
      "learning_rate": 0.0001,
      "loss": 5.2565,
      "loss/crossentropy": 2.010675370693207,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31364670395851135,
      "step": 1246
    },
    {
      "epoch": 0.02496,
      "grad_norm": 3.15625,
      "grad_norm_var": 0.07418619791666667,
      "learning_rate": 0.0001,
      "loss": 5.366,
      "loss/crossentropy": 2.128747880458832,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3106560483574867,
      "step": 1248
    },
    {
      "epoch": 0.025,
      "grad_norm": 3.359375,
      "grad_norm_var": 0.06523335774739583,
      "learning_rate": 0.0001,
      "loss": 5.8005,
      "loss/crossentropy": 2.4563735723495483,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33769866824150085,
      "step": 1250
    },
    {
      "epoch": 0.02504,
      "grad_norm": 3.15625,
      "grad_norm_var": 0.06453450520833333,
      "learning_rate": 0.0001,
      "loss": 5.372,
      "loss/crossentropy": 2.3785592317581177,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.36167748272418976,
      "step": 1252
    },
    {
      "epoch": 0.02508,
      "grad_norm": 3.21875,
      "grad_norm_var": 0.029069010416666666,
      "learning_rate": 0.0001,
      "loss": 5.5506,
      "loss/crossentropy": 2.2642308473587036,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3315223157405853,
      "step": 1254
    },
    {
      "epoch": 0.02512,
      "grad_norm": 3.46875,
      "grad_norm_var": 0.028416951497395832,
      "learning_rate": 0.0001,
      "loss": 5.4608,
      "loss/crossentropy": 2.2246991395950317,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33083613216876984,
      "step": 1256
    },
    {
      "epoch": 0.02516,
      "grad_norm": 3.1875,
      "grad_norm_var": 0.0290435791015625,
      "learning_rate": 0.0001,
      "loss": 5.762,
      "loss/crossentropy": 2.129785656929016,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32672610878944397,
      "step": 1258
    },
    {
      "epoch": 0.0252,
      "grad_norm": 3.125,
      "grad_norm_var": 0.0243804931640625,
      "learning_rate": 0.0001,
      "loss": 5.7297,
      "loss/crossentropy": 2.0835453271865845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3504233658313751,
      "step": 1260
    },
    {
      "epoch": 0.02524,
      "grad_norm": 3.703125,
      "grad_norm_var": 0.024637858072916668,
      "learning_rate": 0.0001,
      "loss": 5.6707,
      "loss/crossentropy": 2.443893313407898,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3636191487312317,
      "step": 1262
    },
    {
      "epoch": 0.02528,
      "grad_norm": 3.21875,
      "grad_norm_var": 0.02197265625,
      "learning_rate": 0.0001,
      "loss": 5.6107,
      "loss/crossentropy": 2.367433190345764,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.35734108090400696,
      "step": 1264
    },
    {
      "epoch": 0.02532,
      "grad_norm": 3.1875,
      "grad_norm_var": 0.02662353515625,
      "learning_rate": 0.0001,
      "loss": 5.049,
      "loss/crossentropy": 1.8102024793624878,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28676700592041016,
      "step": 1266
    },
    {
      "epoch": 0.02536,
      "grad_norm": 3.0,
      "grad_norm_var": 0.03738606770833333,
      "learning_rate": 0.0001,
      "loss": 5.3916,
      "loss/crossentropy": 2.54524827003479,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3380637466907501,
      "step": 1268
    },
    {
      "epoch": 0.0254,
      "grad_norm": 2.984375,
      "grad_norm_var": 0.040913899739583336,
      "learning_rate": 0.0001,
      "loss": 5.6667,
      "loss/crossentropy": 2.4164276123046875,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.36949749290943146,
      "step": 1270
    },
    {
      "epoch": 0.02544,
      "grad_norm": 3.515625,
      "grad_norm_var": 0.043745930989583334,
      "learning_rate": 0.0001,
      "loss": 5.4235,
      "loss/crossentropy": 2.4335602521896362,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3608807325363159,
      "step": 1272
    },
    {
      "epoch": 0.02548,
      "grad_norm": 3.1875,
      "grad_norm_var": 0.0422760009765625,
      "learning_rate": 0.0001,
      "loss": 5.5858,
      "loss/crossentropy": 2.2711308002471924,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3631722033023834,
      "step": 1274
    },
    {
      "epoch": 0.02552,
      "grad_norm": 3.140625,
      "grad_norm_var": 0.0430816650390625,
      "learning_rate": 0.0001,
      "loss": 5.181,
      "loss/crossentropy": 2.378043293952942,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.35628968477249146,
      "step": 1276
    },
    {
      "epoch": 0.02556,
      "grad_norm": 3.1875,
      "grad_norm_var": 0.02431640625,
      "learning_rate": 0.0001,
      "loss": 5.5721,
      "loss/crossentropy": 1.8950039148330688,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3105602264404297,
      "step": 1278
    },
    {
      "epoch": 0.0256,
      "grad_norm": 2.96875,
      "grad_norm_var": 0.026659138997395835,
      "learning_rate": 0.0001,
      "loss": 5.4649,
      "loss/crossentropy": 1.8309656977653503,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27458132803440094,
      "step": 1280
    },
    {
      "epoch": 0.02564,
      "grad_norm": 3.40625,
      "grad_norm_var": 0.034195963541666666,
      "learning_rate": 0.0001,
      "loss": 5.993,
      "loss/crossentropy": 2.3949296474456787,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.37693680822849274,
      "step": 1282
    },
    {
      "epoch": 0.02568,
      "grad_norm": 3.140625,
      "grad_norm_var": 0.026725260416666667,
      "learning_rate": 0.0001,
      "loss": 5.6157,
      "loss/crossentropy": 2.497879147529602,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.36076007783412933,
      "step": 1284
    },
    {
      "epoch": 0.02572,
      "grad_norm": 2.953125,
      "grad_norm_var": 0.027339680989583334,
      "learning_rate": 0.0001,
      "loss": 5.354,
      "loss/crossentropy": 2.108432352542877,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3115152269601822,
      "step": 1286
    },
    {
      "epoch": 0.02576,
      "grad_norm": 3.171875,
      "grad_norm_var": 0.0201568603515625,
      "learning_rate": 0.0001,
      "loss": 5.5424,
      "loss/crossentropy": 2.079313635826111,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31575673818588257,
      "step": 1288
    },
    {
      "epoch": 0.0258,
      "grad_norm": 3.15625,
      "grad_norm_var": 0.020099894205729166,
      "learning_rate": 0.0001,
      "loss": 5.2594,
      "loss/crossentropy": 2.3390332460403442,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3293873071670532,
      "step": 1290
    },
    {
      "epoch": 0.02584,
      "grad_norm": 3.15625,
      "grad_norm_var": 0.019612630208333332,
      "learning_rate": 0.0001,
      "loss": 5.2895,
      "loss/crossentropy": 2.180980920791626,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30600421130657196,
      "step": 1292
    },
    {
      "epoch": 0.02588,
      "grad_norm": 2.9375,
      "grad_norm_var": 0.022858683268229166,
      "learning_rate": 0.0001,
      "loss": 5.2784,
      "loss/crossentropy": 2.0647836327552795,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3114248663187027,
      "step": 1294
    },
    {
      "epoch": 0.02592,
      "grad_norm": 2.84375,
      "grad_norm_var": 0.02958984375,
      "learning_rate": 0.0001,
      "loss": 5.5063,
      "loss/crossentropy": 1.931971788406372,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32757866382598877,
      "step": 1296
    },
    {
      "epoch": 0.02596,
      "grad_norm": 3.09375,
      "grad_norm_var": 0.0202301025390625,
      "learning_rate": 0.0001,
      "loss": 5.6389,
      "loss/crossentropy": 2.1180718541145325,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3272206783294678,
      "step": 1298
    },
    {
      "epoch": 0.026,
      "grad_norm": 3.4375,
      "grad_norm_var": 0.024616495768229166,
      "learning_rate": 0.0001,
      "loss": 5.5069,
      "loss/crossentropy": 1.8535473346710205,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3100028932094574,
      "step": 1300
    },
    {
      "epoch": 0.02604,
      "grad_norm": 4.5,
      "grad_norm_var": 0.16031901041666666,
      "learning_rate": 0.0001,
      "loss": 5.574,
      "loss/crossentropy": 1.9625197052955627,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3074956685304642,
      "step": 1302
    },
    {
      "epoch": 0.02608,
      "grad_norm": 3.25,
      "grad_norm_var": 0.1673736572265625,
      "learning_rate": 0.0001,
      "loss": 5.3764,
      "loss/crossentropy": 2.248521149158478,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33484284579753876,
      "step": 1304
    },
    {
      "epoch": 0.02612,
      "grad_norm": 2.921875,
      "grad_norm_var": 0.1750640869140625,
      "learning_rate": 0.0001,
      "loss": 5.8318,
      "loss/crossentropy": 2.6374051570892334,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3604440838098526,
      "step": 1306
    },
    {
      "epoch": 0.02616,
      "grad_norm": 4.0625,
      "grad_norm_var": 0.22333984375,
      "learning_rate": 0.0001,
      "loss": 5.5151,
      "loss/crossentropy": 2.3213003873825073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3659953773021698,
      "step": 1308
    },
    {
      "epoch": 0.0262,
      "grad_norm": 3.734375,
      "grad_norm_var": 0.21988525390625,
      "learning_rate": 0.0001,
      "loss": 5.9334,
      "loss/crossentropy": 2.3527311086654663,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3638792932033539,
      "step": 1310
    },
    {
      "epoch": 0.02624,
      "grad_norm": 3.140625,
      "grad_norm_var": 0.20551656087239584,
      "learning_rate": 0.0001,
      "loss": 5.4442,
      "loss/crossentropy": 1.6319801807403564,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29070258140563965,
      "step": 1312
    },
    {
      "epoch": 0.02628,
      "grad_norm": 3.125,
      "grad_norm_var": 0.215966796875,
      "learning_rate": 0.0001,
      "loss": 5.1864,
      "loss/crossentropy": 1.986265480518341,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30515219271183014,
      "step": 1314
    },
    {
      "epoch": 0.02632,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.23336181640625,
      "learning_rate": 0.0001,
      "loss": 5.5808,
      "loss/crossentropy": 2.237283766269684,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32876546680927277,
      "step": 1316
    },
    {
      "epoch": 0.02636,
      "grad_norm": 3.859375,
      "grad_norm_var": 0.13362223307291668,
      "learning_rate": 0.0001,
      "loss": 5.4868,
      "loss/crossentropy": 2.215467691421509,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34449851512908936,
      "step": 1318
    },
    {
      "epoch": 0.0264,
      "grad_norm": 3.03125,
      "grad_norm_var": 0.12567952473958333,
      "learning_rate": 0.0001,
      "loss": 5.7427,
      "loss/crossentropy": 2.264952063560486,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.387825608253479,
      "step": 1320
    },
    {
      "epoch": 0.02644,
      "grad_norm": 3.125,
      "grad_norm_var": 0.12009989420572917,
      "learning_rate": 0.0001,
      "loss": 5.5614,
      "loss/crossentropy": 2.0678945779800415,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3228907287120819,
      "step": 1322
    },
    {
      "epoch": 0.02648,
      "grad_norm": 4.40625,
      "grad_norm_var": 0.212841796875,
      "learning_rate": 0.0001,
      "loss": 5.6259,
      "loss/crossentropy": 2.1414765119552612,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31144315004348755,
      "step": 1324
    },
    {
      "epoch": 0.02652,
      "grad_norm": 4.03125,
      "grad_norm_var": 0.23813374837239584,
      "learning_rate": 0.0001,
      "loss": 5.8487,
      "loss/crossentropy": 2.3898890018463135,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3721010088920593,
      "step": 1326
    },
    {
      "epoch": 0.02656,
      "grad_norm": 3.421875,
      "grad_norm_var": 0.23483784993489584,
      "learning_rate": 0.0001,
      "loss": 5.5417,
      "loss/crossentropy": 2.404030442237854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.36486634612083435,
      "step": 1328
    },
    {
      "epoch": 0.0266,
      "grad_norm": 3.15625,
      "grad_norm_var": 0.22004801432291668,
      "learning_rate": 0.0001,
      "loss": 5.7535,
      "loss/crossentropy": 2.1436617970466614,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3091920465230942,
      "step": 1330
    },
    {
      "epoch": 0.02664,
      "grad_norm": 3.1875,
      "grad_norm_var": 0.2074615478515625,
      "learning_rate": 0.0001,
      "loss": 5.2832,
      "loss/crossentropy": 2.106055796146393,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3344803601503372,
      "step": 1332
    },
    {
      "epoch": 0.02668,
      "grad_norm": 3.640625,
      "grad_norm_var": 0.19583333333333333,
      "learning_rate": 0.0001,
      "loss": 6.0091,
      "loss/crossentropy": 2.465435266494751,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34129244089126587,
      "step": 1334
    },
    {
      "epoch": 0.02672,
      "grad_norm": 3.21875,
      "grad_norm_var": 0.18728841145833333,
      "learning_rate": 0.0001,
      "loss": 5.5569,
      "loss/crossentropy": 1.9109253883361816,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30202071368694305,
      "step": 1336
    },
    {
      "epoch": 0.02676,
      "grad_norm": 3.0,
      "grad_norm_var": 0.1890045166015625,
      "learning_rate": 0.0001,
      "loss": 5.7093,
      "loss/crossentropy": 2.267784833908081,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32984335720539093,
      "step": 1338
    },
    {
      "epoch": 0.0268,
      "grad_norm": 3.0625,
      "grad_norm_var": 0.07266337076822917,
      "learning_rate": 0.0001,
      "loss": 5.4432,
      "loss/crossentropy": 2.6131194829940796,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3797858655452728,
      "step": 1340
    },
    {
      "epoch": 0.02684,
      "grad_norm": 3.171875,
      "grad_norm_var": 0.031891886393229166,
      "learning_rate": 0.0001,
      "loss": 5.7456,
      "loss/crossentropy": 2.214663505554199,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.340796560049057,
      "step": 1342
    },
    {
      "epoch": 0.02688,
      "grad_norm": 3.0,
      "grad_norm_var": 0.03245340983072917,
      "learning_rate": 0.0001,
      "loss": 5.4205,
      "loss/crossentropy": 2.0236783027648926,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3101559728384018,
      "step": 1344
    },
    {
      "epoch": 0.02692,
      "grad_norm": 3.078125,
      "grad_norm_var": 0.030565388997395835,
      "learning_rate": 0.0001,
      "loss": 5.2671,
      "loss/crossentropy": 2.3260135650634766,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.339004784822464,
      "step": 1346
    },
    {
      "epoch": 0.02696,
      "grad_norm": 2.96875,
      "grad_norm_var": 0.0317047119140625,
      "learning_rate": 0.0001,
      "loss": 5.6529,
      "loss/crossentropy": 2.177807927131653,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3171197772026062,
      "step": 1348
    },
    {
      "epoch": 0.027,
      "grad_norm": 3.375,
      "grad_norm_var": 0.0202056884765625,
      "learning_rate": 0.0001,
      "loss": 5.4682,
      "loss/crossentropy": 2.351730227470398,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34749266505241394,
      "step": 1350
    },
    {
      "epoch": 0.02704,
      "grad_norm": 2.953125,
      "grad_norm_var": 0.017867024739583334,
      "learning_rate": 0.0001,
      "loss": 5.3692,
      "loss/crossentropy": 2.2959564924240112,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3270048499107361,
      "step": 1352
    },
    {
      "epoch": 0.02708,
      "grad_norm": 3.03125,
      "grad_norm_var": 0.016890462239583334,
      "learning_rate": 0.0001,
      "loss": 5.5243,
      "loss/crossentropy": 2.399070382118225,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32505376636981964,
      "step": 1354
    },
    {
      "epoch": 0.02712,
      "grad_norm": 3.046875,
      "grad_norm_var": 0.01812744140625,
      "learning_rate": 0.0001,
      "loss": 5.3306,
      "loss/crossentropy": 1.9084061980247498,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3073619455099106,
      "step": 1356
    },
    {
      "epoch": 0.02716,
      "grad_norm": 3.078125,
      "grad_norm_var": 0.01441650390625,
      "learning_rate": 0.0001,
      "loss": 5.3942,
      "loss/crossentropy": 2.1204254627227783,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2885961979627609,
      "step": 1358
    },
    {
      "epoch": 0.0272,
      "grad_norm": 3.28125,
      "grad_norm_var": 0.014842732747395834,
      "learning_rate": 0.0001,
      "loss": 5.7313,
      "loss/crossentropy": 2.0167239904403687,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3026747703552246,
      "step": 1360
    },
    {
      "epoch": 0.02724,
      "grad_norm": 3.171875,
      "grad_norm_var": 0.014742024739583333,
      "learning_rate": 0.0001,
      "loss": 5.3987,
      "loss/crossentropy": 1.9588357210159302,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2976878881454468,
      "step": 1362
    },
    {
      "epoch": 0.02728,
      "grad_norm": 2.796875,
      "grad_norm_var": 0.019449869791666668,
      "learning_rate": 0.0001,
      "loss": 5.1658,
      "loss/crossentropy": 1.8169561624526978,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2874959260225296,
      "step": 1364
    },
    {
      "epoch": 0.02732,
      "grad_norm": 3.1875,
      "grad_norm_var": 0.014354451497395834,
      "learning_rate": 0.0001,
      "loss": 5.5128,
      "loss/crossentropy": 2.258527398109436,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34180621802806854,
      "step": 1366
    },
    {
      "epoch": 0.02736,
      "grad_norm": 3.09375,
      "grad_norm_var": 0.017284138997395834,
      "learning_rate": 0.0001,
      "loss": 5.4374,
      "loss/crossentropy": 2.1959571838378906,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3124794214963913,
      "step": 1368
    },
    {
      "epoch": 0.0274,
      "grad_norm": 2.875,
      "grad_norm_var": 0.0198150634765625,
      "learning_rate": 0.0001,
      "loss": 5.2299,
      "loss/crossentropy": 2.1830934286117554,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3260872811079025,
      "step": 1370
    },
    {
      "epoch": 0.02744,
      "grad_norm": 3.203125,
      "grad_norm_var": 0.02017822265625,
      "learning_rate": 0.0001,
      "loss": 5.6831,
      "loss/crossentropy": 2.411653518676758,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3575899302959442,
      "step": 1372
    },
    {
      "epoch": 0.02748,
      "grad_norm": 3.140625,
      "grad_norm_var": 0.021512858072916665,
      "learning_rate": 0.0001,
      "loss": 5.3919,
      "loss/crossentropy": 2.1585222482681274,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30734311044216156,
      "step": 1374
    },
    {
      "epoch": 0.02752,
      "grad_norm": 2.90625,
      "grad_norm_var": 0.023824055989583332,
      "learning_rate": 0.0001,
      "loss": 5.37,
      "loss/crossentropy": 2.3621217012405396,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3373589664697647,
      "step": 1376
    },
    {
      "epoch": 0.02756,
      "grad_norm": 3.109375,
      "grad_norm_var": 0.023160807291666665,
      "learning_rate": 0.0001,
      "loss": 5.4712,
      "loss/crossentropy": 2.1203317046165466,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28425413370132446,
      "step": 1378
    },
    {
      "epoch": 0.0276,
      "grad_norm": 3.296875,
      "grad_norm_var": 0.0223297119140625,
      "learning_rate": 0.0001,
      "loss": 5.5438,
      "loss/crossentropy": 2.2568705081939697,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.296497106552124,
      "step": 1380
    },
    {
      "epoch": 0.02764,
      "grad_norm": 3.5625,
      "grad_norm_var": 0.03665364583333333,
      "learning_rate": 0.0001,
      "loss": 5.5613,
      "loss/crossentropy": 2.260026216506958,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34534430503845215,
      "step": 1382
    },
    {
      "epoch": 0.02768,
      "grad_norm": 5.5625,
      "grad_norm_var": 0.4100494384765625,
      "learning_rate": 0.0001,
      "loss": 5.6217,
      "loss/crossentropy": 1.9400787949562073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32853779196739197,
      "step": 1384
    },
    {
      "epoch": 0.02772,
      "grad_norm": 3.390625,
      "grad_norm_var": 0.39205322265625,
      "learning_rate": 0.0001,
      "loss": 5.2009,
      "loss/crossentropy": 2.168904423713684,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3421178460121155,
      "step": 1386
    },
    {
      "epoch": 0.02776,
      "grad_norm": 2.90625,
      "grad_norm_var": 0.405419921875,
      "learning_rate": 0.0001,
      "loss": 5.3992,
      "loss/crossentropy": 2.3474777936935425,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31982940435409546,
      "step": 1388
    },
    {
      "epoch": 0.0278,
      "grad_norm": 3.140625,
      "grad_norm_var": 0.40615234375,
      "learning_rate": 0.0001,
      "loss": 5.5791,
      "loss/crossentropy": 2.3416868448257446,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3400905281305313,
      "step": 1390
    },
    {
      "epoch": 0.02784,
      "grad_norm": 2.8125,
      "grad_norm_var": 0.4003570556640625,
      "learning_rate": 0.0001,
      "loss": 5.4413,
      "loss/crossentropy": 2.299672842025757,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3143058717250824,
      "step": 1392
    },
    {
      "epoch": 0.02788,
      "grad_norm": 3.15625,
      "grad_norm_var": 0.3945058186848958,
      "learning_rate": 0.0001,
      "loss": 5.6231,
      "loss/crossentropy": 2.3058812618255615,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33790935575962067,
      "step": 1394
    },
    {
      "epoch": 0.02792,
      "grad_norm": 3.03125,
      "grad_norm_var": 0.3947987874348958,
      "learning_rate": 0.0001,
      "loss": 5.324,
      "loss/crossentropy": 2.2137999534606934,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33156970143318176,
      "step": 1396
    },
    {
      "epoch": 0.02796,
      "grad_norm": 4.1875,
      "grad_norm_var": 0.4427571614583333,
      "learning_rate": 0.0001,
      "loss": 5.4567,
      "loss/crossentropy": 2.04184353351593,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32263143360614777,
      "step": 1398
    },
    {
      "epoch": 0.028,
      "grad_norm": 3.265625,
      "grad_norm_var": 0.09589436848958334,
      "learning_rate": 0.0001,
      "loss": 5.339,
      "loss/crossentropy": 2.0377472639083862,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3136487454175949,
      "step": 1400
    },
    {
      "epoch": 0.02804,
      "grad_norm": 3.171875,
      "grad_norm_var": 0.09228413899739583,
      "learning_rate": 0.0001,
      "loss": 5.5838,
      "loss/crossentropy": 2.5366055965423584,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3476262539625168,
      "step": 1402
    },
    {
      "epoch": 0.02808,
      "grad_norm": 3.1875,
      "grad_norm_var": 0.09063212076822917,
      "learning_rate": 0.0001,
      "loss": 5.2447,
      "loss/crossentropy": 2.088012456893921,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31111815571784973,
      "step": 1404
    },
    {
      "epoch": 0.02812,
      "grad_norm": 3.53125,
      "grad_norm_var": 0.10530598958333333,
      "learning_rate": 0.0001,
      "loss": 5.7316,
      "loss/crossentropy": 2.1750329732894897,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3138856291770935,
      "step": 1406
    },
    {
      "epoch": 0.02816,
      "grad_norm": 3.625,
      "grad_norm_var": 0.09811909993489583,
      "learning_rate": 0.0001,
      "loss": 5.4708,
      "loss/crossentropy": 1.977031648159027,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3179774433374405,
      "step": 1408
    },
    {
      "epoch": 0.0282,
      "grad_norm": 3.296875,
      "grad_norm_var": 0.12561848958333333,
      "learning_rate": 0.0001,
      "loss": 5.3815,
      "loss/crossentropy": 2.0594210028648376,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32200056314468384,
      "step": 1410
    },
    {
      "epoch": 0.02824,
      "grad_norm": 3.59375,
      "grad_norm_var": 0.11237691243489584,
      "learning_rate": 0.0001,
      "loss": 5.443,
      "loss/crossentropy": 2.3887473344802856,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3303599953651428,
      "step": 1412
    },
    {
      "epoch": 0.02828,
      "grad_norm": 3.453125,
      "grad_norm_var": 0.06500244140625,
      "learning_rate": 0.0001,
      "loss": 5.5507,
      "loss/crossentropy": 2.188898801803589,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3378629684448242,
      "step": 1414
    },
    {
      "epoch": 0.02832,
      "grad_norm": 3.171875,
      "grad_norm_var": 0.06982014973958334,
      "learning_rate": 0.0001,
      "loss": 5.458,
      "loss/crossentropy": 1.981561303138733,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2814937084913254,
      "step": 1416
    },
    {
      "epoch": 0.02836,
      "grad_norm": 3.1875,
      "grad_norm_var": 0.0756011962890625,
      "learning_rate": 0.0001,
      "loss": 5.271,
      "loss/crossentropy": 2.2141716480255127,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3429889380931854,
      "step": 1418
    },
    {
      "epoch": 0.0284,
      "grad_norm": 3.1875,
      "grad_norm_var": 0.0818267822265625,
      "learning_rate": 0.0001,
      "loss": 5.048,
      "loss/crossentropy": 2.0720977783203125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30601558089256287,
      "step": 1420
    },
    {
      "epoch": 0.02844,
      "grad_norm": 3.21875,
      "grad_norm_var": 0.07434488932291666,
      "learning_rate": 0.0001,
      "loss": 5.6407,
      "loss/crossentropy": 2.0516344904899597,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3182393014431,
      "step": 1422
    },
    {
      "epoch": 0.02848,
      "grad_norm": 3.28125,
      "grad_norm_var": 0.070166015625,
      "learning_rate": 0.0001,
      "loss": 5.6532,
      "loss/crossentropy": 2.161284327507019,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3383013904094696,
      "step": 1424
    },
    {
      "epoch": 0.02852,
      "grad_norm": 3.1875,
      "grad_norm_var": 0.03163960774739583,
      "learning_rate": 0.0001,
      "loss": 5.1594,
      "loss/crossentropy": 1.9955796599388123,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3169983923435211,
      "step": 1426
    },
    {
      "epoch": 0.02856,
      "grad_norm": 2.984375,
      "grad_norm_var": 0.023421223958333334,
      "learning_rate": 0.0001,
      "loss": 5.3951,
      "loss/crossentropy": 2.1046979427337646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30594320595264435,
      "step": 1428
    },
    {
      "epoch": 0.0286,
      "grad_norm": 2.90625,
      "grad_norm_var": 0.020231119791666665,
      "learning_rate": 0.0001,
      "loss": 5.364,
      "loss/crossentropy": 1.9611601829528809,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3039780706167221,
      "step": 1430
    },
    {
      "epoch": 0.02864,
      "grad_norm": 2.9375,
      "grad_norm_var": 0.021610514322916666,
      "learning_rate": 0.0001,
      "loss": 5.8823,
      "loss/crossentropy": 2.256209373474121,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3315662145614624,
      "step": 1432
    },
    {
      "epoch": 0.02868,
      "grad_norm": 2.9375,
      "grad_norm_var": 0.030594889322916666,
      "learning_rate": 0.0001,
      "loss": 5.2821,
      "loss/crossentropy": 2.150561034679413,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33046241104602814,
      "step": 1434
    },
    {
      "epoch": 0.02872,
      "grad_norm": 3.078125,
      "grad_norm_var": 0.024898274739583334,
      "learning_rate": 0.0001,
      "loss": 5.3609,
      "loss/crossentropy": 2.279554605484009,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3621693551540375,
      "step": 1436
    },
    {
      "epoch": 0.02876,
      "grad_norm": 3.015625,
      "grad_norm_var": 0.0259918212890625,
      "learning_rate": 0.0001,
      "loss": 5.4323,
      "loss/crossentropy": 2.0183790922164917,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33361808955669403,
      "step": 1438
    },
    {
      "epoch": 0.0288,
      "grad_norm": 4.15625,
      "grad_norm_var": 0.7352203369140625,
      "learning_rate": 0.0001,
      "loss": 5.4781,
      "loss/crossentropy": 1.964252531528473,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31638333201408386,
      "step": 1440
    },
    {
      "epoch": 0.02884,
      "grad_norm": 3.046875,
      "grad_norm_var": 0.73902587890625,
      "learning_rate": 0.0001,
      "loss": 5.6907,
      "loss/crossentropy": 1.9271634817123413,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3408525586128235,
      "step": 1442
    },
    {
      "epoch": 0.02888,
      "grad_norm": 2.953125,
      "grad_norm_var": 0.7411417643229167,
      "learning_rate": 0.0001,
      "loss": 5.4869,
      "loss/crossentropy": 2.4400887489318848,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33455249667167664,
      "step": 1444
    },
    {
      "epoch": 0.02892,
      "grad_norm": 3.078125,
      "grad_norm_var": 0.7270904541015625,
      "learning_rate": 0.0001,
      "loss": 5.6179,
      "loss/crossentropy": 2.465711832046509,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3528379648923874,
      "step": 1446
    },
    {
      "epoch": 0.02896,
      "grad_norm": 3.328125,
      "grad_norm_var": 0.7147288004557292,
      "learning_rate": 0.0001,
      "loss": 5.5125,
      "loss/crossentropy": 2.269619941711426,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32898105680942535,
      "step": 1448
    },
    {
      "epoch": 0.029,
      "grad_norm": 2.984375,
      "grad_norm_var": 0.68287353515625,
      "learning_rate": 0.0001,
      "loss": 5.5775,
      "loss/crossentropy": 2.0418076515197754,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3170912265777588,
      "step": 1450
    },
    {
      "epoch": 0.02904,
      "grad_norm": 3.15625,
      "grad_norm_var": 0.6841756184895833,
      "learning_rate": 0.0001,
      "loss": 5.5964,
      "loss/crossentropy": 2.291188359260559,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.35694004595279694,
      "step": 1452
    },
    {
      "epoch": 0.02908,
      "grad_norm": 2.75,
      "grad_norm_var": 0.7253163655598959,
      "learning_rate": 0.0001,
      "loss": 5.1608,
      "loss/crossentropy": 2.497802972793579,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3271156847476959,
      "step": 1454
    },
    {
      "epoch": 0.02912,
      "grad_norm": 2.984375,
      "grad_norm_var": 0.0334625244140625,
      "learning_rate": 0.0001,
      "loss": 5.1527,
      "loss/crossentropy": 2.161794900894165,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32509492337703705,
      "step": 1456
    },
    {
      "epoch": 0.02916,
      "grad_norm": 3.3125,
      "grad_norm_var": 0.03470052083333333,
      "learning_rate": 0.0001,
      "loss": 5.7939,
      "loss/crossentropy": 2.632015347480774,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.36461199820041656,
      "step": 1458
    },
    {
      "epoch": 0.0292,
      "grad_norm": 3.03125,
      "grad_norm_var": 0.03372395833333333,
      "learning_rate": 0.0001,
      "loss": 5.6233,
      "loss/crossentropy": 2.2478749752044678,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3311367332935333,
      "step": 1460
    },
    {
      "epoch": 0.02924,
      "grad_norm": 4.03125,
      "grad_norm_var": 0.09263916015625,
      "learning_rate": 0.0001,
      "loss": 5.2528,
      "loss/crossentropy": 1.8451723456382751,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31802159547805786,
      "step": 1462
    },
    {
      "epoch": 0.02928,
      "grad_norm": 3.71875,
      "grad_norm_var": 0.11772359212239583,
      "learning_rate": 0.0001,
      "loss": 5.6469,
      "loss/crossentropy": 2.6684207916259766,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.35483625531196594,
      "step": 1464
    },
    {
      "epoch": 0.02932,
      "grad_norm": 2.921875,
      "grad_norm_var": 0.11543680826822916,
      "learning_rate": 0.0001,
      "loss": 5.2866,
      "loss/crossentropy": 2.497614622116089,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3422156721353531,
      "step": 1466
    },
    {
      "epoch": 0.02936,
      "grad_norm": 4.25,
      "grad_norm_var": 0.19032796223958334,
      "learning_rate": 0.0001,
      "loss": 5.7424,
      "loss/crossentropy": 2.750740170478821,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.377558171749115,
      "step": 1468
    },
    {
      "epoch": 0.0294,
      "grad_norm": 2.8125,
      "grad_norm_var": 0.17550455729166667,
      "learning_rate": 0.0001,
      "loss": 5.1741,
      "loss/crossentropy": 1.9961607456207275,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3166217654943466,
      "step": 1470
    },
    {
      "epoch": 0.02944,
      "grad_norm": 3.359375,
      "grad_norm_var": 0.1655914306640625,
      "learning_rate": 0.0001,
      "loss": 5.3114,
      "loss/crossentropy": 2.0874768495559692,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2912164479494095,
      "step": 1472
    },
    {
      "epoch": 0.02948,
      "grad_norm": 2.953125,
      "grad_norm_var": 0.174072265625,
      "learning_rate": 0.0001,
      "loss": 5.2102,
      "loss/crossentropy": 2.112182080745697,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29954925179481506,
      "step": 1474
    },
    {
      "epoch": 0.02952,
      "grad_norm": 3.6875,
      "grad_norm_var": 0.4074371337890625,
      "learning_rate": 0.0001,
      "loss": 5.7839,
      "loss/crossentropy": 2.1319644451141357,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3002544492483139,
      "step": 1476
    },
    {
      "epoch": 0.02956,
      "grad_norm": 3.0625,
      "grad_norm_var": 0.37743733723958334,
      "learning_rate": 0.0001,
      "loss": 5.8305,
      "loss/crossentropy": 2.3029643297195435,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3262677788734436,
      "step": 1478
    },
    {
      "epoch": 0.0296,
      "grad_norm": 3.15625,
      "grad_norm_var": 0.3826243082682292,
      "learning_rate": 0.0001,
      "loss": 5.6803,
      "loss/crossentropy": 2.8598941564559937,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.38816460967063904,
      "step": 1480
    },
    {
      "epoch": 0.02964,
      "grad_norm": 2.984375,
      "grad_norm_var": 0.37844950358072915,
      "learning_rate": 0.0001,
      "loss": 5.2177,
      "loss/crossentropy": 2.134063720703125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3128468096256256,
      "step": 1482
    },
    {
      "epoch": 0.02968,
      "grad_norm": 3.125,
      "grad_norm_var": 0.31961263020833336,
      "learning_rate": 0.0001,
      "loss": 5.5234,
      "loss/crossentropy": 2.481287717819214,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3209179639816284,
      "step": 1484
    },
    {
      "epoch": 0.02972,
      "grad_norm": 3.3125,
      "grad_norm_var": 0.30549723307291665,
      "learning_rate": 0.0001,
      "loss": 5.3571,
      "loss/crossentropy": 2.0571895837783813,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3121480941772461,
      "step": 1486
    },
    {
      "epoch": 0.02976,
      "grad_norm": 3.203125,
      "grad_norm_var": 0.30614827473958334,
      "learning_rate": 0.0001,
      "loss": 5.2725,
      "loss/crossentropy": 2.1930073499679565,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3031492233276367,
      "step": 1488
    },
    {
      "epoch": 0.0298,
      "grad_norm": 3.046875,
      "grad_norm_var": 0.30426025390625,
      "learning_rate": 0.0001,
      "loss": 5.3256,
      "loss/crossentropy": 2.403902530670166,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32206277549266815,
      "step": 1490
    },
    {
      "epoch": 0.02984,
      "grad_norm": 2.9375,
      "grad_norm_var": 0.0538482666015625,
      "learning_rate": 0.0001,
      "loss": 5.2859,
      "loss/crossentropy": 2.131115198135376,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31651052832603455,
      "step": 1492
    },
    {
      "epoch": 0.02988,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.05826416015625,
      "learning_rate": 0.0001,
      "loss": 5.1232,
      "loss/crossentropy": 2.024750769138336,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3142934888601303,
      "step": 1494
    },
    {
      "epoch": 0.02992,
      "grad_norm": 2.984375,
      "grad_norm_var": 0.06279195149739583,
      "learning_rate": 0.0001,
      "loss": 5.3804,
      "loss/crossentropy": 1.9606707692146301,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28610387444496155,
      "step": 1496
    },
    {
      "epoch": 0.02996,
      "grad_norm": 3.15625,
      "grad_norm_var": 0.062474568684895836,
      "learning_rate": 0.0001,
      "loss": 5.5731,
      "loss/crossentropy": 2.31516432762146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32875190675258636,
      "step": 1498
    },
    {
      "epoch": 0.03,
      "grad_norm": 3.359375,
      "grad_norm_var": 0.0711822509765625,
      "learning_rate": 0.0001,
      "loss": 5.7626,
      "loss/crossentropy": 2.295942783355713,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3379499167203903,
      "step": 1500
    },
    {
      "epoch": 0.03004,
      "grad_norm": 3.21875,
      "grad_norm_var": 0.06961263020833333,
      "learning_rate": 0.0001,
      "loss": 5.5997,
      "loss/crossentropy": 2.1859577894210815,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32143887877464294,
      "step": 1502
    },
    {
      "epoch": 0.03008,
      "grad_norm": 3.28125,
      "grad_norm_var": 0.07023824055989583,
      "learning_rate": 0.0001,
      "loss": 5.4159,
      "loss/crossentropy": 2.1852502822875977,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3582807630300522,
      "step": 1504
    },
    {
      "epoch": 0.03012,
      "grad_norm": 2.75,
      "grad_norm_var": 0.07810770670572917,
      "learning_rate": 0.0001,
      "loss": 5.3011,
      "loss/crossentropy": 2.174897611141205,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33067959547042847,
      "step": 1506
    },
    {
      "epoch": 0.03016,
      "grad_norm": 2.96875,
      "grad_norm_var": 0.04057515462239583,
      "learning_rate": 0.0001,
      "loss": 5.2775,
      "loss/crossentropy": 2.24343740940094,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34563779830932617,
      "step": 1508
    },
    {
      "epoch": 0.0302,
      "grad_norm": 2.90625,
      "grad_norm_var": 0.043473307291666666,
      "learning_rate": 0.0001,
      "loss": 5.317,
      "loss/crossentropy": 1.9828822612762451,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2887475937604904,
      "step": 1510
    },
    {
      "epoch": 0.03024,
      "grad_norm": 3.0,
      "grad_norm_var": 0.03655192057291667,
      "learning_rate": 0.0001,
      "loss": 5.3172,
      "loss/crossentropy": 2.2110393047332764,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32290786504745483,
      "step": 1512
    },
    {
      "epoch": 0.03028,
      "grad_norm": 2.90625,
      "grad_norm_var": 0.04527587890625,
      "learning_rate": 0.0001,
      "loss": 5.2598,
      "loss/crossentropy": 2.3797603845596313,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3262799382209778,
      "step": 1514
    },
    {
      "epoch": 0.03032,
      "grad_norm": 3.1875,
      "grad_norm_var": 0.0281158447265625,
      "learning_rate": 0.0001,
      "loss": 5.4107,
      "loss/crossentropy": 2.0183085799217224,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2915680408477783,
      "step": 1516
    },
    {
      "epoch": 0.03036,
      "grad_norm": 3.78125,
      "grad_norm_var": 0.9888631184895833,
      "learning_rate": 0.0001,
      "loss": 5.4632,
      "loss/crossentropy": 1.875212013721466,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30001460015773773,
      "step": 1518
    },
    {
      "epoch": 0.0304,
      "grad_norm": 3.0625,
      "grad_norm_var": 0.9916300455729167,
      "learning_rate": 0.0001,
      "loss": 5.4406,
      "loss/crossentropy": 2.1000564098358154,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.354188472032547,
      "step": 1520
    },
    {
      "epoch": 0.03044,
      "grad_norm": 2.84375,
      "grad_norm_var": 0.980126953125,
      "learning_rate": 0.0001,
      "loss": 5.4837,
      "loss/crossentropy": 2.071643114089966,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.333335280418396,
      "step": 1522
    },
    {
      "epoch": 0.03048,
      "grad_norm": 3.203125,
      "grad_norm_var": 0.9749664306640625,
      "learning_rate": 0.0001,
      "loss": 5.2716,
      "loss/crossentropy": 2.4253947734832764,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31365686655044556,
      "step": 1524
    },
    {
      "epoch": 0.03052,
      "grad_norm": 3.0625,
      "grad_norm_var": 0.9605133056640625,
      "learning_rate": 0.0001,
      "loss": 5.1601,
      "loss/crossentropy": 1.967090904712677,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30677899718284607,
      "step": 1526
    },
    {
      "epoch": 0.03056,
      "grad_norm": 3.15625,
      "grad_norm_var": 0.954443359375,
      "learning_rate": 0.0001,
      "loss": 5.0971,
      "loss/crossentropy": 2.112701952457428,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3029911667108536,
      "step": 1528
    },
    {
      "epoch": 0.0306,
      "grad_norm": 4.5625,
      "grad_norm_var": 1.0084269205729166,
      "learning_rate": 0.0001,
      "loss": 5.6836,
      "loss/crossentropy": 2.5657063722610474,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3484763503074646,
      "step": 1530
    },
    {
      "epoch": 0.03064,
      "grad_norm": 3.234375,
      "grad_norm_var": 1.110497029622396,
      "learning_rate": 0.0001,
      "loss": 5.3888,
      "loss/crossentropy": 2.1214585304260254,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3243858218193054,
      "step": 1532
    },
    {
      "epoch": 0.03068,
      "grad_norm": 3.234375,
      "grad_norm_var": 0.35461832682291666,
      "learning_rate": 0.0001,
      "loss": 5.4662,
      "loss/crossentropy": 2.427902936935425,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3428474962711334,
      "step": 1534
    },
    {
      "epoch": 0.03072,
      "grad_norm": 3.078125,
      "grad_norm_var": 0.36741434733072914,
      "learning_rate": 0.0001,
      "loss": 5.2956,
      "loss/crossentropy": 1.975690484046936,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3136949688196182,
      "step": 1536
    },
    {
      "epoch": 0.03076,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.38342692057291666,
      "learning_rate": 0.0001,
      "loss": 5.1233,
      "loss/crossentropy": 2.295845150947571,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31703390181064606,
      "step": 1538
    },
    {
      "epoch": 0.0308,
      "grad_norm": 3.171875,
      "grad_norm_var": 0.37892252604166665,
      "learning_rate": 0.0001,
      "loss": 5.8286,
      "loss/crossentropy": 2.117497444152832,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32601243257522583,
      "step": 1540
    },
    {
      "epoch": 0.03084,
      "grad_norm": 3.140625,
      "grad_norm_var": 0.49339192708333335,
      "learning_rate": 0.0001,
      "loss": 5.4065,
      "loss/crossentropy": 2.3824862241744995,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3332519829273224,
      "step": 1542
    },
    {
      "epoch": 0.03088,
      "grad_norm": 3.140625,
      "grad_norm_var": 0.4940582275390625,
      "learning_rate": 0.0001,
      "loss": 5.0672,
      "loss/crossentropy": 1.9037857055664062,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3143462985754013,
      "step": 1544
    },
    {
      "epoch": 0.03092,
      "grad_norm": 3.09375,
      "grad_norm_var": 0.4064737955729167,
      "learning_rate": 0.0001,
      "loss": 5.2328,
      "loss/crossentropy": 1.9191133379936218,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29443541169166565,
      "step": 1546
    },
    {
      "epoch": 0.03096,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.28172200520833335,
      "learning_rate": 0.0001,
      "loss": 5.3402,
      "loss/crossentropy": 2.216462254524231,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31411711871623993,
      "step": 1548
    },
    {
      "epoch": 0.031,
      "grad_norm": 2.96875,
      "grad_norm_var": 0.227685546875,
      "learning_rate": 0.0001,
      "loss": 5.0455,
      "loss/crossentropy": 1.8064754605293274,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2667318135499954,
      "step": 1550
    },
    {
      "epoch": 0.03104,
      "grad_norm": 3.453125,
      "grad_norm_var": 0.46340738932291664,
      "learning_rate": 0.0001,
      "loss": 5.5672,
      "loss/crossentropy": 2.488176465034485,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3422655761241913,
      "step": 1552
    },
    {
      "epoch": 0.03108,
      "grad_norm": 2.859375,
      "grad_norm_var": 0.4529205322265625,
      "learning_rate": 0.0001,
      "loss": 5.4404,
      "loss/crossentropy": 2.5670164823532104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3462950587272644,
      "step": 1554
    },
    {
      "epoch": 0.03112,
      "grad_norm": 3.25,
      "grad_norm_var": 0.46499735514322915,
      "learning_rate": 0.0001,
      "loss": 5.0295,
      "loss/crossentropy": 2.0630581378936768,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29241877794265747,
      "step": 1556
    },
    {
      "epoch": 0.03116,
      "grad_norm": 2.84375,
      "grad_norm_var": 0.30891011555989584,
      "learning_rate": 0.0001,
      "loss": 5.4751,
      "loss/crossentropy": 2.685954213142395,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.37143297493457794,
      "step": 1558
    },
    {
      "epoch": 0.0312,
      "grad_norm": 3.421875,
      "grad_norm_var": 0.311669921875,
      "learning_rate": 0.0001,
      "loss": 5.6995,
      "loss/crossentropy": 1.9786988496780396,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.40276341140270233,
      "step": 1560
    },
    {
      "epoch": 0.03124,
      "grad_norm": 3.0,
      "grad_norm_var": 0.31383056640625,
      "learning_rate": 0.0001,
      "loss": 5.6695,
      "loss/crossentropy": 2.1484411358833313,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3264364004135132,
      "step": 1562
    },
    {
      "epoch": 0.03128,
      "grad_norm": 2.875,
      "grad_norm_var": 0.3103424072265625,
      "learning_rate": 0.0001,
      "loss": 5.3015,
      "loss/crossentropy": 2.1411852836608887,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3156583160161972,
      "step": 1564
    },
    {
      "epoch": 0.03132,
      "grad_norm": 2.90625,
      "grad_norm_var": 0.31302083333333336,
      "learning_rate": 0.0001,
      "loss": 5.2911,
      "loss/crossentropy": 2.1509006023406982,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3066476732492447,
      "step": 1566
    },
    {
      "epoch": 0.03136,
      "grad_norm": 2.90625,
      "grad_norm_var": 0.0416412353515625,
      "learning_rate": 0.0001,
      "loss": 5.1904,
      "loss/crossentropy": 1.7540676593780518,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2705047130584717,
      "step": 1568
    },
    {
      "epoch": 0.0314,
      "grad_norm": 3.046875,
      "grad_norm_var": 0.0431640625,
      "learning_rate": 0.0001,
      "loss": 4.9637,
      "loss/crossentropy": 2.2091184854507446,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2920738309621811,
      "step": 1570
    },
    {
      "epoch": 0.03144,
      "grad_norm": 2.984375,
      "grad_norm_var": 0.045735677083333336,
      "learning_rate": 0.0001,
      "loss": 5.6048,
      "loss/crossentropy": 1.8405091762542725,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2981649935245514,
      "step": 1572
    },
    {
      "epoch": 0.03148,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.0420806884765625,
      "learning_rate": 0.0001,
      "loss": 5.4258,
      "loss/crossentropy": 2.2292457818984985,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30654460191726685,
      "step": 1574
    },
    {
      "epoch": 0.03152,
      "grad_norm": 4.21875,
      "grad_norm_var": 0.12868550618489583,
      "learning_rate": 0.0001,
      "loss": 5.8439,
      "loss/crossentropy": 2.653907895088196,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.35303865373134613,
      "step": 1576
    },
    {
      "epoch": 0.03156,
      "grad_norm": 3.234375,
      "grad_norm_var": 0.13888346354166667,
      "learning_rate": 0.0001,
      "loss": 5.6145,
      "loss/crossentropy": 2.6799376010894775,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.37383997440338135,
      "step": 1578
    },
    {
      "epoch": 0.0316,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.14431864420572918,
      "learning_rate": 0.0001,
      "loss": 5.3321,
      "loss/crossentropy": 2.3438535928726196,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30706922709941864,
      "step": 1580
    },
    {
      "epoch": 0.03164,
      "grad_norm": 2.96875,
      "grad_norm_var": 0.14254150390625,
      "learning_rate": 0.0001,
      "loss": 5.2139,
      "loss/crossentropy": 2.1885964274406433,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30707718431949615,
      "step": 1582
    },
    {
      "epoch": 0.03168,
      "grad_norm": 2.984375,
      "grad_norm_var": 0.14042561848958332,
      "learning_rate": 0.0001,
      "loss": 5.1661,
      "loss/crossentropy": 2.0471584796905518,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31723180413246155,
      "step": 1584
    },
    {
      "epoch": 0.03172,
      "grad_norm": 3.0,
      "grad_norm_var": 0.1363433837890625,
      "learning_rate": 0.0001,
      "loss": 5.227,
      "loss/crossentropy": 2.135176420211792,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29839280247688293,
      "step": 1586
    },
    {
      "epoch": 0.03176,
      "grad_norm": 3.3125,
      "grad_norm_var": 0.13482666015625,
      "learning_rate": 0.0001,
      "loss": 5.6412,
      "loss/crossentropy": 2.4444775581359863,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33574284613132477,
      "step": 1588
    },
    {
      "epoch": 0.0318,
      "grad_norm": 3.140625,
      "grad_norm_var": 0.12195536295572916,
      "learning_rate": 0.0001,
      "loss": 5.4289,
      "loss/crossentropy": 2.1725653409957886,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.310588076710701,
      "step": 1590
    },
    {
      "epoch": 0.03184,
      "grad_norm": 2.875,
      "grad_norm_var": 0.04572652180989583,
      "learning_rate": 0.0001,
      "loss": 5.3727,
      "loss/crossentropy": 2.3610929250717163,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32158929109573364,
      "step": 1592
    },
    {
      "epoch": 0.03188,
      "grad_norm": 2.84375,
      "grad_norm_var": 0.0502105712890625,
      "learning_rate": 0.0001,
      "loss": 4.8794,
      "loss/crossentropy": 1.9271156787872314,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28283432126045227,
      "step": 1594
    },
    {
      "epoch": 0.03192,
      "grad_norm": 3.203125,
      "grad_norm_var": 0.04372456868489583,
      "learning_rate": 0.0001,
      "loss": 5.3912,
      "loss/crossentropy": 2.4196890592575073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3463610112667084,
      "step": 1596
    },
    {
      "epoch": 0.03196,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.0500152587890625,
      "learning_rate": 0.0001,
      "loss": 5.1524,
      "loss/crossentropy": 2.207236647605896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33829018473625183,
      "step": 1598
    },
    {
      "epoch": 0.032,
      "grad_norm": 3.0625,
      "grad_norm_var": 0.04101155598958333,
      "learning_rate": 0.0001,
      "loss": 5.4724,
      "loss/crossentropy": 2.3757678270339966,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3342677056789398,
      "step": 1600
    },
    {
      "epoch": 0.03204,
      "grad_norm": 2.921875,
      "grad_norm_var": 0.04512430826822917,
      "learning_rate": 0.0001,
      "loss": 5.1763,
      "loss/crossentropy": 2.2605016231536865,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3062159866094589,
      "step": 1602
    },
    {
      "epoch": 0.03208,
      "grad_norm": 3.359375,
      "grad_norm_var": 0.04794921875,
      "learning_rate": 0.0001,
      "loss": 5.7139,
      "loss/crossentropy": 2.4536768198013306,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.36668023467063904,
      "step": 1604
    },
    {
      "epoch": 0.03212,
      "grad_norm": 3.09375,
      "grad_norm_var": 0.0523834228515625,
      "learning_rate": 0.0001,
      "loss": 5.137,
      "loss/crossentropy": 1.9870036244392395,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2974477708339691,
      "step": 1606
    },
    {
      "epoch": 0.03216,
      "grad_norm": 3.015625,
      "grad_norm_var": 0.039876302083333336,
      "learning_rate": 0.0001,
      "loss": 5.3926,
      "loss/crossentropy": 2.1852506399154663,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3327452540397644,
      "step": 1608
    },
    {
      "epoch": 0.0322,
      "grad_norm": 3.140625,
      "grad_norm_var": 0.03128255208333333,
      "learning_rate": 0.0001,
      "loss": 5.4964,
      "loss/crossentropy": 2.2226197719573975,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3110218793153763,
      "step": 1610
    },
    {
      "epoch": 0.03224,
      "grad_norm": 3.09375,
      "grad_norm_var": 0.029523722330729165,
      "learning_rate": 0.0001,
      "loss": 5.398,
      "loss/crossentropy": 1.8255922198295593,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28216174244880676,
      "step": 1612
    },
    {
      "epoch": 0.03228,
      "grad_norm": 2.96875,
      "grad_norm_var": 0.023200480143229167,
      "learning_rate": 0.0001,
      "loss": 5.2777,
      "loss/crossentropy": 1.9663920998573303,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3095496743917465,
      "step": 1614
    },
    {
      "epoch": 0.03232,
      "grad_norm": 2.953125,
      "grad_norm_var": 0.023908487955729165,
      "learning_rate": 0.0001,
      "loss": 5.1578,
      "loss/crossentropy": 2.2089942693710327,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3307983875274658,
      "step": 1616
    },
    {
      "epoch": 0.03236,
      "grad_norm": 3.03125,
      "grad_norm_var": 0.01793212890625,
      "learning_rate": 0.0001,
      "loss": 5.3331,
      "loss/crossentropy": 2.261039137840271,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.314799427986145,
      "step": 1618
    },
    {
      "epoch": 0.0324,
      "grad_norm": 2.859375,
      "grad_norm_var": 0.015555826822916667,
      "learning_rate": 0.0001,
      "loss": 5.1674,
      "loss/crossentropy": 2.350824236869812,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3515756279230118,
      "step": 1620
    },
    {
      "epoch": 0.03244,
      "grad_norm": 2.9375,
      "grad_norm_var": 0.010749308268229167,
      "learning_rate": 0.0001,
      "loss": 5.4853,
      "loss/crossentropy": 2.2964736223220825,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3059935122728348,
      "step": 1622
    },
    {
      "epoch": 0.03248,
      "grad_norm": 2.859375,
      "grad_norm_var": 0.016597493489583334,
      "learning_rate": 0.0001,
      "loss": 5.2503,
      "loss/crossentropy": 2.36459481716156,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33763329684734344,
      "step": 1624
    },
    {
      "epoch": 0.03252,
      "grad_norm": 2.984375,
      "grad_norm_var": 0.0126617431640625,
      "learning_rate": 0.0001,
      "loss": 5.2242,
      "loss/crossentropy": 2.165920853614807,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31318019330501556,
      "step": 1626
    },
    {
      "epoch": 0.03256,
      "grad_norm": 2.859375,
      "grad_norm_var": 0.010628255208333333,
      "learning_rate": 0.0001,
      "loss": 5.2544,
      "loss/crossentropy": 2.326790690422058,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3044355511665344,
      "step": 1628
    },
    {
      "epoch": 0.0326,
      "grad_norm": 2.96875,
      "grad_norm_var": 0.01129150390625,
      "learning_rate": 0.0001,
      "loss": 5.3369,
      "loss/crossentropy": 1.8848688006401062,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.306812584400177,
      "step": 1630
    },
    {
      "epoch": 0.03264,
      "grad_norm": 2.953125,
      "grad_norm_var": 0.0134185791015625,
      "learning_rate": 0.0001,
      "loss": 5.351,
      "loss/crossentropy": 2.045863091945648,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3091462701559067,
      "step": 1632
    },
    {
      "epoch": 0.03268,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.01148681640625,
      "learning_rate": 0.0001,
      "loss": 4.9742,
      "loss/crossentropy": 2.0707273483276367,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3038959503173828,
      "step": 1634
    },
    {
      "epoch": 0.03272,
      "grad_norm": 2.859375,
      "grad_norm_var": 0.011237589518229167,
      "learning_rate": 0.0001,
      "loss": 5.2076,
      "loss/crossentropy": 2.0786932706832886,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29513464868068695,
      "step": 1636
    },
    {
      "epoch": 0.03276,
      "grad_norm": 3.203125,
      "grad_norm_var": 0.01900634765625,
      "learning_rate": 0.0001,
      "loss": 5.4237,
      "loss/crossentropy": 2.1527108550071716,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32102274894714355,
      "step": 1638
    },
    {
      "epoch": 0.0328,
      "grad_norm": 3.15625,
      "grad_norm_var": 0.021361287434895834,
      "learning_rate": 0.0001,
      "loss": 5.126,
      "loss/crossentropy": 2.0383081436157227,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3253529220819473,
      "step": 1640
    },
    {
      "epoch": 0.03284,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.021809895833333332,
      "learning_rate": 0.0001,
      "loss": 5.0775,
      "loss/crossentropy": 2.2801902294158936,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32200203835964203,
      "step": 1642
    },
    {
      "epoch": 0.03288,
      "grad_norm": 2.953125,
      "grad_norm_var": 0.022184244791666665,
      "learning_rate": 0.0001,
      "loss": 5.3035,
      "loss/crossentropy": 2.164717435836792,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2997436225414276,
      "step": 1644
    },
    {
      "epoch": 0.03292,
      "grad_norm": 3.1875,
      "grad_norm_var": 0.025191243489583334,
      "learning_rate": 0.0001,
      "loss": 5.5166,
      "loss/crossentropy": 2.389414072036743,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32998231053352356,
      "step": 1646
    },
    {
      "epoch": 0.03296,
      "grad_norm": 2.921875,
      "grad_norm_var": 0.023949178059895833,
      "learning_rate": 0.0001,
      "loss": 5.3225,
      "loss/crossentropy": 2.09418523311615,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33829881250858307,
      "step": 1648
    },
    {
      "epoch": 0.033,
      "grad_norm": 3.109375,
      "grad_norm_var": 0.021800740559895834,
      "learning_rate": 0.0001,
      "loss": 5.261,
      "loss/crossentropy": 2.324030637741089,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30337512493133545,
      "step": 1650
    },
    {
      "epoch": 0.03304,
      "grad_norm": 3.109375,
      "grad_norm_var": 0.020992024739583334,
      "learning_rate": 0.0001,
      "loss": 5.4162,
      "loss/crossentropy": 1.8635556101799011,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27077023684978485,
      "step": 1652
    },
    {
      "epoch": 0.03308,
      "grad_norm": 2.8125,
      "grad_norm_var": 0.0215972900390625,
      "learning_rate": 0.0001,
      "loss": 5.2371,
      "loss/crossentropy": 2.1776190996170044,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30546560883522034,
      "step": 1654
    },
    {
      "epoch": 0.03312,
      "grad_norm": 3.1875,
      "grad_norm_var": 0.021240234375,
      "learning_rate": 0.0001,
      "loss": 5.1721,
      "loss/crossentropy": 2.1003682613372803,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3304767310619354,
      "step": 1656
    },
    {
      "epoch": 0.03316,
      "grad_norm": 3.203125,
      "grad_norm_var": 0.02584228515625,
      "learning_rate": 0.0001,
      "loss": 5.8029,
      "loss/crossentropy": 2.4331823587417603,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3653264045715332,
      "step": 1658
    },
    {
      "epoch": 0.0332,
      "grad_norm": 3.125,
      "grad_norm_var": 0.1084381103515625,
      "learning_rate": 0.0001,
      "loss": 5.7157,
      "loss/crossentropy": 1.9939777851104736,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28948865830898285,
      "step": 1660
    },
    {
      "epoch": 0.03324,
      "grad_norm": 2.9375,
      "grad_norm_var": 0.10896809895833333,
      "learning_rate": 0.0001,
      "loss": 4.933,
      "loss/crossentropy": 1.9093859791755676,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3007328063249588,
      "step": 1662
    },
    {
      "epoch": 0.03328,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.11998697916666666,
      "learning_rate": 0.0001,
      "loss": 5.1861,
      "loss/crossentropy": 2.2847355604171753,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2975463569164276,
      "step": 1664
    },
    {
      "epoch": 0.03332,
      "grad_norm": 3.046875,
      "grad_norm_var": 0.120166015625,
      "learning_rate": 0.0001,
      "loss": 5.3623,
      "loss/crossentropy": 2.0081310868263245,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28596948087215424,
      "step": 1666
    },
    {
      "epoch": 0.03336,
      "grad_norm": 3.125,
      "grad_norm_var": 0.11974283854166666,
      "learning_rate": 0.0001,
      "loss": 5.2338,
      "loss/crossentropy": 2.189584493637085,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30838510394096375,
      "step": 1668
    },
    {
      "epoch": 0.0334,
      "grad_norm": 2.796875,
      "grad_norm_var": 0.12802327473958333,
      "learning_rate": 0.0001,
      "loss": 5.2749,
      "loss/crossentropy": 2.204169988632202,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3091500401496887,
      "step": 1670
    },
    {
      "epoch": 0.03344,
      "grad_norm": 2.75,
      "grad_norm_var": 0.1338775634765625,
      "learning_rate": 0.0001,
      "loss": 5.2755,
      "loss/crossentropy": 2.195966899394989,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30603383481502533,
      "step": 1672
    },
    {
      "epoch": 0.03348,
      "grad_norm": 3.0,
      "grad_norm_var": 0.1251953125,
      "learning_rate": 0.0001,
      "loss": 5.8768,
      "loss/crossentropy": 2.5402153730392456,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3657945841550827,
      "step": 1674
    },
    {
      "epoch": 0.03352,
      "grad_norm": 3.09375,
      "grad_norm_var": 0.0277252197265625,
      "learning_rate": 0.0001,
      "loss": 5.5387,
      "loss/crossentropy": 2.1721729040145874,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29539716243743896,
      "step": 1676
    },
    {
      "epoch": 0.03356,
      "grad_norm": 2.953125,
      "grad_norm_var": 0.027253214518229166,
      "learning_rate": 0.0001,
      "loss": 5.2355,
      "loss/crossentropy": 2.0591543912887573,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29050062596797943,
      "step": 1678
    },
    {
      "epoch": 0.0336,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.025846354166666665,
      "learning_rate": 0.0001,
      "loss": 5.4895,
      "loss/crossentropy": 2.1396639347076416,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3261077404022217,
      "step": 1680
    },
    {
      "epoch": 0.03364,
      "grad_norm": 3.015625,
      "grad_norm_var": 0.024689737955729166,
      "learning_rate": 0.0001,
      "loss": 5.2094,
      "loss/crossentropy": 1.9553123712539673,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3204822689294815,
      "step": 1682
    },
    {
      "epoch": 0.03368,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.023746744791666666,
      "learning_rate": 0.0001,
      "loss": 5.3417,
      "loss/crossentropy": 2.4139883518218994,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32602658867836,
      "step": 1684
    },
    {
      "epoch": 0.03372,
      "grad_norm": 2.90625,
      "grad_norm_var": 0.019624837239583335,
      "learning_rate": 0.0001,
      "loss": 5.3479,
      "loss/crossentropy": 1.9849293231964111,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32610756158828735,
      "step": 1686
    },
    {
      "epoch": 0.03376,
      "grad_norm": 2.953125,
      "grad_norm_var": 0.008622233072916667,
      "learning_rate": 0.0001,
      "loss": 5.6218,
      "loss/crossentropy": 2.698970675468445,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3749641329050064,
      "step": 1688
    },
    {
      "epoch": 0.0338,
      "grad_norm": 3.0625,
      "grad_norm_var": 0.01285400390625,
      "learning_rate": 0.0001,
      "loss": 5.7617,
      "loss/crossentropy": 2.715620517730713,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.35715436935424805,
      "step": 1690
    },
    {
      "epoch": 0.03384,
      "grad_norm": 3.0,
      "grad_norm_var": 0.01201171875,
      "learning_rate": 0.0001,
      "loss": 5.5073,
      "loss/crossentropy": 2.7213666439056396,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34585659205913544,
      "step": 1692
    },
    {
      "epoch": 0.03388,
      "grad_norm": 2.84375,
      "grad_norm_var": 0.016044108072916667,
      "learning_rate": 0.0001,
      "loss": 5.2674,
      "loss/crossentropy": 2.277606725692749,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3140410780906677,
      "step": 1694
    },
    {
      "epoch": 0.03392,
      "grad_norm": 3.765625,
      "grad_norm_var": 0.05364176432291667,
      "learning_rate": 0.0001,
      "loss": 5.2633,
      "loss/crossentropy": 2.332197904586792,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30997559428215027,
      "step": 1696
    },
    {
      "epoch": 0.03396,
      "grad_norm": 2.953125,
      "grad_norm_var": 0.05464579264322917,
      "learning_rate": 0.0001,
      "loss": 5.4323,
      "loss/crossentropy": 2.4413230419158936,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3075388967990875,
      "step": 1698
    },
    {
      "epoch": 0.034,
      "grad_norm": 2.859375,
      "grad_norm_var": 0.0567535400390625,
      "learning_rate": 0.0001,
      "loss": 5.1099,
      "loss/crossentropy": 2.2601696252822876,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3050367534160614,
      "step": 1700
    },
    {
      "epoch": 0.03404,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.06572977701822917,
      "learning_rate": 0.0001,
      "loss": 4.9925,
      "loss/crossentropy": 2.2910414934158325,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3248990923166275,
      "step": 1702
    },
    {
      "epoch": 0.03408,
      "grad_norm": 2.890625,
      "grad_norm_var": 0.06843159993489584,
      "learning_rate": 0.0001,
      "loss": 5.3814,
      "loss/crossentropy": 1.9898682832717896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27573561668395996,
      "step": 1704
    },
    {
      "epoch": 0.03412,
      "grad_norm": 2.96875,
      "grad_norm_var": 0.06398824055989584,
      "learning_rate": 0.0001,
      "loss": 5.1506,
      "loss/crossentropy": 2.1234602332115173,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30724021792411804,
      "step": 1706
    },
    {
      "epoch": 0.03416,
      "grad_norm": 3.390625,
      "grad_norm_var": 0.0764801025390625,
      "learning_rate": 0.0001,
      "loss": 5.5433,
      "loss/crossentropy": 2.34807026386261,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32687780261039734,
      "step": 1708
    },
    {
      "epoch": 0.0342,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.07669169108072917,
      "learning_rate": 0.0001,
      "loss": 5.1249,
      "loss/crossentropy": 2.17264860868454,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3086177706718445,
      "step": 1710
    },
    {
      "epoch": 0.03424,
      "grad_norm": 2.96875,
      "grad_norm_var": 0.0303131103515625,
      "learning_rate": 0.0001,
      "loss": 5.3613,
      "loss/crossentropy": 2.1094497442245483,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29483039677143097,
      "step": 1712
    },
    {
      "epoch": 0.03428,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.03284505208333333,
      "learning_rate": 0.0001,
      "loss": 5.256,
      "loss/crossentropy": 2.2379074692726135,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2799292802810669,
      "step": 1714
    },
    {
      "epoch": 0.03432,
      "grad_norm": 3.078125,
      "grad_norm_var": 0.21389567057291667,
      "learning_rate": 0.0001,
      "loss": 5.5834,
      "loss/crossentropy": 2.4616905450820923,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33721986413002014,
      "step": 1716
    },
    {
      "epoch": 0.03436,
      "grad_norm": 2.953125,
      "grad_norm_var": 0.2042877197265625,
      "learning_rate": 0.0001,
      "loss": 5.4415,
      "loss/crossentropy": 2.383226990699768,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3242805302143097,
      "step": 1718
    },
    {
      "epoch": 0.0344,
      "grad_norm": 2.921875,
      "grad_norm_var": 0.19931538899739584,
      "learning_rate": 0.0001,
      "loss": 5.5514,
      "loss/crossentropy": 2.495948314666748,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3131762146949768,
      "step": 1720
    },
    {
      "epoch": 0.03444,
      "grad_norm": 3.109375,
      "grad_norm_var": 0.19524739583333334,
      "learning_rate": 0.0001,
      "loss": 5.6767,
      "loss/crossentropy": 2.1921653747558594,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.320631667971611,
      "step": 1722
    },
    {
      "epoch": 0.03448,
      "grad_norm": 3.40625,
      "grad_norm_var": 0.20221354166666666,
      "learning_rate": 0.0001,
      "loss": 5.2,
      "loss/crossentropy": 2.0795475840568542,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29602357745170593,
      "step": 1724
    },
    {
      "epoch": 0.03452,
      "grad_norm": 2.75,
      "grad_norm_var": 0.201171875,
      "learning_rate": 0.0001,
      "loss": 4.7489,
      "loss/crossentropy": 1.911207377910614,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.287412166595459,
      "step": 1726
    },
    {
      "epoch": 0.03456,
      "grad_norm": 2.96875,
      "grad_norm_var": 0.20129801432291666,
      "learning_rate": 0.0001,
      "loss": 5.155,
      "loss/crossentropy": 2.0638214349746704,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32240423560142517,
      "step": 1728
    },
    {
      "epoch": 0.0346,
      "grad_norm": 2.75,
      "grad_norm_var": 0.20066630045572917,
      "learning_rate": 0.0001,
      "loss": 5.3464,
      "loss/crossentropy": 2.355573534965515,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3161381185054779,
      "step": 1730
    },
    {
      "epoch": 0.03464,
      "grad_norm": 2.8125,
      "grad_norm_var": 0.04185282389322917,
      "learning_rate": 0.0001,
      "loss": 5.1039,
      "loss/crossentropy": 2.2227123975753784,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31909704208374023,
      "step": 1732
    },
    {
      "epoch": 0.03468,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.03954671223958333,
      "learning_rate": 0.0001,
      "loss": 5.2249,
      "loss/crossentropy": 2.203602910041809,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3238847255706787,
      "step": 1734
    },
    {
      "epoch": 0.03472,
      "grad_norm": 3.34375,
      "grad_norm_var": 0.07062174479166666,
      "learning_rate": 0.0001,
      "loss": 5.0623,
      "loss/crossentropy": 2.2696332335472107,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3174774497747421,
      "step": 1736
    },
    {
      "epoch": 0.03476,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.07385660807291666,
      "learning_rate": 0.0001,
      "loss": 5.0469,
      "loss/crossentropy": 1.8124465942382812,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2645348161458969,
      "step": 1738
    },
    {
      "epoch": 0.0348,
      "grad_norm": 3.015625,
      "grad_norm_var": 0.0578765869140625,
      "learning_rate": 0.0001,
      "loss": 5.589,
      "loss/crossentropy": 2.1951464414596558,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31606370210647583,
      "step": 1740
    },
    {
      "epoch": 0.03484,
      "grad_norm": 3.109375,
      "grad_norm_var": 0.05624593098958333,
      "learning_rate": 0.0001,
      "loss": 5.4867,
      "loss/crossentropy": 2.4413230419158936,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33688417077064514,
      "step": 1742
    },
    {
      "epoch": 0.03488,
      "grad_norm": 3.09375,
      "grad_norm_var": 0.05705973307291667,
      "learning_rate": 0.0001,
      "loss": 5.7898,
      "loss/crossentropy": 2.1357219219207764,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3476516157388687,
      "step": 1744
    },
    {
      "epoch": 0.03492,
      "grad_norm": 2.859375,
      "grad_norm_var": 0.0551666259765625,
      "learning_rate": 0.0001,
      "loss": 5.4465,
      "loss/crossentropy": 2.1557592153549194,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3080083876848221,
      "step": 1746
    },
    {
      "epoch": 0.03496,
      "grad_norm": 2.859375,
      "grad_norm_var": 0.0512603759765625,
      "learning_rate": 0.0001,
      "loss": 5.4949,
      "loss/crossentropy": 2.3549705743789673,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33814217150211334,
      "step": 1748
    },
    {
      "epoch": 0.035,
      "grad_norm": 3.109375,
      "grad_norm_var": 0.04348958333333333,
      "learning_rate": 0.0001,
      "loss": 5.5881,
      "loss/crossentropy": 2.382844924926758,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31944599747657776,
      "step": 1750
    },
    {
      "epoch": 0.03504,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.02138671875,
      "learning_rate": 0.0001,
      "loss": 5.2937,
      "loss/crossentropy": 2.3312920331954956,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3346693813800812,
      "step": 1752
    },
    {
      "epoch": 0.03508,
      "grad_norm": 3.015625,
      "grad_norm_var": 0.019677734375,
      "learning_rate": 0.0001,
      "loss": 5.1981,
      "loss/crossentropy": 2.1921491026878357,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31133508682250977,
      "step": 1754
    },
    {
      "epoch": 0.03512,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.024494425455729166,
      "learning_rate": 0.0001,
      "loss": 4.8796,
      "loss/crossentropy": 2.0229611992836,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26832816004753113,
      "step": 1756
    },
    {
      "epoch": 0.03516,
      "grad_norm": 2.75,
      "grad_norm_var": 0.025764973958333333,
      "learning_rate": 0.0001,
      "loss": 5.0322,
      "loss/crossentropy": 1.8138108849525452,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2576165944337845,
      "step": 1758
    },
    {
      "epoch": 0.0352,
      "grad_norm": 2.75,
      "grad_norm_var": 0.0232574462890625,
      "learning_rate": 0.0001,
      "loss": 5.1276,
      "loss/crossentropy": 2.0019126534461975,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28094957768917084,
      "step": 1760
    },
    {
      "epoch": 0.03524,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.022554524739583335,
      "learning_rate": 0.0001,
      "loss": 5.1776,
      "loss/crossentropy": 2.400240898132324,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30936548113822937,
      "step": 1762
    },
    {
      "epoch": 0.03528,
      "grad_norm": 3.328125,
      "grad_norm_var": 0.03495992024739583,
      "learning_rate": 0.0001,
      "loss": 4.9401,
      "loss/crossentropy": 2.019958734512329,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27741560339927673,
      "step": 1764
    },
    {
      "epoch": 0.03532,
      "grad_norm": 2.875,
      "grad_norm_var": 0.025032552083333333,
      "learning_rate": 0.0001,
      "loss": 5.0912,
      "loss/crossentropy": 1.9099596738815308,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28769225627183914,
      "step": 1766
    },
    {
      "epoch": 0.03536,
      "grad_norm": 2.890625,
      "grad_norm_var": 0.024344889322916667,
      "learning_rate": 0.0001,
      "loss": 5.42,
      "loss/crossentropy": 2.2508283853530884,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2945093661546707,
      "step": 1768
    },
    {
      "epoch": 0.0354,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.0222320556640625,
      "learning_rate": 0.0001,
      "loss": 5.2516,
      "loss/crossentropy": 1.9332409501075745,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26304441690444946,
      "step": 1770
    },
    {
      "epoch": 0.03544,
      "grad_norm": 2.875,
      "grad_norm_var": 0.022557576497395832,
      "learning_rate": 0.0001,
      "loss": 5.1896,
      "loss/crossentropy": 2.143627643585205,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28407415747642517,
      "step": 1772
    },
    {
      "epoch": 0.03548,
      "grad_norm": 2.859375,
      "grad_norm_var": 0.029683430989583332,
      "learning_rate": 0.0001,
      "loss": 5.4763,
      "loss/crossentropy": 2.32085120677948,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3176119029521942,
      "step": 1774
    },
    {
      "epoch": 0.03552,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.028539021809895832,
      "learning_rate": 0.0001,
      "loss": 5.1717,
      "loss/crossentropy": 1.7893801927566528,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27407801151275635,
      "step": 1776
    },
    {
      "epoch": 0.03556,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.027228800455729167,
      "learning_rate": 0.0001,
      "loss": 5.42,
      "loss/crossentropy": 2.206292986869812,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3071517199277878,
      "step": 1778
    },
    {
      "epoch": 0.0356,
      "grad_norm": 2.84375,
      "grad_norm_var": 0.013263956705729166,
      "learning_rate": 0.0001,
      "loss": 5.1879,
      "loss/crossentropy": 2.1285043954849243,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3164139539003372,
      "step": 1780
    },
    {
      "epoch": 0.03564,
      "grad_norm": 3.15625,
      "grad_norm_var": 0.018635050455729166,
      "learning_rate": 0.0001,
      "loss": 5.2826,
      "loss/crossentropy": 2.16570383310318,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3052050769329071,
      "step": 1782
    },
    {
      "epoch": 0.03568,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.019950358072916667,
      "learning_rate": 0.0001,
      "loss": 5.0921,
      "loss/crossentropy": 1.9799941778182983,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2655785381793976,
      "step": 1784
    },
    {
      "epoch": 0.03572,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.0209625244140625,
      "learning_rate": 0.0001,
      "loss": 5.2468,
      "loss/crossentropy": 1.9801498651504517,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2824363112449646,
      "step": 1786
    },
    {
      "epoch": 0.03576,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.020947265625,
      "learning_rate": 0.0001,
      "loss": 5.0131,
      "loss/crossentropy": 1.5805786848068237,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25234321504831314,
      "step": 1788
    },
    {
      "epoch": 0.0358,
      "grad_norm": 3.15625,
      "grad_norm_var": 0.021484375,
      "learning_rate": 0.0001,
      "loss": 5.296,
      "loss/crossentropy": 2.2434048652648926,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2896551638841629,
      "step": 1790
    },
    {
      "epoch": 0.03584,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.022411092122395834,
      "learning_rate": 0.0001,
      "loss": 5.0179,
      "loss/crossentropy": 1.9738762378692627,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2959328889846802,
      "step": 1792
    },
    {
      "epoch": 0.03588,
      "grad_norm": 3.0625,
      "grad_norm_var": 0.0272857666015625,
      "learning_rate": 0.0001,
      "loss": 5.2317,
      "loss/crossentropy": 2.222583770751953,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29915711283683777,
      "step": 1794
    },
    {
      "epoch": 0.03592,
      "grad_norm": 2.8125,
      "grad_norm_var": 0.032942708333333334,
      "learning_rate": 0.0001,
      "loss": 5.4192,
      "loss/crossentropy": 2.188909649848938,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.294817179441452,
      "step": 1796
    },
    {
      "epoch": 0.03596,
      "grad_norm": 3.015625,
      "grad_norm_var": 0.030907185872395833,
      "learning_rate": 0.0001,
      "loss": 5.6303,
      "loss/crossentropy": 2.4745373725891113,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32838208973407745,
      "step": 1798
    },
    {
      "epoch": 0.036,
      "grad_norm": 2.890625,
      "grad_norm_var": 0.028595987955729166,
      "learning_rate": 0.0001,
      "loss": 5.3466,
      "loss/crossentropy": 1.9314215779304504,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29102426767349243,
      "step": 1800
    },
    {
      "epoch": 0.03604,
      "grad_norm": 2.796875,
      "grad_norm_var": 0.026276652018229166,
      "learning_rate": 0.0001,
      "loss": 4.9782,
      "loss/crossentropy": 2.0099900364875793,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30170293152332306,
      "step": 1802
    },
    {
      "epoch": 0.03608,
      "grad_norm": 2.859375,
      "grad_norm_var": 0.024388631184895832,
      "learning_rate": 0.0001,
      "loss": 5.2506,
      "loss/crossentropy": 2.1273564100265503,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3019937574863434,
      "step": 1804
    },
    {
      "epoch": 0.03612,
      "grad_norm": 2.96875,
      "grad_norm_var": 0.016341145833333334,
      "learning_rate": 0.0001,
      "loss": 5.1003,
      "loss/crossentropy": 2.065160095691681,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2817380279302597,
      "step": 1806
    },
    {
      "epoch": 0.03616,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.018561808268229167,
      "learning_rate": 0.0001,
      "loss": 5.1569,
      "loss/crossentropy": 2.262821078300476,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33245067298412323,
      "step": 1808
    },
    {
      "epoch": 0.0362,
      "grad_norm": 2.625,
      "grad_norm_var": 0.020003255208333334,
      "learning_rate": 0.0001,
      "loss": 4.8323,
      "loss/crossentropy": 2.164163827896118,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26008155941963196,
      "step": 1810
    },
    {
      "epoch": 0.03624,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.012398274739583333,
      "learning_rate": 0.0001,
      "loss": 5.2517,
      "loss/crossentropy": 2.147629737854004,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3073730617761612,
      "step": 1812
    },
    {
      "epoch": 0.03628,
      "grad_norm": 2.75,
      "grad_norm_var": 0.023900349934895832,
      "learning_rate": 0.0001,
      "loss": 5.1255,
      "loss/crossentropy": 2.04233980178833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2981158718466759,
      "step": 1814
    },
    {
      "epoch": 0.03632,
      "grad_norm": 3.0,
      "grad_norm_var": 0.026439412434895834,
      "learning_rate": 0.0001,
      "loss": 4.9465,
      "loss/crossentropy": 2.264409065246582,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32124973833560944,
      "step": 1816
    },
    {
      "epoch": 0.03636,
      "grad_norm": 3.109375,
      "grad_norm_var": 0.033610026041666664,
      "learning_rate": 0.0001,
      "loss": 5.4324,
      "loss/crossentropy": 2.092079997062683,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3318764269351959,
      "step": 1818
    },
    {
      "epoch": 0.0364,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.03850504557291667,
      "learning_rate": 0.0001,
      "loss": 5.2766,
      "loss/crossentropy": 2.1007314324378967,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2930498272180557,
      "step": 1820
    },
    {
      "epoch": 0.03644,
      "grad_norm": 2.84375,
      "grad_norm_var": 0.03697001139322917,
      "learning_rate": 0.0001,
      "loss": 5.129,
      "loss/crossentropy": 2.2377375960350037,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30473431944847107,
      "step": 1822
    },
    {
      "epoch": 0.03648,
      "grad_norm": 3.296875,
      "grad_norm_var": 0.04684244791666667,
      "learning_rate": 0.0001,
      "loss": 5.4209,
      "loss/crossentropy": 2.0965787172317505,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30526305735111237,
      "step": 1824
    },
    {
      "epoch": 0.03652,
      "grad_norm": 3.078125,
      "grad_norm_var": 0.04755859375,
      "learning_rate": 0.0001,
      "loss": 5.0359,
      "loss/crossentropy": 2.208711862564087,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2727830111980438,
      "step": 1826
    },
    {
      "epoch": 0.03656,
      "grad_norm": 2.796875,
      "grad_norm_var": 0.050633748372395836,
      "learning_rate": 0.0001,
      "loss": 5.1692,
      "loss/crossentropy": 2.1706738471984863,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3054092824459076,
      "step": 1828
    },
    {
      "epoch": 0.0366,
      "grad_norm": 2.796875,
      "grad_norm_var": 0.037262980143229166,
      "learning_rate": 0.0001,
      "loss": 5.2311,
      "loss/crossentropy": 2.0891621112823486,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3111531287431717,
      "step": 1830
    },
    {
      "epoch": 0.03664,
      "grad_norm": 2.859375,
      "grad_norm_var": 0.03560791015625,
      "learning_rate": 0.0001,
      "loss": 5.0537,
      "loss/crossentropy": 2.0721256732940674,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30227208137512207,
      "step": 1832
    },
    {
      "epoch": 0.03668,
      "grad_norm": 3.015625,
      "grad_norm_var": 0.03023681640625,
      "learning_rate": 0.0001,
      "loss": 5.1267,
      "loss/crossentropy": 2.2015734910964966,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3185647875070572,
      "step": 1834
    },
    {
      "epoch": 0.03672,
      "grad_norm": 3.03125,
      "grad_norm_var": 0.026493326822916666,
      "learning_rate": 0.0001,
      "loss": 5.1973,
      "loss/crossentropy": 2.1985132694244385,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3032165467739105,
      "step": 1836
    },
    {
      "epoch": 0.03676,
      "grad_norm": 2.890625,
      "grad_norm_var": 0.026334635416666665,
      "learning_rate": 0.0001,
      "loss": 5.2041,
      "loss/crossentropy": 2.170067548751831,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3295029550790787,
      "step": 1838
    },
    {
      "epoch": 0.0368,
      "grad_norm": 2.921875,
      "grad_norm_var": 0.014615885416666667,
      "learning_rate": 0.0001,
      "loss": 5.0049,
      "loss/crossentropy": 2.113592267036438,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2775915116071701,
      "step": 1840
    },
    {
      "epoch": 0.03684,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.0117828369140625,
      "learning_rate": 0.0001,
      "loss": 5.4221,
      "loss/crossentropy": 2.1905024647712708,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33297547698020935,
      "step": 1842
    },
    {
      "epoch": 0.03688,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.012919108072916666,
      "learning_rate": 0.0001,
      "loss": 5.1859,
      "loss/crossentropy": 2.252369999885559,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29375749826431274,
      "step": 1844
    },
    {
      "epoch": 0.03692,
      "grad_norm": 3.015625,
      "grad_norm_var": 0.020182291666666668,
      "learning_rate": 0.0001,
      "loss": 4.8942,
      "loss/crossentropy": 1.7526759505271912,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2707225978374481,
      "step": 1846
    },
    {
      "epoch": 0.03696,
      "grad_norm": 3.21875,
      "grad_norm_var": 0.028880818684895834,
      "learning_rate": 0.0001,
      "loss": 5.6529,
      "loss/crossentropy": 2.592544913291931,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3520146906375885,
      "step": 1848
    },
    {
      "epoch": 0.037,
      "grad_norm": 2.8125,
      "grad_norm_var": 0.0278961181640625,
      "learning_rate": 0.0001,
      "loss": 4.9816,
      "loss/crossentropy": 1.8699345588684082,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2852860391139984,
      "step": 1850
    },
    {
      "epoch": 0.03704,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.03209228515625,
      "learning_rate": 0.0001,
      "loss": 5.1326,
      "loss/crossentropy": 2.2219313383102417,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3210798054933548,
      "step": 1852
    },
    {
      "epoch": 0.03708,
      "grad_norm": 2.796875,
      "grad_norm_var": 0.0327056884765625,
      "learning_rate": 0.0001,
      "loss": 5.2301,
      "loss/crossentropy": 1.9926818013191223,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26928839832544327,
      "step": 1854
    },
    {
      "epoch": 0.03712,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.032515462239583334,
      "learning_rate": 0.0001,
      "loss": 5.0372,
      "loss/crossentropy": 2.019917130470276,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28046920895576477,
      "step": 1856
    },
    {
      "epoch": 0.03716,
      "grad_norm": 4.84375,
      "grad_norm_var": 0.2762858072916667,
      "learning_rate": 0.0001,
      "loss": 5.6297,
      "loss/crossentropy": 2.2585690021514893,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30598941445350647,
      "step": 1858
    },
    {
      "epoch": 0.0372,
      "grad_norm": 2.984375,
      "grad_norm_var": 0.27327067057291665,
      "learning_rate": 0.0001,
      "loss": 5.3508,
      "loss/crossentropy": 2.298324942588806,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3494870364665985,
      "step": 1860
    },
    {
      "epoch": 0.03724,
      "grad_norm": 3.125,
      "grad_norm_var": 0.25745035807291666,
      "learning_rate": 0.0001,
      "loss": 5.54,
      "loss/crossentropy": 2.430496573448181,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3227449208498001,
      "step": 1862
    },
    {
      "epoch": 0.03728,
      "grad_norm": 2.953125,
      "grad_norm_var": 0.26183980305989585,
      "learning_rate": 0.0001,
      "loss": 5.1876,
      "loss/crossentropy": 2.090576171875,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2790074646472931,
      "step": 1864
    },
    {
      "epoch": 0.03732,
      "grad_norm": 3.0625,
      "grad_norm_var": 0.25806884765625,
      "learning_rate": 0.0001,
      "loss": 5.1799,
      "loss/crossentropy": 2.2794109582901,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3135898858308792,
      "step": 1866
    },
    {
      "epoch": 0.03736,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.2597076416015625,
      "learning_rate": 0.0001,
      "loss": 4.9333,
      "loss/crossentropy": 2.2433481216430664,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27558377385139465,
      "step": 1868
    },
    {
      "epoch": 0.0374,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.26806233723958334,
      "learning_rate": 0.0001,
      "loss": 5.292,
      "loss/crossentropy": 2.3111730813980103,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3172578364610672,
      "step": 1870
    },
    {
      "epoch": 0.03744,
      "grad_norm": 2.921875,
      "grad_norm_var": 0.270654296875,
      "learning_rate": 0.0001,
      "loss": 5.2364,
      "loss/crossentropy": 2.0028095841407776,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.35644619166851044,
      "step": 1872
    },
    {
      "epoch": 0.03748,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.04501546223958333,
      "learning_rate": 0.0001,
      "loss": 5.2183,
      "loss/crossentropy": 2.347644329071045,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3119680881500244,
      "step": 1874
    },
    {
      "epoch": 0.03752,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.03474833170572917,
      "learning_rate": 0.0001,
      "loss": 5.0787,
      "loss/crossentropy": 2.118954062461853,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28975334763526917,
      "step": 1876
    },
    {
      "epoch": 0.03756,
      "grad_norm": 3.0625,
      "grad_norm_var": 0.033568318684895834,
      "learning_rate": 0.0001,
      "loss": 5.6649,
      "loss/crossentropy": 2.4376548528671265,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3358190506696701,
      "step": 1878
    },
    {
      "epoch": 0.0376,
      "grad_norm": 3.28125,
      "grad_norm_var": 0.042724609375,
      "learning_rate": 0.0001,
      "loss": 5.4924,
      "loss/crossentropy": 2.5907636880874634,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3234570771455765,
      "step": 1880
    },
    {
      "epoch": 0.03764,
      "grad_norm": 2.859375,
      "grad_norm_var": 0.04352925618489583,
      "learning_rate": 0.0001,
      "loss": 5.0298,
      "loss/crossentropy": 1.574956238269806,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26118964701890945,
      "step": 1882
    },
    {
      "epoch": 0.03768,
      "grad_norm": 3.203125,
      "grad_norm_var": 0.04599609375,
      "learning_rate": 0.0001,
      "loss": 5.3339,
      "loss/crossentropy": 2.3571736812591553,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.308853879570961,
      "step": 1884
    },
    {
      "epoch": 0.03772,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.03877665201822917,
      "learning_rate": 0.0001,
      "loss": 5.0807,
      "loss/crossentropy": 2.1266958117485046,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.307782918214798,
      "step": 1886
    },
    {
      "epoch": 0.03776,
      "grad_norm": 3.0,
      "grad_norm_var": 0.03876546223958333,
      "learning_rate": 0.0001,
      "loss": 5.5402,
      "loss/crossentropy": 2.3529077768325806,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32287272810935974,
      "step": 1888
    },
    {
      "epoch": 0.0378,
      "grad_norm": 3.015625,
      "grad_norm_var": 0.03905843098958333,
      "learning_rate": 0.0001,
      "loss": 5.4555,
      "loss/crossentropy": 2.277345299720764,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32319171726703644,
      "step": 1890
    },
    {
      "epoch": 0.03784,
      "grad_norm": 2.859375,
      "grad_norm_var": 0.035008748372395836,
      "learning_rate": 0.0001,
      "loss": 5.01,
      "loss/crossentropy": 2.102940857410431,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.292771652340889,
      "step": 1892
    },
    {
      "epoch": 0.03788,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.03998921712239583,
      "learning_rate": 0.0001,
      "loss": 5.0784,
      "loss/crossentropy": 1.9744033813476562,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2844446450471878,
      "step": 1894
    },
    {
      "epoch": 0.03792,
      "grad_norm": 2.625,
      "grad_norm_var": 0.03250325520833333,
      "learning_rate": 0.0001,
      "loss": 4.9859,
      "loss/crossentropy": 1.8222022652626038,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28611525893211365,
      "step": 1896
    },
    {
      "epoch": 0.03796,
      "grad_norm": 2.96875,
      "grad_norm_var": 0.03135477701822917,
      "learning_rate": 0.0001,
      "loss": 5.0704,
      "loss/crossentropy": 2.1963966488838196,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2988738566637039,
      "step": 1898
    },
    {
      "epoch": 0.038,
      "grad_norm": 3.046875,
      "grad_norm_var": 0.025927734375,
      "learning_rate": 0.0001,
      "loss": 5.1155,
      "loss/crossentropy": 1.9982789754867554,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.312205046415329,
      "step": 1900
    },
    {
      "epoch": 0.03804,
      "grad_norm": 2.75,
      "grad_norm_var": 0.020612589518229165,
      "learning_rate": 0.0001,
      "loss": 5.2097,
      "loss/crossentropy": 2.1999258995056152,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29916079342365265,
      "step": 1902
    },
    {
      "epoch": 0.03808,
      "grad_norm": 2.859375,
      "grad_norm_var": 0.016999308268229166,
      "learning_rate": 0.0001,
      "loss": 5.2233,
      "loss/crossentropy": 2.0725532174110413,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3093830645084381,
      "step": 1904
    },
    {
      "epoch": 0.03812,
      "grad_norm": 3.03125,
      "grad_norm_var": 0.027079264322916668,
      "learning_rate": 0.0001,
      "loss": 5.7014,
      "loss/crossentropy": 2.2504276037216187,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.332836389541626,
      "step": 1906
    },
    {
      "epoch": 0.03816,
      "grad_norm": 2.875,
      "grad_norm_var": 0.026927693684895834,
      "learning_rate": 0.0001,
      "loss": 5.3413,
      "loss/crossentropy": 2.1570577025413513,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3210139572620392,
      "step": 1908
    },
    {
      "epoch": 0.0382,
      "grad_norm": 3.59375,
      "grad_norm_var": 0.05137430826822917,
      "learning_rate": 0.0001,
      "loss": 5.6215,
      "loss/crossentropy": 2.0739041566848755,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3091907352209091,
      "step": 1910
    },
    {
      "epoch": 0.03824,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.0502838134765625,
      "learning_rate": 0.0001,
      "loss": 5.0772,
      "loss/crossentropy": 2.0542168021202087,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2791624963283539,
      "step": 1912
    },
    {
      "epoch": 0.03828,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.0561431884765625,
      "learning_rate": 0.0001,
      "loss": 4.9386,
      "loss/crossentropy": 1.9705287218093872,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25483617186546326,
      "step": 1914
    },
    {
      "epoch": 0.03832,
      "grad_norm": 2.875,
      "grad_norm_var": 0.05819905598958333,
      "learning_rate": 0.0001,
      "loss": 5.054,
      "loss/crossentropy": 2.0234111547470093,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3110152333974838,
      "step": 1916
    },
    {
      "epoch": 0.03836,
      "grad_norm": 3.15625,
      "grad_norm_var": 0.06004130045572917,
      "learning_rate": 0.0001,
      "loss": 5.2723,
      "loss/crossentropy": 2.010735273361206,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2943515181541443,
      "step": 1918
    },
    {
      "epoch": 0.0384,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.06575419108072916,
      "learning_rate": 0.0001,
      "loss": 4.9471,
      "loss/crossentropy": 2.1912686824798584,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27914971113204956,
      "step": 1920
    },
    {
      "epoch": 0.03844,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.0709869384765625,
      "learning_rate": 0.0001,
      "loss": 5.085,
      "loss/crossentropy": 1.9889940023422241,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26166442036628723,
      "step": 1922
    },
    {
      "epoch": 0.03848,
      "grad_norm": 2.96875,
      "grad_norm_var": 0.07668863932291667,
      "learning_rate": 0.0001,
      "loss": 5.3534,
      "loss/crossentropy": 2.154898941516876,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31905338168144226,
      "step": 1924
    },
    {
      "epoch": 0.03852,
      "grad_norm": 2.796875,
      "grad_norm_var": 0.03997395833333333,
      "learning_rate": 0.0001,
      "loss": 5.0847,
      "loss/crossentropy": 2.44269061088562,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28736811876296997,
      "step": 1926
    },
    {
      "epoch": 0.03856,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.04108784993489583,
      "learning_rate": 0.0001,
      "loss": 5.0585,
      "loss/crossentropy": 1.6790328621864319,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32607489824295044,
      "step": 1928
    },
    {
      "epoch": 0.0386,
      "grad_norm": 3.046875,
      "grad_norm_var": 0.05115559895833333,
      "learning_rate": 0.0001,
      "loss": 5.336,
      "loss/crossentropy": 2.0223641991615295,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2871186435222626,
      "step": 1930
    },
    {
      "epoch": 0.03864,
      "grad_norm": 2.8125,
      "grad_norm_var": 0.05756734212239583,
      "learning_rate": 0.0001,
      "loss": 5.549,
      "loss/crossentropy": 2.451051712036133,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33973076939582825,
      "step": 1932
    },
    {
      "epoch": 0.03868,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.052262369791666666,
      "learning_rate": 0.0001,
      "loss": 5.4403,
      "loss/crossentropy": 2.2884862422943115,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29729554057121277,
      "step": 1934
    },
    {
      "epoch": 0.03872,
      "grad_norm": 2.890625,
      "grad_norm_var": 0.04889322916666667,
      "learning_rate": 0.0001,
      "loss": 5.6203,
      "loss/crossentropy": 2.113345444202423,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.36945630609989166,
      "step": 1936
    },
    {
      "epoch": 0.03876,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.03752848307291667,
      "learning_rate": 0.0001,
      "loss": 5.1845,
      "loss/crossentropy": 2.139409840106964,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2899101823568344,
      "step": 1938
    },
    {
      "epoch": 0.0388,
      "grad_norm": 2.953125,
      "grad_norm_var": 0.03622639973958333,
      "learning_rate": 0.0001,
      "loss": 5.5397,
      "loss/crossentropy": 2.1029305458068848,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29830390214920044,
      "step": 1940
    },
    {
      "epoch": 0.03884,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.03816630045572917,
      "learning_rate": 0.0001,
      "loss": 5.1383,
      "loss/crossentropy": 1.7736502885818481,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27648696303367615,
      "step": 1942
    },
    {
      "epoch": 0.03888,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.03942057291666667,
      "learning_rate": 0.0001,
      "loss": 5.0885,
      "loss/crossentropy": 2.0281469225883484,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30329641699790955,
      "step": 1944
    },
    {
      "epoch": 0.03892,
      "grad_norm": 2.890625,
      "grad_norm_var": 0.03528544108072917,
      "learning_rate": 0.0001,
      "loss": 5.3265,
      "loss/crossentropy": 2.404891610145569,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3563212752342224,
      "step": 1946
    },
    {
      "epoch": 0.03896,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.03572489420572917,
      "learning_rate": 0.0001,
      "loss": 5.0657,
      "loss/crossentropy": 2.2187922596931458,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2999056503176689,
      "step": 1948
    },
    {
      "epoch": 0.039,
      "grad_norm": 2.875,
      "grad_norm_var": 0.03566792805989583,
      "learning_rate": 0.0001,
      "loss": 5.0499,
      "loss/crossentropy": 2.3901994228363037,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3288661539554596,
      "step": 1950
    },
    {
      "epoch": 0.03904,
      "grad_norm": 3.15625,
      "grad_norm_var": 0.04069722493489583,
      "learning_rate": 0.0001,
      "loss": 5.252,
      "loss/crossentropy": 2.274617314338684,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2928028404712677,
      "step": 1952
    },
    {
      "epoch": 0.03908,
      "grad_norm": 2.859375,
      "grad_norm_var": 0.03430582682291667,
      "learning_rate": 0.0001,
      "loss": 5.4463,
      "loss/crossentropy": 2.2478950023651123,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3027127981185913,
      "step": 1954
    },
    {
      "epoch": 0.03912,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.030321248372395835,
      "learning_rate": 0.0001,
      "loss": 5.2707,
      "loss/crossentropy": 2.0634876489639282,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29374830424785614,
      "step": 1956
    },
    {
      "epoch": 0.03916,
      "grad_norm": 2.90625,
      "grad_norm_var": 0.0287994384765625,
      "learning_rate": 0.0001,
      "loss": 5.3441,
      "loss/crossentropy": 2.171326994895935,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2995557487010956,
      "step": 1958
    },
    {
      "epoch": 0.0392,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.040266927083333334,
      "learning_rate": 0.0001,
      "loss": 4.9937,
      "loss/crossentropy": 2.142563223838806,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27376608550548553,
      "step": 1960
    },
    {
      "epoch": 0.03924,
      "grad_norm": 5.34375,
      "grad_norm_var": 0.42568359375,
      "learning_rate": 0.0001,
      "loss": 5.4145,
      "loss/crossentropy": 2.3189245462417603,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.37109237909317017,
      "step": 1962
    },
    {
      "epoch": 0.03928,
      "grad_norm": 3.203125,
      "grad_norm_var": 0.40812886555989586,
      "learning_rate": 0.0001,
      "loss": 5.1826,
      "loss/crossentropy": 2.1483139991760254,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32499393820762634,
      "step": 1964
    },
    {
      "epoch": 0.03932,
      "grad_norm": 3.046875,
      "grad_norm_var": 0.404736328125,
      "learning_rate": 0.0001,
      "loss": 5.445,
      "loss/crossentropy": 2.2916383743286133,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29367291927337646,
      "step": 1966
    },
    {
      "epoch": 0.03936,
      "grad_norm": 2.9375,
      "grad_norm_var": 0.3999582926432292,
      "learning_rate": 0.0001,
      "loss": 5.1663,
      "loss/crossentropy": 2.4217371940612793,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31309331953525543,
      "step": 1968
    },
    {
      "epoch": 0.0394,
      "grad_norm": 2.9375,
      "grad_norm_var": 0.40103759765625,
      "learning_rate": 0.0001,
      "loss": 5.2057,
      "loss/crossentropy": 1.9491975903511047,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29921913146972656,
      "step": 1970
    },
    {
      "epoch": 0.03944,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.39126688639322915,
      "learning_rate": 0.0001,
      "loss": 5.1788,
      "loss/crossentropy": 2.144432306289673,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29162150621414185,
      "step": 1972
    },
    {
      "epoch": 0.03948,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.40748291015625,
      "learning_rate": 0.0001,
      "loss": 5.1336,
      "loss/crossentropy": 1.9492529034614563,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2875414192676544,
      "step": 1974
    },
    {
      "epoch": 0.03952,
      "grad_norm": 2.75,
      "grad_norm_var": 0.4002593994140625,
      "learning_rate": 0.0001,
      "loss": 5.053,
      "loss/crossentropy": 1.9269813895225525,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2755381464958191,
      "step": 1976
    },
    {
      "epoch": 0.03956,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.03798726399739583,
      "learning_rate": 0.0001,
      "loss": 4.8879,
      "loss/crossentropy": 2.074360489845276,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2987503558397293,
      "step": 1978
    },
    {
      "epoch": 0.0396,
      "grad_norm": 2.953125,
      "grad_norm_var": 0.02086181640625,
      "learning_rate": 0.0001,
      "loss": 4.8834,
      "loss/crossentropy": 2.257633090019226,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2992274910211563,
      "step": 1980
    },
    {
      "epoch": 0.03964,
      "grad_norm": 2.75,
      "grad_norm_var": 0.01802978515625,
      "learning_rate": 0.0001,
      "loss": 4.9533,
      "loss/crossentropy": 1.8207083940505981,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2752758115530014,
      "step": 1982
    },
    {
      "epoch": 0.03968,
      "grad_norm": 2.875,
      "grad_norm_var": 0.015623982747395833,
      "learning_rate": 0.0001,
      "loss": 5.343,
      "loss/crossentropy": 2.105292797088623,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3131226450204849,
      "step": 1984
    },
    {
      "epoch": 0.03972,
      "grad_norm": 3.484375,
      "grad_norm_var": 0.05191650390625,
      "learning_rate": 0.0001,
      "loss": 5.4785,
      "loss/crossentropy": 2.1191373467445374,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3175530731678009,
      "step": 1986
    },
    {
      "epoch": 0.03976,
      "grad_norm": 2.75,
      "grad_norm_var": 0.051878865559895834,
      "learning_rate": 0.0001,
      "loss": 4.9236,
      "loss/crossentropy": 2.2214397192001343,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3091724067926407,
      "step": 1988
    },
    {
      "epoch": 0.0398,
      "grad_norm": 2.875,
      "grad_norm_var": 0.05133056640625,
      "learning_rate": 0.0001,
      "loss": 5.0031,
      "loss/crossentropy": 1.7347424626350403,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2825440764427185,
      "step": 1990
    },
    {
      "epoch": 0.03984,
      "grad_norm": 2.890625,
      "grad_norm_var": 0.0500396728515625,
      "learning_rate": 0.0001,
      "loss": 5.0951,
      "loss/crossentropy": 2.1566559076309204,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2961925268173218,
      "step": 1992
    },
    {
      "epoch": 0.03988,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.06396077473958334,
      "learning_rate": 0.0001,
      "loss": 4.9195,
      "loss/crossentropy": 2.2129205465316772,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29672613739967346,
      "step": 1994
    },
    {
      "epoch": 0.03992,
      "grad_norm": 3.3125,
      "grad_norm_var": 0.07595113118489584,
      "learning_rate": 0.0001,
      "loss": 5.7534,
      "loss/crossentropy": 2.4702744483947754,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3707122802734375,
      "step": 1996
    },
    {
      "epoch": 0.03996,
      "grad_norm": 2.84375,
      "grad_norm_var": 0.06731669108072917,
      "learning_rate": 0.0001,
      "loss": 5.0296,
      "loss/crossentropy": 2.0463536977767944,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3124735355377197,
      "step": 1998
    },
    {
      "epoch": 0.04,
      "grad_norm": 2.796875,
      "grad_norm_var": 0.07281494140625,
      "learning_rate": 0.0001,
      "loss": 4.959,
      "loss/crossentropy": 2.1550235748291016,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32177163660526276,
      "step": 2000
    },
    {
      "epoch": 0.04004,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.059798177083333334,
      "learning_rate": 0.0001,
      "loss": 5.2078,
      "loss/crossentropy": 2.1312190890312195,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29013920575380325,
      "step": 2002
    },
    {
      "epoch": 0.04008,
      "grad_norm": 2.75,
      "grad_norm_var": 0.06082356770833333,
      "learning_rate": 0.0001,
      "loss": 5.0086,
      "loss/crossentropy": 1.8546085357666016,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2619543671607971,
      "step": 2004
    },
    {
      "epoch": 0.04012,
      "grad_norm": 3.015625,
      "grad_norm_var": 0.0561920166015625,
      "learning_rate": 0.0001,
      "loss": 5.3416,
      "loss/crossentropy": 2.262398660182953,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28742220997810364,
      "step": 2006
    },
    {
      "epoch": 0.04016,
      "grad_norm": 3.109375,
      "grad_norm_var": 0.05734049479166667,
      "learning_rate": 0.0001,
      "loss": 5.4315,
      "loss/crossentropy": 2.156043767929077,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30488699674606323,
      "step": 2008
    },
    {
      "epoch": 0.0402,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.04290364583333333,
      "learning_rate": 0.0001,
      "loss": 4.8761,
      "loss/crossentropy": 1.925516963005066,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2795708477497101,
      "step": 2010
    },
    {
      "epoch": 0.04024,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.031281534830729166,
      "learning_rate": 0.0001,
      "loss": 5.0729,
      "loss/crossentropy": 1.947714388370514,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27192793786525726,
      "step": 2012
    },
    {
      "epoch": 0.04028,
      "grad_norm": 2.625,
      "grad_norm_var": 0.038939412434895834,
      "learning_rate": 0.0001,
      "loss": 4.6214,
      "loss/crossentropy": 1.9584010243415833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2835424840450287,
      "step": 2014
    },
    {
      "epoch": 0.04032,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.04108784993489583,
      "learning_rate": 0.0001,
      "loss": 5.1974,
      "loss/crossentropy": 2.461808919906616,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3104698956012726,
      "step": 2016
    },
    {
      "epoch": 0.04036,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.021903483072916667,
      "learning_rate": 0.0001,
      "loss": 5.286,
      "loss/crossentropy": 2.094545900821686,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30010756850242615,
      "step": 2018
    },
    {
      "epoch": 0.0404,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.021923828125,
      "learning_rate": 0.0001,
      "loss": 5.1274,
      "loss/crossentropy": 2.353589177131653,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29775144159793854,
      "step": 2020
    },
    {
      "epoch": 0.04044,
      "grad_norm": 3.375,
      "grad_norm_var": 10.55537821451823,
      "learning_rate": 0.0001,
      "loss": 5.3359,
      "loss/crossentropy": 2.4468252658843994,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.34534354507923126,
      "step": 2022
    },
    {
      "epoch": 0.04048,
      "grad_norm": 2.984375,
      "grad_norm_var": 10.529678344726562,
      "learning_rate": 0.0001,
      "loss": 5.5028,
      "loss/crossentropy": 2.2037755250930786,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.303710475564003,
      "step": 2024
    },
    {
      "epoch": 0.04052,
      "grad_norm": 2.71875,
      "grad_norm_var": 10.546240234375,
      "learning_rate": 0.0001,
      "loss": 4.9229,
      "loss/crossentropy": 1.9658318161964417,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2829178273677826,
      "step": 2026
    },
    {
      "epoch": 0.04056,
      "grad_norm": 2.734375,
      "grad_norm_var": 10.555729166666667,
      "learning_rate": 0.0001,
      "loss": 4.8996,
      "loss/crossentropy": 2.118351697921753,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29469528794288635,
      "step": 2028
    },
    {
      "epoch": 0.0406,
      "grad_norm": 2.875,
      "grad_norm_var": 10.507957967122396,
      "learning_rate": 0.0001,
      "loss": 5.5817,
      "loss/crossentropy": 2.172826111316681,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.35036201775074005,
      "step": 2030
    },
    {
      "epoch": 0.04064,
      "grad_norm": 3.953125,
      "grad_norm_var": 10.432124837239583,
      "learning_rate": 0.0001,
      "loss": 5.4846,
      "loss/crossentropy": 2.185975730419159,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3397497236728668,
      "step": 2032
    },
    {
      "epoch": 0.04068,
      "grad_norm": 2.78125,
      "grad_norm_var": 10.411026000976562,
      "learning_rate": 0.0001,
      "loss": 5.0145,
      "loss/crossentropy": 2.043874442577362,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3034070134162903,
      "step": 2034
    },
    {
      "epoch": 0.04072,
      "grad_norm": 2.984375,
      "grad_norm_var": 10.380106608072916,
      "learning_rate": 0.0001,
      "loss": 5.3958,
      "loss/crossentropy": 2.3315287828445435,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32198067009449005,
      "step": 2036
    },
    {
      "epoch": 0.04076,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.1112213134765625,
      "learning_rate": 0.0001,
      "loss": 5.0266,
      "loss/crossentropy": 2.05656898021698,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30919137597084045,
      "step": 2038
    },
    {
      "epoch": 0.0408,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.10188395182291667,
      "learning_rate": 0.0001,
      "loss": 5.3205,
      "loss/crossentropy": 2.2451635599136353,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3412973880767822,
      "step": 2040
    },
    {
      "epoch": 0.04084,
      "grad_norm": 3.1875,
      "grad_norm_var": 0.10715738932291667,
      "learning_rate": 0.0001,
      "loss": 5.1734,
      "loss/crossentropy": 2.527924060821533,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3178148865699768,
      "step": 2042
    },
    {
      "epoch": 0.04088,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.10305582682291667,
      "learning_rate": 0.0001,
      "loss": 4.8441,
      "loss/crossentropy": 2.03126460313797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29549214243888855,
      "step": 2044
    },
    {
      "epoch": 0.04092,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.11038004557291667,
      "learning_rate": 0.0001,
      "loss": 5.0134,
      "loss/crossentropy": 2.029997706413269,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27938202023506165,
      "step": 2046
    },
    {
      "epoch": 0.04096,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.03211263020833333,
      "learning_rate": 0.0001,
      "loss": 4.9321,
      "loss/crossentropy": 1.764098048210144,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27712512016296387,
      "step": 2048
    },
    {
      "epoch": 0.041,
      "grad_norm": 3.046875,
      "grad_norm_var": 0.0356597900390625,
      "learning_rate": 0.0001,
      "loss": 5.435,
      "loss/crossentropy": 2.605324864387512,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3540754020214081,
      "step": 2050
    },
    {
      "epoch": 0.04104,
      "grad_norm": 2.875,
      "grad_norm_var": 0.0349761962890625,
      "learning_rate": 0.0001,
      "loss": 5.0207,
      "loss/crossentropy": 1.9333613514900208,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29670488089323044,
      "step": 2052
    },
    {
      "epoch": 0.04108,
      "grad_norm": 3.078125,
      "grad_norm_var": 0.040022786458333334,
      "learning_rate": 0.0001,
      "loss": 5.0056,
      "loss/crossentropy": 1.7876797914505005,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2579014301300049,
      "step": 2054
    },
    {
      "epoch": 0.04112,
      "grad_norm": 3.125,
      "grad_norm_var": 0.0417877197265625,
      "learning_rate": 0.0001,
      "loss": 5.1401,
      "loss/crossentropy": 2.0947588682174683,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3334304690361023,
      "step": 2056
    },
    {
      "epoch": 0.04116,
      "grad_norm": 3.0,
      "grad_norm_var": 0.035965983072916666,
      "learning_rate": 0.0001,
      "loss": 4.9009,
      "loss/crossentropy": 2.1838767528533936,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.306907519698143,
      "step": 2058
    },
    {
      "epoch": 0.0412,
      "grad_norm": 3.015625,
      "grad_norm_var": 0.0341461181640625,
      "learning_rate": 0.0001,
      "loss": 5.3724,
      "loss/crossentropy": 2.2180997133255005,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.314627081155777,
      "step": 2060
    },
    {
      "epoch": 0.04124,
      "grad_norm": 2.875,
      "grad_norm_var": 0.025419108072916665,
      "learning_rate": 0.0001,
      "loss": 4.8362,
      "loss/crossentropy": 1.914646863937378,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2848198413848877,
      "step": 2062
    },
    {
      "epoch": 0.04128,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.0262603759765625,
      "learning_rate": 0.0001,
      "loss": 5.4538,
      "loss/crossentropy": 2.42458713054657,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31268230080604553,
      "step": 2064
    },
    {
      "epoch": 0.04132,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.02457275390625,
      "learning_rate": 0.0001,
      "loss": 5.2497,
      "loss/crossentropy": 2.23202121257782,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30988384783267975,
      "step": 2066
    },
    {
      "epoch": 0.04136,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.0284820556640625,
      "learning_rate": 0.0001,
      "loss": 5.0416,
      "loss/crossentropy": 2.0225483179092407,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26953594386577606,
      "step": 2068
    },
    {
      "epoch": 0.0414,
      "grad_norm": 2.875,
      "grad_norm_var": 0.028669230143229165,
      "learning_rate": 0.0001,
      "loss": 4.8259,
      "loss/crossentropy": 1.8593338131904602,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27706706523895264,
      "step": 2070
    },
    {
      "epoch": 0.04144,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.023387654622395834,
      "learning_rate": 0.0001,
      "loss": 4.9949,
      "loss/crossentropy": 2.373727560043335,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3000074476003647,
      "step": 2072
    },
    {
      "epoch": 0.04148,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.022704060872395834,
      "learning_rate": 0.0001,
      "loss": 4.9438,
      "loss/crossentropy": 1.959564983844757,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26365046203136444,
      "step": 2074
    },
    {
      "epoch": 0.04152,
      "grad_norm": 2.921875,
      "grad_norm_var": 0.014159138997395833,
      "learning_rate": 0.0001,
      "loss": 4.983,
      "loss/crossentropy": 2.0590676069259644,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2816864550113678,
      "step": 2076
    },
    {
      "epoch": 0.04156,
      "grad_norm": 2.84375,
      "grad_norm_var": 0.017121378580729166,
      "learning_rate": 0.0001,
      "loss": 5.2049,
      "loss/crossentropy": 2.147680163383484,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29019051790237427,
      "step": 2078
    },
    {
      "epoch": 0.0416,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.020198567708333334,
      "learning_rate": 0.0001,
      "loss": 5.6103,
      "loss/crossentropy": 2.212267220020294,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2804659754037857,
      "step": 2080
    },
    {
      "epoch": 0.04164,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.020406087239583332,
      "learning_rate": 0.0001,
      "loss": 5.0489,
      "loss/crossentropy": 2.144743025302887,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2946523129940033,
      "step": 2082
    },
    {
      "epoch": 0.04168,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.019172159830729167,
      "learning_rate": 0.0001,
      "loss": 4.9311,
      "loss/crossentropy": 2.3702481985092163,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2923210561275482,
      "step": 2084
    },
    {
      "epoch": 0.04172,
      "grad_norm": 2.96875,
      "grad_norm_var": 0.019684855143229166,
      "learning_rate": 0.0001,
      "loss": 5.2291,
      "loss/crossentropy": 1.9512975811958313,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28704553842544556,
      "step": 2086
    },
    {
      "epoch": 0.04176,
      "grad_norm": 3.25,
      "grad_norm_var": 0.25388895670572914,
      "learning_rate": 0.0001,
      "loss": 4.9572,
      "loss/crossentropy": 2.180745005607605,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28361976146698,
      "step": 2088
    },
    {
      "epoch": 0.0418,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.24712626139322916,
      "learning_rate": 0.0001,
      "loss": 4.8579,
      "loss/crossentropy": 1.9768275022506714,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28042787313461304,
      "step": 2090
    },
    {
      "epoch": 0.04184,
      "grad_norm": 2.90625,
      "grad_norm_var": 0.24544270833333334,
      "learning_rate": 0.0001,
      "loss": 5.2602,
      "loss/crossentropy": 2.148472547531128,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30099035799503326,
      "step": 2092
    },
    {
      "epoch": 0.04188,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.24763081868489584,
      "learning_rate": 0.0001,
      "loss": 5.0152,
      "loss/crossentropy": 2.1698715686798096,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3140450567007065,
      "step": 2094
    },
    {
      "epoch": 0.04192,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.24172770182291667,
      "learning_rate": 0.0001,
      "loss": 4.8679,
      "loss/crossentropy": 2.1142334938049316,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2847675681114197,
      "step": 2096
    },
    {
      "epoch": 0.04196,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.2395660400390625,
      "learning_rate": 0.0001,
      "loss": 5.2185,
      "loss/crossentropy": 2.1908479928970337,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28702451288700104,
      "step": 2098
    },
    {
      "epoch": 0.042,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.23321024576822916,
      "learning_rate": 0.0001,
      "loss": 5.0212,
      "loss/crossentropy": 2.0519612431526184,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29224735498428345,
      "step": 2100
    },
    {
      "epoch": 0.04204,
      "grad_norm": 2.96875,
      "grad_norm_var": 0.2412994384765625,
      "learning_rate": 0.0001,
      "loss": 4.871,
      "loss/crossentropy": 1.9304961562156677,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28785137832164764,
      "step": 2102
    },
    {
      "epoch": 0.04208,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.0134185791015625,
      "learning_rate": 0.0001,
      "loss": 5.2462,
      "loss/crossentropy": 2.297300934791565,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.302143856883049,
      "step": 2104
    },
    {
      "epoch": 0.04212,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.019782511393229167,
      "learning_rate": 0.0001,
      "loss": 5.0491,
      "loss/crossentropy": 2.2764381170272827,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28773219883441925,
      "step": 2106
    },
    {
      "epoch": 0.04216,
      "grad_norm": 2.625,
      "grad_norm_var": 0.019391886393229165,
      "learning_rate": 0.0001,
      "loss": 5.0563,
      "loss/crossentropy": 2.141321837902069,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3108212947845459,
      "step": 2108
    },
    {
      "epoch": 0.0422,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.0185699462890625,
      "learning_rate": 0.0001,
      "loss": 5.0362,
      "loss/crossentropy": 1.9619495272636414,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2938811331987381,
      "step": 2110
    },
    {
      "epoch": 0.04224,
      "grad_norm": 3.015625,
      "grad_norm_var": 0.026325480143229166,
      "learning_rate": 0.0001,
      "loss": 5.4496,
      "loss/crossentropy": 1.9741051197052002,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28835102915763855,
      "step": 2112
    },
    {
      "epoch": 0.04228,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.029255167643229166,
      "learning_rate": 0.0001,
      "loss": 4.9303,
      "loss/crossentropy": 1.9510936737060547,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2839510589838028,
      "step": 2114
    },
    {
      "epoch": 0.04232,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.026753743489583332,
      "learning_rate": 0.0001,
      "loss": 5.2446,
      "loss/crossentropy": 2.0201885104179382,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30627067387104034,
      "step": 2116
    },
    {
      "epoch": 0.04236,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.022493489583333335,
      "learning_rate": 0.0001,
      "loss": 5.1411,
      "loss/crossentropy": 2.4522262811660767,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.308579683303833,
      "step": 2118
    },
    {
      "epoch": 0.0424,
      "grad_norm": 3.09375,
      "grad_norm_var": 0.04345296223958333,
      "learning_rate": 0.0001,
      "loss": 5.5535,
      "loss/crossentropy": 1.9289590120315552,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29346515238285065,
      "step": 2120
    },
    {
      "epoch": 0.04244,
      "grad_norm": 2.8125,
      "grad_norm_var": 0.03437398274739583,
      "learning_rate": 0.0001,
      "loss": 5.0588,
      "loss/crossentropy": 2.2020061016082764,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2992282509803772,
      "step": 2122
    },
    {
      "epoch": 0.04248,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.031494140625,
      "learning_rate": 0.0001,
      "loss": 5.2466,
      "loss/crossentropy": 2.180301785469055,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28323256969451904,
      "step": 2124
    },
    {
      "epoch": 0.04252,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.03435872395833333,
      "learning_rate": 0.0001,
      "loss": 4.9061,
      "loss/crossentropy": 2.1250513792037964,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.272533118724823,
      "step": 2126
    },
    {
      "epoch": 0.04256,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.037873331705729166,
      "learning_rate": 0.0001,
      "loss": 5.4375,
      "loss/crossentropy": 2.3509981632232666,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3719516545534134,
      "step": 2128
    },
    {
      "epoch": 0.0426,
      "grad_norm": 3.078125,
      "grad_norm_var": 0.03508707682291667,
      "learning_rate": 0.0001,
      "loss": 5.3067,
      "loss/crossentropy": 2.135426163673401,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3582882583141327,
      "step": 2130
    },
    {
      "epoch": 0.04264,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.0398590087890625,
      "learning_rate": 0.0001,
      "loss": 5.2406,
      "loss/crossentropy": 2.316452383995056,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30162203311920166,
      "step": 2132
    },
    {
      "epoch": 0.04268,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.03846028645833333,
      "learning_rate": 0.0001,
      "loss": 5.0372,
      "loss/crossentropy": 2.0325432419776917,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.287256121635437,
      "step": 2134
    },
    {
      "epoch": 0.04272,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.0236236572265625,
      "learning_rate": 0.0001,
      "loss": 5.1985,
      "loss/crossentropy": 2.070056974887848,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2552843391895294,
      "step": 2136
    },
    {
      "epoch": 0.04276,
      "grad_norm": 3.078125,
      "grad_norm_var": 0.029157511393229165,
      "learning_rate": 0.0001,
      "loss": 5.0623,
      "loss/crossentropy": 1.7005944848060608,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24984879791736603,
      "step": 2138
    },
    {
      "epoch": 0.0428,
      "grad_norm": 2.75,
      "grad_norm_var": 0.03203125,
      "learning_rate": 0.0001,
      "loss": 5.0862,
      "loss/crossentropy": 1.6700931787490845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2572527676820755,
      "step": 2140
    },
    {
      "epoch": 0.04284,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.03125,
      "learning_rate": 0.0001,
      "loss": 5.0186,
      "loss/crossentropy": 2.3074774742126465,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31137382984161377,
      "step": 2142
    },
    {
      "epoch": 0.04288,
      "grad_norm": 2.84375,
      "grad_norm_var": 0.0229400634765625,
      "learning_rate": 0.0001,
      "loss": 5.1973,
      "loss/crossentropy": 2.103408098220825,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30157215893268585,
      "step": 2144
    },
    {
      "epoch": 0.04292,
      "grad_norm": 2.796875,
      "grad_norm_var": 0.020979817708333334,
      "learning_rate": 0.0001,
      "loss": 4.8206,
      "loss/crossentropy": 1.8602584600448608,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28300634026527405,
      "step": 2146
    },
    {
      "epoch": 0.04296,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.019188435872395833,
      "learning_rate": 0.0001,
      "loss": 5.0525,
      "loss/crossentropy": 2.337582588195801,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28086431324481964,
      "step": 2148
    },
    {
      "epoch": 0.043,
      "grad_norm": 2.796875,
      "grad_norm_var": 0.026439412434895834,
      "learning_rate": 0.0001,
      "loss": 5.2405,
      "loss/crossentropy": 2.2635254859924316,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3113311231136322,
      "step": 2150
    },
    {
      "epoch": 0.04304,
      "grad_norm": 2.96875,
      "grad_norm_var": 0.0277984619140625,
      "learning_rate": 0.0001,
      "loss": 5.187,
      "loss/crossentropy": 2.3971948623657227,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32285284996032715,
      "step": 2152
    },
    {
      "epoch": 0.04308,
      "grad_norm": 2.875,
      "grad_norm_var": 0.021891276041666668,
      "learning_rate": 0.0001,
      "loss": 5.1438,
      "loss/crossentropy": 1.8900776505470276,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28163351118564606,
      "step": 2154
    },
    {
      "epoch": 0.04312,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.020536295572916665,
      "learning_rate": 0.0001,
      "loss": 4.9774,
      "loss/crossentropy": 1.908443808555603,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2901918590068817,
      "step": 2156
    },
    {
      "epoch": 0.04316,
      "grad_norm": 2.75,
      "grad_norm_var": 0.020409138997395833,
      "learning_rate": 0.0001,
      "loss": 4.7808,
      "loss/crossentropy": 1.8003268837928772,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2762569487094879,
      "step": 2158
    },
    {
      "epoch": 0.0432,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.0221832275390625,
      "learning_rate": 0.0001,
      "loss": 5.0477,
      "loss/crossentropy": 1.996739387512207,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24626458436250687,
      "step": 2160
    },
    {
      "epoch": 0.04324,
      "grad_norm": 2.84375,
      "grad_norm_var": 0.018880208333333332,
      "learning_rate": 0.0001,
      "loss": 5.1737,
      "loss/crossentropy": 2.0175461173057556,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31059183180332184,
      "step": 2162
    },
    {
      "epoch": 0.04328,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.019266764322916668,
      "learning_rate": 0.0001,
      "loss": 5.0448,
      "loss/crossentropy": 2.0009909868240356,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2835536003112793,
      "step": 2164
    },
    {
      "epoch": 0.04332,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.017967732747395833,
      "learning_rate": 0.0001,
      "loss": 4.9035,
      "loss/crossentropy": 1.9848785400390625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26657119393348694,
      "step": 2166
    },
    {
      "epoch": 0.04336,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.01207275390625,
      "learning_rate": 0.0001,
      "loss": 4.9108,
      "loss/crossentropy": 2.076065957546234,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2801993191242218,
      "step": 2168
    },
    {
      "epoch": 0.0434,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.011617024739583334,
      "learning_rate": 0.0001,
      "loss": 5.1425,
      "loss/crossentropy": 2.1208528876304626,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3152369260787964,
      "step": 2170
    },
    {
      "epoch": 0.04344,
      "grad_norm": 2.625,
      "grad_norm_var": 0.011400349934895833,
      "learning_rate": 0.0001,
      "loss": 5.0608,
      "loss/crossentropy": 2.1971306204795837,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31046128273010254,
      "step": 2172
    },
    {
      "epoch": 0.04348,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.013158162434895834,
      "learning_rate": 0.0001,
      "loss": 5.2445,
      "loss/crossentropy": 2.275176525115967,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3090529441833496,
      "step": 2174
    },
    {
      "epoch": 0.04352,
      "grad_norm": 2.8125,
      "grad_norm_var": 0.01197509765625,
      "learning_rate": 0.0001,
      "loss": 4.9366,
      "loss/crossentropy": 2.1574501395225525,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.280165433883667,
      "step": 2176
    },
    {
      "epoch": 0.04356,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.012548828125,
      "learning_rate": 0.0001,
      "loss": 5.2338,
      "loss/crossentropy": 2.4236754179000854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3061629384756088,
      "step": 2178
    },
    {
      "epoch": 0.0436,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.012809244791666667,
      "learning_rate": 0.0001,
      "loss": 5.1707,
      "loss/crossentropy": 2.1282758712768555,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3050261586904526,
      "step": 2180
    },
    {
      "epoch": 0.04364,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.008561197916666667,
      "learning_rate": 0.0001,
      "loss": 5.4629,
      "loss/crossentropy": 2.4244707822799683,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33464157581329346,
      "step": 2182
    },
    {
      "epoch": 0.04368,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.010904947916666666,
      "learning_rate": 0.0001,
      "loss": 5.3891,
      "loss/crossentropy": 2.289917469024658,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3181813210248947,
      "step": 2184
    },
    {
      "epoch": 0.04372,
      "grad_norm": 2.890625,
      "grad_norm_var": 0.01031494140625,
      "learning_rate": 0.0001,
      "loss": 5.4207,
      "loss/crossentropy": 2.1540024280548096,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.298043891787529,
      "step": 2186
    },
    {
      "epoch": 0.04376,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.0145660400390625,
      "learning_rate": 0.0001,
      "loss": 4.8595,
      "loss/crossentropy": 1.6615915298461914,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23485098034143448,
      "step": 2188
    },
    {
      "epoch": 0.0438,
      "grad_norm": 2.796875,
      "grad_norm_var": 0.01357421875,
      "learning_rate": 0.0001,
      "loss": 5.0595,
      "loss/crossentropy": 2.352560341358185,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2965056747198105,
      "step": 2190
    },
    {
      "epoch": 0.04384,
      "grad_norm": 2.625,
      "grad_norm_var": 0.0183746337890625,
      "learning_rate": 0.0001,
      "loss": 5.1463,
      "loss/crossentropy": 2.0864007472991943,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2859686613082886,
      "step": 2192
    },
    {
      "epoch": 0.04388,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.016600545247395834,
      "learning_rate": 0.0001,
      "loss": 5.2599,
      "loss/crossentropy": 1.8934992551803589,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24994677305221558,
      "step": 2194
    },
    {
      "epoch": 0.04392,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.02476806640625,
      "learning_rate": 0.0001,
      "loss": 4.9976,
      "loss/crossentropy": 2.2395824193954468,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2719078063964844,
      "step": 2196
    },
    {
      "epoch": 0.04396,
      "grad_norm": 2.84375,
      "grad_norm_var": 0.02515869140625,
      "learning_rate": 0.0001,
      "loss": 5.2631,
      "loss/crossentropy": 2.089230954647064,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2759791761636734,
      "step": 2198
    },
    {
      "epoch": 0.044,
      "grad_norm": 2.75,
      "grad_norm_var": 0.023567708333333333,
      "learning_rate": 0.0001,
      "loss": 5.298,
      "loss/crossentropy": 2.2770241498947144,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31005042791366577,
      "step": 2200
    },
    {
      "epoch": 0.04404,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.028938802083333333,
      "learning_rate": 0.0001,
      "loss": 4.6842,
      "loss/crossentropy": 2.067028760910034,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30712655186653137,
      "step": 2202
    },
    {
      "epoch": 0.04408,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.026395670572916665,
      "learning_rate": 0.0001,
      "loss": 5.0557,
      "loss/crossentropy": 2.4397774934768677,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3508765548467636,
      "step": 2204
    },
    {
      "epoch": 0.04412,
      "grad_norm": 2.890625,
      "grad_norm_var": 0.028880818684895834,
      "learning_rate": 0.0001,
      "loss": 4.966,
      "loss/crossentropy": 1.8136217594146729,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2495090439915657,
      "step": 2206
    },
    {
      "epoch": 0.04416,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.022945149739583334,
      "learning_rate": 0.0001,
      "loss": 5.11,
      "loss/crossentropy": 2.4620203971862793,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31419822573661804,
      "step": 2208
    },
    {
      "epoch": 0.0442,
      "grad_norm": 2.625,
      "grad_norm_var": 0.020042928059895833,
      "learning_rate": 0.0001,
      "loss": 4.9756,
      "loss/crossentropy": 1.8817986249923706,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2705220878124237,
      "step": 2210
    },
    {
      "epoch": 0.04424,
      "grad_norm": 3.125,
      "grad_norm_var": 0.02769775390625,
      "learning_rate": 0.0001,
      "loss": 5.0069,
      "loss/crossentropy": 1.9593598246574402,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.262384794652462,
      "step": 2212
    },
    {
      "epoch": 0.04428,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.03303120930989583,
      "learning_rate": 0.0001,
      "loss": 4.9133,
      "loss/crossentropy": 2.1003851294517517,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28965799510478973,
      "step": 2214
    },
    {
      "epoch": 0.04432,
      "grad_norm": 3.46875,
      "grad_norm_var": 0.07888081868489584,
      "learning_rate": 0.0001,
      "loss": 5.3243,
      "loss/crossentropy": 2.23227858543396,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3120953291654587,
      "step": 2216
    },
    {
      "epoch": 0.04436,
      "grad_norm": 2.625,
      "grad_norm_var": 0.0726226806640625,
      "learning_rate": 0.0001,
      "loss": 5.0901,
      "loss/crossentropy": 1.8880399465560913,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2740217447280884,
      "step": 2218
    },
    {
      "epoch": 0.0444,
      "grad_norm": 2.5,
      "grad_norm_var": 0.07111714680989584,
      "learning_rate": 0.0001,
      "loss": 4.9444,
      "loss/crossentropy": 2.132355511188507,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27536119520664215,
      "step": 2220
    },
    {
      "epoch": 0.04444,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.07419331868489583,
      "learning_rate": 0.0001,
      "loss": 4.7325,
      "loss/crossentropy": 1.831633746623993,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26966987550258636,
      "step": 2222
    },
    {
      "epoch": 0.04448,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.08056538899739583,
      "learning_rate": 0.0001,
      "loss": 5.004,
      "loss/crossentropy": 2.066656529903412,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27303647994995117,
      "step": 2224
    },
    {
      "epoch": 0.04452,
      "grad_norm": 2.75,
      "grad_norm_var": 0.0809234619140625,
      "learning_rate": 0.0001,
      "loss": 4.9265,
      "loss/crossentropy": 2.1416667699813843,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2781240791082382,
      "step": 2226
    },
    {
      "epoch": 0.04456,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.07366536458333334,
      "learning_rate": 0.0001,
      "loss": 5.0701,
      "loss/crossentropy": 1.7953566908836365,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27528999745845795,
      "step": 2228
    },
    {
      "epoch": 0.0446,
      "grad_norm": 2.84375,
      "grad_norm_var": 0.06728515625,
      "learning_rate": 0.0001,
      "loss": 5.0182,
      "loss/crossentropy": 2.1580333709716797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2998732179403305,
      "step": 2230
    },
    {
      "epoch": 0.04464,
      "grad_norm": 2.9375,
      "grad_norm_var": 0.034326171875,
      "learning_rate": 0.0001,
      "loss": 5.3619,
      "loss/crossentropy": 2.1685701608657837,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3457205891609192,
      "step": 2232
    },
    {
      "epoch": 0.04468,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.03345438639322917,
      "learning_rate": 0.0001,
      "loss": 4.7602,
      "loss/crossentropy": 1.9424286484718323,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2728651314973831,
      "step": 2234
    },
    {
      "epoch": 0.04472,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.029736328125,
      "learning_rate": 0.0001,
      "loss": 5.2351,
      "loss/crossentropy": 2.2802772521972656,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30141082406044006,
      "step": 2236
    },
    {
      "epoch": 0.04476,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.03277587890625,
      "learning_rate": 0.0001,
      "loss": 4.8906,
      "loss/crossentropy": 1.9490987062454224,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2573640048503876,
      "step": 2238
    },
    {
      "epoch": 0.0448,
      "grad_norm": 3.171875,
      "grad_norm_var": 0.037679036458333336,
      "learning_rate": 0.0001,
      "loss": 5.2112,
      "loss/crossentropy": 1.993924081325531,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25628305971622467,
      "step": 2240
    },
    {
      "epoch": 0.04484,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.0359283447265625,
      "learning_rate": 0.0001,
      "loss": 5.268,
      "loss/crossentropy": 2.5151875019073486,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.326447993516922,
      "step": 2242
    },
    {
      "epoch": 0.04488,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.04641520182291667,
      "learning_rate": 0.0001,
      "loss": 5.0191,
      "loss/crossentropy": 2.5175565481185913,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3036232739686966,
      "step": 2244
    },
    {
      "epoch": 0.04492,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.05388997395833333,
      "learning_rate": 0.0001,
      "loss": 4.8489,
      "loss/crossentropy": 2.020721971988678,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28087201714515686,
      "step": 2246
    },
    {
      "epoch": 0.04496,
      "grad_norm": 2.96875,
      "grad_norm_var": 0.045703125,
      "learning_rate": 0.0001,
      "loss": 5.6809,
      "loss/crossentropy": 2.4800511598587036,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3100287467241287,
      "step": 2248
    },
    {
      "epoch": 0.045,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.0449127197265625,
      "learning_rate": 0.0001,
      "loss": 4.9055,
      "loss/crossentropy": 1.826172411441803,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2782330811023712,
      "step": 2250
    },
    {
      "epoch": 0.04504,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.04533589680989583,
      "learning_rate": 0.0001,
      "loss": 5.133,
      "loss/crossentropy": 2.256316304206848,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3070906698703766,
      "step": 2252
    },
    {
      "epoch": 0.04508,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.041402180989583336,
      "learning_rate": 0.0001,
      "loss": 5.173,
      "loss/crossentropy": 1.9046601057052612,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2894355356693268,
      "step": 2254
    },
    {
      "epoch": 0.04512,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.03288472493489583,
      "learning_rate": 0.0001,
      "loss": 5.0311,
      "loss/crossentropy": 1.8359373211860657,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2705196440219879,
      "step": 2256
    },
    {
      "epoch": 0.04516,
      "grad_norm": 2.96875,
      "grad_norm_var": 0.0349273681640625,
      "learning_rate": 0.0001,
      "loss": 4.717,
      "loss/crossentropy": 2.096512258052826,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26635921001434326,
      "step": 2258
    },
    {
      "epoch": 0.0452,
      "grad_norm": 2.984375,
      "grad_norm_var": 0.031371053059895834,
      "learning_rate": 0.0001,
      "loss": 5.6736,
      "loss/crossentropy": 2.4621278047561646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3513137400150299,
      "step": 2260
    },
    {
      "epoch": 0.04524,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.020442708333333334,
      "learning_rate": 0.0001,
      "loss": 5.1413,
      "loss/crossentropy": 1.8345229029655457,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2414976954460144,
      "step": 2262
    },
    {
      "epoch": 0.04528,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.026432291666666666,
      "learning_rate": 0.0001,
      "loss": 4.9746,
      "loss/crossentropy": 2.24505877494812,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31794628500938416,
      "step": 2264
    },
    {
      "epoch": 0.04532,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.0285552978515625,
      "learning_rate": 0.0001,
      "loss": 5.1691,
      "loss/crossentropy": 2.2141382694244385,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2925301343202591,
      "step": 2266
    },
    {
      "epoch": 0.04536,
      "grad_norm": 3.1875,
      "grad_norm_var": 0.03681233723958333,
      "learning_rate": 0.0001,
      "loss": 5.0559,
      "loss/crossentropy": 2.1515613794326782,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2843547910451889,
      "step": 2268
    },
    {
      "epoch": 0.0454,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.03422749837239583,
      "learning_rate": 0.0001,
      "loss": 4.6899,
      "loss/crossentropy": 2.1234883666038513,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2981575280427933,
      "step": 2270
    },
    {
      "epoch": 0.04544,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.03806864420572917,
      "learning_rate": 0.0001,
      "loss": 4.9871,
      "loss/crossentropy": 2.1212490797042847,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.276496559381485,
      "step": 2272
    },
    {
      "epoch": 0.04548,
      "grad_norm": 2.921875,
      "grad_norm_var": 0.03574930826822917,
      "learning_rate": 0.0001,
      "loss": 5.2925,
      "loss/crossentropy": 2.4330636262893677,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2815839499235153,
      "step": 2274
    },
    {
      "epoch": 0.04552,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.028238932291666668,
      "learning_rate": 0.0001,
      "loss": 5.2874,
      "loss/crossentropy": 2.110591411590576,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2832919806241989,
      "step": 2276
    },
    {
      "epoch": 0.04556,
      "grad_norm": 2.875,
      "grad_norm_var": 0.20754801432291667,
      "learning_rate": 0.0001,
      "loss": 5.0303,
      "loss/crossentropy": 2.231989800930023,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28183089196681976,
      "step": 2278
    },
    {
      "epoch": 0.0456,
      "grad_norm": 2.84375,
      "grad_norm_var": 0.194873046875,
      "learning_rate": 0.0001,
      "loss": 5.3746,
      "loss/crossentropy": 2.1275558471679688,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29279497265815735,
      "step": 2280
    },
    {
      "epoch": 0.04564,
      "grad_norm": 2.8125,
      "grad_norm_var": 0.1891510009765625,
      "learning_rate": 0.0001,
      "loss": 5.2023,
      "loss/crossentropy": 1.7988306283950806,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2625636160373688,
      "step": 2282
    },
    {
      "epoch": 0.04568,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.18694254557291667,
      "learning_rate": 0.0001,
      "loss": 5.2017,
      "loss/crossentropy": 2.3405990600585938,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30855217576026917,
      "step": 2284
    },
    {
      "epoch": 0.04572,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.18612874348958333,
      "learning_rate": 0.0001,
      "loss": 5.4582,
      "loss/crossentropy": 2.2062121629714966,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30755001306533813,
      "step": 2286
    },
    {
      "epoch": 0.04576,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.18968098958333332,
      "learning_rate": 0.0001,
      "loss": 4.8984,
      "loss/crossentropy": 1.9439310431480408,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26844222843647003,
      "step": 2288
    },
    {
      "epoch": 0.0458,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.19010416666666666,
      "learning_rate": 0.0001,
      "loss": 5.2097,
      "loss/crossentropy": 2.3106162548065186,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30821681022644043,
      "step": 2290
    },
    {
      "epoch": 0.04584,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.19246317545572916,
      "learning_rate": 0.0001,
      "loss": 5.1401,
      "loss/crossentropy": 2.3809561729431152,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3307010233402252,
      "step": 2292
    },
    {
      "epoch": 0.04588,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.01103515625,
      "learning_rate": 0.0001,
      "loss": 5.4066,
      "loss/crossentropy": 2.209702253341675,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2795914113521576,
      "step": 2294
    },
    {
      "epoch": 0.04592,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.0158111572265625,
      "learning_rate": 0.0001,
      "loss": 4.8772,
      "loss/crossentropy": 2.4084372520446777,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31208573281764984,
      "step": 2296
    },
    {
      "epoch": 0.04596,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.0146392822265625,
      "learning_rate": 0.0001,
      "loss": 4.7757,
      "loss/crossentropy": 1.9384723901748657,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2625032365322113,
      "step": 2298
    },
    {
      "epoch": 0.046,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.013704427083333333,
      "learning_rate": 0.0001,
      "loss": 5.2455,
      "loss/crossentropy": 2.150592088699341,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.278359517455101,
      "step": 2300
    },
    {
      "epoch": 0.04604,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.01256103515625,
      "learning_rate": 0.0001,
      "loss": 5.0788,
      "loss/crossentropy": 1.8317970037460327,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2927433103322983,
      "step": 2302
    },
    {
      "epoch": 0.04608,
      "grad_norm": 3.09375,
      "grad_norm_var": 0.029564412434895833,
      "learning_rate": 0.0001,
      "loss": 5.091,
      "loss/crossentropy": 2.323367118835449,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2943577915430069,
      "step": 2304
    },
    {
      "epoch": 0.04612,
      "grad_norm": 2.859375,
      "grad_norm_var": 0.03943583170572917,
      "learning_rate": 0.0001,
      "loss": 5.5301,
      "loss/crossentropy": 2.369907855987549,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2990037202835083,
      "step": 2306
    },
    {
      "epoch": 0.04616,
      "grad_norm": 2.921875,
      "grad_norm_var": 0.04127197265625,
      "learning_rate": 0.0001,
      "loss": 4.7508,
      "loss/crossentropy": 1.691443145275116,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27280642092227936,
      "step": 2308
    },
    {
      "epoch": 0.0462,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.0464508056640625,
      "learning_rate": 0.0001,
      "loss": 4.9413,
      "loss/crossentropy": 2.2883838415145874,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3064710944890976,
      "step": 2310
    },
    {
      "epoch": 0.04624,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.035380045572916664,
      "learning_rate": 0.0001,
      "loss": 5.4165,
      "loss/crossentropy": 2.2042444944381714,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3241504430770874,
      "step": 2312
    },
    {
      "epoch": 0.04628,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.048680623372395836,
      "learning_rate": 0.0001,
      "loss": 4.6657,
      "loss/crossentropy": 1.977793574333191,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2775610163807869,
      "step": 2314
    },
    {
      "epoch": 0.04632,
      "grad_norm": 24.875,
      "grad_norm_var": 30.570881144205728,
      "learning_rate": 0.0001,
      "loss": 5.8585,
      "loss/crossentropy": 2.034530758857727,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27170561254024506,
      "step": 2316
    },
    {
      "epoch": 0.04636,
      "grad_norm": 2.875,
      "grad_norm_var": 30.404881795247395,
      "learning_rate": 0.0001,
      "loss": 5.1565,
      "loss/crossentropy": 2.439123511314392,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2999258190393448,
      "step": 2318
    },
    {
      "epoch": 0.0464,
      "grad_norm": 2.5,
      "grad_norm_var": 30.530557250976564,
      "learning_rate": 0.0001,
      "loss": 4.8907,
      "loss/crossentropy": 2.2600624561309814,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2952795475721359,
      "step": 2320
    },
    {
      "epoch": 0.04644,
      "grad_norm": 2.859375,
      "grad_norm_var": 30.544131469726562,
      "learning_rate": 0.0001,
      "loss": 5.0521,
      "loss/crossentropy": 2.144679367542267,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28713342547416687,
      "step": 2322
    },
    {
      "epoch": 0.04648,
      "grad_norm": 2.890625,
      "grad_norm_var": 30.469155883789064,
      "learning_rate": 0.0001,
      "loss": 5.5054,
      "loss/crossentropy": 2.34474778175354,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.292633980512619,
      "step": 2324
    },
    {
      "epoch": 0.04652,
      "grad_norm": 2.609375,
      "grad_norm_var": 30.491536458333332,
      "learning_rate": 0.0001,
      "loss": 4.5903,
      "loss/crossentropy": 1.96743243932724,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.265610933303833,
      "step": 2326
    },
    {
      "epoch": 0.04656,
      "grad_norm": 2.53125,
      "grad_norm_var": 30.524051920572916,
      "learning_rate": 0.0001,
      "loss": 5.2353,
      "loss/crossentropy": 2.3895785808563232,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3287513107061386,
      "step": 2328
    },
    {
      "epoch": 0.0466,
      "grad_norm": 2.984375,
      "grad_norm_var": 30.446451822916668,
      "learning_rate": 0.0001,
      "loss": 4.9017,
      "loss/crossentropy": 2.0607098937034607,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30049796402454376,
      "step": 2330
    },
    {
      "epoch": 0.04664,
      "grad_norm": 2.96875,
      "grad_norm_var": 0.08322652180989583,
      "learning_rate": 0.0001,
      "loss": 5.1698,
      "loss/crossentropy": 2.162124752998352,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2983020693063736,
      "step": 2332
    },
    {
      "epoch": 0.04668,
      "grad_norm": 2.890625,
      "grad_norm_var": 0.05139567057291667,
      "learning_rate": 0.0001,
      "loss": 5.2702,
      "loss/crossentropy": 2.34970760345459,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3322293907403946,
      "step": 2334
    },
    {
      "epoch": 0.04672,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.04527587890625,
      "learning_rate": 0.0001,
      "loss": 4.9763,
      "loss/crossentropy": 1.9286972284317017,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2752893418073654,
      "step": 2336
    },
    {
      "epoch": 0.04676,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.04159749348958333,
      "learning_rate": 0.0001,
      "loss": 5.1677,
      "loss/crossentropy": 2.182044267654419,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30808278918266296,
      "step": 2338
    },
    {
      "epoch": 0.0468,
      "grad_norm": 2.96875,
      "grad_norm_var": 0.03916727701822917,
      "learning_rate": 0.0001,
      "loss": 5.28,
      "loss/crossentropy": 2.0002610087394714,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2944178581237793,
      "step": 2340
    },
    {
      "epoch": 0.04684,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.034830729166666664,
      "learning_rate": 0.0001,
      "loss": 4.984,
      "loss/crossentropy": 2.0721842646598816,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2769011855125427,
      "step": 2342
    },
    {
      "epoch": 0.04688,
      "grad_norm": 2.625,
      "grad_norm_var": 0.03178609212239583,
      "learning_rate": 0.0001,
      "loss": 5.0911,
      "loss/crossentropy": 1.9710460305213928,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28619086742401123,
      "step": 2344
    },
    {
      "epoch": 0.04692,
      "grad_norm": 3.28125,
      "grad_norm_var": 1.54605712890625,
      "learning_rate": 0.0001,
      "loss": 5.5092,
      "loss/crossentropy": 2.0506762266159058,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31705181300640106,
      "step": 2346
    },
    {
      "epoch": 0.04696,
      "grad_norm": 2.609375,
      "grad_norm_var": 1.5601064046223958,
      "learning_rate": 0.0001,
      "loss": 5.0045,
      "loss/crossentropy": 2.0095930695533752,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2831149846315384,
      "step": 2348
    },
    {
      "epoch": 0.047,
      "grad_norm": 2.890625,
      "grad_norm_var": 1.573631795247396,
      "learning_rate": 0.0001,
      "loss": 5.19,
      "loss/crossentropy": 2.023163616657257,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2813292294740677,
      "step": 2350
    },
    {
      "epoch": 0.04704,
      "grad_norm": 2.671875,
      "grad_norm_var": 1.561424763997396,
      "learning_rate": 0.0001,
      "loss": 5.2907,
      "loss/crossentropy": 2.230435371398926,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30891451239585876,
      "step": 2352
    },
    {
      "epoch": 0.04708,
      "grad_norm": 3.0625,
      "grad_norm_var": 1.5700154622395834,
      "learning_rate": 0.0001,
      "loss": 4.9261,
      "loss/crossentropy": 2.1553521156311035,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29839709401130676,
      "step": 2354
    },
    {
      "epoch": 0.04712,
      "grad_norm": 2.96875,
      "grad_norm_var": 1.5770792643229166,
      "learning_rate": 0.0001,
      "loss": 5.2553,
      "loss/crossentropy": 2.175648272037506,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2972148358821869,
      "step": 2356
    },
    {
      "epoch": 0.04716,
      "grad_norm": 2.703125,
      "grad_norm_var": 1.5980377197265625,
      "learning_rate": 0.0001,
      "loss": 5.0436,
      "loss/crossentropy": 2.3852503299713135,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3166612535715103,
      "step": 2358
    },
    {
      "epoch": 0.0472,
      "grad_norm": 2.75,
      "grad_norm_var": 1.5826456705729166,
      "learning_rate": 0.0001,
      "loss": 5.1827,
      "loss/crossentropy": 2.1905999183654785,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29487256705760956,
      "step": 2360
    },
    {
      "epoch": 0.04724,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.0413482666015625,
      "learning_rate": 0.0001,
      "loss": 4.9295,
      "loss/crossentropy": 1.9224759340286255,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27821336686611176,
      "step": 2362
    },
    {
      "epoch": 0.04728,
      "grad_norm": 3.109375,
      "grad_norm_var": 0.04057515462239583,
      "learning_rate": 0.0001,
      "loss": 5.1078,
      "loss/crossentropy": 2.5025261640548706,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3345927745103836,
      "step": 2364
    },
    {
      "epoch": 0.04732,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.04006245930989583,
      "learning_rate": 0.0001,
      "loss": 5.0502,
      "loss/crossentropy": 2.2385451793670654,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27095621824264526,
      "step": 2366
    },
    {
      "epoch": 0.04736,
      "grad_norm": 4.5,
      "grad_norm_var": 0.22924702962239582,
      "learning_rate": 0.0001,
      "loss": 5.2761,
      "loss/crossentropy": 2.0266553163528442,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2856762409210205,
      "step": 2368
    },
    {
      "epoch": 0.0474,
      "grad_norm": 3.359375,
      "grad_norm_var": 0.24001363118489583,
      "learning_rate": 0.0001,
      "loss": 5.4918,
      "loss/crossentropy": 2.5139355659484863,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3186872750520706,
      "step": 2370
    },
    {
      "epoch": 0.04744,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.24321187337239583,
      "learning_rate": 0.0001,
      "loss": 5.1302,
      "loss/crossentropy": 2.066476881504059,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2889470160007477,
      "step": 2372
    },
    {
      "epoch": 0.04748,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.23772786458333334,
      "learning_rate": 0.0001,
      "loss": 5.0031,
      "loss/crossentropy": 2.0537307262420654,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31795741617679596,
      "step": 2374
    },
    {
      "epoch": 0.04752,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.24492085774739583,
      "learning_rate": 0.0001,
      "loss": 4.9922,
      "loss/crossentropy": 1.9254986643791199,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25411880016326904,
      "step": 2376
    },
    {
      "epoch": 0.04756,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.23855692545572918,
      "learning_rate": 0.0001,
      "loss": 5.0284,
      "loss/crossentropy": 2.221043348312378,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28534361720085144,
      "step": 2378
    },
    {
      "epoch": 0.0476,
      "grad_norm": 2.90625,
      "grad_norm_var": 0.228271484375,
      "learning_rate": 0.0001,
      "loss": 5.2106,
      "loss/crossentropy": 2.3516281843185425,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3128499984741211,
      "step": 2380
    },
    {
      "epoch": 0.04764,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.22349853515625,
      "learning_rate": 0.0001,
      "loss": 5.6266,
      "loss/crossentropy": 2.2144338488578796,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3071902245283127,
      "step": 2382
    },
    {
      "epoch": 0.04768,
      "grad_norm": 2.625,
      "grad_norm_var": 0.0567291259765625,
      "learning_rate": 0.0001,
      "loss": 5.2429,
      "loss/crossentropy": 2.3324203491210938,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.322970449924469,
      "step": 2384
    },
    {
      "epoch": 0.04772,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.03332926432291667,
      "learning_rate": 0.0001,
      "loss": 4.7732,
      "loss/crossentropy": 2.08349871635437,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2774003893136978,
      "step": 2386
    },
    {
      "epoch": 0.04776,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.03902587890625,
      "learning_rate": 0.0001,
      "loss": 4.8585,
      "loss/crossentropy": 1.9565780758857727,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29790589213371277,
      "step": 2388
    },
    {
      "epoch": 0.0478,
      "grad_norm": 2.625,
      "grad_norm_var": 0.022782389322916666,
      "learning_rate": 0.0001,
      "loss": 5.0176,
      "loss/crossentropy": 2.261398434638977,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3143853694200516,
      "step": 2390
    },
    {
      "epoch": 0.04784,
      "grad_norm": 2.96875,
      "grad_norm_var": 0.024494425455729166,
      "learning_rate": 0.0001,
      "loss": 5.0688,
      "loss/crossentropy": 1.9077460169792175,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25433091819286346,
      "step": 2392
    },
    {
      "epoch": 0.04788,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.0240142822265625,
      "learning_rate": 0.0001,
      "loss": 4.9531,
      "loss/crossentropy": 1.9948468208312988,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2598777562379837,
      "step": 2394
    },
    {
      "epoch": 0.04792,
      "grad_norm": 2.9375,
      "grad_norm_var": 0.45455729166666664,
      "learning_rate": 0.0001,
      "loss": 5.0972,
      "loss/crossentropy": 2.1177526116371155,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2775426208972931,
      "step": 2396
    },
    {
      "epoch": 0.04796,
      "grad_norm": 2.5,
      "grad_norm_var": 0.4634348551432292,
      "learning_rate": 0.0001,
      "loss": 4.8571,
      "loss/crossentropy": 2.1756062507629395,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33678852021694183,
      "step": 2398
    },
    {
      "epoch": 0.048,
      "grad_norm": 2.984375,
      "grad_norm_var": 0.4576171875,
      "learning_rate": 0.0001,
      "loss": 5.2617,
      "loss/crossentropy": 2.0923725366592407,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3255026638507843,
      "step": 2400
    },
    {
      "epoch": 0.04804,
      "grad_norm": 2.84375,
      "grad_norm_var": 0.4471181233723958,
      "learning_rate": 0.0001,
      "loss": 4.9421,
      "loss/crossentropy": 1.9236682653427124,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2642124071717262,
      "step": 2402
    },
    {
      "epoch": 0.04808,
      "grad_norm": 2.84375,
      "grad_norm_var": 0.43757222493489584,
      "learning_rate": 0.0001,
      "loss": 5.0604,
      "loss/crossentropy": 2.1742242574691772,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30006398260593414,
      "step": 2404
    },
    {
      "epoch": 0.04812,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.43835347493489585,
      "learning_rate": 0.0001,
      "loss": 4.7077,
      "loss/crossentropy": 1.7445701956748962,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25791122019290924,
      "step": 2406
    },
    {
      "epoch": 0.04816,
      "grad_norm": 4.34375,
      "grad_norm_var": 0.5614735921223958,
      "learning_rate": 0.0001,
      "loss": 5.1289,
      "loss/crossentropy": 1.8616467714309692,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2532489001750946,
      "step": 2408
    },
    {
      "epoch": 0.0482,
      "grad_norm": 3.203125,
      "grad_norm_var": 0.5608723958333334,
      "learning_rate": 0.0001,
      "loss": 5.0486,
      "loss/crossentropy": 1.9146783351898193,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2659924626350403,
      "step": 2410
    },
    {
      "epoch": 0.04824,
      "grad_norm": 2.859375,
      "grad_norm_var": 0.21818745930989583,
      "learning_rate": 0.0001,
      "loss": 5.0972,
      "loss/crossentropy": 2.179564118385315,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29064056277275085,
      "step": 2412
    },
    {
      "epoch": 0.04828,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.21357421875,
      "learning_rate": 0.0001,
      "loss": 4.9578,
      "loss/crossentropy": 2.04409658908844,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27891072630882263,
      "step": 2414
    },
    {
      "epoch": 0.04832,
      "grad_norm": 2.625,
      "grad_norm_var": 0.22568257649739584,
      "learning_rate": 0.0001,
      "loss": 4.8833,
      "loss/crossentropy": 2.590337038040161,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3158426731824875,
      "step": 2416
    },
    {
      "epoch": 0.04836,
      "grad_norm": 2.625,
      "grad_norm_var": 0.23155924479166667,
      "learning_rate": 0.0001,
      "loss": 4.6919,
      "loss/crossentropy": 1.8753941059112549,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2669401317834854,
      "step": 2418
    },
    {
      "epoch": 0.0484,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.23361714680989584,
      "learning_rate": 0.0001,
      "loss": 5.0231,
      "loss/crossentropy": 2.1412659287452698,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2839512526988983,
      "step": 2420
    },
    {
      "epoch": 0.04844,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.23371988932291668,
      "learning_rate": 0.0001,
      "loss": 5.1187,
      "loss/crossentropy": 2.545991063117981,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3331379294395447,
      "step": 2422
    },
    {
      "epoch": 0.04848,
      "grad_norm": 3.25,
      "grad_norm_var": 0.089599609375,
      "learning_rate": 0.0001,
      "loss": 5.1246,
      "loss/crossentropy": 2.12838077545166,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3043065369129181,
      "step": 2424
    },
    {
      "epoch": 0.04852,
      "grad_norm": 2.953125,
      "grad_norm_var": 0.03660380045572917,
      "learning_rate": 0.0001,
      "loss": 5.3821,
      "loss/crossentropy": 2.1983221769332886,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2846619784832001,
      "step": 2426
    },
    {
      "epoch": 0.04856,
      "grad_norm": 3.0625,
      "grad_norm_var": 0.03819986979166667,
      "learning_rate": 0.0001,
      "loss": 5.1044,
      "loss/crossentropy": 2.241136312484741,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30275705456733704,
      "step": 2428
    },
    {
      "epoch": 0.0486,
      "grad_norm": 2.890625,
      "grad_norm_var": 0.03831278483072917,
      "learning_rate": 0.0001,
      "loss": 5.203,
      "loss/crossentropy": 2.097459554672241,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2986721396446228,
      "step": 2430
    },
    {
      "epoch": 0.04864,
      "grad_norm": 4.4375,
      "grad_norm_var": 0.20159403483072916,
      "learning_rate": 0.0001,
      "loss": 5.1158,
      "loss/crossentropy": 2.333081007003784,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2975248098373413,
      "step": 2432
    },
    {
      "epoch": 0.04868,
      "grad_norm": 2.625,
      "grad_norm_var": 0.19758707682291668,
      "learning_rate": 0.0001,
      "loss": 4.9183,
      "loss/crossentropy": 2.3171510696411133,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28452740609645844,
      "step": 2434
    },
    {
      "epoch": 0.04872,
      "grad_norm": 2.375,
      "grad_norm_var": 0.2141754150390625,
      "learning_rate": 0.0001,
      "loss": 4.8853,
      "loss/crossentropy": 1.8334497213363647,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25271379947662354,
      "step": 2436
    },
    {
      "epoch": 0.04876,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.20258687337239584,
      "learning_rate": 0.0001,
      "loss": 5.3809,
      "loss/crossentropy": 2.2712661027908325,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30827929079532623,
      "step": 2438
    },
    {
      "epoch": 0.0488,
      "grad_norm": 3.3125,
      "grad_norm_var": 0.20465087890625,
      "learning_rate": 0.0001,
      "loss": 5.0863,
      "loss/crossentropy": 2.160263180732727,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3078690320253372,
      "step": 2440
    },
    {
      "epoch": 0.04884,
      "grad_norm": 2.96875,
      "grad_norm_var": 0.20429585774739584,
      "learning_rate": 0.0001,
      "loss": 5.2224,
      "loss/crossentropy": 2.071319878101349,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2733730524778366,
      "step": 2442
    },
    {
      "epoch": 0.04888,
      "grad_norm": 2.796875,
      "grad_norm_var": 0.203076171875,
      "learning_rate": 0.0001,
      "loss": 5.1476,
      "loss/crossentropy": 2.0742560029029846,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2835587412118912,
      "step": 2444
    },
    {
      "epoch": 0.04892,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.24807535807291667,
      "learning_rate": 0.0001,
      "loss": 5.0211,
      "loss/crossentropy": 1.8836837410926819,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3089100867509842,
      "step": 2446
    },
    {
      "epoch": 0.04896,
      "grad_norm": 2.875,
      "grad_norm_var": 0.177099609375,
      "learning_rate": 0.0001,
      "loss": 4.9537,
      "loss/crossentropy": 1.8539315462112427,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2697101980447769,
      "step": 2448
    },
    {
      "epoch": 0.049,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.17392578125,
      "learning_rate": 0.0001,
      "loss": 5.1907,
      "loss/crossentropy": 2.219490647315979,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2898380011320114,
      "step": 2450
    },
    {
      "epoch": 0.04904,
      "grad_norm": 3.1875,
      "grad_norm_var": 0.13853251139322917,
      "learning_rate": 0.0001,
      "loss": 5.468,
      "loss/crossentropy": 2.328765392303467,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.4013051837682724,
      "step": 2452
    },
    {
      "epoch": 0.04908,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.14879557291666667,
      "learning_rate": 0.0001,
      "loss": 4.8967,
      "loss/crossentropy": 1.9204192161560059,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25281261652708054,
      "step": 2454
    },
    {
      "epoch": 0.04912,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.15563151041666667,
      "learning_rate": 0.0001,
      "loss": 5.0935,
      "loss/crossentropy": 2.377043604850769,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2928486764431,
      "step": 2456
    },
    {
      "epoch": 0.04916,
      "grad_norm": 2.796875,
      "grad_norm_var": 0.15930989583333333,
      "learning_rate": 0.0001,
      "loss": 5.4528,
      "loss/crossentropy": 2.4364209175109863,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3111976683139801,
      "step": 2458
    },
    {
      "epoch": 0.0492,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.15985921223958333,
      "learning_rate": 0.0001,
      "loss": 5.1357,
      "loss/crossentropy": 2.3738330602645874,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3019126206636429,
      "step": 2460
    },
    {
      "epoch": 0.04924,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.1141265869140625,
      "learning_rate": 0.0001,
      "loss": 5.2876,
      "loss/crossentropy": 2.4575772285461426,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3428986072540283,
      "step": 2462
    },
    {
      "epoch": 0.04928,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.028888956705729166,
      "learning_rate": 0.0001,
      "loss": 4.6505,
      "loss/crossentropy": 1.9849627017974854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27611130475997925,
      "step": 2464
    },
    {
      "epoch": 0.04932,
      "grad_norm": 2.625,
      "grad_norm_var": 0.030402628580729167,
      "learning_rate": 0.0001,
      "loss": 4.8482,
      "loss/crossentropy": 1.9617170691490173,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.257377490401268,
      "step": 2466
    },
    {
      "epoch": 0.04936,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.01871337890625,
      "learning_rate": 0.0001,
      "loss": 5.1739,
      "loss/crossentropy": 2.4031273126602173,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28843145072460175,
      "step": 2468
    },
    {
      "epoch": 0.0494,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.0165435791015625,
      "learning_rate": 0.0001,
      "loss": 4.8452,
      "loss/crossentropy": 1.7263792753219604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25591571629047394,
      "step": 2470
    },
    {
      "epoch": 0.04944,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.0191802978515625,
      "learning_rate": 0.0001,
      "loss": 4.7154,
      "loss/crossentropy": 2.106898784637451,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.280623197555542,
      "step": 2472
    },
    {
      "epoch": 0.04948,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.018358357747395835,
      "learning_rate": 0.0001,
      "loss": 5.0092,
      "loss/crossentropy": 2.325208902359009,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27794161438941956,
      "step": 2474
    },
    {
      "epoch": 0.04952,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.013736979166666666,
      "learning_rate": 0.0001,
      "loss": 5.1128,
      "loss/crossentropy": 2.367414712905884,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31346653401851654,
      "step": 2476
    },
    {
      "epoch": 0.04956,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.013792928059895833,
      "learning_rate": 0.0001,
      "loss": 5.2611,
      "loss/crossentropy": 2.191115140914917,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30560287833213806,
      "step": 2478
    },
    {
      "epoch": 0.0496,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.012430826822916666,
      "learning_rate": 0.0001,
      "loss": 4.9993,
      "loss/crossentropy": 2.2559624314308167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3029457628726959,
      "step": 2480
    },
    {
      "epoch": 0.04964,
      "grad_norm": 3.390625,
      "grad_norm_var": 0.04160868326822917,
      "learning_rate": 0.0001,
      "loss": 5.125,
      "loss/crossentropy": 2.0088382363319397,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2923784404993057,
      "step": 2482
    },
    {
      "epoch": 0.04968,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.03943583170572917,
      "learning_rate": 0.0001,
      "loss": 4.8858,
      "loss/crossentropy": 1.8445329070091248,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2587483897805214,
      "step": 2484
    },
    {
      "epoch": 0.04972,
      "grad_norm": 2.8125,
      "grad_norm_var": 0.03534749348958333,
      "learning_rate": 0.0001,
      "loss": 4.7918,
      "loss/crossentropy": 2.015140950679779,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27480585873126984,
      "step": 2486
    },
    {
      "epoch": 0.04976,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.030833943684895834,
      "learning_rate": 0.0001,
      "loss": 5.1959,
      "loss/crossentropy": 1.918801188468933,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28786011040210724,
      "step": 2488
    },
    {
      "epoch": 0.0498,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.030269368489583334,
      "learning_rate": 0.0001,
      "loss": 5.0108,
      "loss/crossentropy": 1.9899121522903442,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25822708010673523,
      "step": 2490
    },
    {
      "epoch": 0.04984,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.037206013997395836,
      "learning_rate": 0.0001,
      "loss": 4.7259,
      "loss/crossentropy": 2.3775535821914673,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31345370411872864,
      "step": 2492
    },
    {
      "epoch": 0.04988,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.03791402180989583,
      "learning_rate": 0.0001,
      "loss": 4.9496,
      "loss/crossentropy": 2.0874632596969604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2836592495441437,
      "step": 2494
    },
    {
      "epoch": 0.04992,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.046019490559895834,
      "learning_rate": 0.0001,
      "loss": 5.2404,
      "loss/crossentropy": 2.226976454257965,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27826404571533203,
      "step": 2496
    },
    {
      "epoch": 0.04996,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.023900349934895832,
      "learning_rate": 0.0001,
      "loss": 5.1503,
      "loss/crossentropy": 2.4569294452667236,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31981319189071655,
      "step": 2498
    },
    {
      "epoch": 0.05,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.025581868489583333,
      "learning_rate": 0.0001,
      "loss": 4.922,
      "loss/crossentropy": 2.1134212017059326,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28260529041290283,
      "step": 2500
    },
    {
      "epoch": 0.05004,
      "grad_norm": 3.0,
      "grad_norm_var": 0.029195149739583332,
      "learning_rate": 0.0001,
      "loss": 5.3555,
      "loss/crossentropy": 2.2914888858795166,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3182682394981384,
      "step": 2502
    },
    {
      "epoch": 0.05008,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.030078125,
      "learning_rate": 0.0001,
      "loss": 4.9644,
      "loss/crossentropy": 2.3261003494262695,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30558250844478607,
      "step": 2504
    },
    {
      "epoch": 0.05012,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.030598958333333332,
      "learning_rate": 0.0001,
      "loss": 4.9517,
      "loss/crossentropy": 2.351989507675171,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2934701144695282,
      "step": 2506
    },
    {
      "epoch": 0.05016,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.025031534830729167,
      "learning_rate": 0.0001,
      "loss": 4.71,
      "loss/crossentropy": 1.8742690086364746,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26305729895830154,
      "step": 2508
    },
    {
      "epoch": 0.0502,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.0247222900390625,
      "learning_rate": 0.0001,
      "loss": 4.8214,
      "loss/crossentropy": 2.1668856143951416,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2728075534105301,
      "step": 2510
    },
    {
      "epoch": 0.05024,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.0137115478515625,
      "learning_rate": 0.0001,
      "loss": 4.7171,
      "loss/crossentropy": 1.773424208164215,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26808495819568634,
      "step": 2512
    },
    {
      "epoch": 0.05028,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.013109334309895833,
      "learning_rate": 0.0001,
      "loss": 4.9224,
      "loss/crossentropy": 1.7541643977165222,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2600491940975189,
      "step": 2514
    },
    {
      "epoch": 0.05032,
      "grad_norm": 2.796875,
      "grad_norm_var": 0.011324055989583333,
      "learning_rate": 0.0001,
      "loss": 5.129,
      "loss/crossentropy": 1.9693496227264404,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2844541072845459,
      "step": 2516
    },
    {
      "epoch": 0.05036,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.009065755208333333,
      "learning_rate": 0.0001,
      "loss": 4.9202,
      "loss/crossentropy": 1.7539461851119995,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2542608380317688,
      "step": 2518
    },
    {
      "epoch": 0.0504,
      "grad_norm": 2.984375,
      "grad_norm_var": 0.017878214518229168,
      "learning_rate": 0.0001,
      "loss": 5.0686,
      "loss/crossentropy": 2.155138313770294,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3023010194301605,
      "step": 2520
    },
    {
      "epoch": 0.05044,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.018619791666666666,
      "learning_rate": 0.0001,
      "loss": 4.9674,
      "loss/crossentropy": 2.069350838661194,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26573850214481354,
      "step": 2522
    },
    {
      "epoch": 0.05048,
      "grad_norm": 2.5,
      "grad_norm_var": 0.022101847330729167,
      "learning_rate": 0.0001,
      "loss": 5.0295,
      "loss/crossentropy": 2.1864534616470337,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27845603227615356,
      "step": 2524
    },
    {
      "epoch": 0.05052,
      "grad_norm": 2.5,
      "grad_norm_var": 0.023509724934895834,
      "learning_rate": 0.0001,
      "loss": 5.1173,
      "loss/crossentropy": 2.2462135553359985,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29221346974372864,
      "step": 2526
    },
    {
      "epoch": 0.05056,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.030492146809895832,
      "learning_rate": 0.0001,
      "loss": 5.0027,
      "loss/crossentropy": 2.043266773223877,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25857964158058167,
      "step": 2528
    },
    {
      "epoch": 0.0506,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.0299957275390625,
      "learning_rate": 0.0001,
      "loss": 4.9672,
      "loss/crossentropy": 1.8892702460289001,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2879898101091385,
      "step": 2530
    },
    {
      "epoch": 0.05064,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.03238525390625,
      "learning_rate": 0.0001,
      "loss": 4.5332,
      "loss/crossentropy": 1.9220558404922485,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26652154326438904,
      "step": 2532
    },
    {
      "epoch": 0.05068,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.03430989583333333,
      "learning_rate": 0.0001,
      "loss": 4.4176,
      "loss/crossentropy": 1.7282914519309998,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2367371916770935,
      "step": 2534
    },
    {
      "epoch": 0.05072,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.021092732747395832,
      "learning_rate": 0.0001,
      "loss": 4.9589,
      "loss/crossentropy": 1.729803204536438,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25624871999025345,
      "step": 2536
    },
    {
      "epoch": 0.05076,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.020873006184895834,
      "learning_rate": 0.0001,
      "loss": 4.6952,
      "loss/crossentropy": 2.0921449661254883,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2855897545814514,
      "step": 2538
    },
    {
      "epoch": 0.0508,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.015404256184895833,
      "learning_rate": 0.0001,
      "loss": 4.8967,
      "loss/crossentropy": 2.0569751858711243,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2829667925834656,
      "step": 2540
    },
    {
      "epoch": 0.05084,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.015973917643229165,
      "learning_rate": 0.0001,
      "loss": 5.2438,
      "loss/crossentropy": 1.983904242515564,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30939212441444397,
      "step": 2542
    },
    {
      "epoch": 0.05088,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.011864217122395833,
      "learning_rate": 0.0001,
      "loss": 4.9968,
      "loss/crossentropy": 2.2631462812423706,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27867285907268524,
      "step": 2544
    },
    {
      "epoch": 0.05092,
      "grad_norm": 3.078125,
      "grad_norm_var": 0.023856608072916667,
      "learning_rate": 0.0001,
      "loss": 4.9621,
      "loss/crossentropy": 1.9918989539146423,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27703428268432617,
      "step": 2546
    },
    {
      "epoch": 0.05096,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.021110026041666667,
      "learning_rate": 0.0001,
      "loss": 5.2041,
      "loss/crossentropy": 2.1356931924819946,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27815964818000793,
      "step": 2548
    },
    {
      "epoch": 0.051,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.019071451822916665,
      "learning_rate": 0.0001,
      "loss": 4.6514,
      "loss/crossentropy": 2.172751545906067,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2971910834312439,
      "step": 2550
    },
    {
      "epoch": 0.05104,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.019527180989583334,
      "learning_rate": 0.0001,
      "loss": 5.0178,
      "loss/crossentropy": 2.0799094438552856,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30459292232990265,
      "step": 2552
    },
    {
      "epoch": 0.05108,
      "grad_norm": 2.5,
      "grad_norm_var": 0.0211090087890625,
      "learning_rate": 0.0001,
      "loss": 4.8235,
      "loss/crossentropy": 1.7769129872322083,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2526697665452957,
      "step": 2554
    },
    {
      "epoch": 0.05112,
      "grad_norm": 2.8125,
      "grad_norm_var": 0.024494425455729166,
      "learning_rate": 0.0001,
      "loss": 4.8457,
      "loss/crossentropy": 2.044790804386139,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2821648418903351,
      "step": 2556
    },
    {
      "epoch": 0.05116,
      "grad_norm": 2.921875,
      "grad_norm_var": 0.0287109375,
      "learning_rate": 0.0001,
      "loss": 5.5336,
      "loss/crossentropy": 2.3708614110946655,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30810464918613434,
      "step": 2558
    },
    {
      "epoch": 0.0512,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.028709920247395833,
      "learning_rate": 0.0001,
      "loss": 5.2385,
      "loss/crossentropy": 2.2216718196868896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2919304668903351,
      "step": 2560
    },
    {
      "epoch": 0.05124,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.019466145833333334,
      "learning_rate": 0.0001,
      "loss": 5.28,
      "loss/crossentropy": 2.4692097902297974,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30620162189006805,
      "step": 2562
    },
    {
      "epoch": 0.05128,
      "grad_norm": 2.984375,
      "grad_norm_var": 0.026936848958333332,
      "learning_rate": 0.0001,
      "loss": 4.9476,
      "loss/crossentropy": 2.0491623282432556,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25617313385009766,
      "step": 2564
    },
    {
      "epoch": 0.05132,
      "grad_norm": 2.8125,
      "grad_norm_var": 0.023368326822916667,
      "learning_rate": 0.0001,
      "loss": 4.958,
      "loss/crossentropy": 1.8305597305297852,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25237561762332916,
      "step": 2566
    },
    {
      "epoch": 0.05136,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.0223297119140625,
      "learning_rate": 0.0001,
      "loss": 4.9853,
      "loss/crossentropy": 1.9471853971481323,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2712964415550232,
      "step": 2568
    },
    {
      "epoch": 0.0514,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.0198638916015625,
      "learning_rate": 0.0001,
      "loss": 5.0932,
      "loss/crossentropy": 2.575412631034851,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3106851130723953,
      "step": 2570
    },
    {
      "epoch": 0.05144,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.020686848958333334,
      "learning_rate": 0.0001,
      "loss": 4.6755,
      "loss/crossentropy": 2.0210241079330444,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.269680991768837,
      "step": 2572
    },
    {
      "epoch": 0.05148,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.02076416015625,
      "learning_rate": 0.0001,
      "loss": 4.6308,
      "loss/crossentropy": 1.9054389595985413,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24933087825775146,
      "step": 2574
    },
    {
      "epoch": 0.05152,
      "grad_norm": 2.625,
      "grad_norm_var": 0.021284993489583334,
      "learning_rate": 0.0001,
      "loss": 4.9682,
      "loss/crossentropy": 2.142069697380066,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3162301778793335,
      "step": 2576
    },
    {
      "epoch": 0.05156,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.0197662353515625,
      "learning_rate": 0.0001,
      "loss": 5.018,
      "loss/crossentropy": 1.9952309727668762,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2560836151242256,
      "step": 2578
    },
    {
      "epoch": 0.0516,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.016405232747395835,
      "learning_rate": 0.0001,
      "loss": 4.8889,
      "loss/crossentropy": 2.0579317212104797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26214616745710373,
      "step": 2580
    },
    {
      "epoch": 0.05164,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.016927083333333332,
      "learning_rate": 0.0001,
      "loss": 4.6306,
      "loss/crossentropy": 2.076499104499817,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26244185864925385,
      "step": 2582
    },
    {
      "epoch": 0.05168,
      "grad_norm": 2.890625,
      "grad_norm_var": 0.026146443684895833,
      "learning_rate": 0.0001,
      "loss": 4.9393,
      "loss/crossentropy": 2.2277501821517944,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32310059666633606,
      "step": 2584
    },
    {
      "epoch": 0.05172,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.03141988118489583,
      "learning_rate": 0.0001,
      "loss": 5.0929,
      "loss/crossentropy": 2.101436138153076,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26044395565986633,
      "step": 2586
    },
    {
      "epoch": 0.05176,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.030939737955729168,
      "learning_rate": 0.0001,
      "loss": 5.045,
      "loss/crossentropy": 2.2617305517196655,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2768043726682663,
      "step": 2588
    },
    {
      "epoch": 0.0518,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.0349029541015625,
      "learning_rate": 0.0001,
      "loss": 4.989,
      "loss/crossentropy": 2.2669495940208435,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2941794842481613,
      "step": 2590
    },
    {
      "epoch": 0.05184,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.03871968587239583,
      "learning_rate": 0.0001,
      "loss": 4.7676,
      "loss/crossentropy": 1.8102391958236694,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24766983091831207,
      "step": 2592
    },
    {
      "epoch": 0.05188,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.04006754557291667,
      "learning_rate": 0.0001,
      "loss": 5.0022,
      "loss/crossentropy": 2.1426219940185547,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2808763086795807,
      "step": 2594
    },
    {
      "epoch": 0.05192,
      "grad_norm": 2.75,
      "grad_norm_var": 0.0368804931640625,
      "learning_rate": 0.0001,
      "loss": 4.8649,
      "loss/crossentropy": 2.0731321573257446,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28768520057201385,
      "step": 2596
    },
    {
      "epoch": 0.05196,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.03619384765625,
      "learning_rate": 0.0001,
      "loss": 4.8019,
      "loss/crossentropy": 1.8902159333229065,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27209727466106415,
      "step": 2598
    },
    {
      "epoch": 0.052,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.030855305989583335,
      "learning_rate": 0.0001,
      "loss": 4.837,
      "loss/crossentropy": 2.1860097646713257,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3191404938697815,
      "step": 2600
    },
    {
      "epoch": 0.05204,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.027074178059895832,
      "learning_rate": 0.0001,
      "loss": 4.8471,
      "loss/crossentropy": 2.1010658740997314,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2924908995628357,
      "step": 2602
    },
    {
      "epoch": 0.05208,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.0285064697265625,
      "learning_rate": 0.0001,
      "loss": 5.1725,
      "loss/crossentropy": 2.0668399930000305,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2691944092512131,
      "step": 2604
    },
    {
      "epoch": 0.05212,
      "grad_norm": 2.8125,
      "grad_norm_var": 0.023824055989583332,
      "learning_rate": 0.0001,
      "loss": 5.0115,
      "loss/crossentropy": 2.310541272163391,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31448885798454285,
      "step": 2606
    },
    {
      "epoch": 0.05216,
      "grad_norm": 2.75,
      "grad_norm_var": 0.0222808837890625,
      "learning_rate": 0.0001,
      "loss": 4.7731,
      "loss/crossentropy": 2.023577332496643,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28528447449207306,
      "step": 2608
    },
    {
      "epoch": 0.0522,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.0217193603515625,
      "learning_rate": 0.0001,
      "loss": 4.8408,
      "loss/crossentropy": 2.0232901573181152,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25540125370025635,
      "step": 2610
    },
    {
      "epoch": 0.05224,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.019554646809895833,
      "learning_rate": 0.0001,
      "loss": 4.8374,
      "loss/crossentropy": 2.147561550140381,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2914447784423828,
      "step": 2612
    },
    {
      "epoch": 0.05228,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.0193359375,
      "learning_rate": 0.0001,
      "loss": 4.8044,
      "loss/crossentropy": 1.9136184453964233,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27879445254802704,
      "step": 2614
    },
    {
      "epoch": 0.05232,
      "grad_norm": 3.03125,
      "grad_norm_var": 0.0606842041015625,
      "learning_rate": 0.0001,
      "loss": 5.1245,
      "loss/crossentropy": 2.1118494272232056,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2990800142288208,
      "step": 2616
    },
    {
      "epoch": 0.05236,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.0599761962890625,
      "learning_rate": 0.0001,
      "loss": 5.2189,
      "loss/crossentropy": 1.9564325213432312,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2805679142475128,
      "step": 2618
    },
    {
      "epoch": 0.0524,
      "grad_norm": 3.046875,
      "grad_norm_var": 0.06641337076822916,
      "learning_rate": 0.0001,
      "loss": 4.9511,
      "loss/crossentropy": 2.0683051347732544,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2906472980976105,
      "step": 2620
    },
    {
      "epoch": 0.05244,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.06653645833333334,
      "learning_rate": 0.0001,
      "loss": 5.058,
      "loss/crossentropy": 2.0510823130607605,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28148986399173737,
      "step": 2622
    },
    {
      "epoch": 0.05248,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.06256103515625,
      "learning_rate": 0.0001,
      "loss": 5.2113,
      "loss/crossentropy": 2.2972904443740845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32902073860168457,
      "step": 2624
    },
    {
      "epoch": 0.05252,
      "grad_norm": 2.890625,
      "grad_norm_var": 0.0582427978515625,
      "learning_rate": 0.0001,
      "loss": 5.0939,
      "loss/crossentropy": 1.9502179026603699,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2844041585922241,
      "step": 2626
    },
    {
      "epoch": 0.05256,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.05607808430989583,
      "learning_rate": 0.0001,
      "loss": 5.1078,
      "loss/crossentropy": 2.1577298045158386,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2673380598425865,
      "step": 2628
    },
    {
      "epoch": 0.0526,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.05271708170572917,
      "learning_rate": 0.0001,
      "loss": 5.0514,
      "loss/crossentropy": 2.1707664132118225,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2697260081768036,
      "step": 2630
    },
    {
      "epoch": 0.05264,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.017606608072916665,
      "learning_rate": 0.0001,
      "loss": 4.9617,
      "loss/crossentropy": 2.0975311398506165,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27033862471580505,
      "step": 2632
    },
    {
      "epoch": 0.05268,
      "grad_norm": 3.25,
      "grad_norm_var": 0.033869425455729164,
      "learning_rate": 0.0001,
      "loss": 5.1841,
      "loss/crossentropy": 2.197197914123535,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28717951476573944,
      "step": 2634
    },
    {
      "epoch": 0.05272,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.03570556640625,
      "learning_rate": 0.0001,
      "loss": 5.2558,
      "loss/crossentropy": 2.2898266315460205,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31359314918518066,
      "step": 2636
    },
    {
      "epoch": 0.05276,
      "grad_norm": 2.796875,
      "grad_norm_var": 0.0347564697265625,
      "learning_rate": 0.0001,
      "loss": 5.4849,
      "loss/crossentropy": 2.525710701942444,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3199215829372406,
      "step": 2638
    },
    {
      "epoch": 0.0528,
      "grad_norm": 2.796875,
      "grad_norm_var": 0.03349609375,
      "learning_rate": 0.0001,
      "loss": 5.0495,
      "loss/crossentropy": 2.2799761295318604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28715676069259644,
      "step": 2640
    },
    {
      "epoch": 0.05284,
      "grad_norm": 2.796875,
      "grad_norm_var": 0.032373046875,
      "learning_rate": 0.0001,
      "loss": 4.9837,
      "loss/crossentropy": 1.8681190013885498,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25435833632946014,
      "step": 2642
    },
    {
      "epoch": 0.05288,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.03730061848958333,
      "learning_rate": 0.0001,
      "loss": 4.807,
      "loss/crossentropy": 1.9067540168762207,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25444281101226807,
      "step": 2644
    },
    {
      "epoch": 0.05292,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.03732096354166667,
      "learning_rate": 0.0001,
      "loss": 5.0326,
      "loss/crossentropy": 2.370971202850342,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31173495948314667,
      "step": 2646
    },
    {
      "epoch": 0.05296,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.03752848307291667,
      "learning_rate": 0.0001,
      "loss": 5.1148,
      "loss/crossentropy": 2.1829749941825867,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29970400035381317,
      "step": 2648
    },
    {
      "epoch": 0.053,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.020052083333333335,
      "learning_rate": 0.0001,
      "loss": 5.1086,
      "loss/crossentropy": 2.0818498134613037,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27712464332580566,
      "step": 2650
    },
    {
      "epoch": 0.05304,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.010789998372395833,
      "learning_rate": 0.0001,
      "loss": 5.1071,
      "loss/crossentropy": 2.2080377340316772,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27552157640457153,
      "step": 2652
    },
    {
      "epoch": 0.05308,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.01109619140625,
      "learning_rate": 0.0001,
      "loss": 4.9685,
      "loss/crossentropy": 1.7115904092788696,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23266373574733734,
      "step": 2654
    },
    {
      "epoch": 0.05312,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.009598795572916667,
      "learning_rate": 0.0001,
      "loss": 4.8654,
      "loss/crossentropy": 2.1736810207366943,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29696571826934814,
      "step": 2656
    },
    {
      "epoch": 0.05316,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.012886555989583333,
      "learning_rate": 0.0001,
      "loss": 5.0834,
      "loss/crossentropy": 2.2366485595703125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3062315583229065,
      "step": 2658
    },
    {
      "epoch": 0.0532,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.018748982747395834,
      "learning_rate": 0.0001,
      "loss": 5.0888,
      "loss/crossentropy": 1.9835070371627808,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.35663366317749023,
      "step": 2660
    },
    {
      "epoch": 0.05324,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.026688639322916666,
      "learning_rate": 0.0001,
      "loss": 4.6725,
      "loss/crossentropy": 2.148723840713501,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2687191218137741,
      "step": 2662
    },
    {
      "epoch": 0.05328,
      "grad_norm": 2.625,
      "grad_norm_var": 0.02789306640625,
      "learning_rate": 0.0001,
      "loss": 4.9191,
      "loss/crossentropy": 2.2642128467559814,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2738381028175354,
      "step": 2664
    },
    {
      "epoch": 0.05332,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.028483072916666668,
      "learning_rate": 0.0001,
      "loss": 4.8209,
      "loss/crossentropy": 1.839052438735962,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2566695362329483,
      "step": 2666
    },
    {
      "epoch": 0.05336,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.026927693684895834,
      "learning_rate": 0.0001,
      "loss": 4.8771,
      "loss/crossentropy": 2.083684980869293,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2632910907268524,
      "step": 2668
    },
    {
      "epoch": 0.0534,
      "grad_norm": 2.75,
      "grad_norm_var": 0.028434244791666667,
      "learning_rate": 0.0001,
      "loss": 4.8585,
      "loss/crossentropy": 2.2125936150550842,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31363190710544586,
      "step": 2670
    },
    {
      "epoch": 0.05344,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.029157511393229165,
      "learning_rate": 0.0001,
      "loss": 5.0354,
      "loss/crossentropy": 2.2075263261795044,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2850564122200012,
      "step": 2672
    },
    {
      "epoch": 0.05348,
      "grad_norm": 2.859375,
      "grad_norm_var": 0.026656087239583334,
      "learning_rate": 0.0001,
      "loss": 5.0145,
      "loss/crossentropy": 2.0876463651657104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2972792685031891,
      "step": 2674
    },
    {
      "epoch": 0.05352,
      "grad_norm": 2.84375,
      "grad_norm_var": 0.020406087239583332,
      "learning_rate": 0.0001,
      "loss": 5.2755,
      "loss/crossentropy": 2.4033172130584717,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.304116889834404,
      "step": 2676
    },
    {
      "epoch": 0.05356,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.014501953125,
      "learning_rate": 0.0001,
      "loss": 5.0921,
      "loss/crossentropy": 2.30586314201355,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3264722675085068,
      "step": 2678
    },
    {
      "epoch": 0.0536,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.015152994791666667,
      "learning_rate": 0.0001,
      "loss": 5.0941,
      "loss/crossentropy": 2.2175174951553345,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2661540359258652,
      "step": 2680
    },
    {
      "epoch": 0.05364,
      "grad_norm": 2.921875,
      "grad_norm_var": 0.022606404622395833,
      "learning_rate": 0.0001,
      "loss": 5.1162,
      "loss/crossentropy": 2.0583900213241577,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26995618641376495,
      "step": 2682
    },
    {
      "epoch": 0.05368,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.020173136393229166,
      "learning_rate": 0.0001,
      "loss": 4.9357,
      "loss/crossentropy": 2.310309052467346,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29642508924007416,
      "step": 2684
    },
    {
      "epoch": 0.05372,
      "grad_norm": 3.15625,
      "grad_norm_var": 0.03332926432291667,
      "learning_rate": 0.0001,
      "loss": 4.7945,
      "loss/crossentropy": 2.134134352207184,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2619224041700363,
      "step": 2686
    },
    {
      "epoch": 0.05376,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.03306884765625,
      "learning_rate": 0.0001,
      "loss": 4.8411,
      "loss/crossentropy": 1.930562138557434,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27108173072338104,
      "step": 2688
    },
    {
      "epoch": 0.0538,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.03455403645833333,
      "learning_rate": 0.0001,
      "loss": 4.9591,
      "loss/crossentropy": 2.3414769172668457,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.270420178771019,
      "step": 2690
    },
    {
      "epoch": 0.05384,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.0356842041015625,
      "learning_rate": 0.0001,
      "loss": 4.83,
      "loss/crossentropy": 2.1726938486099243,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2710433751344681,
      "step": 2692
    },
    {
      "epoch": 0.05388,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.0361968994140625,
      "learning_rate": 0.0001,
      "loss": 5.0163,
      "loss/crossentropy": 2.220117926597595,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3212582617998123,
      "step": 2694
    },
    {
      "epoch": 0.05392,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.03297119140625,
      "learning_rate": 0.0001,
      "loss": 5.1845,
      "loss/crossentropy": 2.2557668685913086,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3096832036972046,
      "step": 2696
    },
    {
      "epoch": 0.05396,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.0246246337890625,
      "learning_rate": 0.0001,
      "loss": 5.0399,
      "loss/crossentropy": 1.94975346326828,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2530350238084793,
      "step": 2698
    },
    {
      "epoch": 0.054,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.024169921875,
      "learning_rate": 0.0001,
      "loss": 5.19,
      "loss/crossentropy": 2.4622775316238403,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3135389983654022,
      "step": 2700
    },
    {
      "epoch": 0.05404,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.0110748291015625,
      "learning_rate": 0.0001,
      "loss": 5.2005,
      "loss/crossentropy": 2.5367363691329956,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30742934346199036,
      "step": 2702
    },
    {
      "epoch": 0.05408,
      "grad_norm": 2.5,
      "grad_norm_var": 0.014631144205729167,
      "learning_rate": 0.0001,
      "loss": 5.146,
      "loss/crossentropy": 2.5733184814453125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33410580456256866,
      "step": 2704
    },
    {
      "epoch": 0.05412,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.011725870768229167,
      "learning_rate": 0.0001,
      "loss": 4.8888,
      "loss/crossentropy": 1.9339997172355652,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28760699927806854,
      "step": 2706
    },
    {
      "epoch": 0.05416,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.010724894205729167,
      "learning_rate": 0.0001,
      "loss": 4.8719,
      "loss/crossentropy": 1.8515672087669373,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23648252338171005,
      "step": 2708
    },
    {
      "epoch": 0.0542,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.014420572916666667,
      "learning_rate": 0.0001,
      "loss": 4.6598,
      "loss/crossentropy": 2.0973429083824158,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2605738639831543,
      "step": 2710
    },
    {
      "epoch": 0.05424,
      "grad_norm": 2.90625,
      "grad_norm_var": 0.021219889322916668,
      "learning_rate": 0.0001,
      "loss": 5.2795,
      "loss/crossentropy": 2.406570076942444,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2992263287305832,
      "step": 2712
    },
    {
      "epoch": 0.05428,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.024958292643229168,
      "learning_rate": 0.0001,
      "loss": 4.8591,
      "loss/crossentropy": 2.040315330028534,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27726222574710846,
      "step": 2714
    },
    {
      "epoch": 0.05432,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.025178019205729166,
      "learning_rate": 0.0001,
      "loss": 4.7879,
      "loss/crossentropy": 2.250051259994507,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2764698565006256,
      "step": 2716
    },
    {
      "epoch": 0.05436,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.026985677083333333,
      "learning_rate": 0.0001,
      "loss": 4.8813,
      "loss/crossentropy": 2.25112247467041,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3101722151041031,
      "step": 2718
    },
    {
      "epoch": 0.0544,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.028709920247395833,
      "learning_rate": 0.0001,
      "loss": 4.7242,
      "loss/crossentropy": 2.261968731880188,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2805032432079315,
      "step": 2720
    },
    {
      "epoch": 0.05444,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.030078125,
      "learning_rate": 0.0001,
      "loss": 5.0449,
      "loss/crossentropy": 2.376634955406189,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25400323420763016,
      "step": 2722
    },
    {
      "epoch": 0.05448,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.03351949055989583,
      "learning_rate": 0.0001,
      "loss": 5.2325,
      "loss/crossentropy": 2.61246657371521,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3170415759086609,
      "step": 2724
    },
    {
      "epoch": 0.05452,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.0296051025390625,
      "learning_rate": 0.0001,
      "loss": 5.0433,
      "loss/crossentropy": 2.3982752561569214,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2730572074651718,
      "step": 2726
    },
    {
      "epoch": 0.05456,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.0207916259765625,
      "learning_rate": 0.0001,
      "loss": 4.8836,
      "loss/crossentropy": 1.9890516996383667,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26453813910484314,
      "step": 2728
    },
    {
      "epoch": 0.0546,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.016559855143229166,
      "learning_rate": 0.0001,
      "loss": 4.7252,
      "loss/crossentropy": 2.1825047731399536,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28369753062725067,
      "step": 2730
    },
    {
      "epoch": 0.05464,
      "grad_norm": 2.5,
      "grad_norm_var": 0.015360514322916666,
      "learning_rate": 0.0001,
      "loss": 4.9445,
      "loss/crossentropy": 1.9745987057685852,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2662041634321213,
      "step": 2732
    },
    {
      "epoch": 0.05468,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.020783487955729166,
      "learning_rate": 0.0001,
      "loss": 4.7591,
      "loss/crossentropy": 2.2962852716445923,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2986691743135452,
      "step": 2734
    },
    {
      "epoch": 0.05472,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.018342081705729166,
      "learning_rate": 0.0001,
      "loss": 4.9712,
      "loss/crossentropy": 2.0517550110816956,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27293455600738525,
      "step": 2736
    },
    {
      "epoch": 0.05476,
      "grad_norm": 2.859375,
      "grad_norm_var": 0.021434529622395834,
      "learning_rate": 0.0001,
      "loss": 5.4052,
      "loss/crossentropy": 2.327734112739563,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3019224554300308,
      "step": 2738
    },
    {
      "epoch": 0.0548,
      "grad_norm": 2.5,
      "grad_norm_var": 0.018680826822916666,
      "learning_rate": 0.0001,
      "loss": 4.7002,
      "loss/crossentropy": 2.236689567565918,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2816064953804016,
      "step": 2740
    },
    {
      "epoch": 0.05484,
      "grad_norm": 2.75,
      "grad_norm_var": 0.020894368489583332,
      "learning_rate": 0.0001,
      "loss": 5.1321,
      "loss/crossentropy": 2.0209690928459167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2734442874789238,
      "step": 2742
    },
    {
      "epoch": 0.05488,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.026276652018229166,
      "learning_rate": 0.0001,
      "loss": 4.9841,
      "loss/crossentropy": 2.2264864444732666,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28523482382297516,
      "step": 2744
    },
    {
      "epoch": 0.05492,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.024738566080729166,
      "learning_rate": 0.0001,
      "loss": 4.8661,
      "loss/crossentropy": 1.9897980690002441,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26623376458883286,
      "step": 2746
    },
    {
      "epoch": 0.05496,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.024442545572916665,
      "learning_rate": 0.0001,
      "loss": 4.9299,
      "loss/crossentropy": 2.0583779215812683,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2803298681974411,
      "step": 2748
    },
    {
      "epoch": 0.055,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.022001139322916665,
      "learning_rate": 0.0001,
      "loss": 4.9818,
      "loss/crossentropy": 1.8448269367218018,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2606130689382553,
      "step": 2750
    },
    {
      "epoch": 0.05504,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.02301025390625,
      "learning_rate": 0.0001,
      "loss": 5.0885,
      "loss/crossentropy": 2.294836401939392,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29403699934482574,
      "step": 2752
    },
    {
      "epoch": 0.05508,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.02808837890625,
      "learning_rate": 0.0001,
      "loss": 4.4637,
      "loss/crossentropy": 2.2402058839797974,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28176650404930115,
      "step": 2754
    },
    {
      "epoch": 0.05512,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.0298980712890625,
      "learning_rate": 0.0001,
      "loss": 4.7805,
      "loss/crossentropy": 1.7882421612739563,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25275079905986786,
      "step": 2756
    },
    {
      "epoch": 0.05516,
      "grad_norm": 2.8125,
      "grad_norm_var": 0.02919921875,
      "learning_rate": 0.0001,
      "loss": 5.1277,
      "loss/crossentropy": 2.4185458421707153,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30663780868053436,
      "step": 2758
    },
    {
      "epoch": 0.0552,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.0318511962890625,
      "learning_rate": 0.0001,
      "loss": 4.9259,
      "loss/crossentropy": 2.2588642835617065,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26785464584827423,
      "step": 2760
    },
    {
      "epoch": 0.05524,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.034012858072916666,
      "learning_rate": 0.0001,
      "loss": 4.8359,
      "loss/crossentropy": 2.145754337310791,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2710302472114563,
      "step": 2762
    },
    {
      "epoch": 0.05528,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.03648681640625,
      "learning_rate": 0.0001,
      "loss": 4.739,
      "loss/crossentropy": 2.3627569675445557,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2926081120967865,
      "step": 2764
    },
    {
      "epoch": 0.05532,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.03632405598958333,
      "learning_rate": 0.0001,
      "loss": 4.8512,
      "loss/crossentropy": 1.988103210926056,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24308288842439651,
      "step": 2766
    },
    {
      "epoch": 0.05536,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.040445963541666664,
      "learning_rate": 0.0001,
      "loss": 5.1075,
      "loss/crossentropy": 2.2497689723968506,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.313574954867363,
      "step": 2768
    },
    {
      "epoch": 0.0554,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.030492146809895832,
      "learning_rate": 0.0001,
      "loss": 4.8978,
      "loss/crossentropy": 2.2603683471679688,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27997657656669617,
      "step": 2770
    },
    {
      "epoch": 0.05544,
      "grad_norm": 2.796875,
      "grad_norm_var": 0.030614217122395832,
      "learning_rate": 0.0001,
      "loss": 4.9485,
      "loss/crossentropy": 2.2585065364837646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2818940281867981,
      "step": 2772
    },
    {
      "epoch": 0.05548,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.027730305989583332,
      "learning_rate": 0.0001,
      "loss": 5.2222,
      "loss/crossentropy": 2.1413429975509644,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2950669527053833,
      "step": 2774
    },
    {
      "epoch": 0.05552,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.0171875,
      "learning_rate": 0.0001,
      "loss": 5.1567,
      "loss/crossentropy": 1.9994583129882812,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3946940451860428,
      "step": 2776
    },
    {
      "epoch": 0.05556,
      "grad_norm": 2.796875,
      "grad_norm_var": 0.014762369791666667,
      "learning_rate": 0.0001,
      "loss": 5.2877,
      "loss/crossentropy": 2.423824667930603,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30817069113254547,
      "step": 2778
    },
    {
      "epoch": 0.0556,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.013719685872395833,
      "learning_rate": 0.0001,
      "loss": 4.6657,
      "loss/crossentropy": 1.8579126000404358,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24312064796686172,
      "step": 2780
    },
    {
      "epoch": 0.05564,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.018485514322916667,
      "learning_rate": 0.0001,
      "loss": 4.7653,
      "loss/crossentropy": 2.3444939851760864,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28466545045375824,
      "step": 2782
    },
    {
      "epoch": 0.05568,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.03806864420572917,
      "learning_rate": 0.0001,
      "loss": 5.1187,
      "loss/crossentropy": 2.221144914627075,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2694521099328995,
      "step": 2784
    },
    {
      "epoch": 0.05572,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.0376861572265625,
      "learning_rate": 0.0001,
      "loss": 5.0401,
      "loss/crossentropy": 1.9372909665107727,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2698594778776169,
      "step": 2786
    },
    {
      "epoch": 0.05576,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.038834635416666666,
      "learning_rate": 0.0001,
      "loss": 4.834,
      "loss/crossentropy": 2.129204750061035,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27333614230155945,
      "step": 2788
    },
    {
      "epoch": 0.0558,
      "grad_norm": 2.5,
      "grad_norm_var": 0.04045817057291667,
      "learning_rate": 0.0001,
      "loss": 4.8462,
      "loss/crossentropy": 1.6917370557785034,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24290545284748077,
      "step": 2790
    },
    {
      "epoch": 0.05584,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.04096577962239583,
      "learning_rate": 0.0001,
      "loss": 4.6942,
      "loss/crossentropy": 1.7883749604225159,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2554834187030792,
      "step": 2792
    },
    {
      "epoch": 0.05588,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.04560445149739583,
      "learning_rate": 0.0001,
      "loss": 4.6835,
      "loss/crossentropy": 1.867136001586914,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2667583078145981,
      "step": 2794
    },
    {
      "epoch": 0.05592,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.04143473307291667,
      "learning_rate": 0.0001,
      "loss": 4.9686,
      "loss/crossentropy": 2.032800853252411,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2580890506505966,
      "step": 2796
    },
    {
      "epoch": 0.05596,
      "grad_norm": 3.03125,
      "grad_norm_var": 0.04389546712239583,
      "learning_rate": 0.0001,
      "loss": 5.018,
      "loss/crossentropy": 1.9867863655090332,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2694792151451111,
      "step": 2798
    },
    {
      "epoch": 0.056,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.026786295572916667,
      "learning_rate": 0.0001,
      "loss": 4.9137,
      "loss/crossentropy": 2.1026757955551147,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2664051130414009,
      "step": 2800
    },
    {
      "epoch": 0.05604,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.0259918212890625,
      "learning_rate": 0.0001,
      "loss": 4.9883,
      "loss/crossentropy": 2.0649160742759705,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2473129704594612,
      "step": 2802
    },
    {
      "epoch": 0.05608,
      "grad_norm": 2.921875,
      "grad_norm_var": 0.030451456705729168,
      "learning_rate": 0.0001,
      "loss": 4.8959,
      "loss/crossentropy": 2.2110248804092407,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27131715416908264,
      "step": 2804
    },
    {
      "epoch": 0.05612,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.032933553059895836,
      "learning_rate": 0.0001,
      "loss": 4.9603,
      "loss/crossentropy": 2.3483060598373413,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27071496844291687,
      "step": 2806
    },
    {
      "epoch": 0.05616,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.03277587890625,
      "learning_rate": 0.0001,
      "loss": 4.9273,
      "loss/crossentropy": 1.9061944484710693,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2550426125526428,
      "step": 2808
    },
    {
      "epoch": 0.0562,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.0278472900390625,
      "learning_rate": 0.0001,
      "loss": 5.1734,
      "loss/crossentropy": 2.3073103427886963,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30051568150520325,
      "step": 2810
    },
    {
      "epoch": 0.05624,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.027106730143229167,
      "learning_rate": 0.0001,
      "loss": 5.0268,
      "loss/crossentropy": 2.393891453742981,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29226459562778473,
      "step": 2812
    },
    {
      "epoch": 0.05628,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.018578084309895833,
      "learning_rate": 0.0001,
      "loss": 5.0839,
      "loss/crossentropy": 2.3082761764526367,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29539754986763,
      "step": 2814
    },
    {
      "epoch": 0.05632,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.021776326497395835,
      "learning_rate": 0.0001,
      "loss": 4.8973,
      "loss/crossentropy": 2.7815494537353516,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33258646726608276,
      "step": 2816
    },
    {
      "epoch": 0.05636,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.031538899739583334,
      "learning_rate": 0.0001,
      "loss": 4.7095,
      "loss/crossentropy": 2.077984571456909,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.261022225022316,
      "step": 2818
    },
    {
      "epoch": 0.0564,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.0251617431640625,
      "learning_rate": 0.0001,
      "loss": 4.6389,
      "loss/crossentropy": 2.0524688363075256,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2936979830265045,
      "step": 2820
    },
    {
      "epoch": 0.05644,
      "grad_norm": 2.5,
      "grad_norm_var": 0.018212890625,
      "learning_rate": 0.0001,
      "loss": 4.9657,
      "loss/crossentropy": 1.8323140740394592,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26426824927330017,
      "step": 2822
    },
    {
      "epoch": 0.05648,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.0201171875,
      "learning_rate": 0.0001,
      "loss": 4.9513,
      "loss/crossentropy": 2.2290462255477905,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28852027654647827,
      "step": 2824
    },
    {
      "epoch": 0.05652,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.016600545247395834,
      "learning_rate": 0.0001,
      "loss": 4.7828,
      "loss/crossentropy": 1.8788678050041199,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24312467128038406,
      "step": 2826
    },
    {
      "epoch": 0.05656,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.01646728515625,
      "learning_rate": 0.0001,
      "loss": 4.7882,
      "loss/crossentropy": 1.9402090311050415,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25435876101255417,
      "step": 2828
    },
    {
      "epoch": 0.0566,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.0144683837890625,
      "learning_rate": 0.0001,
      "loss": 4.5823,
      "loss/crossentropy": 2.4833847284317017,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27234241366386414,
      "step": 2830
    },
    {
      "epoch": 0.05664,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.0154296875,
      "learning_rate": 0.0001,
      "loss": 5.0521,
      "loss/crossentropy": 2.351561665534973,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28028184175491333,
      "step": 2832
    },
    {
      "epoch": 0.05668,
      "grad_norm": 2.953125,
      "grad_norm_var": 0.019722493489583333,
      "learning_rate": 0.0001,
      "loss": 5.0688,
      "loss/crossentropy": 2.0372352600097656,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26005299389362335,
      "step": 2834
    },
    {
      "epoch": 0.05672,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.021744791666666666,
      "learning_rate": 0.0001,
      "loss": 4.4282,
      "loss/crossentropy": 1.6522082090377808,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26683834940195084,
      "step": 2836
    },
    {
      "epoch": 0.05676,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.023778279622395832,
      "learning_rate": 0.0001,
      "loss": 4.7165,
      "loss/crossentropy": 1.8886643052101135,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23464814573526382,
      "step": 2838
    },
    {
      "epoch": 0.0568,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.023119099934895835,
      "learning_rate": 0.0001,
      "loss": 4.9477,
      "loss/crossentropy": 2.2161459922790527,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29381541907787323,
      "step": 2840
    },
    {
      "epoch": 0.05684,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.024779256184895834,
      "learning_rate": 0.0001,
      "loss": 4.7408,
      "loss/crossentropy": 2.1542043685913086,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2753119468688965,
      "step": 2842
    },
    {
      "epoch": 0.05688,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.026090494791666665,
      "learning_rate": 0.0001,
      "loss": 5.1288,
      "loss/crossentropy": 2.5605628490448,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.329460546374321,
      "step": 2844
    },
    {
      "epoch": 0.05692,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.023224894205729166,
      "learning_rate": 0.0001,
      "loss": 4.9955,
      "loss/crossentropy": 2.1000319719314575,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.273986279964447,
      "step": 2846
    },
    {
      "epoch": 0.05696,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.023558553059895834,
      "learning_rate": 0.0001,
      "loss": 4.6813,
      "loss/crossentropy": 2.1036760210990906,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26293954253196716,
      "step": 2848
    },
    {
      "epoch": 0.057,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.014286295572916666,
      "learning_rate": 0.0001,
      "loss": 4.9799,
      "loss/crossentropy": 2.2130608558654785,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2977828085422516,
      "step": 2850
    },
    {
      "epoch": 0.05704,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.01461181640625,
      "learning_rate": 0.0001,
      "loss": 5.1761,
      "loss/crossentropy": 2.1878823041915894,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27322643995285034,
      "step": 2852
    },
    {
      "epoch": 0.05708,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.012547810872395834,
      "learning_rate": 0.0001,
      "loss": 4.9448,
      "loss/crossentropy": 2.1559258699417114,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32625503838062286,
      "step": 2854
    },
    {
      "epoch": 0.05712,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.015848795572916668,
      "learning_rate": 0.0001,
      "loss": 4.4611,
      "loss/crossentropy": 2.0818992257118225,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2677721679210663,
      "step": 2856
    },
    {
      "epoch": 0.05716,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.016243489583333333,
      "learning_rate": 0.0001,
      "loss": 5.2755,
      "loss/crossentropy": 2.544907331466675,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30028045177459717,
      "step": 2858
    },
    {
      "epoch": 0.0572,
      "grad_norm": 3.203125,
      "grad_norm_var": 0.04491780598958333,
      "learning_rate": 0.0001,
      "loss": 5.2855,
      "loss/crossentropy": 2.5932188034057617,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30431173741817474,
      "step": 2860
    },
    {
      "epoch": 0.05724,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.046219889322916666,
      "learning_rate": 0.0001,
      "loss": 4.9799,
      "loss/crossentropy": 2.1007986068725586,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2989690601825714,
      "step": 2862
    },
    {
      "epoch": 0.05728,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.044169108072916664,
      "learning_rate": 0.0001,
      "loss": 4.7706,
      "loss/crossentropy": 2.0102875232696533,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26936179399490356,
      "step": 2864
    },
    {
      "epoch": 0.05732,
      "grad_norm": 3.109375,
      "grad_norm_var": 0.056559244791666664,
      "learning_rate": 0.0001,
      "loss": 5.2099,
      "loss/crossentropy": 2.3457159996032715,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29194609820842743,
      "step": 2866
    },
    {
      "epoch": 0.05736,
      "grad_norm": 2.625,
      "grad_norm_var": 0.055582682291666664,
      "learning_rate": 0.0001,
      "loss": 4.8768,
      "loss/crossentropy": 2.559054732322693,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3018783777952194,
      "step": 2868
    },
    {
      "epoch": 0.0574,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.05831705729166667,
      "learning_rate": 0.0001,
      "loss": 4.9863,
      "loss/crossentropy": 2.0641059279441833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2789234220981598,
      "step": 2870
    },
    {
      "epoch": 0.05744,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.04683837890625,
      "learning_rate": 0.0001,
      "loss": 4.9222,
      "loss/crossentropy": 2.0819749236106873,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26825186610221863,
      "step": 2872
    },
    {
      "epoch": 0.05748,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.046263631184895834,
      "learning_rate": 0.0001,
      "loss": 4.9586,
      "loss/crossentropy": 2.1114020347595215,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2613547742366791,
      "step": 2874
    },
    {
      "epoch": 0.05752,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.02916259765625,
      "learning_rate": 0.0001,
      "loss": 4.9474,
      "loss/crossentropy": 2.174915075302124,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29503974318504333,
      "step": 2876
    },
    {
      "epoch": 0.05756,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.0265289306640625,
      "learning_rate": 0.0001,
      "loss": 5.0541,
      "loss/crossentropy": 2.4342113733291626,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30623678863048553,
      "step": 2878
    },
    {
      "epoch": 0.0576,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.024918619791666666,
      "learning_rate": 0.0001,
      "loss": 5.0316,
      "loss/crossentropy": 2.0518307089805603,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28422991931438446,
      "step": 2880
    },
    {
      "epoch": 0.05764,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.011921183268229166,
      "learning_rate": 0.0001,
      "loss": 5.0806,
      "loss/crossentropy": 2.5378612279891968,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29730531573295593,
      "step": 2882
    },
    {
      "epoch": 0.05768,
      "grad_norm": 2.5,
      "grad_norm_var": 0.014972941080729166,
      "learning_rate": 0.0001,
      "loss": 4.8685,
      "loss/crossentropy": 2.2670027017593384,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26347288489341736,
      "step": 2884
    },
    {
      "epoch": 0.05772,
      "grad_norm": 3.671875,
      "grad_norm_var": 0.07822265625,
      "learning_rate": 0.0001,
      "loss": 5.0385,
      "loss/crossentropy": 2.351823568344116,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3020637035369873,
      "step": 2886
    },
    {
      "epoch": 0.05776,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.0792633056640625,
      "learning_rate": 0.0001,
      "loss": 4.9699,
      "loss/crossentropy": 2.190839111804962,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2639586254954338,
      "step": 2888
    },
    {
      "epoch": 0.0578,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.08765869140625,
      "learning_rate": 0.0001,
      "loss": 4.6882,
      "loss/crossentropy": 2.148400902748108,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27762140333652496,
      "step": 2890
    },
    {
      "epoch": 0.05784,
      "grad_norm": 2.984375,
      "grad_norm_var": 0.09696858723958333,
      "learning_rate": 0.0001,
      "loss": 5.1033,
      "loss/crossentropy": 2.01130074262619,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2678648605942726,
      "step": 2892
    },
    {
      "epoch": 0.05788,
      "grad_norm": 2.75,
      "grad_norm_var": 0.09986572265625,
      "learning_rate": 0.0001,
      "loss": 5.0829,
      "loss/crossentropy": 2.2239269018173218,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2971164286136627,
      "step": 2894
    },
    {
      "epoch": 0.05792,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.10974934895833334,
      "learning_rate": 0.0001,
      "loss": 4.48,
      "loss/crossentropy": 1.9573850631713867,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.241651751101017,
      "step": 2896
    },
    {
      "epoch": 0.05796,
      "grad_norm": 2.5,
      "grad_norm_var": 0.11169331868489583,
      "learning_rate": 0.0001,
      "loss": 4.8806,
      "loss/crossentropy": 1.9522782564163208,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2685594707727432,
      "step": 2898
    },
    {
      "epoch": 0.058,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.10886128743489583,
      "learning_rate": 0.0001,
      "loss": 4.9335,
      "loss/crossentropy": 2.1742069721221924,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27867090702056885,
      "step": 2900
    },
    {
      "epoch": 0.05804,
      "grad_norm": 5.15625,
      "grad_norm_var": 0.44882405598958336,
      "learning_rate": 0.0001,
      "loss": 4.9164,
      "loss/crossentropy": 2.201782703399658,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2731190174818039,
      "step": 2902
    },
    {
      "epoch": 0.05808,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.4452301025390625,
      "learning_rate": 0.0001,
      "loss": 4.9787,
      "loss/crossentropy": 2.159709095954895,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2940017879009247,
      "step": 2904
    },
    {
      "epoch": 0.05812,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.4297271728515625,
      "learning_rate": 0.0001,
      "loss": 4.9192,
      "loss/crossentropy": 2.1989885568618774,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2776087671518326,
      "step": 2906
    },
    {
      "epoch": 0.05816,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.42688395182291666,
      "learning_rate": 0.0001,
      "loss": 4.8851,
      "loss/crossentropy": 1.9634575247764587,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27917809784412384,
      "step": 2908
    },
    {
      "epoch": 0.0582,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.43651936848958334,
      "learning_rate": 0.0001,
      "loss": 4.7776,
      "loss/crossentropy": 2.1553479433059692,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28876082599163055,
      "step": 2910
    },
    {
      "epoch": 0.05824,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.42451883951822916,
      "learning_rate": 0.0001,
      "loss": 4.881,
      "loss/crossentropy": 2.1035598516464233,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2851293236017227,
      "step": 2912
    },
    {
      "epoch": 0.05828,
      "grad_norm": 2.890625,
      "grad_norm_var": 0.42097880045572916,
      "learning_rate": 0.0001,
      "loss": 5.2733,
      "loss/crossentropy": 2.17076575756073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28258734941482544,
      "step": 2914
    },
    {
      "epoch": 0.05832,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.42477213541666664,
      "learning_rate": 0.0001,
      "loss": 4.9147,
      "loss/crossentropy": 2.2914711236953735,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29451698064804077,
      "step": 2916
    },
    {
      "epoch": 0.05836,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.021410115559895835,
      "learning_rate": 0.0001,
      "loss": 5.1395,
      "loss/crossentropy": 2.554638981819153,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30992935597896576,
      "step": 2918
    },
    {
      "epoch": 0.0584,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.05735575358072917,
      "learning_rate": 0.0001,
      "loss": 4.6262,
      "loss/crossentropy": 1.808376431465149,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22574126720428467,
      "step": 2920
    },
    {
      "epoch": 0.05844,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.05660400390625,
      "learning_rate": 0.0001,
      "loss": 5.1636,
      "loss/crossentropy": 2.1471784114837646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27325738966464996,
      "step": 2922
    },
    {
      "epoch": 0.05848,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.054182942708333334,
      "learning_rate": 0.0001,
      "loss": 5.071,
      "loss/crossentropy": 2.2175731658935547,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3013547956943512,
      "step": 2924
    },
    {
      "epoch": 0.05852,
      "grad_norm": 2.953125,
      "grad_norm_var": 0.04949544270833333,
      "learning_rate": 0.0001,
      "loss": 5.467,
      "loss/crossentropy": 2.369373917579651,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29249751567840576,
      "step": 2926
    },
    {
      "epoch": 0.05856,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.054915364583333334,
      "learning_rate": 0.0001,
      "loss": 4.8778,
      "loss/crossentropy": 2.1758522987365723,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28846102952957153,
      "step": 2928
    },
    {
      "epoch": 0.0586,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.05413411458333333,
      "learning_rate": 0.0001,
      "loss": 4.929,
      "loss/crossentropy": 2.46218478679657,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2824682295322418,
      "step": 2930
    },
    {
      "epoch": 0.05864,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.0606597900390625,
      "learning_rate": 0.0001,
      "loss": 4.5557,
      "loss/crossentropy": 2.058937907218933,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26767465472221375,
      "step": 2932
    },
    {
      "epoch": 0.05868,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.06109619140625,
      "learning_rate": 0.0001,
      "loss": 4.9103,
      "loss/crossentropy": 2.312442421913147,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2952658236026764,
      "step": 2934
    },
    {
      "epoch": 0.05872,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.026911417643229168,
      "learning_rate": 0.0001,
      "loss": 5.1452,
      "loss/crossentropy": 2.18839955329895,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31788623332977295,
      "step": 2936
    },
    {
      "epoch": 0.05876,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.025537109375,
      "learning_rate": 0.0001,
      "loss": 5.1358,
      "loss/crossentropy": 2.1330811977386475,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28829698264598846,
      "step": 2938
    },
    {
      "epoch": 0.0588,
      "grad_norm": 2.9375,
      "grad_norm_var": 0.0294586181640625,
      "learning_rate": 0.0001,
      "loss": 5.1071,
      "loss/crossentropy": 2.124038338661194,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2708826810121536,
      "step": 2940
    },
    {
      "epoch": 0.05884,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.024312337239583332,
      "learning_rate": 0.0001,
      "loss": 4.8479,
      "loss/crossentropy": 2.1934449076652527,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27314090728759766,
      "step": 2942
    },
    {
      "epoch": 0.05888,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.024535115559895834,
      "learning_rate": 0.0001,
      "loss": 4.9904,
      "loss/crossentropy": 1.967636525630951,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23761005699634552,
      "step": 2944
    },
    {
      "epoch": 0.05892,
      "grad_norm": 3.015625,
      "grad_norm_var": 0.034821573893229166,
      "learning_rate": 0.0001,
      "loss": 4.9214,
      "loss/crossentropy": 2.2380826473236084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3402934819459915,
      "step": 2946
    },
    {
      "epoch": 0.05896,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.0299468994140625,
      "learning_rate": 0.0001,
      "loss": 4.7702,
      "loss/crossentropy": 2.227039933204651,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27934837341308594,
      "step": 2948
    },
    {
      "epoch": 0.059,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.03258056640625,
      "learning_rate": 0.0001,
      "loss": 4.789,
      "loss/crossentropy": 1.999170958995819,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.275806725025177,
      "step": 2950
    },
    {
      "epoch": 0.05904,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.03241780598958333,
      "learning_rate": 0.0001,
      "loss": 4.5754,
      "loss/crossentropy": 1.8843520879745483,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26203446090221405,
      "step": 2952
    },
    {
      "epoch": 0.05908,
      "grad_norm": 2.5,
      "grad_norm_var": 0.033426920572916664,
      "learning_rate": 0.0001,
      "loss": 4.5485,
      "loss/crossentropy": 2.0742241740226746,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28777699172496796,
      "step": 2954
    },
    {
      "epoch": 0.05912,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.028206380208333333,
      "learning_rate": 0.0001,
      "loss": 4.9676,
      "loss/crossentropy": 2.2272751331329346,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27761510014533997,
      "step": 2956
    },
    {
      "epoch": 0.05916,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.029878743489583335,
      "learning_rate": 0.0001,
      "loss": 4.6329,
      "loss/crossentropy": 2.2758638858795166,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28144824504852295,
      "step": 2958
    },
    {
      "epoch": 0.0592,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.060530598958333334,
      "learning_rate": 0.0001,
      "loss": 4.9543,
      "loss/crossentropy": 2.245158016681671,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3030128926038742,
      "step": 2960
    },
    {
      "epoch": 0.05924,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.053498331705729166,
      "learning_rate": 0.0001,
      "loss": 4.567,
      "loss/crossentropy": 2.3337708711624146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29228493571281433,
      "step": 2962
    },
    {
      "epoch": 0.05928,
      "grad_norm": 2.5,
      "grad_norm_var": 0.05788472493489583,
      "learning_rate": 0.0001,
      "loss": 4.6477,
      "loss/crossentropy": 2.3765406608581543,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2715871036052704,
      "step": 2964
    },
    {
      "epoch": 0.05932,
      "grad_norm": 2.375,
      "grad_norm_var": 0.0562652587890625,
      "learning_rate": 0.0001,
      "loss": 4.8649,
      "loss/crossentropy": 2.090362787246704,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2505037933588028,
      "step": 2966
    },
    {
      "epoch": 0.05936,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.056396484375,
      "learning_rate": 0.0001,
      "loss": 4.8669,
      "loss/crossentropy": 2.011539399623871,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27219071984291077,
      "step": 2968
    },
    {
      "epoch": 0.0594,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.05693359375,
      "learning_rate": 0.0001,
      "loss": 5.2081,
      "loss/crossentropy": 2.1791563034057617,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2796429842710495,
      "step": 2970
    },
    {
      "epoch": 0.05944,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.05869852701822917,
      "learning_rate": 0.0001,
      "loss": 5.2689,
      "loss/crossentropy": 2.4645297527313232,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3106658458709717,
      "step": 2972
    },
    {
      "epoch": 0.05948,
      "grad_norm": 7.0,
      "grad_norm_var": 1.25670166015625,
      "learning_rate": 0.0001,
      "loss": 5.0715,
      "loss/crossentropy": 2.2050880193710327,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26860976219177246,
      "step": 2974
    },
    {
      "epoch": 0.05952,
      "grad_norm": 2.828125,
      "grad_norm_var": 1.236034138997396,
      "learning_rate": 0.0001,
      "loss": 4.5815,
      "loss/crossentropy": 2.0141921639442444,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27980539202690125,
      "step": 2976
    },
    {
      "epoch": 0.05956,
      "grad_norm": 4.4375,
      "grad_norm_var": 1.3665924072265625,
      "learning_rate": 0.0001,
      "loss": 4.8067,
      "loss/crossentropy": 1.9399088025093079,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26379524916410446,
      "step": 2978
    },
    {
      "epoch": 0.0596,
      "grad_norm": 2.796875,
      "grad_norm_var": 1.3243886311848958,
      "learning_rate": 0.0001,
      "loss": 5.1743,
      "loss/crossentropy": 2.5415157079696655,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.291620597243309,
      "step": 2980
    },
    {
      "epoch": 0.05964,
      "grad_norm": 2.5625,
      "grad_norm_var": 1.3170562744140626,
      "learning_rate": 0.0001,
      "loss": 5.0695,
      "loss/crossentropy": 2.1215542554855347,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26189403235912323,
      "step": 2982
    },
    {
      "epoch": 0.05968,
      "grad_norm": 2.578125,
      "grad_norm_var": 1.32301025390625,
      "learning_rate": 0.0001,
      "loss": 5.0049,
      "loss/crossentropy": 1.7749422788619995,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27546317875385284,
      "step": 2984
    },
    {
      "epoch": 0.05972,
      "grad_norm": 2.59375,
      "grad_norm_var": 1.3371490478515624,
      "learning_rate": 0.0001,
      "loss": 4.8331,
      "loss/crossentropy": 2.3383208513259888,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2790983319282532,
      "step": 2986
    },
    {
      "epoch": 0.05976,
      "grad_norm": 2.5625,
      "grad_norm_var": 1.35230712890625,
      "learning_rate": 0.0001,
      "loss": 4.8656,
      "loss/crossentropy": 2.3688780069351196,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2752939611673355,
      "step": 2988
    },
    {
      "epoch": 0.0598,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.22967020670572916,
      "learning_rate": 0.0001,
      "loss": 4.7299,
      "loss/crossentropy": 2.261468529701233,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2725592106580734,
      "step": 2990
    },
    {
      "epoch": 0.05984,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.2226226806640625,
      "learning_rate": 0.0001,
      "loss": 4.7417,
      "loss/crossentropy": 2.0632028579711914,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2617062032222748,
      "step": 2992
    },
    {
      "epoch": 0.05988,
      "grad_norm": 2.25,
      "grad_norm_var": 0.022151692708333334,
      "learning_rate": 0.0001,
      "loss": 4.4282,
      "loss/crossentropy": 2.1409813165664673,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2657589614391327,
      "step": 2994
    },
    {
      "epoch": 0.05992,
      "grad_norm": 2.921875,
      "grad_norm_var": 0.0258941650390625,
      "learning_rate": 0.0001,
      "loss": 5.1816,
      "loss/crossentropy": 2.431404948234558,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3087555915117264,
      "step": 2996
    },
    {
      "epoch": 0.05996,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.029743448893229166,
      "learning_rate": 0.0001,
      "loss": 4.8253,
      "loss/crossentropy": 2.165238618850708,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29382775723934174,
      "step": 2998
    },
    {
      "epoch": 0.06,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.034566243489583336,
      "learning_rate": 0.0001,
      "loss": 4.6328,
      "loss/crossentropy": 1.9987847208976746,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24466252326965332,
      "step": 3000
    },
    {
      "epoch": 0.06004,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.04084370930989583,
      "learning_rate": 0.0001,
      "loss": 4.5474,
      "loss/crossentropy": 2.1153565049171448,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2435970976948738,
      "step": 3002
    },
    {
      "epoch": 0.06008,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.046402994791666666,
      "learning_rate": 0.0001,
      "loss": 4.7765,
      "loss/crossentropy": 1.8660866618156433,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24413185566663742,
      "step": 3004
    },
    {
      "epoch": 0.06012,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.04309794108072917,
      "learning_rate": 0.0001,
      "loss": 4.7476,
      "loss/crossentropy": 2.1816678047180176,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28010787069797516,
      "step": 3006
    },
    {
      "epoch": 0.06016,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.040827433268229164,
      "learning_rate": 0.0001,
      "loss": 4.5995,
      "loss/crossentropy": 2.1673622131347656,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2659531533718109,
      "step": 3008
    },
    {
      "epoch": 0.0602,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.040913899739583336,
      "learning_rate": 0.0001,
      "loss": 5.2466,
      "loss/crossentropy": 2.277818202972412,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2929365783929825,
      "step": 3010
    },
    {
      "epoch": 0.06024,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.0284576416015625,
      "learning_rate": 0.0001,
      "loss": 4.8904,
      "loss/crossentropy": 2.294926404953003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2753874659538269,
      "step": 3012
    },
    {
      "epoch": 0.06028,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.028173828125,
      "learning_rate": 0.0001,
      "loss": 4.735,
      "loss/crossentropy": 2.0222257375717163,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27295801043510437,
      "step": 3014
    },
    {
      "epoch": 0.06032,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.023628743489583333,
      "learning_rate": 0.0001,
      "loss": 4.818,
      "loss/crossentropy": 2.43736469745636,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2934436649084091,
      "step": 3016
    },
    {
      "epoch": 0.06036,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.020426432291666668,
      "learning_rate": 0.0001,
      "loss": 5.0053,
      "loss/crossentropy": 2.027767241001129,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26353244483470917,
      "step": 3018
    },
    {
      "epoch": 0.0604,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.015034993489583334,
      "learning_rate": 0.0001,
      "loss": 4.9059,
      "loss/crossentropy": 2.150290071964264,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27290941774845123,
      "step": 3020
    },
    {
      "epoch": 0.06044,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.014697265625,
      "learning_rate": 0.0001,
      "loss": 4.8307,
      "loss/crossentropy": 1.8709848523139954,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25272610783576965,
      "step": 3022
    },
    {
      "epoch": 0.06048,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.015949503580729166,
      "learning_rate": 0.0001,
      "loss": 5.0166,
      "loss/crossentropy": 2.2170883417129517,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29401010274887085,
      "step": 3024
    },
    {
      "epoch": 0.06052,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.0109375,
      "learning_rate": 0.0001,
      "loss": 4.6221,
      "loss/crossentropy": 1.728318691253662,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24246910959482193,
      "step": 3026
    },
    {
      "epoch": 0.06056,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.0095123291015625,
      "learning_rate": 0.0001,
      "loss": 4.5597,
      "loss/crossentropy": 1.8453290462493896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24357211589813232,
      "step": 3028
    },
    {
      "epoch": 0.0606,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.010456339518229166,
      "learning_rate": 0.0001,
      "loss": 4.6617,
      "loss/crossentropy": 2.273196220397949,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29239149391651154,
      "step": 3030
    },
    {
      "epoch": 0.06064,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.012165323893229166,
      "learning_rate": 0.0001,
      "loss": 4.9464,
      "loss/crossentropy": 2.031871259212494,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2604901194572449,
      "step": 3032
    },
    {
      "epoch": 0.06068,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.008915201822916666,
      "learning_rate": 0.0001,
      "loss": 5.1618,
      "loss/crossentropy": 2.1781771183013916,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2853371948003769,
      "step": 3034
    },
    {
      "epoch": 0.06072,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.006590779622395833,
      "learning_rate": 0.0001,
      "loss": 4.816,
      "loss/crossentropy": 1.9578353762626648,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23352904617786407,
      "step": 3036
    },
    {
      "epoch": 0.06076,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.012516276041666666,
      "learning_rate": 0.0001,
      "loss": 5.3012,
      "loss/crossentropy": 2.470985770225525,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28890977799892426,
      "step": 3038
    },
    {
      "epoch": 0.0608,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.013361612955729166,
      "learning_rate": 0.0001,
      "loss": 4.9243,
      "loss/crossentropy": 2.056231141090393,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2963344603776932,
      "step": 3040
    },
    {
      "epoch": 0.06084,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.013646443684895834,
      "learning_rate": 0.0001,
      "loss": 4.8612,
      "loss/crossentropy": 2.000037968158722,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27152783423662186,
      "step": 3042
    },
    {
      "epoch": 0.06088,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.0111480712890625,
      "learning_rate": 0.0001,
      "loss": 4.8576,
      "loss/crossentropy": 2.323809027671814,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29995349049568176,
      "step": 3044
    },
    {
      "epoch": 0.06092,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.010407511393229167,
      "learning_rate": 0.0001,
      "loss": 5.0094,
      "loss/crossentropy": 2.06933856010437,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26831966638565063,
      "step": 3046
    },
    {
      "epoch": 0.06096,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.008463541666666666,
      "learning_rate": 0.0001,
      "loss": 5.042,
      "loss/crossentropy": 2.1903880834579468,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2833031117916107,
      "step": 3048
    },
    {
      "epoch": 0.061,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.008893839518229167,
      "learning_rate": 0.0001,
      "loss": 4.9557,
      "loss/crossentropy": 1.910680890083313,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2708408683538437,
      "step": 3050
    },
    {
      "epoch": 0.06104,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.0083404541015625,
      "learning_rate": 0.0001,
      "loss": 4.8902,
      "loss/crossentropy": 2.5203059911727905,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2981104403734207,
      "step": 3052
    },
    {
      "epoch": 0.06108,
      "grad_norm": 2.5,
      "grad_norm_var": 0.00611572265625,
      "learning_rate": 0.0001,
      "loss": 4.68,
      "loss/crossentropy": 1.7918179035186768,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2462354451417923,
      "step": 3054
    },
    {
      "epoch": 0.06112,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.006441243489583333,
      "learning_rate": 0.0001,
      "loss": 4.9908,
      "loss/crossentropy": 2.185121774673462,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3046490252017975,
      "step": 3056
    },
    {
      "epoch": 0.06116,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.01064453125,
      "learning_rate": 0.0001,
      "loss": 4.8041,
      "loss/crossentropy": 1.859390914440155,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2523205131292343,
      "step": 3058
    },
    {
      "epoch": 0.0612,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.011930338541666667,
      "learning_rate": 0.0001,
      "loss": 4.7388,
      "loss/crossentropy": 1.9202255606651306,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26971636712551117,
      "step": 3060
    },
    {
      "epoch": 0.06124,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.011481730143229167,
      "learning_rate": 0.0001,
      "loss": 4.7421,
      "loss/crossentropy": 2.059127449989319,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25721821188926697,
      "step": 3062
    },
    {
      "epoch": 0.06128,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.0154205322265625,
      "learning_rate": 0.0001,
      "loss": 5.1392,
      "loss/crossentropy": 2.3587669134140015,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30162671208381653,
      "step": 3064
    },
    {
      "epoch": 0.06132,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.018050130208333334,
      "learning_rate": 0.0001,
      "loss": 5.2748,
      "loss/crossentropy": 2.239704966545105,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2713697552680969,
      "step": 3066
    },
    {
      "epoch": 0.06136,
      "grad_norm": 2.796875,
      "grad_norm_var": 0.02076416015625,
      "learning_rate": 0.0001,
      "loss": 5.0534,
      "loss/crossentropy": 2.230944514274597,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27418144047260284,
      "step": 3068
    },
    {
      "epoch": 0.0614,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.020563761393229168,
      "learning_rate": 0.0001,
      "loss": 4.9145,
      "loss/crossentropy": 2.196586310863495,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24946682900190353,
      "step": 3070
    },
    {
      "epoch": 0.06144,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.019873046875,
      "learning_rate": 0.0001,
      "loss": 5.058,
      "loss/crossentropy": 2.115275800228119,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2668849602341652,
      "step": 3072
    },
    {
      "epoch": 0.06148,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.014286295572916666,
      "learning_rate": 0.0001,
      "loss": 5.1311,
      "loss/crossentropy": 2.0227994322776794,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2587117701768875,
      "step": 3074
    },
    {
      "epoch": 0.06152,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.016022745768229166,
      "learning_rate": 0.0001,
      "loss": 4.711,
      "loss/crossentropy": 2.231368660926819,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2621329501271248,
      "step": 3076
    },
    {
      "epoch": 0.06156,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.0190093994140625,
      "learning_rate": 0.0001,
      "loss": 4.6707,
      "loss/crossentropy": 2.469847083091736,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2758704572916031,
      "step": 3078
    },
    {
      "epoch": 0.0616,
      "grad_norm": 2.5,
      "grad_norm_var": 0.01666259765625,
      "learning_rate": 0.0001,
      "loss": 4.5753,
      "loss/crossentropy": 2.257850766181946,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28196004033088684,
      "step": 3080
    },
    {
      "epoch": 0.06164,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.014969889322916667,
      "learning_rate": 0.0001,
      "loss": 4.8783,
      "loss/crossentropy": 2.52754008769989,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.301498681306839,
      "step": 3082
    },
    {
      "epoch": 0.06168,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.012279256184895834,
      "learning_rate": 0.0001,
      "loss": 4.6595,
      "loss/crossentropy": 2.120129644870758,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30073782801628113,
      "step": 3084
    },
    {
      "epoch": 0.06172,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.0231353759765625,
      "learning_rate": 0.0001,
      "loss": 5.017,
      "loss/crossentropy": 2.1483632922172546,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29400819540023804,
      "step": 3086
    },
    {
      "epoch": 0.06176,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.022297159830729166,
      "learning_rate": 0.0001,
      "loss": 5.2447,
      "loss/crossentropy": 2.5588048696517944,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3099561035633087,
      "step": 3088
    },
    {
      "epoch": 0.0618,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.019852701822916666,
      "learning_rate": 0.0001,
      "loss": 4.5507,
      "loss/crossentropy": 2.338167190551758,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2638262137770653,
      "step": 3090
    },
    {
      "epoch": 0.06184,
      "grad_norm": 2.5,
      "grad_norm_var": 0.019603474934895834,
      "learning_rate": 0.0001,
      "loss": 5.0471,
      "loss/crossentropy": 2.61361825466156,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29420773684978485,
      "step": 3092
    },
    {
      "epoch": 0.06188,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.019108072916666666,
      "learning_rate": 0.0001,
      "loss": 4.6931,
      "loss/crossentropy": 2.1078842878341675,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27283959090709686,
      "step": 3094
    },
    {
      "epoch": 0.06192,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.023958333333333335,
      "learning_rate": 0.0001,
      "loss": 4.463,
      "loss/crossentropy": 2.1501123905181885,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3052368611097336,
      "step": 3096
    },
    {
      "epoch": 0.06196,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.026656087239583334,
      "learning_rate": 0.0001,
      "loss": 5.0482,
      "loss/crossentropy": 2.3622714281082153,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2851791977882385,
      "step": 3098
    },
    {
      "epoch": 0.062,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.05203450520833333,
      "learning_rate": 0.0001,
      "loss": 4.3506,
      "loss/crossentropy": 1.9757406115531921,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24451570957899094,
      "step": 3100
    },
    {
      "epoch": 0.06204,
      "grad_norm": 2.859375,
      "grad_norm_var": 0.07011311848958333,
      "learning_rate": 0.0001,
      "loss": 5.0409,
      "loss/crossentropy": 2.1915838718414307,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2749434858560562,
      "step": 3102
    },
    {
      "epoch": 0.06208,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.07214253743489583,
      "learning_rate": 0.0001,
      "loss": 4.8408,
      "loss/crossentropy": 2.130649447441101,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2590179592370987,
      "step": 3104
    },
    {
      "epoch": 0.06212,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.06317952473958334,
      "learning_rate": 0.0001,
      "loss": 4.9083,
      "loss/crossentropy": 2.4478825330734253,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30854369699954987,
      "step": 3106
    },
    {
      "epoch": 0.06216,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.06524149576822917,
      "learning_rate": 0.0001,
      "loss": 4.8249,
      "loss/crossentropy": 1.7108858227729797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22502756118774414,
      "step": 3108
    },
    {
      "epoch": 0.0622,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.06788736979166667,
      "learning_rate": 0.0001,
      "loss": 4.5355,
      "loss/crossentropy": 1.79804128408432,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24767793715000153,
      "step": 3110
    },
    {
      "epoch": 0.06224,
      "grad_norm": 3.28125,
      "grad_norm_var": 0.07480061848958333,
      "learning_rate": 0.0001,
      "loss": 4.8779,
      "loss/crossentropy": 1.899846076965332,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23476070165634155,
      "step": 3112
    },
    {
      "epoch": 0.06228,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.0790679931640625,
      "learning_rate": 0.0001,
      "loss": 4.7506,
      "loss/crossentropy": 1.8458155393600464,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24417708814144135,
      "step": 3114
    },
    {
      "epoch": 0.06232,
      "grad_norm": 2.5,
      "grad_norm_var": 0.06979166666666667,
      "learning_rate": 0.0001,
      "loss": 4.814,
      "loss/crossentropy": 1.947302520275116,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2566901594400406,
      "step": 3116
    },
    {
      "epoch": 0.06236,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.046923828125,
      "learning_rate": 0.0001,
      "loss": 4.8424,
      "loss/crossentropy": 2.039161205291748,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26955537497997284,
      "step": 3118
    },
    {
      "epoch": 0.0624,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.04488016764322917,
      "learning_rate": 0.0001,
      "loss": 5.1656,
      "loss/crossentropy": 2.1528661251068115,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.291052982211113,
      "step": 3120
    },
    {
      "epoch": 0.06244,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.04644775390625,
      "learning_rate": 0.0001,
      "loss": 4.7941,
      "loss/crossentropy": 2.4113996028900146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2962254136800766,
      "step": 3122
    },
    {
      "epoch": 0.06248,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.044722493489583334,
      "learning_rate": 0.0001,
      "loss": 4.8463,
      "loss/crossentropy": 2.0718825459480286,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24676042795181274,
      "step": 3124
    },
    {
      "epoch": 0.06252,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.04442952473958333,
      "learning_rate": 0.0001,
      "loss": 4.9723,
      "loss/crossentropy": 2.433539032936096,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.263284370303154,
      "step": 3126
    },
    {
      "epoch": 0.06256,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.0087799072265625,
      "learning_rate": 0.0001,
      "loss": 4.7688,
      "loss/crossentropy": 2.047150671482086,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2655043303966522,
      "step": 3128
    },
    {
      "epoch": 0.0626,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.008219401041666666,
      "learning_rate": 0.0001,
      "loss": 4.9257,
      "loss/crossentropy": 1.8897674679756165,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2314881831407547,
      "step": 3130
    },
    {
      "epoch": 0.06264,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.00797119140625,
      "learning_rate": 0.0001,
      "loss": 4.8053,
      "loss/crossentropy": 2.1436809301376343,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26931750774383545,
      "step": 3132
    },
    {
      "epoch": 0.06268,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.0081695556640625,
      "learning_rate": 0.0001,
      "loss": 4.67,
      "loss/crossentropy": 1.9773973226547241,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26880529522895813,
      "step": 3134
    },
    {
      "epoch": 0.06272,
      "grad_norm": 2.984375,
      "grad_norm_var": 0.021903483072916667,
      "learning_rate": 0.0001,
      "loss": 5.1236,
      "loss/crossentropy": 2.208973228931427,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29979653656482697,
      "step": 3136
    },
    {
      "epoch": 0.06276,
      "grad_norm": 2.8125,
      "grad_norm_var": 0.026676432291666666,
      "learning_rate": 0.0001,
      "loss": 5.0619,
      "loss/crossentropy": 2.64120090007782,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29249200224876404,
      "step": 3138
    },
    {
      "epoch": 0.0628,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.0306549072265625,
      "learning_rate": 0.0001,
      "loss": 5.0152,
      "loss/crossentropy": 2.293095588684082,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2689897269010544,
      "step": 3140
    },
    {
      "epoch": 0.06284,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.028758748372395834,
      "learning_rate": 0.0001,
      "loss": 4.71,
      "loss/crossentropy": 2.455227494239807,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3202812373638153,
      "step": 3142
    },
    {
      "epoch": 0.06288,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.0253814697265625,
      "learning_rate": 0.0001,
      "loss": 4.9622,
      "loss/crossentropy": 1.9667487740516663,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2702512592077255,
      "step": 3144
    },
    {
      "epoch": 0.06292,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.024637858072916668,
      "learning_rate": 0.0001,
      "loss": 4.7756,
      "loss/crossentropy": 1.8181490898132324,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2310035452246666,
      "step": 3146
    },
    {
      "epoch": 0.06296,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.0373687744140625,
      "learning_rate": 0.0001,
      "loss": 4.9556,
      "loss/crossentropy": 1.9985284805297852,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2944463640451431,
      "step": 3148
    },
    {
      "epoch": 0.063,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.04142964680989583,
      "learning_rate": 0.0001,
      "loss": 5.0316,
      "loss/crossentropy": 2.3850419521331787,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29605095088481903,
      "step": 3150
    },
    {
      "epoch": 0.06304,
      "grad_norm": 4.125,
      "grad_norm_var": 0.17021077473958332,
      "learning_rate": 0.0001,
      "loss": 5.2062,
      "loss/crossentropy": 2.3815460205078125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30247962474823,
      "step": 3152
    },
    {
      "epoch": 0.06308,
      "grad_norm": 3.421875,
      "grad_norm_var": 0.19840087890625,
      "learning_rate": 0.0001,
      "loss": 4.5851,
      "loss/crossentropy": 1.995104193687439,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2506560683250427,
      "step": 3154
    },
    {
      "epoch": 0.06312,
      "grad_norm": 3.328125,
      "grad_norm_var": 0.2305084228515625,
      "learning_rate": 0.0001,
      "loss": 4.5885,
      "loss/crossentropy": 2.2037696838378906,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2716705799102783,
      "step": 3156
    },
    {
      "epoch": 0.06316,
      "grad_norm": 2.84375,
      "grad_norm_var": 0.23944905598958333,
      "learning_rate": 0.0001,
      "loss": 4.7155,
      "loss/crossentropy": 2.067265272140503,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25495442003011703,
      "step": 3158
    },
    {
      "epoch": 0.0632,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.23763020833333334,
      "learning_rate": 0.0001,
      "loss": 5.1219,
      "loss/crossentropy": 2.227355480194092,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27113111317157745,
      "step": 3160
    },
    {
      "epoch": 0.06324,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.23921610514322916,
      "learning_rate": 0.0001,
      "loss": 4.6777,
      "loss/crossentropy": 1.9077441096305847,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2630976662039757,
      "step": 3162
    },
    {
      "epoch": 0.06328,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.23515523274739583,
      "learning_rate": 0.0001,
      "loss": 5.0084,
      "loss/crossentropy": 2.1737552881240845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28365984559059143,
      "step": 3164
    },
    {
      "epoch": 0.06332,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.22542317708333334,
      "learning_rate": 0.0001,
      "loss": 4.9822,
      "loss/crossentropy": 1.9357402920722961,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27245980501174927,
      "step": 3166
    },
    {
      "epoch": 0.06336,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.0995269775390625,
      "learning_rate": 0.0001,
      "loss": 4.7681,
      "loss/crossentropy": 1.8484191298484802,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2933400571346283,
      "step": 3168
    },
    {
      "epoch": 0.0634,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.057738240559895834,
      "learning_rate": 0.0001,
      "loss": 5.0719,
      "loss/crossentropy": 2.3085306882858276,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2926720678806305,
      "step": 3170
    },
    {
      "epoch": 0.06344,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.040380859375,
      "learning_rate": 0.0001,
      "loss": 5.1464,
      "loss/crossentropy": 2.080895185470581,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2638559564948082,
      "step": 3172
    },
    {
      "epoch": 0.06348,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.03287353515625,
      "learning_rate": 0.0001,
      "loss": 4.6857,
      "loss/crossentropy": 2.035506248474121,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2692716419696808,
      "step": 3174
    },
    {
      "epoch": 0.06352,
      "grad_norm": 2.859375,
      "grad_norm_var": 0.035252888997395836,
      "learning_rate": 0.0001,
      "loss": 4.9479,
      "loss/crossentropy": 2.1712740659713745,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26579485833644867,
      "step": 3176
    },
    {
      "epoch": 0.06356,
      "grad_norm": 2.96875,
      "grad_norm_var": 0.04252827962239583,
      "learning_rate": 0.0001,
      "loss": 5.1654,
      "loss/crossentropy": 2.2256147861480713,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27605514228343964,
      "step": 3178
    },
    {
      "epoch": 0.0636,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.046507771809895834,
      "learning_rate": 0.0001,
      "loss": 4.6489,
      "loss/crossentropy": 2.0917118191719055,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2560323476791382,
      "step": 3180
    },
    {
      "epoch": 0.06364,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.045491536458333336,
      "learning_rate": 0.0001,
      "loss": 4.9833,
      "loss/crossentropy": 2.3109938502311707,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27040477097034454,
      "step": 3182
    },
    {
      "epoch": 0.06368,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.03984273274739583,
      "learning_rate": 0.0001,
      "loss": 5.1441,
      "loss/crossentropy": 2.3505672812461853,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29180124402046204,
      "step": 3184
    },
    {
      "epoch": 0.06372,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.04014383951822917,
      "learning_rate": 0.0001,
      "loss": 5.0224,
      "loss/crossentropy": 2.2471169233322144,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2676118314266205,
      "step": 3186
    },
    {
      "epoch": 0.06376,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.0264312744140625,
      "learning_rate": 0.0001,
      "loss": 4.8247,
      "loss/crossentropy": 2.3312125205993652,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29243919253349304,
      "step": 3188
    },
    {
      "epoch": 0.0638,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.02681884765625,
      "learning_rate": 0.0001,
      "loss": 4.9374,
      "loss/crossentropy": 2.1709930896759033,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2786664366722107,
      "step": 3190
    },
    {
      "epoch": 0.06384,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.02476806640625,
      "learning_rate": 0.0001,
      "loss": 4.7391,
      "loss/crossentropy": 1.8477665185928345,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26132869720458984,
      "step": 3192
    },
    {
      "epoch": 0.06388,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.01724853515625,
      "learning_rate": 0.0001,
      "loss": 4.8167,
      "loss/crossentropy": 2.2102121114730835,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26027651876211166,
      "step": 3194
    },
    {
      "epoch": 0.06392,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.0168365478515625,
      "learning_rate": 0.0001,
      "loss": 4.6381,
      "loss/crossentropy": 2.0011088252067566,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25629256665706635,
      "step": 3196
    },
    {
      "epoch": 0.06396,
      "grad_norm": 2.625,
      "grad_norm_var": 0.01685791015625,
      "learning_rate": 0.0001,
      "loss": 4.9215,
      "loss/crossentropy": 2.319412350654602,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28315384685993195,
      "step": 3198
    },
    {
      "epoch": 0.064,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.0145172119140625,
      "learning_rate": 0.0001,
      "loss": 4.7936,
      "loss/crossentropy": 2.192026138305664,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.264492504298687,
      "step": 3200
    },
    {
      "epoch": 0.06404,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.017024739583333334,
      "learning_rate": 0.0001,
      "loss": 4.8178,
      "loss/crossentropy": 2.1745734214782715,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29369065165519714,
      "step": 3202
    },
    {
      "epoch": 0.06408,
      "grad_norm": 3.109375,
      "grad_norm_var": 0.042313639322916666,
      "learning_rate": 0.0001,
      "loss": 5.577,
      "loss/crossentropy": 2.5039013624191284,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2835303544998169,
      "step": 3204
    },
    {
      "epoch": 0.06412,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.04075419108072917,
      "learning_rate": 0.0001,
      "loss": 5.0308,
      "loss/crossentropy": 2.4255706071853638,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3014884740114212,
      "step": 3206
    },
    {
      "epoch": 0.06416,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.03873291015625,
      "learning_rate": 0.0001,
      "loss": 5.002,
      "loss/crossentropy": 2.266150116920471,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2698349952697754,
      "step": 3208
    },
    {
      "epoch": 0.0642,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.03805338541666667,
      "learning_rate": 0.0001,
      "loss": 5.0173,
      "loss/crossentropy": 2.2042946815490723,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28830648958683014,
      "step": 3210
    },
    {
      "epoch": 0.06424,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.03578999837239583,
      "learning_rate": 0.0001,
      "loss": 5.0292,
      "loss/crossentropy": 2.4034690856933594,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2834039032459259,
      "step": 3212
    },
    {
      "epoch": 0.06428,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.0444000244140625,
      "learning_rate": 0.0001,
      "loss": 4.6655,
      "loss/crossentropy": 2.1347755193710327,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26405099034309387,
      "step": 3214
    },
    {
      "epoch": 0.06432,
      "grad_norm": 3.34375,
      "grad_norm_var": 0.07415262858072917,
      "learning_rate": 0.0001,
      "loss": 5.0862,
      "loss/crossentropy": 2.018262207508087,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27144598215818405,
      "step": 3216
    },
    {
      "epoch": 0.06436,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.076953125,
      "learning_rate": 0.0001,
      "loss": 4.3969,
      "loss/crossentropy": 2.0254003405570984,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24976836144924164,
      "step": 3218
    },
    {
      "epoch": 0.0644,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.06026102701822917,
      "learning_rate": 0.0001,
      "loss": 4.6779,
      "loss/crossentropy": 2.1952659487724304,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2786559462547302,
      "step": 3220
    },
    {
      "epoch": 0.06444,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.06024983723958333,
      "learning_rate": 0.0001,
      "loss": 4.8828,
      "loss/crossentropy": 2.1383036375045776,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2570757120847702,
      "step": 3222
    },
    {
      "epoch": 0.06448,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.05915425618489583,
      "learning_rate": 0.0001,
      "loss": 4.8248,
      "loss/crossentropy": 2.267812967300415,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2774328589439392,
      "step": 3224
    },
    {
      "epoch": 0.06452,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.0587066650390625,
      "learning_rate": 0.0001,
      "loss": 4.9389,
      "loss/crossentropy": 1.906205415725708,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2528962790966034,
      "step": 3226
    },
    {
      "epoch": 0.06456,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.05895182291666667,
      "learning_rate": 0.0001,
      "loss": 4.9823,
      "loss/crossentropy": 2.1628336906433105,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27087944746017456,
      "step": 3228
    },
    {
      "epoch": 0.0646,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.05807291666666667,
      "learning_rate": 0.0001,
      "loss": 4.5893,
      "loss/crossentropy": 1.8845162391662598,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2241554707288742,
      "step": 3230
    },
    {
      "epoch": 0.06464,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.019237263997395834,
      "learning_rate": 0.0001,
      "loss": 4.9143,
      "loss/crossentropy": 2.4157146215438843,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2770439088344574,
      "step": 3232
    },
    {
      "epoch": 0.06468,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.021825154622395832,
      "learning_rate": 0.0001,
      "loss": 4.832,
      "loss/crossentropy": 1.9979816675186157,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23696578294038773,
      "step": 3234
    },
    {
      "epoch": 0.06472,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.016364542643229167,
      "learning_rate": 0.0001,
      "loss": 5.1123,
      "loss/crossentropy": 2.1790190935134888,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.252600260078907,
      "step": 3236
    },
    {
      "epoch": 0.06476,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.016624959309895833,
      "learning_rate": 0.0001,
      "loss": 4.5943,
      "loss/crossentropy": 2.1612111926078796,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27929094433784485,
      "step": 3238
    },
    {
      "epoch": 0.0648,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.08124898274739584,
      "learning_rate": 0.0001,
      "loss": 4.8293,
      "loss/crossentropy": 2.261025071144104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3034510314464569,
      "step": 3240
    },
    {
      "epoch": 0.06484,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.081689453125,
      "learning_rate": 0.0001,
      "loss": 4.8243,
      "loss/crossentropy": 2.02247554063797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2655429244041443,
      "step": 3242
    },
    {
      "epoch": 0.06488,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.08289286295572916,
      "learning_rate": 0.0001,
      "loss": 4.9428,
      "loss/crossentropy": 2.495269775390625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32837189733982086,
      "step": 3244
    },
    {
      "epoch": 0.06492,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.07550455729166666,
      "learning_rate": 0.0001,
      "loss": 4.9928,
      "loss/crossentropy": 2.365166425704956,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27000221610069275,
      "step": 3246
    },
    {
      "epoch": 0.06496,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.0746246337890625,
      "learning_rate": 0.0001,
      "loss": 4.837,
      "loss/crossentropy": 1.8728906512260437,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2711311876773834,
      "step": 3248
    },
    {
      "epoch": 0.065,
      "grad_norm": 2.375,
      "grad_norm_var": 0.07427978515625,
      "learning_rate": 0.0001,
      "loss": 4.7916,
      "loss/crossentropy": 1.916531264781952,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23036544770002365,
      "step": 3250
    },
    {
      "epoch": 0.06504,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.07463277180989583,
      "learning_rate": 0.0001,
      "loss": 4.9399,
      "loss/crossentropy": 1.996503233909607,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24317056685686111,
      "step": 3252
    },
    {
      "epoch": 0.06508,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.07366536458333334,
      "learning_rate": 0.0001,
      "loss": 4.8746,
      "loss/crossentropy": 2.101921260356903,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2775810658931732,
      "step": 3254
    },
    {
      "epoch": 0.06512,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.005125935872395833,
      "learning_rate": 0.0001,
      "loss": 4.7526,
      "loss/crossentropy": 1.9286046028137207,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23964455723762512,
      "step": 3256
    },
    {
      "epoch": 0.06516,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.005464680989583333,
      "learning_rate": 0.0001,
      "loss": 4.8216,
      "loss/crossentropy": 2.224915862083435,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27516382932662964,
      "step": 3258
    },
    {
      "epoch": 0.0652,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.0133697509765625,
      "learning_rate": 0.0001,
      "loss": 4.6553,
      "loss/crossentropy": 2.259337306022644,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27185751497745514,
      "step": 3260
    },
    {
      "epoch": 0.06524,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.013117472330729166,
      "learning_rate": 0.0001,
      "loss": 4.7095,
      "loss/crossentropy": 2.1081044673919678,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23740330338478088,
      "step": 3262
    },
    {
      "epoch": 0.06528,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.0131744384765625,
      "learning_rate": 0.0001,
      "loss": 4.6375,
      "loss/crossentropy": 1.9032058119773865,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2508246824145317,
      "step": 3264
    },
    {
      "epoch": 0.06532,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.013353474934895833,
      "learning_rate": 0.0001,
      "loss": 4.7165,
      "loss/crossentropy": 2.0773792266845703,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2797544598579407,
      "step": 3266
    },
    {
      "epoch": 0.06536,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.013509114583333334,
      "learning_rate": 0.0001,
      "loss": 4.9136,
      "loss/crossentropy": 2.1591526865959167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2540442571043968,
      "step": 3268
    },
    {
      "epoch": 0.0654,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.014351399739583333,
      "learning_rate": 0.0001,
      "loss": 4.6248,
      "loss/crossentropy": 1.7735809683799744,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24107928574085236,
      "step": 3270
    },
    {
      "epoch": 0.06544,
      "grad_norm": 2.5,
      "grad_norm_var": 0.014647420247395833,
      "learning_rate": 0.0001,
      "loss": 4.7568,
      "loss/crossentropy": 2.069553792476654,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26356737315654755,
      "step": 3272
    },
    {
      "epoch": 0.06548,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.014623006184895834,
      "learning_rate": 0.0001,
      "loss": 4.7274,
      "loss/crossentropy": 1.9688642024993896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29453104734420776,
      "step": 3274
    },
    {
      "epoch": 0.06552,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.012043253580729166,
      "learning_rate": 0.0001,
      "loss": 4.4264,
      "loss/crossentropy": 1.785252034664154,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23213861882686615,
      "step": 3276
    },
    {
      "epoch": 0.06556,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.013255818684895834,
      "learning_rate": 0.0001,
      "loss": 4.779,
      "loss/crossentropy": 2.155774712562561,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25758183747529984,
      "step": 3278
    },
    {
      "epoch": 0.0656,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.016161092122395835,
      "learning_rate": 0.0001,
      "loss": 5.0453,
      "loss/crossentropy": 2.0403348803520203,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25007252395153046,
      "step": 3280
    },
    {
      "epoch": 0.06564,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.015208943684895834,
      "learning_rate": 0.0001,
      "loss": 4.7161,
      "loss/crossentropy": 1.9963608384132385,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2608266994357109,
      "step": 3282
    },
    {
      "epoch": 0.06568,
      "grad_norm": 2.75,
      "grad_norm_var": 0.019364420572916666,
      "learning_rate": 0.0001,
      "loss": 4.9275,
      "loss/crossentropy": 2.2021098732948303,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28361018002033234,
      "step": 3284
    },
    {
      "epoch": 0.06572,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.01812744140625,
      "learning_rate": 0.0001,
      "loss": 4.6533,
      "loss/crossentropy": 2.0363903641700745,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2641526162624359,
      "step": 3286
    },
    {
      "epoch": 0.06576,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.020466105143229166,
      "learning_rate": 0.0001,
      "loss": 4.6884,
      "loss/crossentropy": 2.1052531003952026,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24344927817583084,
      "step": 3288
    },
    {
      "epoch": 0.0658,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.02027587890625,
      "learning_rate": 0.0001,
      "loss": 4.9246,
      "loss/crossentropy": 2.0664029717445374,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25679293274879456,
      "step": 3290
    },
    {
      "epoch": 0.06584,
      "grad_norm": 2.625,
      "grad_norm_var": 0.013678995768229167,
      "learning_rate": 0.0001,
      "loss": 4.7802,
      "loss/crossentropy": 2.168351709842682,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2759328931570053,
      "step": 3292
    },
    {
      "epoch": 0.06588,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.013703409830729167,
      "learning_rate": 0.0001,
      "loss": 4.8481,
      "loss/crossentropy": 1.9305949211120605,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26165173947811127,
      "step": 3294
    },
    {
      "epoch": 0.06592,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.017476399739583332,
      "learning_rate": 0.0001,
      "loss": 4.6942,
      "loss/crossentropy": 1.9257155060768127,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24499841034412384,
      "step": 3296
    },
    {
      "epoch": 0.06596,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.017513020833333334,
      "learning_rate": 0.0001,
      "loss": 4.8941,
      "loss/crossentropy": 2.1406426429748535,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26676414906978607,
      "step": 3298
    },
    {
      "epoch": 0.066,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.016422526041666666,
      "learning_rate": 0.0001,
      "loss": 4.8293,
      "loss/crossentropy": 2.099781036376953,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2916935384273529,
      "step": 3300
    },
    {
      "epoch": 0.06604,
      "grad_norm": 2.875,
      "grad_norm_var": 0.023094685872395833,
      "learning_rate": 0.0001,
      "loss": 4.9504,
      "loss/crossentropy": 2.1077913641929626,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2667912393808365,
      "step": 3302
    },
    {
      "epoch": 0.06608,
      "grad_norm": 3.90625,
      "grad_norm_var": 0.1297271728515625,
      "learning_rate": 0.0001,
      "loss": 5.2952,
      "loss/crossentropy": 2.2583223581314087,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2907385528087616,
      "step": 3304
    },
    {
      "epoch": 0.06612,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.13405659993489583,
      "learning_rate": 0.0001,
      "loss": 4.5813,
      "loss/crossentropy": 1.847477912902832,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2272188812494278,
      "step": 3306
    },
    {
      "epoch": 0.06616,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.1392486572265625,
      "learning_rate": 0.0001,
      "loss": 4.6502,
      "loss/crossentropy": 1.7610225677490234,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23973071575164795,
      "step": 3308
    },
    {
      "epoch": 0.0662,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.1464019775390625,
      "learning_rate": 0.0001,
      "loss": 4.7123,
      "loss/crossentropy": 1.977916419506073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25456882268190384,
      "step": 3310
    },
    {
      "epoch": 0.06624,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.14095052083333334,
      "learning_rate": 0.0001,
      "loss": 4.7353,
      "loss/crossentropy": 2.2196428775787354,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2563377171754837,
      "step": 3312
    },
    {
      "epoch": 0.06628,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.14537760416666667,
      "learning_rate": 0.0001,
      "loss": 4.6907,
      "loss/crossentropy": 2.0861470699310303,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27121224999427795,
      "step": 3314
    },
    {
      "epoch": 0.06632,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.14287821451822916,
      "learning_rate": 0.0001,
      "loss": 4.6484,
      "loss/crossentropy": 1.9716283679008484,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.277963787317276,
      "step": 3316
    },
    {
      "epoch": 0.06636,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.1375,
      "learning_rate": 0.0001,
      "loss": 4.7985,
      "loss/crossentropy": 2.5604729652404785,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31031742691993713,
      "step": 3318
    },
    {
      "epoch": 0.0664,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.022337849934895834,
      "learning_rate": 0.0001,
      "loss": 5.4329,
      "loss/crossentropy": 2.2672252655029297,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29225634038448334,
      "step": 3320
    },
    {
      "epoch": 0.06644,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.020015462239583334,
      "learning_rate": 0.0001,
      "loss": 5.0557,
      "loss/crossentropy": 2.0992931723594666,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2648598402738571,
      "step": 3322
    },
    {
      "epoch": 0.06648,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.014827473958333334,
      "learning_rate": 0.0001,
      "loss": 5.0895,
      "loss/crossentropy": 2.208917260169983,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2998420298099518,
      "step": 3324
    },
    {
      "epoch": 0.06652,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.012791951497395834,
      "learning_rate": 0.0001,
      "loss": 4.8834,
      "loss/crossentropy": 2.290796995162964,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2638905793428421,
      "step": 3326
    },
    {
      "epoch": 0.06656,
      "grad_norm": 2.375,
      "grad_norm_var": 0.012214152018229167,
      "learning_rate": 0.0001,
      "loss": 4.8521,
      "loss/crossentropy": 2.4156445264816284,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28743064403533936,
      "step": 3328
    },
    {
      "epoch": 0.0666,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.012132771809895833,
      "learning_rate": 0.0001,
      "loss": 5.2205,
      "loss/crossentropy": 2.4604904651641846,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27585768699645996,
      "step": 3330
    },
    {
      "epoch": 0.06664,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.013818359375,
      "learning_rate": 0.0001,
      "loss": 4.8296,
      "loss/crossentropy": 1.8613844513893127,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27038969844579697,
      "step": 3332
    },
    {
      "epoch": 0.06668,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.014046223958333333,
      "learning_rate": 0.0001,
      "loss": 4.8777,
      "loss/crossentropy": 2.232776403427124,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2686954140663147,
      "step": 3334
    },
    {
      "epoch": 0.06672,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.0120758056640625,
      "learning_rate": 0.0001,
      "loss": 4.9536,
      "loss/crossentropy": 2.070033550262451,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26176171004772186,
      "step": 3336
    },
    {
      "epoch": 0.06676,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.01187744140625,
      "learning_rate": 0.0001,
      "loss": 4.8616,
      "loss/crossentropy": 2.38937509059906,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2539200633764267,
      "step": 3338
    },
    {
      "epoch": 0.0668,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.010933430989583333,
      "learning_rate": 0.0001,
      "loss": 5.1104,
      "loss/crossentropy": 2.272845983505249,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2856733053922653,
      "step": 3340
    },
    {
      "epoch": 0.06684,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.01181640625,
      "learning_rate": 0.0001,
      "loss": 4.7935,
      "loss/crossentropy": 2.427489161491394,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30572691559791565,
      "step": 3342
    },
    {
      "epoch": 0.06688,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.009407552083333333,
      "learning_rate": 0.0001,
      "loss": 4.9801,
      "loss/crossentropy": 2.4701327085494995,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28469331562519073,
      "step": 3344
    },
    {
      "epoch": 0.06692,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.006917317708333333,
      "learning_rate": 0.0001,
      "loss": 4.815,
      "loss/crossentropy": 2.0733558535575867,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.292859822511673,
      "step": 3346
    },
    {
      "epoch": 0.06696,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.0061187744140625,
      "learning_rate": 0.0001,
      "loss": 4.8379,
      "loss/crossentropy": 2.301755905151367,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3174070864915848,
      "step": 3348
    },
    {
      "epoch": 0.067,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.012580362955729167,
      "learning_rate": 0.0001,
      "loss": 4.3359,
      "loss/crossentropy": 2.039419114589691,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24543144553899765,
      "step": 3350
    },
    {
      "epoch": 0.06704,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.0144683837890625,
      "learning_rate": 0.0001,
      "loss": 4.7938,
      "loss/crossentropy": 1.9247611165046692,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2595005929470062,
      "step": 3352
    },
    {
      "epoch": 0.06708,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.01630859375,
      "learning_rate": 0.0001,
      "loss": 4.8294,
      "loss/crossentropy": 2.224974751472473,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26880575716495514,
      "step": 3354
    },
    {
      "epoch": 0.06712,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.0195220947265625,
      "learning_rate": 0.0001,
      "loss": 4.8922,
      "loss/crossentropy": 2.2601993083953857,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.279633030295372,
      "step": 3356
    },
    {
      "epoch": 0.06716,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.017085774739583334,
      "learning_rate": 0.0001,
      "loss": 4.5911,
      "loss/crossentropy": 1.8156417608261108,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22118539363145828,
      "step": 3358
    },
    {
      "epoch": 0.0672,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.020140584309895834,
      "learning_rate": 0.0001,
      "loss": 5.2032,
      "loss/crossentropy": 2.207859516143799,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2771998345851898,
      "step": 3360
    },
    {
      "epoch": 0.06724,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.02017822265625,
      "learning_rate": 0.0001,
      "loss": 4.9352,
      "loss/crossentropy": 1.9886083602905273,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2545766308903694,
      "step": 3362
    },
    {
      "epoch": 0.06728,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.025777180989583332,
      "learning_rate": 0.0001,
      "loss": 4.4323,
      "loss/crossentropy": 1.7046592235565186,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2207612618803978,
      "step": 3364
    },
    {
      "epoch": 0.06732,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.021122233072916666,
      "learning_rate": 0.0001,
      "loss": 5.0209,
      "loss/crossentropy": 2.0283663868904114,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2743644416332245,
      "step": 3366
    },
    {
      "epoch": 0.06736,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.019498697916666665,
      "learning_rate": 0.0001,
      "loss": 4.4788,
      "loss/crossentropy": 1.975772500038147,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2505730241537094,
      "step": 3368
    },
    {
      "epoch": 0.0674,
      "grad_norm": 2.5,
      "grad_norm_var": 0.019950358072916667,
      "learning_rate": 0.0001,
      "loss": 4.6863,
      "loss/crossentropy": 1.9021872282028198,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25430944561958313,
      "step": 3370
    },
    {
      "epoch": 0.06744,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.0182037353515625,
      "learning_rate": 0.0001,
      "loss": 5.1859,
      "loss/crossentropy": 2.3888463973999023,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27896443009376526,
      "step": 3372
    },
    {
      "epoch": 0.06748,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.02086181640625,
      "learning_rate": 0.0001,
      "loss": 4.6041,
      "loss/crossentropy": 1.7844690680503845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2536455988883972,
      "step": 3374
    },
    {
      "epoch": 0.06752,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.021442667643229166,
      "learning_rate": 0.0001,
      "loss": 4.4538,
      "loss/crossentropy": 1.919598639011383,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26129309833049774,
      "step": 3376
    },
    {
      "epoch": 0.06756,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.022835286458333333,
      "learning_rate": 0.0001,
      "loss": 5.0568,
      "loss/crossentropy": 2.2292110919952393,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25612247735261917,
      "step": 3378
    },
    {
      "epoch": 0.0676,
      "grad_norm": 3.4375,
      "grad_norm_var": 0.0779296875,
      "learning_rate": 0.0001,
      "loss": 4.8674,
      "loss/crossentropy": 2.2235841751098633,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2607487738132477,
      "step": 3380
    },
    {
      "epoch": 0.06764,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.08089192708333333,
      "learning_rate": 0.0001,
      "loss": 4.5872,
      "loss/crossentropy": 2.1375235319137573,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2344205006957054,
      "step": 3382
    },
    {
      "epoch": 0.06768,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.08501688639322917,
      "learning_rate": 0.0001,
      "loss": 4.6076,
      "loss/crossentropy": 2.020237445831299,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2550392523407936,
      "step": 3384
    },
    {
      "epoch": 0.06772,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.07911783854166667,
      "learning_rate": 0.0001,
      "loss": 4.6939,
      "loss/crossentropy": 2.138959765434265,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2742607295513153,
      "step": 3386
    },
    {
      "epoch": 0.06776,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.07864481608072917,
      "learning_rate": 0.0001,
      "loss": 4.9065,
      "loss/crossentropy": 2.521559953689575,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2892928719520569,
      "step": 3388
    },
    {
      "epoch": 0.0678,
      "grad_norm": 2.375,
      "grad_norm_var": 0.07815348307291667,
      "learning_rate": 0.0001,
      "loss": 5.049,
      "loss/crossentropy": 2.169550120830536,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27750439941883087,
      "step": 3390
    },
    {
      "epoch": 0.06784,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.07464090983072917,
      "learning_rate": 0.0001,
      "loss": 4.7401,
      "loss/crossentropy": 1.975584864616394,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2557516545057297,
      "step": 3392
    },
    {
      "epoch": 0.06788,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.07617899576822916,
      "learning_rate": 0.0001,
      "loss": 4.9207,
      "loss/crossentropy": 2.5837322473526,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3444886952638626,
      "step": 3394
    },
    {
      "epoch": 0.06792,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.022294108072916666,
      "learning_rate": 0.0001,
      "loss": 4.743,
      "loss/crossentropy": 1.963110864162445,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24120041728019714,
      "step": 3396
    },
    {
      "epoch": 0.06796,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.0211090087890625,
      "learning_rate": 0.0001,
      "loss": 4.5759,
      "loss/crossentropy": 2.381603956222534,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26319295167922974,
      "step": 3398
    },
    {
      "epoch": 0.068,
      "grad_norm": 2.25,
      "grad_norm_var": 0.023078409830729167,
      "learning_rate": 0.0001,
      "loss": 4.5642,
      "loss/crossentropy": 2.054026961326599,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24540280550718307,
      "step": 3400
    },
    {
      "epoch": 0.06804,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.0187652587890625,
      "learning_rate": 0.0001,
      "loss": 4.3661,
      "loss/crossentropy": 2.047453820705414,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2524164840579033,
      "step": 3402
    },
    {
      "epoch": 0.06808,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.018317667643229167,
      "learning_rate": 0.0001,
      "loss": 4.7839,
      "loss/crossentropy": 1.8616933226585388,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23905867338180542,
      "step": 3404
    },
    {
      "epoch": 0.06812,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.016422526041666666,
      "learning_rate": 0.0001,
      "loss": 4.6932,
      "loss/crossentropy": 2.4304568767547607,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2802550047636032,
      "step": 3406
    },
    {
      "epoch": 0.06816,
      "grad_norm": 2.5,
      "grad_norm_var": 0.01318359375,
      "learning_rate": 0.0001,
      "loss": 4.7005,
      "loss/crossentropy": 1.8178748488426208,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24266308546066284,
      "step": 3408
    },
    {
      "epoch": 0.0682,
      "grad_norm": 2.625,
      "grad_norm_var": 0.0116363525390625,
      "learning_rate": 0.0001,
      "loss": 5.0152,
      "loss/crossentropy": 2.025859773159027,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.282450333237648,
      "step": 3410
    },
    {
      "epoch": 0.06824,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.010383097330729167,
      "learning_rate": 0.0001,
      "loss": 4.737,
      "loss/crossentropy": 2.032994568347931,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26047058403491974,
      "step": 3412
    },
    {
      "epoch": 0.06828,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.007840983072916667,
      "learning_rate": 0.0001,
      "loss": 4.6428,
      "loss/crossentropy": 2.2468607425689697,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2814445495605469,
      "step": 3414
    },
    {
      "epoch": 0.06832,
      "grad_norm": 2.75,
      "grad_norm_var": 0.03629150390625,
      "learning_rate": 0.0001,
      "loss": 4.6854,
      "loss/crossentropy": 2.2534161806106567,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3231179416179657,
      "step": 3416
    },
    {
      "epoch": 0.06836,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.03443603515625,
      "learning_rate": 0.0001,
      "loss": 4.9599,
      "loss/crossentropy": 2.1678181886672974,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2840191572904587,
      "step": 3418
    },
    {
      "epoch": 0.0684,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.03453776041666667,
      "learning_rate": 0.0001,
      "loss": 4.8667,
      "loss/crossentropy": 2.053459882736206,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.271483838558197,
      "step": 3420
    },
    {
      "epoch": 0.06844,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.03243815104166667,
      "learning_rate": 0.0001,
      "loss": 4.714,
      "loss/crossentropy": 2.2278919219970703,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2781473994255066,
      "step": 3422
    },
    {
      "epoch": 0.06848,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.0384674072265625,
      "learning_rate": 0.0001,
      "loss": 4.5262,
      "loss/crossentropy": 1.9582479000091553,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24171434342861176,
      "step": 3424
    },
    {
      "epoch": 0.06852,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.04121805826822917,
      "learning_rate": 0.0001,
      "loss": 4.4734,
      "loss/crossentropy": 1.808964192867279,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22458729147911072,
      "step": 3426
    },
    {
      "epoch": 0.06856,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.04296468098958333,
      "learning_rate": 0.0001,
      "loss": 4.464,
      "loss/crossentropy": 1.9225260019302368,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2508438527584076,
      "step": 3428
    },
    {
      "epoch": 0.0686,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.044465128580729166,
      "learning_rate": 0.0001,
      "loss": 4.9647,
      "loss/crossentropy": 2.2446881532669067,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26919613778591156,
      "step": 3430
    },
    {
      "epoch": 0.06864,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.017899576822916666,
      "learning_rate": 0.0001,
      "loss": 4.6865,
      "loss/crossentropy": 2.2047033309936523,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2581355720758438,
      "step": 3432
    },
    {
      "epoch": 0.06868,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.015729777018229165,
      "learning_rate": 0.0001,
      "loss": 4.7139,
      "loss/crossentropy": 2.1223543882369995,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25814756751060486,
      "step": 3434
    },
    {
      "epoch": 0.06872,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.016813151041666665,
      "learning_rate": 0.0001,
      "loss": 4.6771,
      "loss/crossentropy": 2.1641053557395935,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25222497433423996,
      "step": 3436
    },
    {
      "epoch": 0.06876,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.0189605712890625,
      "learning_rate": 0.0001,
      "loss": 4.6972,
      "loss/crossentropy": 1.9569795727729797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2681007981300354,
      "step": 3438
    },
    {
      "epoch": 0.0688,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.013216145833333333,
      "learning_rate": 0.0001,
      "loss": 4.8214,
      "loss/crossentropy": 2.212220251560211,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28191742300987244,
      "step": 3440
    },
    {
      "epoch": 0.06884,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.011546834309895834,
      "learning_rate": 0.0001,
      "loss": 4.4162,
      "loss/crossentropy": 1.9564262628555298,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2560906559228897,
      "step": 3442
    },
    {
      "epoch": 0.06888,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.012043253580729166,
      "learning_rate": 0.0001,
      "loss": 4.6837,
      "loss/crossentropy": 1.8553346395492554,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24626825004816055,
      "step": 3444
    },
    {
      "epoch": 0.06892,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.012007649739583333,
      "learning_rate": 0.0001,
      "loss": 4.7914,
      "loss/crossentropy": 1.9803723692893982,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2712126225233078,
      "step": 3446
    },
    {
      "epoch": 0.06896,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.010423787434895833,
      "learning_rate": 0.0001,
      "loss": 4.857,
      "loss/crossentropy": 1.9914751648902893,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2565220594406128,
      "step": 3448
    },
    {
      "epoch": 0.069,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.010758463541666667,
      "learning_rate": 0.0001,
      "loss": 4.8013,
      "loss/crossentropy": 2.2114094495773315,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28365010023117065,
      "step": 3450
    },
    {
      "epoch": 0.06904,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.009577433268229166,
      "learning_rate": 0.0001,
      "loss": 5.0322,
      "loss/crossentropy": 2.4366514682769775,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30788426101207733,
      "step": 3452
    },
    {
      "epoch": 0.06908,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.010595703125,
      "learning_rate": 0.0001,
      "loss": 4.7902,
      "loss/crossentropy": 2.304569959640503,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2835633158683777,
      "step": 3454
    },
    {
      "epoch": 0.06912,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.027318318684895832,
      "learning_rate": 0.0001,
      "loss": 5.151,
      "loss/crossentropy": 2.2518080472946167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.278719499707222,
      "step": 3456
    },
    {
      "epoch": 0.06916,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.027904256184895834,
      "learning_rate": 0.0001,
      "loss": 5.1688,
      "loss/crossentropy": 2.333768129348755,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28694969415664673,
      "step": 3458
    },
    {
      "epoch": 0.0692,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.0260650634765625,
      "learning_rate": 0.0001,
      "loss": 5.1421,
      "loss/crossentropy": 2.3534432649612427,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3005864769220352,
      "step": 3460
    },
    {
      "epoch": 0.06924,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.022981770833333335,
      "learning_rate": 0.0001,
      "loss": 4.9502,
      "loss/crossentropy": 2.0703017711639404,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25969168543815613,
      "step": 3462
    },
    {
      "epoch": 0.06928,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.023502604166666666,
      "learning_rate": 0.0001,
      "loss": 5.0258,
      "loss/crossentropy": 2.167420506477356,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32322582602500916,
      "step": 3464
    },
    {
      "epoch": 0.06932,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.021222941080729165,
      "learning_rate": 0.0001,
      "loss": 4.5342,
      "loss/crossentropy": 2.0845181941986084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26804475486278534,
      "step": 3466
    },
    {
      "epoch": 0.06936,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.021219889322916668,
      "learning_rate": 0.0001,
      "loss": 5.0459,
      "loss/crossentropy": 2.4165114164352417,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2866530567407608,
      "step": 3468
    },
    {
      "epoch": 0.0694,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.022359212239583332,
      "learning_rate": 0.0001,
      "loss": 5.0938,
      "loss/crossentropy": 2.4152863025665283,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30185502767562866,
      "step": 3470
    },
    {
      "epoch": 0.06944,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.014924112955729167,
      "learning_rate": 0.0001,
      "loss": 4.6362,
      "loss/crossentropy": 2.1878501176834106,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2736133933067322,
      "step": 3472
    },
    {
      "epoch": 0.06948,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.015262858072916666,
      "learning_rate": 0.0001,
      "loss": 4.812,
      "loss/crossentropy": 2.055173695087433,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2609568238258362,
      "step": 3474
    },
    {
      "epoch": 0.06952,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.015550740559895833,
      "learning_rate": 0.0001,
      "loss": 4.8201,
      "loss/crossentropy": 2.050000250339508,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2713121324777603,
      "step": 3476
    },
    {
      "epoch": 0.06956,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.019169108072916666,
      "learning_rate": 0.0001,
      "loss": 4.9916,
      "loss/crossentropy": 2.227464199066162,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2710970640182495,
      "step": 3478
    },
    {
      "epoch": 0.0696,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.016828409830729165,
      "learning_rate": 0.0001,
      "loss": 4.7435,
      "loss/crossentropy": 2.096015691757202,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29556676745414734,
      "step": 3480
    },
    {
      "epoch": 0.06964,
      "grad_norm": 2.921875,
      "grad_norm_var": 0.028804524739583334,
      "learning_rate": 0.0001,
      "loss": 4.6738,
      "loss/crossentropy": 1.9252901673316956,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2976933419704437,
      "step": 3482
    },
    {
      "epoch": 0.06968,
      "grad_norm": 2.25,
      "grad_norm_var": 0.03385009765625,
      "learning_rate": 0.0001,
      "loss": 4.7679,
      "loss/crossentropy": 2.258090019226074,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27399829030036926,
      "step": 3484
    },
    {
      "epoch": 0.06972,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.03192952473958333,
      "learning_rate": 0.0001,
      "loss": 4.7614,
      "loss/crossentropy": 2.2776867151260376,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29224833846092224,
      "step": 3486
    },
    {
      "epoch": 0.06976,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.031819661458333336,
      "learning_rate": 0.0001,
      "loss": 5.1628,
      "loss/crossentropy": 2.097061276435852,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26103661954402924,
      "step": 3488
    },
    {
      "epoch": 0.0698,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.03264567057291667,
      "learning_rate": 0.0001,
      "loss": 4.7364,
      "loss/crossentropy": 2.206419885158539,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2775641232728958,
      "step": 3490
    },
    {
      "epoch": 0.06984,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.035563151041666664,
      "learning_rate": 0.0001,
      "loss": 4.7216,
      "loss/crossentropy": 1.962704062461853,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2421455979347229,
      "step": 3492
    },
    {
      "epoch": 0.06988,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.032698567708333334,
      "learning_rate": 0.0001,
      "loss": 4.7358,
      "loss/crossentropy": 2.223302483558655,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2709382176399231,
      "step": 3494
    },
    {
      "epoch": 0.06992,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.034764607747395836,
      "learning_rate": 0.0001,
      "loss": 4.8563,
      "loss/crossentropy": 2.259950876235962,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29514479637145996,
      "step": 3496
    },
    {
      "epoch": 0.06996,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.021141560872395833,
      "learning_rate": 0.0001,
      "loss": 4.6147,
      "loss/crossentropy": 2.2337416410446167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2585422098636627,
      "step": 3498
    },
    {
      "epoch": 0.07,
      "grad_norm": 2.75,
      "grad_norm_var": 0.022663370768229166,
      "learning_rate": 0.0001,
      "loss": 4.9278,
      "loss/crossentropy": 2.131904423236847,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32493171095848083,
      "step": 3500
    },
    {
      "epoch": 0.07004,
      "grad_norm": 3.109375,
      "grad_norm_var": 0.04553629557291667,
      "learning_rate": 0.0001,
      "loss": 4.9285,
      "loss/crossentropy": 2.461912155151367,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2917899489402771,
      "step": 3502
    },
    {
      "epoch": 0.07008,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.04366861979166667,
      "learning_rate": 0.0001,
      "loss": 4.6466,
      "loss/crossentropy": 2.05659943819046,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24786780774593353,
      "step": 3504
    },
    {
      "epoch": 0.07012,
      "grad_norm": 2.625,
      "grad_norm_var": 0.0412261962890625,
      "learning_rate": 0.0001,
      "loss": 4.7282,
      "loss/crossentropy": 2.3972705602645874,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29012058675289154,
      "step": 3506
    },
    {
      "epoch": 0.07016,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.03945210774739583,
      "learning_rate": 0.0001,
      "loss": 4.6561,
      "loss/crossentropy": 1.8465647101402283,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26938022673130035,
      "step": 3508
    },
    {
      "epoch": 0.0702,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.0434722900390625,
      "learning_rate": 0.0001,
      "loss": 4.9007,
      "loss/crossentropy": 2.2447493076324463,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27260421216487885,
      "step": 3510
    },
    {
      "epoch": 0.07024,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.042464192708333334,
      "learning_rate": 0.0001,
      "loss": 4.7774,
      "loss/crossentropy": 2.4258209466934204,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.285652831196785,
      "step": 3512
    },
    {
      "epoch": 0.07028,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.04248046875,
      "learning_rate": 0.0001,
      "loss": 4.7353,
      "loss/crossentropy": 2.1033068895339966,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2628294378519058,
      "step": 3514
    },
    {
      "epoch": 0.07032,
      "grad_norm": 2.5,
      "grad_norm_var": 0.03819071451822917,
      "learning_rate": 0.0001,
      "loss": 4.8036,
      "loss/crossentropy": 2.2740964889526367,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27228833734989166,
      "step": 3516
    },
    {
      "epoch": 0.07036,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.015104166666666667,
      "learning_rate": 0.0001,
      "loss": 4.7699,
      "loss/crossentropy": 2.2315655946731567,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26613348722457886,
      "step": 3518
    },
    {
      "epoch": 0.0704,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.014720662434895834,
      "learning_rate": 0.0001,
      "loss": 4.6038,
      "loss/crossentropy": 1.9001839756965637,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.248238705098629,
      "step": 3520
    },
    {
      "epoch": 0.07044,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.013655598958333333,
      "learning_rate": 0.0001,
      "loss": 4.7034,
      "loss/crossentropy": 2.0940937399864197,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28850243985652924,
      "step": 3522
    },
    {
      "epoch": 0.07048,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.014338175455729166,
      "learning_rate": 0.0001,
      "loss": 4.9187,
      "loss/crossentropy": 1.9088054299354553,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2577759325504303,
      "step": 3524
    },
    {
      "epoch": 0.07052,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.01031494140625,
      "learning_rate": 0.0001,
      "loss": 4.7852,
      "loss/crossentropy": 2.1965672969818115,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2696071192622185,
      "step": 3526
    },
    {
      "epoch": 0.07056,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.010770670572916667,
      "learning_rate": 0.0001,
      "loss": 4.7657,
      "loss/crossentropy": 2.245758891105652,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2646239101886749,
      "step": 3528
    },
    {
      "epoch": 0.0706,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.013895670572916666,
      "learning_rate": 0.0001,
      "loss": 4.7794,
      "loss/crossentropy": 2.180204927921295,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2582162171602249,
      "step": 3530
    },
    {
      "epoch": 0.07064,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.019562784830729166,
      "learning_rate": 0.0001,
      "loss": 5.04,
      "loss/crossentropy": 2.193474531173706,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.40557755529880524,
      "step": 3532
    },
    {
      "epoch": 0.07068,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.021361287434895834,
      "learning_rate": 0.0001,
      "loss": 4.9562,
      "loss/crossentropy": 2.2667607069015503,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28741554915905,
      "step": 3534
    },
    {
      "epoch": 0.07072,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.022093709309895834,
      "learning_rate": 0.0001,
      "loss": 4.8215,
      "loss/crossentropy": 1.9890388250350952,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2612537443637848,
      "step": 3536
    },
    {
      "epoch": 0.07076,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.02392578125,
      "learning_rate": 0.0001,
      "loss": 4.7544,
      "loss/crossentropy": 1.9390615820884705,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23259633034467697,
      "step": 3538
    },
    {
      "epoch": 0.0708,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.023193359375,
      "learning_rate": 0.0001,
      "loss": 4.8227,
      "loss/crossentropy": 2.3050389289855957,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2738967537879944,
      "step": 3540
    },
    {
      "epoch": 0.07084,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.025862630208333334,
      "learning_rate": 0.0001,
      "loss": 4.4323,
      "loss/crossentropy": 2.3832077980041504,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29863911867141724,
      "step": 3542
    },
    {
      "epoch": 0.07088,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.02822265625,
      "learning_rate": 0.0001,
      "loss": 4.7101,
      "loss/crossentropy": 1.8186699748039246,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22366883605718613,
      "step": 3544
    },
    {
      "epoch": 0.07092,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.024235026041666666,
      "learning_rate": 0.0001,
      "loss": 4.8151,
      "loss/crossentropy": 2.2650288343429565,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2733971029520035,
      "step": 3546
    },
    {
      "epoch": 0.07096,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.0191070556640625,
      "learning_rate": 0.0001,
      "loss": 4.7514,
      "loss/crossentropy": 2.432945966720581,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.269818976521492,
      "step": 3548
    },
    {
      "epoch": 0.071,
      "grad_norm": 4.15625,
      "grad_norm_var": 0.19250386555989582,
      "learning_rate": 0.0001,
      "loss": 4.9461,
      "loss/crossentropy": 2.021497666835785,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2488800287246704,
      "step": 3550
    },
    {
      "epoch": 0.07104,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.2001129150390625,
      "learning_rate": 0.0001,
      "loss": 5.0341,
      "loss/crossentropy": 2.0840535163879395,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2780974507331848,
      "step": 3552
    },
    {
      "epoch": 0.07108,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.19724833170572917,
      "learning_rate": 0.0001,
      "loss": 4.8123,
      "loss/crossentropy": 2.352238416671753,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2830911874771118,
      "step": 3554
    },
    {
      "epoch": 0.07112,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.19795633951822916,
      "learning_rate": 0.0001,
      "loss": 4.9632,
      "loss/crossentropy": 2.395397186279297,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2861028015613556,
      "step": 3556
    },
    {
      "epoch": 0.07116,
      "grad_norm": 2.84375,
      "grad_norm_var": 0.19630533854166668,
      "learning_rate": 0.0001,
      "loss": 5.0203,
      "loss/crossentropy": 2.6454248428344727,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28888577222824097,
      "step": 3558
    },
    {
      "epoch": 0.0712,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.19394429524739584,
      "learning_rate": 0.0001,
      "loss": 4.8228,
      "loss/crossentropy": 2.21127188205719,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2676347494125366,
      "step": 3560
    },
    {
      "epoch": 0.07124,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.19245503743489584,
      "learning_rate": 0.0001,
      "loss": 4.7502,
      "loss/crossentropy": 2.0187097787857056,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24967321753501892,
      "step": 3562
    },
    {
      "epoch": 0.07128,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.19409077962239582,
      "learning_rate": 0.0001,
      "loss": 4.9548,
      "loss/crossentropy": 2.1822619438171387,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24893341958522797,
      "step": 3564
    },
    {
      "epoch": 0.07132,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.03178609212239583,
      "learning_rate": 0.0001,
      "loss": 5.0912,
      "loss/crossentropy": 2.486607313156128,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29665203392505646,
      "step": 3566
    },
    {
      "epoch": 0.07136,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.020048014322916665,
      "learning_rate": 0.0001,
      "loss": 5.0494,
      "loss/crossentropy": 2.2631163597106934,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.268096499145031,
      "step": 3568
    },
    {
      "epoch": 0.0714,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.02760009765625,
      "learning_rate": 0.0001,
      "loss": 4.7668,
      "loss/crossentropy": 2.3393882513046265,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29145532846450806,
      "step": 3570
    },
    {
      "epoch": 0.07144,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.03980712890625,
      "learning_rate": 0.0001,
      "loss": 4.8281,
      "loss/crossentropy": 2.007299244403839,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30512700974941254,
      "step": 3572
    },
    {
      "epoch": 0.07148,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.0369049072265625,
      "learning_rate": 0.0001,
      "loss": 4.7891,
      "loss/crossentropy": 1.9879329800605774,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22510841488838196,
      "step": 3574
    },
    {
      "epoch": 0.07152,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.03443603515625,
      "learning_rate": 0.0001,
      "loss": 4.9688,
      "loss/crossentropy": 2.387833833694458,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3007010221481323,
      "step": 3576
    },
    {
      "epoch": 0.07156,
      "grad_norm": 2.625,
      "grad_norm_var": 0.03664957682291667,
      "learning_rate": 0.0001,
      "loss": 4.7975,
      "loss/crossentropy": 2.3306411504745483,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.269440695643425,
      "step": 3578
    },
    {
      "epoch": 0.0716,
      "grad_norm": 2.5,
      "grad_norm_var": 0.03462626139322917,
      "learning_rate": 0.0001,
      "loss": 4.5348,
      "loss/crossentropy": 1.8359156847000122,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23353691399097443,
      "step": 3580
    },
    {
      "epoch": 0.07164,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.033080037434895834,
      "learning_rate": 0.0001,
      "loss": 4.9226,
      "loss/crossentropy": 2.257680654525757,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2764490246772766,
      "step": 3582
    },
    {
      "epoch": 0.07168,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.03310445149739583,
      "learning_rate": 0.0001,
      "loss": 4.6434,
      "loss/crossentropy": 2.589483857154846,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28172188997268677,
      "step": 3584
    },
    {
      "epoch": 0.07172,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.04182840983072917,
      "learning_rate": 0.0001,
      "loss": 4.5434,
      "loss/crossentropy": 1.8202016949653625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22330023348331451,
      "step": 3586
    },
    {
      "epoch": 0.07176,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.028609212239583334,
      "learning_rate": 0.0001,
      "loss": 4.8344,
      "loss/crossentropy": 2.0311816334724426,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2622206509113312,
      "step": 3588
    },
    {
      "epoch": 0.0718,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.028400675455729166,
      "learning_rate": 0.0001,
      "loss": 4.5158,
      "loss/crossentropy": 1.7991753220558167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23093865811824799,
      "step": 3590
    },
    {
      "epoch": 0.07184,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.030924479166666668,
      "learning_rate": 0.0001,
      "loss": 4.3148,
      "loss/crossentropy": 1.6141473054885864,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21914401650428772,
      "step": 3592
    },
    {
      "epoch": 0.07188,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.0289703369140625,
      "learning_rate": 0.0001,
      "loss": 4.8007,
      "loss/crossentropy": 2.3337208032608032,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2612123265862465,
      "step": 3594
    },
    {
      "epoch": 0.07192,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.0322662353515625,
      "learning_rate": 0.0001,
      "loss": 4.5811,
      "loss/crossentropy": 2.191028594970703,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2598741352558136,
      "step": 3596
    },
    {
      "epoch": 0.07196,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.03178609212239583,
      "learning_rate": 0.0001,
      "loss": 4.9562,
      "loss/crossentropy": 2.0293691158294678,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27143266797065735,
      "step": 3598
    },
    {
      "epoch": 0.072,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.03437398274739583,
      "learning_rate": 0.0001,
      "loss": 5.1335,
      "loss/crossentropy": 2.1257725954055786,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33300966024398804,
      "step": 3600
    },
    {
      "epoch": 0.07204,
      "grad_norm": 2.375,
      "grad_norm_var": 0.0171875,
      "learning_rate": 0.0001,
      "loss": 4.8223,
      "loss/crossentropy": 2.296278953552246,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28724825382232666,
      "step": 3602
    },
    {
      "epoch": 0.07208,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.016731770833333333,
      "learning_rate": 0.0001,
      "loss": 4.8925,
      "loss/crossentropy": 2.258358597755432,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26510895788669586,
      "step": 3604
    },
    {
      "epoch": 0.07212,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.0170318603515625,
      "learning_rate": 0.0001,
      "loss": 5.0383,
      "loss/crossentropy": 2.0454649925231934,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2725224494934082,
      "step": 3606
    },
    {
      "epoch": 0.07216,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.015250651041666667,
      "learning_rate": 0.0001,
      "loss": 4.6582,
      "loss/crossentropy": 2.1844204664230347,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2729914039373398,
      "step": 3608
    },
    {
      "epoch": 0.0722,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.015608723958333333,
      "learning_rate": 0.0001,
      "loss": 4.4613,
      "loss/crossentropy": 1.8897106647491455,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26307350397109985,
      "step": 3610
    },
    {
      "epoch": 0.07224,
      "grad_norm": 2.5,
      "grad_norm_var": 0.013377888997395834,
      "learning_rate": 0.0001,
      "loss": 4.5695,
      "loss/crossentropy": 1.9441962838172913,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24454529583454132,
      "step": 3612
    },
    {
      "epoch": 0.07228,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.063623046875,
      "learning_rate": 0.0001,
      "loss": 4.7654,
      "loss/crossentropy": 2.10969078540802,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2900787442922592,
      "step": 3614
    },
    {
      "epoch": 0.07232,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.06301676432291667,
      "learning_rate": 0.0001,
      "loss": 4.5195,
      "loss/crossentropy": 2.1384644508361816,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2722831964492798,
      "step": 3616
    },
    {
      "epoch": 0.07236,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.06412760416666667,
      "learning_rate": 0.0001,
      "loss": 4.4995,
      "loss/crossentropy": 2.0648157596588135,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24860350787639618,
      "step": 3618
    },
    {
      "epoch": 0.0724,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.06441650390625,
      "learning_rate": 0.0001,
      "loss": 4.7863,
      "loss/crossentropy": 2.2188034057617188,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2476331740617752,
      "step": 3620
    },
    {
      "epoch": 0.07244,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.06516927083333333,
      "learning_rate": 0.0001,
      "loss": 4.792,
      "loss/crossentropy": 2.1361395120620728,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24962469190359116,
      "step": 3622
    },
    {
      "epoch": 0.07248,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.06457417805989583,
      "learning_rate": 0.0001,
      "loss": 4.739,
      "loss/crossentropy": 2.2646392583847046,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.260420486330986,
      "step": 3624
    },
    {
      "epoch": 0.07252,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.0630523681640625,
      "learning_rate": 0.0001,
      "loss": 4.6937,
      "loss/crossentropy": 2.2822424173355103,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2585812509059906,
      "step": 3626
    },
    {
      "epoch": 0.07256,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.06852925618489583,
      "learning_rate": 0.0001,
      "loss": 4.2868,
      "loss/crossentropy": 1.8197516798973083,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24321961402893066,
      "step": 3628
    },
    {
      "epoch": 0.0726,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.007649739583333333,
      "learning_rate": 0.0001,
      "loss": 4.4835,
      "loss/crossentropy": 2.1475032567977905,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2687895894050598,
      "step": 3630
    },
    {
      "epoch": 0.07264,
      "grad_norm": 2.25,
      "grad_norm_var": 0.006538899739583334,
      "learning_rate": 0.0001,
      "loss": 4.2353,
      "loss/crossentropy": 1.9497992992401123,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24942665547132492,
      "step": 3632
    },
    {
      "epoch": 0.07268,
      "grad_norm": 2.5,
      "grad_norm_var": 0.007059733072916667,
      "learning_rate": 0.0001,
      "loss": 4.7455,
      "loss/crossentropy": 1.8786492347717285,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23713821917772293,
      "step": 3634
    },
    {
      "epoch": 0.07272,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.00738525390625,
      "learning_rate": 0.0001,
      "loss": 5.089,
      "loss/crossentropy": 2.474532127380371,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3133770227432251,
      "step": 3636
    },
    {
      "epoch": 0.07276,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.0073394775390625,
      "learning_rate": 0.0001,
      "loss": 4.912,
      "loss/crossentropy": 2.1231455206871033,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2547219544649124,
      "step": 3638
    },
    {
      "epoch": 0.0728,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.009007771809895834,
      "learning_rate": 0.0001,
      "loss": 4.4727,
      "loss/crossentropy": 2.0511630177497864,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26947300136089325,
      "step": 3640
    },
    {
      "epoch": 0.07284,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.01060791015625,
      "learning_rate": 0.0001,
      "loss": 4.7332,
      "loss/crossentropy": 1.7076187133789062,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21764510869979858,
      "step": 3642
    },
    {
      "epoch": 0.07288,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.009723917643229166,
      "learning_rate": 0.0001,
      "loss": 4.8396,
      "loss/crossentropy": 2.069926142692566,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26851218193769455,
      "step": 3644
    },
    {
      "epoch": 0.07292,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.010054524739583333,
      "learning_rate": 0.0001,
      "loss": 5.0802,
      "loss/crossentropy": 2.1369277238845825,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27878354489803314,
      "step": 3646
    },
    {
      "epoch": 0.07296,
      "grad_norm": 2.375,
      "grad_norm_var": 0.008072916666666667,
      "learning_rate": 0.0001,
      "loss": 4.3659,
      "loss/crossentropy": 2.095974624156952,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25285808742046356,
      "step": 3648
    },
    {
      "epoch": 0.073,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.011888631184895833,
      "learning_rate": 0.0001,
      "loss": 4.9301,
      "loss/crossentropy": 2.240627646446228,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24894578754901886,
      "step": 3650
    },
    {
      "epoch": 0.07304,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.013899739583333333,
      "learning_rate": 0.0001,
      "loss": 4.6588,
      "loss/crossentropy": 2.2270851135253906,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26908691227436066,
      "step": 3652
    },
    {
      "epoch": 0.07308,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.017039998372395834,
      "learning_rate": 0.0001,
      "loss": 4.7385,
      "loss/crossentropy": 2.2684017419815063,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28442947566509247,
      "step": 3654
    },
    {
      "epoch": 0.07312,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.07785542805989583,
      "learning_rate": 0.0001,
      "loss": 4.7585,
      "loss/crossentropy": 2.0922030806541443,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2621888816356659,
      "step": 3656
    },
    {
      "epoch": 0.07316,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.08323567708333333,
      "learning_rate": 0.0001,
      "loss": 4.7425,
      "loss/crossentropy": 2.0134947896003723,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2717447876930237,
      "step": 3658
    },
    {
      "epoch": 0.0732,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.08206278483072917,
      "learning_rate": 0.0001,
      "loss": 4.5573,
      "loss/crossentropy": 1.9246947765350342,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25095127522945404,
      "step": 3660
    },
    {
      "epoch": 0.07324,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.08561197916666667,
      "learning_rate": 0.0001,
      "loss": 4.8616,
      "loss/crossentropy": 2.0655113458633423,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24166538566350937,
      "step": 3662
    },
    {
      "epoch": 0.07328,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.08198954264322916,
      "learning_rate": 0.0001,
      "loss": 4.9137,
      "loss/crossentropy": 2.2706735730171204,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26484307646751404,
      "step": 3664
    },
    {
      "epoch": 0.07332,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.0883453369140625,
      "learning_rate": 0.0001,
      "loss": 4.2911,
      "loss/crossentropy": 1.7969809770584106,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23407897353172302,
      "step": 3666
    },
    {
      "epoch": 0.07336,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.09109700520833333,
      "learning_rate": 0.0001,
      "loss": 4.7081,
      "loss/crossentropy": 2.0398870706558228,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2502745985984802,
      "step": 3668
    },
    {
      "epoch": 0.0734,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.09381103515625,
      "learning_rate": 0.0001,
      "loss": 4.8738,
      "loss/crossentropy": 2.090283453464508,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2936585247516632,
      "step": 3670
    },
    {
      "epoch": 0.07344,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.035008748372395836,
      "learning_rate": 0.0001,
      "loss": 4.8305,
      "loss/crossentropy": 2.286925792694092,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27951307594776154,
      "step": 3672
    },
    {
      "epoch": 0.07348,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.032059733072916666,
      "learning_rate": 0.0001,
      "loss": 4.4885,
      "loss/crossentropy": 2.0264610052108765,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24908316135406494,
      "step": 3674
    },
    {
      "epoch": 0.07352,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.0372467041015625,
      "learning_rate": 0.0001,
      "loss": 4.9239,
      "loss/crossentropy": 2.1947755217552185,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2865990996360779,
      "step": 3676
    },
    {
      "epoch": 0.07356,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.0422515869140625,
      "learning_rate": 0.0001,
      "loss": 4.8784,
      "loss/crossentropy": 2.0050416588783264,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23362033069133759,
      "step": 3678
    },
    {
      "epoch": 0.0736,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.041112263997395836,
      "learning_rate": 0.0001,
      "loss": 4.7423,
      "loss/crossentropy": 1.8935424089431763,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2532464489340782,
      "step": 3680
    },
    {
      "epoch": 0.07364,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.0350250244140625,
      "learning_rate": 0.0001,
      "loss": 4.7389,
      "loss/crossentropy": 2.0181053280830383,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2544455900788307,
      "step": 3682
    },
    {
      "epoch": 0.07368,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.03144124348958333,
      "learning_rate": 0.0001,
      "loss": 4.7099,
      "loss/crossentropy": 2.1172796487808228,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27609144151210785,
      "step": 3684
    },
    {
      "epoch": 0.07372,
      "grad_norm": 2.375,
      "grad_norm_var": 0.023737589518229168,
      "learning_rate": 0.0001,
      "loss": 4.7185,
      "loss/crossentropy": 2.3926355838775635,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28892992436885834,
      "step": 3686
    },
    {
      "epoch": 0.07376,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.026883951822916665,
      "learning_rate": 0.0001,
      "loss": 4.9086,
      "loss/crossentropy": 2.2512835264205933,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26166096329689026,
      "step": 3688
    },
    {
      "epoch": 0.0738,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.0264312744140625,
      "learning_rate": 0.0001,
      "loss": 4.6311,
      "loss/crossentropy": 2.0656538009643555,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25399819016456604,
      "step": 3690
    },
    {
      "epoch": 0.07384,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.0222808837890625,
      "learning_rate": 0.0001,
      "loss": 4.9565,
      "loss/crossentropy": 2.454928994178772,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2964669317007065,
      "step": 3692
    },
    {
      "epoch": 0.07388,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.009859212239583333,
      "learning_rate": 0.0001,
      "loss": 4.5703,
      "loss/crossentropy": 1.988040804862976,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27345800399780273,
      "step": 3694
    },
    {
      "epoch": 0.07392,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.013752237955729166,
      "learning_rate": 0.0001,
      "loss": 4.9418,
      "loss/crossentropy": 1.910742998123169,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24037255346775055,
      "step": 3696
    },
    {
      "epoch": 0.07396,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.013700358072916667,
      "learning_rate": 0.0001,
      "loss": 4.6541,
      "loss/crossentropy": 2.19545578956604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2778017520904541,
      "step": 3698
    },
    {
      "epoch": 0.074,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.0127838134765625,
      "learning_rate": 0.0001,
      "loss": 4.3839,
      "loss/crossentropy": 2.436691403388977,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26254843175411224,
      "step": 3700
    },
    {
      "epoch": 0.07404,
      "grad_norm": 2.5,
      "grad_norm_var": 0.012398274739583333,
      "learning_rate": 0.0001,
      "loss": 4.833,
      "loss/crossentropy": 2.7458308935165405,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27952516078948975,
      "step": 3702
    },
    {
      "epoch": 0.07408,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.014876302083333333,
      "learning_rate": 0.0001,
      "loss": 4.8308,
      "loss/crossentropy": 2.2321633100509644,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2633504122495651,
      "step": 3704
    },
    {
      "epoch": 0.07412,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.013801066080729167,
      "learning_rate": 0.0001,
      "loss": 4.8159,
      "loss/crossentropy": 1.9883576035499573,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25319087505340576,
      "step": 3706
    },
    {
      "epoch": 0.07416,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.0147857666015625,
      "learning_rate": 0.0001,
      "loss": 4.5546,
      "loss/crossentropy": 1.7647870182991028,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21732009947299957,
      "step": 3708
    },
    {
      "epoch": 0.0742,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.046793619791666664,
      "learning_rate": 0.0001,
      "loss": 4.9271,
      "loss/crossentropy": 2.1113381385803223,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25888970494270325,
      "step": 3710
    },
    {
      "epoch": 0.07424,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.04690348307291667,
      "learning_rate": 0.0001,
      "loss": 4.5878,
      "loss/crossentropy": 1.975549578666687,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23899925500154495,
      "step": 3712
    },
    {
      "epoch": 0.07428,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.04664306640625,
      "learning_rate": 0.0001,
      "loss": 4.9262,
      "loss/crossentropy": 2.0562495589256287,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31428879499435425,
      "step": 3714
    },
    {
      "epoch": 0.07432,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.04951883951822917,
      "learning_rate": 0.0001,
      "loss": 4.3719,
      "loss/crossentropy": 2.114805221557617,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25172004848718643,
      "step": 3716
    },
    {
      "epoch": 0.07436,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.052783203125,
      "learning_rate": 0.0001,
      "loss": 4.6032,
      "loss/crossentropy": 2.1865739822387695,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.252632200717926,
      "step": 3718
    },
    {
      "epoch": 0.0744,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.04739176432291667,
      "learning_rate": 0.0001,
      "loss": 4.8493,
      "loss/crossentropy": 2.2550876140594482,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27425335347652435,
      "step": 3720
    },
    {
      "epoch": 0.07444,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.047379557291666666,
      "learning_rate": 0.0001,
      "loss": 4.9072,
      "loss/crossentropy": 2.293414354324341,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26979324221611023,
      "step": 3722
    },
    {
      "epoch": 0.07448,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.043745930989583334,
      "learning_rate": 0.0001,
      "loss": 4.4962,
      "loss/crossentropy": 2.014510452747345,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25709769129753113,
      "step": 3724
    },
    {
      "epoch": 0.07452,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.015746053059895834,
      "learning_rate": 0.0001,
      "loss": 4.6485,
      "loss/crossentropy": 2.0332603454589844,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2585446834564209,
      "step": 3726
    },
    {
      "epoch": 0.07456,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.0165191650390625,
      "learning_rate": 0.0001,
      "loss": 4.7268,
      "loss/crossentropy": 1.9425334930419922,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22921039909124374,
      "step": 3728
    },
    {
      "epoch": 0.0746,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.019090779622395835,
      "learning_rate": 0.0001,
      "loss": 4.9306,
      "loss/crossentropy": 2.1233898997306824,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26014500856399536,
      "step": 3730
    },
    {
      "epoch": 0.07464,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.023566691080729167,
      "learning_rate": 0.0001,
      "loss": 4.9958,
      "loss/crossentropy": 2.3929240703582764,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2791339308023453,
      "step": 3732
    },
    {
      "epoch": 0.07468,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.021370442708333333,
      "learning_rate": 0.0001,
      "loss": 4.6072,
      "loss/crossentropy": 2.163137674331665,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26793332397937775,
      "step": 3734
    },
    {
      "epoch": 0.07472,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.024738566080729166,
      "learning_rate": 0.0001,
      "loss": 4.723,
      "loss/crossentropy": 2.1300129294395447,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2533607929944992,
      "step": 3736
    },
    {
      "epoch": 0.07476,
      "grad_norm": 2.5,
      "grad_norm_var": 0.028055826822916668,
      "learning_rate": 0.0001,
      "loss": 4.7232,
      "loss/crossentropy": 1.9808942675590515,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25632843375205994,
      "step": 3738
    },
    {
      "epoch": 0.0748,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.03328348795572917,
      "learning_rate": 0.0001,
      "loss": 4.8219,
      "loss/crossentropy": 2.161437451839447,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2529585212469101,
      "step": 3740
    },
    {
      "epoch": 0.07484,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.03369140625,
      "learning_rate": 0.0001,
      "loss": 4.6541,
      "loss/crossentropy": 1.852737545967102,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2333993762731552,
      "step": 3742
    },
    {
      "epoch": 0.07488,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.03439127604166667,
      "learning_rate": 0.0001,
      "loss": 4.4355,
      "loss/crossentropy": 1.664733350276947,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20309723168611526,
      "step": 3744
    },
    {
      "epoch": 0.07492,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.03276265462239583,
      "learning_rate": 0.0001,
      "loss": 4.4554,
      "loss/crossentropy": 2.1815799474716187,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2470541000366211,
      "step": 3746
    },
    {
      "epoch": 0.07496,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.024657185872395834,
      "learning_rate": 0.0001,
      "loss": 4.7423,
      "loss/crossentropy": 1.9546562433242798,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2685271203517914,
      "step": 3748
    },
    {
      "epoch": 0.075,
      "grad_norm": 2.25,
      "grad_norm_var": 0.024738566080729166,
      "learning_rate": 0.0001,
      "loss": 4.5171,
      "loss/crossentropy": 1.920817255973816,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2558089941740036,
      "step": 3750
    },
    {
      "epoch": 0.07504,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.024982706705729166,
      "learning_rate": 0.0001,
      "loss": 4.4795,
      "loss/crossentropy": 1.7570490837097168,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2265816479921341,
      "step": 3752
    },
    {
      "epoch": 0.07508,
      "grad_norm": 2.890625,
      "grad_norm_var": 0.203564453125,
      "learning_rate": 0.0001,
      "loss": 4.756,
      "loss/crossentropy": 2.1815105676651,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2536233216524124,
      "step": 3754
    },
    {
      "epoch": 0.07512,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.19650777180989584,
      "learning_rate": 0.0001,
      "loss": 4.7178,
      "loss/crossentropy": 2.1385504603385925,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2445325404405594,
      "step": 3756
    },
    {
      "epoch": 0.07516,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.19650777180989584,
      "learning_rate": 0.0001,
      "loss": 4.6449,
      "loss/crossentropy": 1.7325092554092407,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2419673353433609,
      "step": 3758
    },
    {
      "epoch": 0.0752,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.19572652180989583,
      "learning_rate": 0.0001,
      "loss": 4.6096,
      "loss/crossentropy": 2.358627676963806,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2713842839002609,
      "step": 3760
    },
    {
      "epoch": 0.07524,
      "grad_norm": 3.078125,
      "grad_norm_var": 0.20392252604166666,
      "learning_rate": 0.0001,
      "loss": 5.1712,
      "loss/crossentropy": 2.048672080039978,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3007172644138336,
      "step": 3762
    },
    {
      "epoch": 0.07528,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.19885660807291666,
      "learning_rate": 0.0001,
      "loss": 4.8473,
      "loss/crossentropy": 2.2967183589935303,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28548331558704376,
      "step": 3764
    },
    {
      "epoch": 0.07532,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.19228108723958334,
      "learning_rate": 0.0001,
      "loss": 4.7541,
      "loss/crossentropy": 2.1280438899993896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2825516611337662,
      "step": 3766
    },
    {
      "epoch": 0.07536,
      "grad_norm": 2.375,
      "grad_norm_var": 0.19146728515625,
      "learning_rate": 0.0001,
      "loss": 4.8404,
      "loss/crossentropy": 2.5528002977371216,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28681764006614685,
      "step": 3768
    },
    {
      "epoch": 0.0754,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.0549468994140625,
      "learning_rate": 0.0001,
      "loss": 4.729,
      "loss/crossentropy": 2.235885262489319,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.259520560503006,
      "step": 3770
    },
    {
      "epoch": 0.07544,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.05718994140625,
      "learning_rate": 0.0001,
      "loss": 4.4705,
      "loss/crossentropy": 1.8836966753005981,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2360890954732895,
      "step": 3772
    },
    {
      "epoch": 0.07548,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.056538899739583336,
      "learning_rate": 0.0001,
      "loss": 4.9291,
      "loss/crossentropy": 2.3396376371383667,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30392636358737946,
      "step": 3774
    },
    {
      "epoch": 0.07552,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.051301066080729166,
      "learning_rate": 0.0001,
      "loss": 4.7518,
      "loss/crossentropy": 2.4024877548217773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2858506590127945,
      "step": 3776
    },
    {
      "epoch": 0.07556,
      "grad_norm": 2.25,
      "grad_norm_var": 0.0353515625,
      "learning_rate": 0.0001,
      "loss": 4.4073,
      "loss/crossentropy": 2.138229727745056,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25458595901727676,
      "step": 3778
    },
    {
      "epoch": 0.0756,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.0359375,
      "learning_rate": 0.0001,
      "loss": 4.3882,
      "loss/crossentropy": 1.8413254618644714,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23022352159023285,
      "step": 3780
    },
    {
      "epoch": 0.07564,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.034764607747395836,
      "learning_rate": 0.0001,
      "loss": 4.613,
      "loss/crossentropy": 1.8554572463035583,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25644390285015106,
      "step": 3782
    },
    {
      "epoch": 0.07568,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.018452962239583332,
      "learning_rate": 0.0001,
      "loss": 4.7013,
      "loss/crossentropy": 2.0096731781959534,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25370142608880997,
      "step": 3784
    },
    {
      "epoch": 0.07572,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.020018513997395834,
      "learning_rate": 0.0001,
      "loss": 4.9317,
      "loss/crossentropy": 1.7932413220405579,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2426912784576416,
      "step": 3786
    },
    {
      "epoch": 0.07576,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0263336181640625,
      "learning_rate": 0.0001,
      "loss": 4.1599,
      "loss/crossentropy": 2.0372042655944824,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24203844368457794,
      "step": 3788
    },
    {
      "epoch": 0.0758,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.021773274739583334,
      "learning_rate": 0.0001,
      "loss": 4.3627,
      "loss/crossentropy": 1.8986076712608337,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24614746868610382,
      "step": 3790
    },
    {
      "epoch": 0.07584,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.022135416666666668,
      "learning_rate": 0.0001,
      "loss": 4.563,
      "loss/crossentropy": 1.8080393075942993,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23206676542758942,
      "step": 3792
    },
    {
      "epoch": 0.07588,
      "grad_norm": 2.25,
      "grad_norm_var": 0.015067545572916667,
      "learning_rate": 0.0001,
      "loss": 4.6857,
      "loss/crossentropy": 1.7578041553497314,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21686340868473053,
      "step": 3794
    },
    {
      "epoch": 0.07592,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.015087890625,
      "learning_rate": 0.0001,
      "loss": 4.4938,
      "loss/crossentropy": 2.0115376710891724,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2651440501213074,
      "step": 3796
    },
    {
      "epoch": 0.07596,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.015803019205729168,
      "learning_rate": 0.0001,
      "loss": 4.598,
      "loss/crossentropy": 2.028555393218994,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26161982119083405,
      "step": 3798
    },
    {
      "epoch": 0.076,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.016429646809895834,
      "learning_rate": 0.0001,
      "loss": 4.8315,
      "loss/crossentropy": 2.158663272857666,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28110067546367645,
      "step": 3800
    },
    {
      "epoch": 0.07604,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.019652303059895834,
      "learning_rate": 0.0001,
      "loss": 5.2376,
      "loss/crossentropy": 2.2959556579589844,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28477419912815094,
      "step": 3802
    },
    {
      "epoch": 0.07608,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.014997355143229167,
      "learning_rate": 0.0001,
      "loss": 4.5289,
      "loss/crossentropy": 2.149766206741333,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25216003507375717,
      "step": 3804
    },
    {
      "epoch": 0.07612,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.01416015625,
      "learning_rate": 0.0001,
      "loss": 4.7496,
      "loss/crossentropy": 1.9866302609443665,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26193149387836456,
      "step": 3806
    },
    {
      "epoch": 0.07616,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.0122955322265625,
      "learning_rate": 0.0001,
      "loss": 4.9968,
      "loss/crossentropy": 2.4230403900146484,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27897247672080994,
      "step": 3808
    },
    {
      "epoch": 0.0762,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.012613932291666666,
      "learning_rate": 0.0001,
      "loss": 4.9133,
      "loss/crossentropy": 2.2995522022247314,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27623192965984344,
      "step": 3810
    },
    {
      "epoch": 0.07624,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.013923136393229167,
      "learning_rate": 0.0001,
      "loss": 4.6632,
      "loss/crossentropy": 2.167468547821045,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26121728122234344,
      "step": 3812
    },
    {
      "epoch": 0.07628,
      "grad_norm": 2.5,
      "grad_norm_var": 0.013353474934895833,
      "learning_rate": 0.0001,
      "loss": 4.8435,
      "loss/crossentropy": 2.3259944915771484,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29026439785957336,
      "step": 3814
    },
    {
      "epoch": 0.07632,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.041825358072916666,
      "learning_rate": 0.0001,
      "loss": 4.8704,
      "loss/crossentropy": 2.18080472946167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2550061345100403,
      "step": 3816
    },
    {
      "epoch": 0.07636,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.03871968587239583,
      "learning_rate": 0.0001,
      "loss": 4.5681,
      "loss/crossentropy": 2.1185330748558044,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25014883279800415,
      "step": 3818
    },
    {
      "epoch": 0.0764,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.039383951822916666,
      "learning_rate": 0.0001,
      "loss": 4.5776,
      "loss/crossentropy": 1.9028193354606628,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22502654790878296,
      "step": 3820
    },
    {
      "epoch": 0.07644,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.039713541666666664,
      "learning_rate": 0.0001,
      "loss": 5.0188,
      "loss/crossentropy": 2.266402840614319,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24728389829397202,
      "step": 3822
    },
    {
      "epoch": 0.07648,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.04279683430989583,
      "learning_rate": 0.0001,
      "loss": 4.7036,
      "loss/crossentropy": 2.0918440222740173,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2669295519590378,
      "step": 3824
    },
    {
      "epoch": 0.07652,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.04456380208333333,
      "learning_rate": 0.0001,
      "loss": 4.1446,
      "loss/crossentropy": 1.6120481491088867,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21910656243562698,
      "step": 3826
    },
    {
      "epoch": 0.07656,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.044331868489583336,
      "learning_rate": 0.0001,
      "loss": 4.6352,
      "loss/crossentropy": 2.2294809818267822,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2720891535282135,
      "step": 3828
    },
    {
      "epoch": 0.0766,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.0524322509765625,
      "learning_rate": 0.0001,
      "loss": 4.873,
      "loss/crossentropy": 2.2588730454444885,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2613115608692169,
      "step": 3830
    },
    {
      "epoch": 0.07664,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.02584228515625,
      "learning_rate": 0.0001,
      "loss": 4.5881,
      "loss/crossentropy": 2.2658292055130005,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2729395925998688,
      "step": 3832
    },
    {
      "epoch": 0.07668,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.02662353515625,
      "learning_rate": 0.0001,
      "loss": 5.1787,
      "loss/crossentropy": 2.314574718475342,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28539060056209564,
      "step": 3834
    },
    {
      "epoch": 0.07672,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.024738566080729166,
      "learning_rate": 0.0001,
      "loss": 5.0388,
      "loss/crossentropy": 2.4684417247772217,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29843954741954803,
      "step": 3836
    },
    {
      "epoch": 0.07676,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.025944010416666666,
      "learning_rate": 0.0001,
      "loss": 4.5976,
      "loss/crossentropy": 2.528733253479004,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2696636766195297,
      "step": 3838
    },
    {
      "epoch": 0.0768,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.025340779622395834,
      "learning_rate": 0.0001,
      "loss": 4.6449,
      "loss/crossentropy": 2.203901529312134,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26598773896694183,
      "step": 3840
    },
    {
      "epoch": 0.07684,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.021907552083333334,
      "learning_rate": 0.0001,
      "loss": 4.7322,
      "loss/crossentropy": 1.9192892909049988,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25631098449230194,
      "step": 3842
    },
    {
      "epoch": 0.07688,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.022077433268229165,
      "learning_rate": 0.0001,
      "loss": 4.6372,
      "loss/crossentropy": 1.8314838409423828,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22842589765787125,
      "step": 3844
    },
    {
      "epoch": 0.07692,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.013395182291666667,
      "learning_rate": 0.0001,
      "loss": 4.6023,
      "loss/crossentropy": 2.2416744232177734,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24890189617872238,
      "step": 3846
    },
    {
      "epoch": 0.07696,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.012580362955729167,
      "learning_rate": 0.0001,
      "loss": 4.8454,
      "loss/crossentropy": 2.034749209880829,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29580502212047577,
      "step": 3848
    },
    {
      "epoch": 0.077,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.0117095947265625,
      "learning_rate": 0.0001,
      "loss": 4.5923,
      "loss/crossentropy": 1.9982805848121643,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2618003487586975,
      "step": 3850
    },
    {
      "epoch": 0.07704,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.0127105712890625,
      "learning_rate": 0.0001,
      "loss": 4.7704,
      "loss/crossentropy": 2.065816104412079,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2571987137198448,
      "step": 3852
    },
    {
      "epoch": 0.07708,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.0100982666015625,
      "learning_rate": 0.0001,
      "loss": 4.7493,
      "loss/crossentropy": 1.933334231376648,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24824900180101395,
      "step": 3854
    },
    {
      "epoch": 0.07712,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.010514322916666667,
      "learning_rate": 0.0001,
      "loss": 4.5805,
      "loss/crossentropy": 1.9197405576705933,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2559673935174942,
      "step": 3856
    },
    {
      "epoch": 0.07716,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.010252888997395833,
      "learning_rate": 0.0001,
      "loss": 4.429,
      "loss/crossentropy": 2.307250142097473,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2613677680492401,
      "step": 3858
    },
    {
      "epoch": 0.0772,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.01011962890625,
      "learning_rate": 0.0001,
      "loss": 4.4494,
      "loss/crossentropy": 2.1120635271072388,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24830741435289383,
      "step": 3860
    },
    {
      "epoch": 0.07724,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.039013671875,
      "learning_rate": 0.0001,
      "loss": 4.8585,
      "loss/crossentropy": 2.404169201850891,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2960711419582367,
      "step": 3862
    },
    {
      "epoch": 0.07728,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.038899739583333336,
      "learning_rate": 0.0001,
      "loss": 5.0257,
      "loss/crossentropy": 2.2490307688713074,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2569812461733818,
      "step": 3864
    },
    {
      "epoch": 0.07732,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.0386871337890625,
      "learning_rate": 0.0001,
      "loss": 4.9086,
      "loss/crossentropy": 2.0773178339004517,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2505042105913162,
      "step": 3866
    },
    {
      "epoch": 0.07736,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.038899739583333336,
      "learning_rate": 0.0001,
      "loss": 4.4621,
      "loss/crossentropy": 1.83626389503479,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22876176983118057,
      "step": 3868
    },
    {
      "epoch": 0.0774,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.03911031087239583,
      "learning_rate": 0.0001,
      "loss": 4.5298,
      "loss/crossentropy": 1.8159971833229065,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22787011414766312,
      "step": 3870
    },
    {
      "epoch": 0.07744,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.0378326416015625,
      "learning_rate": 0.0001,
      "loss": 4.5995,
      "loss/crossentropy": 2.0361026525497437,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23802263289690018,
      "step": 3872
    },
    {
      "epoch": 0.07748,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.04433492024739583,
      "learning_rate": 0.0001,
      "loss": 4.9506,
      "loss/crossentropy": 2.2464375495910645,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26598919928073883,
      "step": 3874
    },
    {
      "epoch": 0.07752,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.0408355712890625,
      "learning_rate": 0.0001,
      "loss": 4.9416,
      "loss/crossentropy": 2.163287401199341,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29068198800086975,
      "step": 3876
    },
    {
      "epoch": 0.07756,
      "grad_norm": 2.5,
      "grad_norm_var": 0.22014567057291667,
      "learning_rate": 0.0001,
      "loss": 4.9478,
      "loss/crossentropy": 2.1638875007629395,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2609352171421051,
      "step": 3878
    },
    {
      "epoch": 0.0776,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.21923421223958334,
      "learning_rate": 0.0001,
      "loss": 4.6247,
      "loss/crossentropy": 1.9216270446777344,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25062238425016403,
      "step": 3880
    },
    {
      "epoch": 0.07764,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.21585184733072918,
      "learning_rate": 0.0001,
      "loss": 5.0786,
      "loss/crossentropy": 2.036958694458008,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2897229939699173,
      "step": 3882
    },
    {
      "epoch": 0.07768,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.2094635009765625,
      "learning_rate": 0.0001,
      "loss": 4.6062,
      "loss/crossentropy": 2.1493492126464844,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.258526012301445,
      "step": 3884
    },
    {
      "epoch": 0.07772,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.20414937337239583,
      "learning_rate": 0.0001,
      "loss": 4.468,
      "loss/crossentropy": 2.0496288537979126,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2386137992143631,
      "step": 3886
    },
    {
      "epoch": 0.07776,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.20829671223958332,
      "learning_rate": 0.0001,
      "loss": 4.2402,
      "loss/crossentropy": 1.5763422846794128,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2227308303117752,
      "step": 3888
    },
    {
      "epoch": 0.0778,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.20852864583333333,
      "learning_rate": 0.0001,
      "loss": 5.0582,
      "loss/crossentropy": 2.5032416582107544,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3047761619091034,
      "step": 3890
    },
    {
      "epoch": 0.07784,
      "grad_norm": 2.375,
      "grad_norm_var": 0.21199544270833334,
      "learning_rate": 0.0001,
      "loss": 4.7694,
      "loss/crossentropy": 2.3609601259231567,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27848154306411743,
      "step": 3892
    },
    {
      "epoch": 0.07788,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.0202056884765625,
      "learning_rate": 0.0001,
      "loss": 4.776,
      "loss/crossentropy": 2.188078999519348,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2570301741361618,
      "step": 3894
    },
    {
      "epoch": 0.07792,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.018733723958333334,
      "learning_rate": 0.0001,
      "loss": 4.8374,
      "loss/crossentropy": 1.9860637784004211,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25376833230257034,
      "step": 3896
    },
    {
      "epoch": 0.07796,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.010399373372395833,
      "learning_rate": 0.0001,
      "loss": 4.404,
      "loss/crossentropy": 2.0886037945747375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24766233563423157,
      "step": 3898
    },
    {
      "epoch": 0.078,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.009845987955729166,
      "learning_rate": 0.0001,
      "loss": 4.6833,
      "loss/crossentropy": 2.373010039329529,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27916407585144043,
      "step": 3900
    },
    {
      "epoch": 0.07804,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.009837849934895834,
      "learning_rate": 0.0001,
      "loss": 4.6295,
      "loss/crossentropy": 1.6733890771865845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2211885154247284,
      "step": 3902
    },
    {
      "epoch": 0.07808,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.010184733072916667,
      "learning_rate": 0.0001,
      "loss": 4.6588,
      "loss/crossentropy": 2.0506675243377686,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27091431617736816,
      "step": 3904
    },
    {
      "epoch": 0.07812,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.008784993489583334,
      "learning_rate": 0.0001,
      "loss": 4.712,
      "loss/crossentropy": 2.3200724124908447,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26983049511909485,
      "step": 3906
    },
    {
      "epoch": 0.07816,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.0164459228515625,
      "learning_rate": 0.0001,
      "loss": 4.7171,
      "loss/crossentropy": 1.928814709186554,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2423655241727829,
      "step": 3908
    },
    {
      "epoch": 0.0782,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.016389973958333335,
      "learning_rate": 0.0001,
      "loss": 4.6944,
      "loss/crossentropy": 2.007555842399597,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2444767728447914,
      "step": 3910
    },
    {
      "epoch": 0.07824,
      "grad_norm": 2.875,
      "grad_norm_var": 0.027469889322916666,
      "learning_rate": 0.0001,
      "loss": 4.7955,
      "loss/crossentropy": 2.2054057121276855,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2849784791469574,
      "step": 3912
    },
    {
      "epoch": 0.07828,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.03986002604166667,
      "learning_rate": 0.0001,
      "loss": 4.6885,
      "loss/crossentropy": 2.331532597541809,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2820790112018585,
      "step": 3914
    },
    {
      "epoch": 0.07832,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.036519368489583336,
      "learning_rate": 0.0001,
      "loss": 4.6967,
      "loss/crossentropy": 2.142041563987732,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25799722969532013,
      "step": 3916
    },
    {
      "epoch": 0.07836,
      "grad_norm": 2.625,
      "grad_norm_var": 0.0394683837890625,
      "learning_rate": 0.0001,
      "loss": 4.5415,
      "loss/crossentropy": 2.010735809803009,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24202881753444672,
      "step": 3918
    },
    {
      "epoch": 0.0784,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.03970947265625,
      "learning_rate": 0.0001,
      "loss": 5.037,
      "loss/crossentropy": 2.382808804512024,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27038049697875977,
      "step": 3920
    },
    {
      "epoch": 0.07844,
      "grad_norm": 2.375,
      "grad_norm_var": 0.0372955322265625,
      "learning_rate": 0.0001,
      "loss": 4.7708,
      "loss/crossentropy": 2.099658191204071,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2557126358151436,
      "step": 3922
    },
    {
      "epoch": 0.07848,
      "grad_norm": 2.375,
      "grad_norm_var": 0.03968098958333333,
      "learning_rate": 0.0001,
      "loss": 4.3775,
      "loss/crossentropy": 1.7840275764465332,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22102414071559906,
      "step": 3924
    },
    {
      "epoch": 0.07852,
      "grad_norm": 2.5,
      "grad_norm_var": 0.039159138997395836,
      "learning_rate": 0.0001,
      "loss": 4.6637,
      "loss/crossentropy": 1.8730725049972534,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23219943791627884,
      "step": 3926
    },
    {
      "epoch": 0.07856,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.034520467122395836,
      "learning_rate": 0.0001,
      "loss": 4.4432,
      "loss/crossentropy": 1.9218623638153076,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23333143442869186,
      "step": 3928
    },
    {
      "epoch": 0.0786,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.0152740478515625,
      "learning_rate": 0.0001,
      "loss": 4.8944,
      "loss/crossentropy": 1.9885727763175964,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2596626430749893,
      "step": 3930
    },
    {
      "epoch": 0.07864,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.014989217122395834,
      "learning_rate": 0.0001,
      "loss": 4.9358,
      "loss/crossentropy": 2.397018015384674,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2790713906288147,
      "step": 3932
    },
    {
      "epoch": 0.07868,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.014383951822916666,
      "learning_rate": 0.0001,
      "loss": 4.3062,
      "loss/crossentropy": 1.7345170378684998,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23668452352285385,
      "step": 3934
    },
    {
      "epoch": 0.07872,
      "grad_norm": 2.25,
      "grad_norm_var": 0.013309733072916666,
      "learning_rate": 0.0001,
      "loss": 4.6173,
      "loss/crossentropy": 1.8630162477493286,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2371346428990364,
      "step": 3936
    },
    {
      "epoch": 0.07876,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.010367838541666667,
      "learning_rate": 0.0001,
      "loss": 4.5774,
      "loss/crossentropy": 2.0738128423690796,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24340355396270752,
      "step": 3938
    },
    {
      "epoch": 0.0788,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.010835774739583333,
      "learning_rate": 0.0001,
      "loss": 5.0027,
      "loss/crossentropy": 2.2932467460632324,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2527567446231842,
      "step": 3940
    },
    {
      "epoch": 0.07884,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.009956868489583333,
      "learning_rate": 0.0001,
      "loss": 4.6785,
      "loss/crossentropy": 2.0000113248825073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23461253196001053,
      "step": 3942
    },
    {
      "epoch": 0.07888,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.008430989583333333,
      "learning_rate": 0.0001,
      "loss": 4.7307,
      "loss/crossentropy": 2.0753955841064453,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2535083442926407,
      "step": 3944
    },
    {
      "epoch": 0.07892,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.008153279622395834,
      "learning_rate": 0.0001,
      "loss": 4.6906,
      "loss/crossentropy": 2.167261242866516,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24734552949666977,
      "step": 3946
    },
    {
      "epoch": 0.07896,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.0108795166015625,
      "learning_rate": 0.0001,
      "loss": 4.7347,
      "loss/crossentropy": 1.9755831956863403,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24861737340688705,
      "step": 3948
    },
    {
      "epoch": 0.079,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.009261067708333333,
      "learning_rate": 0.0001,
      "loss": 4.6813,
      "loss/crossentropy": 2.195417881011963,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2518697530031204,
      "step": 3950
    },
    {
      "epoch": 0.07904,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.01920166015625,
      "learning_rate": 0.0001,
      "loss": 5.0172,
      "loss/crossentropy": 2.5771371126174927,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28423887491226196,
      "step": 3952
    },
    {
      "epoch": 0.07908,
      "grad_norm": 2.375,
      "grad_norm_var": 0.017870076497395835,
      "learning_rate": 0.0001,
      "loss": 4.7071,
      "loss/crossentropy": 1.683276355266571,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2075405865907669,
      "step": 3954
    },
    {
      "epoch": 0.07912,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.018993123372395834,
      "learning_rate": 0.0001,
      "loss": 4.7128,
      "loss/crossentropy": 2.2983756065368652,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28010235726833344,
      "step": 3956
    },
    {
      "epoch": 0.07916,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.020361328125,
      "learning_rate": 0.0001,
      "loss": 4.7896,
      "loss/crossentropy": 2.3263272047042847,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28625819087028503,
      "step": 3958
    },
    {
      "epoch": 0.0792,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.0204742431640625,
      "learning_rate": 0.0001,
      "loss": 4.5201,
      "loss/crossentropy": 1.9820871353149414,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2594129145145416,
      "step": 3960
    },
    {
      "epoch": 0.07924,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.025634765625,
      "learning_rate": 0.0001,
      "loss": 4.4754,
      "loss/crossentropy": 1.8991515636444092,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23694587498903275,
      "step": 3962
    },
    {
      "epoch": 0.07928,
      "grad_norm": 2.921875,
      "grad_norm_var": 0.03843994140625,
      "learning_rate": 0.0001,
      "loss": 4.9865,
      "loss/crossentropy": 2.485508918762207,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2957809865474701,
      "step": 3964
    },
    {
      "epoch": 0.07932,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.037262980143229166,
      "learning_rate": 0.0001,
      "loss": 4.8871,
      "loss/crossentropy": 2.156081974506378,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27474651485681534,
      "step": 3966
    },
    {
      "epoch": 0.07936,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.028544108072916668,
      "learning_rate": 0.0001,
      "loss": 4.8694,
      "loss/crossentropy": 2.0370571613311768,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2732074484229088,
      "step": 3968
    },
    {
      "epoch": 0.0794,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.028251139322916667,
      "learning_rate": 0.0001,
      "loss": 5.062,
      "loss/crossentropy": 2.3039989471435547,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31054411828517914,
      "step": 3970
    },
    {
      "epoch": 0.07944,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.028641764322916666,
      "learning_rate": 0.0001,
      "loss": 4.7875,
      "loss/crossentropy": 2.2280107736587524,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2711277902126312,
      "step": 3972
    },
    {
      "epoch": 0.07948,
      "grad_norm": 2.625,
      "grad_norm_var": 0.02935791015625,
      "learning_rate": 0.0001,
      "loss": 4.8992,
      "loss/crossentropy": 2.0609869956970215,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24142058193683624,
      "step": 3974
    },
    {
      "epoch": 0.07952,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.03277587890625,
      "learning_rate": 0.0001,
      "loss": 4.1391,
      "loss/crossentropy": 2.0541876554489136,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24537865817546844,
      "step": 3976
    },
    {
      "epoch": 0.07956,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.6102701822916666,
      "learning_rate": 0.0001,
      "loss": 4.9591,
      "loss/crossentropy": 2.344236969947815,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2770863175392151,
      "step": 3978
    },
    {
      "epoch": 0.0796,
      "grad_norm": 2.625,
      "grad_norm_var": 0.65465087890625,
      "learning_rate": 0.0001,
      "loss": 4.7232,
      "loss/crossentropy": 1.7899338603019714,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23951984196901321,
      "step": 3980
    },
    {
      "epoch": 0.07964,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.6591471354166667,
      "learning_rate": 0.0001,
      "loss": 4.6771,
      "loss/crossentropy": 2.3253976106643677,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2691802680492401,
      "step": 3982
    },
    {
      "epoch": 0.07968,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.65885009765625,
      "learning_rate": 0.0001,
      "loss": 4.8223,
      "loss/crossentropy": 2.14141583442688,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2603989467024803,
      "step": 3984
    },
    {
      "epoch": 0.07972,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.6512196858723959,
      "learning_rate": 0.0001,
      "loss": 4.9059,
      "loss/crossentropy": 2.262465476989746,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.291474387049675,
      "step": 3986
    },
    {
      "epoch": 0.07976,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.66064453125,
      "learning_rate": 0.0001,
      "loss": 4.5626,
      "loss/crossentropy": 2.1835561990737915,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23615659773349762,
      "step": 3988
    },
    {
      "epoch": 0.0798,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.6716145833333333,
      "learning_rate": 0.0001,
      "loss": 4.8482,
      "loss/crossentropy": 2.020140767097473,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2593151703476906,
      "step": 3990
    },
    {
      "epoch": 0.07984,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.6518513997395833,
      "learning_rate": 0.0001,
      "loss": 4.7986,
      "loss/crossentropy": 2.277661681175232,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2721617519855499,
      "step": 3992
    },
    {
      "epoch": 0.07988,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.09846089680989584,
      "learning_rate": 0.0001,
      "loss": 4.5113,
      "loss/crossentropy": 2.1883193254470825,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2610222101211548,
      "step": 3994
    },
    {
      "epoch": 0.07992,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.01754150390625,
      "learning_rate": 0.0001,
      "loss": 4.5987,
      "loss/crossentropy": 2.152850031852722,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27418845891952515,
      "step": 3996
    },
    {
      "epoch": 0.07996,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.017513020833333334,
      "learning_rate": 0.0001,
      "loss": 4.8214,
      "loss/crossentropy": 2.3313716650009155,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27529503405094147,
      "step": 3998
    },
    {
      "epoch": 0.08,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.020319620768229168,
      "learning_rate": 0.0001,
      "loss": 5.086,
      "loss/crossentropy": 2.250498414039612,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26368021965026855,
      "step": 4000
    },
    {
      "epoch": 0.08004,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.011188761393229166,
      "learning_rate": 0.0001,
      "loss": 4.8005,
      "loss/crossentropy": 2.322459101676941,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2868216335773468,
      "step": 4002
    },
    {
      "epoch": 0.08008,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.0142730712890625,
      "learning_rate": 0.0001,
      "loss": 4.8693,
      "loss/crossentropy": 1.9340506792068481,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2316983863711357,
      "step": 4004
    },
    {
      "epoch": 0.08012,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.0150054931640625,
      "learning_rate": 0.0001,
      "loss": 4.7395,
      "loss/crossentropy": 1.8635645508766174,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22694773972034454,
      "step": 4006
    },
    {
      "epoch": 0.08016,
      "grad_norm": 10.375,
      "grad_norm_var": 3.9615631103515625,
      "learning_rate": 0.0001,
      "loss": 4.8916,
      "loss/crossentropy": 1.9252317547798157,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24893560260534286,
      "step": 4008
    },
    {
      "epoch": 0.0802,
      "grad_norm": 2.671875,
      "grad_norm_var": 3.9093424479166665,
      "learning_rate": 0.0001,
      "loss": 5.2636,
      "loss/crossentropy": 2.1964328289031982,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2567693591117859,
      "step": 4010
    },
    {
      "epoch": 0.08024,
      "grad_norm": 2.671875,
      "grad_norm_var": 3.8960113525390625,
      "learning_rate": 0.0001,
      "loss": 4.9054,
      "loss/crossentropy": 2.296012043952942,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27360107749700546,
      "step": 4012
    },
    {
      "epoch": 0.08028,
      "grad_norm": 2.453125,
      "grad_norm_var": 3.9009724934895833,
      "learning_rate": 0.0001,
      "loss": 4.8894,
      "loss/crossentropy": 2.360015869140625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27541905641555786,
      "step": 4014
    },
    {
      "epoch": 0.08032,
      "grad_norm": 2.390625,
      "grad_norm_var": 3.906591796875,
      "learning_rate": 0.0001,
      "loss": 4.8865,
      "loss/crossentropy": 2.36370050907135,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27881547808647156,
      "step": 4016
    },
    {
      "epoch": 0.08036,
      "grad_norm": 2.25,
      "grad_norm_var": 3.9068593343098956,
      "learning_rate": 0.0001,
      "loss": 4.6461,
      "loss/crossentropy": 1.8704780340194702,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2512069493532181,
      "step": 4018
    },
    {
      "epoch": 0.0804,
      "grad_norm": 2.4375,
      "grad_norm_var": 3.9156483968098956,
      "learning_rate": 0.0001,
      "loss": 4.733,
      "loss/crossentropy": 2.1989234685897827,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2755106985569,
      "step": 4020
    },
    {
      "epoch": 0.08044,
      "grad_norm": 2.21875,
      "grad_norm_var": 3.9420237223307293,
      "learning_rate": 0.0001,
      "loss": 4.3471,
      "loss/crossentropy": 1.9905433058738708,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2579897418618202,
      "step": 4022
    },
    {
      "epoch": 0.08048,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.08837788899739583,
      "learning_rate": 0.0001,
      "loss": 4.9025,
      "loss/crossentropy": 2.270000696182251,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2693728804588318,
      "step": 4024
    },
    {
      "epoch": 0.08052,
      "grad_norm": 2.375,
      "grad_norm_var": 0.08504130045572916,
      "learning_rate": 0.0001,
      "loss": 4.7569,
      "loss/crossentropy": 2.178301692008972,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2617819905281067,
      "step": 4026
    },
    {
      "epoch": 0.08056,
      "grad_norm": 2.25,
      "grad_norm_var": 0.08346354166666667,
      "learning_rate": 0.0001,
      "loss": 4.687,
      "loss/crossentropy": 2.518654465675354,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29282887279987335,
      "step": 4028
    },
    {
      "epoch": 0.0806,
      "grad_norm": 2.25,
      "grad_norm_var": 0.08859049479166667,
      "learning_rate": 0.0001,
      "loss": 4.4825,
      "loss/crossentropy": 1.9181422591209412,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23210199177265167,
      "step": 4030
    },
    {
      "epoch": 0.08064,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.08816630045572917,
      "learning_rate": 0.0001,
      "loss": 4.6407,
      "loss/crossentropy": 2.343222141265869,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28908614814281464,
      "step": 4032
    },
    {
      "epoch": 0.08068,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.0302886962890625,
      "learning_rate": 0.0001,
      "loss": 4.6879,
      "loss/crossentropy": 2.0816845893859863,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2571762502193451,
      "step": 4034
    },
    {
      "epoch": 0.08072,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.029878743489583335,
      "learning_rate": 0.0001,
      "loss": 4.3324,
      "loss/crossentropy": 1.9679544568061829,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2316160574555397,
      "step": 4036
    },
    {
      "epoch": 0.08076,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.029195149739583332,
      "learning_rate": 0.0001,
      "loss": 4.7335,
      "loss/crossentropy": 2.1553682684898376,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25725623965263367,
      "step": 4038
    },
    {
      "epoch": 0.0808,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.004964192708333333,
      "learning_rate": 0.0001,
      "loss": 4.6892,
      "loss/crossentropy": 2.0269790291786194,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26298412680625916,
      "step": 4040
    },
    {
      "epoch": 0.08084,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.0060943603515625,
      "learning_rate": 0.0001,
      "loss": 4.6686,
      "loss/crossentropy": 1.984773874282837,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24288517236709595,
      "step": 4042
    },
    {
      "epoch": 0.08088,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.0051910400390625,
      "learning_rate": 0.0001,
      "loss": 4.9282,
      "loss/crossentropy": 2.178356111049652,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2684163451194763,
      "step": 4044
    },
    {
      "epoch": 0.08092,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.0053670247395833336,
      "learning_rate": 0.0001,
      "loss": 4.8191,
      "loss/crossentropy": 2.235984683036804,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26377667486667633,
      "step": 4046
    },
    {
      "epoch": 0.08096,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.005826822916666667,
      "learning_rate": 0.0001,
      "loss": 4.7026,
      "loss/crossentropy": 2.085321545600891,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23403701931238174,
      "step": 4048
    },
    {
      "epoch": 0.081,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.0052642822265625,
      "learning_rate": 0.0001,
      "loss": 4.9932,
      "loss/crossentropy": 2.419228672981262,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27236658334732056,
      "step": 4050
    },
    {
      "epoch": 0.08104,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.0054972330729166664,
      "learning_rate": 0.0001,
      "loss": 4.6105,
      "loss/crossentropy": 2.153620958328247,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28071053326129913,
      "step": 4052
    },
    {
      "epoch": 0.08108,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.012743123372395833,
      "learning_rate": 0.0001,
      "loss": 4.8775,
      "loss/crossentropy": 2.1466477513313293,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27491800487041473,
      "step": 4054
    },
    {
      "epoch": 0.08112,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.013932291666666667,
      "learning_rate": 0.0001,
      "loss": 4.3707,
      "loss/crossentropy": 2.2020710706710815,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25149518996477127,
      "step": 4056
    },
    {
      "epoch": 0.08116,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.013623046875,
      "learning_rate": 0.0001,
      "loss": 4.8458,
      "loss/crossentropy": 2.264205574989319,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27612583339214325,
      "step": 4058
    },
    {
      "epoch": 0.0812,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.016434733072916666,
      "learning_rate": 0.0001,
      "loss": 4.8644,
      "loss/crossentropy": 2.269905209541321,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2692303955554962,
      "step": 4060
    },
    {
      "epoch": 0.08124,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.018505859375,
      "learning_rate": 0.0001,
      "loss": 4.5057,
      "loss/crossentropy": 1.920631766319275,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22550494968891144,
      "step": 4062
    },
    {
      "epoch": 0.08128,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.019749959309895832,
      "learning_rate": 0.0001,
      "loss": 5.0796,
      "loss/crossentropy": 2.307617664337158,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22719035297632217,
      "step": 4064
    },
    {
      "epoch": 0.08132,
      "grad_norm": 2.375,
      "grad_norm_var": 0.022102864583333333,
      "learning_rate": 0.0001,
      "loss": 4.6167,
      "loss/crossentropy": 2.113444685935974,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.240354023873806,
      "step": 4066
    },
    {
      "epoch": 0.08136,
      "grad_norm": 2.375,
      "grad_norm_var": 0.02232666015625,
      "learning_rate": 0.0001,
      "loss": 4.9152,
      "loss/crossentropy": 2.4516230821609497,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27590544521808624,
      "step": 4068
    },
    {
      "epoch": 0.0814,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.08772379557291667,
      "learning_rate": 0.0001,
      "loss": 4.5976,
      "loss/crossentropy": 1.8287339806556702,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22503511607646942,
      "step": 4070
    },
    {
      "epoch": 0.08144,
      "grad_norm": 2.5,
      "grad_norm_var": 0.08479715983072916,
      "learning_rate": 0.0001,
      "loss": 5.1623,
      "loss/crossentropy": 2.3468997478485107,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2773839682340622,
      "step": 4072
    },
    {
      "epoch": 0.08148,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.08782145182291666,
      "learning_rate": 0.0001,
      "loss": 4.5413,
      "loss/crossentropy": 2.1307512521743774,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24247504770755768,
      "step": 4074
    },
    {
      "epoch": 0.08152,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.08982645670572917,
      "learning_rate": 0.0001,
      "loss": 4.7447,
      "loss/crossentropy": 2.248755097389221,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2696859538555145,
      "step": 4076
    },
    {
      "epoch": 0.08156,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.09045817057291666,
      "learning_rate": 0.0001,
      "loss": 4.2948,
      "loss/crossentropy": 2.0233980417251587,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2370016872882843,
      "step": 4078
    },
    {
      "epoch": 0.0816,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.0923828125,
      "learning_rate": 0.0001,
      "loss": 4.432,
      "loss/crossentropy": 1.9536627531051636,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23242933303117752,
      "step": 4080
    },
    {
      "epoch": 0.08164,
      "grad_norm": 2.375,
      "grad_norm_var": 0.0889801025390625,
      "learning_rate": 0.0001,
      "loss": 4.5037,
      "loss/crossentropy": 1.9631904363632202,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2337196245789528,
      "step": 4082
    },
    {
      "epoch": 0.08168,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.08935139973958334,
      "learning_rate": 0.0001,
      "loss": 4.7406,
      "loss/crossentropy": 2.193789482116699,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2687358558177948,
      "step": 4084
    },
    {
      "epoch": 0.08172,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.020210774739583333,
      "learning_rate": 0.0001,
      "loss": 4.6454,
      "loss/crossentropy": 2.308240056037903,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2679348289966583,
      "step": 4086
    },
    {
      "epoch": 0.08176,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.021903483072916667,
      "learning_rate": 0.0001,
      "loss": 4.9126,
      "loss/crossentropy": 2.343047261238098,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3056950569152832,
      "step": 4088
    },
    {
      "epoch": 0.0818,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.022526041666666666,
      "learning_rate": 0.0001,
      "loss": 4.7315,
      "loss/crossentropy": 1.9583085179328918,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23911744356155396,
      "step": 4090
    },
    {
      "epoch": 0.08184,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.021198527018229166,
      "learning_rate": 0.0001,
      "loss": 4.6559,
      "loss/crossentropy": 2.341569185256958,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26053962111473083,
      "step": 4092
    },
    {
      "epoch": 0.08188,
      "grad_norm": 2.921875,
      "grad_norm_var": 0.034601847330729164,
      "learning_rate": 0.0001,
      "loss": 4.5773,
      "loss/crossentropy": 2.068669080734253,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25753986835479736,
      "step": 4094
    },
    {
      "epoch": 0.08192,
      "grad_norm": 2.625,
      "grad_norm_var": 0.04309488932291667,
      "learning_rate": 0.0001,
      "loss": 5.0253,
      "loss/crossentropy": 2.1461241841316223,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2628704681992531,
      "step": 4096
    },
    {
      "epoch": 0.08196,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.04810791015625,
      "learning_rate": 0.0001,
      "loss": 4.5587,
      "loss/crossentropy": 2.0718055963516235,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24923217296600342,
      "step": 4098
    },
    {
      "epoch": 0.082,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.0503082275390625,
      "learning_rate": 0.0001,
      "loss": 4.379,
      "loss/crossentropy": 1.9812004566192627,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2316955253481865,
      "step": 4100
    },
    {
      "epoch": 0.08204,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.0465240478515625,
      "learning_rate": 0.0001,
      "loss": 4.7909,
      "loss/crossentropy": 2.2669100761413574,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2806926518678665,
      "step": 4102
    },
    {
      "epoch": 0.08208,
      "grad_norm": 2.375,
      "grad_norm_var": 0.043196614583333334,
      "learning_rate": 0.0001,
      "loss": 4.7502,
      "loss/crossentropy": 2.0620261430740356,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2610047310590744,
      "step": 4104
    },
    {
      "epoch": 0.08212,
      "grad_norm": 3.046875,
      "grad_norm_var": 0.06297098795572917,
      "learning_rate": 0.0001,
      "loss": 4.6672,
      "loss/crossentropy": 2.249971866607666,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2783341556787491,
      "step": 4106
    },
    {
      "epoch": 0.08216,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.06320699055989583,
      "learning_rate": 0.0001,
      "loss": 4.834,
      "loss/crossentropy": 2.1064823865890503,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27534550428390503,
      "step": 4108
    },
    {
      "epoch": 0.0822,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.05406901041666667,
      "learning_rate": 0.0001,
      "loss": 4.3944,
      "loss/crossentropy": 1.886509656906128,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2315160632133484,
      "step": 4110
    },
    {
      "epoch": 0.08224,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.039290364583333334,
      "learning_rate": 0.0001,
      "loss": 4.2969,
      "loss/crossentropy": 1.6429635286331177,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20593100041151047,
      "step": 4112
    },
    {
      "epoch": 0.08228,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.0370025634765625,
      "learning_rate": 0.0001,
      "loss": 4.4581,
      "loss/crossentropy": 2.3236618041992188,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2674623131752014,
      "step": 4114
    },
    {
      "epoch": 0.08232,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.03658447265625,
      "learning_rate": 0.0001,
      "loss": 4.9734,
      "loss/crossentropy": 2.1479567885398865,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25003983825445175,
      "step": 4116
    },
    {
      "epoch": 0.08236,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.03611551920572917,
      "learning_rate": 0.0001,
      "loss": 5.0477,
      "loss/crossentropy": 2.140569031238556,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24322029948234558,
      "step": 4118
    },
    {
      "epoch": 0.0824,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.03762613932291667,
      "learning_rate": 0.0001,
      "loss": 4.6061,
      "loss/crossentropy": 2.126375436782837,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27629759907722473,
      "step": 4120
    },
    {
      "epoch": 0.08244,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.015843709309895832,
      "learning_rate": 0.0001,
      "loss": 4.9143,
      "loss/crossentropy": 2.3699214458465576,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2692546397447586,
      "step": 4122
    },
    {
      "epoch": 0.08248,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.010904947916666666,
      "learning_rate": 0.0001,
      "loss": 4.5672,
      "loss/crossentropy": 2.013331353664398,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24442073702812195,
      "step": 4124
    },
    {
      "epoch": 0.08252,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.010152180989583334,
      "learning_rate": 0.0001,
      "loss": 4.4841,
      "loss/crossentropy": 2.1869460344314575,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.250150203704834,
      "step": 4126
    },
    {
      "epoch": 0.08256,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.0127838134765625,
      "learning_rate": 0.0001,
      "loss": 4.4564,
      "loss/crossentropy": 2.2725884914398193,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2810261696577072,
      "step": 4128
    },
    {
      "epoch": 0.0826,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.013688151041666667,
      "learning_rate": 0.0001,
      "loss": 4.7311,
      "loss/crossentropy": 1.9190022945404053,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2693525403738022,
      "step": 4130
    },
    {
      "epoch": 0.08264,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.016112263997395834,
      "learning_rate": 0.0001,
      "loss": 4.7967,
      "loss/crossentropy": 2.5477795600891113,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27040669322013855,
      "step": 4132
    },
    {
      "epoch": 0.08268,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.0113677978515625,
      "learning_rate": 0.0001,
      "loss": 4.7617,
      "loss/crossentropy": 2.231198728084564,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28101250529289246,
      "step": 4134
    },
    {
      "epoch": 0.08272,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.011693318684895834,
      "learning_rate": 0.0001,
      "loss": 4.6334,
      "loss/crossentropy": 2.17776882648468,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24743208289146423,
      "step": 4136
    },
    {
      "epoch": 0.08276,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.011799112955729166,
      "learning_rate": 0.0001,
      "loss": 5.0233,
      "loss/crossentropy": 2.418373703956604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2954525351524353,
      "step": 4138
    },
    {
      "epoch": 0.0828,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.010969034830729167,
      "learning_rate": 0.0001,
      "loss": 4.5564,
      "loss/crossentropy": 2.054605543613434,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23863950371742249,
      "step": 4140
    },
    {
      "epoch": 0.08284,
      "grad_norm": 2.5,
      "grad_norm_var": 0.011872355143229167,
      "learning_rate": 0.0001,
      "loss": 4.8983,
      "loss/crossentropy": 2.054013967514038,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29851172864437103,
      "step": 4142
    },
    {
      "epoch": 0.08288,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.008610026041666666,
      "learning_rate": 0.0001,
      "loss": 4.7425,
      "loss/crossentropy": 2.193961024284363,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2563214898109436,
      "step": 4144
    },
    {
      "epoch": 0.08292,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.008382161458333334,
      "learning_rate": 0.0001,
      "loss": 4.7995,
      "loss/crossentropy": 2.460008382797241,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26705074310302734,
      "step": 4146
    },
    {
      "epoch": 0.08296,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.005060831705729167,
      "learning_rate": 0.0001,
      "loss": 4.894,
      "loss/crossentropy": 2.508321523666382,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26217761635780334,
      "step": 4148
    },
    {
      "epoch": 0.083,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.0065582275390625,
      "learning_rate": 0.0001,
      "loss": 4.6103,
      "loss/crossentropy": 1.8445284366607666,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23396535962820053,
      "step": 4150
    },
    {
      "epoch": 0.08304,
      "grad_norm": 2.375,
      "grad_norm_var": 0.005952962239583333,
      "learning_rate": 0.0001,
      "loss": 4.8048,
      "loss/crossentropy": 2.433600902557373,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2825329154729843,
      "step": 4152
    },
    {
      "epoch": 0.08308,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.0053212483723958336,
      "learning_rate": 0.0001,
      "loss": 4.8632,
      "loss/crossentropy": 2.386221170425415,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2941686511039734,
      "step": 4154
    },
    {
      "epoch": 0.08312,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.0052154541015625,
      "learning_rate": 0.0001,
      "loss": 4.7486,
      "loss/crossentropy": 1.9578949809074402,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24851053953170776,
      "step": 4156
    },
    {
      "epoch": 0.08316,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.004150390625,
      "learning_rate": 0.0001,
      "loss": 4.6443,
      "loss/crossentropy": 2.0034408569335938,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25524984300136566,
      "step": 4158
    },
    {
      "epoch": 0.0832,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.0069488525390625,
      "learning_rate": 0.0001,
      "loss": 4.6856,
      "loss/crossentropy": 2.3342589139938354,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31023962795734406,
      "step": 4160
    },
    {
      "epoch": 0.08324,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.042867024739583336,
      "learning_rate": 0.0001,
      "loss": 4.6762,
      "loss/crossentropy": 2.3941839933395386,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2641746401786804,
      "step": 4162
    },
    {
      "epoch": 0.08328,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.04168294270833333,
      "learning_rate": 0.0001,
      "loss": 4.7149,
      "loss/crossentropy": 2.371219038963318,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24197939038276672,
      "step": 4164
    },
    {
      "epoch": 0.08332,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.04442952473958333,
      "learning_rate": 0.0001,
      "loss": 4.7949,
      "loss/crossentropy": 2.133378028869629,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25449906289577484,
      "step": 4166
    },
    {
      "epoch": 0.08336,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.04496968587239583,
      "learning_rate": 0.0001,
      "loss": 4.5974,
      "loss/crossentropy": 1.7460412979125977,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2157151699066162,
      "step": 4168
    },
    {
      "epoch": 0.0834,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.046873982747395834,
      "learning_rate": 0.0001,
      "loss": 4.7234,
      "loss/crossentropy": 2.215083122253418,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2823774367570877,
      "step": 4170
    },
    {
      "epoch": 0.08344,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.04820048014322917,
      "learning_rate": 0.0001,
      "loss": 4.5189,
      "loss/crossentropy": 2.0528377890586853,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2551003098487854,
      "step": 4172
    },
    {
      "epoch": 0.08348,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.05054423014322917,
      "learning_rate": 0.0001,
      "loss": 4.4008,
      "loss/crossentropy": 1.7953855395317078,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23185917735099792,
      "step": 4174
    },
    {
      "epoch": 0.08352,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.058934529622395836,
      "learning_rate": 0.0001,
      "loss": 4.4879,
      "loss/crossentropy": 2.0794734954833984,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23386041820049286,
      "step": 4176
    },
    {
      "epoch": 0.08356,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.031086222330729166,
      "learning_rate": 0.0001,
      "loss": 4.3802,
      "loss/crossentropy": 2.1685845851898193,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26074862480163574,
      "step": 4178
    },
    {
      "epoch": 0.0836,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.031412760416666664,
      "learning_rate": 0.0001,
      "loss": 4.5507,
      "loss/crossentropy": 2.1495825052261353,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2689145505428314,
      "step": 4180
    },
    {
      "epoch": 0.08364,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.024967447916666666,
      "learning_rate": 0.0001,
      "loss": 4.5258,
      "loss/crossentropy": 2.043331503868103,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2498578578233719,
      "step": 4182
    },
    {
      "epoch": 0.08368,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.07787984212239583,
      "learning_rate": 0.0001,
      "loss": 5.0198,
      "loss/crossentropy": 2.04026997089386,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2873340845108032,
      "step": 4184
    },
    {
      "epoch": 0.08372,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.09128316243489583,
      "learning_rate": 0.0001,
      "loss": 4.9537,
      "loss/crossentropy": 2.4653968811035156,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2911294251680374,
      "step": 4186
    },
    {
      "epoch": 0.08376,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.09215494791666666,
      "learning_rate": 0.0001,
      "loss": 4.6589,
      "loss/crossentropy": 2.2960848808288574,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23984040319919586,
      "step": 4188
    },
    {
      "epoch": 0.0838,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.09599202473958333,
      "learning_rate": 0.0001,
      "loss": 4.4324,
      "loss/crossentropy": 2.011807084083557,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2523125037550926,
      "step": 4190
    },
    {
      "epoch": 0.08384,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.0886871337890625,
      "learning_rate": 0.0001,
      "loss": 4.8437,
      "loss/crossentropy": 2.0016889572143555,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2291206791996956,
      "step": 4192
    },
    {
      "epoch": 0.08388,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.09378255208333333,
      "learning_rate": 0.0001,
      "loss": 4.3604,
      "loss/crossentropy": 1.97197824716568,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2595779076218605,
      "step": 4194
    },
    {
      "epoch": 0.08392,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.09763895670572917,
      "learning_rate": 0.0001,
      "loss": 4.5823,
      "loss/crossentropy": 2.2910103797912598,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24870596826076508,
      "step": 4196
    },
    {
      "epoch": 0.08396,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.1001617431640625,
      "learning_rate": 0.0001,
      "loss": 4.562,
      "loss/crossentropy": 2.1453208923339844,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25001347810029984,
      "step": 4198
    },
    {
      "epoch": 0.084,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.0398834228515625,
      "learning_rate": 0.0001,
      "loss": 4.8835,
      "loss/crossentropy": 2.1935043334960938,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26807525753974915,
      "step": 4200
    },
    {
      "epoch": 0.08404,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.009765625,
      "learning_rate": 0.0001,
      "loss": 4.5912,
      "loss/crossentropy": 2.039341926574707,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2460380420088768,
      "step": 4202
    },
    {
      "epoch": 0.08408,
      "grad_norm": 3.09375,
      "grad_norm_var": 0.0478179931640625,
      "learning_rate": 0.0001,
      "loss": 4.8243,
      "loss/crossentropy": 2.4660122394561768,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3042101263999939,
      "step": 4204
    },
    {
      "epoch": 0.08412,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.08088785807291667,
      "learning_rate": 0.0001,
      "loss": 4.8635,
      "loss/crossentropy": 1.9346272349357605,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25924334675073624,
      "step": 4206
    },
    {
      "epoch": 0.08416,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.08311258951822917,
      "learning_rate": 0.0001,
      "loss": 4.5152,
      "loss/crossentropy": 2.0120063424110413,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24327433109283447,
      "step": 4208
    },
    {
      "epoch": 0.0842,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.0852935791015625,
      "learning_rate": 0.0001,
      "loss": 4.6148,
      "loss/crossentropy": 2.2359931468963623,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25374244898557663,
      "step": 4210
    },
    {
      "epoch": 0.08424,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.08185221354166666,
      "learning_rate": 0.0001,
      "loss": 4.5751,
      "loss/crossentropy": 2.0038134455680847,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22315364331007004,
      "step": 4212
    },
    {
      "epoch": 0.08428,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.08567301432291667,
      "learning_rate": 0.0001,
      "loss": 4.721,
      "loss/crossentropy": 2.2041471004486084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2655494213104248,
      "step": 4214
    },
    {
      "epoch": 0.08432,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.08399149576822916,
      "learning_rate": 0.0001,
      "loss": 4.8091,
      "loss/crossentropy": 2.344551682472229,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2743126451969147,
      "step": 4216
    },
    {
      "epoch": 0.08436,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.08025614420572917,
      "learning_rate": 0.0001,
      "loss": 4.7162,
      "loss/crossentropy": 1.9694250226020813,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24425261467695236,
      "step": 4218
    },
    {
      "epoch": 0.0844,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.05671284993489583,
      "learning_rate": 0.0001,
      "loss": 4.8526,
      "loss/crossentropy": 2.164921760559082,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.272469699382782,
      "step": 4220
    },
    {
      "epoch": 0.08444,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.02642822265625,
      "learning_rate": 0.0001,
      "loss": 4.5513,
      "loss/crossentropy": 1.944575309753418,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23848393559455872,
      "step": 4222
    },
    {
      "epoch": 0.08448,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.025386555989583334,
      "learning_rate": 0.0001,
      "loss": 4.7416,
      "loss/crossentropy": 2.278227686882019,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2558315545320511,
      "step": 4224
    },
    {
      "epoch": 0.08452,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.01802978515625,
      "learning_rate": 0.0001,
      "loss": 4.7318,
      "loss/crossentropy": 2.035117268562317,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2585935667157173,
      "step": 4226
    },
    {
      "epoch": 0.08456,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.020466105143229166,
      "learning_rate": 0.0001,
      "loss": 4.5674,
      "loss/crossentropy": 2.0172035694122314,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23832131922245026,
      "step": 4228
    },
    {
      "epoch": 0.0846,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.011165364583333334,
      "learning_rate": 0.0001,
      "loss": 4.8113,
      "loss/crossentropy": 2.0574535727500916,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23829226195812225,
      "step": 4230
    },
    {
      "epoch": 0.08464,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.010773722330729167,
      "learning_rate": 0.0001,
      "loss": 4.6776,
      "loss/crossentropy": 2.5003366470336914,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27905476093292236,
      "step": 4232
    },
    {
      "epoch": 0.08468,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.012043253580729166,
      "learning_rate": 0.0001,
      "loss": 4.9137,
      "loss/crossentropy": 2.207367777824402,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27426937222480774,
      "step": 4234
    },
    {
      "epoch": 0.08472,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.01627197265625,
      "learning_rate": 0.0001,
      "loss": 4.716,
      "loss/crossentropy": 2.240189790725708,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26208513230085373,
      "step": 4236
    },
    {
      "epoch": 0.08476,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.016600545247395834,
      "learning_rate": 0.0001,
      "loss": 4.4569,
      "loss/crossentropy": 2.1357412338256836,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24108420312404633,
      "step": 4238
    },
    {
      "epoch": 0.0848,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.013004557291666666,
      "learning_rate": 0.0001,
      "loss": 4.7249,
      "loss/crossentropy": 2.1073816418647766,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25213342159986496,
      "step": 4240
    },
    {
      "epoch": 0.08484,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.013102213541666666,
      "learning_rate": 0.0001,
      "loss": 4.9558,
      "loss/crossentropy": 2.158124566078186,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2505848854780197,
      "step": 4242
    },
    {
      "epoch": 0.08488,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.011555989583333334,
      "learning_rate": 0.0001,
      "loss": 4.7743,
      "loss/crossentropy": 2.253539562225342,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2675466388463974,
      "step": 4244
    },
    {
      "epoch": 0.08492,
      "grad_norm": 2.5,
      "grad_norm_var": 0.01396484375,
      "learning_rate": 0.0001,
      "loss": 4.272,
      "loss/crossentropy": 1.7170023918151855,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20157357305288315,
      "step": 4246
    },
    {
      "epoch": 0.08496,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.0188385009765625,
      "learning_rate": 0.0001,
      "loss": 4.4479,
      "loss/crossentropy": 2.082640767097473,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2600134015083313,
      "step": 4248
    },
    {
      "epoch": 0.085,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.0165679931640625,
      "learning_rate": 0.0001,
      "loss": 4.4176,
      "loss/crossentropy": 2.044301390647888,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23220707476139069,
      "step": 4250
    },
    {
      "epoch": 0.08504,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.0168121337890625,
      "learning_rate": 0.0001,
      "loss": 4.648,
      "loss/crossentropy": 2.293405532836914,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28027066588401794,
      "step": 4252
    },
    {
      "epoch": 0.08508,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.016649373372395835,
      "learning_rate": 0.0001,
      "loss": 4.55,
      "loss/crossentropy": 2.2604206800460815,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25545646995306015,
      "step": 4254
    },
    {
      "epoch": 0.08512,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.023949178059895833,
      "learning_rate": 0.0001,
      "loss": 4.6258,
      "loss/crossentropy": 2.118361234664917,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24799348413944244,
      "step": 4256
    },
    {
      "epoch": 0.08516,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.022847493489583332,
      "learning_rate": 0.0001,
      "loss": 4.6751,
      "loss/crossentropy": 1.9369969964027405,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24676478654146194,
      "step": 4258
    },
    {
      "epoch": 0.0852,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.021728515625,
      "learning_rate": 0.0001,
      "loss": 4.5197,
      "loss/crossentropy": 2.075170874595642,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21742676943540573,
      "step": 4260
    },
    {
      "epoch": 0.08524,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.018973795572916667,
      "learning_rate": 0.0001,
      "loss": 4.4112,
      "loss/crossentropy": 2.056099236011505,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2490842342376709,
      "step": 4262
    },
    {
      "epoch": 0.08528,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.014436848958333333,
      "learning_rate": 0.0001,
      "loss": 4.6169,
      "loss/crossentropy": 2.2279993891716003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24964337795972824,
      "step": 4264
    },
    {
      "epoch": 0.08532,
      "grad_norm": 2.375,
      "grad_norm_var": 0.011393229166666666,
      "learning_rate": 0.0001,
      "loss": 4.6686,
      "loss/crossentropy": 2.1645933389663696,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24576786905527115,
      "step": 4266
    },
    {
      "epoch": 0.08536,
      "grad_norm": 2.25,
      "grad_norm_var": 0.010887654622395833,
      "learning_rate": 0.0001,
      "loss": 4.4458,
      "loss/crossentropy": 1.9033920764923096,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23393237590789795,
      "step": 4268
    },
    {
      "epoch": 0.0854,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.011865234375,
      "learning_rate": 0.0001,
      "loss": 4.4022,
      "loss/crossentropy": 2.153634190559387,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2496839165687561,
      "step": 4270
    },
    {
      "epoch": 0.08544,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.0045206705729166664,
      "learning_rate": 0.0001,
      "loss": 4.4781,
      "loss/crossentropy": 1.9188589453697205,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23544569313526154,
      "step": 4272
    },
    {
      "epoch": 0.08548,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.004264322916666666,
      "learning_rate": 0.0001,
      "loss": 4.704,
      "loss/crossentropy": 2.4337977170944214,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2951700836420059,
      "step": 4274
    },
    {
      "epoch": 0.08552,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.003902180989583333,
      "learning_rate": 0.0001,
      "loss": 4.7051,
      "loss/crossentropy": 1.9108383059501648,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24459081888198853,
      "step": 4276
    },
    {
      "epoch": 0.08556,
      "grad_norm": 2.375,
      "grad_norm_var": 0.003123982747395833,
      "learning_rate": 0.0001,
      "loss": 4.3751,
      "loss/crossentropy": 1.6632736921310425,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22495487332344055,
      "step": 4278
    },
    {
      "epoch": 0.0856,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.021312459309895834,
      "learning_rate": 0.0001,
      "loss": 4.8144,
      "loss/crossentropy": 2.519997477531433,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2657178193330765,
      "step": 4280
    },
    {
      "epoch": 0.08564,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.025275675455729167,
      "learning_rate": 0.0001,
      "loss": 4.6837,
      "loss/crossentropy": 2.150822162628174,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30192580074071884,
      "step": 4282
    },
    {
      "epoch": 0.08568,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.023763020833333332,
      "learning_rate": 0.0001,
      "loss": 4.7411,
      "loss/crossentropy": 1.9970062971115112,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24695321917533875,
      "step": 4284
    },
    {
      "epoch": 0.08572,
      "grad_norm": 2.25,
      "grad_norm_var": 0.024372355143229166,
      "learning_rate": 0.0001,
      "loss": 4.5361,
      "loss/crossentropy": 2.3136098384857178,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25082121044397354,
      "step": 4286
    },
    {
      "epoch": 0.08576,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.025992838541666667,
      "learning_rate": 0.0001,
      "loss": 4.9171,
      "loss/crossentropy": 2.112035870552063,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27019062638282776,
      "step": 4288
    },
    {
      "epoch": 0.0858,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.025992838541666667,
      "learning_rate": 0.0001,
      "loss": 4.4985,
      "loss/crossentropy": 2.068653643131256,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24297921359539032,
      "step": 4290
    },
    {
      "epoch": 0.08584,
      "grad_norm": 2.375,
      "grad_norm_var": 0.025520833333333333,
      "learning_rate": 0.0001,
      "loss": 4.5182,
      "loss/crossentropy": 1.9013578295707703,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23058265447616577,
      "step": 4292
    },
    {
      "epoch": 0.08588,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.0294830322265625,
      "learning_rate": 0.0001,
      "loss": 4.6825,
      "loss/crossentropy": 2.149984359741211,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25223904848098755,
      "step": 4294
    },
    {
      "epoch": 0.08592,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.014469401041666666,
      "learning_rate": 0.0001,
      "loss": 4.4109,
      "loss/crossentropy": 1.894010066986084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2323223054409027,
      "step": 4296
    },
    {
      "epoch": 0.08596,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.010123697916666667,
      "learning_rate": 0.0001,
      "loss": 4.7653,
      "loss/crossentropy": 2.3351621627807617,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27194930613040924,
      "step": 4298
    },
    {
      "epoch": 0.086,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.0097320556640625,
      "learning_rate": 0.0001,
      "loss": 4.741,
      "loss/crossentropy": 2.224352180957794,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24884501099586487,
      "step": 4300
    },
    {
      "epoch": 0.08604,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.008854166666666666,
      "learning_rate": 0.0001,
      "loss": 4.6592,
      "loss/crossentropy": 1.908318042755127,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24747492372989655,
      "step": 4302
    },
    {
      "epoch": 0.08608,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.00758056640625,
      "learning_rate": 0.0001,
      "loss": 4.8566,
      "loss/crossentropy": 2.1990396976470947,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27217794954776764,
      "step": 4304
    },
    {
      "epoch": 0.08612,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.0098297119140625,
      "learning_rate": 0.0001,
      "loss": 4.6432,
      "loss/crossentropy": 2.3146010637283325,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26875850558280945,
      "step": 4306
    },
    {
      "epoch": 0.08616,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.015208943684895834,
      "learning_rate": 0.0001,
      "loss": 4.8254,
      "loss/crossentropy": 2.2507941722869873,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27052539587020874,
      "step": 4308
    },
    {
      "epoch": 0.0862,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.013199869791666667,
      "learning_rate": 0.0001,
      "loss": 4.4067,
      "loss/crossentropy": 1.9077125787734985,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22987178713083267,
      "step": 4310
    },
    {
      "epoch": 0.08624,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.01353759765625,
      "learning_rate": 0.0001,
      "loss": 4.4822,
      "loss/crossentropy": 1.951395332813263,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2392890453338623,
      "step": 4312
    },
    {
      "epoch": 0.08628,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.033854166666666664,
      "learning_rate": 0.0001,
      "loss": 4.5371,
      "loss/crossentropy": 1.9426860213279724,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24374966323375702,
      "step": 4314
    },
    {
      "epoch": 0.08632,
      "grad_norm": 3.390625,
      "grad_norm_var": 0.09062398274739583,
      "learning_rate": 0.0001,
      "loss": 5.253,
      "loss/crossentropy": 2.2508288621902466,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3862452507019043,
      "step": 4316
    },
    {
      "epoch": 0.08636,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.09058329264322916,
      "learning_rate": 0.0001,
      "loss": 4.5288,
      "loss/crossentropy": 2.1161463260650635,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2557579278945923,
      "step": 4318
    },
    {
      "epoch": 0.0864,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.09374593098958334,
      "learning_rate": 0.0001,
      "loss": 5.1146,
      "loss/crossentropy": 2.2570544481277466,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31447017192840576,
      "step": 4320
    },
    {
      "epoch": 0.08644,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.08740234375,
      "learning_rate": 0.0001,
      "loss": 4.9724,
      "loss/crossentropy": 2.3211100101470947,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26107798516750336,
      "step": 4322
    },
    {
      "epoch": 0.08648,
      "grad_norm": 2.25,
      "grad_norm_var": 0.09231669108072917,
      "learning_rate": 0.0001,
      "loss": 4.5236,
      "loss/crossentropy": 2.1451058387756348,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23633568733930588,
      "step": 4324
    },
    {
      "epoch": 0.08652,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.09463602701822917,
      "learning_rate": 0.0001,
      "loss": 4.5828,
      "loss/crossentropy": 1.9880141615867615,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23688867688179016,
      "step": 4326
    },
    {
      "epoch": 0.08656,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.09724934895833333,
      "learning_rate": 0.0001,
      "loss": 4.7098,
      "loss/crossentropy": 2.021056890487671,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2543798238039017,
      "step": 4328
    },
    {
      "epoch": 0.0866,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.0814453125,
      "learning_rate": 0.0001,
      "loss": 4.6439,
      "loss/crossentropy": 2.1323755979537964,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27410852909088135,
      "step": 4330
    },
    {
      "epoch": 0.08664,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.014940388997395833,
      "learning_rate": 0.0001,
      "loss": 4.6945,
      "loss/crossentropy": 1.9674875736236572,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24818265438079834,
      "step": 4332
    },
    {
      "epoch": 0.08668,
      "grad_norm": 2.25,
      "grad_norm_var": 0.01490478515625,
      "learning_rate": 0.0001,
      "loss": 4.1911,
      "loss/crossentropy": 2.0466583967208862,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2567761391401291,
      "step": 4334
    },
    {
      "epoch": 0.08672,
      "grad_norm": 2.375,
      "grad_norm_var": 0.004423014322916667,
      "learning_rate": 0.0001,
      "loss": 4.5937,
      "loss/crossentropy": 2.138857126235962,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26430967450141907,
      "step": 4336
    },
    {
      "epoch": 0.08676,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.003902180989583333,
      "learning_rate": 0.0001,
      "loss": 4.7168,
      "loss/crossentropy": 2.164841413497925,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24389629065990448,
      "step": 4338
    },
    {
      "epoch": 0.0868,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.005322265625,
      "learning_rate": 0.0001,
      "loss": 4.6017,
      "loss/crossentropy": 2.2220189571380615,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24361558258533478,
      "step": 4340
    },
    {
      "epoch": 0.08684,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.004792277018229167,
      "learning_rate": 0.0001,
      "loss": 4.3088,
      "loss/crossentropy": 1.7106285095214844,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21558403968811035,
      "step": 4342
    },
    {
      "epoch": 0.08688,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.0053670247395833336,
      "learning_rate": 0.0001,
      "loss": 4.1647,
      "loss/crossentropy": 1.9200173020362854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2502904310822487,
      "step": 4344
    },
    {
      "epoch": 0.08692,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.020796712239583334,
      "learning_rate": 0.0001,
      "loss": 4.6817,
      "loss/crossentropy": 1.883722960948944,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24818243086338043,
      "step": 4346
    },
    {
      "epoch": 0.08696,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.026146443684895833,
      "learning_rate": 0.0001,
      "loss": 4.7509,
      "loss/crossentropy": 2.2069878578186035,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2518744319677353,
      "step": 4348
    },
    {
      "epoch": 0.087,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.026656087239583334,
      "learning_rate": 0.0001,
      "loss": 4.7235,
      "loss/crossentropy": 2.2158325910568237,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2551049590110779,
      "step": 4350
    },
    {
      "epoch": 0.08704,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.02730712890625,
      "learning_rate": 0.0001,
      "loss": 4.8406,
      "loss/crossentropy": 2.0580105781555176,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2742728739976883,
      "step": 4352
    },
    {
      "epoch": 0.08708,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.02906494140625,
      "learning_rate": 0.0001,
      "loss": 4.5267,
      "loss/crossentropy": 2.190276265144348,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.274506613612175,
      "step": 4354
    },
    {
      "epoch": 0.08712,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.027242024739583332,
      "learning_rate": 0.0001,
      "loss": 4.7499,
      "loss/crossentropy": 2.2595328092575073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25295622646808624,
      "step": 4356
    },
    {
      "epoch": 0.08716,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.027391560872395835,
      "learning_rate": 0.0001,
      "loss": 5.1247,
      "loss/crossentropy": 2.322342872619629,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2490948587656021,
      "step": 4358
    },
    {
      "epoch": 0.0872,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.024665323893229167,
      "learning_rate": 0.0001,
      "loss": 4.7045,
      "loss/crossentropy": 2.108223795890808,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25404803454875946,
      "step": 4360
    },
    {
      "epoch": 0.08724,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.013825480143229167,
      "learning_rate": 0.0001,
      "loss": 4.5332,
      "loss/crossentropy": 2.1363136768341064,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2679157853126526,
      "step": 4362
    },
    {
      "epoch": 0.08728,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.0076171875,
      "learning_rate": 0.0001,
      "loss": 4.6418,
      "loss/crossentropy": 2.3207738399505615,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2725762128829956,
      "step": 4364
    },
    {
      "epoch": 0.08732,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.004813639322916666,
      "learning_rate": 0.0001,
      "loss": 4.7431,
      "loss/crossentropy": 2.3179128170013428,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27275949716567993,
      "step": 4366
    },
    {
      "epoch": 0.08736,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.005182902018229167,
      "learning_rate": 0.0001,
      "loss": 4.7355,
      "loss/crossentropy": 2.2130206823349,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2787477523088455,
      "step": 4368
    },
    {
      "epoch": 0.0874,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.004548136393229167,
      "learning_rate": 0.0001,
      "loss": 4.6193,
      "loss/crossentropy": 2.3350926637649536,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26483266800642014,
      "step": 4370
    },
    {
      "epoch": 0.08744,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.005736287434895833,
      "learning_rate": 0.0001,
      "loss": 4.7329,
      "loss/crossentropy": 1.9162638187408447,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23748356848955154,
      "step": 4372
    },
    {
      "epoch": 0.08748,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.006322224934895833,
      "learning_rate": 0.0001,
      "loss": 4.7045,
      "loss/crossentropy": 2.2708429098129272,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2709425985813141,
      "step": 4374
    },
    {
      "epoch": 0.08752,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.011604817708333333,
      "learning_rate": 0.0001,
      "loss": 4.3481,
      "loss/crossentropy": 1.7216318845748901,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22498781234025955,
      "step": 4376
    },
    {
      "epoch": 0.08756,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.011937459309895834,
      "learning_rate": 0.0001,
      "loss": 4.4261,
      "loss/crossentropy": 2.144331693649292,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25400668382644653,
      "step": 4378
    },
    {
      "epoch": 0.0876,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.011750284830729167,
      "learning_rate": 0.0001,
      "loss": 4.5617,
      "loss/crossentropy": 2.305369734764099,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29137127101421356,
      "step": 4380
    },
    {
      "epoch": 0.08764,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.018115234375,
      "learning_rate": 0.0001,
      "loss": 4.6861,
      "loss/crossentropy": 2.1156765818595886,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24203064292669296,
      "step": 4382
    },
    {
      "epoch": 0.08768,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.016630045572916665,
      "learning_rate": 0.0001,
      "loss": 4.4544,
      "loss/crossentropy": 1.9081769585609436,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22396781295537949,
      "step": 4384
    },
    {
      "epoch": 0.08772,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.017769368489583333,
      "learning_rate": 0.0001,
      "loss": 4.704,
      "loss/crossentropy": 2.0938609838485718,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27330371737480164,
      "step": 4386
    },
    {
      "epoch": 0.08776,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.017748006184895835,
      "learning_rate": 0.0001,
      "loss": 4.1179,
      "loss/crossentropy": 1.9685207605361938,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2287725731730461,
      "step": 4388
    },
    {
      "epoch": 0.0878,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.016109212239583334,
      "learning_rate": 0.0001,
      "loss": 4.5233,
      "loss/crossentropy": 1.7536925673484802,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2345883920788765,
      "step": 4390
    },
    {
      "epoch": 0.08784,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.029150390625,
      "learning_rate": 0.0001,
      "loss": 4.3897,
      "loss/crossentropy": 2.145567834377289,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26126645505428314,
      "step": 4392
    },
    {
      "epoch": 0.08788,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.028348795572916665,
      "learning_rate": 0.0001,
      "loss": 4.4677,
      "loss/crossentropy": 2.2211687564849854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3332909345626831,
      "step": 4394
    },
    {
      "epoch": 0.08792,
      "grad_norm": 2.375,
      "grad_norm_var": 0.028539021809895832,
      "learning_rate": 0.0001,
      "loss": 4.5545,
      "loss/crossentropy": 1.919084072113037,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2480143904685974,
      "step": 4396
    },
    {
      "epoch": 0.08796,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.025121053059895832,
      "learning_rate": 0.0001,
      "loss": 4.6408,
      "loss/crossentropy": 2.1769548654556274,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.269208699464798,
      "step": 4398
    },
    {
      "epoch": 0.088,
      "grad_norm": 2.5,
      "grad_norm_var": 0.028706868489583332,
      "learning_rate": 0.0001,
      "loss": 4.7372,
      "loss/crossentropy": 1.8480086922645569,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2432136833667755,
      "step": 4400
    },
    {
      "epoch": 0.08804,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.026838175455729165,
      "learning_rate": 0.0001,
      "loss": 4.7303,
      "loss/crossentropy": 2.1948903799057007,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28328536450862885,
      "step": 4402
    },
    {
      "epoch": 0.08808,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.022554524739583335,
      "learning_rate": 0.0001,
      "loss": 4.6929,
      "loss/crossentropy": 2.163570761680603,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.282375693321228,
      "step": 4404
    },
    {
      "epoch": 0.08812,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.022359212239583332,
      "learning_rate": 0.0001,
      "loss": 4.6882,
      "loss/crossentropy": 2.3737696409225464,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26703887432813644,
      "step": 4406
    },
    {
      "epoch": 0.08816,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.015192667643229166,
      "learning_rate": 0.0001,
      "loss": 4.6959,
      "loss/crossentropy": 2.2449779510498047,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2744368612766266,
      "step": 4408
    },
    {
      "epoch": 0.0882,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.015348307291666667,
      "learning_rate": 0.0001,
      "loss": 4.6602,
      "loss/crossentropy": 2.1167399287223816,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28688907623291016,
      "step": 4410
    },
    {
      "epoch": 0.08824,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.017118326822916665,
      "learning_rate": 0.0001,
      "loss": 4.3777,
      "loss/crossentropy": 2.2249929904937744,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24534232914447784,
      "step": 4412
    },
    {
      "epoch": 0.08828,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.019261678059895832,
      "learning_rate": 0.0001,
      "loss": 4.7013,
      "loss/crossentropy": 2.172566771507263,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2783561646938324,
      "step": 4414
    },
    {
      "epoch": 0.08832,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.018684895833333333,
      "learning_rate": 0.0001,
      "loss": 4.3536,
      "loss/crossentropy": 2.0709031224250793,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2419627606868744,
      "step": 4416
    },
    {
      "epoch": 0.08836,
      "grad_norm": 3.359375,
      "grad_norm_var": 0.0735260009765625,
      "learning_rate": 0.0001,
      "loss": 4.8378,
      "loss/crossentropy": 2.2390655279159546,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27328142523765564,
      "step": 4418
    },
    {
      "epoch": 0.0884,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.07681884765625,
      "learning_rate": 0.0001,
      "loss": 4.6879,
      "loss/crossentropy": 2.061118960380554,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23206621408462524,
      "step": 4420
    },
    {
      "epoch": 0.08844,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.09147847493489583,
      "learning_rate": 0.0001,
      "loss": 4.3028,
      "loss/crossentropy": 1.4919558763504028,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17881463468074799,
      "step": 4422
    },
    {
      "epoch": 0.08848,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.08662109375,
      "learning_rate": 0.0001,
      "loss": 4.7705,
      "loss/crossentropy": 2.267301082611084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27718164026737213,
      "step": 4424
    },
    {
      "epoch": 0.08852,
      "grad_norm": 2.375,
      "grad_norm_var": 0.0862457275390625,
      "learning_rate": 0.0001,
      "loss": 4.635,
      "loss/crossentropy": 1.9008094668388367,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22865734994411469,
      "step": 4426
    },
    {
      "epoch": 0.08856,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.090966796875,
      "learning_rate": 0.0001,
      "loss": 4.385,
      "loss/crossentropy": 1.8788060545921326,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23521529138088226,
      "step": 4428
    },
    {
      "epoch": 0.0886,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.08844401041666666,
      "learning_rate": 0.0001,
      "loss": 4.6703,
      "loss/crossentropy": 2.0775802731513977,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3402235209941864,
      "step": 4430
    },
    {
      "epoch": 0.08864,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.08642171223958334,
      "learning_rate": 0.0001,
      "loss": 4.5722,
      "loss/crossentropy": 2.0950201749801636,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2665044367313385,
      "step": 4432
    },
    {
      "epoch": 0.08868,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.026167805989583334,
      "learning_rate": 0.0001,
      "loss": 4.9235,
      "loss/crossentropy": 2.328200340270996,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24911007285118103,
      "step": 4434
    },
    {
      "epoch": 0.08872,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.023273722330729166,
      "learning_rate": 0.0001,
      "loss": 4.7462,
      "loss/crossentropy": 2.1840142011642456,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2855416387319565,
      "step": 4436
    },
    {
      "epoch": 0.08876,
      "grad_norm": 2.25,
      "grad_norm_var": 0.015559895833333334,
      "learning_rate": 0.0001,
      "loss": 4.1399,
      "loss/crossentropy": 1.6167555451393127,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21646380424499512,
      "step": 4438
    },
    {
      "epoch": 0.0888,
      "grad_norm": 2.25,
      "grad_norm_var": 0.015543619791666666,
      "learning_rate": 0.0001,
      "loss": 4.5274,
      "loss/crossentropy": 1.9902858138084412,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.237789124250412,
      "step": 4440
    },
    {
      "epoch": 0.08884,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.016357421875,
      "learning_rate": 0.0001,
      "loss": 4.6682,
      "loss/crossentropy": 2.4779287576675415,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2726414203643799,
      "step": 4442
    },
    {
      "epoch": 0.08888,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.0116119384765625,
      "learning_rate": 0.0001,
      "loss": 4.7415,
      "loss/crossentropy": 2.0914896726608276,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22887279838323593,
      "step": 4444
    },
    {
      "epoch": 0.08892,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.010570271809895834,
      "learning_rate": 0.0001,
      "loss": 4.498,
      "loss/crossentropy": 2.0526055693626404,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2509382963180542,
      "step": 4446
    },
    {
      "epoch": 0.08896,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.010347493489583333,
      "learning_rate": 0.0001,
      "loss": 4.4306,
      "loss/crossentropy": 1.9779084920883179,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2320951297879219,
      "step": 4448
    },
    {
      "epoch": 0.089,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.010399373372395833,
      "learning_rate": 0.0001,
      "loss": 4.7725,
      "loss/crossentropy": 2.2081698179244995,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29023079574108124,
      "step": 4450
    },
    {
      "epoch": 0.08904,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.0074127197265625,
      "learning_rate": 0.0001,
      "loss": 4.542,
      "loss/crossentropy": 1.834806501865387,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23430980741977692,
      "step": 4452
    },
    {
      "epoch": 0.08908,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.006884765625,
      "learning_rate": 0.0001,
      "loss": 4.7087,
      "loss/crossentropy": 2.4750468730926514,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27945323288440704,
      "step": 4454
    },
    {
      "epoch": 0.08912,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.007255045572916666,
      "learning_rate": 0.0001,
      "loss": 4.6822,
      "loss/crossentropy": 2.1766942739486694,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28881968557834625,
      "step": 4456
    },
    {
      "epoch": 0.08916,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.009065755208333333,
      "learning_rate": 0.0001,
      "loss": 4.9204,
      "loss/crossentropy": 2.265195846557617,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2616717368364334,
      "step": 4458
    },
    {
      "epoch": 0.0892,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.011473592122395833,
      "learning_rate": 0.0001,
      "loss": 4.5747,
      "loss/crossentropy": 2.2438716888427734,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2630545049905777,
      "step": 4460
    },
    {
      "epoch": 0.08924,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.012572224934895833,
      "learning_rate": 0.0001,
      "loss": 4.3404,
      "loss/crossentropy": 2.060324013233185,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23639824986457825,
      "step": 4462
    },
    {
      "epoch": 0.08928,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.0331207275390625,
      "learning_rate": 0.0001,
      "loss": 4.733,
      "loss/crossentropy": 1.8830525279045105,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21016474813222885,
      "step": 4464
    },
    {
      "epoch": 0.08932,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.03435872395833333,
      "learning_rate": 0.0001,
      "loss": 4.7571,
      "loss/crossentropy": 2.3420846462249756,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28108468651771545,
      "step": 4466
    },
    {
      "epoch": 0.08936,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.0337310791015625,
      "learning_rate": 0.0001,
      "loss": 4.6851,
      "loss/crossentropy": 1.9140342473983765,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22587041556835175,
      "step": 4468
    },
    {
      "epoch": 0.0894,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.032613118489583336,
      "learning_rate": 0.0001,
      "loss": 4.79,
      "loss/crossentropy": 1.9753122925758362,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24506878852844238,
      "step": 4470
    },
    {
      "epoch": 0.08944,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.032389322916666664,
      "learning_rate": 0.0001,
      "loss": 4.7769,
      "loss/crossentropy": 2.0735195875167847,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2763114273548126,
      "step": 4472
    },
    {
      "epoch": 0.08948,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.034403483072916664,
      "learning_rate": 0.0001,
      "loss": 4.3593,
      "loss/crossentropy": 1.7784460186958313,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22829323261976242,
      "step": 4474
    },
    {
      "epoch": 0.08952,
      "grad_norm": 2.625,
      "grad_norm_var": 0.051041666666666666,
      "learning_rate": 0.0001,
      "loss": 4.9451,
      "loss/crossentropy": 2.1188095808029175,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27410270273685455,
      "step": 4476
    },
    {
      "epoch": 0.08956,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.051634724934895834,
      "learning_rate": 0.0001,
      "loss": 4.3392,
      "loss/crossentropy": 2.320235252380371,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27514997124671936,
      "step": 4478
    },
    {
      "epoch": 0.0896,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.03766988118489583,
      "learning_rate": 0.0001,
      "loss": 4.8345,
      "loss/crossentropy": 2.3023080825805664,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2516366094350815,
      "step": 4480
    },
    {
      "epoch": 0.08964,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.039567057291666666,
      "learning_rate": 0.0001,
      "loss": 4.6698,
      "loss/crossentropy": 1.9677563905715942,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2362382560968399,
      "step": 4482
    },
    {
      "epoch": 0.08968,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.03951416015625,
      "learning_rate": 0.0001,
      "loss": 4.9159,
      "loss/crossentropy": 2.2005198001861572,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2760336175560951,
      "step": 4484
    },
    {
      "epoch": 0.08972,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0478912353515625,
      "learning_rate": 0.0001,
      "loss": 4.4418,
      "loss/crossentropy": 1.9799931049346924,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25132423639297485,
      "step": 4486
    },
    {
      "epoch": 0.08976,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.10212300618489584,
      "learning_rate": 0.0001,
      "loss": 4.9643,
      "loss/crossentropy": 2.205371141433716,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2695635259151459,
      "step": 4488
    },
    {
      "epoch": 0.0898,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.09692281087239583,
      "learning_rate": 0.0001,
      "loss": 4.9695,
      "loss/crossentropy": 2.3500062227249146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2736963629722595,
      "step": 4490
    },
    {
      "epoch": 0.08984,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.07867431640625,
      "learning_rate": 0.0001,
      "loss": 4.9083,
      "loss/crossentropy": 2.386352837085724,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2472890019416809,
      "step": 4492
    },
    {
      "epoch": 0.08988,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.07701416015625,
      "learning_rate": 0.0001,
      "loss": 4.3598,
      "loss/crossentropy": 1.9863982200622559,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2288871705532074,
      "step": 4494
    },
    {
      "epoch": 0.08992,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.07757059733072917,
      "learning_rate": 0.0001,
      "loss": 4.6157,
      "loss/crossentropy": 2.4088594913482666,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24304527044296265,
      "step": 4496
    },
    {
      "epoch": 0.08996,
      "grad_norm": 2.5,
      "grad_norm_var": 0.07517801920572917,
      "learning_rate": 0.0001,
      "loss": 4.7457,
      "loss/crossentropy": 2.1663140058517456,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2681227922439575,
      "step": 4498
    },
    {
      "epoch": 0.09,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.0810455322265625,
      "learning_rate": 0.0001,
      "loss": 4.2199,
      "loss/crossentropy": 1.9233570098876953,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23169977217912674,
      "step": 4500
    },
    {
      "epoch": 0.09004,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.07579752604166666,
      "learning_rate": 0.0001,
      "loss": 4.4677,
      "loss/crossentropy": 2.2940425872802734,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24809539318084717,
      "step": 4502
    },
    {
      "epoch": 0.09008,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.015771484375,
      "learning_rate": 0.0001,
      "loss": 4.5333,
      "loss/crossentropy": 2.1874176263809204,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26286616921424866,
      "step": 4504
    },
    {
      "epoch": 0.09012,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.015868123372395834,
      "learning_rate": 0.0001,
      "loss": 4.5029,
      "loss/crossentropy": 2.190543472766876,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26766398549079895,
      "step": 4506
    },
    {
      "epoch": 0.09016,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.010823567708333334,
      "learning_rate": 0.0001,
      "loss": 4.464,
      "loss/crossentropy": 2.132491707801819,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2514026165008545,
      "step": 4508
    },
    {
      "epoch": 0.0902,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.0107818603515625,
      "learning_rate": 0.0001,
      "loss": 4.5032,
      "loss/crossentropy": 2.1492353677749634,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25663119554519653,
      "step": 4510
    },
    {
      "epoch": 0.09024,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.011454264322916666,
      "learning_rate": 0.0001,
      "loss": 4.6166,
      "loss/crossentropy": 2.2471927404403687,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24812395125627518,
      "step": 4512
    },
    {
      "epoch": 0.09028,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.007013956705729167,
      "learning_rate": 0.0001,
      "loss": 4.5651,
      "loss/crossentropy": 2.1944304704666138,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26567137241363525,
      "step": 4514
    },
    {
      "epoch": 0.09032,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.006883748372395833,
      "learning_rate": 0.0001,
      "loss": 4.3054,
      "loss/crossentropy": 1.7537739872932434,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21784386038780212,
      "step": 4516
    },
    {
      "epoch": 0.09036,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.006981404622395834,
      "learning_rate": 0.0001,
      "loss": 4.6961,
      "loss/crossentropy": 2.133580207824707,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27865441143512726,
      "step": 4518
    },
    {
      "epoch": 0.0904,
      "grad_norm": 2.5,
      "grad_norm_var": 0.008723958333333334,
      "learning_rate": 0.0001,
      "loss": 4.5486,
      "loss/crossentropy": 2.0858315229415894,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26390860974788666,
      "step": 4520
    },
    {
      "epoch": 0.09044,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.008101399739583333,
      "learning_rate": 0.0001,
      "loss": 4.6833,
      "loss/crossentropy": 1.9084516763687134,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23013630509376526,
      "step": 4522
    },
    {
      "epoch": 0.09048,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.00859375,
      "learning_rate": 0.0001,
      "loss": 4.5817,
      "loss/crossentropy": 2.2123712301254272,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2739466577768326,
      "step": 4524
    },
    {
      "epoch": 0.09052,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.009691365559895833,
      "learning_rate": 0.0001,
      "loss": 4.9898,
      "loss/crossentropy": 2.3532934188842773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25509175658226013,
      "step": 4526
    },
    {
      "epoch": 0.09056,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.008137003580729166,
      "learning_rate": 0.0001,
      "loss": 4.7786,
      "loss/crossentropy": 2.1543048620224,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23639082163572311,
      "step": 4528
    },
    {
      "epoch": 0.0906,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.0109527587890625,
      "learning_rate": 0.0001,
      "loss": 4.647,
      "loss/crossentropy": 2.1322286128997803,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23874164372682571,
      "step": 4530
    },
    {
      "epoch": 0.09064,
      "grad_norm": 2.25,
      "grad_norm_var": 0.01011962890625,
      "learning_rate": 0.0001,
      "loss": 4.7623,
      "loss/crossentropy": 2.5552597045898438,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24949757009744644,
      "step": 4532
    },
    {
      "epoch": 0.09068,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.0117095947265625,
      "learning_rate": 0.0001,
      "loss": 4.6999,
      "loss/crossentropy": 2.1248819231987,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2795845717191696,
      "step": 4534
    },
    {
      "epoch": 0.09072,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.014655558268229167,
      "learning_rate": 0.0001,
      "loss": 4.7739,
      "loss/crossentropy": 1.985984206199646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22764238715171814,
      "step": 4536
    },
    {
      "epoch": 0.09076,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.01519775390625,
      "learning_rate": 0.0001,
      "loss": 4.64,
      "loss/crossentropy": 2.220720648765564,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2693287283182144,
      "step": 4538
    },
    {
      "epoch": 0.0908,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.01324462890625,
      "learning_rate": 0.0001,
      "loss": 4.486,
      "loss/crossentropy": 1.9541595578193665,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2419680804014206,
      "step": 4540
    },
    {
      "epoch": 0.09084,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.013916015625,
      "learning_rate": 0.0001,
      "loss": 4.963,
      "loss/crossentropy": 2.275113582611084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2562567666172981,
      "step": 4542
    },
    {
      "epoch": 0.09088,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.017122395833333335,
      "learning_rate": 0.0001,
      "loss": 4.7535,
      "loss/crossentropy": 2.4411803483963013,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25798996537923813,
      "step": 4544
    },
    {
      "epoch": 0.09092,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.019823201497395835,
      "learning_rate": 0.0001,
      "loss": 4.7339,
      "loss/crossentropy": 2.035769820213318,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25313758105039597,
      "step": 4546
    },
    {
      "epoch": 0.09096,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.018701171875,
      "learning_rate": 0.0001,
      "loss": 4.1331,
      "loss/crossentropy": 1.9237529039382935,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24456002563238144,
      "step": 4548
    },
    {
      "epoch": 0.091,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.02281494140625,
      "learning_rate": 0.0001,
      "loss": 4.6762,
      "loss/crossentropy": 2.179704189300537,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2681535929441452,
      "step": 4550
    },
    {
      "epoch": 0.09104,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.024137369791666665,
      "learning_rate": 0.0001,
      "loss": 4.6899,
      "loss/crossentropy": 2.013023316860199,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22643990069627762,
      "step": 4552
    },
    {
      "epoch": 0.09108,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.024201456705729166,
      "learning_rate": 0.0001,
      "loss": 4.6527,
      "loss/crossentropy": 2.173883557319641,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26495447754859924,
      "step": 4554
    },
    {
      "epoch": 0.09112,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.025121053059895832,
      "learning_rate": 0.0001,
      "loss": 4.5323,
      "loss/crossentropy": 1.9398870468139648,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24307211488485336,
      "step": 4556
    },
    {
      "epoch": 0.09116,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.025423177083333335,
      "learning_rate": 0.0001,
      "loss": 4.2028,
      "loss/crossentropy": 1.8551223874092102,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21659143269062042,
      "step": 4558
    },
    {
      "epoch": 0.0912,
      "grad_norm": 2.375,
      "grad_norm_var": 0.0254058837890625,
      "learning_rate": 0.0001,
      "loss": 4.7813,
      "loss/crossentropy": 2.104207456111908,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26515253633260727,
      "step": 4560
    },
    {
      "epoch": 0.09124,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.023875935872395834,
      "learning_rate": 0.0001,
      "loss": 4.6241,
      "loss/crossentropy": 2.31631863117218,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25454505532979965,
      "step": 4562
    },
    {
      "epoch": 0.09128,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.021800740559895834,
      "learning_rate": 0.0001,
      "loss": 4.5732,
      "loss/crossentropy": 2.331356406211853,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2667195200920105,
      "step": 4564
    },
    {
      "epoch": 0.09132,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.0176910400390625,
      "learning_rate": 0.0001,
      "loss": 4.6714,
      "loss/crossentropy": 2.2126184701919556,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24132181704044342,
      "step": 4566
    },
    {
      "epoch": 0.09136,
      "grad_norm": 2.25,
      "grad_norm_var": 0.013167317708333333,
      "learning_rate": 0.0001,
      "loss": 4.5753,
      "loss/crossentropy": 2.330659508705139,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2838585078716278,
      "step": 4568
    },
    {
      "epoch": 0.0914,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.0132720947265625,
      "learning_rate": 0.0001,
      "loss": 4.4935,
      "loss/crossentropy": 2.167214035987854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24021611362695694,
      "step": 4570
    },
    {
      "epoch": 0.09144,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.014046223958333333,
      "learning_rate": 0.0001,
      "loss": 4.5847,
      "loss/crossentropy": 1.770102322101593,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2108836993575096,
      "step": 4572
    },
    {
      "epoch": 0.09148,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.015425618489583333,
      "learning_rate": 0.0001,
      "loss": 4.2906,
      "loss/crossentropy": 1.9293717741966248,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2186967208981514,
      "step": 4574
    },
    {
      "epoch": 0.09152,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.014188639322916667,
      "learning_rate": 0.0001,
      "loss": 4.8128,
      "loss/crossentropy": 2.4099135398864746,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25587645173072815,
      "step": 4576
    },
    {
      "epoch": 0.09156,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.06220296223958333,
      "learning_rate": 0.0001,
      "loss": 4.5727,
      "loss/crossentropy": 1.7967870831489563,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2521464377641678,
      "step": 4578
    },
    {
      "epoch": 0.0916,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.0616851806640625,
      "learning_rate": 0.0001,
      "loss": 4.7604,
      "loss/crossentropy": 1.989583432674408,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22892683744430542,
      "step": 4580
    },
    {
      "epoch": 0.09164,
      "grad_norm": 2.625,
      "grad_norm_var": 0.0626953125,
      "learning_rate": 0.0001,
      "loss": 4.4986,
      "loss/crossentropy": 2.167198657989502,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24552703648805618,
      "step": 4582
    },
    {
      "epoch": 0.09168,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.06129150390625,
      "learning_rate": 0.0001,
      "loss": 4.8237,
      "loss/crossentropy": 2.213137984275818,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2614079788327217,
      "step": 4584
    },
    {
      "epoch": 0.09172,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.06194661458333333,
      "learning_rate": 0.0001,
      "loss": 4.977,
      "loss/crossentropy": 2.3586392998695374,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2527218610048294,
      "step": 4586
    },
    {
      "epoch": 0.09176,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.06054280598958333,
      "learning_rate": 0.0001,
      "loss": 4.5118,
      "loss/crossentropy": 2.326598286628723,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2684827446937561,
      "step": 4588
    },
    {
      "epoch": 0.0918,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.055562337239583336,
      "learning_rate": 0.0001,
      "loss": 4.4296,
      "loss/crossentropy": 2.1365907192230225,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2705482095479965,
      "step": 4590
    },
    {
      "epoch": 0.09184,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.056005859375,
      "learning_rate": 0.0001,
      "loss": 4.6895,
      "loss/crossentropy": 1.816661775112152,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2188858687877655,
      "step": 4592
    },
    {
      "epoch": 0.09188,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.020731608072916668,
      "learning_rate": 0.0001,
      "loss": 4.5953,
      "loss/crossentropy": 2.1533923149108887,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2372177392244339,
      "step": 4594
    },
    {
      "epoch": 0.09192,
      "grad_norm": 2.984375,
      "grad_norm_var": 0.04504801432291667,
      "learning_rate": 0.0001,
      "loss": 4.9472,
      "loss/crossentropy": 2.2175614833831787,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27555912733078003,
      "step": 4596
    },
    {
      "epoch": 0.09196,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.04644775390625,
      "learning_rate": 0.0001,
      "loss": 5.11,
      "loss/crossentropy": 2.583792209625244,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2970864772796631,
      "step": 4598
    },
    {
      "epoch": 0.092,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.046751912434895834,
      "learning_rate": 0.0001,
      "loss": 4.5761,
      "loss/crossentropy": 2.227868676185608,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.259210504591465,
      "step": 4600
    },
    {
      "epoch": 0.09204,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.047118123372395834,
      "learning_rate": 0.0001,
      "loss": 4.4125,
      "loss/crossentropy": 2.146941900253296,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26443855464458466,
      "step": 4602
    },
    {
      "epoch": 0.09208,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.04439697265625,
      "learning_rate": 0.0001,
      "loss": 4.8655,
      "loss/crossentropy": 2.129795551300049,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2451881766319275,
      "step": 4604
    },
    {
      "epoch": 0.09212,
      "grad_norm": 2.8125,
      "grad_norm_var": 0.0506256103515625,
      "learning_rate": 0.0001,
      "loss": 4.895,
      "loss/crossentropy": 2.2696213722229004,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26037923991680145,
      "step": 4606
    },
    {
      "epoch": 0.09216,
      "grad_norm": 3.03125,
      "grad_norm_var": 0.07105712890625,
      "learning_rate": 0.0001,
      "loss": 4.7424,
      "loss/crossentropy": 2.1916056871414185,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2648691013455391,
      "step": 4608
    },
    {
      "epoch": 0.0922,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.07119038899739584,
      "learning_rate": 0.0001,
      "loss": 4.5021,
      "loss/crossentropy": 1.975761890411377,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22248996049165726,
      "step": 4610
    },
    {
      "epoch": 0.09224,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.053511555989583334,
      "learning_rate": 0.0001,
      "loss": 4.6596,
      "loss/crossentropy": 2.403375506401062,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24553030729293823,
      "step": 4612
    },
    {
      "epoch": 0.09228,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.049637858072916666,
      "learning_rate": 0.0001,
      "loss": 4.4542,
      "loss/crossentropy": 1.9468475580215454,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24023501574993134,
      "step": 4614
    },
    {
      "epoch": 0.09232,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.0534088134765625,
      "learning_rate": 0.0001,
      "loss": 4.8095,
      "loss/crossentropy": 2.013557195663452,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23829826712608337,
      "step": 4616
    },
    {
      "epoch": 0.09236,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.0519439697265625,
      "learning_rate": 0.0001,
      "loss": 4.5736,
      "loss/crossentropy": 2.0552549958229065,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2391991689801216,
      "step": 4618
    },
    {
      "epoch": 0.0924,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.053694661458333334,
      "learning_rate": 0.0001,
      "loss": 4.4091,
      "loss/crossentropy": 1.8311110734939575,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23821169883012772,
      "step": 4620
    },
    {
      "epoch": 0.09244,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.04431966145833333,
      "learning_rate": 0.0001,
      "loss": 4.5034,
      "loss/crossentropy": 2.0346454977989197,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23688799142837524,
      "step": 4622
    },
    {
      "epoch": 0.09248,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.016304524739583333,
      "learning_rate": 0.0001,
      "loss": 4.5329,
      "loss/crossentropy": 1.8475716710090637,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23444947600364685,
      "step": 4624
    },
    {
      "epoch": 0.09252,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.0136383056640625,
      "learning_rate": 0.0001,
      "loss": 4.4315,
      "loss/crossentropy": 2.075626790523529,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2663211151957512,
      "step": 4626
    },
    {
      "epoch": 0.09256,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.0130035400390625,
      "learning_rate": 0.0001,
      "loss": 4.9214,
      "loss/crossentropy": 2.413028359413147,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29516373574733734,
      "step": 4628
    },
    {
      "epoch": 0.0926,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.014143880208333333,
      "learning_rate": 0.0001,
      "loss": 4.6878,
      "loss/crossentropy": 1.8549358248710632,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23275860399007797,
      "step": 4630
    },
    {
      "epoch": 0.09264,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.0076405843098958336,
      "learning_rate": 0.0001,
      "loss": 4.8594,
      "loss/crossentropy": 2.273250460624695,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2511187419295311,
      "step": 4632
    },
    {
      "epoch": 0.09268,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.013093058268229167,
      "learning_rate": 0.0001,
      "loss": 4.336,
      "loss/crossentropy": 1.929758369922638,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25462278723716736,
      "step": 4634
    },
    {
      "epoch": 0.09272,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.013158162434895834,
      "learning_rate": 0.0001,
      "loss": 4.5888,
      "loss/crossentropy": 2.0929598212242126,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23592843115329742,
      "step": 4636
    },
    {
      "epoch": 0.09276,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.015282185872395833,
      "learning_rate": 0.0001,
      "loss": 4.4595,
      "loss/crossentropy": 1.973824143409729,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24049720913171768,
      "step": 4638
    },
    {
      "epoch": 0.0928,
      "grad_norm": 2.25,
      "grad_norm_var": 0.014207967122395833,
      "learning_rate": 0.0001,
      "loss": 4.2606,
      "loss/crossentropy": 1.8606626987457275,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21698658913373947,
      "step": 4640
    },
    {
      "epoch": 0.09284,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.01875,
      "learning_rate": 0.0001,
      "loss": 4.9827,
      "loss/crossentropy": 2.241386890411377,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2831972986459732,
      "step": 4642
    },
    {
      "epoch": 0.09288,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.0166656494140625,
      "learning_rate": 0.0001,
      "loss": 4.7046,
      "loss/crossentropy": 2.2538920640945435,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2482161968946457,
      "step": 4644
    },
    {
      "epoch": 0.09292,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.01812744140625,
      "learning_rate": 0.0001,
      "loss": 4.8557,
      "loss/crossentropy": 2.067206382751465,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26665763556957245,
      "step": 4646
    },
    {
      "epoch": 0.09296,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.021751912434895833,
      "learning_rate": 0.0001,
      "loss": 4.9474,
      "loss/crossentropy": 2.33401358127594,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2752760946750641,
      "step": 4648
    },
    {
      "epoch": 0.093,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.04914449055989583,
      "learning_rate": 0.0001,
      "loss": 4.8658,
      "loss/crossentropy": 2.1510735750198364,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24588150531053543,
      "step": 4650
    },
    {
      "epoch": 0.09304,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.049397786458333336,
      "learning_rate": 0.0001,
      "loss": 4.579,
      "loss/crossentropy": 2.086448848247528,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.261296346783638,
      "step": 4652
    },
    {
      "epoch": 0.09308,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.0446685791015625,
      "learning_rate": 0.0001,
      "loss": 4.7495,
      "loss/crossentropy": 1.5951193571090698,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2312234491109848,
      "step": 4654
    },
    {
      "epoch": 0.09312,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.04052734375,
      "learning_rate": 0.0001,
      "loss": 4.2273,
      "loss/crossentropy": 2.0098360776901245,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22354383766651154,
      "step": 4656
    },
    {
      "epoch": 0.09316,
      "grad_norm": 2.375,
      "grad_norm_var": 0.04079488118489583,
      "learning_rate": 0.0001,
      "loss": 4.6047,
      "loss/crossentropy": 2.263219714164734,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26900506019592285,
      "step": 4658
    },
    {
      "epoch": 0.0932,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.0485504150390625,
      "learning_rate": 0.0001,
      "loss": 4.5575,
      "loss/crossentropy": 1.9481555819511414,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24665354192256927,
      "step": 4660
    },
    {
      "epoch": 0.09324,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.04988505045572917,
      "learning_rate": 0.0001,
      "loss": 4.5895,
      "loss/crossentropy": 1.8751367926597595,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24356309324502945,
      "step": 4662
    },
    {
      "epoch": 0.09328,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.06110026041666667,
      "learning_rate": 0.0001,
      "loss": 4.7739,
      "loss/crossentropy": 2.343896746635437,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2558425962924957,
      "step": 4664
    },
    {
      "epoch": 0.09332,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.032811482747395836,
      "learning_rate": 0.0001,
      "loss": 4.779,
      "loss/crossentropy": 2.067797303199768,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25030098110437393,
      "step": 4666
    },
    {
      "epoch": 0.09336,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.029816691080729166,
      "learning_rate": 0.0001,
      "loss": 4.7427,
      "loss/crossentropy": 2.4221161603927612,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26934675872325897,
      "step": 4668
    },
    {
      "epoch": 0.0934,
      "grad_norm": 2.8125,
      "grad_norm_var": 0.038182576497395836,
      "learning_rate": 0.0001,
      "loss": 4.845,
      "loss/crossentropy": 2.2733672857284546,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2536270022392273,
      "step": 4670
    },
    {
      "epoch": 0.09344,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.0369537353515625,
      "learning_rate": 0.0001,
      "loss": 4.7721,
      "loss/crossentropy": 2.179157257080078,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24323320388793945,
      "step": 4672
    },
    {
      "epoch": 0.09348,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.03875325520833333,
      "learning_rate": 0.0001,
      "loss": 4.8574,
      "loss/crossentropy": 2.3613970279693604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2567686140537262,
      "step": 4674
    },
    {
      "epoch": 0.09352,
      "grad_norm": 2.5,
      "grad_norm_var": 0.029683430989583332,
      "learning_rate": 0.0001,
      "loss": 4.787,
      "loss/crossentropy": 2.0755810141563416,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2349473536014557,
      "step": 4676
    },
    {
      "epoch": 0.09356,
      "grad_norm": 2.25,
      "grad_norm_var": 0.03134358723958333,
      "learning_rate": 0.0001,
      "loss": 4.6235,
      "loss/crossentropy": 1.9971619248390198,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22000454366207123,
      "step": 4678
    },
    {
      "epoch": 0.0936,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.020068359375,
      "learning_rate": 0.0001,
      "loss": 4.5307,
      "loss/crossentropy": 1.9419977068901062,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21052303910255432,
      "step": 4680
    },
    {
      "epoch": 0.09364,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.021122233072916666,
      "learning_rate": 0.0001,
      "loss": 4.6314,
      "loss/crossentropy": 2.1175334453582764,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23567666858434677,
      "step": 4682
    },
    {
      "epoch": 0.09368,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.021122233072916666,
      "learning_rate": 0.0001,
      "loss": 4.5955,
      "loss/crossentropy": 2.0300605297088623,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29177258908748627,
      "step": 4684
    },
    {
      "epoch": 0.09372,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.009228515625,
      "learning_rate": 0.0001,
      "loss": 4.5583,
      "loss/crossentropy": 2.052473723888397,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22480525821447372,
      "step": 4686
    },
    {
      "epoch": 0.09376,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.008275349934895834,
      "learning_rate": 0.0001,
      "loss": 4.4925,
      "loss/crossentropy": 2.2506834268569946,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26876674592494965,
      "step": 4688
    },
    {
      "epoch": 0.0938,
      "grad_norm": 2.125,
      "grad_norm_var": 0.007515462239583334,
      "learning_rate": 0.0001,
      "loss": 4.291,
      "loss/crossentropy": 1.930393099784851,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22246946394443512,
      "step": 4690
    },
    {
      "epoch": 0.09384,
      "grad_norm": 2.5,
      "grad_norm_var": 0.007616170247395833,
      "learning_rate": 0.0001,
      "loss": 4.9317,
      "loss/crossentropy": 2.5477651357650757,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27808643877506256,
      "step": 4692
    },
    {
      "epoch": 0.09388,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.009033203125,
      "learning_rate": 0.0001,
      "loss": 4.5313,
      "loss/crossentropy": 2.1059221625328064,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.253792941570282,
      "step": 4694
    },
    {
      "epoch": 0.09392,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.0091217041015625,
      "learning_rate": 0.0001,
      "loss": 4.607,
      "loss/crossentropy": 2.0058358907699585,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23942459374666214,
      "step": 4696
    },
    {
      "epoch": 0.09396,
      "grad_norm": 2.25,
      "grad_norm_var": 0.0103515625,
      "learning_rate": 0.0001,
      "loss": 4.2024,
      "loss/crossentropy": 1.9118947982788086,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2300974577665329,
      "step": 4698
    },
    {
      "epoch": 0.094,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.013337198893229167,
      "learning_rate": 0.0001,
      "loss": 4.6541,
      "loss/crossentropy": 2.1172733902931213,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24785596132278442,
      "step": 4700
    },
    {
      "epoch": 0.09404,
      "grad_norm": 2.5,
      "grad_norm_var": 0.016044108072916667,
      "learning_rate": 0.0001,
      "loss": 4.9005,
      "loss/crossentropy": 2.0064170956611633,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25045372545719147,
      "step": 4702
    },
    {
      "epoch": 0.09408,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.015738932291666667,
      "learning_rate": 0.0001,
      "loss": 4.8139,
      "loss/crossentropy": 2.5816656351089478,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27938568592071533,
      "step": 4704
    },
    {
      "epoch": 0.09412,
      "grad_norm": 2.25,
      "grad_norm_var": 0.016597493489583334,
      "learning_rate": 0.0001,
      "loss": 4.866,
      "loss/crossentropy": 2.5768171548843384,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27975398302078247,
      "step": 4706
    },
    {
      "epoch": 0.09416,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.0153717041015625,
      "learning_rate": 0.0001,
      "loss": 4.6075,
      "loss/crossentropy": 2.323893189430237,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2608431279659271,
      "step": 4708
    },
    {
      "epoch": 0.0942,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.01357421875,
      "learning_rate": 0.0001,
      "loss": 4.3564,
      "loss/crossentropy": 1.7910810708999634,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21380099654197693,
      "step": 4710
    },
    {
      "epoch": 0.09424,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.015848795572916668,
      "learning_rate": 0.0001,
      "loss": 4.6374,
      "loss/crossentropy": 2.214052677154541,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24833911657333374,
      "step": 4712
    },
    {
      "epoch": 0.09428,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.0158355712890625,
      "learning_rate": 0.0001,
      "loss": 4.9987,
      "loss/crossentropy": 2.0850380063056946,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2419341504573822,
      "step": 4714
    },
    {
      "epoch": 0.09432,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.014058430989583334,
      "learning_rate": 0.0001,
      "loss": 4.2961,
      "loss/crossentropy": 2.0707927346229553,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2499033510684967,
      "step": 4716
    },
    {
      "epoch": 0.09436,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.0125396728515625,
      "learning_rate": 0.0001,
      "loss": 4.4057,
      "loss/crossentropy": 2.11221444606781,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25916673243045807,
      "step": 4718
    },
    {
      "epoch": 0.0944,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.012718709309895833,
      "learning_rate": 0.0001,
      "loss": 4.6057,
      "loss/crossentropy": 1.8400230407714844,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22771906107664108,
      "step": 4720
    },
    {
      "epoch": 0.09444,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.008234659830729166,
      "learning_rate": 0.0001,
      "loss": 4.4929,
      "loss/crossentropy": 2.2271865606307983,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2574647441506386,
      "step": 4722
    },
    {
      "epoch": 0.09448,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.008649698893229167,
      "learning_rate": 0.0001,
      "loss": 4.4629,
      "loss/crossentropy": 2.4532920122146606,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27140843868255615,
      "step": 4724
    },
    {
      "epoch": 0.09452,
      "grad_norm": 2.375,
      "grad_norm_var": 0.009504191080729167,
      "learning_rate": 0.0001,
      "loss": 4.731,
      "loss/crossentropy": 1.951303780078888,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2333502620458603,
      "step": 4726
    },
    {
      "epoch": 0.09456,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.009137980143229167,
      "learning_rate": 0.0001,
      "loss": 4.6334,
      "loss/crossentropy": 2.209821343421936,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24776015430688858,
      "step": 4728
    },
    {
      "epoch": 0.0946,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.010872395833333333,
      "learning_rate": 0.0001,
      "loss": 4.861,
      "loss/crossentropy": 2.434941530227661,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2747645229101181,
      "step": 4730
    },
    {
      "epoch": 0.09464,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.010602823893229167,
      "learning_rate": 0.0001,
      "loss": 4.6145,
      "loss/crossentropy": 2.051652252674103,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24515582621097565,
      "step": 4732
    },
    {
      "epoch": 0.09468,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.009797159830729167,
      "learning_rate": 0.0001,
      "loss": 4.6804,
      "loss/crossentropy": 2.0039377212524414,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23073314130306244,
      "step": 4734
    },
    {
      "epoch": 0.09472,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.19903971354166666,
      "learning_rate": 0.0001,
      "loss": 4.8333,
      "loss/crossentropy": 2.166410982608795,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25871724635362625,
      "step": 4736
    },
    {
      "epoch": 0.09476,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.2066802978515625,
      "learning_rate": 0.0001,
      "loss": 4.2739,
      "loss/crossentropy": 1.9289153218269348,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23183659464120865,
      "step": 4738
    },
    {
      "epoch": 0.0948,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.20640360514322917,
      "learning_rate": 0.0001,
      "loss": 4.5903,
      "loss/crossentropy": 2.270771861076355,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26335832476615906,
      "step": 4740
    },
    {
      "epoch": 0.09484,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.206494140625,
      "learning_rate": 0.0001,
      "loss": 4.7142,
      "loss/crossentropy": 2.395651936531067,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28527122735977173,
      "step": 4742
    },
    {
      "epoch": 0.09488,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.20437723795572918,
      "learning_rate": 0.0001,
      "loss": 4.5083,
      "loss/crossentropy": 1.8597867488861084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.224149189889431,
      "step": 4744
    },
    {
      "epoch": 0.09492,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.20693257649739583,
      "learning_rate": 0.0001,
      "loss": 4.4415,
      "loss/crossentropy": 1.7795116305351257,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23233074694871902,
      "step": 4746
    },
    {
      "epoch": 0.09496,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.20255533854166666,
      "learning_rate": 0.0001,
      "loss": 4.7749,
      "loss/crossentropy": 1.9449282884597778,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2524118423461914,
      "step": 4748
    },
    {
      "epoch": 0.095,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.2005035400390625,
      "learning_rate": 0.0001,
      "loss": 4.7047,
      "loss/crossentropy": 2.195169448852539,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2349853590130806,
      "step": 4750
    },
    {
      "epoch": 0.09504,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.022468058268229167,
      "learning_rate": 0.0001,
      "loss": 4.5268,
      "loss/crossentropy": 1.6628928184509277,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1975274682044983,
      "step": 4752
    },
    {
      "epoch": 0.09508,
      "grad_norm": 2.25,
      "grad_norm_var": 0.018488566080729168,
      "learning_rate": 0.0001,
      "loss": 4.5862,
      "loss/crossentropy": 2.0991236567497253,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23341374844312668,
      "step": 4754
    },
    {
      "epoch": 0.09512,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.017024739583333334,
      "learning_rate": 0.0001,
      "loss": 4.8338,
      "loss/crossentropy": 2.350286066532135,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2648291736841202,
      "step": 4756
    },
    {
      "epoch": 0.09516,
      "grad_norm": 2.375,
      "grad_norm_var": 0.017430623372395832,
      "learning_rate": 0.0001,
      "loss": 4.5788,
      "loss/crossentropy": 2.02384877204895,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24653150886297226,
      "step": 4758
    },
    {
      "epoch": 0.0952,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.01734619140625,
      "learning_rate": 0.0001,
      "loss": 4.7052,
      "loss/crossentropy": 2.264349341392517,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25881223380565643,
      "step": 4760
    },
    {
      "epoch": 0.09524,
      "grad_norm": 2.890625,
      "grad_norm_var": 0.03779296875,
      "learning_rate": 0.0001,
      "loss": 4.8216,
      "loss/crossentropy": 2.197356939315796,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2565019279718399,
      "step": 4762
    },
    {
      "epoch": 0.09528,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.038655598958333336,
      "learning_rate": 0.0001,
      "loss": 4.3849,
      "loss/crossentropy": 1.8794787526130676,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21771979331970215,
      "step": 4764
    },
    {
      "epoch": 0.09532,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.03795572916666667,
      "learning_rate": 0.0001,
      "loss": 4.7391,
      "loss/crossentropy": 2.107520580291748,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2350333333015442,
      "step": 4766
    },
    {
      "epoch": 0.09536,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.03297119140625,
      "learning_rate": 0.0001,
      "loss": 4.5194,
      "loss/crossentropy": 2.082249402999878,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24860269576311111,
      "step": 4768
    },
    {
      "epoch": 0.0954,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.03062744140625,
      "learning_rate": 0.0001,
      "loss": 4.3985,
      "loss/crossentropy": 1.9632289409637451,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24099770188331604,
      "step": 4770
    },
    {
      "epoch": 0.09544,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.02994384765625,
      "learning_rate": 0.0001,
      "loss": 4.4526,
      "loss/crossentropy": 2.3403327465057373,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2573448717594147,
      "step": 4772
    },
    {
      "epoch": 0.09548,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.03181050618489583,
      "learning_rate": 0.0001,
      "loss": 4.7711,
      "loss/crossentropy": 2.302455425262451,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28050975501537323,
      "step": 4774
    },
    {
      "epoch": 0.09552,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.031103515625,
      "learning_rate": 0.0001,
      "loss": 4.6092,
      "loss/crossentropy": 2.2875213623046875,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23910009860992432,
      "step": 4776
    },
    {
      "epoch": 0.09556,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.022721354166666666,
      "learning_rate": 0.0001,
      "loss": 4.5237,
      "loss/crossentropy": 2.0440263748168945,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24640783667564392,
      "step": 4778
    },
    {
      "epoch": 0.0956,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.02613525390625,
      "learning_rate": 0.0001,
      "loss": 4.6328,
      "loss/crossentropy": 2.1597142219543457,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24566050618886948,
      "step": 4780
    },
    {
      "epoch": 0.09564,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.027765909830729168,
      "learning_rate": 0.0001,
      "loss": 4.7138,
      "loss/crossentropy": 2.2846235036849976,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2598261833190918,
      "step": 4782
    },
    {
      "epoch": 0.09568,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.029683430989583332,
      "learning_rate": 0.0001,
      "loss": 4.4527,
      "loss/crossentropy": 2.032243251800537,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2488306611776352,
      "step": 4784
    },
    {
      "epoch": 0.09572,
      "grad_norm": 2.5,
      "grad_norm_var": 0.037230428059895834,
      "learning_rate": 0.0001,
      "loss": 4.7651,
      "loss/crossentropy": 2.4999172687530518,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25949685275554657,
      "step": 4786
    },
    {
      "epoch": 0.09576,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.0375152587890625,
      "learning_rate": 0.0001,
      "loss": 4.6068,
      "loss/crossentropy": 2.1610575914382935,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2414923906326294,
      "step": 4788
    },
    {
      "epoch": 0.0958,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.03135477701822917,
      "learning_rate": 0.0001,
      "loss": 4.5309,
      "loss/crossentropy": 1.8679919838905334,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21853071451187134,
      "step": 4790
    },
    {
      "epoch": 0.09584,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.031281534830729166,
      "learning_rate": 0.0001,
      "loss": 4.697,
      "loss/crossentropy": 2.072615623474121,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24943216145038605,
      "step": 4792
    },
    {
      "epoch": 0.09588,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.032624308268229166,
      "learning_rate": 0.0001,
      "loss": 4.5555,
      "loss/crossentropy": 2.420115113258362,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2840816229581833,
      "step": 4794
    },
    {
      "epoch": 0.09592,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.0455078125,
      "learning_rate": 0.0001,
      "loss": 5.102,
      "loss/crossentropy": 2.2809360027313232,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2739071249961853,
      "step": 4796
    },
    {
      "epoch": 0.09596,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.04319559733072917,
      "learning_rate": 0.0001,
      "loss": 4.5418,
      "loss/crossentropy": 2.1809465289115906,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24680403620004654,
      "step": 4798
    },
    {
      "epoch": 0.096,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.044611612955729164,
      "learning_rate": 0.0001,
      "loss": 4.6994,
      "loss/crossentropy": 1.9710316061973572,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2598233222961426,
      "step": 4800
    },
    {
      "epoch": 0.09604,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.035477701822916666,
      "learning_rate": 0.0001,
      "loss": 4.9811,
      "loss/crossentropy": 2.4378572702407837,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2665309011936188,
      "step": 4802
    },
    {
      "epoch": 0.09608,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.03759358723958333,
      "learning_rate": 0.0001,
      "loss": 4.604,
      "loss/crossentropy": 1.8578800559043884,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24051420390605927,
      "step": 4804
    },
    {
      "epoch": 0.09612,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.034520467122395836,
      "learning_rate": 0.0001,
      "loss": 4.4567,
      "loss/crossentropy": 2.191601037979126,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25017624348402023,
      "step": 4806
    },
    {
      "epoch": 0.09616,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.038374837239583334,
      "learning_rate": 0.0001,
      "loss": 4.5777,
      "loss/crossentropy": 2.2077550888061523,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31426818668842316,
      "step": 4808
    },
    {
      "epoch": 0.0962,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.030659993489583332,
      "learning_rate": 0.0001,
      "loss": 4.917,
      "loss/crossentropy": 2.2983503341674805,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26998060941696167,
      "step": 4810
    },
    {
      "epoch": 0.09624,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.020824178059895834,
      "learning_rate": 0.0001,
      "loss": 4.8691,
      "loss/crossentropy": 2.0875505208969116,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2673826217651367,
      "step": 4812
    },
    {
      "epoch": 0.09628,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.031403605143229166,
      "learning_rate": 0.0001,
      "loss": 4.0736,
      "loss/crossentropy": 1.9439318776130676,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2379719465970993,
      "step": 4814
    },
    {
      "epoch": 0.09632,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.03611551920572917,
      "learning_rate": 0.0001,
      "loss": 4.4698,
      "loss/crossentropy": 1.9822518229484558,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24425340443849564,
      "step": 4816
    },
    {
      "epoch": 0.09636,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.03902994791666667,
      "learning_rate": 0.0001,
      "loss": 4.475,
      "loss/crossentropy": 2.0238336324691772,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2409110590815544,
      "step": 4818
    },
    {
      "epoch": 0.0964,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.03707275390625,
      "learning_rate": 0.0001,
      "loss": 4.5401,
      "loss/crossentropy": 2.454450249671936,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26127950847148895,
      "step": 4820
    },
    {
      "epoch": 0.09644,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.038407389322916666,
      "learning_rate": 0.0001,
      "loss": 4.5682,
      "loss/crossentropy": 2.210579752922058,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23662539571523666,
      "step": 4822
    },
    {
      "epoch": 0.09648,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.034601847330729164,
      "learning_rate": 0.0001,
      "loss": 4.5527,
      "loss/crossentropy": 1.8737664222717285,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22884630411863327,
      "step": 4824
    },
    {
      "epoch": 0.09652,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.0335357666015625,
      "learning_rate": 0.0001,
      "loss": 4.6221,
      "loss/crossentropy": 2.1939562559127808,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25145241618156433,
      "step": 4826
    },
    {
      "epoch": 0.09656,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.0278472900390625,
      "learning_rate": 0.0001,
      "loss": 4.6685,
      "loss/crossentropy": 2.133625030517578,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25248992443084717,
      "step": 4828
    },
    {
      "epoch": 0.0966,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.0197906494140625,
      "learning_rate": 0.0001,
      "loss": 4.2854,
      "loss/crossentropy": 2.259738326072693,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24727293848991394,
      "step": 4830
    },
    {
      "epoch": 0.09664,
      "grad_norm": 2.984375,
      "grad_norm_var": 0.043635050455729164,
      "learning_rate": 0.0001,
      "loss": 4.7497,
      "loss/crossentropy": 2.0186268091201782,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27940231561660767,
      "step": 4832
    },
    {
      "epoch": 0.09668,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.04185282389322917,
      "learning_rate": 0.0001,
      "loss": 4.5867,
      "loss/crossentropy": 1.9686395525932312,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23021705448627472,
      "step": 4834
    },
    {
      "epoch": 0.09672,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.0424224853515625,
      "learning_rate": 0.0001,
      "loss": 4.6157,
      "loss/crossentropy": 2.1677820682525635,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2766892910003662,
      "step": 4836
    },
    {
      "epoch": 0.09676,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.04670308430989583,
      "learning_rate": 0.0001,
      "loss": 4.3751,
      "loss/crossentropy": 1.8458788990974426,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23294126987457275,
      "step": 4838
    },
    {
      "epoch": 0.0968,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.0495269775390625,
      "learning_rate": 0.0001,
      "loss": 4.8931,
      "loss/crossentropy": 1.7898097038269043,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23436476290225983,
      "step": 4840
    },
    {
      "epoch": 0.09684,
      "grad_norm": 2.375,
      "grad_norm_var": 0.04843343098958333,
      "learning_rate": 0.0001,
      "loss": 4.5169,
      "loss/crossentropy": 2.4594497680664062,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2618473023176193,
      "step": 4842
    },
    {
      "epoch": 0.09688,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.04752197265625,
      "learning_rate": 0.0001,
      "loss": 4.4841,
      "loss/crossentropy": 2.061558425426483,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2523345798254013,
      "step": 4844
    },
    {
      "epoch": 0.09692,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.04462890625,
      "learning_rate": 0.0001,
      "loss": 4.5606,
      "loss/crossentropy": 2.060658574104309,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2493506520986557,
      "step": 4846
    },
    {
      "epoch": 0.09696,
      "grad_norm": 2.25,
      "grad_norm_var": 0.016169230143229168,
      "learning_rate": 0.0001,
      "loss": 4.8028,
      "loss/crossentropy": 2.201029062271118,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2629493921995163,
      "step": 4848
    },
    {
      "epoch": 0.097,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.01519775390625,
      "learning_rate": 0.0001,
      "loss": 4.7404,
      "loss/crossentropy": 2.199273705482483,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27054519951343536,
      "step": 4850
    },
    {
      "epoch": 0.09704,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0189453125,
      "learning_rate": 0.0001,
      "loss": 4.3008,
      "loss/crossentropy": 2.2466784715652466,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2678111642599106,
      "step": 4852
    },
    {
      "epoch": 0.09708,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.022054036458333332,
      "learning_rate": 0.0001,
      "loss": 4.6762,
      "loss/crossentropy": 1.9152815341949463,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2441672906279564,
      "step": 4854
    },
    {
      "epoch": 0.09712,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.017577107747395834,
      "learning_rate": 0.0001,
      "loss": 4.3818,
      "loss/crossentropy": 2.092438578605652,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24061349034309387,
      "step": 4856
    },
    {
      "epoch": 0.09716,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.01802978515625,
      "learning_rate": 0.0001,
      "loss": 4.3224,
      "loss/crossentropy": 1.8620384335517883,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23738765716552734,
      "step": 4858
    },
    {
      "epoch": 0.0972,
      "grad_norm": 2.953125,
      "grad_norm_var": 0.04138997395833333,
      "learning_rate": 0.0001,
      "loss": 4.8798,
      "loss/crossentropy": 2.199701428413391,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26521213352680206,
      "step": 4860
    },
    {
      "epoch": 0.09724,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.042170206705729164,
      "learning_rate": 0.0001,
      "loss": 4.7598,
      "loss/crossentropy": 2.3637821674346924,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2707534506917,
      "step": 4862
    },
    {
      "epoch": 0.09728,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.045491536458333336,
      "learning_rate": 0.0001,
      "loss": 4.5184,
      "loss/crossentropy": 2.207235813140869,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28080131858587265,
      "step": 4864
    },
    {
      "epoch": 0.09732,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.0466461181640625,
      "learning_rate": 0.0001,
      "loss": 4.3798,
      "loss/crossentropy": 1.901595950126648,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23459318280220032,
      "step": 4866
    },
    {
      "epoch": 0.09736,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.042769368489583334,
      "learning_rate": 0.0001,
      "loss": 4.4289,
      "loss/crossentropy": 2.298312544822693,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2580094337463379,
      "step": 4868
    },
    {
      "epoch": 0.0974,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.03658854166666667,
      "learning_rate": 0.0001,
      "loss": 4.7576,
      "loss/crossentropy": 2.29680597782135,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24749226868152618,
      "step": 4870
    },
    {
      "epoch": 0.09744,
      "grad_norm": 2.25,
      "grad_norm_var": 0.0361968994140625,
      "learning_rate": 0.0001,
      "loss": 4.5839,
      "loss/crossentropy": 2.1169378757476807,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2565944790840149,
      "step": 4872
    },
    {
      "epoch": 0.09748,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.04035542805989583,
      "learning_rate": 0.0001,
      "loss": 4.5364,
      "loss/crossentropy": 2.0863184928894043,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2527791038155556,
      "step": 4874
    },
    {
      "epoch": 0.09752,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.015625,
      "learning_rate": 0.0001,
      "loss": 4.6176,
      "loss/crossentropy": 2.146193563938141,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23968011140823364,
      "step": 4876
    },
    {
      "epoch": 0.09756,
      "grad_norm": 2.375,
      "grad_norm_var": 0.01060791015625,
      "learning_rate": 0.0001,
      "loss": 4.5598,
      "loss/crossentropy": 2.1125508546829224,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2498009204864502,
      "step": 4878
    },
    {
      "epoch": 0.0976,
      "grad_norm": 2.375,
      "grad_norm_var": 0.004130045572916667,
      "learning_rate": 0.0001,
      "loss": 4.772,
      "loss/crossentropy": 2.13166081905365,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2640175372362137,
      "step": 4880
    },
    {
      "epoch": 0.09764,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.0051910400390625,
      "learning_rate": 0.0001,
      "loss": 4.5276,
      "loss/crossentropy": 2.047860622406006,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23980721086263657,
      "step": 4882
    },
    {
      "epoch": 0.09768,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.010477701822916666,
      "learning_rate": 0.0001,
      "loss": 4.8519,
      "loss/crossentropy": 2.362569808959961,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27908293902873993,
      "step": 4884
    },
    {
      "epoch": 0.09772,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.012398274739583333,
      "learning_rate": 0.0001,
      "loss": 4.2709,
      "loss/crossentropy": 1.8781500458717346,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22341010719537735,
      "step": 4886
    },
    {
      "epoch": 0.09776,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.019807942708333335,
      "learning_rate": 0.0001,
      "loss": 4.9408,
      "loss/crossentropy": 2.1138893365859985,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2533891350030899,
      "step": 4888
    },
    {
      "epoch": 0.0978,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.0188140869140625,
      "learning_rate": 0.0001,
      "loss": 4.5981,
      "loss/crossentropy": 2.2212090492248535,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.279159352183342,
      "step": 4890
    },
    {
      "epoch": 0.09784,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.0177886962890625,
      "learning_rate": 0.0001,
      "loss": 4.7245,
      "loss/crossentropy": 1.9747707843780518,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24078013002872467,
      "step": 4892
    },
    {
      "epoch": 0.09788,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.020015462239583334,
      "learning_rate": 0.0001,
      "loss": 4.4051,
      "loss/crossentropy": 1.8607316613197327,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25342857837677,
      "step": 4894
    },
    {
      "epoch": 0.09792,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.020963541666666665,
      "learning_rate": 0.0001,
      "loss": 4.6759,
      "loss/crossentropy": 2.245271682739258,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24260863661766052,
      "step": 4896
    },
    {
      "epoch": 0.09796,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.017560831705729165,
      "learning_rate": 0.0001,
      "loss": 4.5006,
      "loss/crossentropy": 2.0503702759742737,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25429168343544006,
      "step": 4898
    },
    {
      "epoch": 0.098,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.017236328125,
      "learning_rate": 0.0001,
      "loss": 4.421,
      "loss/crossentropy": 1.7784077525138855,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2249627709388733,
      "step": 4900
    },
    {
      "epoch": 0.09804,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.0114898681640625,
      "learning_rate": 0.0001,
      "loss": 4.5398,
      "loss/crossentropy": 1.9827336072921753,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22001181542873383,
      "step": 4902
    },
    {
      "epoch": 0.09808,
      "grad_norm": 2.53125,
      "grad_norm_var": 1.638996378580729,
      "learning_rate": 0.0001,
      "loss": 4.8149,
      "loss/crossentropy": 2.1003565788269043,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2550586014986038,
      "step": 4904
    },
    {
      "epoch": 0.09812,
      "grad_norm": 2.484375,
      "grad_norm_var": 1.6366933186848958,
      "learning_rate": 0.0001,
      "loss": 4.5099,
      "loss/crossentropy": 1.9565055966377258,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25481177121400833,
      "step": 4906
    },
    {
      "epoch": 0.09816,
      "grad_norm": 2.234375,
      "grad_norm_var": 1.6479777018229167,
      "learning_rate": 0.0001,
      "loss": 4.6214,
      "loss/crossentropy": 2.1693456172943115,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23336851596832275,
      "step": 4908
    },
    {
      "epoch": 0.0982,
      "grad_norm": 2.40625,
      "grad_norm_var": 1.6428995768229167,
      "learning_rate": 0.0001,
      "loss": 4.7637,
      "loss/crossentropy": 2.050541341304779,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25821831077337265,
      "step": 4910
    },
    {
      "epoch": 0.09824,
      "grad_norm": 2.40625,
      "grad_norm_var": 1.62906494140625,
      "learning_rate": 0.0001,
      "loss": 4.8515,
      "loss/crossentropy": 2.168497681617737,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2674206495285034,
      "step": 4912
    },
    {
      "epoch": 0.09828,
      "grad_norm": 2.609375,
      "grad_norm_var": 1.6097320556640624,
      "learning_rate": 0.0001,
      "loss": 5.1667,
      "loss/crossentropy": 2.147577404975891,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26900771260261536,
      "step": 4914
    },
    {
      "epoch": 0.09832,
      "grad_norm": 2.09375,
      "grad_norm_var": 1.6371734619140625,
      "learning_rate": 0.0001,
      "loss": 4.3958,
      "loss/crossentropy": 2.4436198472976685,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2504816800355911,
      "step": 4916
    },
    {
      "epoch": 0.09836,
      "grad_norm": 2.25,
      "grad_norm_var": 1.6447265625,
      "learning_rate": 0.0001,
      "loss": 4.7244,
      "loss/crossentropy": 2.097359538078308,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24005448818206787,
      "step": 4918
    },
    {
      "epoch": 0.0984,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0267974853515625,
      "learning_rate": 0.0001,
      "loss": 4.2305,
      "loss/crossentropy": 1.8467384576797485,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22932368516921997,
      "step": 4920
    },
    {
      "epoch": 0.09844,
      "grad_norm": 2.859375,
      "grad_norm_var": 0.04263916015625,
      "learning_rate": 0.0001,
      "loss": 4.4407,
      "loss/crossentropy": 2.1454135179519653,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2634875178337097,
      "step": 4922
    },
    {
      "epoch": 0.09848,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.05164286295572917,
      "learning_rate": 0.0001,
      "loss": 4.5775,
      "loss/crossentropy": 1.9837967157363892,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23980768024921417,
      "step": 4924
    },
    {
      "epoch": 0.09852,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.054032389322916666,
      "learning_rate": 0.0001,
      "loss": 4.6888,
      "loss/crossentropy": 2.099667489528656,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23482084274291992,
      "step": 4926
    },
    {
      "epoch": 0.09856,
      "grad_norm": 3.125,
      "grad_norm_var": 0.08787333170572917,
      "learning_rate": 0.0001,
      "loss": 4.5685,
      "loss/crossentropy": 1.8467332124710083,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.256004735827446,
      "step": 4928
    },
    {
      "epoch": 0.0986,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.08548075358072917,
      "learning_rate": 0.0001,
      "loss": 4.4474,
      "loss/crossentropy": 2.036003887653351,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2377806007862091,
      "step": 4930
    },
    {
      "epoch": 0.09864,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.07834370930989583,
      "learning_rate": 0.0001,
      "loss": 4.8258,
      "loss/crossentropy": 2.091457724571228,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24746537953615189,
      "step": 4932
    },
    {
      "epoch": 0.09868,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.07929280598958334,
      "learning_rate": 0.0001,
      "loss": 4.6583,
      "loss/crossentropy": 2.245227336883545,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24141517281532288,
      "step": 4934
    },
    {
      "epoch": 0.09872,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.0661285400390625,
      "learning_rate": 0.0001,
      "loss": 4.6514,
      "loss/crossentropy": 2.295682668685913,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28035247325897217,
      "step": 4936
    },
    {
      "epoch": 0.09876,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.05524800618489583,
      "learning_rate": 0.0001,
      "loss": 4.5525,
      "loss/crossentropy": 1.9251704812049866,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23609354346990585,
      "step": 4938
    },
    {
      "epoch": 0.0988,
      "grad_norm": 2.5,
      "grad_norm_var": 0.048314412434895836,
      "learning_rate": 0.0001,
      "loss": 5.1583,
      "loss/crossentropy": 2.289652466773987,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30698196589946747,
      "step": 4940
    },
    {
      "epoch": 0.09884,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0534820556640625,
      "learning_rate": 0.0001,
      "loss": 4.7759,
      "loss/crossentropy": 2.3339043855667114,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2449272722005844,
      "step": 4942
    },
    {
      "epoch": 0.09888,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.01949462890625,
      "learning_rate": 0.0001,
      "loss": 4.6678,
      "loss/crossentropy": 2.0610195994377136,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2553517669439316,
      "step": 4944
    },
    {
      "epoch": 0.09892,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.018122355143229168,
      "learning_rate": 0.0001,
      "loss": 4.5849,
      "loss/crossentropy": 2.1093825101852417,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24767974764108658,
      "step": 4946
    },
    {
      "epoch": 0.09896,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.01416015625,
      "learning_rate": 0.0001,
      "loss": 4.3631,
      "loss/crossentropy": 2.1742878556251526,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24640457332134247,
      "step": 4948
    },
    {
      "epoch": 0.099,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.0144195556640625,
      "learning_rate": 0.0001,
      "loss": 4.8319,
      "loss/crossentropy": 2.4237486124038696,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2610916793346405,
      "step": 4950
    },
    {
      "epoch": 0.09904,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.015445963541666666,
      "learning_rate": 0.0001,
      "loss": 4.3397,
      "loss/crossentropy": 2.197754144668579,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2275979220867157,
      "step": 4952
    },
    {
      "epoch": 0.09908,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.01529541015625,
      "learning_rate": 0.0001,
      "loss": 4.4533,
      "loss/crossentropy": 2.267225503921509,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23391032963991165,
      "step": 4954
    },
    {
      "epoch": 0.09912,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.009504191080729167,
      "learning_rate": 0.0001,
      "loss": 4.47,
      "loss/crossentropy": 2.04893159866333,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23763196170330048,
      "step": 4956
    },
    {
      "epoch": 0.09916,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.010716756184895834,
      "learning_rate": 0.0001,
      "loss": 4.97,
      "loss/crossentropy": 2.4489223957061768,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27698180079460144,
      "step": 4958
    },
    {
      "epoch": 0.0992,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.011930338541666667,
      "learning_rate": 0.0001,
      "loss": 4.8604,
      "loss/crossentropy": 2.3654375076293945,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2495984137058258,
      "step": 4960
    },
    {
      "epoch": 0.09924,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.011812337239583333,
      "learning_rate": 0.0001,
      "loss": 4.7671,
      "loss/crossentropy": 1.8583308458328247,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21832667291164398,
      "step": 4962
    },
    {
      "epoch": 0.09928,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.01177978515625,
      "learning_rate": 0.0001,
      "loss": 4.4951,
      "loss/crossentropy": 2.0762425661087036,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2280900850892067,
      "step": 4964
    },
    {
      "epoch": 0.09932,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.012532552083333334,
      "learning_rate": 0.0001,
      "loss": 4.7461,
      "loss/crossentropy": 2.228640913963318,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26595622301101685,
      "step": 4966
    },
    {
      "epoch": 0.09936,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.0111236572265625,
      "learning_rate": 0.0001,
      "loss": 4.521,
      "loss/crossentropy": 2.1447466611862183,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2542327791452408,
      "step": 4968
    },
    {
      "epoch": 0.0994,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.01226806640625,
      "learning_rate": 0.0001,
      "loss": 4.3959,
      "loss/crossentropy": 2.0750887989997864,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2362435683608055,
      "step": 4970
    },
    {
      "epoch": 0.09944,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.009666951497395833,
      "learning_rate": 0.0001,
      "loss": 4.6705,
      "loss/crossentropy": 1.9413353204727173,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24418477714061737,
      "step": 4972
    },
    {
      "epoch": 0.09948,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.00904541015625,
      "learning_rate": 0.0001,
      "loss": 4.609,
      "loss/crossentropy": 2.102766752243042,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24170882254838943,
      "step": 4974
    },
    {
      "epoch": 0.09952,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.0090240478515625,
      "learning_rate": 0.0001,
      "loss": 4.7568,
      "loss/crossentropy": 2.431061267852783,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2855495512485504,
      "step": 4976
    },
    {
      "epoch": 0.09956,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.010448201497395834,
      "learning_rate": 0.0001,
      "loss": 4.7449,
      "loss/crossentropy": 1.9656312465667725,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26410341262817383,
      "step": 4978
    },
    {
      "epoch": 0.0996,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.01627197265625,
      "learning_rate": 0.0001,
      "loss": 5.0286,
      "loss/crossentropy": 2.2365923523902893,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27946531772613525,
      "step": 4980
    },
    {
      "epoch": 0.09964,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.015184529622395833,
      "learning_rate": 0.0001,
      "loss": 4.5791,
      "loss/crossentropy": 2.203595757484436,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25247688591480255,
      "step": 4982
    },
    {
      "epoch": 0.09968,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.0163482666015625,
      "learning_rate": 0.0001,
      "loss": 4.6824,
      "loss/crossentropy": 2.1462446451187134,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2539386674761772,
      "step": 4984
    },
    {
      "epoch": 0.09972,
      "grad_norm": 2.25,
      "grad_norm_var": 0.014127604166666667,
      "learning_rate": 0.0001,
      "loss": 4.7121,
      "loss/crossentropy": 2.4518587589263916,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24366125464439392,
      "step": 4986
    },
    {
      "epoch": 0.09976,
      "grad_norm": 2.25,
      "grad_norm_var": 0.01402587890625,
      "learning_rate": 0.0001,
      "loss": 4.5131,
      "loss/crossentropy": 2.003869950771332,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22874485701322556,
      "step": 4988
    },
    {
      "epoch": 0.0998,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.017476399739583332,
      "learning_rate": 0.0001,
      "loss": 4.3221,
      "loss/crossentropy": 2.0251912474632263,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2240670546889305,
      "step": 4990
    },
    {
      "epoch": 0.09984,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.020166015625,
      "learning_rate": 0.0001,
      "loss": 4.5297,
      "loss/crossentropy": 2.199779748916626,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22916750609874725,
      "step": 4992
    },
    {
      "epoch": 0.09988,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.0207672119140625,
      "learning_rate": 0.0001,
      "loss": 4.8066,
      "loss/crossentropy": 2.3852288722991943,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26489999890327454,
      "step": 4994
    },
    {
      "epoch": 0.09992,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.03874409993489583,
      "learning_rate": 0.0001,
      "loss": 4.7245,
      "loss/crossentropy": 1.9446094632148743,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24112944304943085,
      "step": 4996
    },
    {
      "epoch": 0.09996,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.04269205729166667,
      "learning_rate": 0.0001,
      "loss": 4.7063,
      "loss/crossentropy": 2.585115671157837,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26535044610500336,
      "step": 4998
    },
    {
      "epoch": 0.1,
      "grad_norm": 2.8125,
      "grad_norm_var": 1.14400634765625,
      "learning_rate": 0.0001,
      "loss": 4.6848,
      "loss/crossentropy": 1.9871427416801453,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24747569859027863,
      "step": 5000
    },
    {
      "epoch": 0.10004,
      "grad_norm": 2.359375,
      "grad_norm_var": 1.1424967447916667,
      "learning_rate": 0.0001,
      "loss": 4.6058,
      "loss/crossentropy": 1.8981972336769104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24131165444850922,
      "step": 5002
    },
    {
      "epoch": 0.10008,
      "grad_norm": 2.28125,
      "grad_norm_var": 1.1522786458333334,
      "learning_rate": 0.0001,
      "loss": 4.5731,
      "loss/crossentropy": 2.323825240135193,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2584778293967247,
      "step": 5004
    },
    {
      "epoch": 0.10012,
      "grad_norm": 2.390625,
      "grad_norm_var": 1.1363118489583333,
      "learning_rate": 0.0001,
      "loss": 4.7065,
      "loss/crossentropy": 1.728028118610382,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2997850477695465,
      "step": 5006
    },
    {
      "epoch": 0.10016,
      "grad_norm": 2.28125,
      "grad_norm_var": 1.12437744140625,
      "learning_rate": 0.0001,
      "loss": 4.8001,
      "loss/crossentropy": 2.1486289501190186,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25335805118083954,
      "step": 5008
    },
    {
      "epoch": 0.1002,
      "grad_norm": 2.5,
      "grad_norm_var": 1.1099761962890624,
      "learning_rate": 0.0001,
      "loss": 4.936,
      "loss/crossentropy": 2.3132145404815674,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26033517718315125,
      "step": 5010
    },
    {
      "epoch": 0.10024,
      "grad_norm": 2.671875,
      "grad_norm_var": 1.1134928385416667,
      "learning_rate": 0.0001,
      "loss": 4.493,
      "loss/crossentropy": 1.9233656525611877,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2471245899796486,
      "step": 5012
    },
    {
      "epoch": 0.10028,
      "grad_norm": 2.21875,
      "grad_norm_var": 1.1301747639973958,
      "learning_rate": 0.0001,
      "loss": 4.5221,
      "loss/crossentropy": 1.9435511827468872,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24730819463729858,
      "step": 5014
    },
    {
      "epoch": 0.10032,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.018928019205729167,
      "learning_rate": 0.0001,
      "loss": 4.3895,
      "loss/crossentropy": 2.2031294107437134,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23213820159435272,
      "step": 5016
    },
    {
      "epoch": 0.10036,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.0276763916015625,
      "learning_rate": 0.0001,
      "loss": 4.7247,
      "loss/crossentropy": 2.285850405693054,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25809091329574585,
      "step": 5018
    },
    {
      "epoch": 0.1004,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.025211588541666666,
      "learning_rate": 0.0001,
      "loss": 4.7697,
      "loss/crossentropy": 1.8660435676574707,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2436714917421341,
      "step": 5020
    },
    {
      "epoch": 0.10044,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.025877888997395834,
      "learning_rate": 0.0001,
      "loss": 4.3716,
      "loss/crossentropy": 2.0659420490264893,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24248096346855164,
      "step": 5022
    },
    {
      "epoch": 0.10048,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.030516560872395834,
      "learning_rate": 0.0001,
      "loss": 4.7093,
      "loss/crossentropy": 2.213133215904236,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2423061951994896,
      "step": 5024
    },
    {
      "epoch": 0.10052,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.0313385009765625,
      "learning_rate": 0.0001,
      "loss": 4.5288,
      "loss/crossentropy": 2.3052343130111694,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24757324904203415,
      "step": 5026
    },
    {
      "epoch": 0.10056,
      "grad_norm": 2.796875,
      "grad_norm_var": 0.039094034830729166,
      "learning_rate": 0.0001,
      "loss": 4.8709,
      "loss/crossentropy": 2.226990580558777,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25892695784568787,
      "step": 5028
    },
    {
      "epoch": 0.1006,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.04163004557291667,
      "learning_rate": 0.0001,
      "loss": 4.9444,
      "loss/crossentropy": 2.3460742235183716,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27760128676891327,
      "step": 5030
    },
    {
      "epoch": 0.10064,
      "grad_norm": 2.125,
      "grad_norm_var": 0.04345703125,
      "learning_rate": 0.0001,
      "loss": 4.3597,
      "loss/crossentropy": 1.9782095551490784,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2349972277879715,
      "step": 5032
    },
    {
      "epoch": 0.10068,
      "grad_norm": 2.96875,
      "grad_norm_var": 0.06148681640625,
      "learning_rate": 0.0001,
      "loss": 4.5602,
      "loss/crossentropy": 1.847929298877716,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22401423752307892,
      "step": 5034
    },
    {
      "epoch": 0.10072,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.06073811848958333,
      "learning_rate": 0.0001,
      "loss": 4.4989,
      "loss/crossentropy": 2.172071158885956,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2656550332903862,
      "step": 5036
    },
    {
      "epoch": 0.10076,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.07026265462239584,
      "learning_rate": 0.0001,
      "loss": 4.3892,
      "loss/crossentropy": 2.3497499227523804,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26085225492715836,
      "step": 5038
    },
    {
      "epoch": 0.1008,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.07284749348958333,
      "learning_rate": 0.0001,
      "loss": 4.2583,
      "loss/crossentropy": 2.0916348695755005,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21129868924617767,
      "step": 5040
    },
    {
      "epoch": 0.10084,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.07073160807291666,
      "learning_rate": 0.0001,
      "loss": 4.8931,
      "loss/crossentropy": 2.243077278137207,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2657035142183304,
      "step": 5042
    },
    {
      "epoch": 0.10088,
      "grad_norm": 10.0,
      "grad_norm_var": 3.600194295247396,
      "learning_rate": 0.0001,
      "loss": 4.8887,
      "loss/crossentropy": 1.9361066222190857,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.41255413740873337,
      "step": 5044
    },
    {
      "epoch": 0.10092,
      "grad_norm": 3.703125,
      "grad_norm_var": 3.6162760416666666,
      "learning_rate": 0.0001,
      "loss": 4.7632,
      "loss/crossentropy": 2.0139313340187073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2591235190629959,
      "step": 5046
    },
    {
      "epoch": 0.10096,
      "grad_norm": 2.203125,
      "grad_norm_var": 3.589704386393229,
      "learning_rate": 0.0001,
      "loss": 4.6642,
      "loss/crossentropy": 2.335710287094116,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26457205414772034,
      "step": 5048
    },
    {
      "epoch": 0.101,
      "grad_norm": 2.34375,
      "grad_norm_var": 3.604325358072917,
      "learning_rate": 0.0001,
      "loss": 4.3329,
      "loss/crossentropy": 1.9269848465919495,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23867928236722946,
      "step": 5050
    },
    {
      "epoch": 0.10104,
      "grad_norm": 2.34375,
      "grad_norm_var": 3.627415974934896,
      "learning_rate": 0.0001,
      "loss": 4.4982,
      "loss/crossentropy": 1.9732608795166016,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24961821734905243,
      "step": 5052
    },
    {
      "epoch": 0.10108,
      "grad_norm": 2.359375,
      "grad_norm_var": 3.6287506103515623,
      "learning_rate": 0.0001,
      "loss": 4.7218,
      "loss/crossentropy": 2.2659696340560913,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24580512940883636,
      "step": 5054
    },
    {
      "epoch": 0.10112,
      "grad_norm": 2.84375,
      "grad_norm_var": 3.57880859375,
      "learning_rate": 0.0001,
      "loss": 4.4318,
      "loss/crossentropy": 1.6600720882415771,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20283473283052444,
      "step": 5056
    },
    {
      "epoch": 0.10116,
      "grad_norm": 2.375,
      "grad_norm_var": 3.602855428059896,
      "learning_rate": 0.0001,
      "loss": 4.8249,
      "loss/crossentropy": 2.0175185799598694,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23329314589500427,
      "step": 5058
    },
    {
      "epoch": 0.1012,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.146484375,
      "learning_rate": 0.0001,
      "loss": 4.5132,
      "loss/crossentropy": 2.259668231010437,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2692077234387398,
      "step": 5060
    },
    {
      "epoch": 0.10124,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.02144775390625,
      "learning_rate": 0.0001,
      "loss": 4.6392,
      "loss/crossentropy": 2.260953903198242,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28225430846214294,
      "step": 5062
    },
    {
      "epoch": 0.10128,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.021686808268229166,
      "learning_rate": 0.0001,
      "loss": 4.3624,
      "loss/crossentropy": 1.9721892476081848,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2336483597755432,
      "step": 5064
    },
    {
      "epoch": 0.10132,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.021512858072916665,
      "learning_rate": 0.0001,
      "loss": 4.6257,
      "loss/crossentropy": 2.188947319984436,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23959602415561676,
      "step": 5066
    },
    {
      "epoch": 0.10136,
      "grad_norm": 3.078125,
      "grad_norm_var": 0.04843343098958333,
      "learning_rate": 0.0001,
      "loss": 4.5601,
      "loss/crossentropy": 1.7914190292358398,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2234276980161667,
      "step": 5068
    },
    {
      "epoch": 0.1014,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.049925740559895834,
      "learning_rate": 0.0001,
      "loss": 4.5806,
      "loss/crossentropy": 2.1124663949012756,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22523616254329681,
      "step": 5070
    },
    {
      "epoch": 0.10144,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.03764546712239583,
      "learning_rate": 0.0001,
      "loss": 4.7098,
      "loss/crossentropy": 2.1146361231803894,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22483345866203308,
      "step": 5072
    },
    {
      "epoch": 0.10148,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.04121805826822917,
      "learning_rate": 0.0001,
      "loss": 4.5076,
      "loss/crossentropy": 2.335755705833435,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.256914846599102,
      "step": 5074
    },
    {
      "epoch": 0.10152,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.04346415201822917,
      "learning_rate": 0.0001,
      "loss": 4.8665,
      "loss/crossentropy": 2.2911819219589233,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26038119196891785,
      "step": 5076
    },
    {
      "epoch": 0.10156,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.047587076822916664,
      "learning_rate": 0.0001,
      "loss": 4.592,
      "loss/crossentropy": 2.059769034385681,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24004006385803223,
      "step": 5078
    },
    {
      "epoch": 0.1016,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.04888916015625,
      "learning_rate": 0.0001,
      "loss": 4.5028,
      "loss/crossentropy": 2.141201138496399,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22454539686441422,
      "step": 5080
    },
    {
      "epoch": 0.10164,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.050146484375,
      "learning_rate": 0.0001,
      "loss": 4.7527,
      "loss/crossentropy": 1.9538633823394775,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24477297067642212,
      "step": 5082
    },
    {
      "epoch": 0.10168,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.00992431640625,
      "learning_rate": 0.0001,
      "loss": 4.4799,
      "loss/crossentropy": 2.1555078625679016,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2425938919186592,
      "step": 5084
    },
    {
      "epoch": 0.10172,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.009235636393229166,
      "learning_rate": 0.0001,
      "loss": 4.5583,
      "loss/crossentropy": 2.2306214570999146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2445632517337799,
      "step": 5086
    },
    {
      "epoch": 0.10176,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.010179646809895833,
      "learning_rate": 0.0001,
      "loss": 4.1075,
      "loss/crossentropy": 1.9713392853736877,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22239256650209427,
      "step": 5088
    },
    {
      "epoch": 0.1018,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.011799112955729166,
      "learning_rate": 0.0001,
      "loss": 4.5181,
      "loss/crossentropy": 1.951128602027893,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24310563504695892,
      "step": 5090
    },
    {
      "epoch": 0.10184,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.011051432291666666,
      "learning_rate": 0.0001,
      "loss": 4.3617,
      "loss/crossentropy": 2.0100057125091553,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23054375499486923,
      "step": 5092
    },
    {
      "epoch": 0.10188,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.011042277018229166,
      "learning_rate": 0.0001,
      "loss": 4.4573,
      "loss/crossentropy": 2.1898789405822754,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24956130981445312,
      "step": 5094
    },
    {
      "epoch": 0.10192,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.021512858072916665,
      "learning_rate": 0.0001,
      "loss": 4.6576,
      "loss/crossentropy": 1.5666239857673645,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20010025054216385,
      "step": 5096
    },
    {
      "epoch": 0.10196,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.022468058268229167,
      "learning_rate": 0.0001,
      "loss": 4.2762,
      "loss/crossentropy": 1.884951651096344,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2484818547964096,
      "step": 5098
    },
    {
      "epoch": 0.102,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.022554524739583335,
      "learning_rate": 0.0001,
      "loss": 4.5207,
      "loss/crossentropy": 1.976080298423767,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23380043357610703,
      "step": 5100
    },
    {
      "epoch": 0.10204,
      "grad_norm": 2.375,
      "grad_norm_var": 0.0229156494140625,
      "learning_rate": 0.0001,
      "loss": 4.592,
      "loss/crossentropy": 2.1262658834457397,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2441401332616806,
      "step": 5102
    },
    {
      "epoch": 0.10208,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.08055013020833333,
      "learning_rate": 0.0001,
      "loss": 4.2735,
      "loss/crossentropy": 1.7588757276535034,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2101762592792511,
      "step": 5104
    },
    {
      "epoch": 0.10212,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.08059488932291667,
      "learning_rate": 0.0001,
      "loss": 4.6721,
      "loss/crossentropy": 2.292783260345459,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26928654313087463,
      "step": 5106
    },
    {
      "epoch": 0.10216,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.4834218343098958,
      "learning_rate": 0.0001,
      "loss": 4.7475,
      "loss/crossentropy": 2.0670888423919678,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2644767463207245,
      "step": 5108
    },
    {
      "epoch": 0.1022,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.4729563395182292,
      "learning_rate": 0.0001,
      "loss": 4.4113,
      "loss/crossentropy": 2.0522598028182983,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2410094290971756,
      "step": 5110
    },
    {
      "epoch": 0.10224,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.4729075113932292,
      "learning_rate": 0.0001,
      "loss": 4.4914,
      "loss/crossentropy": 2.0756974816322327,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27077721059322357,
      "step": 5112
    },
    {
      "epoch": 0.10228,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.47226155598958336,
      "learning_rate": 0.0001,
      "loss": 4.6524,
      "loss/crossentropy": 2.1569767594337463,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23582034558057785,
      "step": 5114
    },
    {
      "epoch": 0.10232,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.4847819010416667,
      "learning_rate": 0.0001,
      "loss": 4.2821,
      "loss/crossentropy": 1.9736077785491943,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23984025418758392,
      "step": 5116
    },
    {
      "epoch": 0.10236,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.4942698160807292,
      "learning_rate": 0.0001,
      "loss": 4.3047,
      "loss/crossentropy": 2.1400066614151,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2347380667924881,
      "step": 5118
    },
    {
      "epoch": 0.1024,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.46923726399739585,
      "learning_rate": 0.0001,
      "loss": 4.3531,
      "loss/crossentropy": 1.989999234676361,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22508147358894348,
      "step": 5120
    },
    {
      "epoch": 0.10244,
      "grad_norm": 2.5,
      "grad_norm_var": 0.4704498291015625,
      "learning_rate": 0.0001,
      "loss": 4.8817,
      "loss/crossentropy": 2.132390856742859,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3074522316455841,
      "step": 5122
    },
    {
      "epoch": 0.10248,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.011002604166666667,
      "learning_rate": 0.0001,
      "loss": 4.3606,
      "loss/crossentropy": 1.7906856536865234,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2126675397157669,
      "step": 5124
    },
    {
      "epoch": 0.10252,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.0112457275390625,
      "learning_rate": 0.0001,
      "loss": 4.7045,
      "loss/crossentropy": 2.0576369762420654,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27211636304855347,
      "step": 5126
    },
    {
      "epoch": 0.10256,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.01099853515625,
      "learning_rate": 0.0001,
      "loss": 4.5402,
      "loss/crossentropy": 2.1174184679985046,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2344469577074051,
      "step": 5128
    },
    {
      "epoch": 0.1026,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.0090972900390625,
      "learning_rate": 0.0001,
      "loss": 4.7227,
      "loss/crossentropy": 1.9139717817306519,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2362786829471588,
      "step": 5130
    },
    {
      "epoch": 0.10264,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.008495076497395834,
      "learning_rate": 0.0001,
      "loss": 4.4801,
      "loss/crossentropy": 2.022357940673828,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2307513877749443,
      "step": 5132
    },
    {
      "epoch": 0.10268,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.007721964518229167,
      "learning_rate": 0.0001,
      "loss": 4.3963,
      "loss/crossentropy": 2.038477897644043,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22871223092079163,
      "step": 5134
    },
    {
      "epoch": 0.10272,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.007079060872395833,
      "learning_rate": 0.0001,
      "loss": 4.7283,
      "loss/crossentropy": 2.0895442962646484,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23719681799411774,
      "step": 5136
    },
    {
      "epoch": 0.10276,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.008820597330729167,
      "learning_rate": 0.0001,
      "loss": 4.7059,
      "loss/crossentropy": 2.17978835105896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2519204765558243,
      "step": 5138
    },
    {
      "epoch": 0.1028,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.00953369140625,
      "learning_rate": 0.0001,
      "loss": 4.7318,
      "loss/crossentropy": 2.19089937210083,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24376338720321655,
      "step": 5140
    },
    {
      "epoch": 0.10284,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.011002604166666667,
      "learning_rate": 0.0001,
      "loss": 4.3961,
      "loss/crossentropy": 2.018259823322296,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23707401752471924,
      "step": 5142
    },
    {
      "epoch": 0.10288,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.012287394205729166,
      "learning_rate": 0.0001,
      "loss": 4.7556,
      "loss/crossentropy": 2.110401153564453,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23409561812877655,
      "step": 5144
    },
    {
      "epoch": 0.10292,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.012239583333333333,
      "learning_rate": 0.0001,
      "loss": 4.615,
      "loss/crossentropy": 2.2096832990646362,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2580249160528183,
      "step": 5146
    },
    {
      "epoch": 0.10296,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.015803019205729168,
      "learning_rate": 0.0001,
      "loss": 4.5218,
      "loss/crossentropy": 2.1825822591781616,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23592937737703323,
      "step": 5148
    },
    {
      "epoch": 0.103,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.01451416015625,
      "learning_rate": 0.0001,
      "loss": 4.6945,
      "loss/crossentropy": 2.2440234422683716,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23741164803504944,
      "step": 5150
    },
    {
      "epoch": 0.10304,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.015071614583333334,
      "learning_rate": 0.0001,
      "loss": 4.962,
      "loss/crossentropy": 2.3818799257278442,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2642097622156143,
      "step": 5152
    },
    {
      "epoch": 0.10308,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.014378865559895834,
      "learning_rate": 0.0001,
      "loss": 4.5706,
      "loss/crossentropy": 2.1905806064605713,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26945509016513824,
      "step": 5154
    },
    {
      "epoch": 0.10312,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.017671712239583335,
      "learning_rate": 0.0001,
      "loss": 4.5678,
      "loss/crossentropy": 2.096913695335388,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2540033459663391,
      "step": 5156
    },
    {
      "epoch": 0.10316,
      "grad_norm": 2.375,
      "grad_norm_var": 0.015721638997395832,
      "learning_rate": 0.0001,
      "loss": 4.5286,
      "loss/crossentropy": 1.7916489243507385,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22504088282585144,
      "step": 5158
    },
    {
      "epoch": 0.1032,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.014240519205729166,
      "learning_rate": 0.0001,
      "loss": 4.6256,
      "loss/crossentropy": 1.9366079568862915,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25474467873573303,
      "step": 5160
    },
    {
      "epoch": 0.10324,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.014533487955729167,
      "learning_rate": 0.0001,
      "loss": 4.5054,
      "loss/crossentropy": 2.0233771800994873,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23394355177879333,
      "step": 5162
    },
    {
      "epoch": 0.10328,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.010693359375,
      "learning_rate": 0.0001,
      "loss": 4.7803,
      "loss/crossentropy": 2.442312717437744,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28390438854694366,
      "step": 5164
    },
    {
      "epoch": 0.10332,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.014435831705729167,
      "learning_rate": 0.0001,
      "loss": 4.7849,
      "loss/crossentropy": 1.9547526836395264,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2298036813735962,
      "step": 5166
    },
    {
      "epoch": 0.10336,
      "grad_norm": 2.25,
      "grad_norm_var": 0.018766276041666665,
      "learning_rate": 0.0001,
      "loss": 4.3225,
      "loss/crossentropy": 1.7974739074707031,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21959447860717773,
      "step": 5168
    },
    {
      "epoch": 0.1034,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.020075480143229168,
      "learning_rate": 0.0001,
      "loss": 4.5438,
      "loss/crossentropy": 1.9860564470291138,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24661653488874435,
      "step": 5170
    },
    {
      "epoch": 0.10344,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.019050089518229167,
      "learning_rate": 0.0001,
      "loss": 4.5084,
      "loss/crossentropy": 1.6198940873146057,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20592768490314484,
      "step": 5172
    },
    {
      "epoch": 0.10348,
      "grad_norm": 2.25,
      "grad_norm_var": 0.019554646809895833,
      "learning_rate": 0.0001,
      "loss": 4.4831,
      "loss/crossentropy": 2.193474531173706,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24437790364027023,
      "step": 5174
    },
    {
      "epoch": 0.10352,
      "grad_norm": 2.5,
      "grad_norm_var": 0.022557576497395832,
      "learning_rate": 0.0001,
      "loss": 4.9209,
      "loss/crossentropy": 2.221992254257202,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2931511402130127,
      "step": 5176
    },
    {
      "epoch": 0.10356,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.022272745768229168,
      "learning_rate": 0.0001,
      "loss": 4.4611,
      "loss/crossentropy": 2.006419837474823,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2335921749472618,
      "step": 5178
    },
    {
      "epoch": 0.1036,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.022175089518229166,
      "learning_rate": 0.0001,
      "loss": 4.4477,
      "loss/crossentropy": 2.2861804962158203,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2549041658639908,
      "step": 5180
    },
    {
      "epoch": 0.10364,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.014378865559895834,
      "learning_rate": 0.0001,
      "loss": 4.4079,
      "loss/crossentropy": 2.217998743057251,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24741299450397491,
      "step": 5182
    },
    {
      "epoch": 0.10368,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.0123931884765625,
      "learning_rate": 0.0001,
      "loss": 4.7344,
      "loss/crossentropy": 2.1875526905059814,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23636415600776672,
      "step": 5184
    },
    {
      "epoch": 0.10372,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.014850870768229166,
      "learning_rate": 0.0001,
      "loss": 4.4629,
      "loss/crossentropy": 1.8408135175704956,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23278112709522247,
      "step": 5186
    },
    {
      "epoch": 0.10376,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.014188639322916667,
      "learning_rate": 0.0001,
      "loss": 4.6848,
      "loss/crossentropy": 1.7936646342277527,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23859456181526184,
      "step": 5188
    },
    {
      "epoch": 0.1038,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.014058430989583334,
      "learning_rate": 0.0001,
      "loss": 4.3825,
      "loss/crossentropy": 2.0800318717956543,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23155340552330017,
      "step": 5190
    },
    {
      "epoch": 0.10384,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.0066802978515625,
      "learning_rate": 0.0001,
      "loss": 4.9645,
      "loss/crossentropy": 2.277778387069702,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25029345601797104,
      "step": 5192
    },
    {
      "epoch": 0.10388,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0081695556640625,
      "learning_rate": 0.0001,
      "loss": 4.2486,
      "loss/crossentropy": 1.9658478498458862,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22655482590198517,
      "step": 5194
    },
    {
      "epoch": 0.10392,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.01060791015625,
      "learning_rate": 0.0001,
      "loss": 4.7503,
      "loss/crossentropy": 2.214509129524231,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.254493810236454,
      "step": 5196
    },
    {
      "epoch": 0.10396,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.0102203369140625,
      "learning_rate": 0.0001,
      "loss": 4.3648,
      "loss/crossentropy": 1.9465742707252502,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23288530111312866,
      "step": 5198
    },
    {
      "epoch": 0.104,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.010277303059895833,
      "learning_rate": 0.0001,
      "loss": 4.3216,
      "loss/crossentropy": 2.062779188156128,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21423730999231339,
      "step": 5200
    },
    {
      "epoch": 0.10404,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.010350545247395834,
      "learning_rate": 0.0001,
      "loss": 4.1645,
      "loss/crossentropy": 1.777470588684082,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22998665273189545,
      "step": 5202
    },
    {
      "epoch": 0.10408,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.05396728515625,
      "learning_rate": 0.0001,
      "loss": 4.8979,
      "loss/crossentropy": 2.2505980730056763,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30352361500263214,
      "step": 5204
    },
    {
      "epoch": 0.10412,
      "grad_norm": 3.359375,
      "grad_norm_var": 0.12148335774739584,
      "learning_rate": 0.0001,
      "loss": 4.4093,
      "loss/crossentropy": 1.8989517092704773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2210940569639206,
      "step": 5206
    },
    {
      "epoch": 0.10416,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.12923177083333334,
      "learning_rate": 0.0001,
      "loss": 4.5535,
      "loss/crossentropy": 2.378798723220825,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25575730204582214,
      "step": 5208
    },
    {
      "epoch": 0.1042,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.11901041666666666,
      "learning_rate": 0.0001,
      "loss": 4.6763,
      "loss/crossentropy": 1.7642006278038025,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22105304896831512,
      "step": 5210
    },
    {
      "epoch": 0.10424,
      "grad_norm": 2.375,
      "grad_norm_var": 0.11607666015625,
      "learning_rate": 0.0001,
      "loss": 4.8945,
      "loss/crossentropy": 2.188746988773346,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24192160367965698,
      "step": 5212
    },
    {
      "epoch": 0.10428,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.11298421223958334,
      "learning_rate": 0.0001,
      "loss": 4.6001,
      "loss/crossentropy": 2.116630494594574,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2879187613725662,
      "step": 5214
    },
    {
      "epoch": 0.10432,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.11013895670572917,
      "learning_rate": 0.0001,
      "loss": 4.4932,
      "loss/crossentropy": 1.8329599499702454,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21425092220306396,
      "step": 5216
    },
    {
      "epoch": 0.10436,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.1064849853515625,
      "learning_rate": 0.0001,
      "loss": 4.3966,
      "loss/crossentropy": 2.1063259840011597,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23704984784126282,
      "step": 5218
    },
    {
      "epoch": 0.1044,
      "grad_norm": 2.25,
      "grad_norm_var": 0.08238525390625,
      "learning_rate": 0.0001,
      "loss": 4.4821,
      "loss/crossentropy": 1.7994996309280396,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21408653259277344,
      "step": 5220
    },
    {
      "epoch": 0.10444,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.0281158447265625,
      "learning_rate": 0.0001,
      "loss": 4.2437,
      "loss/crossentropy": 1.880006492137909,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24362730979919434,
      "step": 5222
    },
    {
      "epoch": 0.10448,
      "grad_norm": 2.25,
      "grad_norm_var": 0.017853800455729166,
      "learning_rate": 0.0001,
      "loss": 4.2726,
      "loss/crossentropy": 2.010268449783325,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22580985724925995,
      "step": 5224
    },
    {
      "epoch": 0.10452,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.0177886962890625,
      "learning_rate": 0.0001,
      "loss": 4.6839,
      "loss/crossentropy": 2.057162046432495,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23878254741430283,
      "step": 5226
    },
    {
      "epoch": 0.10456,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.019603474934895834,
      "learning_rate": 0.0001,
      "loss": 4.3774,
      "loss/crossentropy": 2.0615930557250977,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22890077531337738,
      "step": 5228
    },
    {
      "epoch": 0.1046,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.020817057291666666,
      "learning_rate": 0.0001,
      "loss": 4.7693,
      "loss/crossentropy": 1.9013472199440002,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22539222240447998,
      "step": 5230
    },
    {
      "epoch": 0.10464,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.019482421875,
      "learning_rate": 0.0001,
      "loss": 4.5531,
      "loss/crossentropy": 2.239185929298401,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24706681817770004,
      "step": 5232
    },
    {
      "epoch": 0.10468,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.020540364583333335,
      "learning_rate": 0.0001,
      "loss": 4.3902,
      "loss/crossentropy": 1.9881523251533508,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24434638023376465,
      "step": 5234
    },
    {
      "epoch": 0.10472,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.020524088541666666,
      "learning_rate": 0.0001,
      "loss": 4.7908,
      "loss/crossentropy": 1.9529212713241577,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2321469634771347,
      "step": 5236
    },
    {
      "epoch": 0.10476,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.01041259765625,
      "learning_rate": 0.0001,
      "loss": 4.8045,
      "loss/crossentropy": 2.196424722671509,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24054434895515442,
      "step": 5238
    },
    {
      "epoch": 0.1048,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.009781901041666667,
      "learning_rate": 0.0001,
      "loss": 4.5895,
      "loss/crossentropy": 2.082987070083618,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2760557308793068,
      "step": 5240
    },
    {
      "epoch": 0.10484,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.011652628580729166,
      "learning_rate": 0.0001,
      "loss": 4.5961,
      "loss/crossentropy": 2.2369720935821533,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25305214524269104,
      "step": 5242
    },
    {
      "epoch": 0.10488,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0119537353515625,
      "learning_rate": 0.0001,
      "loss": 4.7268,
      "loss/crossentropy": 2.3372031450271606,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26765232533216476,
      "step": 5244
    },
    {
      "epoch": 0.10492,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.0113677978515625,
      "learning_rate": 0.0001,
      "loss": 4.6148,
      "loss/crossentropy": 2.23944628238678,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26508912444114685,
      "step": 5246
    },
    {
      "epoch": 0.10496,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.012105305989583334,
      "learning_rate": 0.0001,
      "loss": 4.3453,
      "loss/crossentropy": 2.2701858282089233,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2551077604293823,
      "step": 5248
    },
    {
      "epoch": 0.105,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.010887654622395833,
      "learning_rate": 0.0001,
      "loss": 4.6455,
      "loss/crossentropy": 2.293464779853821,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25016437470912933,
      "step": 5250
    },
    {
      "epoch": 0.10504,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.0109527587890625,
      "learning_rate": 0.0001,
      "loss": 4.5798,
      "loss/crossentropy": 2.3072171211242676,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27824972569942474,
      "step": 5252
    },
    {
      "epoch": 0.10508,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.010498046875,
      "learning_rate": 0.0001,
      "loss": 4.5948,
      "loss/crossentropy": 1.9855756759643555,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2312464788556099,
      "step": 5254
    },
    {
      "epoch": 0.10512,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.009129842122395834,
      "learning_rate": 0.0001,
      "loss": 4.8104,
      "loss/crossentropy": 1.9584077596664429,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2386016771197319,
      "step": 5256
    },
    {
      "epoch": 0.10516,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.010741170247395833,
      "learning_rate": 0.0001,
      "loss": 4.6103,
      "loss/crossentropy": 2.2184669375419617,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25914129614830017,
      "step": 5258
    },
    {
      "epoch": 0.1052,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0122222900390625,
      "learning_rate": 0.0001,
      "loss": 4.1221,
      "loss/crossentropy": 1.6798554062843323,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2037271112203598,
      "step": 5260
    },
    {
      "epoch": 0.10524,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.011995442708333333,
      "learning_rate": 0.0001,
      "loss": 4.2718,
      "loss/crossentropy": 1.8675006031990051,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21148693561553955,
      "step": 5262
    },
    {
      "epoch": 0.10528,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.012336222330729167,
      "learning_rate": 0.0001,
      "loss": 4.5136,
      "loss/crossentropy": 1.9439855813980103,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24611759185791016,
      "step": 5264
    },
    {
      "epoch": 0.10532,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.013923136393229167,
      "learning_rate": 0.0001,
      "loss": 4.5298,
      "loss/crossentropy": 2.1550748348236084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2428945228457451,
      "step": 5266
    },
    {
      "epoch": 0.10536,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.0117584228515625,
      "learning_rate": 0.0001,
      "loss": 4.6777,
      "loss/crossentropy": 1.9924054741859436,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21866007149219513,
      "step": 5268
    },
    {
      "epoch": 0.1054,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.011571248372395834,
      "learning_rate": 0.0001,
      "loss": 4.5359,
      "loss/crossentropy": 2.0413920879364014,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2579498365521431,
      "step": 5270
    },
    {
      "epoch": 0.10544,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.016380818684895833,
      "learning_rate": 0.0001,
      "loss": 4.6406,
      "loss/crossentropy": 2.062632381916046,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24655035883188248,
      "step": 5272
    },
    {
      "epoch": 0.10548,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.021833292643229165,
      "learning_rate": 0.0001,
      "loss": 4.8133,
      "loss/crossentropy": 2.0620386600494385,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3041655272245407,
      "step": 5274
    },
    {
      "epoch": 0.10552,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.016657511393229168,
      "learning_rate": 0.0001,
      "loss": 4.4775,
      "loss/crossentropy": 1.966421365737915,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22333864122629166,
      "step": 5276
    },
    {
      "epoch": 0.10556,
      "grad_norm": 2.25,
      "grad_norm_var": 0.017650349934895834,
      "learning_rate": 0.0001,
      "loss": 4.4323,
      "loss/crossentropy": 1.9120238423347473,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22857370972633362,
      "step": 5278
    },
    {
      "epoch": 0.1056,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.019701131184895835,
      "learning_rate": 0.0001,
      "loss": 4.5179,
      "loss/crossentropy": 2.084389805793762,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2556762397289276,
      "step": 5280
    },
    {
      "epoch": 0.10564,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.019758097330729165,
      "learning_rate": 0.0001,
      "loss": 4.4552,
      "loss/crossentropy": 1.8707188367843628,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22955116629600525,
      "step": 5282
    },
    {
      "epoch": 0.10568,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.019775390625,
      "learning_rate": 0.0001,
      "loss": 4.3538,
      "loss/crossentropy": 1.8243692517280579,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21088172495365143,
      "step": 5284
    },
    {
      "epoch": 0.10572,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.0192535400390625,
      "learning_rate": 0.0001,
      "loss": 4.5046,
      "loss/crossentropy": 2.111305356025696,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2595943957567215,
      "step": 5286
    },
    {
      "epoch": 0.10576,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.015697224934895834,
      "learning_rate": 0.0001,
      "loss": 4.4598,
      "loss/crossentropy": 2.3729283809661865,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2632211595773697,
      "step": 5288
    },
    {
      "epoch": 0.1058,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.007249959309895833,
      "learning_rate": 0.0001,
      "loss": 4.6256,
      "loss/crossentropy": 2.3542696237564087,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25645585358142853,
      "step": 5290
    },
    {
      "epoch": 0.10584,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.01002197265625,
      "learning_rate": 0.0001,
      "loss": 4.4863,
      "loss/crossentropy": 2.0140068531036377,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23033145815134048,
      "step": 5292
    },
    {
      "epoch": 0.10588,
      "grad_norm": 2.375,
      "grad_norm_var": 0.0099761962890625,
      "learning_rate": 0.0001,
      "loss": 4.8126,
      "loss/crossentropy": 1.9499077796936035,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21458172798156738,
      "step": 5294
    },
    {
      "epoch": 0.10592,
      "grad_norm": 2.25,
      "grad_norm_var": 0.0084625244140625,
      "learning_rate": 0.0001,
      "loss": 4.479,
      "loss/crossentropy": 2.1434344053268433,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2552379444241524,
      "step": 5296
    },
    {
      "epoch": 0.10596,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.007819620768229167,
      "learning_rate": 0.0001,
      "loss": 4.7149,
      "loss/crossentropy": 1.9951340556144714,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21683495491743088,
      "step": 5298
    },
    {
      "epoch": 0.106,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.02330322265625,
      "learning_rate": 0.0001,
      "loss": 4.7855,
      "loss/crossentropy": 2.380235195159912,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2756577730178833,
      "step": 5300
    },
    {
      "epoch": 0.10604,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.02330322265625,
      "learning_rate": 0.0001,
      "loss": 4.8596,
      "loss/crossentropy": 2.298241972923279,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2598598450422287,
      "step": 5302
    },
    {
      "epoch": 0.10608,
      "grad_norm": 2.5,
      "grad_norm_var": 0.032136027018229166,
      "learning_rate": 0.0001,
      "loss": 4.8615,
      "loss/crossentropy": 2.093233823776245,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23809552192687988,
      "step": 5304
    },
    {
      "epoch": 0.10612,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.031493123372395834,
      "learning_rate": 0.0001,
      "loss": 4.5271,
      "loss/crossentropy": 2.0901564955711365,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24874016642570496,
      "step": 5306
    },
    {
      "epoch": 0.10616,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.0304107666015625,
      "learning_rate": 0.0001,
      "loss": 4.3193,
      "loss/crossentropy": 1.8029736280441284,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20973137766122818,
      "step": 5308
    },
    {
      "epoch": 0.1062,
      "grad_norm": 2.25,
      "grad_norm_var": 0.03316650390625,
      "learning_rate": 0.0001,
      "loss": 4.4255,
      "loss/crossentropy": 2.4068437814712524,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25310443341732025,
      "step": 5310
    },
    {
      "epoch": 0.10624,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.031029256184895833,
      "learning_rate": 0.0001,
      "loss": 4.4499,
      "loss/crossentropy": 2.1125503182411194,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2455870360136032,
      "step": 5312
    },
    {
      "epoch": 0.10628,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.0323150634765625,
      "learning_rate": 0.0001,
      "loss": 4.4752,
      "loss/crossentropy": 2.0995737314224243,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2418016716837883,
      "step": 5314
    },
    {
      "epoch": 0.10632,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.0212890625,
      "learning_rate": 0.0001,
      "loss": 4.5873,
      "loss/crossentropy": 1.8753212690353394,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24125799536705017,
      "step": 5316
    },
    {
      "epoch": 0.10636,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.021451822916666665,
      "learning_rate": 0.0001,
      "loss": 4.7269,
      "loss/crossentropy": 2.0175408720970154,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23516137897968292,
      "step": 5318
    },
    {
      "epoch": 0.1064,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.00914306640625,
      "learning_rate": 0.0001,
      "loss": 4.4953,
      "loss/crossentropy": 2.2671592235565186,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2356845736503601,
      "step": 5320
    },
    {
      "epoch": 0.10644,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.009468587239583333,
      "learning_rate": 0.0001,
      "loss": 4.5328,
      "loss/crossentropy": 2.142452359199524,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26422591507434845,
      "step": 5322
    },
    {
      "epoch": 0.10648,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.007225545247395834,
      "learning_rate": 0.0001,
      "loss": 4.4328,
      "loss/crossentropy": 1.9664896726608276,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25856246054172516,
      "step": 5324
    },
    {
      "epoch": 0.10652,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.007222493489583333,
      "learning_rate": 0.0001,
      "loss": 4.5531,
      "loss/crossentropy": 2.168110191822052,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23274105042219162,
      "step": 5326
    },
    {
      "epoch": 0.10656,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.00552978515625,
      "learning_rate": 0.0001,
      "loss": 4.5242,
      "loss/crossentropy": 2.006514251232147,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2532104551792145,
      "step": 5328
    },
    {
      "epoch": 0.1066,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.0070220947265625,
      "learning_rate": 0.0001,
      "loss": 4.5593,
      "loss/crossentropy": 2.462701439857483,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2771739661693573,
      "step": 5330
    },
    {
      "epoch": 0.10664,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.0076324462890625,
      "learning_rate": 0.0001,
      "loss": 4.4076,
      "loss/crossentropy": 2.0889209508895874,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2479858472943306,
      "step": 5332
    },
    {
      "epoch": 0.10668,
      "grad_norm": 2.25,
      "grad_norm_var": 0.008356730143229166,
      "learning_rate": 0.0001,
      "loss": 4.329,
      "loss/crossentropy": 1.8056100606918335,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22194840013980865,
      "step": 5334
    },
    {
      "epoch": 0.10672,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.0075185139973958336,
      "learning_rate": 0.0001,
      "loss": 4.703,
      "loss/crossentropy": 2.32460880279541,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28185322880744934,
      "step": 5336
    },
    {
      "epoch": 0.10676,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0075185139973958336,
      "learning_rate": 0.0001,
      "loss": 4.6388,
      "loss/crossentropy": 2.238978862762451,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24314726889133453,
      "step": 5338
    },
    {
      "epoch": 0.1068,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.00914306640625,
      "learning_rate": 0.0001,
      "loss": 4.4161,
      "loss/crossentropy": 1.8914734721183777,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20767460763454437,
      "step": 5340
    },
    {
      "epoch": 0.10684,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.010445149739583333,
      "learning_rate": 0.0001,
      "loss": 4.5628,
      "loss/crossentropy": 1.9704068899154663,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23617815226316452,
      "step": 5342
    },
    {
      "epoch": 0.10688,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.013374837239583333,
      "learning_rate": 0.0001,
      "loss": 4.214,
      "loss/crossentropy": 1.8539690971374512,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2218247577548027,
      "step": 5344
    },
    {
      "epoch": 0.10692,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.012596638997395833,
      "learning_rate": 0.0001,
      "loss": 4.6077,
      "loss/crossentropy": 1.982038140296936,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23828908801078796,
      "step": 5346
    },
    {
      "epoch": 0.10696,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.013313802083333333,
      "learning_rate": 0.0001,
      "loss": 4.2879,
      "loss/crossentropy": 1.8247870802879333,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22487633675336838,
      "step": 5348
    },
    {
      "epoch": 0.107,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.011253865559895833,
      "learning_rate": 0.0001,
      "loss": 4.382,
      "loss/crossentropy": 2.0704214572906494,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24332743138074875,
      "step": 5350
    },
    {
      "epoch": 0.10704,
      "grad_norm": 2.5,
      "grad_norm_var": 0.013509114583333334,
      "learning_rate": 0.0001,
      "loss": 4.709,
      "loss/crossentropy": 2.037345290184021,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2515557184815407,
      "step": 5352
    },
    {
      "epoch": 0.10708,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.020702107747395834,
      "learning_rate": 0.0001,
      "loss": 4.5427,
      "loss/crossentropy": 2.0561426877975464,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24030926823616028,
      "step": 5354
    },
    {
      "epoch": 0.10712,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.017943318684895834,
      "learning_rate": 0.0001,
      "loss": 4.3108,
      "loss/crossentropy": 1.6871100068092346,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2526541203260422,
      "step": 5356
    },
    {
      "epoch": 0.10716,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.01978759765625,
      "learning_rate": 0.0001,
      "loss": 4.5238,
      "loss/crossentropy": 2.0133201479911804,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23663413524627686,
      "step": 5358
    },
    {
      "epoch": 0.1072,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.0170562744140625,
      "learning_rate": 0.0001,
      "loss": 4.557,
      "loss/crossentropy": 2.0627574920654297,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24395380914211273,
      "step": 5360
    },
    {
      "epoch": 0.10724,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.020929972330729168,
      "learning_rate": 0.0001,
      "loss": 4.5495,
      "loss/crossentropy": 2.280818462371826,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2806383967399597,
      "step": 5362
    },
    {
      "epoch": 0.10728,
      "grad_norm": 2.5,
      "grad_norm_var": 0.022684733072916668,
      "learning_rate": 0.0001,
      "loss": 4.4743,
      "loss/crossentropy": 2.002636671066284,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23986798524856567,
      "step": 5364
    },
    {
      "epoch": 0.10732,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.022298177083333332,
      "learning_rate": 0.0001,
      "loss": 4.6816,
      "loss/crossentropy": 2.042721927165985,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23199205100536346,
      "step": 5366
    },
    {
      "epoch": 0.10736,
      "grad_norm": 2.25,
      "grad_norm_var": 0.021540323893229168,
      "learning_rate": 0.0001,
      "loss": 4.3225,
      "loss/crossentropy": 2.1047908663749695,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23987916857004166,
      "step": 5368
    },
    {
      "epoch": 0.1074,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.014774576822916666,
      "learning_rate": 0.0001,
      "loss": 4.4827,
      "loss/crossentropy": 2.0513075590133667,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.242512047290802,
      "step": 5370
    },
    {
      "epoch": 0.10744,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.015913899739583334,
      "learning_rate": 0.0001,
      "loss": 4.4452,
      "loss/crossentropy": 1.9151215553283691,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23007020354270935,
      "step": 5372
    },
    {
      "epoch": 0.10748,
      "grad_norm": 2.125,
      "grad_norm_var": 0.019303385416666666,
      "learning_rate": 0.0001,
      "loss": 4.2344,
      "loss/crossentropy": 1.8759313821792603,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2068658247590065,
      "step": 5374
    },
    {
      "epoch": 0.10752,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.020807902018229168,
      "learning_rate": 0.0001,
      "loss": 4.4667,
      "loss/crossentropy": 2.1500572562217712,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24392293393611908,
      "step": 5376
    },
    {
      "epoch": 0.10756,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.018355305989583334,
      "learning_rate": 0.0001,
      "loss": 4.0917,
      "loss/crossentropy": 1.6089633703231812,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20190145075321198,
      "step": 5378
    },
    {
      "epoch": 0.1076,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.014452107747395833,
      "learning_rate": 0.0001,
      "loss": 4.6521,
      "loss/crossentropy": 2.1967561841011047,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2749984338879585,
      "step": 5380
    },
    {
      "epoch": 0.10764,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.011311848958333334,
      "learning_rate": 0.0001,
      "loss": 4.0843,
      "loss/crossentropy": 1.8293656706809998,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20650158822536469,
      "step": 5382
    },
    {
      "epoch": 0.10768,
      "grad_norm": 2.125,
      "grad_norm_var": 0.011847941080729167,
      "learning_rate": 0.0001,
      "loss": 4.3441,
      "loss/crossentropy": 2.3964673280715942,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2814205437898636,
      "step": 5384
    },
    {
      "epoch": 0.10772,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.021842447916666667,
      "learning_rate": 0.0001,
      "loss": 4.8872,
      "loss/crossentropy": 2.4995274543762207,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28704003244638443,
      "step": 5386
    },
    {
      "epoch": 0.10776,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.019462076822916667,
      "learning_rate": 0.0001,
      "loss": 4.6066,
      "loss/crossentropy": 2.0650060176849365,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23606212437152863,
      "step": 5388
    },
    {
      "epoch": 0.1078,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.01640625,
      "learning_rate": 0.0001,
      "loss": 4.3723,
      "loss/crossentropy": 2.3049341440200806,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23550046980381012,
      "step": 5390
    },
    {
      "epoch": 0.10784,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.013505045572916667,
      "learning_rate": 0.0001,
      "loss": 4.813,
      "loss/crossentropy": 2.2687143087387085,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2854095697402954,
      "step": 5392
    },
    {
      "epoch": 0.10788,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.014557902018229167,
      "learning_rate": 0.0001,
      "loss": 4.6267,
      "loss/crossentropy": 2.029325544834137,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23444775491952896,
      "step": 5394
    },
    {
      "epoch": 0.10792,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.013895670572916666,
      "learning_rate": 0.0001,
      "loss": 4.5214,
      "loss/crossentropy": 2.2012031078338623,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27986764907836914,
      "step": 5396
    },
    {
      "epoch": 0.10796,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.012626139322916667,
      "learning_rate": 0.0001,
      "loss": 4.65,
      "loss/crossentropy": 2.2396020889282227,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24839117377996445,
      "step": 5398
    },
    {
      "epoch": 0.108,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.009566243489583333,
      "learning_rate": 0.0001,
      "loss": 4.4634,
      "loss/crossentropy": 2.1481886506080627,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2545444592833519,
      "step": 5400
    },
    {
      "epoch": 0.10804,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.006005859375,
      "learning_rate": 0.0001,
      "loss": 4.6109,
      "loss/crossentropy": 1.9799351692199707,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23296385258436203,
      "step": 5402
    },
    {
      "epoch": 0.10808,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.006180826822916667,
      "learning_rate": 0.0001,
      "loss": 4.5612,
      "loss/crossentropy": 1.845237910747528,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2316332384943962,
      "step": 5404
    },
    {
      "epoch": 0.10812,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.005692545572916667,
      "learning_rate": 0.0001,
      "loss": 4.4825,
      "loss/crossentropy": 2.078865647315979,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.251836359500885,
      "step": 5406
    },
    {
      "epoch": 0.10816,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.006245930989583333,
      "learning_rate": 0.0001,
      "loss": 4.4409,
      "loss/crossentropy": 2.031971752643585,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23658733069896698,
      "step": 5408
    },
    {
      "epoch": 0.1082,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.00504150390625,
      "learning_rate": 0.0001,
      "loss": 4.3034,
      "loss/crossentropy": 1.8173908591270447,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2169174626469612,
      "step": 5410
    },
    {
      "epoch": 0.10824,
      "grad_norm": 2.25,
      "grad_norm_var": 0.0086090087890625,
      "learning_rate": 0.0001,
      "loss": 4.838,
      "loss/crossentropy": 2.2501285672187805,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.240458145737648,
      "step": 5412
    },
    {
      "epoch": 0.10828,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.015168253580729167,
      "learning_rate": 0.0001,
      "loss": 4.5449,
      "loss/crossentropy": 2.256573438644409,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26129382848739624,
      "step": 5414
    },
    {
      "epoch": 0.10832,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.0153717041015625,
      "learning_rate": 0.0001,
      "loss": 4.7704,
      "loss/crossentropy": 2.2014705538749695,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2334136888384819,
      "step": 5416
    },
    {
      "epoch": 0.10836,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.013016764322916667,
      "learning_rate": 0.0001,
      "loss": 4.4046,
      "loss/crossentropy": 1.8590609431266785,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2271011471748352,
      "step": 5418
    },
    {
      "epoch": 0.1084,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.017378743489583334,
      "learning_rate": 0.0001,
      "loss": 4.9419,
      "loss/crossentropy": 2.2923961877822876,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25833888351917267,
      "step": 5420
    },
    {
      "epoch": 0.10844,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.017723592122395833,
      "learning_rate": 0.0001,
      "loss": 4.4535,
      "loss/crossentropy": 2.1932299733161926,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26209479570388794,
      "step": 5422
    },
    {
      "epoch": 0.10848,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.016185506184895834,
      "learning_rate": 0.0001,
      "loss": 4.7057,
      "loss/crossentropy": 2.3909924030303955,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2525275945663452,
      "step": 5424
    },
    {
      "epoch": 0.10852,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.011324055989583333,
      "learning_rate": 0.0001,
      "loss": 4.7509,
      "loss/crossentropy": 2.423817992210388,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2635423541069031,
      "step": 5426
    },
    {
      "epoch": 0.10856,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.009227498372395834,
      "learning_rate": 0.0001,
      "loss": 4.7082,
      "loss/crossentropy": 1.9641632437705994,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2460889369249344,
      "step": 5428
    },
    {
      "epoch": 0.1086,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.0077707926432291664,
      "learning_rate": 0.0001,
      "loss": 4.6347,
      "loss/crossentropy": 2.027769148349762,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2357894703745842,
      "step": 5430
    },
    {
      "epoch": 0.10864,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.0128570556640625,
      "learning_rate": 0.0001,
      "loss": 4.4833,
      "loss/crossentropy": 2.122319996356964,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2445123866200447,
      "step": 5432
    },
    {
      "epoch": 0.10868,
      "grad_norm": 3.53125,
      "grad_norm_var": 0.0993072509765625,
      "learning_rate": 0.0001,
      "loss": 4.6332,
      "loss/crossentropy": 1.8631052374839783,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23456327617168427,
      "step": 5434
    },
    {
      "epoch": 0.10872,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.1000885009765625,
      "learning_rate": 0.0001,
      "loss": 4.6022,
      "loss/crossentropy": 2.184281885623932,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24073782563209534,
      "step": 5436
    },
    {
      "epoch": 0.10876,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.09521077473958334,
      "learning_rate": 0.0001,
      "loss": 4.7912,
      "loss/crossentropy": 1.9587833881378174,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22247321158647537,
      "step": 5438
    },
    {
      "epoch": 0.1088,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.09562886555989583,
      "learning_rate": 0.0001,
      "loss": 4.5185,
      "loss/crossentropy": 2.334655284881592,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24817728251218796,
      "step": 5440
    },
    {
      "epoch": 0.10884,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.10161031087239583,
      "learning_rate": 0.0001,
      "loss": 4.3817,
      "loss/crossentropy": 2.1424371004104614,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2622353136539459,
      "step": 5442
    },
    {
      "epoch": 0.10888,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.10598958333333333,
      "learning_rate": 0.0001,
      "loss": 4.5876,
      "loss/crossentropy": 2.0363662242889404,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2450125440955162,
      "step": 5444
    },
    {
      "epoch": 0.10892,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.110791015625,
      "learning_rate": 0.0001,
      "loss": 4.4015,
      "loss/crossentropy": 2.0536006689071655,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23283181339502335,
      "step": 5446
    },
    {
      "epoch": 0.10896,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.10741780598958334,
      "learning_rate": 0.0001,
      "loss": 4.5194,
      "loss/crossentropy": 2.2678059339523315,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24513862282037735,
      "step": 5448
    },
    {
      "epoch": 0.109,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.02115478515625,
      "learning_rate": 0.0001,
      "loss": 4.7404,
      "loss/crossentropy": 2.406686782836914,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2901146113872528,
      "step": 5450
    },
    {
      "epoch": 0.10904,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.017801920572916668,
      "learning_rate": 0.0001,
      "loss": 4.4724,
      "loss/crossentropy": 2.352605938911438,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25215400755405426,
      "step": 5452
    },
    {
      "epoch": 0.10908,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.014546712239583334,
      "learning_rate": 0.0001,
      "loss": 4.5593,
      "loss/crossentropy": 1.9139850735664368,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22682765871286392,
      "step": 5454
    },
    {
      "epoch": 0.10912,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.01441650390625,
      "learning_rate": 0.0001,
      "loss": 4.6163,
      "loss/crossentropy": 2.3240445852279663,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24574412405490875,
      "step": 5456
    },
    {
      "epoch": 0.10916,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.0123687744140625,
      "learning_rate": 0.0001,
      "loss": 4.3356,
      "loss/crossentropy": 1.9347040057182312,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2159302830696106,
      "step": 5458
    },
    {
      "epoch": 0.1092,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.012360636393229167,
      "learning_rate": 0.0001,
      "loss": 4.6539,
      "loss/crossentropy": 1.8933109641075134,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.284725621342659,
      "step": 5460
    },
    {
      "epoch": 0.10924,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.0129058837890625,
      "learning_rate": 0.0001,
      "loss": 4.2488,
      "loss/crossentropy": 2.3611297607421875,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2651517689228058,
      "step": 5462
    },
    {
      "epoch": 0.10928,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0235504150390625,
      "learning_rate": 0.0001,
      "loss": 4.4564,
      "loss/crossentropy": 1.828608751296997,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2337760180234909,
      "step": 5464
    },
    {
      "epoch": 0.10932,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.015412394205729167,
      "learning_rate": 0.0001,
      "loss": 4.321,
      "loss/crossentropy": 2.1374374628067017,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2525453567504883,
      "step": 5466
    },
    {
      "epoch": 0.10936,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0150543212890625,
      "learning_rate": 0.0001,
      "loss": 4.5307,
      "loss/crossentropy": 1.8054441213607788,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2252344712615013,
      "step": 5468
    },
    {
      "epoch": 0.1094,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.022972615559895833,
      "learning_rate": 0.0001,
      "loss": 4.616,
      "loss/crossentropy": 2.1468498706817627,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2604861631989479,
      "step": 5470
    },
    {
      "epoch": 0.10944,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.022684733072916668,
      "learning_rate": 0.0001,
      "loss": 4.7298,
      "loss/crossentropy": 2.2180548906326294,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2603686898946762,
      "step": 5472
    },
    {
      "epoch": 0.10948,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.022261555989583334,
      "learning_rate": 0.0001,
      "loss": 4.5263,
      "loss/crossentropy": 1.9773708581924438,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23676057159900665,
      "step": 5474
    },
    {
      "epoch": 0.10952,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.023485310872395835,
      "learning_rate": 0.0001,
      "loss": 4.6156,
      "loss/crossentropy": 1.9277283549308777,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22822558879852295,
      "step": 5476
    },
    {
      "epoch": 0.10956,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.022484334309895833,
      "learning_rate": 0.0001,
      "loss": 4.5529,
      "loss/crossentropy": 2.0625431537628174,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2565220817923546,
      "step": 5478
    },
    {
      "epoch": 0.1096,
      "grad_norm": 2.25,
      "grad_norm_var": 0.01441650390625,
      "learning_rate": 0.0001,
      "loss": 4.7956,
      "loss/crossentropy": 2.383894443511963,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2563806623220444,
      "step": 5480
    },
    {
      "epoch": 0.10964,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.0120758056640625,
      "learning_rate": 0.0001,
      "loss": 4.5226,
      "loss/crossentropy": 2.409442663192749,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2637571170926094,
      "step": 5482
    },
    {
      "epoch": 0.10968,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.0151763916015625,
      "learning_rate": 0.0001,
      "loss": 4.743,
      "loss/crossentropy": 2.1789854764938354,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23846548050642014,
      "step": 5484
    },
    {
      "epoch": 0.10972,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.011839803059895833,
      "learning_rate": 0.0001,
      "loss": 4.4108,
      "loss/crossentropy": 2.127842903137207,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2622772455215454,
      "step": 5486
    },
    {
      "epoch": 0.10976,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.01412353515625,
      "learning_rate": 0.0001,
      "loss": 4.6032,
      "loss/crossentropy": 2.107556462287903,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24714312702417374,
      "step": 5488
    },
    {
      "epoch": 0.1098,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.022459920247395834,
      "learning_rate": 0.0001,
      "loss": 4.6525,
      "loss/crossentropy": 2.1959601640701294,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23568396270275116,
      "step": 5490
    },
    {
      "epoch": 0.10984,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.021826171875,
      "learning_rate": 0.0001,
      "loss": 4.5584,
      "loss/crossentropy": 2.1246761083602905,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24083472788333893,
      "step": 5492
    },
    {
      "epoch": 0.10988,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.025031534830729167,
      "learning_rate": 0.0001,
      "loss": 4.6436,
      "loss/crossentropy": 2.091724157333374,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2527346611022949,
      "step": 5494
    },
    {
      "epoch": 0.10992,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.026056925455729168,
      "learning_rate": 0.0001,
      "loss": 4.3207,
      "loss/crossentropy": 1.8898470997810364,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20916878432035446,
      "step": 5496
    },
    {
      "epoch": 0.10996,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.025951131184895834,
      "learning_rate": 0.0001,
      "loss": 4.399,
      "loss/crossentropy": 2.1901716589927673,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2409309297800064,
      "step": 5498
    },
    {
      "epoch": 0.11,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.017626953125,
      "learning_rate": 0.0001,
      "loss": 4.6897,
      "loss/crossentropy": 2.1018574237823486,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24559657275676727,
      "step": 5500
    },
    {
      "epoch": 0.11004,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.019449869791666668,
      "learning_rate": 0.0001,
      "loss": 3.8159,
      "loss/crossentropy": 2.0575350522994995,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23721858859062195,
      "step": 5502
    },
    {
      "epoch": 0.11008,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.019710286458333334,
      "learning_rate": 0.0001,
      "loss": 4.6347,
      "loss/crossentropy": 2.1846336126327515,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2378462702035904,
      "step": 5504
    },
    {
      "epoch": 0.11012,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.014574178059895833,
      "learning_rate": 0.0001,
      "loss": 4.4028,
      "loss/crossentropy": 2.1359363198280334,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26211391389369965,
      "step": 5506
    },
    {
      "epoch": 0.11016,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.01416015625,
      "learning_rate": 0.0001,
      "loss": 4.5621,
      "loss/crossentropy": 2.236825942993164,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2557792589068413,
      "step": 5508
    },
    {
      "epoch": 0.1102,
      "grad_norm": 2.25,
      "grad_norm_var": 0.009521484375,
      "learning_rate": 0.0001,
      "loss": 4.3234,
      "loss/crossentropy": 2.3140580654144287,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2554958164691925,
      "step": 5510
    },
    {
      "epoch": 0.11024,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.00904541015625,
      "learning_rate": 0.0001,
      "loss": 4.4382,
      "loss/crossentropy": 1.7190355062484741,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19964009523391724,
      "step": 5512
    },
    {
      "epoch": 0.11028,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.014631144205729167,
      "learning_rate": 0.0001,
      "loss": 4.3831,
      "loss/crossentropy": 1.8326427340507507,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2199154868721962,
      "step": 5514
    },
    {
      "epoch": 0.11032,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.01549072265625,
      "learning_rate": 0.0001,
      "loss": 4.4494,
      "loss/crossentropy": 1.9013121724128723,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20293578505516052,
      "step": 5516
    },
    {
      "epoch": 0.11036,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0138671875,
      "learning_rate": 0.0001,
      "loss": 4.7056,
      "loss/crossentropy": 2.0221983790397644,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23435519635677338,
      "step": 5518
    },
    {
      "epoch": 0.1104,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.0140533447265625,
      "learning_rate": 0.0001,
      "loss": 4.6157,
      "loss/crossentropy": 2.153970956802368,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22127598524093628,
      "step": 5520
    },
    {
      "epoch": 0.11044,
      "grad_norm": 2.125,
      "grad_norm_var": 0.010835774739583333,
      "learning_rate": 0.0001,
      "loss": 4.4516,
      "loss/crossentropy": 1.8674496412277222,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.232261061668396,
      "step": 5522
    },
    {
      "epoch": 0.11048,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.012495930989583333,
      "learning_rate": 0.0001,
      "loss": 4.6528,
      "loss/crossentropy": 2.1575759649276733,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22218701988458633,
      "step": 5524
    },
    {
      "epoch": 0.11052,
      "grad_norm": 2.25,
      "grad_norm_var": 0.012434895833333333,
      "learning_rate": 0.0001,
      "loss": 4.5553,
      "loss/crossentropy": 2.054452419281006,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25137215852737427,
      "step": 5526
    },
    {
      "epoch": 0.11056,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.022639973958333334,
      "learning_rate": 0.0001,
      "loss": 4.4789,
      "loss/crossentropy": 1.966478705406189,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23289234936237335,
      "step": 5528
    },
    {
      "epoch": 0.1106,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.017626953125,
      "learning_rate": 0.0001,
      "loss": 4.687,
      "loss/crossentropy": 2.171034336090088,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23373593389987946,
      "step": 5530
    },
    {
      "epoch": 0.11064,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.022362263997395833,
      "learning_rate": 0.0001,
      "loss": 4.142,
      "loss/crossentropy": 2.2416292428970337,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24835532158613205,
      "step": 5532
    },
    {
      "epoch": 0.11068,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.02720947265625,
      "learning_rate": 0.0001,
      "loss": 4.6252,
      "loss/crossentropy": 2.2599780559539795,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24948878586292267,
      "step": 5534
    },
    {
      "epoch": 0.11072,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.028902180989583335,
      "learning_rate": 0.0001,
      "loss": 4.6984,
      "loss/crossentropy": 2.2292014360427856,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26616473495960236,
      "step": 5536
    },
    {
      "epoch": 0.11076,
      "grad_norm": 2.375,
      "grad_norm_var": 0.027424112955729166,
      "learning_rate": 0.0001,
      "loss": 4.336,
      "loss/crossentropy": 1.9285388588905334,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22964124381542206,
      "step": 5538
    },
    {
      "epoch": 0.1108,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.0251953125,
      "learning_rate": 0.0001,
      "loss": 4.2861,
      "loss/crossentropy": 1.864789366722107,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2264304906129837,
      "step": 5540
    },
    {
      "epoch": 0.11084,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.0250885009765625,
      "learning_rate": 0.0001,
      "loss": 4.5163,
      "loss/crossentropy": 1.8676912188529968,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21919699758291245,
      "step": 5542
    },
    {
      "epoch": 0.11088,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.0190338134765625,
      "learning_rate": 0.0001,
      "loss": 4.3612,
      "loss/crossentropy": 2.34523469209671,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25638002157211304,
      "step": 5544
    },
    {
      "epoch": 0.11092,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.020992024739583334,
      "learning_rate": 0.0001,
      "loss": 4.5165,
      "loss/crossentropy": 2.2903120517730713,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25111711025238037,
      "step": 5546
    },
    {
      "epoch": 0.11096,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.017671712239583335,
      "learning_rate": 0.0001,
      "loss": 4.5336,
      "loss/crossentropy": 2.2106658220291138,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22340595722198486,
      "step": 5548
    },
    {
      "epoch": 0.111,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.013700358072916667,
      "learning_rate": 0.0001,
      "loss": 4.6305,
      "loss/crossentropy": 2.0777581334114075,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2446284517645836,
      "step": 5550
    },
    {
      "epoch": 0.11104,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.010758463541666667,
      "learning_rate": 0.0001,
      "loss": 4.5507,
      "loss/crossentropy": 2.131237506866455,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24113191664218903,
      "step": 5552
    },
    {
      "epoch": 0.11108,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.018778483072916668,
      "learning_rate": 0.0001,
      "loss": 4.6337,
      "loss/crossentropy": 2.190987467765808,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2528213635087013,
      "step": 5554
    },
    {
      "epoch": 0.11112,
      "grad_norm": 2.25,
      "grad_norm_var": 0.019527180989583334,
      "learning_rate": 0.0001,
      "loss": 4.4889,
      "loss/crossentropy": 2.26843523979187,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27781064808368683,
      "step": 5556
    },
    {
      "epoch": 0.11116,
      "grad_norm": 2.25,
      "grad_norm_var": 0.019527180989583334,
      "learning_rate": 0.0001,
      "loss": 4.2242,
      "loss/crossentropy": 1.9507999420166016,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23172564804553986,
      "step": 5558
    },
    {
      "epoch": 0.1112,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.016966756184895834,
      "learning_rate": 0.0001,
      "loss": 4.4821,
      "loss/crossentropy": 2.0738234519958496,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24497026205062866,
      "step": 5560
    },
    {
      "epoch": 0.11124,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.019261678059895832,
      "learning_rate": 0.0001,
      "loss": 4.9501,
      "loss/crossentropy": 2.273179054260254,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2917838394641876,
      "step": 5562
    },
    {
      "epoch": 0.11128,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.017899576822916666,
      "learning_rate": 0.0001,
      "loss": 4.774,
      "loss/crossentropy": 2.085157036781311,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2488701120018959,
      "step": 5564
    },
    {
      "epoch": 0.11132,
      "grad_norm": 3.109375,
      "grad_norm_var": 0.05950113932291667,
      "learning_rate": 0.0001,
      "loss": 4.2869,
      "loss/crossentropy": 2.0528116822242737,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24717991054058075,
      "step": 5566
    },
    {
      "epoch": 0.11136,
      "grad_norm": 7.0,
      "grad_norm_var": 1.3981597900390625,
      "learning_rate": 0.0001,
      "loss": 4.4443,
      "loss/crossentropy": 2.0651500821113586,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26806148886680603,
      "step": 5568
    },
    {
      "epoch": 0.1114,
      "grad_norm": 2.328125,
      "grad_norm_var": 1.4721588134765624,
      "learning_rate": 0.0001,
      "loss": 4.6162,
      "loss/crossentropy": 2.1860616207122803,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23045460134744644,
      "step": 5570
    },
    {
      "epoch": 0.11144,
      "grad_norm": 2.328125,
      "grad_norm_var": 1.460399373372396,
      "learning_rate": 0.0001,
      "loss": 4.3629,
      "loss/crossentropy": 1.6931262016296387,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20816385000944138,
      "step": 5572
    },
    {
      "epoch": 0.11148,
      "grad_norm": 2.28125,
      "grad_norm_var": 1.4581858317057292,
      "learning_rate": 0.0001,
      "loss": 4.3376,
      "loss/crossentropy": 2.199341118335724,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2490597665309906,
      "step": 5574
    },
    {
      "epoch": 0.11152,
      "grad_norm": 2.484375,
      "grad_norm_var": 1.4561513264973958,
      "learning_rate": 0.0001,
      "loss": 4.6627,
      "loss/crossentropy": 2.2010069489479065,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24095547199249268,
      "step": 5576
    },
    {
      "epoch": 0.11156,
      "grad_norm": 2.234375,
      "grad_norm_var": 1.4810129801432292,
      "learning_rate": 0.0001,
      "loss": 4.5243,
      "loss/crossentropy": 1.9907150864601135,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2122703418135643,
      "step": 5578
    },
    {
      "epoch": 0.1116,
      "grad_norm": 2.21875,
      "grad_norm_var": 1.5023508707682292,
      "learning_rate": 0.0001,
      "loss": 4.2618,
      "loss/crossentropy": 2.196335554122925,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25111766904592514,
      "step": 5580
    },
    {
      "epoch": 0.11164,
      "grad_norm": 2.171875,
      "grad_norm_var": 1.5003000895182292,
      "learning_rate": 0.0001,
      "loss": 4.5104,
      "loss/crossentropy": 2.0762988924980164,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23773670941591263,
      "step": 5582
    },
    {
      "epoch": 0.11168,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.15530192057291667,
      "learning_rate": 0.0001,
      "loss": 4.3166,
      "loss/crossentropy": 2.0803143978118896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22162869572639465,
      "step": 5584
    },
    {
      "epoch": 0.11172,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.0084136962890625,
      "learning_rate": 0.0001,
      "loss": 4.5596,
      "loss/crossentropy": 2.1821994185447693,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26280316710472107,
      "step": 5586
    },
    {
      "epoch": 0.11176,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.011847941080729167,
      "learning_rate": 0.0001,
      "loss": 4.4328,
      "loss/crossentropy": 2.1899439096450806,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24677179753780365,
      "step": 5588
    },
    {
      "epoch": 0.1118,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.0116851806640625,
      "learning_rate": 0.0001,
      "loss": 4.4361,
      "loss/crossentropy": 2.334734559059143,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2481135129928589,
      "step": 5590
    },
    {
      "epoch": 0.11184,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.007291666666666667,
      "learning_rate": 0.0001,
      "loss": 4.7838,
      "loss/crossentropy": 2.2976341247558594,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2454354390501976,
      "step": 5592
    },
    {
      "epoch": 0.11188,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.007477823893229167,
      "learning_rate": 0.0001,
      "loss": 4.7148,
      "loss/crossentropy": 2.3243749141693115,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2735731601715088,
      "step": 5594
    },
    {
      "epoch": 0.11192,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.007372029622395833,
      "learning_rate": 0.0001,
      "loss": 4.7124,
      "loss/crossentropy": 2.0328271985054016,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21919244527816772,
      "step": 5596
    },
    {
      "epoch": 0.11196,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.0077789306640625,
      "learning_rate": 0.0001,
      "loss": 4.4584,
      "loss/crossentropy": 2.249367594718933,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2577860951423645,
      "step": 5598
    },
    {
      "epoch": 0.112,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.009505208333333333,
      "learning_rate": 0.0001,
      "loss": 4.2594,
      "loss/crossentropy": 1.9844761490821838,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2142435386776924,
      "step": 5600
    },
    {
      "epoch": 0.11204,
      "grad_norm": 2.375,
      "grad_norm_var": 0.0111328125,
      "learning_rate": 0.0001,
      "loss": 4.6043,
      "loss/crossentropy": 2.1334372758865356,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23296397179365158,
      "step": 5602
    },
    {
      "epoch": 0.11208,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.0078084309895833336,
      "learning_rate": 0.0001,
      "loss": 4.5308,
      "loss/crossentropy": 2.119946002960205,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23398507386446,
      "step": 5604
    },
    {
      "epoch": 0.11212,
      "grad_norm": 2.25,
      "grad_norm_var": 0.008040364583333333,
      "learning_rate": 0.0001,
      "loss": 4.5011,
      "loss/crossentropy": 2.0414544343948364,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23420867323875427,
      "step": 5606
    },
    {
      "epoch": 0.11216,
      "grad_norm": 2.125,
      "grad_norm_var": 0.009130859375,
      "learning_rate": 0.0001,
      "loss": 4.2092,
      "loss/crossentropy": 1.9592725038528442,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23922864347696304,
      "step": 5608
    },
    {
      "epoch": 0.1122,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.011188761393229166,
      "learning_rate": 0.0001,
      "loss": 4.4795,
      "loss/crossentropy": 2.150269627571106,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2528265416622162,
      "step": 5610
    },
    {
      "epoch": 0.11224,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.3099772135416667,
      "learning_rate": 0.0001,
      "loss": 4.6234,
      "loss/crossentropy": 2.0591378211975098,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2430240362882614,
      "step": 5612
    },
    {
      "epoch": 0.11228,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.30794169108072916,
      "learning_rate": 0.0001,
      "loss": 4.4251,
      "loss/crossentropy": 2.2132861614227295,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22676552087068558,
      "step": 5614
    },
    {
      "epoch": 0.11232,
      "grad_norm": 2.375,
      "grad_norm_var": 0.3002237955729167,
      "learning_rate": 0.0001,
      "loss": 4.4332,
      "loss/crossentropy": 1.9607917070388794,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23068836331367493,
      "step": 5616
    },
    {
      "epoch": 0.11236,
      "grad_norm": 2.25,
      "grad_norm_var": 0.3021321614583333,
      "learning_rate": 0.0001,
      "loss": 4.549,
      "loss/crossentropy": 2.2245940566062927,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25029121339321136,
      "step": 5618
    },
    {
      "epoch": 0.1124,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.30278218587239586,
      "learning_rate": 0.0001,
      "loss": 4.4539,
      "loss/crossentropy": 2.2511253356933594,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25143079459667206,
      "step": 5620
    },
    {
      "epoch": 0.11244,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.3026194254557292,
      "learning_rate": 0.0001,
      "loss": 4.4632,
      "loss/crossentropy": 2.2945470809936523,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2196483463048935,
      "step": 5622
    },
    {
      "epoch": 0.11248,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.29273681640625,
      "learning_rate": 0.0001,
      "loss": 4.8769,
      "loss/crossentropy": 2.2266393899917603,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24020668864250183,
      "step": 5624
    },
    {
      "epoch": 0.11252,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.29136454264322914,
      "learning_rate": 0.0001,
      "loss": 4.742,
      "loss/crossentropy": 2.2835845947265625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27199871838092804,
      "step": 5626
    },
    {
      "epoch": 0.11256,
      "grad_norm": 2.859375,
      "grad_norm_var": 0.026383463541666666,
      "learning_rate": 0.0001,
      "loss": 4.4213,
      "loss/crossentropy": 1.9576718211174011,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21716968715190887,
      "step": 5628
    },
    {
      "epoch": 0.1126,
      "grad_norm": 2.375,
      "grad_norm_var": 0.044611612955729164,
      "learning_rate": 0.0001,
      "loss": 4.7695,
      "loss/crossentropy": 2.0955676436424255,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24259109795093536,
      "step": 5630
    },
    {
      "epoch": 0.11264,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.044840494791666664,
      "learning_rate": 0.0001,
      "loss": 4.4127,
      "loss/crossentropy": 2.119523346424103,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2478165179491043,
      "step": 5632
    },
    {
      "epoch": 0.11268,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.0444488525390625,
      "learning_rate": 0.0001,
      "loss": 4.5591,
      "loss/crossentropy": 2.189425826072693,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24532486498355865,
      "step": 5634
    },
    {
      "epoch": 0.11272,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.0467193603515625,
      "learning_rate": 0.0001,
      "loss": 4.4969,
      "loss/crossentropy": 2.215874433517456,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24533041566610336,
      "step": 5636
    },
    {
      "epoch": 0.11276,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.04868876139322917,
      "learning_rate": 0.0001,
      "loss": 4.5657,
      "loss/crossentropy": 2.226451873779297,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2545652836561203,
      "step": 5638
    },
    {
      "epoch": 0.1128,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.05181884765625,
      "learning_rate": 0.0001,
      "loss": 4.4779,
      "loss/crossentropy": 2.0343592762947083,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2503928989171982,
      "step": 5640
    },
    {
      "epoch": 0.11284,
      "grad_norm": 2.25,
      "grad_norm_var": 0.0535552978515625,
      "learning_rate": 0.0001,
      "loss": 4.6253,
      "loss/crossentropy": 2.142001748085022,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24624411761760712,
      "step": 5642
    },
    {
      "epoch": 0.11288,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.039383951822916666,
      "learning_rate": 0.0001,
      "loss": 4.4212,
      "loss/crossentropy": 1.822394609451294,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21315739303827286,
      "step": 5644
    },
    {
      "epoch": 0.11292,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.018863932291666666,
      "learning_rate": 0.0001,
      "loss": 4.1611,
      "loss/crossentropy": 2.3221731185913086,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24407917261123657,
      "step": 5646
    },
    {
      "epoch": 0.11296,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.012531534830729166,
      "learning_rate": 0.0001,
      "loss": 4.5591,
      "loss/crossentropy": 1.9784467816352844,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22578515857458115,
      "step": 5648
    },
    {
      "epoch": 0.113,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.0142730712890625,
      "learning_rate": 0.0001,
      "loss": 4.636,
      "loss/crossentropy": 2.2148635387420654,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25031210482120514,
      "step": 5650
    },
    {
      "epoch": 0.11304,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.012369791666666666,
      "learning_rate": 0.0001,
      "loss": 4.4644,
      "loss/crossentropy": 1.9204095602035522,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22073335200548172,
      "step": 5652
    },
    {
      "epoch": 0.11308,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.009764607747395833,
      "learning_rate": 0.0001,
      "loss": 4.6601,
      "loss/crossentropy": 2.092605173587799,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24161123484373093,
      "step": 5654
    },
    {
      "epoch": 0.11312,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.0163726806640625,
      "learning_rate": 0.0001,
      "loss": 4.5977,
      "loss/crossentropy": 1.9546263217926025,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22741927206516266,
      "step": 5656
    },
    {
      "epoch": 0.11316,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.016974894205729167,
      "learning_rate": 0.0001,
      "loss": 4.6585,
      "loss/crossentropy": 2.256605863571167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2560018301010132,
      "step": 5658
    },
    {
      "epoch": 0.1132,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.0157135009765625,
      "learning_rate": 0.0001,
      "loss": 4.3605,
      "loss/crossentropy": 2.24527370929718,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.270521879196167,
      "step": 5660
    },
    {
      "epoch": 0.11324,
      "grad_norm": 2.5,
      "grad_norm_var": 0.014029947916666667,
      "learning_rate": 0.0001,
      "loss": 4.5429,
      "loss/crossentropy": 1.8154722452163696,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22786997258663177,
      "step": 5662
    },
    {
      "epoch": 0.11328,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.0143951416015625,
      "learning_rate": 0.0001,
      "loss": 4.4079,
      "loss/crossentropy": 2.135699689388275,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.253468282520771,
      "step": 5664
    },
    {
      "epoch": 0.11332,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.015087890625,
      "learning_rate": 0.0001,
      "loss": 4.5381,
      "loss/crossentropy": 2.15896338224411,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25651170313358307,
      "step": 5666
    },
    {
      "epoch": 0.11336,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.012035115559895834,
      "learning_rate": 0.0001,
      "loss": 4.7877,
      "loss/crossentropy": 2.115864336490631,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28546491265296936,
      "step": 5668
    },
    {
      "epoch": 0.1134,
      "grad_norm": 2.25,
      "grad_norm_var": 0.012214152018229167,
      "learning_rate": 0.0001,
      "loss": 4.4283,
      "loss/crossentropy": 2.2036256790161133,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2397892326116562,
      "step": 5670
    },
    {
      "epoch": 0.11344,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.008430989583333333,
      "learning_rate": 0.0001,
      "loss": 4.598,
      "loss/crossentropy": 2.4966647624969482,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2984588146209717,
      "step": 5672
    },
    {
      "epoch": 0.11348,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.008202107747395833,
      "learning_rate": 0.0001,
      "loss": 4.5289,
      "loss/crossentropy": 2.051860749721527,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2748461365699768,
      "step": 5674
    },
    {
      "epoch": 0.11352,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.007938639322916666,
      "learning_rate": 0.0001,
      "loss": 4.597,
      "loss/crossentropy": 2.046416461467743,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2320190668106079,
      "step": 5676
    },
    {
      "epoch": 0.11356,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.0054514567057291664,
      "learning_rate": 0.0001,
      "loss": 4.7389,
      "loss/crossentropy": 2.2385342121124268,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21859309077262878,
      "step": 5678
    },
    {
      "epoch": 0.1136,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.006843058268229166,
      "learning_rate": 0.0001,
      "loss": 4.2487,
      "loss/crossentropy": 1.8511550426483154,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22065181285142899,
      "step": 5680
    },
    {
      "epoch": 0.11364,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.031477864583333334,
      "learning_rate": 0.0001,
      "loss": 4.642,
      "loss/crossentropy": 2.304056167602539,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29507844150066376,
      "step": 5682
    },
    {
      "epoch": 0.11368,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.03329671223958333,
      "learning_rate": 0.0001,
      "loss": 4.5519,
      "loss/crossentropy": 1.993275225162506,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33826952427625656,
      "step": 5684
    },
    {
      "epoch": 0.11372,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.03498433430989583,
      "learning_rate": 0.0001,
      "loss": 4.4729,
      "loss/crossentropy": 2.1836347579956055,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24539872258901596,
      "step": 5686
    },
    {
      "epoch": 0.11376,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.03504231770833333,
      "learning_rate": 0.0001,
      "loss": 4.5908,
      "loss/crossentropy": 1.9467885494232178,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2180013582110405,
      "step": 5688
    },
    {
      "epoch": 0.1138,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.04907938639322917,
      "learning_rate": 0.0001,
      "loss": 4.2888,
      "loss/crossentropy": 1.9907563924789429,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2551623433828354,
      "step": 5690
    },
    {
      "epoch": 0.11384,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.049153645833333336,
      "learning_rate": 0.0001,
      "loss": 4.2186,
      "loss/crossentropy": 1.9452654719352722,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24263620376586914,
      "step": 5692
    },
    {
      "epoch": 0.11388,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.050093587239583334,
      "learning_rate": 0.0001,
      "loss": 4.5816,
      "loss/crossentropy": 2.2448811531066895,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2508034110069275,
      "step": 5694
    },
    {
      "epoch": 0.11392,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.055712890625,
      "learning_rate": 0.0001,
      "loss": 4.1868,
      "loss/crossentropy": 1.991935908794403,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22709138691425323,
      "step": 5696
    },
    {
      "epoch": 0.11396,
      "grad_norm": 2.625,
      "grad_norm_var": 0.47700907389322916,
      "learning_rate": 0.0001,
      "loss": 4.8208,
      "loss/crossentropy": 2.0479623675346375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23980189859867096,
      "step": 5698
    },
    {
      "epoch": 0.114,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.4779581705729167,
      "learning_rate": 0.0001,
      "loss": 4.4742,
      "loss/crossentropy": 1.9319151639938354,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2295292615890503,
      "step": 5700
    },
    {
      "epoch": 0.11404,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.47330322265625,
      "learning_rate": 0.0001,
      "loss": 4.32,
      "loss/crossentropy": 1.985447645187378,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.246543288230896,
      "step": 5702
    },
    {
      "epoch": 0.11408,
      "grad_norm": 2.84375,
      "grad_norm_var": 0.478271484375,
      "learning_rate": 0.0001,
      "loss": 4.6602,
      "loss/crossentropy": 2.0016521215438843,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25966253876686096,
      "step": 5704
    },
    {
      "epoch": 0.11412,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.47038472493489586,
      "learning_rate": 0.0001,
      "loss": 4.9207,
      "loss/crossentropy": 2.112374246120453,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3251144737005234,
      "step": 5706
    },
    {
      "epoch": 0.11416,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.4698720296223958,
      "learning_rate": 0.0001,
      "loss": 4.5223,
      "loss/crossentropy": 2.0931158661842346,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2275928407907486,
      "step": 5708
    },
    {
      "epoch": 0.1142,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.46104227701822914,
      "learning_rate": 0.0001,
      "loss": 4.4481,
      "loss/crossentropy": 1.9867743849754333,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2419440597295761,
      "step": 5710
    },
    {
      "epoch": 0.11424,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.4639312744140625,
      "learning_rate": 0.0001,
      "loss": 4.3874,
      "loss/crossentropy": 2.1822216510772705,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2437271624803543,
      "step": 5712
    },
    {
      "epoch": 0.11428,
      "grad_norm": 6.46875,
      "grad_norm_var": 1.1088775634765624,
      "learning_rate": 0.0001,
      "loss": 4.5307,
      "loss/crossentropy": 2.412580370903015,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3222763389348984,
      "step": 5714
    },
    {
      "epoch": 0.11432,
      "grad_norm": 2.28125,
      "grad_norm_var": 1.1019490559895833,
      "learning_rate": 0.0001,
      "loss": 4.2491,
      "loss/crossentropy": 1.676392376422882,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24499019235372543,
      "step": 5716
    },
    {
      "epoch": 0.11436,
      "grad_norm": 2.328125,
      "grad_norm_var": 1.1039876302083333,
      "learning_rate": 0.0001,
      "loss": 4.4741,
      "loss/crossentropy": 1.7818856835365295,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22935030609369278,
      "step": 5718
    },
    {
      "epoch": 0.1144,
      "grad_norm": 2.40625,
      "grad_norm_var": 1.0944732666015624,
      "learning_rate": 0.0001,
      "loss": 4.5445,
      "loss/crossentropy": 2.012014925479889,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.231942281126976,
      "step": 5720
    },
    {
      "epoch": 0.11444,
      "grad_norm": 2.296875,
      "grad_norm_var": 1.1023834228515625,
      "learning_rate": 0.0001,
      "loss": 4.5061,
      "loss/crossentropy": 2.3663965463638306,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.254768967628479,
      "step": 5722
    },
    {
      "epoch": 0.11448,
      "grad_norm": 2.34375,
      "grad_norm_var": 1.107982381184896,
      "learning_rate": 0.0001,
      "loss": 4.5878,
      "loss/crossentropy": 2.343206286430359,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26609115302562714,
      "step": 5724
    },
    {
      "epoch": 0.11452,
      "grad_norm": 2.21875,
      "grad_norm_var": 1.1131337483723958,
      "learning_rate": 0.0001,
      "loss": 4.5047,
      "loss/crossentropy": 1.8696978092193604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24282050132751465,
      "step": 5726
    },
    {
      "epoch": 0.11456,
      "grad_norm": 2.0625,
      "grad_norm_var": 1.1170644124348958,
      "learning_rate": 0.0001,
      "loss": 4.2198,
      "loss/crossentropy": 2.1430450677871704,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23211582750082016,
      "step": 5728
    },
    {
      "epoch": 0.1146,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.013841756184895833,
      "learning_rate": 0.0001,
      "loss": 4.6868,
      "loss/crossentropy": 2.2231308221817017,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23736387491226196,
      "step": 5730
    },
    {
      "epoch": 0.11464,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.014069620768229167,
      "learning_rate": 0.0001,
      "loss": 4.4911,
      "loss/crossentropy": 1.8789280652999878,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24081497639417648,
      "step": 5732
    },
    {
      "epoch": 0.11468,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.014875284830729167,
      "learning_rate": 0.0001,
      "loss": 4.4478,
      "loss/crossentropy": 2.0491732358932495,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20530463755130768,
      "step": 5734
    },
    {
      "epoch": 0.11472,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.012516276041666666,
      "learning_rate": 0.0001,
      "loss": 4.4352,
      "loss/crossentropy": 2.067046642303467,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24701344966888428,
      "step": 5736
    },
    {
      "epoch": 0.11476,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.012791951497395834,
      "learning_rate": 0.0001,
      "loss": 4.4515,
      "loss/crossentropy": 2.0207647681236267,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2243770807981491,
      "step": 5738
    },
    {
      "epoch": 0.1148,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.013084920247395833,
      "learning_rate": 0.0001,
      "loss": 4.9205,
      "loss/crossentropy": 2.230514347553253,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24377264082431793,
      "step": 5740
    },
    {
      "epoch": 0.11484,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.013997395833333334,
      "learning_rate": 0.0001,
      "loss": 4.3831,
      "loss/crossentropy": 2.38068687915802,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24760407954454422,
      "step": 5742
    },
    {
      "epoch": 0.11488,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.012483723958333333,
      "learning_rate": 0.0001,
      "loss": 4.5598,
      "loss/crossentropy": 2.238909125328064,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2503668889403343,
      "step": 5744
    },
    {
      "epoch": 0.11492,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.015265909830729167,
      "learning_rate": 0.0001,
      "loss": 4.4848,
      "loss/crossentropy": 1.7423101663589478,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20424582809209824,
      "step": 5746
    },
    {
      "epoch": 0.11496,
      "grad_norm": 2.25,
      "grad_norm_var": 0.0181304931640625,
      "learning_rate": 0.0001,
      "loss": 4.6754,
      "loss/crossentropy": 2.5906827449798584,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24731651693582535,
      "step": 5748
    },
    {
      "epoch": 0.115,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.025830078125,
      "learning_rate": 0.0001,
      "loss": 4.7427,
      "loss/crossentropy": 2.418861746788025,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2505089193582535,
      "step": 5750
    },
    {
      "epoch": 0.11504,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.03658447265625,
      "learning_rate": 0.0001,
      "loss": 4.869,
      "loss/crossentropy": 2.158658504486084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2381347045302391,
      "step": 5752
    },
    {
      "epoch": 0.11508,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.03860270182291667,
      "learning_rate": 0.0001,
      "loss": 4.5073,
      "loss/crossentropy": 2.0938435196876526,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23639065027236938,
      "step": 5754
    },
    {
      "epoch": 0.11512,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0441314697265625,
      "learning_rate": 0.0001,
      "loss": 4.2227,
      "loss/crossentropy": 2.023799479007721,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24510329961776733,
      "step": 5756
    },
    {
      "epoch": 0.11516,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.03854166666666667,
      "learning_rate": 0.0001,
      "loss": 4.5155,
      "loss/crossentropy": 2.3589184284210205,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2778936177492142,
      "step": 5758
    },
    {
      "epoch": 0.1152,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.041731770833333334,
      "learning_rate": 0.0001,
      "loss": 4.4209,
      "loss/crossentropy": 2.4897998571395874,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23546544462442398,
      "step": 5760
    },
    {
      "epoch": 0.11524,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.041747029622395834,
      "learning_rate": 0.0001,
      "loss": 4.4619,
      "loss/crossentropy": 2.0426196455955505,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22708147019147873,
      "step": 5762
    },
    {
      "epoch": 0.11528,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.0397369384765625,
      "learning_rate": 0.0001,
      "loss": 4.6661,
      "loss/crossentropy": 2.2582051753997803,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25646351277828217,
      "step": 5764
    },
    {
      "epoch": 0.11532,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.03430582682291667,
      "learning_rate": 0.0001,
      "loss": 4.5929,
      "loss/crossentropy": 2.181105613708496,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2380223199725151,
      "step": 5766
    },
    {
      "epoch": 0.11536,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.026200358072916666,
      "learning_rate": 0.0001,
      "loss": 4.5078,
      "loss/crossentropy": 2.236580967903137,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26678355038166046,
      "step": 5768
    },
    {
      "epoch": 0.1154,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.020361328125,
      "learning_rate": 0.0001,
      "loss": 4.6232,
      "loss/crossentropy": 1.9383749961853027,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24920085072517395,
      "step": 5770
    },
    {
      "epoch": 0.11544,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.018529256184895832,
      "learning_rate": 0.0001,
      "loss": 4.6044,
      "loss/crossentropy": 2.2856688499450684,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.258881650865078,
      "step": 5772
    },
    {
      "epoch": 0.11548,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.0239410400390625,
      "learning_rate": 0.0001,
      "loss": 4.8391,
      "loss/crossentropy": 2.2897390127182007,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24453644454479218,
      "step": 5774
    },
    {
      "epoch": 0.11552,
      "grad_norm": 2.25,
      "grad_norm_var": 0.014549763997395833,
      "learning_rate": 0.0001,
      "loss": 4.5698,
      "loss/crossentropy": 2.0502785444259644,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24106161296367645,
      "step": 5776
    },
    {
      "epoch": 0.11556,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.01441650390625,
      "learning_rate": 0.0001,
      "loss": 4.5665,
      "loss/crossentropy": 2.344050645828247,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24690424650907516,
      "step": 5778
    },
    {
      "epoch": 0.1156,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.021239217122395834,
      "learning_rate": 0.0001,
      "loss": 4.5273,
      "loss/crossentropy": 2.2274389266967773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25618284940719604,
      "step": 5780
    },
    {
      "epoch": 0.11564,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.022184244791666665,
      "learning_rate": 0.0001,
      "loss": 4.4888,
      "loss/crossentropy": 2.063184678554535,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22606099396944046,
      "step": 5782
    },
    {
      "epoch": 0.11568,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.0206451416015625,
      "learning_rate": 0.0001,
      "loss": 4.2333,
      "loss/crossentropy": 2.093947410583496,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23588553071022034,
      "step": 5784
    },
    {
      "epoch": 0.11572,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0246490478515625,
      "learning_rate": 0.0001,
      "loss": 4.426,
      "loss/crossentropy": 2.1599318981170654,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20322780311107635,
      "step": 5786
    },
    {
      "epoch": 0.11576,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.02506103515625,
      "learning_rate": 0.0001,
      "loss": 4.5398,
      "loss/crossentropy": 2.274693012237549,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2704206556081772,
      "step": 5788
    },
    {
      "epoch": 0.1158,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.0222076416015625,
      "learning_rate": 0.0001,
      "loss": 4.6755,
      "loss/crossentropy": 1.9712103009223938,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28300437331199646,
      "step": 5790
    },
    {
      "epoch": 0.11584,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.024344889322916667,
      "learning_rate": 0.0001,
      "loss": 4.2035,
      "loss/crossentropy": 2.027747690677643,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24419110268354416,
      "step": 5792
    },
    {
      "epoch": 0.11588,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.024251302083333332,
      "learning_rate": 0.0001,
      "loss": 4.5614,
      "loss/crossentropy": 2.162364959716797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2522001415491104,
      "step": 5794
    },
    {
      "epoch": 0.11592,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.0166656494140625,
      "learning_rate": 0.0001,
      "loss": 4.4711,
      "loss/crossentropy": 2.558881998062134,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26168718934059143,
      "step": 5796
    },
    {
      "epoch": 0.11596,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0162994384765625,
      "learning_rate": 0.0001,
      "loss": 4.4108,
      "loss/crossentropy": 2.1027071475982666,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24049362540245056,
      "step": 5798
    },
    {
      "epoch": 0.116,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.015034993489583334,
      "learning_rate": 0.0001,
      "loss": 4.6059,
      "loss/crossentropy": 2.488932490348816,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2697457820177078,
      "step": 5800
    },
    {
      "epoch": 0.11604,
      "grad_norm": 2.375,
      "grad_norm_var": 0.0121734619140625,
      "learning_rate": 0.0001,
      "loss": 4.6706,
      "loss/crossentropy": 2.0441418886184692,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23980429768562317,
      "step": 5802
    },
    {
      "epoch": 0.11608,
      "grad_norm": 2.375,
      "grad_norm_var": 0.0107421875,
      "learning_rate": 0.0001,
      "loss": 4.8471,
      "loss/crossentropy": 2.2873395681381226,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3105090409517288,
      "step": 5804
    },
    {
      "epoch": 0.11612,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.00826416015625,
      "learning_rate": 0.0001,
      "loss": 4.5388,
      "loss/crossentropy": 1.8773444890975952,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20640414953231812,
      "step": 5806
    },
    {
      "epoch": 0.11616,
      "grad_norm": 2.375,
      "grad_norm_var": 0.007502237955729167,
      "learning_rate": 0.0001,
      "loss": 4.7338,
      "loss/crossentropy": 2.3736027479171753,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2679155319929123,
      "step": 5808
    },
    {
      "epoch": 0.1162,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.010676066080729166,
      "learning_rate": 0.0001,
      "loss": 4.0496,
      "loss/crossentropy": 2.071690082550049,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21868111193180084,
      "step": 5810
    },
    {
      "epoch": 0.11624,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.015425618489583333,
      "learning_rate": 0.0001,
      "loss": 4.4029,
      "loss/crossentropy": 2.0531184673309326,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24457989633083344,
      "step": 5812
    },
    {
      "epoch": 0.11628,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.015718587239583335,
      "learning_rate": 0.0001,
      "loss": 4.6511,
      "loss/crossentropy": 2.1161271929740906,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22809523344039917,
      "step": 5814
    },
    {
      "epoch": 0.11632,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0196929931640625,
      "learning_rate": 0.0001,
      "loss": 4.2736,
      "loss/crossentropy": 1.6557151675224304,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20782940834760666,
      "step": 5816
    },
    {
      "epoch": 0.11636,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.03704020182291667,
      "learning_rate": 0.0001,
      "loss": 4.5977,
      "loss/crossentropy": 2.2499040365219116,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2713513821363449,
      "step": 5818
    },
    {
      "epoch": 0.1164,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.03723958333333333,
      "learning_rate": 0.0001,
      "loss": 4.6417,
      "loss/crossentropy": 2.3616446256637573,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2613145411014557,
      "step": 5820
    },
    {
      "epoch": 0.11644,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.0353515625,
      "learning_rate": 0.0001,
      "loss": 4.3744,
      "loss/crossentropy": 2.0932790637016296,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23843754082918167,
      "step": 5822
    },
    {
      "epoch": 0.11648,
      "grad_norm": 2.25,
      "grad_norm_var": 0.03540751139322917,
      "learning_rate": 0.0001,
      "loss": 4.4089,
      "loss/crossentropy": 2.128177046775818,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26579540967941284,
      "step": 5824
    },
    {
      "epoch": 0.11652,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.031281534830729166,
      "learning_rate": 0.0001,
      "loss": 4.6942,
      "loss/crossentropy": 2.332372784614563,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27953268587589264,
      "step": 5826
    },
    {
      "epoch": 0.11656,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0263092041015625,
      "learning_rate": 0.0001,
      "loss": 4.4615,
      "loss/crossentropy": 2.360959053039551,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26249848306179047,
      "step": 5828
    },
    {
      "epoch": 0.1166,
      "grad_norm": 2.0,
      "grad_norm_var": 0.030745442708333334,
      "learning_rate": 0.0001,
      "loss": 3.9757,
      "loss/crossentropy": 1.9800339341163635,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22766301035881042,
      "step": 5830
    },
    {
      "epoch": 0.11664,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.029442342122395833,
      "learning_rate": 0.0001,
      "loss": 4.3287,
      "loss/crossentropy": 2.082980155944824,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2368428260087967,
      "step": 5832
    },
    {
      "epoch": 0.11668,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.011986287434895833,
      "learning_rate": 0.0001,
      "loss": 4.4441,
      "loss/crossentropy": 2.093027710914612,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21121951937675476,
      "step": 5834
    },
    {
      "epoch": 0.11672,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.011031087239583333,
      "learning_rate": 0.0001,
      "loss": 4.5008,
      "loss/crossentropy": 2.1329175233840942,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25303877890110016,
      "step": 5836
    },
    {
      "epoch": 0.11676,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.022932942708333334,
      "learning_rate": 0.0001,
      "loss": 4.5174,
      "loss/crossentropy": 1.79305762052536,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20448636263608932,
      "step": 5838
    },
    {
      "epoch": 0.1168,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.024933878580729166,
      "learning_rate": 0.0001,
      "loss": 4.2722,
      "loss/crossentropy": 1.957836627960205,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22745755314826965,
      "step": 5840
    },
    {
      "epoch": 0.11684,
      "grad_norm": 2.25,
      "grad_norm_var": 0.025007120768229165,
      "learning_rate": 0.0001,
      "loss": 4.3773,
      "loss/crossentropy": 2.1167174577713013,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2437898963689804,
      "step": 5842
    },
    {
      "epoch": 0.11688,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.024800618489583332,
      "learning_rate": 0.0001,
      "loss": 4.4856,
      "loss/crossentropy": 2.3288447856903076,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2355937883257866,
      "step": 5844
    },
    {
      "epoch": 0.11692,
      "grad_norm": 2.125,
      "grad_norm_var": 0.022337849934895834,
      "learning_rate": 0.0001,
      "loss": 4.4774,
      "loss/crossentropy": 2.2526416778564453,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24917340278625488,
      "step": 5846
    },
    {
      "epoch": 0.11696,
      "grad_norm": 2.25,
      "grad_norm_var": 0.0198883056640625,
      "learning_rate": 0.0001,
      "loss": 4.391,
      "loss/crossentropy": 2.194224774837494,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2625589966773987,
      "step": 5848
    },
    {
      "epoch": 0.117,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.020116170247395832,
      "learning_rate": 0.0001,
      "loss": 4.4955,
      "loss/crossentropy": 2.0156877040863037,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22422078251838684,
      "step": 5850
    },
    {
      "epoch": 0.11704,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.01962890625,
      "learning_rate": 0.0001,
      "loss": 4.6231,
      "loss/crossentropy": 2.2480785846710205,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24705884605646133,
      "step": 5852
    },
    {
      "epoch": 0.11708,
      "grad_norm": 2.25,
      "grad_norm_var": 0.0078521728515625,
      "learning_rate": 0.0001,
      "loss": 4.4817,
      "loss/crossentropy": 2.0915993452072144,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24669666588306427,
      "step": 5854
    },
    {
      "epoch": 0.11712,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.0062652587890625,
      "learning_rate": 0.0001,
      "loss": 4.4444,
      "loss/crossentropy": 2.1283876299858093,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24228712916374207,
      "step": 5856
    },
    {
      "epoch": 0.11716,
      "grad_norm": 2.625,
      "grad_norm_var": 0.015523274739583334,
      "learning_rate": 0.0001,
      "loss": 4.6582,
      "loss/crossentropy": 2.028861939907074,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2281404584646225,
      "step": 5858
    },
    {
      "epoch": 0.1172,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.015852864583333334,
      "learning_rate": 0.0001,
      "loss": 4.2293,
      "loss/crossentropy": 2.154610753059387,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24233703315258026,
      "step": 5860
    },
    {
      "epoch": 0.11724,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.015787760416666668,
      "learning_rate": 0.0001,
      "loss": 4.5059,
      "loss/crossentropy": 1.9396602511405945,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22303076088428497,
      "step": 5862
    },
    {
      "epoch": 0.11728,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.021061197916666666,
      "learning_rate": 0.0001,
      "loss": 4.7447,
      "loss/crossentropy": 2.053893029689789,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26897912472486496,
      "step": 5864
    },
    {
      "epoch": 0.11732,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0263336181640625,
      "learning_rate": 0.0001,
      "loss": 4.3529,
      "loss/crossentropy": 1.990949273109436,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22180908918380737,
      "step": 5866
    },
    {
      "epoch": 0.11736,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.030085245768229168,
      "learning_rate": 0.0001,
      "loss": 4.6434,
      "loss/crossentropy": 2.0929455161094666,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24055248498916626,
      "step": 5868
    },
    {
      "epoch": 0.1174,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.03168843587239583,
      "learning_rate": 0.0001,
      "loss": 4.5465,
      "loss/crossentropy": 2.1476733684539795,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2416895627975464,
      "step": 5870
    },
    {
      "epoch": 0.11744,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.027962239583333333,
      "learning_rate": 0.0001,
      "loss": 4.5633,
      "loss/crossentropy": 1.74330335855484,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22001302987337112,
      "step": 5872
    },
    {
      "epoch": 0.11748,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.021507771809895833,
      "learning_rate": 0.0001,
      "loss": 4.2725,
      "loss/crossentropy": 1.8903921246528625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21950766444206238,
      "step": 5874
    },
    {
      "epoch": 0.11752,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.022069295247395832,
      "learning_rate": 0.0001,
      "loss": 4.3673,
      "loss/crossentropy": 1.798406720161438,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20271167159080505,
      "step": 5876
    },
    {
      "epoch": 0.11756,
      "grad_norm": 2.25,
      "grad_norm_var": 0.021089680989583335,
      "learning_rate": 0.0001,
      "loss": 4.7128,
      "loss/crossentropy": 1.9651959538459778,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21256517618894577,
      "step": 5878
    },
    {
      "epoch": 0.1176,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.016844685872395834,
      "learning_rate": 0.0001,
      "loss": 4.4158,
      "loss/crossentropy": 2.039245307445526,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23225219547748566,
      "step": 5880
    },
    {
      "epoch": 0.11764,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.013329060872395833,
      "learning_rate": 0.0001,
      "loss": 4.8741,
      "loss/crossentropy": 2.500381350517273,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31757834553718567,
      "step": 5882
    },
    {
      "epoch": 0.11768,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.018195597330729167,
      "learning_rate": 0.0001,
      "loss": 4.6463,
      "loss/crossentropy": 2.0540305972099304,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2474198415875435,
      "step": 5884
    },
    {
      "epoch": 0.11772,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.016462198893229165,
      "learning_rate": 0.0001,
      "loss": 4.5047,
      "loss/crossentropy": 2.072624385356903,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2482014298439026,
      "step": 5886
    },
    {
      "epoch": 0.11776,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.017853800455729166,
      "learning_rate": 0.0001,
      "loss": 4.4025,
      "loss/crossentropy": 2.153423309326172,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.234448604285717,
      "step": 5888
    },
    {
      "epoch": 0.1178,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.025325520833333334,
      "learning_rate": 0.0001,
      "loss": 4.2698,
      "loss/crossentropy": 1.5941627621650696,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20982372760772705,
      "step": 5890
    },
    {
      "epoch": 0.11784,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.025886027018229167,
      "learning_rate": 0.0001,
      "loss": 4.5168,
      "loss/crossentropy": 2.2858930826187134,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24936140328645706,
      "step": 5892
    },
    {
      "epoch": 0.11788,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.028076171875,
      "learning_rate": 0.0001,
      "loss": 4.536,
      "loss/crossentropy": 2.333559274673462,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26152122020721436,
      "step": 5894
    },
    {
      "epoch": 0.11792,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.027106730143229167,
      "learning_rate": 0.0001,
      "loss": 4.412,
      "loss/crossentropy": 2.134613037109375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24929091334342957,
      "step": 5896
    },
    {
      "epoch": 0.11796,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.026423136393229168,
      "learning_rate": 0.0001,
      "loss": 4.4052,
      "loss/crossentropy": 2.179764688014984,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24523546546697617,
      "step": 5898
    },
    {
      "epoch": 0.118,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0204742431640625,
      "learning_rate": 0.0001,
      "loss": 4.3431,
      "loss/crossentropy": 2.1184223294258118,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21715932339429855,
      "step": 5900
    },
    {
      "epoch": 0.11804,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.021882120768229166,
      "learning_rate": 0.0001,
      "loss": 4.5606,
      "loss/crossentropy": 2.024593770503998,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23016374558210373,
      "step": 5902
    },
    {
      "epoch": 0.11808,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.0204986572265625,
      "learning_rate": 0.0001,
      "loss": 4.3646,
      "loss/crossentropy": 2.077186107635498,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2538782134652138,
      "step": 5904
    },
    {
      "epoch": 0.11812,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.013190714518229167,
      "learning_rate": 0.0001,
      "loss": 4.9135,
      "loss/crossentropy": 2.2535945177078247,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25025132298469543,
      "step": 5906
    },
    {
      "epoch": 0.11816,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.01064453125,
      "learning_rate": 0.0001,
      "loss": 4.5512,
      "loss/crossentropy": 2.5321284532546997,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2729681059718132,
      "step": 5908
    },
    {
      "epoch": 0.1182,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.006403605143229167,
      "learning_rate": 0.0001,
      "loss": 4.332,
      "loss/crossentropy": 2.043885111808777,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23721691966056824,
      "step": 5910
    },
    {
      "epoch": 0.11824,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.005204264322916667,
      "learning_rate": 0.0001,
      "loss": 4.3901,
      "loss/crossentropy": 2.1343676447868347,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23325734585523605,
      "step": 5912
    },
    {
      "epoch": 0.11828,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0074045817057291664,
      "learning_rate": 0.0001,
      "loss": 4.3551,
      "loss/crossentropy": 2.1164477467536926,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24526448547840118,
      "step": 5914
    },
    {
      "epoch": 0.11832,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0064737955729166664,
      "learning_rate": 0.0001,
      "loss": 4.5162,
      "loss/crossentropy": 2.268216848373413,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24837128818035126,
      "step": 5916
    },
    {
      "epoch": 0.11836,
      "grad_norm": 2.25,
      "grad_norm_var": 0.0057037353515625,
      "learning_rate": 0.0001,
      "loss": 4.4636,
      "loss/crossentropy": 2.074695885181427,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24956009536981583,
      "step": 5918
    },
    {
      "epoch": 0.1184,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.007372029622395833,
      "learning_rate": 0.0001,
      "loss": 4.3165,
      "loss/crossentropy": 1.9369722604751587,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2316487580537796,
      "step": 5920
    },
    {
      "epoch": 0.11844,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.00611572265625,
      "learning_rate": 0.0001,
      "loss": 4.2789,
      "loss/crossentropy": 2.2189531326293945,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23884039372205734,
      "step": 5922
    },
    {
      "epoch": 0.11848,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.007323201497395833,
      "learning_rate": 0.0001,
      "loss": 4.5965,
      "loss/crossentropy": 2.3833028078079224,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2504591718316078,
      "step": 5924
    },
    {
      "epoch": 0.11852,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.007005818684895833,
      "learning_rate": 0.0001,
      "loss": 4.6346,
      "loss/crossentropy": 2.0443845987319946,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23422807455062866,
      "step": 5926
    },
    {
      "epoch": 0.11856,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.0073964436848958336,
      "learning_rate": 0.0001,
      "loss": 4.9061,
      "loss/crossentropy": 2.223625063896179,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24985776841640472,
      "step": 5928
    },
    {
      "epoch": 0.1186,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.005126953125,
      "learning_rate": 0.0001,
      "loss": 4.493,
      "loss/crossentropy": 2.1353545784950256,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22846727818250656,
      "step": 5930
    },
    {
      "epoch": 0.11864,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.005159505208333333,
      "learning_rate": 0.0001,
      "loss": 4.5786,
      "loss/crossentropy": 2.0497827529907227,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21461189538240433,
      "step": 5932
    },
    {
      "epoch": 0.11868,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.004715983072916667,
      "learning_rate": 0.0001,
      "loss": 4.5071,
      "loss/crossentropy": 1.9257362484931946,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2216043472290039,
      "step": 5934
    },
    {
      "epoch": 0.11872,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.0034006754557291668,
      "learning_rate": 0.0001,
      "loss": 4.4212,
      "loss/crossentropy": 2.0458216071128845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23162957280874252,
      "step": 5936
    },
    {
      "epoch": 0.11876,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0033274332682291666,
      "learning_rate": 0.0001,
      "loss": 4.6109,
      "loss/crossentropy": 2.0786396861076355,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22303655743598938,
      "step": 5938
    },
    {
      "epoch": 0.1188,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.04327799479166667,
      "learning_rate": 0.0001,
      "loss": 4.4861,
      "loss/crossentropy": 1.9800568222999573,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22236331552267075,
      "step": 5940
    },
    {
      "epoch": 0.11884,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0502349853515625,
      "learning_rate": 0.0001,
      "loss": 4.056,
      "loss/crossentropy": 1.882250189781189,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21664132922887802,
      "step": 5942
    },
    {
      "epoch": 0.11888,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.05025634765625,
      "learning_rate": 0.0001,
      "loss": 4.5965,
      "loss/crossentropy": 2.3682695627212524,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2698971778154373,
      "step": 5944
    },
    {
      "epoch": 0.11892,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.0505523681640625,
      "learning_rate": 0.0001,
      "loss": 4.6364,
      "loss/crossentropy": 2.225574493408203,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2642917186021805,
      "step": 5946
    },
    {
      "epoch": 0.11896,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.05032145182291667,
      "learning_rate": 0.0001,
      "loss": 4.3157,
      "loss/crossentropy": 1.8634169697761536,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2170402556657791,
      "step": 5948
    },
    {
      "epoch": 0.119,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.05123291015625,
      "learning_rate": 0.0001,
      "loss": 5.0392,
      "loss/crossentropy": 2.4500025510787964,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27845144271850586,
      "step": 5950
    },
    {
      "epoch": 0.11904,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.054108683268229166,
      "learning_rate": 0.0001,
      "loss": 4.1179,
      "loss/crossentropy": 2.1532052755355835,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24952176213264465,
      "step": 5952
    },
    {
      "epoch": 0.11908,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0561676025390625,
      "learning_rate": 0.0001,
      "loss": 4.2592,
      "loss/crossentropy": 2.066560387611389,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2356283888220787,
      "step": 5954
    },
    {
      "epoch": 0.11912,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.01021728515625,
      "learning_rate": 0.0001,
      "loss": 4.6187,
      "loss/crossentropy": 1.9679089784622192,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23729706555604935,
      "step": 5956
    },
    {
      "epoch": 0.11916,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.0071441650390625,
      "learning_rate": 0.0001,
      "loss": 4.3463,
      "loss/crossentropy": 2.3490394353866577,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26400282233953476,
      "step": 5958
    },
    {
      "epoch": 0.1192,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.008576456705729167,
      "learning_rate": 0.0001,
      "loss": 4.3629,
      "loss/crossentropy": 2.145757555961609,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23329483717679977,
      "step": 5960
    },
    {
      "epoch": 0.11924,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.005399576822916667,
      "learning_rate": 0.0001,
      "loss": 4.2376,
      "loss/crossentropy": 2.0764617919921875,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22371648252010345,
      "step": 5962
    },
    {
      "epoch": 0.11928,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.010380045572916666,
      "learning_rate": 0.0001,
      "loss": 4.8106,
      "loss/crossentropy": 2.199389696121216,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23210449516773224,
      "step": 5964
    },
    {
      "epoch": 0.11932,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.01064453125,
      "learning_rate": 0.0001,
      "loss": 4.3515,
      "loss/crossentropy": 1.9008439183235168,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21474837511777878,
      "step": 5966
    },
    {
      "epoch": 0.11936,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0148590087890625,
      "learning_rate": 0.0001,
      "loss": 4.4603,
      "loss/crossentropy": 2.1779539585113525,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23485051095485687,
      "step": 5968
    },
    {
      "epoch": 0.1194,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.014176432291666667,
      "learning_rate": 0.0001,
      "loss": 4.428,
      "loss/crossentropy": 2.267147421836853,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24924689531326294,
      "step": 5970
    },
    {
      "epoch": 0.11944,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.016292317708333334,
      "learning_rate": 0.0001,
      "loss": 4.0856,
      "loss/crossentropy": 2.0618110299110413,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24352984875440598,
      "step": 5972
    },
    {
      "epoch": 0.11948,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.023758951822916666,
      "learning_rate": 0.0001,
      "loss": 4.5802,
      "loss/crossentropy": 2.1419676542282104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23680058121681213,
      "step": 5974
    },
    {
      "epoch": 0.11952,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.023368326822916667,
      "learning_rate": 0.0001,
      "loss": 4.6554,
      "loss/crossentropy": 2.0376622080802917,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22515031695365906,
      "step": 5976
    },
    {
      "epoch": 0.11956,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.025739542643229165,
      "learning_rate": 0.0001,
      "loss": 4.4213,
      "loss/crossentropy": 1.7675965428352356,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.219834603369236,
      "step": 5978
    },
    {
      "epoch": 0.1196,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.024030558268229165,
      "learning_rate": 0.0001,
      "loss": 4.2886,
      "loss/crossentropy": 1.8919905424118042,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2256327122449875,
      "step": 5980
    },
    {
      "epoch": 0.11964,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.0228424072265625,
      "learning_rate": 0.0001,
      "loss": 4.6148,
      "loss/crossentropy": 2.287980794906616,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25571687519550323,
      "step": 5982
    },
    {
      "epoch": 0.11968,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.0169830322265625,
      "learning_rate": 0.0001,
      "loss": 4.3527,
      "loss/crossentropy": 2.1030094027519226,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23401429504156113,
      "step": 5984
    },
    {
      "epoch": 0.11972,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.0166412353515625,
      "learning_rate": 0.0001,
      "loss": 4.8857,
      "loss/crossentropy": 2.469533920288086,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2548582851886749,
      "step": 5986
    },
    {
      "epoch": 0.11976,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.018047841389973958,
      "learning_rate": 0.0001,
      "loss": 4.167,
      "loss/crossentropy": 2.140324354171753,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22603372484445572,
      "step": 5988
    },
    {
      "epoch": 0.1198,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.015773264567057292,
      "learning_rate": 0.0001,
      "loss": 4.3848,
      "loss/crossentropy": 2.1848061084747314,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22712922096252441,
      "step": 5990
    },
    {
      "epoch": 0.11984,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.015380605061848959,
      "learning_rate": 0.0001,
      "loss": 4.6457,
      "loss/crossentropy": 2.2872358560562134,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2352529615163803,
      "step": 5992
    },
    {
      "epoch": 0.11988,
      "grad_norm": 2.25,
      "grad_norm_var": 0.012318674723307292,
      "learning_rate": 0.0001,
      "loss": 4.6136,
      "loss/crossentropy": 2.082811713218689,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24107889831066132,
      "step": 5994
    },
    {
      "epoch": 0.11992,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.013602447509765626,
      "learning_rate": 0.0001,
      "loss": 4.2089,
      "loss/crossentropy": 2.23664391040802,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23429522663354874,
      "step": 5996
    },
    {
      "epoch": 0.11996,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.013242340087890625,
      "learning_rate": 0.0001,
      "loss": 4.5237,
      "loss/crossentropy": 2.451270341873169,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2541816979646683,
      "step": 5998
    },
    {
      "epoch": 0.12,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.013561757405598958,
      "learning_rate": 0.0001,
      "loss": 4.5559,
      "loss/crossentropy": 2.1744157671928406,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22349119931459427,
      "step": 6000
    },
    {
      "epoch": 0.12004,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.013171132405598958,
      "learning_rate": 0.0001,
      "loss": 4.5532,
      "loss/crossentropy": 2.0316836833953857,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.259635865688324,
      "step": 6002
    },
    {
      "epoch": 0.12008,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.022337849934895834,
      "learning_rate": 0.0001,
      "loss": 4.3674,
      "loss/crossentropy": 2.0989437103271484,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25056491047143936,
      "step": 6004
    },
    {
      "epoch": 0.12012,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.02633056640625,
      "learning_rate": 0.0001,
      "loss": 4.0883,
      "loss/crossentropy": 1.9609100818634033,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22969383746385574,
      "step": 6006
    },
    {
      "epoch": 0.12016,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.027665201822916666,
      "learning_rate": 0.0001,
      "loss": 4.2094,
      "loss/crossentropy": 2.077883243560791,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22850078344345093,
      "step": 6008
    },
    {
      "epoch": 0.1202,
      "grad_norm": 2.34375,
      "grad_norm_var": 1.6465779622395833,
      "learning_rate": 0.0001,
      "loss": 4.5659,
      "loss/crossentropy": 1.7967591285705566,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2126556932926178,
      "step": 6010
    },
    {
      "epoch": 0.12024,
      "grad_norm": 2.234375,
      "grad_norm_var": 1.62222900390625,
      "learning_rate": 0.0001,
      "loss": 4.4018,
      "loss/crossentropy": 1.6516226530075073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2188895046710968,
      "step": 6012
    },
    {
      "epoch": 0.12028,
      "grad_norm": 2.21875,
      "grad_norm_var": 1.62984619140625,
      "learning_rate": 0.0001,
      "loss": 4.3303,
      "loss/crossentropy": 2.161388635635376,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23447516560554504,
      "step": 6014
    },
    {
      "epoch": 0.12032,
      "grad_norm": 2.328125,
      "grad_norm_var": 1.6235636393229167,
      "learning_rate": 0.0001,
      "loss": 4.5112,
      "loss/crossentropy": 1.9205461740493774,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23797179013490677,
      "step": 6016
    },
    {
      "epoch": 0.12036,
      "grad_norm": 2.3125,
      "grad_norm_var": 1.616844685872396,
      "learning_rate": 0.0001,
      "loss": 4.8019,
      "loss/crossentropy": 2.025223135948181,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25120896100997925,
      "step": 6018
    },
    {
      "epoch": 0.1204,
      "grad_norm": 2.25,
      "grad_norm_var": 1.6347005208333334,
      "learning_rate": 0.0001,
      "loss": 4.4819,
      "loss/crossentropy": 1.957942008972168,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22228525578975677,
      "step": 6020
    },
    {
      "epoch": 0.12044,
      "grad_norm": 2.171875,
      "grad_norm_var": 1.674201456705729,
      "learning_rate": 0.0001,
      "loss": 4.6193,
      "loss/crossentropy": 2.3325445652008057,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26250994950532913,
      "step": 6022
    },
    {
      "epoch": 0.12048,
      "grad_norm": 2.03125,
      "grad_norm_var": 1.6957997639973958,
      "learning_rate": 0.0001,
      "loss": 4.264,
      "loss/crossentropy": 2.0049667954444885,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22990728169679642,
      "step": 6024
    },
    {
      "epoch": 0.12052,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.08680013020833334,
      "learning_rate": 0.0001,
      "loss": 4.5996,
      "loss/crossentropy": 2.0473387241363525,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23628919571638107,
      "step": 6026
    },
    {
      "epoch": 0.12056,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.09107640584309896,
      "learning_rate": 0.0001,
      "loss": 4.109,
      "loss/crossentropy": 2.013141930103302,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21728236973285675,
      "step": 6028
    },
    {
      "epoch": 0.1206,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.09145278930664062,
      "learning_rate": 0.0001,
      "loss": 4.3869,
      "loss/crossentropy": 2.1269132494926453,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25139085948467255,
      "step": 6030
    },
    {
      "epoch": 0.12064,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.09058405558268229,
      "learning_rate": 0.0001,
      "loss": 4.5568,
      "loss/crossentropy": 2.5267512798309326,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27998340129852295,
      "step": 6032
    },
    {
      "epoch": 0.12068,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.10746027628580729,
      "learning_rate": 0.0001,
      "loss": 4.2615,
      "loss/crossentropy": 1.8502249717712402,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22863731533288956,
      "step": 6034
    },
    {
      "epoch": 0.12072,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.10850601196289063,
      "learning_rate": 0.0001,
      "loss": 4.2197,
      "loss/crossentropy": 1.7754456400871277,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23809141665697098,
      "step": 6036
    },
    {
      "epoch": 0.12076,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.033760325113932295,
      "learning_rate": 0.0001,
      "loss": 4.4714,
      "loss/crossentropy": 1.9596920609474182,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2111937776207924,
      "step": 6038
    },
    {
      "epoch": 0.1208,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.02995580037434896,
      "learning_rate": 0.0001,
      "loss": 4.3079,
      "loss/crossentropy": 1.888563334941864,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21698038280010223,
      "step": 6040
    },
    {
      "epoch": 0.12084,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.031404368082682294,
      "learning_rate": 0.0001,
      "loss": 4.2412,
      "loss/crossentropy": 2.1646993160247803,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22735021263360977,
      "step": 6042
    },
    {
      "epoch": 0.12088,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0258941650390625,
      "learning_rate": 0.0001,
      "loss": 4.525,
      "loss/crossentropy": 2.0790343284606934,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2193024456501007,
      "step": 6044
    },
    {
      "epoch": 0.12092,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.025211588541666666,
      "learning_rate": 0.0001,
      "loss": 4.3538,
      "loss/crossentropy": 2.2733768224716187,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.249516561627388,
      "step": 6046
    },
    {
      "epoch": 0.12096,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.02431640625,
      "learning_rate": 0.0001,
      "loss": 4.7286,
      "loss/crossentropy": 2.5003533363342285,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22928690910339355,
      "step": 6048
    },
    {
      "epoch": 0.121,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.03629150390625,
      "learning_rate": 0.0001,
      "loss": 4.6269,
      "loss/crossentropy": 1.9606900215148926,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2152085080742836,
      "step": 6050
    },
    {
      "epoch": 0.12104,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.0384765625,
      "learning_rate": 0.0001,
      "loss": 4.5724,
      "loss/crossentropy": 2.266395926475525,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26055608689785004,
      "step": 6052
    },
    {
      "epoch": 0.12108,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.037821451822916664,
      "learning_rate": 0.0001,
      "loss": 4.5841,
      "loss/crossentropy": 2.1753041744232178,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2327822595834732,
      "step": 6054
    },
    {
      "epoch": 0.12112,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.041304524739583334,
      "learning_rate": 0.0001,
      "loss": 4.2198,
      "loss/crossentropy": 1.775630235671997,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2073052078485489,
      "step": 6056
    },
    {
      "epoch": 0.12116,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.03975321451822917,
      "learning_rate": 0.0001,
      "loss": 4.486,
      "loss/crossentropy": 2.415855050086975,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.270721398293972,
      "step": 6058
    },
    {
      "epoch": 0.1212,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.039453125,
      "learning_rate": 0.0001,
      "loss": 4.6647,
      "loss/crossentropy": 2.2122162580490112,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24723708629608154,
      "step": 6060
    },
    {
      "epoch": 0.12124,
      "grad_norm": 2.375,
      "grad_norm_var": 0.041792805989583334,
      "learning_rate": 0.0001,
      "loss": 4.3803,
      "loss/crossentropy": 1.9540830850601196,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21385761350393295,
      "step": 6062
    },
    {
      "epoch": 0.12128,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.0413726806640625,
      "learning_rate": 0.0001,
      "loss": 4.979,
      "loss/crossentropy": 1.923313319683075,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22986505925655365,
      "step": 6064
    },
    {
      "epoch": 0.12132,
      "grad_norm": 3.03125,
      "grad_norm_var": 0.05182291666666667,
      "learning_rate": 0.0001,
      "loss": 4.6895,
      "loss/crossentropy": 2.4042444229125977,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30044034123420715,
      "step": 6066
    },
    {
      "epoch": 0.12136,
      "grad_norm": 2.25,
      "grad_norm_var": 0.049225870768229166,
      "learning_rate": 0.0001,
      "loss": 4.5857,
      "loss/crossentropy": 2.2610549926757812,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2628016769886017,
      "step": 6068
    },
    {
      "epoch": 0.1214,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0513671875,
      "learning_rate": 0.0001,
      "loss": 4.5024,
      "loss/crossentropy": 1.9100797176361084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22663169354200363,
      "step": 6070
    },
    {
      "epoch": 0.12144,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.04830729166666667,
      "learning_rate": 0.0001,
      "loss": 4.5916,
      "loss/crossentropy": 2.3971651792526245,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26831263303756714,
      "step": 6072
    },
    {
      "epoch": 0.12148,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.0492095947265625,
      "learning_rate": 0.0001,
      "loss": 4.4553,
      "loss/crossentropy": 2.106821596622467,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23786011338233948,
      "step": 6074
    },
    {
      "epoch": 0.12152,
      "grad_norm": 2.125,
      "grad_norm_var": 0.052469889322916664,
      "learning_rate": 0.0001,
      "loss": 4.4225,
      "loss/crossentropy": 2.1920535564422607,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23812127113342285,
      "step": 6076
    },
    {
      "epoch": 0.12156,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.049088541666666666,
      "learning_rate": 0.0001,
      "loss": 4.2314,
      "loss/crossentropy": 2.3014419078826904,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2380141019821167,
      "step": 6078
    },
    {
      "epoch": 0.1216,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.04903055826822917,
      "learning_rate": 0.0001,
      "loss": 4.4558,
      "loss/crossentropy": 2.1721781492233276,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24158813059329987,
      "step": 6080
    },
    {
      "epoch": 0.12164,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.013232421875,
      "learning_rate": 0.0001,
      "loss": 4.5192,
      "loss/crossentropy": 2.1230934858322144,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23764102160930634,
      "step": 6082
    },
    {
      "epoch": 0.12168,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.0166412353515625,
      "learning_rate": 0.0001,
      "loss": 4.2147,
      "loss/crossentropy": 1.9570311307907104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24687693268060684,
      "step": 6084
    },
    {
      "epoch": 0.12172,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.014997355143229167,
      "learning_rate": 0.0001,
      "loss": 4.2063,
      "loss/crossentropy": 2.272383213043213,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2567671462893486,
      "step": 6086
    },
    {
      "epoch": 0.12176,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.009598795572916667,
      "learning_rate": 0.0001,
      "loss": 4.645,
      "loss/crossentropy": 2.216492176055908,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24377377331256866,
      "step": 6088
    },
    {
      "epoch": 0.1218,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.007201131184895833,
      "learning_rate": 0.0001,
      "loss": 4.5956,
      "loss/crossentropy": 2.25177001953125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23176074773073196,
      "step": 6090
    },
    {
      "epoch": 0.12184,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.009098307291666666,
      "learning_rate": 0.0001,
      "loss": 4.1966,
      "loss/crossentropy": 2.2452452182769775,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24528006464242935,
      "step": 6092
    },
    {
      "epoch": 0.12188,
      "grad_norm": 2.25,
      "grad_norm_var": 0.0091949462890625,
      "learning_rate": 0.0001,
      "loss": 4.3548,
      "loss/crossentropy": 2.078445553779602,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2276337668299675,
      "step": 6094
    },
    {
      "epoch": 0.12192,
      "grad_norm": 2.125,
      "grad_norm_var": 0.00982666015625,
      "learning_rate": 0.0001,
      "loss": 4.4435,
      "loss/crossentropy": 2.2938032150268555,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2784377336502075,
      "step": 6096
    },
    {
      "epoch": 0.12196,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0136627197265625,
      "learning_rate": 0.0001,
      "loss": 4.1588,
      "loss/crossentropy": 2.028180480003357,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23716723918914795,
      "step": 6098
    },
    {
      "epoch": 0.122,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.020580037434895834,
      "learning_rate": 0.0001,
      "loss": 4.6345,
      "loss/crossentropy": 2.4959323406219482,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2510451450943947,
      "step": 6100
    },
    {
      "epoch": 0.12204,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.023856608072916667,
      "learning_rate": 0.0001,
      "loss": 4.6639,
      "loss/crossentropy": 2.186043620109558,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23511512577533722,
      "step": 6102
    },
    {
      "epoch": 0.12208,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.024442545572916665,
      "learning_rate": 0.0001,
      "loss": 4.6727,
      "loss/crossentropy": 2.6631078720092773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25689610838890076,
      "step": 6104
    },
    {
      "epoch": 0.12212,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.025609334309895832,
      "learning_rate": 0.0001,
      "loss": 4.7943,
      "loss/crossentropy": 2.310486674308777,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26937828958034515,
      "step": 6106
    },
    {
      "epoch": 0.12216,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.0247711181640625,
      "learning_rate": 0.0001,
      "loss": 4.9892,
      "loss/crossentropy": 2.2036240100860596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24968907237052917,
      "step": 6108
    },
    {
      "epoch": 0.1222,
      "grad_norm": 2.984375,
      "grad_norm_var": 0.058251953125,
      "learning_rate": 0.0001,
      "loss": 4.3527,
      "loss/crossentropy": 1.9434874057769775,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23247701674699783,
      "step": 6110
    },
    {
      "epoch": 0.12224,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.05579020182291667,
      "learning_rate": 0.0001,
      "loss": 4.4149,
      "loss/crossentropy": 2.0525330305099487,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24810528755187988,
      "step": 6112
    },
    {
      "epoch": 0.12228,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.04482320149739583,
      "learning_rate": 0.0001,
      "loss": 4.5205,
      "loss/crossentropy": 2.0085532665252686,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2514277398586273,
      "step": 6114
    },
    {
      "epoch": 0.12232,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.04120686848958333,
      "learning_rate": 0.0001,
      "loss": 4.5745,
      "loss/crossentropy": 1.7393967509269714,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2239851951599121,
      "step": 6116
    },
    {
      "epoch": 0.12236,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.04327799479166667,
      "learning_rate": 0.0001,
      "loss": 4.7332,
      "loss/crossentropy": 1.8714343905448914,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22823140025138855,
      "step": 6118
    },
    {
      "epoch": 0.1224,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.04192708333333333,
      "learning_rate": 0.0001,
      "loss": 4.6177,
      "loss/crossentropy": 1.9353562593460083,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22261886298656464,
      "step": 6120
    },
    {
      "epoch": 0.12244,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.0469635009765625,
      "learning_rate": 0.0001,
      "loss": 4.3154,
      "loss/crossentropy": 2.0409420132637024,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2319282591342926,
      "step": 6122
    },
    {
      "epoch": 0.12248,
      "grad_norm": 2.125,
      "grad_norm_var": 0.04882405598958333,
      "learning_rate": 0.0001,
      "loss": 4.4764,
      "loss/crossentropy": 2.2461307048797607,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25047267973423004,
      "step": 6124
    },
    {
      "epoch": 0.12252,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.014012654622395834,
      "learning_rate": 0.0001,
      "loss": 4.4863,
      "loss/crossentropy": 2.1060246229171753,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24702349305152893,
      "step": 6126
    },
    {
      "epoch": 0.12256,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.01718724568684896,
      "learning_rate": 0.0001,
      "loss": 3.9093,
      "loss/crossentropy": 1.7481068968772888,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19611438363790512,
      "step": 6128
    },
    {
      "epoch": 0.1226,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.017329661051432292,
      "learning_rate": 0.0001,
      "loss": 4.5506,
      "loss/crossentropy": 1.8341861963272095,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20660528540611267,
      "step": 6130
    },
    {
      "epoch": 0.12264,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.018507639567057293,
      "learning_rate": 0.0001,
      "loss": 4.3388,
      "loss/crossentropy": 1.7332024574279785,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21615543216466904,
      "step": 6132
    },
    {
      "epoch": 0.12268,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.013203684488932292,
      "learning_rate": 0.0001,
      "loss": 4.3027,
      "loss/crossentropy": 1.8959643244743347,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.219951793551445,
      "step": 6134
    },
    {
      "epoch": 0.12272,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.012601470947265625,
      "learning_rate": 0.0001,
      "loss": 4.6945,
      "loss/crossentropy": 2.234723210334778,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2342153787612915,
      "step": 6136
    },
    {
      "epoch": 0.12276,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.011637115478515625,
      "learning_rate": 0.0001,
      "loss": 4.7394,
      "loss/crossentropy": 2.2297592759132385,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24097590148448944,
      "step": 6138
    },
    {
      "epoch": 0.1228,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.013586171468098958,
      "learning_rate": 0.0001,
      "loss": 4.592,
      "loss/crossentropy": 2.013442814350128,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2532622739672661,
      "step": 6140
    },
    {
      "epoch": 0.12284,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.014833323160807292,
      "learning_rate": 0.0001,
      "loss": 4.417,
      "loss/crossentropy": 2.3066656589508057,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24480555206537247,
      "step": 6142
    },
    {
      "epoch": 0.12288,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.014241536458333334,
      "learning_rate": 0.0001,
      "loss": 4.4156,
      "loss/crossentropy": 1.9281310439109802,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22363708168268204,
      "step": 6144
    },
    {
      "epoch": 0.12292,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.016792805989583333,
      "learning_rate": 0.0001,
      "loss": 4.4314,
      "loss/crossentropy": 1.965875267982483,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22153983265161514,
      "step": 6146
    },
    {
      "epoch": 0.12296,
      "grad_norm": 2.8125,
      "grad_norm_var": 0.03536783854166667,
      "learning_rate": 0.0001,
      "loss": 4.6139,
      "loss/crossentropy": 1.811126947402954,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21779045462608337,
      "step": 6148
    },
    {
      "epoch": 0.123,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.0311676025390625,
      "learning_rate": 0.0001,
      "loss": 4.4855,
      "loss/crossentropy": 2.5114762783050537,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2542252689599991,
      "step": 6150
    },
    {
      "epoch": 0.12304,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.028271484375,
      "learning_rate": 0.0001,
      "loss": 4.1269,
      "loss/crossentropy": 1.8784565925598145,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24637068808078766,
      "step": 6152
    },
    {
      "epoch": 0.12308,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.028685506184895834,
      "learning_rate": 0.0001,
      "loss": 4.3377,
      "loss/crossentropy": 1.6900760531425476,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2170872688293457,
      "step": 6154
    },
    {
      "epoch": 0.12312,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.02867431640625,
      "learning_rate": 0.0001,
      "loss": 4.5705,
      "loss/crossentropy": 2.413783550262451,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2624947279691696,
      "step": 6156
    },
    {
      "epoch": 0.12316,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.030126953125,
      "learning_rate": 0.0001,
      "loss": 4.3463,
      "loss/crossentropy": 2.0598058104515076,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2071695551276207,
      "step": 6158
    },
    {
      "epoch": 0.1232,
      "grad_norm": 2.375,
      "grad_norm_var": 0.029781087239583334,
      "learning_rate": 0.0001,
      "loss": 4.6702,
      "loss/crossentropy": 2.0407246947288513,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24353720247745514,
      "step": 6160
    },
    {
      "epoch": 0.12324,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.028450520833333333,
      "learning_rate": 0.0001,
      "loss": 4.3116,
      "loss/crossentropy": 1.9608840346336365,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20965111255645752,
      "step": 6162
    },
    {
      "epoch": 0.12328,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.01640625,
      "learning_rate": 0.0001,
      "loss": 4.2146,
      "loss/crossentropy": 2.0231454372406006,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.219897098839283,
      "step": 6164
    },
    {
      "epoch": 0.12332,
      "grad_norm": 2.25,
      "grad_norm_var": 0.015999348958333333,
      "learning_rate": 0.0001,
      "loss": 4.614,
      "loss/crossentropy": 2.217663288116455,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2466331645846367,
      "step": 6166
    },
    {
      "epoch": 0.12336,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.01597900390625,
      "learning_rate": 0.0001,
      "loss": 4.4135,
      "loss/crossentropy": 1.6825732588768005,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1999632865190506,
      "step": 6168
    },
    {
      "epoch": 0.1234,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.020897420247395833,
      "learning_rate": 0.0001,
      "loss": 4.4552,
      "loss/crossentropy": 2.288950800895691,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26286834478378296,
      "step": 6170
    },
    {
      "epoch": 0.12344,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.023225911458333335,
      "learning_rate": 0.0001,
      "loss": 4.3376,
      "loss/crossentropy": 1.8202561140060425,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22252517193555832,
      "step": 6172
    },
    {
      "epoch": 0.12348,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.019710286458333334,
      "learning_rate": 0.0001,
      "loss": 4.5023,
      "loss/crossentropy": 2.1007159948349,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24983422458171844,
      "step": 6174
    },
    {
      "epoch": 0.12352,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.0199371337890625,
      "learning_rate": 0.0001,
      "loss": 4.4513,
      "loss/crossentropy": 2.0953084230422974,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25545743107795715,
      "step": 6176
    },
    {
      "epoch": 0.12356,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.019001261393229166,
      "learning_rate": 0.0001,
      "loss": 4.4315,
      "loss/crossentropy": 2.107246518135071,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22741339355707169,
      "step": 6178
    },
    {
      "epoch": 0.1236,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.013118489583333334,
      "learning_rate": 0.0001,
      "loss": 4.4982,
      "loss/crossentropy": 2.1219626665115356,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22499094158411026,
      "step": 6180
    },
    {
      "epoch": 0.12364,
      "grad_norm": 2.25,
      "grad_norm_var": 0.013093058268229167,
      "learning_rate": 0.0001,
      "loss": 4.4263,
      "loss/crossentropy": 1.8892266154289246,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2255062311887741,
      "step": 6182
    },
    {
      "epoch": 0.12368,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.015315755208333334,
      "learning_rate": 0.0001,
      "loss": 4.5884,
      "loss/crossentropy": 2.0765860080718994,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24712087213993073,
      "step": 6184
    },
    {
      "epoch": 0.12372,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.010074869791666666,
      "learning_rate": 0.0001,
      "loss": 4.5431,
      "loss/crossentropy": 2.301337718963623,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24419523775577545,
      "step": 6186
    },
    {
      "epoch": 0.12376,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.008373006184895834,
      "learning_rate": 0.0001,
      "loss": 4.5043,
      "loss/crossentropy": 2.0489712953567505,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25768278539180756,
      "step": 6188
    },
    {
      "epoch": 0.1238,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0128082275390625,
      "learning_rate": 0.0001,
      "loss": 4.2654,
      "loss/crossentropy": 1.9328826069831848,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22916094958782196,
      "step": 6190
    },
    {
      "epoch": 0.12384,
      "grad_norm": 2.25,
      "grad_norm_var": 0.01171875,
      "learning_rate": 0.0001,
      "loss": 4.1574,
      "loss/crossentropy": 1.6330446004867554,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20600398629903793,
      "step": 6192
    },
    {
      "epoch": 0.12388,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.0111724853515625,
      "learning_rate": 0.0001,
      "loss": 4.4769,
      "loss/crossentropy": 1.988203227519989,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22711507230997086,
      "step": 6194
    },
    {
      "epoch": 0.12392,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.01590143839518229,
      "learning_rate": 0.0001,
      "loss": 4.125,
      "loss/crossentropy": 2.036049246788025,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2110549360513687,
      "step": 6196
    },
    {
      "epoch": 0.12396,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.01624120076497396,
      "learning_rate": 0.0001,
      "loss": 4.5546,
      "loss/crossentropy": 2.217681884765625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23705793917179108,
      "step": 6198
    },
    {
      "epoch": 0.124,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.015276845296223958,
      "learning_rate": 0.0001,
      "loss": 4.4803,
      "loss/crossentropy": 2.3878824710845947,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24000737071037292,
      "step": 6200
    },
    {
      "epoch": 0.12404,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.015852610270182293,
      "learning_rate": 0.0001,
      "loss": 4.4754,
      "loss/crossentropy": 2.2790249586105347,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26150786131620407,
      "step": 6202
    },
    {
      "epoch": 0.12408,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.015036773681640626,
      "learning_rate": 0.0001,
      "loss": 4.4703,
      "loss/crossentropy": 2.251123785972595,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25231797993183136,
      "step": 6204
    },
    {
      "epoch": 0.12412,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.010802968343098959,
      "learning_rate": 0.0001,
      "loss": 4.5294,
      "loss/crossentropy": 1.8977670073509216,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21783004701137543,
      "step": 6206
    },
    {
      "epoch": 0.12416,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.012894439697265624,
      "learning_rate": 0.0001,
      "loss": 4.6458,
      "loss/crossentropy": 2.385319232940674,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29223111271858215,
      "step": 6208
    },
    {
      "epoch": 0.1242,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.015964508056640625,
      "learning_rate": 0.0001,
      "loss": 4.6555,
      "loss/crossentropy": 1.9274529218673706,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22827968001365662,
      "step": 6210
    },
    {
      "epoch": 0.12424,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.01226806640625,
      "learning_rate": 0.0001,
      "loss": 4.8232,
      "loss/crossentropy": 2.1861478090286255,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2424854040145874,
      "step": 6212
    },
    {
      "epoch": 0.12428,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.01207275390625,
      "learning_rate": 0.0001,
      "loss": 4.3002,
      "loss/crossentropy": 2.2234357595443726,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2471691370010376,
      "step": 6214
    },
    {
      "epoch": 0.12432,
      "grad_norm": 2.625,
      "grad_norm_var": 0.022541300455729166,
      "learning_rate": 0.0001,
      "loss": 4.5216,
      "loss/crossentropy": 2.365513563156128,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2640291824936867,
      "step": 6216
    },
    {
      "epoch": 0.12436,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.021955362955729165,
      "learning_rate": 0.0001,
      "loss": 4.282,
      "loss/crossentropy": 1.964316964149475,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22341662645339966,
      "step": 6218
    },
    {
      "epoch": 0.1244,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.021805826822916666,
      "learning_rate": 0.0001,
      "loss": 4.7078,
      "loss/crossentropy": 2.3704408407211304,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24599966406822205,
      "step": 6220
    },
    {
      "epoch": 0.12444,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0193756103515625,
      "learning_rate": 0.0001,
      "loss": 4.7558,
      "loss/crossentropy": 2.1461408138275146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25205330550670624,
      "step": 6222
    },
    {
      "epoch": 0.12448,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.018745930989583333,
      "learning_rate": 0.0001,
      "loss": 4.7032,
      "loss/crossentropy": 2.3997104167938232,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23371660709381104,
      "step": 6224
    },
    {
      "epoch": 0.12452,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0212890625,
      "learning_rate": 0.0001,
      "loss": 4.1842,
      "loss/crossentropy": 1.7976875305175781,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21662414073944092,
      "step": 6226
    },
    {
      "epoch": 0.12456,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.02008056640625,
      "learning_rate": 0.0001,
      "loss": 4.2214,
      "loss/crossentropy": 1.9929583668708801,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21378497034311295,
      "step": 6228
    },
    {
      "epoch": 0.1246,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.02017822265625,
      "learning_rate": 0.0001,
      "loss": 4.2982,
      "loss/crossentropy": 1.8853623867034912,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21267645806074142,
      "step": 6230
    },
    {
      "epoch": 0.12464,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.008454386393229167,
      "learning_rate": 0.0001,
      "loss": 4.2638,
      "loss/crossentropy": 2.2534812688827515,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23003943264484406,
      "step": 6232
    },
    {
      "epoch": 0.12468,
      "grad_norm": 2.0,
      "grad_norm_var": 0.010863240559895833,
      "learning_rate": 0.0001,
      "loss": 4.2504,
      "loss/crossentropy": 2.026564121246338,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2172931283712387,
      "step": 6234
    },
    {
      "epoch": 0.12472,
      "grad_norm": 2.25,
      "grad_norm_var": 0.01021728515625,
      "learning_rate": 0.0001,
      "loss": 4.3555,
      "loss/crossentropy": 1.9323118925094604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22170638293027878,
      "step": 6236
    },
    {
      "epoch": 0.12476,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.007938639322916666,
      "learning_rate": 0.0001,
      "loss": 4.479,
      "loss/crossentropy": 2.056011915206909,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2656880244612694,
      "step": 6238
    },
    {
      "epoch": 0.1248,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.00830078125,
      "learning_rate": 0.0001,
      "loss": 4.6288,
      "loss/crossentropy": 2.095108926296234,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23341026157140732,
      "step": 6240
    },
    {
      "epoch": 0.12484,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.011555989583333334,
      "learning_rate": 0.0001,
      "loss": 4.737,
      "loss/crossentropy": 2.0198334455490112,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24095112830400467,
      "step": 6242
    },
    {
      "epoch": 0.12488,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.012105305989583334,
      "learning_rate": 0.0001,
      "loss": 4.3745,
      "loss/crossentropy": 2.0397544503211975,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22567399591207504,
      "step": 6244
    },
    {
      "epoch": 0.12492,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0120513916015625,
      "learning_rate": 0.0001,
      "loss": 4.3633,
      "loss/crossentropy": 1.9094319343566895,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22690637409687042,
      "step": 6246
    },
    {
      "epoch": 0.12496,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.011295572916666666,
      "learning_rate": 0.0001,
      "loss": 4.6474,
      "loss/crossentropy": 2.27765429019928,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25075703859329224,
      "step": 6248
    },
    {
      "epoch": 0.125,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.017023722330729168,
      "learning_rate": 0.0001,
      "loss": 4.5769,
      "loss/crossentropy": 2.0027456283569336,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24267998337745667,
      "step": 6250
    },
    {
      "epoch": 0.12504,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.017723592122395833,
      "learning_rate": 0.0001,
      "loss": 4.5398,
      "loss/crossentropy": 2.276857614517212,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24048637598752975,
      "step": 6252
    },
    {
      "epoch": 0.12508,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.018065388997395834,
      "learning_rate": 0.0001,
      "loss": 4.7603,
      "loss/crossentropy": 2.1234214305877686,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23481453210115433,
      "step": 6254
    },
    {
      "epoch": 0.12512,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.019624837239583335,
      "learning_rate": 0.0001,
      "loss": 4.7537,
      "loss/crossentropy": 2.201158881187439,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23473594337701797,
      "step": 6256
    },
    {
      "epoch": 0.12516,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.017220052083333333,
      "learning_rate": 0.0001,
      "loss": 4.7202,
      "loss/crossentropy": 2.3437804579734802,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24881581962108612,
      "step": 6258
    },
    {
      "epoch": 0.1252,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.015461222330729166,
      "learning_rate": 0.0001,
      "loss": 4.5938,
      "loss/crossentropy": 2.0984586477279663,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24332843720912933,
      "step": 6260
    },
    {
      "epoch": 0.12524,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.016942342122395832,
      "learning_rate": 0.0001,
      "loss": 4.2317,
      "loss/crossentropy": 1.7946080565452576,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21510492265224457,
      "step": 6262
    },
    {
      "epoch": 0.12528,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.016161092122395835,
      "learning_rate": 0.0001,
      "loss": 4.3889,
      "loss/crossentropy": 2.022711932659149,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22890903800725937,
      "step": 6264
    },
    {
      "epoch": 0.12532,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.009577433268229166,
      "learning_rate": 0.0001,
      "loss": 4.288,
      "loss/crossentropy": 1.9752087593078613,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21972095221281052,
      "step": 6266
    },
    {
      "epoch": 0.12536,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.008625284830729166,
      "learning_rate": 0.0001,
      "loss": 4.4952,
      "loss/crossentropy": 1.7267251014709473,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21305006742477417,
      "step": 6268
    },
    {
      "epoch": 0.1254,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.008869425455729166,
      "learning_rate": 0.0001,
      "loss": 4.2164,
      "loss/crossentropy": 2.249786615371704,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24875187873840332,
      "step": 6270
    },
    {
      "epoch": 0.12544,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.012190755208333333,
      "learning_rate": 0.0001,
      "loss": 4.693,
      "loss/crossentropy": 2.402994990348816,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25536587834358215,
      "step": 6272
    },
    {
      "epoch": 0.12548,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.016243489583333333,
      "learning_rate": 0.0001,
      "loss": 4.606,
      "loss/crossentropy": 2.240913987159729,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3013365715742111,
      "step": 6274
    },
    {
      "epoch": 0.12552,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.019071451822916665,
      "learning_rate": 0.0001,
      "loss": 4.4691,
      "loss/crossentropy": 2.0767332911491394,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22113215178251266,
      "step": 6276
    },
    {
      "epoch": 0.12556,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.02041015625,
      "learning_rate": 0.0001,
      "loss": 4.608,
      "loss/crossentropy": 1.8625048995018005,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20321352779865265,
      "step": 6278
    },
    {
      "epoch": 0.1256,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.024072265625,
      "learning_rate": 0.0001,
      "loss": 4.0386,
      "loss/crossentropy": 1.9143638610839844,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22085773944854736,
      "step": 6280
    },
    {
      "epoch": 0.12564,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.023192342122395834,
      "learning_rate": 0.0001,
      "loss": 4.372,
      "loss/crossentropy": 2.3756210803985596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26027245819568634,
      "step": 6282
    },
    {
      "epoch": 0.12568,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.02427978515625,
      "learning_rate": 0.0001,
      "loss": 4.5526,
      "loss/crossentropy": 1.9310896396636963,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2086925357580185,
      "step": 6284
    },
    {
      "epoch": 0.12572,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.021675618489583333,
      "learning_rate": 0.0001,
      "loss": 4.4751,
      "loss/crossentropy": 2.1757423877716064,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.241075336933136,
      "step": 6286
    },
    {
      "epoch": 0.12576,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.021284993489583334,
      "learning_rate": 0.0001,
      "loss": 4.0946,
      "loss/crossentropy": 2.3057546615600586,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24882248044013977,
      "step": 6288
    },
    {
      "epoch": 0.1258,
      "grad_norm": 3.140625,
      "grad_norm_var": 0.06883036295572917,
      "learning_rate": 0.0001,
      "loss": 4.5211,
      "loss/crossentropy": 2.1551633477211,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2796258181333542,
      "step": 6290
    },
    {
      "epoch": 0.12584,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.06608072916666667,
      "learning_rate": 0.0001,
      "loss": 4.4915,
      "loss/crossentropy": 2.0627459287643433,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24781616777181625,
      "step": 6292
    },
    {
      "epoch": 0.12588,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.06463216145833334,
      "learning_rate": 0.0001,
      "loss": 4.5896,
      "loss/crossentropy": 1.795321524143219,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2113223671913147,
      "step": 6294
    },
    {
      "epoch": 0.12592,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.06004130045572917,
      "learning_rate": 0.0001,
      "loss": 4.4988,
      "loss/crossentropy": 2.323319673538208,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24748852849006653,
      "step": 6296
    },
    {
      "epoch": 0.12596,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.05998942057291667,
      "learning_rate": 0.0001,
      "loss": 4.4801,
      "loss/crossentropy": 2.053200662136078,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2278411090373993,
      "step": 6298
    },
    {
      "epoch": 0.126,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.060155232747395836,
      "learning_rate": 0.0001,
      "loss": 4.4605,
      "loss/crossentropy": 2.102539896965027,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26071713864803314,
      "step": 6300
    },
    {
      "epoch": 0.12604,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.059554036458333334,
      "learning_rate": 0.0001,
      "loss": 4.4613,
      "loss/crossentropy": 1.9316620826721191,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23007915914058685,
      "step": 6302
    },
    {
      "epoch": 0.12608,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.06288655598958333,
      "learning_rate": 0.0001,
      "loss": 4.4194,
      "loss/crossentropy": 1.9966526627540588,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2424854189157486,
      "step": 6304
    },
    {
      "epoch": 0.12612,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.022541300455729166,
      "learning_rate": 0.0001,
      "loss": 4.2382,
      "loss/crossentropy": 1.895868957042694,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2255951091647148,
      "step": 6306
    },
    {
      "epoch": 0.12616,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.023688761393229167,
      "learning_rate": 0.0001,
      "loss": 4.7758,
      "loss/crossentropy": 2.3897345066070557,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24461720883846283,
      "step": 6308
    },
    {
      "epoch": 0.1262,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.022858683268229166,
      "learning_rate": 0.0001,
      "loss": 4.8275,
      "loss/crossentropy": 2.059292197227478,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29264035820961,
      "step": 6310
    },
    {
      "epoch": 0.12624,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.0221343994140625,
      "learning_rate": 0.0001,
      "loss": 4.5098,
      "loss/crossentropy": 1.8142406344413757,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22127485275268555,
      "step": 6312
    },
    {
      "epoch": 0.12628,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0245025634765625,
      "learning_rate": 0.0001,
      "loss": 4.2349,
      "loss/crossentropy": 2.0186127424240112,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.217330664396286,
      "step": 6314
    },
    {
      "epoch": 0.12632,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.024820963541666668,
      "learning_rate": 0.0001,
      "loss": 4.0102,
      "loss/crossentropy": 1.9439889192581177,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2108919695019722,
      "step": 6316
    },
    {
      "epoch": 0.12636,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.024690755208333335,
      "learning_rate": 0.0001,
      "loss": 4.8672,
      "loss/crossentropy": 2.617791175842285,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2756097912788391,
      "step": 6318
    },
    {
      "epoch": 0.1264,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.017609659830729166,
      "learning_rate": 0.0001,
      "loss": 4.3918,
      "loss/crossentropy": 1.9074286818504333,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22491587698459625,
      "step": 6320
    },
    {
      "epoch": 0.12644,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.016331990559895832,
      "learning_rate": 0.0001,
      "loss": 4.4304,
      "loss/crossentropy": 2.1414809226989746,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22703612595796585,
      "step": 6322
    },
    {
      "epoch": 0.12648,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.0154205322265625,
      "learning_rate": 0.0001,
      "loss": 4.4982,
      "loss/crossentropy": 2.291784942150116,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24490328133106232,
      "step": 6324
    },
    {
      "epoch": 0.12652,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0143218994140625,
      "learning_rate": 0.0001,
      "loss": 4.228,
      "loss/crossentropy": 1.938249409198761,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22611317038536072,
      "step": 6326
    },
    {
      "epoch": 0.12656,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0187408447265625,
      "learning_rate": 0.0001,
      "loss": 4.4932,
      "loss/crossentropy": 1.7510024905204773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30439992994070053,
      "step": 6328
    },
    {
      "epoch": 0.1266,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.022907511393229166,
      "learning_rate": 0.0001,
      "loss": 4.4351,
      "loss/crossentropy": 2.3168221712112427,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25660980492830276,
      "step": 6330
    },
    {
      "epoch": 0.12664,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.024409993489583334,
      "learning_rate": 0.0001,
      "loss": 4.5002,
      "loss/crossentropy": 1.9957427978515625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21420849114656448,
      "step": 6332
    },
    {
      "epoch": 0.12668,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.025104777018229166,
      "learning_rate": 0.0001,
      "loss": 4.3583,
      "loss/crossentropy": 2.4371464252471924,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2576342821121216,
      "step": 6334
    },
    {
      "epoch": 0.12672,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.021458943684895832,
      "learning_rate": 0.0001,
      "loss": 4.2497,
      "loss/crossentropy": 1.910677433013916,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21360614150762558,
      "step": 6336
    },
    {
      "epoch": 0.12676,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.020750935872395834,
      "learning_rate": 0.0001,
      "loss": 4.2679,
      "loss/crossentropy": 2.057362914085388,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23079442232847214,
      "step": 6338
    },
    {
      "epoch": 0.1268,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.0191070556640625,
      "learning_rate": 0.0001,
      "loss": 4.1818,
      "loss/crossentropy": 2.0249438881874084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24038050323724747,
      "step": 6340
    },
    {
      "epoch": 0.12684,
      "grad_norm": 2.25,
      "grad_norm_var": 0.019074503580729166,
      "learning_rate": 0.0001,
      "loss": 4.5374,
      "loss/crossentropy": 2.0371533632278442,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2355760782957077,
      "step": 6342
    },
    {
      "epoch": 0.12688,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.0130523681640625,
      "learning_rate": 0.0001,
      "loss": 4.4849,
      "loss/crossentropy": 2.3101218938827515,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24161705374717712,
      "step": 6344
    },
    {
      "epoch": 0.12692,
      "grad_norm": 2.25,
      "grad_norm_var": 0.010261027018229167,
      "learning_rate": 0.0001,
      "loss": 4.6399,
      "loss/crossentropy": 2.1990097761154175,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2537754699587822,
      "step": 6346
    },
    {
      "epoch": 0.12696,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0136383056640625,
      "learning_rate": 0.0001,
      "loss": 4.2697,
      "loss/crossentropy": 2.1783597469329834,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2106732353568077,
      "step": 6348
    },
    {
      "epoch": 0.127,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.012987263997395833,
      "learning_rate": 0.0001,
      "loss": 4.5127,
      "loss/crossentropy": 2.2316598892211914,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23841773718595505,
      "step": 6350
    },
    {
      "epoch": 0.12704,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.00943603515625,
      "learning_rate": 0.0001,
      "loss": 4.5651,
      "loss/crossentropy": 2.0329924821853638,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22755083441734314,
      "step": 6352
    },
    {
      "epoch": 0.12708,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.014525349934895833,
      "learning_rate": 0.0001,
      "loss": 4.1888,
      "loss/crossentropy": 1.9174052476882935,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20000722259283066,
      "step": 6354
    },
    {
      "epoch": 0.12712,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.0154693603515625,
      "learning_rate": 0.0001,
      "loss": 4.3321,
      "loss/crossentropy": 1.968774676322937,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2194407731294632,
      "step": 6356
    },
    {
      "epoch": 0.12716,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.0183013916015625,
      "learning_rate": 0.0001,
      "loss": 4.5768,
      "loss/crossentropy": 2.1767213344573975,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24483858048915863,
      "step": 6358
    },
    {
      "epoch": 0.1272,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.018391927083333332,
      "learning_rate": 0.0001,
      "loss": 4.5063,
      "loss/crossentropy": 1.8390987515449524,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21677500754594803,
      "step": 6360
    },
    {
      "epoch": 0.12724,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.016039021809895835,
      "learning_rate": 0.0001,
      "loss": 4.2635,
      "loss/crossentropy": 2.1923086643218994,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21362057328224182,
      "step": 6362
    },
    {
      "epoch": 0.12728,
      "grad_norm": 2.25,
      "grad_norm_var": 0.010347493489583333,
      "learning_rate": 0.0001,
      "loss": 4.3104,
      "loss/crossentropy": 2.030683994293213,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22884567826986313,
      "step": 6364
    },
    {
      "epoch": 0.12732,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.00816650390625,
      "learning_rate": 0.0001,
      "loss": 4.452,
      "loss/crossentropy": 2.330837845802307,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24233026057481766,
      "step": 6366
    },
    {
      "epoch": 0.12736,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.012040201822916667,
      "learning_rate": 0.0001,
      "loss": 4.6404,
      "loss/crossentropy": 2.001612663269043,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2332654967904091,
      "step": 6368
    },
    {
      "epoch": 0.1274,
      "grad_norm": 2.25,
      "grad_norm_var": 0.011750284830729167,
      "learning_rate": 0.0001,
      "loss": 4.7101,
      "loss/crossentropy": 1.9234941601753235,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23741928488016129,
      "step": 6370
    },
    {
      "epoch": 0.12744,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.012262980143229166,
      "learning_rate": 0.0001,
      "loss": 4.2164,
      "loss/crossentropy": 2.059934139251709,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24736596643924713,
      "step": 6372
    },
    {
      "epoch": 0.12748,
      "grad_norm": 2.125,
      "grad_norm_var": 0.010798136393229166,
      "learning_rate": 0.0001,
      "loss": 4.3916,
      "loss/crossentropy": 2.25182843208313,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24349220097064972,
      "step": 6374
    },
    {
      "epoch": 0.12752,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.010888671875,
      "learning_rate": 0.0001,
      "loss": 4.3742,
      "loss/crossentropy": 1.845405638217926,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2138313353061676,
      "step": 6376
    },
    {
      "epoch": 0.12756,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.0164215087890625,
      "learning_rate": 0.0001,
      "loss": 4.568,
      "loss/crossentropy": 1.8833998441696167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25304871797561646,
      "step": 6378
    },
    {
      "epoch": 0.1276,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.017072550455729165,
      "learning_rate": 0.0001,
      "loss": 4.2282,
      "loss/crossentropy": 2.1066314578056335,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2231372445821762,
      "step": 6380
    },
    {
      "epoch": 0.12764,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.018561808268229167,
      "learning_rate": 0.0001,
      "loss": 4.3381,
      "loss/crossentropy": 2.024670898914337,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22260528802871704,
      "step": 6382
    },
    {
      "epoch": 0.12768,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.01763916015625,
      "learning_rate": 0.0001,
      "loss": 4.8132,
      "loss/crossentropy": 2.425115466117859,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2429298236966133,
      "step": 6384
    },
    {
      "epoch": 0.12772,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.016597493489583334,
      "learning_rate": 0.0001,
      "loss": 4.2387,
      "loss/crossentropy": 2.1847925186157227,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2495381161570549,
      "step": 6386
    },
    {
      "epoch": 0.12776,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.015746053059895834,
      "learning_rate": 0.0001,
      "loss": 4.6919,
      "loss/crossentropy": 2.1463611125946045,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2303348332643509,
      "step": 6388
    },
    {
      "epoch": 0.1278,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.014839680989583333,
      "learning_rate": 0.0001,
      "loss": 4.2714,
      "loss/crossentropy": 1.9755331873893738,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2232731431722641,
      "step": 6390
    },
    {
      "epoch": 0.12784,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.015973917643229165,
      "learning_rate": 0.0001,
      "loss": 4.5255,
      "loss/crossentropy": 2.0741465091705322,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22912710905075073,
      "step": 6392
    },
    {
      "epoch": 0.12788,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.009471638997395834,
      "learning_rate": 0.0001,
      "loss": 4.3834,
      "loss/crossentropy": 2.1314439177513123,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22910036891698837,
      "step": 6394
    },
    {
      "epoch": 0.12792,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0107086181640625,
      "learning_rate": 0.0001,
      "loss": 4.1615,
      "loss/crossentropy": 1.976862907409668,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.233707495033741,
      "step": 6396
    },
    {
      "epoch": 0.12796,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.010807291666666666,
      "learning_rate": 0.0001,
      "loss": 4.3939,
      "loss/crossentropy": 2.0534666180610657,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21424879133701324,
      "step": 6398
    },
    {
      "epoch": 0.128,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.008088175455729167,
      "learning_rate": 0.0001,
      "loss": 4.6647,
      "loss/crossentropy": 2.2200992107391357,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2836414724588394,
      "step": 6400
    },
    {
      "epoch": 0.12804,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.007542928059895833,
      "learning_rate": 0.0001,
      "loss": 4.3802,
      "loss/crossentropy": 1.8642286658287048,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21438255906105042,
      "step": 6402
    },
    {
      "epoch": 0.12808,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.005464680989583333,
      "learning_rate": 0.0001,
      "loss": 4.3868,
      "loss/crossentropy": 1.9571613073349,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22038634121418,
      "step": 6404
    },
    {
      "epoch": 0.12812,
      "grad_norm": 2.125,
      "grad_norm_var": 0.008980305989583333,
      "learning_rate": 0.0001,
      "loss": 4.6888,
      "loss/crossentropy": 2.0420787930488586,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2532814294099808,
      "step": 6406
    },
    {
      "epoch": 0.12816,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.008349609375,
      "learning_rate": 0.0001,
      "loss": 4.3357,
      "loss/crossentropy": 1.7511736750602722,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2067284658551216,
      "step": 6408
    },
    {
      "epoch": 0.1282,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.0095123291015625,
      "learning_rate": 0.0001,
      "loss": 4.5568,
      "loss/crossentropy": 2.0310762524604797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2763645648956299,
      "step": 6410
    },
    {
      "epoch": 0.12824,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.008617146809895834,
      "learning_rate": 0.0001,
      "loss": 4.575,
      "loss/crossentropy": 2.112699866294861,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2409205138683319,
      "step": 6412
    },
    {
      "epoch": 0.12828,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.14166259765625,
      "learning_rate": 0.0001,
      "loss": 4.3023,
      "loss/crossentropy": 1.8243904113769531,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2079915553331375,
      "step": 6414
    },
    {
      "epoch": 0.12832,
      "grad_norm": 2.125,
      "grad_norm_var": 0.14143473307291668,
      "learning_rate": 0.0001,
      "loss": 4.5395,
      "loss/crossentropy": 1.8310211896896362,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21304991096258163,
      "step": 6416
    },
    {
      "epoch": 0.12836,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.14011128743489584,
      "learning_rate": 0.0001,
      "loss": 4.3806,
      "loss/crossentropy": 1.9653990268707275,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22207710891962051,
      "step": 6418
    },
    {
      "epoch": 0.1284,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.13982645670572916,
      "learning_rate": 0.0001,
      "loss": 4.5869,
      "loss/crossentropy": 2.0583431124687195,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23768695443868637,
      "step": 6420
    },
    {
      "epoch": 0.12844,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.1391998291015625,
      "learning_rate": 0.0001,
      "loss": 4.6512,
      "loss/crossentropy": 2.1162279844284058,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23963302373886108,
      "step": 6422
    },
    {
      "epoch": 0.12848,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.13347066243489583,
      "learning_rate": 0.0001,
      "loss": 4.6268,
      "loss/crossentropy": 2.068794012069702,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21967273205518723,
      "step": 6424
    },
    {
      "epoch": 0.12852,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.13585611979166667,
      "learning_rate": 0.0001,
      "loss": 4.4004,
      "loss/crossentropy": 2.165997266769409,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22991405427455902,
      "step": 6426
    },
    {
      "epoch": 0.12856,
      "grad_norm": 2.25,
      "grad_norm_var": 0.14031473795572916,
      "learning_rate": 0.0001,
      "loss": 4.1816,
      "loss/crossentropy": 1.9116491675376892,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22784583270549774,
      "step": 6428
    },
    {
      "epoch": 0.1286,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.0087890625,
      "learning_rate": 0.0001,
      "loss": 4.5268,
      "loss/crossentropy": 2.296347498893738,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2618508040904999,
      "step": 6430
    },
    {
      "epoch": 0.12864,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0100250244140625,
      "learning_rate": 0.0001,
      "loss": 4.0363,
      "loss/crossentropy": 2.3961373567581177,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27667778730392456,
      "step": 6432
    },
    {
      "epoch": 0.12868,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.009598795572916667,
      "learning_rate": 0.0001,
      "loss": 4.3993,
      "loss/crossentropy": 1.9141033291816711,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22818633913993835,
      "step": 6434
    },
    {
      "epoch": 0.12872,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.0070149739583333336,
      "learning_rate": 0.0001,
      "loss": 4.481,
      "loss/crossentropy": 2.4431718587875366,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24852856248617172,
      "step": 6436
    },
    {
      "epoch": 0.12876,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.006494140625,
      "learning_rate": 0.0001,
      "loss": 4.5153,
      "loss/crossentropy": 1.9343949556350708,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22181283682584763,
      "step": 6438
    },
    {
      "epoch": 0.1288,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.005549112955729167,
      "learning_rate": 0.0001,
      "loss": 4.4547,
      "loss/crossentropy": 2.203734040260315,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23536919057369232,
      "step": 6440
    },
    {
      "epoch": 0.12884,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0055572509765625,
      "learning_rate": 0.0001,
      "loss": 4.354,
      "loss/crossentropy": 2.0419046878814697,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2121758982539177,
      "step": 6442
    },
    {
      "epoch": 0.12888,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.0054595947265625,
      "learning_rate": 0.0001,
      "loss": 4.5738,
      "loss/crossentropy": 2.3551554679870605,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23853591084480286,
      "step": 6444
    },
    {
      "epoch": 0.12892,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.004813639322916666,
      "learning_rate": 0.0001,
      "loss": 4.4334,
      "loss/crossentropy": 1.9079387784004211,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22546496242284775,
      "step": 6446
    },
    {
      "epoch": 0.12896,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.00390625,
      "learning_rate": 0.0001,
      "loss": 4.4188,
      "loss/crossentropy": 1.9344156980514526,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21923892199993134,
      "step": 6448
    },
    {
      "epoch": 0.129,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.0047271728515625,
      "learning_rate": 0.0001,
      "loss": 4.3219,
      "loss/crossentropy": 1.7627189755439758,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20726975798606873,
      "step": 6450
    },
    {
      "epoch": 0.12904,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.0048736572265625,
      "learning_rate": 0.0001,
      "loss": 4.4045,
      "loss/crossentropy": 1.9988782405853271,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23156572133302689,
      "step": 6452
    },
    {
      "epoch": 0.12908,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.004621378580729167,
      "learning_rate": 0.0001,
      "loss": 4.6551,
      "loss/crossentropy": 2.3970296382904053,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24723558872938156,
      "step": 6454
    },
    {
      "epoch": 0.12912,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.00455322265625,
      "learning_rate": 0.0001,
      "loss": 4.3785,
      "loss/crossentropy": 2.009088099002838,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2188432812690735,
      "step": 6456
    },
    {
      "epoch": 0.12916,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.005882771809895834,
      "learning_rate": 0.0001,
      "loss": 4.9189,
      "loss/crossentropy": 2.166573464870453,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22506655752658844,
      "step": 6458
    },
    {
      "epoch": 0.1292,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.00670166015625,
      "learning_rate": 0.0001,
      "loss": 4.3281,
      "loss/crossentropy": 2.1567386388778687,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23143760859966278,
      "step": 6460
    },
    {
      "epoch": 0.12924,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.007307942708333333,
      "learning_rate": 0.0001,
      "loss": 3.9726,
      "loss/crossentropy": 1.8458901643753052,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22627578675746918,
      "step": 6462
    },
    {
      "epoch": 0.12928,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.00738525390625,
      "learning_rate": 0.0001,
      "loss": 4.1416,
      "loss/crossentropy": 1.7933887243270874,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1944974958896637,
      "step": 6464
    },
    {
      "epoch": 0.12932,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.0077707926432291664,
      "learning_rate": 0.0001,
      "loss": 4.3712,
      "loss/crossentropy": 2.14457631111145,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2461908757686615,
      "step": 6466
    },
    {
      "epoch": 0.12936,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.011563873291015625,
      "learning_rate": 0.0001,
      "loss": 4.2253,
      "loss/crossentropy": 2.2191081047058105,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24845656007528305,
      "step": 6468
    },
    {
      "epoch": 0.1294,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.012284088134765624,
      "learning_rate": 0.0001,
      "loss": 4.3992,
      "loss/crossentropy": 2.2655181884765625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24961909651756287,
      "step": 6470
    },
    {
      "epoch": 0.12944,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.012617746988932291,
      "learning_rate": 0.0001,
      "loss": 4.2431,
      "loss/crossentropy": 1.9992872476577759,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21180924773216248,
      "step": 6472
    },
    {
      "epoch": 0.12948,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.009348297119140625,
      "learning_rate": 0.0001,
      "loss": 4.2865,
      "loss/crossentropy": 2.1950103044509888,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23401658236980438,
      "step": 6474
    },
    {
      "epoch": 0.12952,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.010027821858723958,
      "learning_rate": 0.0001,
      "loss": 4.2614,
      "loss/crossentropy": 2.148743689060211,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2146785706281662,
      "step": 6476
    },
    {
      "epoch": 0.12956,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.010253651936848959,
      "learning_rate": 0.0001,
      "loss": 4.7196,
      "loss/crossentropy": 2.4312193393707275,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2596089243888855,
      "step": 6478
    },
    {
      "epoch": 0.1296,
      "grad_norm": 2.25,
      "grad_norm_var": 0.011844635009765625,
      "learning_rate": 0.0001,
      "loss": 4.7232,
      "loss/crossentropy": 2.061104893684387,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24564718455076218,
      "step": 6480
    },
    {
      "epoch": 0.12964,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.011224110921223959,
      "learning_rate": 0.0001,
      "loss": 4.6833,
      "loss/crossentropy": 2.1994687914848328,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2352369725704193,
      "step": 6482
    },
    {
      "epoch": 0.12968,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.010692342122395834,
      "learning_rate": 0.0001,
      "loss": 4.2378,
      "loss/crossentropy": 2.085163116455078,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2309635877609253,
      "step": 6484
    },
    {
      "epoch": 0.12972,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0098297119140625,
      "learning_rate": 0.0001,
      "loss": 4.5964,
      "loss/crossentropy": 2.3360198736190796,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24846713244915009,
      "step": 6486
    },
    {
      "epoch": 0.12976,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.010602823893229167,
      "learning_rate": 0.0001,
      "loss": 4.4454,
      "loss/crossentropy": 2.089003086090088,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24980145692825317,
      "step": 6488
    },
    {
      "epoch": 0.1298,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0108795166015625,
      "learning_rate": 0.0001,
      "loss": 4.4446,
      "loss/crossentropy": 2.0884299874305725,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23875930905342102,
      "step": 6490
    },
    {
      "epoch": 0.12984,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.01002197265625,
      "learning_rate": 0.0001,
      "loss": 4.1661,
      "loss/crossentropy": 1.9070702195167542,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21942836046218872,
      "step": 6492
    },
    {
      "epoch": 0.12988,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.02047119140625,
      "learning_rate": 0.0001,
      "loss": 4.6558,
      "loss/crossentropy": 2.1617711782455444,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22820374369621277,
      "step": 6494
    },
    {
      "epoch": 0.12992,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.0203765869140625,
      "learning_rate": 0.0001,
      "loss": 4.0938,
      "loss/crossentropy": 1.7782898545265198,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22162485867738724,
      "step": 6496
    },
    {
      "epoch": 0.12996,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.021028645833333335,
      "learning_rate": 0.0001,
      "loss": 4.4198,
      "loss/crossentropy": 2.1364612579345703,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22490044683218002,
      "step": 6498
    },
    {
      "epoch": 0.13,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.020213826497395834,
      "learning_rate": 0.0001,
      "loss": 4.569,
      "loss/crossentropy": 2.282773971557617,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24553906172513962,
      "step": 6500
    },
    {
      "epoch": 0.13004,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.020905558268229166,
      "learning_rate": 0.0001,
      "loss": 4.4965,
      "loss/crossentropy": 2.008001983165741,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22152648121118546,
      "step": 6502
    },
    {
      "epoch": 0.13008,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.021833292643229165,
      "learning_rate": 0.0001,
      "loss": 4.4535,
      "loss/crossentropy": 2.1681981086730957,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26258186995983124,
      "step": 6504
    },
    {
      "epoch": 0.13012,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.021761067708333335,
      "learning_rate": 0.0001,
      "loss": 4.6254,
      "loss/crossentropy": 2.5754435062408447,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26315446197986603,
      "step": 6506
    },
    {
      "epoch": 0.13016,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.021187337239583333,
      "learning_rate": 0.0001,
      "loss": 4.1505,
      "loss/crossentropy": 1.7897658348083496,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22884277999401093,
      "step": 6508
    },
    {
      "epoch": 0.1302,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.016844685872395834,
      "learning_rate": 0.0001,
      "loss": 4.6315,
      "loss/crossentropy": 1.8081435561180115,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21860769391059875,
      "step": 6510
    },
    {
      "epoch": 0.13024,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.0140289306640625,
      "learning_rate": 0.0001,
      "loss": 4.7371,
      "loss/crossentropy": 2.243759036064148,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2545628473162651,
      "step": 6512
    },
    {
      "epoch": 0.13028,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.03447265625,
      "learning_rate": 0.0001,
      "loss": 4.6147,
      "loss/crossentropy": 2.069986939430237,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20862630754709244,
      "step": 6514
    },
    {
      "epoch": 0.13032,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.0349761962890625,
      "learning_rate": 0.0001,
      "loss": 4.4758,
      "loss/crossentropy": 2.052341878414154,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2460884153842926,
      "step": 6516
    },
    {
      "epoch": 0.13036,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.0317047119140625,
      "learning_rate": 0.0001,
      "loss": 4.6369,
      "loss/crossentropy": 2.3376708030700684,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25382688641548157,
      "step": 6518
    },
    {
      "epoch": 0.1304,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0329498291015625,
      "learning_rate": 0.0001,
      "loss": 4.3048,
      "loss/crossentropy": 1.8329171538352966,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2135135680437088,
      "step": 6520
    },
    {
      "epoch": 0.13044,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.035791015625,
      "learning_rate": 0.0001,
      "loss": 4.6582,
      "loss/crossentropy": 2.077217698097229,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2265045866370201,
      "step": 6522
    },
    {
      "epoch": 0.13048,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.036783854166666664,
      "learning_rate": 0.0001,
      "loss": 4.2189,
      "loss/crossentropy": 2.0677568912506104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2468869537115097,
      "step": 6524
    },
    {
      "epoch": 0.13052,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.03591206868489583,
      "learning_rate": 0.0001,
      "loss": 4.5246,
      "loss/crossentropy": 2.0692074298858643,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2602302059531212,
      "step": 6526
    },
    {
      "epoch": 0.13056,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.03583984375,
      "learning_rate": 0.0001,
      "loss": 4.6709,
      "loss/crossentropy": 1.9767250418663025,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2232941836118698,
      "step": 6528
    },
    {
      "epoch": 0.1306,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.01480712890625,
      "learning_rate": 0.0001,
      "loss": 4.4851,
      "loss/crossentropy": 1.7538996934890747,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20786649733781815,
      "step": 6530
    },
    {
      "epoch": 0.13064,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.01578369140625,
      "learning_rate": 0.0001,
      "loss": 4.5598,
      "loss/crossentropy": 2.061249256134033,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23238955438137054,
      "step": 6532
    },
    {
      "epoch": 0.13068,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.0167877197265625,
      "learning_rate": 0.0001,
      "loss": 4.237,
      "loss/crossentropy": 1.9083253145217896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22536182403564453,
      "step": 6534
    },
    {
      "epoch": 0.13072,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.017952473958333333,
      "learning_rate": 0.0001,
      "loss": 4.3004,
      "loss/crossentropy": 2.052153766155243,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2343958392739296,
      "step": 6536
    },
    {
      "epoch": 0.13076,
      "grad_norm": 2.125,
      "grad_norm_var": 0.010091145833333334,
      "learning_rate": 0.0001,
      "loss": 4.3638,
      "loss/crossentropy": 1.9979270100593567,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20102836191654205,
      "step": 6538
    },
    {
      "epoch": 0.1308,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.012031809488932291,
      "learning_rate": 0.0001,
      "loss": 4.019,
      "loss/crossentropy": 1.618333637714386,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21146921813488007,
      "step": 6540
    },
    {
      "epoch": 0.13084,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.011940256754557291,
      "learning_rate": 0.0001,
      "loss": 4.2514,
      "loss/crossentropy": 1.771790623664856,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21661869436502457,
      "step": 6542
    },
    {
      "epoch": 0.13088,
      "grad_norm": 2.25,
      "grad_norm_var": 0.007165273030598958,
      "learning_rate": 0.0001,
      "loss": 4.3528,
      "loss/crossentropy": 2.2347733974456787,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2507361173629761,
      "step": 6544
    },
    {
      "epoch": 0.13092,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.012389882405598959,
      "learning_rate": 0.0001,
      "loss": 4.4333,
      "loss/crossentropy": 2.226397395133972,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.257301464676857,
      "step": 6546
    },
    {
      "epoch": 0.13096,
      "grad_norm": 2.125,
      "grad_norm_var": 0.012776438395182292,
      "learning_rate": 0.0001,
      "loss": 4.1629,
      "loss/crossentropy": 1.9884281158447266,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2213538959622383,
      "step": 6548
    },
    {
      "epoch": 0.131,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.01755549112955729,
      "learning_rate": 0.0001,
      "loss": 4.549,
      "loss/crossentropy": 2.0790398120880127,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2259911745786667,
      "step": 6550
    },
    {
      "epoch": 0.13104,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.016294097900390624,
      "learning_rate": 0.0001,
      "loss": 4.5118,
      "loss/crossentropy": 2.1309107542037964,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23984672129154205,
      "step": 6552
    },
    {
      "epoch": 0.13108,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.01744562784830729,
      "learning_rate": 0.0001,
      "loss": 4.4528,
      "loss/crossentropy": 1.9651963114738464,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21699358522891998,
      "step": 6554
    },
    {
      "epoch": 0.13112,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.014598592122395834,
      "learning_rate": 0.0001,
      "loss": 4.436,
      "loss/crossentropy": 2.405247449874878,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24624405801296234,
      "step": 6556
    },
    {
      "epoch": 0.13116,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0154296875,
      "learning_rate": 0.0001,
      "loss": 4.3984,
      "loss/crossentropy": 2.215611457824707,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24835016578435898,
      "step": 6558
    },
    {
      "epoch": 0.1312,
      "grad_norm": 4.125,
      "grad_norm_var": 0.2436920166015625,
      "learning_rate": 0.0001,
      "loss": 4.3876,
      "loss/crossentropy": 2.1731653809547424,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24633550643920898,
      "step": 6560
    },
    {
      "epoch": 0.13124,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.24501953125,
      "learning_rate": 0.0001,
      "loss": 4.5101,
      "loss/crossentropy": 2.0507587790489197,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25702129304409027,
      "step": 6562
    },
    {
      "epoch": 0.13128,
      "grad_norm": 2.375,
      "grad_norm_var": 0.2433502197265625,
      "learning_rate": 0.0001,
      "loss": 4.3464,
      "loss/crossentropy": 2.2088446617126465,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22610870003700256,
      "step": 6564
    },
    {
      "epoch": 0.13132,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.23931884765625,
      "learning_rate": 0.0001,
      "loss": 4.3612,
      "loss/crossentropy": 1.6911352276802063,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2188590243458748,
      "step": 6566
    },
    {
      "epoch": 0.13136,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.24976806640625,
      "learning_rate": 0.0001,
      "loss": 3.9795,
      "loss/crossentropy": 1.807699978351593,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21844930946826935,
      "step": 6568
    },
    {
      "epoch": 0.1314,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.24514872233072918,
      "learning_rate": 0.0001,
      "loss": 4.4293,
      "loss/crossentropy": 2.292602300643921,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.242776520550251,
      "step": 6570
    },
    {
      "epoch": 0.13144,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.24806315104166668,
      "learning_rate": 0.0001,
      "loss": 4.07,
      "loss/crossentropy": 1.5262329578399658,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1674257293343544,
      "step": 6572
    },
    {
      "epoch": 0.13148,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.24504801432291667,
      "learning_rate": 0.0001,
      "loss": 4.4084,
      "loss/crossentropy": 2.180319309234619,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2260192409157753,
      "step": 6574
    },
    {
      "epoch": 0.13152,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.0216705322265625,
      "learning_rate": 0.0001,
      "loss": 4.544,
      "loss/crossentropy": 2.188440203666687,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23389383405447006,
      "step": 6576
    },
    {
      "epoch": 0.13156,
      "grad_norm": 2.125,
      "grad_norm_var": 0.03427327473958333,
      "learning_rate": 0.0001,
      "loss": 4.7595,
      "loss/crossentropy": 2.20253586769104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3217846751213074,
      "step": 6578
    },
    {
      "epoch": 0.1316,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.034764607747395836,
      "learning_rate": 0.0001,
      "loss": 4.2771,
      "loss/crossentropy": 2.1901479959487915,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24521923065185547,
      "step": 6580
    },
    {
      "epoch": 0.13164,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.04170633951822917,
      "learning_rate": 0.0001,
      "loss": 4.7105,
      "loss/crossentropy": 2.0719348192214966,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2640562355518341,
      "step": 6582
    },
    {
      "epoch": 0.13168,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.040339152018229164,
      "learning_rate": 0.0001,
      "loss": 4.1487,
      "loss/crossentropy": 2.331762194633484,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24890532344579697,
      "step": 6584
    },
    {
      "epoch": 0.13172,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.04429931640625,
      "learning_rate": 0.0001,
      "loss": 4.6013,
      "loss/crossentropy": 1.5143779516220093,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1957392692565918,
      "step": 6586
    },
    {
      "epoch": 0.13176,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.04088134765625,
      "learning_rate": 0.0001,
      "loss": 4.4311,
      "loss/crossentropy": 2.301910698413849,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27353671938180923,
      "step": 6588
    },
    {
      "epoch": 0.1318,
      "grad_norm": 2.84375,
      "grad_norm_var": 0.11796773274739583,
      "learning_rate": 0.0001,
      "loss": 4.5035,
      "loss/crossentropy": 2.018579602241516,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24261966347694397,
      "step": 6590
    },
    {
      "epoch": 0.13184,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.11685282389322917,
      "learning_rate": 0.0001,
      "loss": 4.6931,
      "loss/crossentropy": 1.9749983549118042,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20727626234292984,
      "step": 6592
    },
    {
      "epoch": 0.13188,
      "grad_norm": 2.0,
      "grad_norm_var": 0.11741536458333333,
      "learning_rate": 0.0001,
      "loss": 4.1929,
      "loss/crossentropy": 2.040414035320282,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22879169881343842,
      "step": 6594
    },
    {
      "epoch": 0.13192,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.11551106770833333,
      "learning_rate": 0.0001,
      "loss": 4.3331,
      "loss/crossentropy": 1.941766619682312,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22591909021139145,
      "step": 6596
    },
    {
      "epoch": 0.13196,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.10946858723958333,
      "learning_rate": 0.0001,
      "loss": 4.3659,
      "loss/crossentropy": 1.8487200140953064,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23654203116893768,
      "step": 6598
    },
    {
      "epoch": 0.132,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.10340067545572916,
      "learning_rate": 0.0001,
      "loss": 4.225,
      "loss/crossentropy": 1.9297338724136353,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2016594037413597,
      "step": 6600
    },
    {
      "epoch": 0.13204,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.11005757649739584,
      "learning_rate": 0.0001,
      "loss": 4.3643,
      "loss/crossentropy": 2.048487663269043,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20044995844364166,
      "step": 6602
    },
    {
      "epoch": 0.13208,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.11295572916666667,
      "learning_rate": 0.0001,
      "loss": 4.425,
      "loss/crossentropy": 2.3620439767837524,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24750277400016785,
      "step": 6604
    },
    {
      "epoch": 0.13212,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.0158203125,
      "learning_rate": 0.0001,
      "loss": 4.2578,
      "loss/crossentropy": 2.1324113607406616,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23109012842178345,
      "step": 6606
    },
    {
      "epoch": 0.13216,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.015620930989583334,
      "learning_rate": 0.0001,
      "loss": 4.4775,
      "loss/crossentropy": 2.021477997303009,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2152409330010414,
      "step": 6608
    },
    {
      "epoch": 0.1322,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.012669881184895834,
      "learning_rate": 0.0001,
      "loss": 4.6221,
      "loss/crossentropy": 2.1693036556243896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.234656922519207,
      "step": 6610
    },
    {
      "epoch": 0.13224,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.008854166666666666,
      "learning_rate": 0.0001,
      "loss": 4.1469,
      "loss/crossentropy": 2.1138017177581787,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2239084094762802,
      "step": 6612
    },
    {
      "epoch": 0.13228,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.008772786458333333,
      "learning_rate": 0.0001,
      "loss": 4.4862,
      "loss/crossentropy": 2.2540252208709717,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2614182382822037,
      "step": 6614
    },
    {
      "epoch": 0.13232,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.008837890625,
      "learning_rate": 0.0001,
      "loss": 4.2131,
      "loss/crossentropy": 2.033502757549286,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2177339717745781,
      "step": 6616
    },
    {
      "epoch": 0.13236,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.010054524739583333,
      "learning_rate": 0.0001,
      "loss": 4.2648,
      "loss/crossentropy": 2.2490646839141846,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22369390726089478,
      "step": 6618
    },
    {
      "epoch": 0.1324,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.0150543212890625,
      "learning_rate": 0.0001,
      "loss": 4.3571,
      "loss/crossentropy": 2.14319908618927,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24394190311431885,
      "step": 6620
    },
    {
      "epoch": 0.13244,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.01627197265625,
      "learning_rate": 0.0001,
      "loss": 4.5136,
      "loss/crossentropy": 2.045474410057068,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22738997638225555,
      "step": 6622
    },
    {
      "epoch": 0.13248,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.015816243489583333,
      "learning_rate": 0.0001,
      "loss": 4.5773,
      "loss/crossentropy": 2.1853290796279907,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2565506473183632,
      "step": 6624
    },
    {
      "epoch": 0.13252,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.017015584309895835,
      "learning_rate": 0.0001,
      "loss": 4.275,
      "loss/crossentropy": 2.1161083579063416,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22949891537427902,
      "step": 6626
    },
    {
      "epoch": 0.13256,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0167633056640625,
      "learning_rate": 0.0001,
      "loss": 4.3154,
      "loss/crossentropy": 2.167048454284668,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23087909072637558,
      "step": 6628
    },
    {
      "epoch": 0.1326,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.0156158447265625,
      "learning_rate": 0.0001,
      "loss": 4.2777,
      "loss/crossentropy": 2.1544610261917114,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25153525173664093,
      "step": 6630
    },
    {
      "epoch": 0.13264,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.018244425455729168,
      "learning_rate": 0.0001,
      "loss": 4.4778,
      "loss/crossentropy": 2.0319228768348694,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22067170590162277,
      "step": 6632
    },
    {
      "epoch": 0.13268,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0162261962890625,
      "learning_rate": 0.0001,
      "loss": 4.4398,
      "loss/crossentropy": 1.9723476767539978,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21380966901779175,
      "step": 6634
    },
    {
      "epoch": 0.13272,
      "grad_norm": 2.796875,
      "grad_norm_var": 0.031590779622395836,
      "learning_rate": 0.0001,
      "loss": 4.3474,
      "loss/crossentropy": 2.2833333015441895,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23245477676391602,
      "step": 6636
    },
    {
      "epoch": 0.13276,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.03132222493489583,
      "learning_rate": 0.0001,
      "loss": 4.4522,
      "loss/crossentropy": 2.13793683052063,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22665268182754517,
      "step": 6638
    },
    {
      "epoch": 0.1328,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.04592692057291667,
      "learning_rate": 0.0001,
      "loss": 4.6386,
      "loss/crossentropy": 2.188693881034851,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2471916377544403,
      "step": 6640
    },
    {
      "epoch": 0.13284,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.0438629150390625,
      "learning_rate": 0.0001,
      "loss": 4.6394,
      "loss/crossentropy": 2.169856071472168,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23427975177764893,
      "step": 6642
    },
    {
      "epoch": 0.13288,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.04365132649739583,
      "learning_rate": 0.0001,
      "loss": 4.6498,
      "loss/crossentropy": 2.1600061655044556,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23366540670394897,
      "step": 6644
    },
    {
      "epoch": 0.13292,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.044188435872395834,
      "learning_rate": 0.0001,
      "loss": 4.4521,
      "loss/crossentropy": 1.822945475578308,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2203340008854866,
      "step": 6646
    },
    {
      "epoch": 0.13296,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.045735677083333336,
      "learning_rate": 0.0001,
      "loss": 4.3263,
      "loss/crossentropy": 1.8908653259277344,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2320382297039032,
      "step": 6648
    },
    {
      "epoch": 0.133,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.04363606770833333,
      "learning_rate": 0.0001,
      "loss": 4.5065,
      "loss/crossentropy": 2.126000165939331,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24388836324214935,
      "step": 6650
    },
    {
      "epoch": 0.13304,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.0242828369140625,
      "learning_rate": 0.0001,
      "loss": 4.5787,
      "loss/crossentropy": 2.434928297996521,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24083568900823593,
      "step": 6652
    },
    {
      "epoch": 0.13308,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.023824055989583332,
      "learning_rate": 0.0001,
      "loss": 4.5538,
      "loss/crossentropy": 2.2186567783355713,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2220132276415825,
      "step": 6654
    },
    {
      "epoch": 0.13312,
      "grad_norm": 2.25,
      "grad_norm_var": 0.006257120768229167,
      "learning_rate": 0.0001,
      "loss": 4.4934,
      "loss/crossentropy": 1.849799931049347,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21551364660263062,
      "step": 6656
    },
    {
      "epoch": 0.13316,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0072662353515625,
      "learning_rate": 0.0001,
      "loss": 4.2237,
      "loss/crossentropy": 2.082044243812561,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21274320781230927,
      "step": 6658
    },
    {
      "epoch": 0.1332,
      "grad_norm": 2.125,
      "grad_norm_var": 0.010838826497395834,
      "learning_rate": 0.0001,
      "loss": 4.5884,
      "loss/crossentropy": 2.1957098245620728,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23078418523073196,
      "step": 6660
    },
    {
      "epoch": 0.13324,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.16033528645833334,
      "learning_rate": 0.0001,
      "loss": 4.519,
      "loss/crossentropy": 2.228309690952301,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27178408950567245,
      "step": 6662
    },
    {
      "epoch": 0.13328,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.156298828125,
      "learning_rate": 0.0001,
      "loss": 4.5987,
      "loss/crossentropy": 1.8185940384864807,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2145048901438713,
      "step": 6664
    },
    {
      "epoch": 0.13332,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.1566558837890625,
      "learning_rate": 0.0001,
      "loss": 4.4722,
      "loss/crossentropy": 2.198649048805237,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2262621968984604,
      "step": 6666
    },
    {
      "epoch": 0.13336,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.159130859375,
      "learning_rate": 0.0001,
      "loss": 4.5729,
      "loss/crossentropy": 2.2075835466384888,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23531068861484528,
      "step": 6668
    },
    {
      "epoch": 0.1334,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.16243387858072916,
      "learning_rate": 0.0001,
      "loss": 4.2913,
      "loss/crossentropy": 1.9719768166542053,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2206977903842926,
      "step": 6670
    },
    {
      "epoch": 0.13344,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.17021484375,
      "learning_rate": 0.0001,
      "loss": 4.2144,
      "loss/crossentropy": 2.304553985595703,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23418369889259338,
      "step": 6672
    },
    {
      "epoch": 0.13348,
      "grad_norm": 2.25,
      "grad_norm_var": 0.1682281494140625,
      "learning_rate": 0.0001,
      "loss": 4.4485,
      "loss/crossentropy": 2.212409734725952,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23888318240642548,
      "step": 6674
    },
    {
      "epoch": 0.13352,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.19371337890625,
      "learning_rate": 0.0001,
      "loss": 4.176,
      "loss/crossentropy": 2.001897156238556,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2147163525223732,
      "step": 6676
    },
    {
      "epoch": 0.13356,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.043680826822916664,
      "learning_rate": 0.0001,
      "loss": 4.4245,
      "loss/crossentropy": 2.216760039329529,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24913202226161957,
      "step": 6678
    },
    {
      "epoch": 0.1336,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.04597142537434896,
      "learning_rate": 0.0001,
      "loss": 4.1862,
      "loss/crossentropy": 1.8190750479698181,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.200186125934124,
      "step": 6680
    },
    {
      "epoch": 0.13364,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.047548166910807294,
      "learning_rate": 0.0001,
      "loss": 4.7271,
      "loss/crossentropy": 2.311274528503418,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22400956600904465,
      "step": 6682
    },
    {
      "epoch": 0.13368,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.5293841044108073,
      "learning_rate": 0.0001,
      "loss": 4.3852,
      "loss/crossentropy": 2.0381893515586853,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23101551085710526,
      "step": 6684
    },
    {
      "epoch": 0.13372,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.5331776936848959,
      "learning_rate": 0.0001,
      "loss": 4.1185,
      "loss/crossentropy": 1.7441503405570984,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20208899676799774,
      "step": 6686
    },
    {
      "epoch": 0.13376,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.5252593994140625,
      "learning_rate": 0.0001,
      "loss": 4.3835,
      "loss/crossentropy": 1.8874938488006592,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2255900353193283,
      "step": 6688
    },
    {
      "epoch": 0.1338,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.5284006754557292,
      "learning_rate": 0.0001,
      "loss": 4.2563,
      "loss/crossentropy": 2.2768125534057617,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21636968851089478,
      "step": 6690
    },
    {
      "epoch": 0.13384,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.5063954671223958,
      "learning_rate": 0.0001,
      "loss": 4.278,
      "loss/crossentropy": 2.0253931283950806,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28937554359436035,
      "step": 6692
    },
    {
      "epoch": 0.13388,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.5079661051432292,
      "learning_rate": 0.0001,
      "loss": 4.4205,
      "loss/crossentropy": 2.1940718293190002,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23384064435958862,
      "step": 6694
    },
    {
      "epoch": 0.13392,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.4987993876139323,
      "learning_rate": 0.0001,
      "loss": 4.5066,
      "loss/crossentropy": 2.071534514427185,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2146512120962143,
      "step": 6696
    },
    {
      "epoch": 0.13396,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.5017534891764323,
      "learning_rate": 0.0001,
      "loss": 4.348,
      "loss/crossentropy": 1.8530714511871338,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21837462484836578,
      "step": 6698
    },
    {
      "epoch": 0.134,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.013396962483723959,
      "learning_rate": 0.0001,
      "loss": 4.521,
      "loss/crossentropy": 2.210664451122284,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2199932038784027,
      "step": 6700
    },
    {
      "epoch": 0.13404,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.011506144205729167,
      "learning_rate": 0.0001,
      "loss": 4.4226,
      "loss/crossentropy": 1.8530223965644836,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.209881991147995,
      "step": 6702
    },
    {
      "epoch": 0.13408,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.011839803059895833,
      "learning_rate": 0.0001,
      "loss": 4.4302,
      "loss/crossentropy": 1.8609183430671692,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22096765041351318,
      "step": 6704
    },
    {
      "epoch": 0.13412,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.021214803059895832,
      "learning_rate": 0.0001,
      "loss": 4.8429,
      "loss/crossentropy": 2.33315110206604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26414938271045685,
      "step": 6706
    },
    {
      "epoch": 0.13416,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.04798075358072917,
      "learning_rate": 0.0001,
      "loss": 4.6054,
      "loss/crossentropy": 2.2656116485595703,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2769838646054268,
      "step": 6708
    },
    {
      "epoch": 0.1342,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0466705322265625,
      "learning_rate": 0.0001,
      "loss": 4.4875,
      "loss/crossentropy": 2.2131590843200684,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23640615493059158,
      "step": 6710
    },
    {
      "epoch": 0.13424,
      "grad_norm": 2.25,
      "grad_norm_var": 0.044831339518229166,
      "learning_rate": 0.0001,
      "loss": 4.1554,
      "loss/crossentropy": 1.8667671084403992,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.212738998234272,
      "step": 6712
    },
    {
      "epoch": 0.13428,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.04480692545572917,
      "learning_rate": 0.0001,
      "loss": 4.0958,
      "loss/crossentropy": 1.9699830412864685,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23507894575595856,
      "step": 6714
    },
    {
      "epoch": 0.13432,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.04632161458333333,
      "learning_rate": 0.0001,
      "loss": 4.1571,
      "loss/crossentropy": 1.8108918070793152,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2192004919052124,
      "step": 6716
    },
    {
      "epoch": 0.13436,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.042740885416666666,
      "learning_rate": 0.0001,
      "loss": 4.4983,
      "loss/crossentropy": 2.0528674125671387,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22221814841032028,
      "step": 6718
    },
    {
      "epoch": 0.1344,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.046305338541666664,
      "learning_rate": 0.0001,
      "loss": 4.2544,
      "loss/crossentropy": 1.7881956696510315,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21330178529024124,
      "step": 6720
    },
    {
      "epoch": 0.13444,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.039159138997395836,
      "learning_rate": 0.0001,
      "loss": 4.5424,
      "loss/crossentropy": 2.2016018629074097,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22042546421289444,
      "step": 6722
    },
    {
      "epoch": 0.13448,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.006322224934895833,
      "learning_rate": 0.0001,
      "loss": 4.3159,
      "loss/crossentropy": 1.9661846160888672,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21542846411466599,
      "step": 6724
    },
    {
      "epoch": 0.13452,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.031037394205729166,
      "learning_rate": 0.0001,
      "loss": 4.4473,
      "loss/crossentropy": 2.1073482036590576,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23370730876922607,
      "step": 6726
    },
    {
      "epoch": 0.13456,
      "grad_norm": 2.25,
      "grad_norm_var": 0.030826822916666666,
      "learning_rate": 0.0001,
      "loss": 4.6069,
      "loss/crossentropy": 2.1937917470932007,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22414422780275345,
      "step": 6728
    },
    {
      "epoch": 0.1346,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.030436197916666668,
      "learning_rate": 0.0001,
      "loss": 4.2937,
      "loss/crossentropy": 2.078732967376709,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21227504312992096,
      "step": 6730
    },
    {
      "epoch": 0.13464,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.030301920572916665,
      "learning_rate": 0.0001,
      "loss": 4.3043,
      "loss/crossentropy": 1.9002525806427002,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21004119515419006,
      "step": 6732
    },
    {
      "epoch": 0.13468,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.030794270833333335,
      "learning_rate": 0.0001,
      "loss": 4.5156,
      "loss/crossentropy": 2.055518925189972,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21610142290592194,
      "step": 6734
    },
    {
      "epoch": 0.13472,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.028473917643229166,
      "learning_rate": 0.0001,
      "loss": 4.4889,
      "loss/crossentropy": 2.0521084666252136,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23528365790843964,
      "step": 6736
    },
    {
      "epoch": 0.13476,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.02730712890625,
      "learning_rate": 0.0001,
      "loss": 4.5051,
      "loss/crossentropy": 2.2536301612854004,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21544227004051208,
      "step": 6738
    },
    {
      "epoch": 0.1348,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.027179972330729166,
      "learning_rate": 0.0001,
      "loss": 4.5164,
      "loss/crossentropy": 2.2610143423080444,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2532905787229538,
      "step": 6740
    },
    {
      "epoch": 0.13484,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.00836181640625,
      "learning_rate": 0.0001,
      "loss": 4.1301,
      "loss/crossentropy": 1.8840081095695496,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21514790505170822,
      "step": 6742
    },
    {
      "epoch": 0.13488,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0084136962890625,
      "learning_rate": 0.0001,
      "loss": 4.2532,
      "loss/crossentropy": 2.0841002464294434,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2328692153096199,
      "step": 6744
    },
    {
      "epoch": 0.13492,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.010798136393229166,
      "learning_rate": 0.0001,
      "loss": 4.6335,
      "loss/crossentropy": 2.507196068763733,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2736222892999649,
      "step": 6746
    },
    {
      "epoch": 0.13496,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.016527303059895835,
      "learning_rate": 0.0001,
      "loss": 4.189,
      "loss/crossentropy": 2.0180357098579407,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22328195720911026,
      "step": 6748
    },
    {
      "epoch": 0.135,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.016097005208333334,
      "learning_rate": 0.0001,
      "loss": 4.2854,
      "loss/crossentropy": 2.2457560300827026,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26572495698928833,
      "step": 6750
    },
    {
      "epoch": 0.13504,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.031493123372395834,
      "learning_rate": 0.0001,
      "loss": 4.885,
      "loss/crossentropy": 2.1280853748321533,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22716474533081055,
      "step": 6752
    },
    {
      "epoch": 0.13508,
      "grad_norm": 2.25,
      "grad_norm_var": 0.03216145833333333,
      "learning_rate": 0.0001,
      "loss": 4.3901,
      "loss/crossentropy": 2.1843650341033936,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24630828201770782,
      "step": 6754
    },
    {
      "epoch": 0.13512,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.03390299479166667,
      "learning_rate": 0.0001,
      "loss": 4.3584,
      "loss/crossentropy": 1.7955012917518616,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20614011585712433,
      "step": 6756
    },
    {
      "epoch": 0.13516,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.028563435872395834,
      "learning_rate": 0.0001,
      "loss": 4.3546,
      "loss/crossentropy": 1.9315852522850037,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22918011993169785,
      "step": 6758
    },
    {
      "epoch": 0.1352,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.028055826822916668,
      "learning_rate": 0.0001,
      "loss": 4.2659,
      "loss/crossentropy": 1.982887327671051,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21635843813419342,
      "step": 6760
    },
    {
      "epoch": 0.13524,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.029215494791666668,
      "learning_rate": 0.0001,
      "loss": 4.2828,
      "loss/crossentropy": 2.25021892786026,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22931701689958572,
      "step": 6762
    },
    {
      "epoch": 0.13528,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.02252197265625,
      "learning_rate": 0.0001,
      "loss": 4.5991,
      "loss/crossentropy": 2.5220746994018555,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2406606376171112,
      "step": 6764
    },
    {
      "epoch": 0.13532,
      "grad_norm": 2.375,
      "grad_norm_var": 0.022386678059895835,
      "learning_rate": 0.0001,
      "loss": 4.5143,
      "loss/crossentropy": 1.8115127086639404,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.220379076898098,
      "step": 6766
    },
    {
      "epoch": 0.13536,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.010856119791666667,
      "learning_rate": 0.0001,
      "loss": 4.5113,
      "loss/crossentropy": 1.8998088240623474,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22993376106023788,
      "step": 6768
    },
    {
      "epoch": 0.1354,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.01051025390625,
      "learning_rate": 0.0001,
      "loss": 4.28,
      "loss/crossentropy": 2.0183660984039307,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21991200000047684,
      "step": 6770
    },
    {
      "epoch": 0.13544,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.009847005208333334,
      "learning_rate": 0.0001,
      "loss": 4.6224,
      "loss/crossentropy": 2.1927448511123657,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2536699026823044,
      "step": 6772
    },
    {
      "epoch": 0.13548,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.014404296875,
      "learning_rate": 0.0001,
      "loss": 4.196,
      "loss/crossentropy": 1.92184317111969,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20716708898544312,
      "step": 6774
    },
    {
      "epoch": 0.13552,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.016813151041666665,
      "learning_rate": 0.0001,
      "loss": 4.7779,
      "loss/crossentropy": 2.2437468767166138,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26381388306617737,
      "step": 6776
    },
    {
      "epoch": 0.13556,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.016722615559895834,
      "learning_rate": 0.0001,
      "loss": 4.2907,
      "loss/crossentropy": 2.087414026260376,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21566492319107056,
      "step": 6778
    },
    {
      "epoch": 0.1356,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.015653483072916665,
      "learning_rate": 0.0001,
      "loss": 4.4273,
      "loss/crossentropy": 2.1936367750167847,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22492723166942596,
      "step": 6780
    },
    {
      "epoch": 0.13564,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.014046223958333333,
      "learning_rate": 0.0001,
      "loss": 4.2992,
      "loss/crossentropy": 1.7642306685447693,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20992937684059143,
      "step": 6782
    },
    {
      "epoch": 0.13568,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.014371744791666667,
      "learning_rate": 0.0001,
      "loss": 4.3593,
      "loss/crossentropy": 2.01781964302063,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22552715986967087,
      "step": 6784
    },
    {
      "epoch": 0.13572,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.016402180989583334,
      "learning_rate": 0.0001,
      "loss": 4.4708,
      "loss/crossentropy": 2.0788158774375916,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24181769788265228,
      "step": 6786
    },
    {
      "epoch": 0.13576,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.01539306640625,
      "learning_rate": 0.0001,
      "loss": 4.2163,
      "loss/crossentropy": 2.0424017310142517,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21705424785614014,
      "step": 6788
    },
    {
      "epoch": 0.1358,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.019025675455729165,
      "learning_rate": 0.0001,
      "loss": 4.1182,
      "loss/crossentropy": 1.6175345182418823,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1899409219622612,
      "step": 6790
    },
    {
      "epoch": 0.13584,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.016852823893229167,
      "learning_rate": 0.0001,
      "loss": 4.2914,
      "loss/crossentropy": 2.004386007785797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22395263612270355,
      "step": 6792
    },
    {
      "epoch": 0.13588,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.019245402018229166,
      "learning_rate": 0.0001,
      "loss": 4.2182,
      "loss/crossentropy": 1.9224132895469666,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2326791137456894,
      "step": 6794
    },
    {
      "epoch": 0.13592,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.019755045572916668,
      "learning_rate": 0.0001,
      "loss": 4.4768,
      "loss/crossentropy": 1.8331453204154968,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21586360037326813,
      "step": 6796
    },
    {
      "epoch": 0.13596,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0197906494140625,
      "learning_rate": 0.0001,
      "loss": 4.3059,
      "loss/crossentropy": 2.535244107246399,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24744001775979996,
      "step": 6798
    },
    {
      "epoch": 0.136,
      "grad_norm": 2.25,
      "grad_norm_var": 0.017704264322916666,
      "learning_rate": 0.0001,
      "loss": 4.4444,
      "loss/crossentropy": 2.0433249473571777,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2502296343445778,
      "step": 6800
    },
    {
      "epoch": 0.13604,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0221588134765625,
      "learning_rate": 0.0001,
      "loss": 4.4619,
      "loss/crossentropy": 2.35608172416687,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2536633685231209,
      "step": 6802
    },
    {
      "epoch": 0.13608,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.022362263997395833,
      "learning_rate": 0.0001,
      "loss": 4.4493,
      "loss/crossentropy": 2.1230265498161316,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2276441976428032,
      "step": 6804
    },
    {
      "epoch": 0.13612,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.015925089518229168,
      "learning_rate": 0.0001,
      "loss": 4.2125,
      "loss/crossentropy": 2.0186346769332886,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22158686816692352,
      "step": 6806
    },
    {
      "epoch": 0.13616,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.0151763916015625,
      "learning_rate": 0.0001,
      "loss": 4.6065,
      "loss/crossentropy": 2.4136343002319336,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22668009996414185,
      "step": 6808
    },
    {
      "epoch": 0.1362,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.012495930989583333,
      "learning_rate": 0.0001,
      "loss": 4.3372,
      "loss/crossentropy": 2.1241788268089294,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23394256830215454,
      "step": 6810
    },
    {
      "epoch": 0.13624,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.017145792643229168,
      "learning_rate": 0.0001,
      "loss": 4.6063,
      "loss/crossentropy": 1.9051874279975891,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2560805454850197,
      "step": 6812
    },
    {
      "epoch": 0.13628,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.015458170572916667,
      "learning_rate": 0.0001,
      "loss": 4.2419,
      "loss/crossentropy": 1.9248363375663757,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22457116842269897,
      "step": 6814
    },
    {
      "epoch": 0.13632,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.022264607747395835,
      "learning_rate": 0.0001,
      "loss": 4.5953,
      "loss/crossentropy": 2.3065048456192017,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.245137557387352,
      "step": 6816
    },
    {
      "epoch": 0.13636,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.0152496337890625,
      "learning_rate": 0.0001,
      "loss": 4.6096,
      "loss/crossentropy": 2.152611255645752,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22794383764266968,
      "step": 6818
    },
    {
      "epoch": 0.1364,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.016242472330729167,
      "learning_rate": 0.0001,
      "loss": 4.129,
      "loss/crossentropy": 1.9548735618591309,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20764236897230148,
      "step": 6820
    },
    {
      "epoch": 0.13644,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.016136678059895833,
      "learning_rate": 0.0001,
      "loss": 4.3501,
      "loss/crossentropy": 2.11979341506958,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22793393582105637,
      "step": 6822
    },
    {
      "epoch": 0.13648,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.016748046875,
      "learning_rate": 0.0001,
      "loss": 4.3338,
      "loss/crossentropy": 2.351949691772461,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2366446554660797,
      "step": 6824
    },
    {
      "epoch": 0.13652,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.018843587239583334,
      "learning_rate": 0.0001,
      "loss": 4.3764,
      "loss/crossentropy": 2.1197460889816284,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24759536981582642,
      "step": 6826
    },
    {
      "epoch": 0.13656,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.013264973958333334,
      "learning_rate": 0.0001,
      "loss": 4.1204,
      "loss/crossentropy": 1.8491687178611755,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20777105540037155,
      "step": 6828
    },
    {
      "epoch": 0.1366,
      "grad_norm": 2.25,
      "grad_norm_var": 0.013459269205729167,
      "learning_rate": 0.0001,
      "loss": 4.4867,
      "loss/crossentropy": 1.964136004447937,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2368694394826889,
      "step": 6830
    },
    {
      "epoch": 0.13664,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.006859334309895834,
      "learning_rate": 0.0001,
      "loss": 4.469,
      "loss/crossentropy": 1.8988104462623596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19548720866441727,
      "step": 6832
    },
    {
      "epoch": 0.13668,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0059855143229166664,
      "learning_rate": 0.0001,
      "loss": 4.3104,
      "loss/crossentropy": 1.757002353668213,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21120281517505646,
      "step": 6834
    },
    {
      "epoch": 0.13672,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.005952962239583333,
      "learning_rate": 0.0001,
      "loss": 4.4239,
      "loss/crossentropy": 1.9414420127868652,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22160177677869797,
      "step": 6836
    },
    {
      "epoch": 0.13676,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0072174072265625,
      "learning_rate": 0.0001,
      "loss": 4.4294,
      "loss/crossentropy": 2.280028223991394,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24696747958660126,
      "step": 6838
    },
    {
      "epoch": 0.1368,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.006787109375,
      "learning_rate": 0.0001,
      "loss": 4.523,
      "loss/crossentropy": 2.106986403465271,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24022039771080017,
      "step": 6840
    },
    {
      "epoch": 0.13684,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.0349517822265625,
      "learning_rate": 0.0001,
      "loss": 4.2609,
      "loss/crossentropy": 1.9540700912475586,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1968660056591034,
      "step": 6842
    },
    {
      "epoch": 0.13688,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.03394775390625,
      "learning_rate": 0.0001,
      "loss": 4.2181,
      "loss/crossentropy": 1.6771780252456665,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20230162143707275,
      "step": 6844
    },
    {
      "epoch": 0.13692,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.04419657389322917,
      "learning_rate": 0.0001,
      "loss": 4.7567,
      "loss/crossentropy": 2.059873402118683,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.276339128613472,
      "step": 6846
    },
    {
      "epoch": 0.13696,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.04684244791666667,
      "learning_rate": 0.0001,
      "loss": 4.2512,
      "loss/crossentropy": 1.7943353056907654,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1893974468111992,
      "step": 6848
    },
    {
      "epoch": 0.137,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.046223958333333336,
      "learning_rate": 0.0001,
      "loss": 4.687,
      "loss/crossentropy": 2.314136028289795,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24898302555084229,
      "step": 6850
    },
    {
      "epoch": 0.13704,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.04383036295572917,
      "learning_rate": 0.0001,
      "loss": 4.4257,
      "loss/crossentropy": 2.0062466263771057,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23169535398483276,
      "step": 6852
    },
    {
      "epoch": 0.13708,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.044873046875,
      "learning_rate": 0.0001,
      "loss": 4.5787,
      "loss/crossentropy": 2.3600821495056152,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25484780967235565,
      "step": 6854
    },
    {
      "epoch": 0.13712,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.043675740559895836,
      "learning_rate": 0.0001,
      "loss": 4.2113,
      "loss/crossentropy": 1.885023295879364,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.255520723760128,
      "step": 6856
    },
    {
      "epoch": 0.13716,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.019806925455729166,
      "learning_rate": 0.0001,
      "loss": 4.6493,
      "loss/crossentropy": 2.2864162921905518,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22469403594732285,
      "step": 6858
    },
    {
      "epoch": 0.1372,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.018961588541666668,
      "learning_rate": 0.0001,
      "loss": 4.4017,
      "loss/crossentropy": 1.908643126487732,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22016740590333939,
      "step": 6860
    },
    {
      "epoch": 0.13724,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.010282389322916667,
      "learning_rate": 0.0001,
      "loss": 4.7255,
      "loss/crossentropy": 2.1028786301612854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23632919788360596,
      "step": 6862
    },
    {
      "epoch": 0.13728,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.01539306640625,
      "learning_rate": 0.0001,
      "loss": 4.4043,
      "loss/crossentropy": 2.0363592505455017,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22131157666444778,
      "step": 6864
    },
    {
      "epoch": 0.13732,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.0158843994140625,
      "learning_rate": 0.0001,
      "loss": 4.4357,
      "loss/crossentropy": 2.030495524406433,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.233994759619236,
      "step": 6866
    },
    {
      "epoch": 0.13736,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.018033854166666665,
      "learning_rate": 0.0001,
      "loss": 4.3036,
      "loss/crossentropy": 1.6365603804588318,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1909056007862091,
      "step": 6868
    },
    {
      "epoch": 0.1374,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.015217081705729166,
      "learning_rate": 0.0001,
      "loss": 4.655,
      "loss/crossentropy": 2.205111026763916,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2267644703388214,
      "step": 6870
    },
    {
      "epoch": 0.13744,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.016437784830729166,
      "learning_rate": 0.0001,
      "loss": 4.4207,
      "loss/crossentropy": 2.0179646015167236,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2307143434882164,
      "step": 6872
    },
    {
      "epoch": 0.13748,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.018290201822916668,
      "learning_rate": 0.0001,
      "loss": 4.2154,
      "loss/crossentropy": 1.9697463512420654,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2207925096154213,
      "step": 6874
    },
    {
      "epoch": 0.13752,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.019189453125,
      "learning_rate": 0.0001,
      "loss": 4.5691,
      "loss/crossentropy": 2.186875820159912,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25762687623500824,
      "step": 6876
    },
    {
      "epoch": 0.13756,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.017867024739583334,
      "learning_rate": 0.0001,
      "loss": 4.2713,
      "loss/crossentropy": 2.3203837871551514,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23470903187990189,
      "step": 6878
    },
    {
      "epoch": 0.1376,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0066640218098958336,
      "learning_rate": 0.0001,
      "loss": 4.4154,
      "loss/crossentropy": 2.2642472982406616,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2454553246498108,
      "step": 6880
    },
    {
      "epoch": 0.13764,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.008687337239583334,
      "learning_rate": 0.0001,
      "loss": 4.4076,
      "loss/crossentropy": 1.9313859343528748,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21797804534435272,
      "step": 6882
    },
    {
      "epoch": 0.13768,
      "grad_norm": 3.4375,
      "grad_norm_var": 0.11204020182291667,
      "learning_rate": 0.0001,
      "loss": 4.7846,
      "loss/crossentropy": 2.5469977855682373,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2651800066232681,
      "step": 6884
    },
    {
      "epoch": 0.13772,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.11030171712239584,
      "learning_rate": 0.0001,
      "loss": 4.4893,
      "loss/crossentropy": 2.549328088760376,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2566085457801819,
      "step": 6886
    },
    {
      "epoch": 0.13776,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.1126617431640625,
      "learning_rate": 0.0001,
      "loss": 4.4927,
      "loss/crossentropy": 2.3094369769096375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24709751456975937,
      "step": 6888
    },
    {
      "epoch": 0.1378,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.1097808837890625,
      "learning_rate": 0.0001,
      "loss": 4.2412,
      "loss/crossentropy": 1.5071046948432922,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19776180386543274,
      "step": 6890
    },
    {
      "epoch": 0.13784,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.1101226806640625,
      "learning_rate": 0.0001,
      "loss": 4.3705,
      "loss/crossentropy": 2.0064221620559692,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20836234837770462,
      "step": 6892
    },
    {
      "epoch": 0.13788,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.10816650390625,
      "learning_rate": 0.0001,
      "loss": 4.3608,
      "loss/crossentropy": 2.1216301321983337,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22151901572942734,
      "step": 6894
    },
    {
      "epoch": 0.13792,
      "grad_norm": 2.25,
      "grad_norm_var": 0.1073150634765625,
      "learning_rate": 0.0001,
      "loss": 4.4168,
      "loss/crossentropy": 1.8417679071426392,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21999332308769226,
      "step": 6896
    },
    {
      "epoch": 0.13796,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.10695699055989584,
      "learning_rate": 0.0001,
      "loss": 4.7005,
      "loss/crossentropy": 2.4651769399642944,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27029043436050415,
      "step": 6898
    },
    {
      "epoch": 0.138,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.014188639322916667,
      "learning_rate": 0.0001,
      "loss": 4.2985,
      "loss/crossentropy": 1.7225988507270813,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20930497348308563,
      "step": 6900
    },
    {
      "epoch": 0.13804,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.015360514322916666,
      "learning_rate": 0.0001,
      "loss": 4.1156,
      "loss/crossentropy": 2.1218297481536865,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21276423335075378,
      "step": 6902
    },
    {
      "epoch": 0.13808,
      "grad_norm": 2.25,
      "grad_norm_var": 0.010856119791666667,
      "learning_rate": 0.0001,
      "loss": 4.2706,
      "loss/crossentropy": 2.040019452571869,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2438269630074501,
      "step": 6904
    },
    {
      "epoch": 0.13812,
      "grad_norm": 15.8125,
      "grad_norm_var": 11.600536092122395,
      "learning_rate": 0.0001,
      "loss": 4.5041,
      "loss/crossentropy": 1.8229625225067139,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22197778522968292,
      "step": 6906
    },
    {
      "epoch": 0.13816,
      "grad_norm": 2.25,
      "grad_norm_var": 11.543973795572917,
      "learning_rate": 0.0001,
      "loss": 4.7087,
      "loss/crossentropy": 2.453408360481262,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24046239256858826,
      "step": 6908
    },
    {
      "epoch": 0.1382,
      "grad_norm": 2.046875,
      "grad_norm_var": 11.55152587890625,
      "learning_rate": 0.0001,
      "loss": 4.4806,
      "loss/crossentropy": 2.2724320888519287,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23739346861839294,
      "step": 6910
    },
    {
      "epoch": 0.13824,
      "grad_norm": 2.125,
      "grad_norm_var": 11.562272135416666,
      "learning_rate": 0.0001,
      "loss": 4.2747,
      "loss/crossentropy": 2.2382686138153076,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22692064195871353,
      "step": 6912
    },
    {
      "epoch": 0.13828,
      "grad_norm": 2.3125,
      "grad_norm_var": 11.54869384765625,
      "learning_rate": 0.0001,
      "loss": 4.6867,
      "loss/crossentropy": 2.021562337875366,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24746088683605194,
      "step": 6914
    },
    {
      "epoch": 0.13832,
      "grad_norm": 2.21875,
      "grad_norm_var": 11.546556599934895,
      "learning_rate": 0.0001,
      "loss": 4.3247,
      "loss/crossentropy": 2.1071943044662476,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2144002616405487,
      "step": 6916
    },
    {
      "epoch": 0.13836,
      "grad_norm": 2.28125,
      "grad_norm_var": 11.550846354166667,
      "learning_rate": 0.0001,
      "loss": 4.2686,
      "loss/crossentropy": 1.9641517400741577,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22584721446037292,
      "step": 6918
    },
    {
      "epoch": 0.1384,
      "grad_norm": 2.078125,
      "grad_norm_var": 11.559130859375,
      "learning_rate": 0.0001,
      "loss": 4.3194,
      "loss/crossentropy": 2.4430564641952515,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24351171404123306,
      "step": 6920
    },
    {
      "epoch": 0.13844,
      "grad_norm": 2.125,
      "grad_norm_var": 0.022484334309895833,
      "learning_rate": 0.0001,
      "loss": 4.4202,
      "loss/crossentropy": 2.2237725257873535,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2248261496424675,
      "step": 6922
    },
    {
      "epoch": 0.13848,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.015412394205729167,
      "learning_rate": 0.0001,
      "loss": 4.2028,
      "loss/crossentropy": 1.7291913628578186,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19805100560188293,
      "step": 6924
    },
    {
      "epoch": 0.13852,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.013923136393229167,
      "learning_rate": 0.0001,
      "loss": 4.3972,
      "loss/crossentropy": 1.807108223438263,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20386626571416855,
      "step": 6926
    },
    {
      "epoch": 0.13856,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.01422119140625,
      "learning_rate": 0.0001,
      "loss": 4.5188,
      "loss/crossentropy": 2.510676622390747,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24666880816221237,
      "step": 6928
    },
    {
      "epoch": 0.1386,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.007515462239583334,
      "learning_rate": 0.0001,
      "loss": 4.2006,
      "loss/crossentropy": 1.9420115947723389,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2214776575565338,
      "step": 6930
    },
    {
      "epoch": 0.13864,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.010837554931640625,
      "learning_rate": 0.0001,
      "loss": 4.4445,
      "loss/crossentropy": 2.2288190722465515,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23719585686922073,
      "step": 6932
    },
    {
      "epoch": 0.13868,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.011043294270833334,
      "learning_rate": 0.0001,
      "loss": 4.071,
      "loss/crossentropy": 2.04274183511734,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21653369069099426,
      "step": 6934
    },
    {
      "epoch": 0.13872,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.0121002197265625,
      "learning_rate": 0.0001,
      "loss": 4.4041,
      "loss/crossentropy": 1.9149779081344604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19941364973783493,
      "step": 6936
    },
    {
      "epoch": 0.13876,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.011554972330729166,
      "learning_rate": 0.0001,
      "loss": 4.2577,
      "loss/crossentropy": 1.7983179092407227,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18979590386152267,
      "step": 6938
    },
    {
      "epoch": 0.1388,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.023164876302083335,
      "learning_rate": 0.0001,
      "loss": 4.3314,
      "loss/crossentropy": 2.1919915080070496,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22652295976877213,
      "step": 6940
    },
    {
      "epoch": 0.13884,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.023152669270833332,
      "learning_rate": 0.0001,
      "loss": 4.494,
      "loss/crossentropy": 2.0362821221351624,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24296525120735168,
      "step": 6942
    },
    {
      "epoch": 0.13888,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.023653157552083335,
      "learning_rate": 0.0001,
      "loss": 4.4135,
      "loss/crossentropy": 2.0371538400650024,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2217850610613823,
      "step": 6944
    },
    {
      "epoch": 0.13892,
      "grad_norm": 2.5,
      "grad_norm_var": 0.0268310546875,
      "learning_rate": 0.0001,
      "loss": 4.3371,
      "loss/crossentropy": 1.9137988686561584,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2218162938952446,
      "step": 6946
    },
    {
      "epoch": 0.13896,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.03438898722330729,
      "learning_rate": 0.0001,
      "loss": 4.6521,
      "loss/crossentropy": 2.3215843439102173,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2576482892036438,
      "step": 6948
    },
    {
      "epoch": 0.139,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.026590983072916668,
      "learning_rate": 0.0001,
      "loss": 4.6004,
      "loss/crossentropy": 2.169154405593872,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24629274010658264,
      "step": 6950
    },
    {
      "epoch": 0.13904,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.03173421223958333,
      "learning_rate": 0.0001,
      "loss": 4.3549,
      "loss/crossentropy": 2.1355135440826416,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2316850870847702,
      "step": 6952
    },
    {
      "epoch": 0.13908,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.03183186848958333,
      "learning_rate": 0.0001,
      "loss": 4.3462,
      "loss/crossentropy": 2.264985144138336,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23467915505170822,
      "step": 6954
    },
    {
      "epoch": 0.13912,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.025926717122395835,
      "learning_rate": 0.0001,
      "loss": 4.6559,
      "loss/crossentropy": 2.1007654666900635,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2376948595046997,
      "step": 6956
    },
    {
      "epoch": 0.13916,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.02603759765625,
      "learning_rate": 0.0001,
      "loss": 4.4871,
      "loss/crossentropy": 2.284608840942383,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24811238050460815,
      "step": 6958
    },
    {
      "epoch": 0.1392,
      "grad_norm": 2.125,
      "grad_norm_var": 0.025911458333333335,
      "learning_rate": 0.0001,
      "loss": 4.2338,
      "loss/crossentropy": 1.657732367515564,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19541934877634048,
      "step": 6960
    },
    {
      "epoch": 0.13924,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.022391764322916667,
      "learning_rate": 0.0001,
      "loss": 4.3832,
      "loss/crossentropy": 1.9607325792312622,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2269211858510971,
      "step": 6962
    },
    {
      "epoch": 0.13928,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.006571451822916667,
      "learning_rate": 0.0001,
      "loss": 4.3759,
      "loss/crossentropy": 1.7454752326011658,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21780573576688766,
      "step": 6964
    },
    {
      "epoch": 0.13932,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.010350545247395834,
      "learning_rate": 0.0001,
      "loss": 4.7664,
      "loss/crossentropy": 2.001866638660431,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24929454922676086,
      "step": 6966
    },
    {
      "epoch": 0.13936,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.008006795247395834,
      "learning_rate": 0.0001,
      "loss": 4.4181,
      "loss/crossentropy": 1.9167855978012085,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22114858031272888,
      "step": 6968
    },
    {
      "epoch": 0.1394,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.008003743489583333,
      "learning_rate": 0.0001,
      "loss": 4.2284,
      "loss/crossentropy": 2.0324739813804626,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23271320760250092,
      "step": 6970
    },
    {
      "epoch": 0.13944,
      "grad_norm": 2.25,
      "grad_norm_var": 0.016307576497395834,
      "learning_rate": 0.0001,
      "loss": 4.5375,
      "loss/crossentropy": 2.162013590335846,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22832238674163818,
      "step": 6972
    },
    {
      "epoch": 0.13948,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.018944295247395833,
      "learning_rate": 0.0001,
      "loss": 4.1406,
      "loss/crossentropy": 2.074672818183899,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2402098923921585,
      "step": 6974
    },
    {
      "epoch": 0.13952,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.019758097330729165,
      "learning_rate": 0.0001,
      "loss": 4.4221,
      "loss/crossentropy": 1.9982299208641052,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21126113086938858,
      "step": 6976
    },
    {
      "epoch": 0.13956,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.021751912434895833,
      "learning_rate": 0.0001,
      "loss": 4.3391,
      "loss/crossentropy": 1.944049894809723,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2296794354915619,
      "step": 6978
    },
    {
      "epoch": 0.1396,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.02047119140625,
      "learning_rate": 0.0001,
      "loss": 4.4344,
      "loss/crossentropy": 2.308506488800049,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2589666247367859,
      "step": 6980
    },
    {
      "epoch": 0.13964,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.017724609375,
      "learning_rate": 0.0001,
      "loss": 4.2867,
      "loss/crossentropy": 2.129163682460785,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21905581653118134,
      "step": 6982
    },
    {
      "epoch": 0.13968,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.017902628580729166,
      "learning_rate": 0.0001,
      "loss": 4.3023,
      "loss/crossentropy": 1.8560669422149658,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2119361013174057,
      "step": 6984
    },
    {
      "epoch": 0.13972,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.018602498372395835,
      "learning_rate": 0.0001,
      "loss": 4.1212,
      "loss/crossentropy": 1.8194095492362976,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20872193574905396,
      "step": 6986
    },
    {
      "epoch": 0.13976,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.005501302083333334,
      "learning_rate": 0.0001,
      "loss": 4.2612,
      "loss/crossentropy": 2.0200153589248657,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20160746574401855,
      "step": 6988
    },
    {
      "epoch": 0.1398,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.007298787434895833,
      "learning_rate": 0.0001,
      "loss": 4.2757,
      "loss/crossentropy": 1.982479751110077,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23396103084087372,
      "step": 6990
    },
    {
      "epoch": 0.13984,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.006494140625,
      "learning_rate": 0.0001,
      "loss": 4.5075,
      "loss/crossentropy": 2.17054283618927,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23827539384365082,
      "step": 6992
    },
    {
      "epoch": 0.13988,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.0056640625,
      "learning_rate": 0.0001,
      "loss": 4.1794,
      "loss/crossentropy": 1.619499921798706,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1917721927165985,
      "step": 6994
    },
    {
      "epoch": 0.13992,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.005353800455729167,
      "learning_rate": 0.0001,
      "loss": 4.3833,
      "loss/crossentropy": 2.1082500219345093,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23768241703510284,
      "step": 6996
    },
    {
      "epoch": 0.13996,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.004076131184895833,
      "learning_rate": 0.0001,
      "loss": 4.6731,
      "loss/crossentropy": 1.8480825424194336,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21046122163534164,
      "step": 6998
    },
    {
      "epoch": 0.14,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.008463541666666666,
      "learning_rate": 0.0001,
      "loss": 4.5285,
      "loss/crossentropy": 2.0547631978988647,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22980494797229767,
      "step": 7000
    },
    {
      "epoch": 0.14004,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.007835896809895833,
      "learning_rate": 0.0001,
      "loss": 4.4625,
      "loss/crossentropy": 2.0695141553878784,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2100546732544899,
      "step": 7002
    },
    {
      "epoch": 0.14008,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.012277984619140625,
      "learning_rate": 0.0001,
      "loss": 4.3716,
      "loss/crossentropy": 2.105263113975525,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24552470445632935,
      "step": 7004
    },
    {
      "epoch": 0.14012,
      "grad_norm": 2.375,
      "grad_norm_var": 0.014427693684895833,
      "learning_rate": 0.0001,
      "loss": 4.3566,
      "loss/crossentropy": 2.03000670671463,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24780434370040894,
      "step": 7006
    },
    {
      "epoch": 0.14016,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.0173980712890625,
      "learning_rate": 0.0001,
      "loss": 4.4758,
      "loss/crossentropy": 2.288944959640503,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2760937511920929,
      "step": 7008
    },
    {
      "epoch": 0.1402,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.06544570922851563,
      "learning_rate": 0.0001,
      "loss": 3.9326,
      "loss/crossentropy": 1.790147304534912,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1905786320567131,
      "step": 7010
    },
    {
      "epoch": 0.14024,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.07765884399414062,
      "learning_rate": 0.0001,
      "loss": 4.6364,
      "loss/crossentropy": 2.008346378803253,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23138166218996048,
      "step": 7012
    },
    {
      "epoch": 0.14028,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.07974014282226563,
      "learning_rate": 0.0001,
      "loss": 4.2304,
      "loss/crossentropy": 1.9694496393203735,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21347863972187042,
      "step": 7014
    },
    {
      "epoch": 0.14032,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.07948989868164062,
      "learning_rate": 0.0001,
      "loss": 4.3685,
      "loss/crossentropy": 2.0907286405563354,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23349857330322266,
      "step": 7016
    },
    {
      "epoch": 0.14036,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.07850316365559896,
      "learning_rate": 0.0001,
      "loss": 4.6454,
      "loss/crossentropy": 2.161414623260498,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23841089010238647,
      "step": 7018
    },
    {
      "epoch": 0.1404,
      "grad_norm": 2.25,
      "grad_norm_var": 0.07315266927083333,
      "learning_rate": 0.0001,
      "loss": 4.4868,
      "loss/crossentropy": 2.1402887105941772,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25493840873241425,
      "step": 7020
    },
    {
      "epoch": 0.14044,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.06809666951497396,
      "learning_rate": 0.0001,
      "loss": 4.1248,
      "loss/crossentropy": 2.0703811049461365,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24300381541252136,
      "step": 7022
    },
    {
      "epoch": 0.14048,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.0716875712076823,
      "learning_rate": 0.0001,
      "loss": 4.6165,
      "loss/crossentropy": 2.152569532394409,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23347805440425873,
      "step": 7024
    },
    {
      "epoch": 0.14052,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.03453369140625,
      "learning_rate": 0.0001,
      "loss": 4.1642,
      "loss/crossentropy": 2.0831095576286316,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2198955938220024,
      "step": 7026
    },
    {
      "epoch": 0.14056,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.019498697916666665,
      "learning_rate": 0.0001,
      "loss": 4.4929,
      "loss/crossentropy": 2.1631508469581604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23971489816904068,
      "step": 7028
    },
    {
      "epoch": 0.1406,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.019481404622395834,
      "learning_rate": 0.0001,
      "loss": 4.4687,
      "loss/crossentropy": 2.1683043241500854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2628757208585739,
      "step": 7030
    },
    {
      "epoch": 0.14064,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.07377827962239583,
      "learning_rate": 0.0001,
      "loss": 4.522,
      "loss/crossentropy": 2.021001398563385,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22922180593013763,
      "step": 7032
    },
    {
      "epoch": 0.14068,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.0725494384765625,
      "learning_rate": 0.0001,
      "loss": 4.7729,
      "loss/crossentropy": 2.267430543899536,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27525072544813156,
      "step": 7034
    },
    {
      "epoch": 0.14072,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0788726806640625,
      "learning_rate": 0.0001,
      "loss": 4.3118,
      "loss/crossentropy": 2.066729426383972,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21959475427865982,
      "step": 7036
    },
    {
      "epoch": 0.14076,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.08162333170572916,
      "learning_rate": 0.0001,
      "loss": 4.3203,
      "loss/crossentropy": 1.7972697019577026,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21773608028888702,
      "step": 7038
    },
    {
      "epoch": 0.1408,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.08068745930989583,
      "learning_rate": 0.0001,
      "loss": 4.0745,
      "loss/crossentropy": 1.751904845237732,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19929176568984985,
      "step": 7040
    },
    {
      "epoch": 0.14084,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.07649332682291667,
      "learning_rate": 0.0001,
      "loss": 4.4188,
      "loss/crossentropy": 1.8432873487472534,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21235650032758713,
      "step": 7042
    },
    {
      "epoch": 0.14088,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.07618815104166667,
      "learning_rate": 0.0001,
      "loss": 4.2343,
      "loss/crossentropy": 1.9589285850524902,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21506928652524948,
      "step": 7044
    },
    {
      "epoch": 0.14092,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0788726806640625,
      "learning_rate": 0.0001,
      "loss": 4.229,
      "loss/crossentropy": 2.3658028841018677,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23405643552541733,
      "step": 7046
    },
    {
      "epoch": 0.14096,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.008837890625,
      "learning_rate": 0.0001,
      "loss": 4.3922,
      "loss/crossentropy": 2.088135540485382,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22823208570480347,
      "step": 7048
    },
    {
      "epoch": 0.141,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.006810506184895833,
      "learning_rate": 0.0001,
      "loss": 4.2367,
      "loss/crossentropy": 1.9309821724891663,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21409911662340164,
      "step": 7050
    },
    {
      "epoch": 0.14104,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.006636555989583333,
      "learning_rate": 0.0001,
      "loss": 4.5237,
      "loss/crossentropy": 2.5411492586135864,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25017087161540985,
      "step": 7052
    },
    {
      "epoch": 0.14108,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.007352701822916667,
      "learning_rate": 0.0001,
      "loss": 4.4998,
      "loss/crossentropy": 2.3210322856903076,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23967693746089935,
      "step": 7054
    },
    {
      "epoch": 0.14112,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.009370930989583333,
      "learning_rate": 0.0001,
      "loss": 4.4607,
      "loss/crossentropy": 2.054674744606018,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23270255327224731,
      "step": 7056
    },
    {
      "epoch": 0.14116,
      "grad_norm": 2.125,
      "grad_norm_var": 0.008968098958333334,
      "learning_rate": 0.0001,
      "loss": 4.5423,
      "loss/crossentropy": 2.545789122581482,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25560182332992554,
      "step": 7058
    },
    {
      "epoch": 0.1412,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.008861287434895834,
      "learning_rate": 0.0001,
      "loss": 4.2682,
      "loss/crossentropy": 2.262348175048828,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24763159453868866,
      "step": 7060
    },
    {
      "epoch": 0.14124,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.008447265625,
      "learning_rate": 0.0001,
      "loss": 4.618,
      "loss/crossentropy": 2.1045475602149963,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2472379505634308,
      "step": 7062
    },
    {
      "epoch": 0.14128,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0093170166015625,
      "learning_rate": 0.0001,
      "loss": 4.2312,
      "loss/crossentropy": 1.5632115006446838,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19010942429304123,
      "step": 7064
    },
    {
      "epoch": 0.14132,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.0116851806640625,
      "learning_rate": 0.0001,
      "loss": 4.2638,
      "loss/crossentropy": 2.0847875475883484,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21371345967054367,
      "step": 7066
    },
    {
      "epoch": 0.14136,
      "grad_norm": 2.25,
      "grad_norm_var": 0.018016560872395834,
      "learning_rate": 0.0001,
      "loss": 4.5171,
      "loss/crossentropy": 2.2243804931640625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23065787553787231,
      "step": 7068
    },
    {
      "epoch": 0.1414,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.016927083333333332,
      "learning_rate": 0.0001,
      "loss": 4.2812,
      "loss/crossentropy": 1.9477753639221191,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21616356819868088,
      "step": 7070
    },
    {
      "epoch": 0.14144,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.024169921875,
      "learning_rate": 0.0001,
      "loss": 4.7005,
      "loss/crossentropy": 2.2598072290420532,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25795431435108185,
      "step": 7072
    },
    {
      "epoch": 0.14148,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.022652180989583333,
      "learning_rate": 0.0001,
      "loss": 4.4345,
      "loss/crossentropy": 1.8817242980003357,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21974974125623703,
      "step": 7074
    },
    {
      "epoch": 0.14152,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0256744384765625,
      "learning_rate": 0.0001,
      "loss": 4.5688,
      "loss/crossentropy": 2.5275847911834717,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24608048796653748,
      "step": 7076
    },
    {
      "epoch": 0.14156,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0265045166015625,
      "learning_rate": 0.0001,
      "loss": 4.3937,
      "loss/crossentropy": 2.400865852832794,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22960034757852554,
      "step": 7078
    },
    {
      "epoch": 0.1416,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0237457275390625,
      "learning_rate": 0.0001,
      "loss": 4.4005,
      "loss/crossentropy": 1.908901333808899,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22271078824996948,
      "step": 7080
    },
    {
      "epoch": 0.14164,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.02213134765625,
      "learning_rate": 0.0001,
      "loss": 4.1384,
      "loss/crossentropy": 2.3330780267715454,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23944097012281418,
      "step": 7082
    },
    {
      "epoch": 0.14168,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.018941243489583332,
      "learning_rate": 0.0001,
      "loss": 4.3516,
      "loss/crossentropy": 2.332213521003723,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23576530069112778,
      "step": 7084
    },
    {
      "epoch": 0.14172,
      "grad_norm": 2.125,
      "grad_norm_var": 0.022435506184895832,
      "learning_rate": 0.0001,
      "loss": 4.4654,
      "loss/crossentropy": 2.2269067764282227,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25341375917196274,
      "step": 7086
    },
    {
      "epoch": 0.14176,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.01470947265625,
      "learning_rate": 0.0001,
      "loss": 4.2491,
      "loss/crossentropy": 2.461983561515808,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24491792172193527,
      "step": 7088
    },
    {
      "epoch": 0.1418,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.013952382405598958,
      "learning_rate": 0.0001,
      "loss": 4.2348,
      "loss/crossentropy": 2.428719997406006,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2394469603896141,
      "step": 7090
    },
    {
      "epoch": 0.14184,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.011433664957682292,
      "learning_rate": 0.0001,
      "loss": 4.2242,
      "loss/crossentropy": 2.32351291179657,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23441501706838608,
      "step": 7092
    },
    {
      "epoch": 0.14188,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.011482493082682291,
      "learning_rate": 0.0001,
      "loss": 4.4155,
      "loss/crossentropy": 2.1165764331817627,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2350979596376419,
      "step": 7094
    },
    {
      "epoch": 0.14192,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.014288075764973958,
      "learning_rate": 0.0001,
      "loss": 4.4952,
      "loss/crossentropy": 2.150681734085083,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2475891336798668,
      "step": 7096
    },
    {
      "epoch": 0.14196,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.013079579671223958,
      "learning_rate": 0.0001,
      "loss": 4.2753,
      "loss/crossentropy": 2.038177013397217,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2293965071439743,
      "step": 7098
    },
    {
      "epoch": 0.142,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.014062245686848959,
      "learning_rate": 0.0001,
      "loss": 4.298,
      "loss/crossentropy": 1.899521827697754,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20856370776891708,
      "step": 7100
    },
    {
      "epoch": 0.14204,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.010593414306640625,
      "learning_rate": 0.0001,
      "loss": 4.3449,
      "loss/crossentropy": 1.9807924032211304,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22756796330213547,
      "step": 7102
    },
    {
      "epoch": 0.14208,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009421539306640626,
      "learning_rate": 0.0001,
      "loss": 4.2893,
      "loss/crossentropy": 2.158667206764221,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22594892233610153,
      "step": 7104
    },
    {
      "epoch": 0.14212,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.007811482747395833,
      "learning_rate": 0.0001,
      "loss": 4.3751,
      "loss/crossentropy": 2.3133270144462585,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22925584018230438,
      "step": 7106
    },
    {
      "epoch": 0.14216,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010107421875,
      "learning_rate": 0.0001,
      "loss": 4.2762,
      "loss/crossentropy": 1.9796301126480103,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21650104224681854,
      "step": 7108
    },
    {
      "epoch": 0.1422,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.010054524739583333,
      "learning_rate": 0.0001,
      "loss": 4.2778,
      "loss/crossentropy": 2.092659056186676,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23407263308763504,
      "step": 7110
    },
    {
      "epoch": 0.14224,
      "grad_norm": 2.125,
      "grad_norm_var": 0.009056599934895833,
      "learning_rate": 0.0001,
      "loss": 4.6078,
      "loss/crossentropy": 1.970819890499115,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2262839823961258,
      "step": 7112
    },
    {
      "epoch": 0.14228,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.011923980712890626,
      "learning_rate": 0.0001,
      "loss": 4.0642,
      "loss/crossentropy": 1.5877107381820679,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.182430237531662,
      "step": 7114
    },
    {
      "epoch": 0.14232,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.020336659749348958,
      "learning_rate": 0.0001,
      "loss": 4.6997,
      "loss/crossentropy": 2.3208755254745483,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25018931180238724,
      "step": 7116
    },
    {
      "epoch": 0.14236,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.01904271443684896,
      "learning_rate": 0.0001,
      "loss": 4.4483,
      "loss/crossentropy": 2.1348973512649536,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23516173660755157,
      "step": 7118
    },
    {
      "epoch": 0.1424,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.02533543904622396,
      "learning_rate": 0.0001,
      "loss": 4.3443,
      "loss/crossentropy": 2.1442995071411133,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23094037175178528,
      "step": 7120
    },
    {
      "epoch": 0.14244,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.025608062744140625,
      "learning_rate": 0.0001,
      "loss": 4.3297,
      "loss/crossentropy": 2.24001145362854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2479296177625656,
      "step": 7122
    },
    {
      "epoch": 0.14248,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.038917795817057295,
      "learning_rate": 0.0001,
      "loss": 4.4122,
      "loss/crossentropy": 1.8284733891487122,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23670874536037445,
      "step": 7124
    },
    {
      "epoch": 0.14252,
      "grad_norm": 2.375,
      "grad_norm_var": 0.04146499633789062,
      "learning_rate": 0.0001,
      "loss": 4.6223,
      "loss/crossentropy": 2.1003533601760864,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24235840141773224,
      "step": 7126
    },
    {
      "epoch": 0.14256,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.04201024373372396,
      "learning_rate": 0.0001,
      "loss": 4.2591,
      "loss/crossentropy": 2.3661316633224487,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.258208692073822,
      "step": 7128
    },
    {
      "epoch": 0.1426,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.030989583333333334,
      "learning_rate": 0.0001,
      "loss": 4.3288,
      "loss/crossentropy": 2.2374593019485474,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2543400973081589,
      "step": 7130
    },
    {
      "epoch": 0.14264,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.02945556640625,
      "learning_rate": 0.0001,
      "loss": 4.5939,
      "loss/crossentropy": 1.9141342639923096,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23257911205291748,
      "step": 7132
    },
    {
      "epoch": 0.14268,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.031224568684895832,
      "learning_rate": 0.0001,
      "loss": 4.1893,
      "loss/crossentropy": 1.992666780948639,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2298717424273491,
      "step": 7134
    },
    {
      "epoch": 0.14272,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.033503214518229164,
      "learning_rate": 0.0001,
      "loss": 4.2665,
      "loss/crossentropy": 1.9794283509254456,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2112378552556038,
      "step": 7136
    },
    {
      "epoch": 0.14276,
      "grad_norm": 2.5,
      "grad_norm_var": 0.0382232666015625,
      "learning_rate": 0.0001,
      "loss": 4.3801,
      "loss/crossentropy": 2.1011139154434204,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27278490364551544,
      "step": 7138
    },
    {
      "epoch": 0.1428,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.02340087890625,
      "learning_rate": 0.0001,
      "loss": 4.6989,
      "loss/crossentropy": 2.3489880561828613,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24445360153913498,
      "step": 7140
    },
    {
      "epoch": 0.14284,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.0173736572265625,
      "learning_rate": 0.0001,
      "loss": 4.3418,
      "loss/crossentropy": 2.011172831058502,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.210151307284832,
      "step": 7142
    },
    {
      "epoch": 0.14288,
      "grad_norm": 2.25,
      "grad_norm_var": 0.017366536458333335,
      "learning_rate": 0.0001,
      "loss": 4.3488,
      "loss/crossentropy": 1.963642418384552,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22939135879278183,
      "step": 7144
    },
    {
      "epoch": 0.14292,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.016194661458333332,
      "learning_rate": 0.0001,
      "loss": 4.5166,
      "loss/crossentropy": 2.2739341259002686,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23235367238521576,
      "step": 7146
    },
    {
      "epoch": 0.14296,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.018973795572916667,
      "learning_rate": 0.0001,
      "loss": 4.5977,
      "loss/crossentropy": 2.282576322555542,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24144183099269867,
      "step": 7148
    },
    {
      "epoch": 0.143,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.02329279581705729,
      "learning_rate": 0.0001,
      "loss": 4.141,
      "loss/crossentropy": 1.7847901582717896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19184929877519608,
      "step": 7150
    },
    {
      "epoch": 0.14304,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.021740468343098958,
      "learning_rate": 0.0001,
      "loss": 4.3379,
      "loss/crossentropy": 2.165170907974243,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.223430335521698,
      "step": 7152
    },
    {
      "epoch": 0.14308,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.018873850504557293,
      "learning_rate": 0.0001,
      "loss": 4.3407,
      "loss/crossentropy": 2.0395787954330444,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21662698686122894,
      "step": 7154
    },
    {
      "epoch": 0.14312,
      "grad_norm": 2.125,
      "grad_norm_var": 0.01907323201497396,
      "learning_rate": 0.0001,
      "loss": 4.4936,
      "loss/crossentropy": 2.014316141605377,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23184175789356232,
      "step": 7156
    },
    {
      "epoch": 0.14316,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.019419097900390626,
      "learning_rate": 0.0001,
      "loss": 4.5612,
      "loss/crossentropy": 2.2581117153167725,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24055174738168716,
      "step": 7158
    },
    {
      "epoch": 0.1432,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.021022288004557292,
      "learning_rate": 0.0001,
      "loss": 4.3853,
      "loss/crossentropy": 2.0905630588531494,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2719188630580902,
      "step": 7160
    },
    {
      "epoch": 0.14324,
      "grad_norm": 2.25,
      "grad_norm_var": 0.025233713785807292,
      "learning_rate": 0.0001,
      "loss": 4.6485,
      "loss/crossentropy": 2.414529800415039,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24188701063394547,
      "step": 7162
    },
    {
      "epoch": 0.14328,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.02240778605143229,
      "learning_rate": 0.0001,
      "loss": 4.3605,
      "loss/crossentropy": 2.028432607650757,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2300800457596779,
      "step": 7164
    },
    {
      "epoch": 0.14332,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.03299153645833333,
      "learning_rate": 0.0001,
      "loss": 4.2947,
      "loss/crossentropy": 2.096144199371338,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23311930894851685,
      "step": 7166
    },
    {
      "epoch": 0.14336,
      "grad_norm": 2.25,
      "grad_norm_var": 0.029622395833333332,
      "learning_rate": 0.0001,
      "loss": 4.4375,
      "loss/crossentropy": 2.259281277656555,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25558799505233765,
      "step": 7168
    },
    {
      "epoch": 0.1434,
      "grad_norm": 2.25,
      "grad_norm_var": 0.0219390869140625,
      "learning_rate": 0.0001,
      "loss": 4.4364,
      "loss/crossentropy": 2.0766254663467407,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22725434601306915,
      "step": 7170
    },
    {
      "epoch": 0.14344,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.024312337239583332,
      "learning_rate": 0.0001,
      "loss": 4.4695,
      "loss/crossentropy": 2.26702618598938,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22584324330091476,
      "step": 7172
    },
    {
      "epoch": 0.14348,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.027534993489583333,
      "learning_rate": 0.0001,
      "loss": 4.3417,
      "loss/crossentropy": 2.1933096647262573,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23658160120248795,
      "step": 7174
    },
    {
      "epoch": 0.14352,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.027372233072916665,
      "learning_rate": 0.0001,
      "loss": 4.351,
      "loss/crossentropy": 2.2003660202026367,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22349942475557327,
      "step": 7176
    },
    {
      "epoch": 0.14356,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.025145467122395834,
      "learning_rate": 0.0001,
      "loss": 4.684,
      "loss/crossentropy": 2.4630067348480225,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.253003865480423,
      "step": 7178
    },
    {
      "epoch": 0.1436,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.027082316080729165,
      "learning_rate": 0.0001,
      "loss": 4.6875,
      "loss/crossentropy": 2.264480948448181,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27604997158050537,
      "step": 7180
    },
    {
      "epoch": 0.14364,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.00712890625,
      "learning_rate": 0.0001,
      "loss": 4.3064,
      "loss/crossentropy": 2.1641955375671387,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22587595880031586,
      "step": 7182
    },
    {
      "epoch": 0.14368,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.00943603515625,
      "learning_rate": 0.0001,
      "loss": 4.6137,
      "loss/crossentropy": 2.1432350873947144,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24324018508195877,
      "step": 7184
    },
    {
      "epoch": 0.14372,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.014611562093098959,
      "learning_rate": 0.0001,
      "loss": 4.3394,
      "loss/crossentropy": 1.7448238134384155,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1751260682940483,
      "step": 7186
    },
    {
      "epoch": 0.14376,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.015295155843098958,
      "learning_rate": 0.0001,
      "loss": 4.3602,
      "loss/crossentropy": 2.3202184438705444,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2314094603061676,
      "step": 7188
    },
    {
      "epoch": 0.1438,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.019681549072265624,
      "learning_rate": 0.0001,
      "loss": 4.187,
      "loss/crossentropy": 1.970094919204712,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24998464435338974,
      "step": 7190
    },
    {
      "epoch": 0.14384,
      "grad_norm": 2.125,
      "grad_norm_var": 0.02075780232747396,
      "learning_rate": 0.0001,
      "loss": 4.215,
      "loss/crossentropy": 2.1331114768981934,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2008385732769966,
      "step": 7192
    },
    {
      "epoch": 0.14388,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.021345774332682293,
      "learning_rate": 0.0001,
      "loss": 4.3976,
      "loss/crossentropy": 2.1659106016159058,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22314336150884628,
      "step": 7194
    },
    {
      "epoch": 0.14392,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.018507639567057293,
      "learning_rate": 0.0001,
      "loss": 4.6324,
      "loss/crossentropy": 2.3382883071899414,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26393643021583557,
      "step": 7196
    },
    {
      "epoch": 0.14396,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.020499420166015626,
      "learning_rate": 0.0001,
      "loss": 4.7988,
      "loss/crossentropy": 2.1325554847717285,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2457222416996956,
      "step": 7198
    },
    {
      "epoch": 0.144,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.03144709269205729,
      "learning_rate": 0.0001,
      "loss": 4.3175,
      "loss/crossentropy": 1.7927106022834778,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20284093916416168,
      "step": 7200
    },
    {
      "epoch": 0.14404,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.03050715128580729,
      "learning_rate": 0.0001,
      "loss": 4.1676,
      "loss/crossentropy": 1.9799031615257263,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23475942015647888,
      "step": 7202
    },
    {
      "epoch": 0.14408,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.02939020792643229,
      "learning_rate": 0.0001,
      "loss": 4.2627,
      "loss/crossentropy": 2.0590370893478394,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2160736471414566,
      "step": 7204
    },
    {
      "epoch": 0.14412,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.02800267537434896,
      "learning_rate": 0.0001,
      "loss": 4.4397,
      "loss/crossentropy": 1.9866149425506592,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23413674533367157,
      "step": 7206
    },
    {
      "epoch": 0.14416,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.03551610310872396,
      "learning_rate": 0.0001,
      "loss": 4.526,
      "loss/crossentropy": 2.1320748925209045,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24469739198684692,
      "step": 7208
    },
    {
      "epoch": 0.1442,
      "grad_norm": 2.375,
      "grad_norm_var": 0.034708404541015626,
      "learning_rate": 0.0001,
      "loss": 4.5834,
      "loss/crossentropy": 2.225857973098755,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23838083446025848,
      "step": 7210
    },
    {
      "epoch": 0.14424,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.03794733683268229,
      "learning_rate": 0.0001,
      "loss": 3.9952,
      "loss/crossentropy": 1.9118528962135315,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2064301297068596,
      "step": 7212
    },
    {
      "epoch": 0.14428,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.03806940714518229,
      "learning_rate": 0.0001,
      "loss": 4.182,
      "loss/crossentropy": 1.8142234086990356,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2172461450099945,
      "step": 7214
    },
    {
      "epoch": 0.14432,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.02540868123372396,
      "learning_rate": 0.0001,
      "loss": 4.4446,
      "loss/crossentropy": 1.9308255910873413,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2236044555902481,
      "step": 7216
    },
    {
      "epoch": 0.14436,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.022391764322916667,
      "learning_rate": 0.0001,
      "loss": 4.4597,
      "loss/crossentropy": 1.9821211695671082,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21110422909259796,
      "step": 7218
    },
    {
      "epoch": 0.1444,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.023176066080729165,
      "learning_rate": 0.0001,
      "loss": 4.6732,
      "loss/crossentropy": 2.216045379638672,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23433538526296616,
      "step": 7220
    },
    {
      "epoch": 0.14444,
      "grad_norm": 2.5,
      "grad_norm_var": 0.024442545572916665,
      "learning_rate": 0.0001,
      "loss": 4.6089,
      "loss/crossentropy": 2.2303662300109863,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2390453889966011,
      "step": 7222
    },
    {
      "epoch": 0.14448,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.017020670572916667,
      "learning_rate": 0.0001,
      "loss": 4.0805,
      "loss/crossentropy": 2.2152082920074463,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22196897864341736,
      "step": 7224
    },
    {
      "epoch": 0.14452,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.014860026041666667,
      "learning_rate": 0.0001,
      "loss": 4.2843,
      "loss/crossentropy": 2.134513795375824,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21964208781719208,
      "step": 7226
    },
    {
      "epoch": 0.14456,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.015262858072916666,
      "learning_rate": 0.0001,
      "loss": 4.0289,
      "loss/crossentropy": 1.6803861260414124,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18217483162879944,
      "step": 7228
    },
    {
      "epoch": 0.1446,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0152252197265625,
      "learning_rate": 0.0001,
      "loss": 4.1353,
      "loss/crossentropy": 1.6597792506217957,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20317095518112183,
      "step": 7230
    },
    {
      "epoch": 0.14464,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.017704264322916666,
      "learning_rate": 0.0001,
      "loss": 4.2978,
      "loss/crossentropy": 1.776586651802063,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21835225820541382,
      "step": 7232
    },
    {
      "epoch": 0.14468,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.017154947916666666,
      "learning_rate": 0.0001,
      "loss": 4.1092,
      "loss/crossentropy": 1.7347259521484375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21964067220687866,
      "step": 7234
    },
    {
      "epoch": 0.14472,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.015380859375,
      "learning_rate": 0.0001,
      "loss": 4.2069,
      "loss/crossentropy": 1.79097181558609,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21776803582906723,
      "step": 7236
    },
    {
      "epoch": 0.14476,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0069244384765625,
      "learning_rate": 0.0001,
      "loss": 4.0335,
      "loss/crossentropy": 2.051329553127289,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21669812500476837,
      "step": 7238
    },
    {
      "epoch": 0.1448,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.0070220947265625,
      "learning_rate": 0.0001,
      "loss": 4.3206,
      "loss/crossentropy": 1.965324580669403,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2283879667520523,
      "step": 7240
    },
    {
      "epoch": 0.14484,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.010692342122395834,
      "learning_rate": 0.0001,
      "loss": 4.5952,
      "loss/crossentropy": 2.248784363269806,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24906039983034134,
      "step": 7242
    },
    {
      "epoch": 0.14488,
      "grad_norm": 2.125,
      "grad_norm_var": 0.010529581705729167,
      "learning_rate": 0.0001,
      "loss": 4.3321,
      "loss/crossentropy": 1.9946890473365784,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22583268582820892,
      "step": 7244
    },
    {
      "epoch": 0.14492,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.010660807291666666,
      "learning_rate": 0.0001,
      "loss": 4.1436,
      "loss/crossentropy": 2.2306413650512695,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22068945318460464,
      "step": 7246
    },
    {
      "epoch": 0.14496,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.00758056640625,
      "learning_rate": 0.0001,
      "loss": 4.2159,
      "loss/crossentropy": 2.110253095626831,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22045104205608368,
      "step": 7248
    },
    {
      "epoch": 0.145,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.008210245768229167,
      "learning_rate": 0.0001,
      "loss": 4.0693,
      "loss/crossentropy": 1.928157925605774,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2043258175253868,
      "step": 7250
    },
    {
      "epoch": 0.14504,
      "grad_norm": 2.0,
      "grad_norm_var": 0.009110514322916667,
      "learning_rate": 0.0001,
      "loss": 4.2458,
      "loss/crossentropy": 2.2674691677093506,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23258862644433975,
      "step": 7252
    },
    {
      "epoch": 0.14508,
      "grad_norm": 2.765625,
      "grad_norm_var": 0.03052978515625,
      "learning_rate": 0.0001,
      "loss": 4.5107,
      "loss/crossentropy": 2.2825024127960205,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.263367660343647,
      "step": 7254
    },
    {
      "epoch": 0.14512,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.040185546875,
      "learning_rate": 0.0001,
      "loss": 4.2394,
      "loss/crossentropy": 2.1546168327331543,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23073332011699677,
      "step": 7256
    },
    {
      "epoch": 0.14516,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.040816243489583334,
      "learning_rate": 0.0001,
      "loss": 4.5504,
      "loss/crossentropy": 2.0490044951438904,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24161820113658905,
      "step": 7258
    },
    {
      "epoch": 0.1452,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.04038798014322917,
      "learning_rate": 0.0001,
      "loss": 4.6468,
      "loss/crossentropy": 2.115446150302887,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2537624090909958,
      "step": 7260
    },
    {
      "epoch": 0.14524,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.22388407389322917,
      "learning_rate": 0.0001,
      "loss": 4.1457,
      "loss/crossentropy": 2.0302165746688843,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21026766300201416,
      "step": 7262
    },
    {
      "epoch": 0.14528,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.2228179931640625,
      "learning_rate": 0.0001,
      "loss": 4.4077,
      "loss/crossentropy": 2.102527379989624,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2354147508740425,
      "step": 7264
    },
    {
      "epoch": 0.14532,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.21614176432291668,
      "learning_rate": 0.0001,
      "loss": 4.1108,
      "loss/crossentropy": 2.0095282793045044,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22679834067821503,
      "step": 7266
    },
    {
      "epoch": 0.14536,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.20706278483072918,
      "learning_rate": 0.0001,
      "loss": 4.3849,
      "loss/crossentropy": 1.8988603353500366,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21238050609827042,
      "step": 7268
    },
    {
      "epoch": 0.1454,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.19975484212239583,
      "learning_rate": 0.0001,
      "loss": 4.6614,
      "loss/crossentropy": 2.186660885810852,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24453039467334747,
      "step": 7270
    },
    {
      "epoch": 0.14544,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.20244852701822916,
      "learning_rate": 0.0001,
      "loss": 4.1409,
      "loss/crossentropy": 1.8927155137062073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2147793024778366,
      "step": 7272
    },
    {
      "epoch": 0.14548,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.20608317057291667,
      "learning_rate": 0.0001,
      "loss": 4.1375,
      "loss/crossentropy": 1.8969642519950867,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21910040825605392,
      "step": 7274
    },
    {
      "epoch": 0.14552,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.20545247395833333,
      "learning_rate": 0.0001,
      "loss": 4.3325,
      "loss/crossentropy": 2.090053617954254,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22576116025447845,
      "step": 7276
    },
    {
      "epoch": 0.14556,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.010480753580729167,
      "learning_rate": 0.0001,
      "loss": 4.7263,
      "loss/crossentropy": 2.1606650352478027,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22937766462564468,
      "step": 7278
    },
    {
      "epoch": 0.1456,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.013451131184895833,
      "learning_rate": 0.0001,
      "loss": 4.0924,
      "loss/crossentropy": 1.9946333765983582,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20145532488822937,
      "step": 7280
    },
    {
      "epoch": 0.14564,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.013834635416666666,
      "learning_rate": 0.0001,
      "loss": 4.6519,
      "loss/crossentropy": 2.0958545207977295,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22669509798288345,
      "step": 7282
    },
    {
      "epoch": 0.14568,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.016109212239583334,
      "learning_rate": 0.0001,
      "loss": 4.4224,
      "loss/crossentropy": 2.0515894889831543,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2520884945988655,
      "step": 7284
    },
    {
      "epoch": 0.14572,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.016063435872395834,
      "learning_rate": 0.0001,
      "loss": 4.1572,
      "loss/crossentropy": 2.034587264060974,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21523287147283554,
      "step": 7286
    },
    {
      "epoch": 0.14576,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.016080729166666665,
      "learning_rate": 0.0001,
      "loss": 3.8649,
      "loss/crossentropy": 1.6578314900398254,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19603776931762695,
      "step": 7288
    },
    {
      "epoch": 0.1458,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.014167277018229167,
      "learning_rate": 0.0001,
      "loss": 4.2443,
      "loss/crossentropy": 2.0019100308418274,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22880420833826065,
      "step": 7290
    },
    {
      "epoch": 0.14584,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.018723297119140624,
      "learning_rate": 0.0001,
      "loss": 3.981,
      "loss/crossentropy": 2.068517565727234,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19675085693597794,
      "step": 7292
    },
    {
      "epoch": 0.14588,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.013734690348307292,
      "learning_rate": 0.0001,
      "loss": 4.4037,
      "loss/crossentropy": 2.000797212123871,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20464950054883957,
      "step": 7294
    },
    {
      "epoch": 0.14592,
      "grad_norm": 2.125,
      "grad_norm_var": 0.013734690348307292,
      "learning_rate": 0.0001,
      "loss": 4.4132,
      "loss/crossentropy": 2.1914668679237366,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2406519278883934,
      "step": 7296
    },
    {
      "epoch": 0.14596,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.01685358683268229,
      "learning_rate": 0.0001,
      "loss": 4.4737,
      "loss/crossentropy": 2.123211979866028,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.250032439827919,
      "step": 7298
    },
    {
      "epoch": 0.146,
      "grad_norm": 2.625,
      "grad_norm_var": 0.028507232666015625,
      "learning_rate": 0.0001,
      "loss": 4.718,
      "loss/crossentropy": 2.0686148405075073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22581970691680908,
      "step": 7300
    },
    {
      "epoch": 0.14604,
      "grad_norm": 2.25,
      "grad_norm_var": 0.029288482666015626,
      "learning_rate": 0.0001,
      "loss": 4.2295,
      "loss/crossentropy": 2.141907751560211,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21503636240959167,
      "step": 7302
    },
    {
      "epoch": 0.14608,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.028436024983723957,
      "learning_rate": 0.0001,
      "loss": 4.1493,
      "loss/crossentropy": 1.6741206645965576,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20490698516368866,
      "step": 7304
    },
    {
      "epoch": 0.14612,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.027854156494140626,
      "learning_rate": 0.0001,
      "loss": 4.1259,
      "loss/crossentropy": 1.8094561696052551,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21328043192625046,
      "step": 7306
    },
    {
      "epoch": 0.14616,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.020970662434895832,
      "learning_rate": 0.0001,
      "loss": 4.3715,
      "loss/crossentropy": 2.204083800315857,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24204879999160767,
      "step": 7308
    },
    {
      "epoch": 0.1462,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.021126302083333333,
      "learning_rate": 0.0001,
      "loss": 4.2614,
      "loss/crossentropy": 2.0166819095611572,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2219880372285843,
      "step": 7310
    },
    {
      "epoch": 0.14624,
      "grad_norm": 2.25,
      "grad_norm_var": 0.022684733072916668,
      "learning_rate": 0.0001,
      "loss": 4.3768,
      "loss/crossentropy": 2.4667757749557495,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2880199924111366,
      "step": 7312
    },
    {
      "epoch": 0.14628,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.019287109375,
      "learning_rate": 0.0001,
      "loss": 4.2005,
      "loss/crossentropy": 2.1497310400009155,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21792854368686676,
      "step": 7314
    },
    {
      "epoch": 0.14632,
      "grad_norm": 2.25,
      "grad_norm_var": 0.0062896728515625,
      "learning_rate": 0.0001,
      "loss": 4.3911,
      "loss/crossentropy": 2.179584264755249,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23200812935829163,
      "step": 7316
    },
    {
      "epoch": 0.14636,
      "grad_norm": 6.0625,
      "grad_norm_var": 0.9582590738932292,
      "learning_rate": 0.0001,
      "loss": 4.1545,
      "loss/crossentropy": 1.4067250490188599,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.204126738011837,
      "step": 7318
    },
    {
      "epoch": 0.1464,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.9473592122395833,
      "learning_rate": 0.0001,
      "loss": 4.2439,
      "loss/crossentropy": 1.9879435896873474,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.247541606426239,
      "step": 7320
    },
    {
      "epoch": 0.14644,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.9377919514973958,
      "learning_rate": 0.0001,
      "loss": 4.1795,
      "loss/crossentropy": 1.62649005651474,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2054424211382866,
      "step": 7322
    },
    {
      "epoch": 0.14648,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.9344228108723959,
      "learning_rate": 0.0001,
      "loss": 4.1479,
      "loss/crossentropy": 1.9012435674667358,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2240126132965088,
      "step": 7324
    },
    {
      "epoch": 0.14652,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.948193359375,
      "learning_rate": 0.0001,
      "loss": 3.6841,
      "loss/crossentropy": 1.8239200115203857,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19691675901412964,
      "step": 7326
    },
    {
      "epoch": 0.14656,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.94068603515625,
      "learning_rate": 0.0001,
      "loss": 4.2223,
      "loss/crossentropy": 1.9689037799835205,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21301231533288956,
      "step": 7328
    },
    {
      "epoch": 0.1466,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.9363433837890625,
      "learning_rate": 0.0001,
      "loss": 4.3733,
      "loss/crossentropy": 2.1064560413360596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23990632593631744,
      "step": 7330
    },
    {
      "epoch": 0.14664,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.9397125244140625,
      "learning_rate": 0.0001,
      "loss": 4.2765,
      "loss/crossentropy": 2.0871987342834473,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23108436167240143,
      "step": 7332
    },
    {
      "epoch": 0.14668,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.0209869384765625,
      "learning_rate": 0.0001,
      "loss": 4.4517,
      "loss/crossentropy": 2.3061007857322693,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24406791478395462,
      "step": 7334
    },
    {
      "epoch": 0.14672,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.016943359375,
      "learning_rate": 0.0001,
      "loss": 4.4278,
      "loss/crossentropy": 1.9471244812011719,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20859277993440628,
      "step": 7336
    },
    {
      "epoch": 0.14676,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.0131744384765625,
      "learning_rate": 0.0001,
      "loss": 4.188,
      "loss/crossentropy": 2.0628740191459656,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25501881539821625,
      "step": 7338
    },
    {
      "epoch": 0.1468,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.014481353759765624,
      "learning_rate": 0.0001,
      "loss": 4.4329,
      "loss/crossentropy": 1.8065250515937805,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20222238451242447,
      "step": 7340
    },
    {
      "epoch": 0.14684,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.012910715738932292,
      "learning_rate": 0.0001,
      "loss": 4.5431,
      "loss/crossentropy": 2.134244918823242,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22319403290748596,
      "step": 7342
    },
    {
      "epoch": 0.14688,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.012359364827473959,
      "learning_rate": 0.0001,
      "loss": 4.3098,
      "loss/crossentropy": 2.5807924270629883,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26865454018116,
      "step": 7344
    },
    {
      "epoch": 0.14692,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.015421295166015625,
      "learning_rate": 0.0001,
      "loss": 4.684,
      "loss/crossentropy": 2.4128278493881226,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25771988928318024,
      "step": 7346
    },
    {
      "epoch": 0.14696,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.017561594645182293,
      "learning_rate": 0.0001,
      "loss": 4.0915,
      "loss/crossentropy": 1.7323983907699585,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2146565169095993,
      "step": 7348
    },
    {
      "epoch": 0.147,
      "grad_norm": 2.125,
      "grad_norm_var": 0.015553538004557292,
      "learning_rate": 0.0001,
      "loss": 4.5511,
      "loss/crossentropy": 2.036192536354065,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21692586690187454,
      "step": 7350
    },
    {
      "epoch": 0.14704,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.012320709228515626,
      "learning_rate": 0.0001,
      "loss": 4.4173,
      "loss/crossentropy": 1.9586528539657593,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21543999761343002,
      "step": 7352
    },
    {
      "epoch": 0.14708,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.011805979410807292,
      "learning_rate": 0.0001,
      "loss": 4.4506,
      "loss/crossentropy": 2.3444113731384277,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24080512672662735,
      "step": 7354
    },
    {
      "epoch": 0.14712,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.0097076416015625,
      "learning_rate": 0.0001,
      "loss": 4.4865,
      "loss/crossentropy": 2.3060439825057983,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24567800760269165,
      "step": 7356
    },
    {
      "epoch": 0.14716,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.01279296875,
      "learning_rate": 0.0001,
      "loss": 4.5975,
      "loss/crossentropy": 2.2267106771469116,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2345375493168831,
      "step": 7358
    },
    {
      "epoch": 0.1472,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.014481608072916667,
      "learning_rate": 0.0001,
      "loss": 4.7052,
      "loss/crossentropy": 2.2470518350601196,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2417411357164383,
      "step": 7360
    },
    {
      "epoch": 0.14724,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.012953440348307291,
      "learning_rate": 0.0001,
      "loss": 4.3299,
      "loss/crossentropy": 2.200543165206909,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24434109032154083,
      "step": 7362
    },
    {
      "epoch": 0.14728,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.010227203369140625,
      "learning_rate": 0.0001,
      "loss": 4.3282,
      "loss/crossentropy": 1.995844304561615,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22591491788625717,
      "step": 7364
    },
    {
      "epoch": 0.14732,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.011580149332682291,
      "learning_rate": 0.0001,
      "loss": 4.3354,
      "loss/crossentropy": 1.9180658459663391,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22776676714420319,
      "step": 7366
    },
    {
      "epoch": 0.14736,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.011840565999348959,
      "learning_rate": 0.0001,
      "loss": 4.3662,
      "loss/crossentropy": 2.473931312561035,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26284685730934143,
      "step": 7368
    },
    {
      "epoch": 0.1474,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.011744944254557292,
      "learning_rate": 0.0001,
      "loss": 4.3167,
      "loss/crossentropy": 2.0392738580703735,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22323766350746155,
      "step": 7370
    },
    {
      "epoch": 0.14744,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.012094879150390625,
      "learning_rate": 0.0001,
      "loss": 4.3032,
      "loss/crossentropy": 1.9847410917282104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22324485331773758,
      "step": 7372
    },
    {
      "epoch": 0.14748,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.008459218343098958,
      "learning_rate": 0.0001,
      "loss": 4.2794,
      "loss/crossentropy": 1.930326521396637,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24155349284410477,
      "step": 7374
    },
    {
      "epoch": 0.14752,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0061724344889322914,
      "learning_rate": 0.0001,
      "loss": 4.0648,
      "loss/crossentropy": 1.825449824333191,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2087552770972252,
      "step": 7376
    },
    {
      "epoch": 0.14756,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.004325358072916666,
      "learning_rate": 0.0001,
      "loss": 4.2405,
      "loss/crossentropy": 2.156645655632019,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2519562169909477,
      "step": 7378
    },
    {
      "epoch": 0.1476,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.008333333333333333,
      "learning_rate": 0.0001,
      "loss": 4.5316,
      "loss/crossentropy": 2.255813479423523,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23737946152687073,
      "step": 7380
    },
    {
      "epoch": 0.14764,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.00797119140625,
      "learning_rate": 0.0001,
      "loss": 4.5795,
      "loss/crossentropy": 2.47933566570282,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23890959471464157,
      "step": 7382
    },
    {
      "epoch": 0.14768,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.007710774739583333,
      "learning_rate": 0.0001,
      "loss": 4.5328,
      "loss/crossentropy": 2.139566659927368,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22334300726652145,
      "step": 7384
    },
    {
      "epoch": 0.14772,
      "grad_norm": 2.125,
      "grad_norm_var": 0.007673136393229167,
      "learning_rate": 0.0001,
      "loss": 4.1581,
      "loss/crossentropy": 1.6182149052619934,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19345169514417648,
      "step": 7386
    },
    {
      "epoch": 0.14776,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.010758463541666667,
      "learning_rate": 0.0001,
      "loss": 4.4571,
      "loss/crossentropy": 1.9984004497528076,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24823245406150818,
      "step": 7388
    },
    {
      "epoch": 0.1478,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.010445149739583333,
      "learning_rate": 0.0001,
      "loss": 4.3218,
      "loss/crossentropy": 2.5892586708068848,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2489030361175537,
      "step": 7390
    },
    {
      "epoch": 0.14784,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.009598795572916667,
      "learning_rate": 0.0001,
      "loss": 4.6077,
      "loss/crossentropy": 2.3723479509353638,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23511765897274017,
      "step": 7392
    },
    {
      "epoch": 0.14788,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.0145660400390625,
      "learning_rate": 0.0001,
      "loss": 4.3902,
      "loss/crossentropy": 2.001940071582794,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22256013005971909,
      "step": 7394
    },
    {
      "epoch": 0.14792,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.013850911458333334,
      "learning_rate": 0.0001,
      "loss": 4.4422,
      "loss/crossentropy": 2.2255555391311646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23905682563781738,
      "step": 7396
    },
    {
      "epoch": 0.14796,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.013277180989583333,
      "learning_rate": 0.0001,
      "loss": 4.6217,
      "loss/crossentropy": 2.5670583248138428,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25223178416490555,
      "step": 7398
    },
    {
      "epoch": 0.148,
      "grad_norm": 2.203125,
      "grad_norm_var": 2.3002278645833334,
      "learning_rate": 0.0001,
      "loss": 4.6365,
      "loss/crossentropy": 2.16109037399292,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2601509317755699,
      "step": 7400
    },
    {
      "epoch": 0.14804,
      "grad_norm": 2.09375,
      "grad_norm_var": 2.3012847900390625,
      "learning_rate": 0.0001,
      "loss": 4.4169,
      "loss/crossentropy": 2.1484315395355225,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21474219858646393,
      "step": 7402
    },
    {
      "epoch": 0.14808,
      "grad_norm": 4.3125,
      "grad_norm_var": 2.48385009765625,
      "learning_rate": 0.0001,
      "loss": 4.7452,
      "loss/crossentropy": 2.183099091053009,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25882233679294586,
      "step": 7404
    },
    {
      "epoch": 0.14812,
      "grad_norm": 2.0625,
      "grad_norm_var": 2.491097005208333,
      "learning_rate": 0.0001,
      "loss": 3.9739,
      "loss/crossentropy": 1.795831561088562,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20870305597782135,
      "step": 7406
    },
    {
      "epoch": 0.14816,
      "grad_norm": 2.09375,
      "grad_norm_var": 2.5058553059895834,
      "learning_rate": 0.0001,
      "loss": 4.3433,
      "loss/crossentropy": 2.202280640602112,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2150556966662407,
      "step": 7408
    },
    {
      "epoch": 0.1482,
      "grad_norm": 2.03125,
      "grad_norm_var": 2.5274251302083335,
      "learning_rate": 0.0001,
      "loss": 4.3063,
      "loss/crossentropy": 2.2616937160491943,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24934212118387222,
      "step": 7410
    },
    {
      "epoch": 0.14824,
      "grad_norm": 2.171875,
      "grad_norm_var": 2.51627197265625,
      "learning_rate": 0.0001,
      "loss": 4.3303,
      "loss/crossentropy": 1.909091055393219,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23191271722316742,
      "step": 7412
    },
    {
      "epoch": 0.14828,
      "grad_norm": 2.09375,
      "grad_norm_var": 2.5269765218098956,
      "learning_rate": 0.0001,
      "loss": 4.4224,
      "loss/crossentropy": 2.1383588314056396,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22893162816762924,
      "step": 7414
    },
    {
      "epoch": 0.14832,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.2983062744140625,
      "learning_rate": 0.0001,
      "loss": 4.3664,
      "loss/crossentropy": 2.0628907680511475,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23075110465288162,
      "step": 7416
    },
    {
      "epoch": 0.14836,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.29704488118489586,
      "learning_rate": 0.0001,
      "loss": 4.2109,
      "loss/crossentropy": 1.8996745347976685,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21171879768371582,
      "step": 7418
    },
    {
      "epoch": 0.1484,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0032867431640625,
      "learning_rate": 0.0001,
      "loss": 4.4241,
      "loss/crossentropy": 2.2367645502090454,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2563689202070236,
      "step": 7420
    },
    {
      "epoch": 0.14844,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.003934733072916667,
      "learning_rate": 0.0001,
      "loss": 4.2326,
      "loss/crossentropy": 2.274489164352417,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2462785318493843,
      "step": 7422
    },
    {
      "epoch": 0.14848,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.00562744140625,
      "learning_rate": 0.0001,
      "loss": 4.5288,
      "loss/crossentropy": 2.227620482444763,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24729500710964203,
      "step": 7424
    },
    {
      "epoch": 0.14852,
      "grad_norm": 2.0,
      "grad_norm_var": 0.006864420572916667,
      "learning_rate": 0.0001,
      "loss": 4.0913,
      "loss/crossentropy": 1.9911785125732422,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.216292105615139,
      "step": 7426
    },
    {
      "epoch": 0.14856,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.008199055989583334,
      "learning_rate": 0.0001,
      "loss": 4.4071,
      "loss/crossentropy": 1.8715736865997314,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2289058193564415,
      "step": 7428
    },
    {
      "epoch": 0.1486,
      "grad_norm": 2.125,
      "grad_norm_var": 0.007991536458333334,
      "learning_rate": 0.0001,
      "loss": 4.4671,
      "loss/crossentropy": 2.0192378759384155,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2327374964952469,
      "step": 7430
    },
    {
      "epoch": 0.14864,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.008226521809895833,
      "learning_rate": 0.0001,
      "loss": 4.213,
      "loss/crossentropy": 1.9407023191452026,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22586089372634888,
      "step": 7432
    },
    {
      "epoch": 0.14868,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.01011962890625,
      "learning_rate": 0.0001,
      "loss": 4.7816,
      "loss/crossentropy": 2.6349592208862305,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2605717331171036,
      "step": 7434
    },
    {
      "epoch": 0.14872,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.009989420572916666,
      "learning_rate": 0.0001,
      "loss": 4.5511,
      "loss/crossentropy": 2.3015077114105225,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2292959839105606,
      "step": 7436
    },
    {
      "epoch": 0.14876,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.013231404622395833,
      "learning_rate": 0.0001,
      "loss": 4.1663,
      "loss/crossentropy": 2.081148624420166,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21435046195983887,
      "step": 7438
    },
    {
      "epoch": 0.1488,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.011735026041666667,
      "learning_rate": 0.0001,
      "loss": 4.4342,
      "loss/crossentropy": 2.235422372817993,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23794984817504883,
      "step": 7440
    },
    {
      "epoch": 0.14884,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.011295572916666666,
      "learning_rate": 0.0001,
      "loss": 4.4595,
      "loss/crossentropy": 2.1423263549804688,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2200494110584259,
      "step": 7442
    },
    {
      "epoch": 0.14888,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.018342081705729166,
      "learning_rate": 0.0001,
      "loss": 4.514,
      "loss/crossentropy": 2.3156672716140747,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25282321125268936,
      "step": 7444
    },
    {
      "epoch": 0.14892,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.020051066080729166,
      "learning_rate": 0.0001,
      "loss": 4.3107,
      "loss/crossentropy": 1.7777396440505981,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2104710191488266,
      "step": 7446
    },
    {
      "epoch": 0.14896,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.023341623942057292,
      "learning_rate": 0.0001,
      "loss": 3.874,
      "loss/crossentropy": 1.8015541434288025,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1976182758808136,
      "step": 7448
    },
    {
      "epoch": 0.149,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.04948298136393229,
      "learning_rate": 0.0001,
      "loss": 4.5647,
      "loss/crossentropy": 2.36995792388916,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23767317831516266,
      "step": 7450
    },
    {
      "epoch": 0.14904,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.050142161051432294,
      "learning_rate": 0.0001,
      "loss": 4.7568,
      "loss/crossentropy": 2.2867971062660217,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2666979879140854,
      "step": 7452
    },
    {
      "epoch": 0.14908,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.043794504801432294,
      "learning_rate": 0.0001,
      "loss": 4.773,
      "loss/crossentropy": 2.247913956642151,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2122594192624092,
      "step": 7454
    },
    {
      "epoch": 0.14912,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.04197362263997396,
      "learning_rate": 0.0001,
      "loss": 4.273,
      "loss/crossentropy": 2.0020886063575745,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22448039799928665,
      "step": 7456
    },
    {
      "epoch": 0.14916,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.043702952067057294,
      "learning_rate": 0.0001,
      "loss": 4.3139,
      "loss/crossentropy": 2.2290679216384888,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21489766240119934,
      "step": 7458
    },
    {
      "epoch": 0.1492,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.049344635009765624,
      "learning_rate": 0.0001,
      "loss": 4.5919,
      "loss/crossentropy": 2.0447877049446106,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22640856355428696,
      "step": 7460
    },
    {
      "epoch": 0.14924,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.045904286702473956,
      "learning_rate": 0.0001,
      "loss": 4.4614,
      "loss/crossentropy": 2.395468831062317,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2395479902625084,
      "step": 7462
    },
    {
      "epoch": 0.14928,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.04019775390625,
      "learning_rate": 0.0001,
      "loss": 4.1552,
      "loss/crossentropy": 2.117182433605194,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22302603721618652,
      "step": 7464
    },
    {
      "epoch": 0.14932,
      "grad_norm": 2.125,
      "grad_norm_var": 0.019498697916666665,
      "learning_rate": 0.0001,
      "loss": 4.4046,
      "loss/crossentropy": 2.206045985221863,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2548774778842926,
      "step": 7466
    },
    {
      "epoch": 0.14936,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0194244384765625,
      "learning_rate": 0.0001,
      "loss": 4.1531,
      "loss/crossentropy": 1.9933450818061829,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22455794364213943,
      "step": 7468
    },
    {
      "epoch": 0.1494,
      "grad_norm": 2.125,
      "grad_norm_var": 0.017731730143229166,
      "learning_rate": 0.0001,
      "loss": 4.2744,
      "loss/crossentropy": 2.216577649116516,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24432705342769623,
      "step": 7470
    },
    {
      "epoch": 0.14944,
      "grad_norm": 2.25,
      "grad_norm_var": 0.018163045247395832,
      "learning_rate": 0.0001,
      "loss": 4.6072,
      "loss/crossentropy": 2.4282405376434326,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2599884122610092,
      "step": 7472
    },
    {
      "epoch": 0.14948,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.018089803059895833,
      "learning_rate": 0.0001,
      "loss": 4.2833,
      "loss/crossentropy": 2.2677053213119507,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24509359151124954,
      "step": 7474
    },
    {
      "epoch": 0.14952,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.007877604166666666,
      "learning_rate": 0.0001,
      "loss": 4.1271,
      "loss/crossentropy": 1.9610475897789001,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19867657870054245,
      "step": 7476
    },
    {
      "epoch": 0.14956,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.012565104166666667,
      "learning_rate": 0.0001,
      "loss": 4.6286,
      "loss/crossentropy": 2.1379209756851196,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24773475527763367,
      "step": 7478
    },
    {
      "epoch": 0.1496,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0125152587890625,
      "learning_rate": 0.0001,
      "loss": 4.3593,
      "loss/crossentropy": 2.295411467552185,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2417762354016304,
      "step": 7480
    },
    {
      "epoch": 0.14964,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.013374837239583333,
      "learning_rate": 0.0001,
      "loss": 4.4839,
      "loss/crossentropy": 2.0983279943466187,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22220823168754578,
      "step": 7482
    },
    {
      "epoch": 0.14968,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.015086873372395834,
      "learning_rate": 0.0001,
      "loss": 4.3977,
      "loss/crossentropy": 2.130508065223694,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23387905955314636,
      "step": 7484
    },
    {
      "epoch": 0.14972,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.015869140625,
      "learning_rate": 0.0001,
      "loss": 4.545,
      "loss/crossentropy": 2.0727924704551697,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23198049515485764,
      "step": 7486
    },
    {
      "epoch": 0.14976,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.015523274739583334,
      "learning_rate": 0.0001,
      "loss": 4.3722,
      "loss/crossentropy": 2.1407171487808228,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23349495232105255,
      "step": 7488
    },
    {
      "epoch": 0.1498,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.016063435872395834,
      "learning_rate": 0.0001,
      "loss": 4.3483,
      "loss/crossentropy": 2.123879909515381,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2230711579322815,
      "step": 7490
    },
    {
      "epoch": 0.14984,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.02194188435872396,
      "learning_rate": 0.0001,
      "loss": 3.8687,
      "loss/crossentropy": 2.1362847685813904,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22217128425836563,
      "step": 7492
    },
    {
      "epoch": 0.14988,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.01625544230143229,
      "learning_rate": 0.0001,
      "loss": 4.4753,
      "loss/crossentropy": 2.2171897292137146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.237472265958786,
      "step": 7494
    },
    {
      "epoch": 0.14992,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.016841379801432292,
      "learning_rate": 0.0001,
      "loss": 4.2439,
      "loss/crossentropy": 1.9622138142585754,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19700734317302704,
      "step": 7496
    },
    {
      "epoch": 0.14996,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.01666234334309896,
      "learning_rate": 0.0001,
      "loss": 4.1296,
      "loss/crossentropy": 2.300232410430908,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26260019838809967,
      "step": 7498
    },
    {
      "epoch": 0.15,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.013765207926432292,
      "learning_rate": 0.0001,
      "loss": 4.4466,
      "loss/crossentropy": 1.9829946756362915,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21770135313272476,
      "step": 7500
    },
    {
      "epoch": 0.15004,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.010935211181640625,
      "learning_rate": 0.0001,
      "loss": 4.017,
      "loss/crossentropy": 1.8421878218650818,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20965785533189774,
      "step": 7502
    },
    {
      "epoch": 0.15008,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.010267893473307291,
      "learning_rate": 0.0001,
      "loss": 4.3111,
      "loss/crossentropy": 2.0559566020965576,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21752064675092697,
      "step": 7504
    },
    {
      "epoch": 0.15012,
      "grad_norm": 2.25,
      "grad_norm_var": 0.009834543863932291,
      "learning_rate": 0.0001,
      "loss": 4.5526,
      "loss/crossentropy": 1.862777590751648,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2026660442352295,
      "step": 7506
    },
    {
      "epoch": 0.15016,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0064198811848958336,
      "learning_rate": 0.0001,
      "loss": 4.0238,
      "loss/crossentropy": 2.340041399002075,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23489895462989807,
      "step": 7508
    },
    {
      "epoch": 0.1502,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.006224568684895833,
      "learning_rate": 0.0001,
      "loss": 4.5434,
      "loss/crossentropy": 2.3596150875091553,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23775358498096466,
      "step": 7510
    },
    {
      "epoch": 0.15024,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.005980428059895833,
      "learning_rate": 0.0001,
      "loss": 4.4637,
      "loss/crossentropy": 1.883503019809723,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22191710770130157,
      "step": 7512
    },
    {
      "epoch": 0.15028,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.006723785400390625,
      "learning_rate": 0.0001,
      "loss": 4.1485,
      "loss/crossentropy": 2.12862491607666,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2177818939089775,
      "step": 7514
    },
    {
      "epoch": 0.15032,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.005936431884765625,
      "learning_rate": 0.0001,
      "loss": 4.167,
      "loss/crossentropy": 1.9326343536376953,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20991001278162003,
      "step": 7516
    },
    {
      "epoch": 0.15036,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.008070627848307291,
      "learning_rate": 0.0001,
      "loss": 4.4552,
      "loss/crossentropy": 2.1532761454582214,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2308126464486122,
      "step": 7518
    },
    {
      "epoch": 0.1504,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.007867177327473959,
      "learning_rate": 0.0001,
      "loss": 4.42,
      "loss/crossentropy": 1.876515507698059,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1940685734152794,
      "step": 7520
    },
    {
      "epoch": 0.15044,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.008937327067057292,
      "learning_rate": 0.0001,
      "loss": 4.3515,
      "loss/crossentropy": 2.3677611351013184,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24787116795778275,
      "step": 7522
    },
    {
      "epoch": 0.15048,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.010892486572265625,
      "learning_rate": 0.0001,
      "loss": 4.0489,
      "loss/crossentropy": 1.8500076532363892,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19271107017993927,
      "step": 7524
    },
    {
      "epoch": 0.15052,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.010432688395182292,
      "learning_rate": 0.0001,
      "loss": 4.3298,
      "loss/crossentropy": 1.8013980984687805,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2181655913591385,
      "step": 7526
    },
    {
      "epoch": 0.15056,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.010361480712890624,
      "learning_rate": 0.0001,
      "loss": 4.4958,
      "loss/crossentropy": 2.6469568014144897,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2441466525197029,
      "step": 7528
    },
    {
      "epoch": 0.1506,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.009227498372395834,
      "learning_rate": 0.0001,
      "loss": 4.1317,
      "loss/crossentropy": 1.7992960214614868,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2209397256374359,
      "step": 7530
    },
    {
      "epoch": 0.15064,
      "grad_norm": 2.625,
      "grad_norm_var": 0.023258463541666666,
      "learning_rate": 0.0001,
      "loss": 4.9501,
      "loss/crossentropy": 2.1914783120155334,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23764144629240036,
      "step": 7532
    },
    {
      "epoch": 0.15068,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.02906494140625,
      "learning_rate": 0.0001,
      "loss": 4.4833,
      "loss/crossentropy": 2.0035120844841003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21483591943979263,
      "step": 7534
    },
    {
      "epoch": 0.15072,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.03235677083333333,
      "learning_rate": 0.0001,
      "loss": 4.2674,
      "loss/crossentropy": 2.0218639969825745,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20919294655323029,
      "step": 7536
    },
    {
      "epoch": 0.15076,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.03219401041666667,
      "learning_rate": 0.0001,
      "loss": 4.1091,
      "loss/crossentropy": 2.006688416004181,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21247616410255432,
      "step": 7538
    },
    {
      "epoch": 0.1508,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.03542378743489583,
      "learning_rate": 0.0001,
      "loss": 4.5424,
      "loss/crossentropy": 2.1565613746643066,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30109211802482605,
      "step": 7540
    },
    {
      "epoch": 0.15084,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.03603413899739583,
      "learning_rate": 0.0001,
      "loss": 4.3621,
      "loss/crossentropy": 1.9676685333251953,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23605409264564514,
      "step": 7542
    },
    {
      "epoch": 0.15088,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.051493326822916664,
      "learning_rate": 0.0001,
      "loss": 4.4178,
      "loss/crossentropy": 2.1150137186050415,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22903620451688766,
      "step": 7544
    },
    {
      "epoch": 0.15092,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.051253255208333334,
      "learning_rate": 0.0001,
      "loss": 4.3186,
      "loss/crossentropy": 2.1894554495811462,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24362927675247192,
      "step": 7546
    },
    {
      "epoch": 0.15096,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.0410308837890625,
      "learning_rate": 0.0001,
      "loss": 4.4529,
      "loss/crossentropy": 1.9624019861221313,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22628726810216904,
      "step": 7548
    },
    {
      "epoch": 0.151,
      "grad_norm": 2.0,
      "grad_norm_var": 0.04338277180989583,
      "learning_rate": 0.0001,
      "loss": 4.2795,
      "loss/crossentropy": 2.1547625064849854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2295984849333763,
      "step": 7550
    },
    {
      "epoch": 0.15104,
      "grad_norm": 2.125,
      "grad_norm_var": 0.03857014973958333,
      "learning_rate": 0.0001,
      "loss": 4.4638,
      "loss/crossentropy": 2.229305863380432,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2427791878581047,
      "step": 7552
    },
    {
      "epoch": 0.15108,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.037093098958333334,
      "learning_rate": 0.0001,
      "loss": 4.7319,
      "loss/crossentropy": 2.4998362064361572,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23718395829200745,
      "step": 7554
    },
    {
      "epoch": 0.15112,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.033722941080729166,
      "learning_rate": 0.0001,
      "loss": 4.2118,
      "loss/crossentropy": 2.319428563117981,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26024487614631653,
      "step": 7556
    },
    {
      "epoch": 0.15116,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0357818603515625,
      "learning_rate": 0.0001,
      "loss": 4.2451,
      "loss/crossentropy": 1.861966609954834,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2121218591928482,
      "step": 7558
    },
    {
      "epoch": 0.1512,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.015751139322916666,
      "learning_rate": 0.0001,
      "loss": 4.5963,
      "loss/crossentropy": 2.1688510179519653,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22341617196798325,
      "step": 7560
    },
    {
      "epoch": 0.15124,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.013932291666666667,
      "learning_rate": 0.0001,
      "loss": 4.4808,
      "loss/crossentropy": 2.2833406925201416,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22648481279611588,
      "step": 7562
    },
    {
      "epoch": 0.15128,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.0138671875,
      "learning_rate": 0.0001,
      "loss": 4.4126,
      "loss/crossentropy": 2.224393129348755,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23414459079504013,
      "step": 7564
    },
    {
      "epoch": 0.15132,
      "grad_norm": 2.25,
      "grad_norm_var": 0.0098297119140625,
      "learning_rate": 0.0001,
      "loss": 4.253,
      "loss/crossentropy": 2.003828763961792,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2166791632771492,
      "step": 7566
    },
    {
      "epoch": 0.15136,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.01168212890625,
      "learning_rate": 0.0001,
      "loss": 4.0904,
      "loss/crossentropy": 2.1813005208969116,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28873007744550705,
      "step": 7568
    },
    {
      "epoch": 0.1514,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.028696441650390626,
      "learning_rate": 0.0001,
      "loss": 4.3824,
      "loss/crossentropy": 2.3818061351776123,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24916332960128784,
      "step": 7570
    },
    {
      "epoch": 0.15144,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.030460357666015625,
      "learning_rate": 0.0001,
      "loss": 4.1825,
      "loss/crossentropy": 2.041518449783325,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22352956235408783,
      "step": 7572
    },
    {
      "epoch": 0.15148,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0326904296875,
      "learning_rate": 0.0001,
      "loss": 3.8531,
      "loss/crossentropy": 1.856759488582611,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20227209478616714,
      "step": 7574
    },
    {
      "epoch": 0.15152,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.03157450358072917,
      "learning_rate": 0.0001,
      "loss": 4.237,
      "loss/crossentropy": 1.9612281918525696,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22188737243413925,
      "step": 7576
    },
    {
      "epoch": 0.15156,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0337554931640625,
      "learning_rate": 0.0001,
      "loss": 4.0405,
      "loss/crossentropy": 1.9610649943351746,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2282957062125206,
      "step": 7578
    },
    {
      "epoch": 0.1516,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.03400472005208333,
      "learning_rate": 0.0001,
      "loss": 4.1253,
      "loss/crossentropy": 1.9239189624786377,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22926658391952515,
      "step": 7580
    },
    {
      "epoch": 0.15164,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.033299763997395836,
      "learning_rate": 0.0001,
      "loss": 4.548,
      "loss/crossentropy": 2.425737738609314,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24260805547237396,
      "step": 7582
    },
    {
      "epoch": 0.15168,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.03242085774739583,
      "learning_rate": 0.0001,
      "loss": 4.2894,
      "loss/crossentropy": 1.9602521061897278,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23231150209903717,
      "step": 7584
    },
    {
      "epoch": 0.15172,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.007954661051432292,
      "learning_rate": 0.0001,
      "loss": 4.3131,
      "loss/crossentropy": 2.3671375513076782,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24326395988464355,
      "step": 7586
    },
    {
      "epoch": 0.15176,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.007675933837890625,
      "learning_rate": 0.0001,
      "loss": 4.1297,
      "loss/crossentropy": 2.0128119587898254,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20082567632198334,
      "step": 7588
    },
    {
      "epoch": 0.1518,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.0085845947265625,
      "learning_rate": 0.0001,
      "loss": 4.5813,
      "loss/crossentropy": 2.3719125986099243,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.243422269821167,
      "step": 7590
    },
    {
      "epoch": 0.15184,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.010575358072916667,
      "learning_rate": 0.0001,
      "loss": 4.6177,
      "loss/crossentropy": 2.088695764541626,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23578013479709625,
      "step": 7592
    },
    {
      "epoch": 0.15188,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.01099853515625,
      "learning_rate": 0.0001,
      "loss": 4.3719,
      "loss/crossentropy": 2.151872456073761,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2496410757303238,
      "step": 7594
    },
    {
      "epoch": 0.15192,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.017281087239583333,
      "learning_rate": 0.0001,
      "loss": 4.6041,
      "loss/crossentropy": 2.020963430404663,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25003430247306824,
      "step": 7596
    },
    {
      "epoch": 0.15196,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.0199615478515625,
      "learning_rate": 0.0001,
      "loss": 4.4439,
      "loss/crossentropy": 1.9290395379066467,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20771078765392303,
      "step": 7598
    },
    {
      "epoch": 0.152,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.020536295572916665,
      "learning_rate": 0.0001,
      "loss": 4.2461,
      "loss/crossentropy": 2.0420188307762146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2375432327389717,
      "step": 7600
    },
    {
      "epoch": 0.15204,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.019169108072916666,
      "learning_rate": 0.0001,
      "loss": 4.4291,
      "loss/crossentropy": 2.474969744682312,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2535504847764969,
      "step": 7602
    },
    {
      "epoch": 0.15208,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.015550740559895833,
      "learning_rate": 0.0001,
      "loss": 4.3487,
      "loss/crossentropy": 2.177125334739685,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24919381737709045,
      "step": 7604
    },
    {
      "epoch": 0.15212,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.015510050455729167,
      "learning_rate": 0.0001,
      "loss": 4.4719,
      "loss/crossentropy": 2.348747491836548,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22336618602275848,
      "step": 7606
    },
    {
      "epoch": 0.15216,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.014676920572916667,
      "learning_rate": 0.0001,
      "loss": 4.4502,
      "loss/crossentropy": 1.718321442604065,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19830843806266785,
      "step": 7608
    },
    {
      "epoch": 0.1522,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.015901692708333335,
      "learning_rate": 0.0001,
      "loss": 4.4505,
      "loss/crossentropy": 2.2954800128936768,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23203244805335999,
      "step": 7610
    },
    {
      "epoch": 0.15224,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.013570149739583334,
      "learning_rate": 0.0001,
      "loss": 4.1141,
      "loss/crossentropy": 1.6918454766273499,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18312305957078934,
      "step": 7612
    },
    {
      "epoch": 0.15228,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.012727864583333333,
      "learning_rate": 0.0001,
      "loss": 4.1312,
      "loss/crossentropy": 1.887774109840393,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2151220142841339,
      "step": 7614
    },
    {
      "epoch": 0.15232,
      "grad_norm": 2.125,
      "grad_norm_var": 0.01373291015625,
      "learning_rate": 0.0001,
      "loss": 4.1274,
      "loss/crossentropy": 2.0903998613357544,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22486191242933273,
      "step": 7616
    },
    {
      "epoch": 0.15236,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.016310373942057293,
      "learning_rate": 0.0001,
      "loss": 4.0659,
      "loss/crossentropy": 1.929358720779419,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20021560043096542,
      "step": 7618
    },
    {
      "epoch": 0.1524,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.03484064737955729,
      "learning_rate": 0.0001,
      "loss": 4.5726,
      "loss/crossentropy": 1.9812661409378052,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21395482122898102,
      "step": 7620
    },
    {
      "epoch": 0.15244,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.03358942667643229,
      "learning_rate": 0.0001,
      "loss": 4.15,
      "loss/crossentropy": 2.148552179336548,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22805052995681763,
      "step": 7622
    },
    {
      "epoch": 0.15248,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.03416926066080729,
      "learning_rate": 0.0001,
      "loss": 4.3471,
      "loss/crossentropy": 2.012804687023163,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2248745858669281,
      "step": 7624
    },
    {
      "epoch": 0.15252,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.03463312784830729,
      "learning_rate": 0.0001,
      "loss": 4.3948,
      "loss/crossentropy": 2.3378156423568726,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2457970678806305,
      "step": 7626
    },
    {
      "epoch": 0.15256,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.036382802327473956,
      "learning_rate": 0.0001,
      "loss": 4.5414,
      "loss/crossentropy": 2.0815274119377136,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22083784639835358,
      "step": 7628
    },
    {
      "epoch": 0.1526,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.039589182535807295,
      "learning_rate": 0.0001,
      "loss": 4.2609,
      "loss/crossentropy": 2.172307014465332,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25988608598709106,
      "step": 7630
    },
    {
      "epoch": 0.15264,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.03921305338541667,
      "learning_rate": 0.0001,
      "loss": 4.3711,
      "loss/crossentropy": 1.973683476448059,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20379862189292908,
      "step": 7632
    },
    {
      "epoch": 0.15268,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.039406077067057295,
      "learning_rate": 0.0001,
      "loss": 4.3394,
      "loss/crossentropy": 2.175020456314087,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22480874508619308,
      "step": 7634
    },
    {
      "epoch": 0.15272,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.022304026285807292,
      "learning_rate": 0.0001,
      "loss": 4.2176,
      "loss/crossentropy": 2.063227415084839,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2324352264404297,
      "step": 7636
    },
    {
      "epoch": 0.15276,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.019769032796223957,
      "learning_rate": 0.0001,
      "loss": 4.4654,
      "loss/crossentropy": 1.9297555088996887,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2171614021062851,
      "step": 7638
    },
    {
      "epoch": 0.1528,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.019421132405598958,
      "learning_rate": 0.0001,
      "loss": 4.4698,
      "loss/crossentropy": 1.9808542132377625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21765583008527756,
      "step": 7640
    },
    {
      "epoch": 0.15284,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.018790435791015626,
      "learning_rate": 0.0001,
      "loss": 4.4145,
      "loss/crossentropy": 2.009860336780548,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20963389426469803,
      "step": 7642
    },
    {
      "epoch": 0.15288,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.017319488525390624,
      "learning_rate": 0.0001,
      "loss": 4.2649,
      "loss/crossentropy": 1.912703514099121,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23796956986188889,
      "step": 7644
    },
    {
      "epoch": 0.15292,
      "grad_norm": 2.25,
      "grad_norm_var": 0.012341054280598958,
      "learning_rate": 0.0001,
      "loss": 4.4965,
      "loss/crossentropy": 2.2945642471313477,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2621624022722244,
      "step": 7646
    },
    {
      "epoch": 0.15296,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.011067708333333334,
      "learning_rate": 0.0001,
      "loss": 4.2573,
      "loss/crossentropy": 2.185902237892151,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23890018463134766,
      "step": 7648
    },
    {
      "epoch": 0.153,
      "grad_norm": 2.546875,
      "grad_norm_var": 0.018648274739583335,
      "learning_rate": 0.0001,
      "loss": 4.5815,
      "loss/crossentropy": 1.8625503778457642,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2206135168671608,
      "step": 7650
    },
    {
      "epoch": 0.15304,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.020684560139973957,
      "learning_rate": 0.0001,
      "loss": 3.9142,
      "loss/crossentropy": 1.8815893530845642,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21166741847991943,
      "step": 7652
    },
    {
      "epoch": 0.15308,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.020979563395182293,
      "learning_rate": 0.0001,
      "loss": 4.2277,
      "loss/crossentropy": 2.156697630882263,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22994951903820038,
      "step": 7654
    },
    {
      "epoch": 0.15312,
      "grad_norm": 2.25,
      "grad_norm_var": 0.02005182902018229,
      "learning_rate": 0.0001,
      "loss": 4.366,
      "loss/crossentropy": 1.6800576448440552,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21905134618282318,
      "step": 7656
    },
    {
      "epoch": 0.15316,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.02061945597330729,
      "learning_rate": 0.0001,
      "loss": 4.1446,
      "loss/crossentropy": 1.9215145707130432,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2034405767917633,
      "step": 7658
    },
    {
      "epoch": 0.1532,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.02380956013997396,
      "learning_rate": 0.0001,
      "loss": 4.6475,
      "loss/crossentropy": 2.153718650341034,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25605182349681854,
      "step": 7660
    },
    {
      "epoch": 0.15324,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.02466608683268229,
      "learning_rate": 0.0001,
      "loss": 4.2676,
      "loss/crossentropy": 1.8782889246940613,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20574549585580826,
      "step": 7662
    },
    {
      "epoch": 0.15328,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.031040191650390625,
      "learning_rate": 0.0001,
      "loss": 4.2171,
      "loss/crossentropy": 2.003354489803314,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24152260273694992,
      "step": 7664
    },
    {
      "epoch": 0.15332,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.022299957275390626,
      "learning_rate": 0.0001,
      "loss": 4.5828,
      "loss/crossentropy": 2.217758059501648,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24700726568698883,
      "step": 7666
    },
    {
      "epoch": 0.15336,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.017757161458333334,
      "learning_rate": 0.0001,
      "loss": 4.3418,
      "loss/crossentropy": 1.934537410736084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2163502648472786,
      "step": 7668
    },
    {
      "epoch": 0.1534,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.020340983072916666,
      "learning_rate": 0.0001,
      "loss": 4.3,
      "loss/crossentropy": 2.007661819458008,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21769292652606964,
      "step": 7670
    },
    {
      "epoch": 0.15344,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.020653279622395833,
      "learning_rate": 0.0001,
      "loss": 4.3082,
      "loss/crossentropy": 2.1586949825286865,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2453690618276596,
      "step": 7672
    },
    {
      "epoch": 0.15348,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0197174072265625,
      "learning_rate": 0.0001,
      "loss": 4.4433,
      "loss/crossentropy": 2.2201706171035767,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22639702260494232,
      "step": 7674
    },
    {
      "epoch": 0.15352,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.015104166666666667,
      "learning_rate": 0.0001,
      "loss": 4.2337,
      "loss/crossentropy": 2.059146285057068,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22411910444498062,
      "step": 7676
    },
    {
      "epoch": 0.15356,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.015397135416666667,
      "learning_rate": 0.0001,
      "loss": 4.2318,
      "loss/crossentropy": 1.9768954515457153,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21798508614301682,
      "step": 7678
    },
    {
      "epoch": 0.1536,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.003413899739583333,
      "learning_rate": 0.0001,
      "loss": 4.4823,
      "loss/crossentropy": 1.8555094003677368,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19453337788581848,
      "step": 7680
    },
    {
      "epoch": 0.15364,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0036783854166666666,
      "learning_rate": 0.0001,
      "loss": 4.0379,
      "loss/crossentropy": 1.5948917865753174,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17746463418006897,
      "step": 7682
    },
    {
      "epoch": 0.15368,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0034464518229166668,
      "learning_rate": 0.0001,
      "loss": 4.2901,
      "loss/crossentropy": 1.8917757868766785,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2103574424982071,
      "step": 7684
    },
    {
      "epoch": 0.15372,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0030670166015625,
      "learning_rate": 0.0001,
      "loss": 4.3326,
      "loss/crossentropy": 2.090232729911804,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2199823334813118,
      "step": 7686
    },
    {
      "epoch": 0.15376,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.004979451497395833,
      "learning_rate": 0.0001,
      "loss": 4.3695,
      "loss/crossentropy": 1.8155178427696228,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21044857800006866,
      "step": 7688
    },
    {
      "epoch": 0.1538,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0063629150390625,
      "learning_rate": 0.0001,
      "loss": 4.5003,
      "loss/crossentropy": 2.31532621383667,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2437051385641098,
      "step": 7690
    },
    {
      "epoch": 0.15384,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0065419514973958336,
      "learning_rate": 0.0001,
      "loss": 4.3815,
      "loss/crossentropy": 1.9688079357147217,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22686263918876648,
      "step": 7692
    },
    {
      "epoch": 0.15388,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.010001373291015626,
      "learning_rate": 0.0001,
      "loss": 3.9937,
      "loss/crossentropy": 1.9029017686843872,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20635761320590973,
      "step": 7694
    },
    {
      "epoch": 0.15392,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.010587565104166667,
      "learning_rate": 0.0001,
      "loss": 3.91,
      "loss/crossentropy": 1.9817028641700745,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2080235257744789,
      "step": 7696
    },
    {
      "epoch": 0.15396,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.017463175455729167,
      "learning_rate": 0.0001,
      "loss": 4.3301,
      "loss/crossentropy": 2.3392014503479004,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26226382702589035,
      "step": 7698
    },
    {
      "epoch": 0.154,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.021320597330729166,
      "learning_rate": 0.0001,
      "loss": 4.0381,
      "loss/crossentropy": 1.7265403866767883,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21790936589241028,
      "step": 7700
    },
    {
      "epoch": 0.15404,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.02276585896809896,
      "learning_rate": 0.0001,
      "loss": 4.1725,
      "loss/crossentropy": 2.024384081363678,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21987473219633102,
      "step": 7702
    },
    {
      "epoch": 0.15408,
      "grad_norm": 2.5,
      "grad_norm_var": 0.03117650349934896,
      "learning_rate": 0.0001,
      "loss": 4.6702,
      "loss/crossentropy": 2.1840893030166626,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24231631308794022,
      "step": 7704
    },
    {
      "epoch": 0.15412,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.03001683553059896,
      "learning_rate": 0.0001,
      "loss": 4.2197,
      "loss/crossentropy": 2.1950928568840027,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21808429062366486,
      "step": 7706
    },
    {
      "epoch": 0.15416,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.03029352823893229,
      "learning_rate": 0.0001,
      "loss": 4.3215,
      "loss/crossentropy": 1.9541537165641785,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22109197825193405,
      "step": 7708
    },
    {
      "epoch": 0.1542,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0257476806640625,
      "learning_rate": 0.0001,
      "loss": 4.1966,
      "loss/crossentropy": 2.0232877135276794,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.214762382209301,
      "step": 7710
    },
    {
      "epoch": 0.15424,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.023859659830729168,
      "learning_rate": 0.0001,
      "loss": 4.0012,
      "loss/crossentropy": 2.003768503665924,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21820105612277985,
      "step": 7712
    },
    {
      "epoch": 0.15428,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.020026652018229167,
      "learning_rate": 0.0001,
      "loss": 4.2471,
      "loss/crossentropy": 2.007221221923828,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2203196883201599,
      "step": 7714
    },
    {
      "epoch": 0.15432,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.015900675455729166,
      "learning_rate": 0.0001,
      "loss": 4.3201,
      "loss/crossentropy": 2.0134615898132324,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21916545927524567,
      "step": 7716
    },
    {
      "epoch": 0.15436,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.016013336181640626,
      "learning_rate": 0.0001,
      "loss": 4.3821,
      "loss/crossentropy": 1.9012999534606934,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25758640468120575,
      "step": 7718
    },
    {
      "epoch": 0.1544,
      "grad_norm": 2.25,
      "grad_norm_var": 0.008727773030598959,
      "learning_rate": 0.0001,
      "loss": 4.1555,
      "loss/crossentropy": 2.074169874191284,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22667942196130753,
      "step": 7720
    },
    {
      "epoch": 0.15444,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.009059397379557292,
      "learning_rate": 0.0001,
      "loss": 4.4355,
      "loss/crossentropy": 2.070925295352936,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20353248715400696,
      "step": 7722
    },
    {
      "epoch": 0.15448,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.011163075764973959,
      "learning_rate": 0.0001,
      "loss": 4.5676,
      "loss/crossentropy": 2.289568066596985,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23175117373466492,
      "step": 7724
    },
    {
      "epoch": 0.15452,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.012827301025390625,
      "learning_rate": 0.0001,
      "loss": 4.6152,
      "loss/crossentropy": 2.21374249458313,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23585008084774017,
      "step": 7726
    },
    {
      "epoch": 0.15456,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.010791015625,
      "learning_rate": 0.0001,
      "loss": 4.5575,
      "loss/crossentropy": 2.15897136926651,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24143048375844955,
      "step": 7728
    },
    {
      "epoch": 0.1546,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.01142578125,
      "learning_rate": 0.0001,
      "loss": 4.4722,
      "loss/crossentropy": 2.134206771850586,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24344487488269806,
      "step": 7730
    },
    {
      "epoch": 0.15464,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.011180623372395834,
      "learning_rate": 0.0001,
      "loss": 4.5827,
      "loss/crossentropy": 2.3832513093948364,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24878299236297607,
      "step": 7732
    },
    {
      "epoch": 0.15468,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.0102203369140625,
      "learning_rate": 0.0001,
      "loss": 4.0722,
      "loss/crossentropy": 1.917544960975647,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20134451985359192,
      "step": 7734
    },
    {
      "epoch": 0.15472,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.011767578125,
      "learning_rate": 0.0001,
      "loss": 4.4499,
      "loss/crossentropy": 2.1081286668777466,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23577219247817993,
      "step": 7736
    },
    {
      "epoch": 0.15476,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.0137603759765625,
      "learning_rate": 0.0001,
      "loss": 4.7942,
      "loss/crossentropy": 2.214662790298462,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23744845390319824,
      "step": 7738
    },
    {
      "epoch": 0.1548,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.013895670572916666,
      "learning_rate": 0.0001,
      "loss": 4.4325,
      "loss/crossentropy": 1.995256781578064,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21037640422582626,
      "step": 7740
    },
    {
      "epoch": 0.15484,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.024006144205729166,
      "learning_rate": 0.0001,
      "loss": 4.0701,
      "loss/crossentropy": 2.2877765893936157,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23090071976184845,
      "step": 7742
    },
    {
      "epoch": 0.15488,
      "grad_norm": 2.125,
      "grad_norm_var": 0.022652180989583333,
      "learning_rate": 0.0001,
      "loss": 4.6167,
      "loss/crossentropy": 2.23935329914093,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24617131054401398,
      "step": 7744
    },
    {
      "epoch": 0.15492,
      "grad_norm": 2.125,
      "grad_norm_var": 0.023005167643229168,
      "learning_rate": 0.0001,
      "loss": 4.4886,
      "loss/crossentropy": 2.15006422996521,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2235095053911209,
      "step": 7746
    },
    {
      "epoch": 0.15496,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.024144490559895832,
      "learning_rate": 0.0001,
      "loss": 4.1873,
      "loss/crossentropy": 1.9917905926704407,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21606986224651337,
      "step": 7748
    },
    {
      "epoch": 0.155,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.025804646809895835,
      "learning_rate": 0.0001,
      "loss": 4.6347,
      "loss/crossentropy": 2.303179979324341,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2541813999414444,
      "step": 7750
    },
    {
      "epoch": 0.15504,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.022786458333333332,
      "learning_rate": 0.0001,
      "loss": 4.3647,
      "loss/crossentropy": 2.231510281562805,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2479737550020218,
      "step": 7752
    },
    {
      "epoch": 0.15508,
      "grad_norm": 2.25,
      "grad_norm_var": 0.01778132120768229,
      "learning_rate": 0.0001,
      "loss": 4.1714,
      "loss/crossentropy": 2.0530437231063843,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20220057666301727,
      "step": 7754
    },
    {
      "epoch": 0.15512,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.015909830729166668,
      "learning_rate": 0.0001,
      "loss": 4.1978,
      "loss/crossentropy": 2.0889222025871277,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22389977425336838,
      "step": 7756
    },
    {
      "epoch": 0.15516,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.0131256103515625,
      "learning_rate": 0.0001,
      "loss": 4.4748,
      "loss/crossentropy": 2.3807711601257324,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2397037297487259,
      "step": 7758
    },
    {
      "epoch": 0.1552,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.01297607421875,
      "learning_rate": 0.0001,
      "loss": 4.3382,
      "loss/crossentropy": 1.8144067525863647,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19052604585886002,
      "step": 7760
    },
    {
      "epoch": 0.15524,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.013240559895833334,
      "learning_rate": 0.0001,
      "loss": 4.5358,
      "loss/crossentropy": 2.295349955558777,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2168404459953308,
      "step": 7762
    },
    {
      "epoch": 0.15528,
      "grad_norm": 11.8125,
      "grad_norm_var": 5.868936920166016,
      "learning_rate": 0.0001,
      "loss": 4.1706,
      "loss/crossentropy": 1.7281805276870728,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.230300635099411,
      "step": 7764
    },
    {
      "epoch": 0.15532,
      "grad_norm": 2.375,
      "grad_norm_var": 5.861083730061849,
      "learning_rate": 0.0001,
      "loss": 4.3867,
      "loss/crossentropy": 2.2153135538101196,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24419061839580536,
      "step": 7766
    },
    {
      "epoch": 0.15536,
      "grad_norm": 2.03125,
      "grad_norm_var": 5.890169270833334,
      "learning_rate": 0.0001,
      "loss": 4.2047,
      "loss/crossentropy": 2.0259060859680176,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2119704708456993,
      "step": 7768
    },
    {
      "epoch": 0.1554,
      "grad_norm": 2.21875,
      "grad_norm_var": 5.871726226806641,
      "learning_rate": 0.0001,
      "loss": 4.3405,
      "loss/crossentropy": 2.2399297952651978,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23252833634614944,
      "step": 7770
    },
    {
      "epoch": 0.15544,
      "grad_norm": 2.109375,
      "grad_norm_var": 5.86380615234375,
      "learning_rate": 0.0001,
      "loss": 4.3287,
      "loss/crossentropy": 2.0974661111831665,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2085232511162758,
      "step": 7772
    },
    {
      "epoch": 0.15548,
      "grad_norm": 2.21875,
      "grad_norm_var": 5.861717732747396,
      "learning_rate": 0.0001,
      "loss": 4.4141,
      "loss/crossentropy": 2.0121108293533325,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24354346096515656,
      "step": 7774
    },
    {
      "epoch": 0.15552,
      "grad_norm": 2.171875,
      "grad_norm_var": 5.843431599934896,
      "learning_rate": 0.0001,
      "loss": 4.3167,
      "loss/crossentropy": 2.1463273763656616,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22797952592372894,
      "step": 7776
    },
    {
      "epoch": 0.15556,
      "grad_norm": 1.9140625,
      "grad_norm_var": 5.875705718994141,
      "learning_rate": 0.0001,
      "loss": 3.9814,
      "loss/crossentropy": 1.6362827122211456,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18291430547833443,
      "step": 7778
    },
    {
      "epoch": 0.1556,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.016283162434895835,
      "learning_rate": 0.0001,
      "loss": 4.3534,
      "loss/crossentropy": 2.4894620180130005,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24631594866514206,
      "step": 7780
    },
    {
      "epoch": 0.15564,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.0119781494140625,
      "learning_rate": 0.0001,
      "loss": 4.5799,
      "loss/crossentropy": 2.346967577934265,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20964853465557098,
      "step": 7782
    },
    {
      "epoch": 0.15568,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.010282135009765625,
      "learning_rate": 0.0001,
      "loss": 4.5036,
      "loss/crossentropy": 2.0165189504623413,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22410035878419876,
      "step": 7784
    },
    {
      "epoch": 0.15572,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.009417470296223958,
      "learning_rate": 0.0001,
      "loss": 4.0735,
      "loss/crossentropy": 1.7486848831176758,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20792805403470993,
      "step": 7786
    },
    {
      "epoch": 0.15576,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009905751546223958,
      "learning_rate": 0.0001,
      "loss": 4.1321,
      "loss/crossentropy": 1.9615037441253662,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22578728944063187,
      "step": 7788
    },
    {
      "epoch": 0.1558,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.009069569905598958,
      "learning_rate": 0.0001,
      "loss": 4.326,
      "loss/crossentropy": 1.8386783003807068,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21898606419563293,
      "step": 7790
    },
    {
      "epoch": 0.15584,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.011533355712890625,
      "learning_rate": 0.0001,
      "loss": 4.6429,
      "loss/crossentropy": 2.1383039951324463,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2322411835193634,
      "step": 7792
    },
    {
      "epoch": 0.15588,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.008918253580729167,
      "learning_rate": 0.0001,
      "loss": 4.3568,
      "loss/crossentropy": 2.510488271713257,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24672146886587143,
      "step": 7794
    },
    {
      "epoch": 0.15592,
      "grad_norm": 2.125,
      "grad_norm_var": 0.009309895833333333,
      "learning_rate": 0.0001,
      "loss": 4.4328,
      "loss/crossentropy": 2.03993421792984,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23064473271369934,
      "step": 7796
    },
    {
      "epoch": 0.15596,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.007306925455729167,
      "learning_rate": 0.0001,
      "loss": 4.4833,
      "loss/crossentropy": 2.305809736251831,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21916814893484116,
      "step": 7798
    },
    {
      "epoch": 0.156,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.007796223958333333,
      "learning_rate": 0.0001,
      "loss": 4.1218,
      "loss/crossentropy": 1.8330454230308533,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20832987129688263,
      "step": 7800
    },
    {
      "epoch": 0.15604,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.007420857747395833,
      "learning_rate": 0.0001,
      "loss": 4.2579,
      "loss/crossentropy": 1.9194663166999817,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22163349390029907,
      "step": 7802
    },
    {
      "epoch": 0.15608,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.006917317708333333,
      "learning_rate": 0.0001,
      "loss": 4.2219,
      "loss/crossentropy": 1.798878252506256,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19532181322574615,
      "step": 7804
    },
    {
      "epoch": 0.15612,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0067942301432291664,
      "learning_rate": 0.0001,
      "loss": 4.2949,
      "loss/crossentropy": 1.730432152748108,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20092248916625977,
      "step": 7806
    },
    {
      "epoch": 0.15616,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.01304931640625,
      "learning_rate": 0.0001,
      "loss": 4.3109,
      "loss/crossentropy": 2.1426968574523926,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22185371816158295,
      "step": 7808
    },
    {
      "epoch": 0.1562,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.012848917643229167,
      "learning_rate": 0.0001,
      "loss": 4.4308,
      "loss/crossentropy": 1.983969271183014,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22203146666288376,
      "step": 7810
    },
    {
      "epoch": 0.15624,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.016532389322916667,
      "learning_rate": 0.0001,
      "loss": 4.2114,
      "loss/crossentropy": 2.2948192954063416,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23453453928232193,
      "step": 7812
    },
    {
      "epoch": 0.15628,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0175933837890625,
      "learning_rate": 0.0001,
      "loss": 4.287,
      "loss/crossentropy": 1.9190048575401306,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22457829862833023,
      "step": 7814
    },
    {
      "epoch": 0.15632,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.01529541015625,
      "learning_rate": 0.0001,
      "loss": 4.1721,
      "loss/crossentropy": 2.0268847346305847,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21628276258707047,
      "step": 7816
    },
    {
      "epoch": 0.15636,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.017411295572916666,
      "learning_rate": 0.0001,
      "loss": 4.3553,
      "loss/crossentropy": 2.0050706267356873,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20108875632286072,
      "step": 7818
    },
    {
      "epoch": 0.1564,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.024470774332682292,
      "learning_rate": 0.0001,
      "loss": 4.2059,
      "loss/crossentropy": 2.096635937690735,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22623063623905182,
      "step": 7820
    },
    {
      "epoch": 0.15644,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.024580637613932293,
      "learning_rate": 0.0001,
      "loss": 4.3901,
      "loss/crossentropy": 1.9188768863677979,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21088901162147522,
      "step": 7822
    },
    {
      "epoch": 0.15648,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.018304189046223957,
      "learning_rate": 0.0001,
      "loss": 4.4062,
      "loss/crossentropy": 2.3639817237854004,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2608294039964676,
      "step": 7824
    },
    {
      "epoch": 0.15652,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.017765045166015625,
      "learning_rate": 0.0001,
      "loss": 4.2123,
      "loss/crossentropy": 1.9716956615447998,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23327408730983734,
      "step": 7826
    },
    {
      "epoch": 0.15656,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.08131688435872396,
      "learning_rate": 0.0001,
      "loss": 4.1377,
      "loss/crossentropy": 2.004276990890503,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22930586338043213,
      "step": 7828
    },
    {
      "epoch": 0.1566,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.08247858683268229,
      "learning_rate": 0.0001,
      "loss": 4.4703,
      "loss/crossentropy": 1.7996181845664978,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21030305325984955,
      "step": 7830
    },
    {
      "epoch": 0.15664,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.09555435180664062,
      "learning_rate": 0.0001,
      "loss": 4.7938,
      "loss/crossentropy": 2.192178189754486,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2232302725315094,
      "step": 7832
    },
    {
      "epoch": 0.15668,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.09538345336914063,
      "learning_rate": 0.0001,
      "loss": 4.2381,
      "loss/crossentropy": 1.7093925476074219,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19697313755750656,
      "step": 7834
    },
    {
      "epoch": 0.15672,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.09130859375,
      "learning_rate": 0.0001,
      "loss": 4.2534,
      "loss/crossentropy": 1.915247917175293,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21003933250904083,
      "step": 7836
    },
    {
      "epoch": 0.15676,
      "grad_norm": 2.375,
      "grad_norm_var": 0.091162109375,
      "learning_rate": 0.0001,
      "loss": 4.3825,
      "loss/crossentropy": 2.188641667366028,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23260314762592316,
      "step": 7838
    },
    {
      "epoch": 0.1568,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.09501953125,
      "learning_rate": 0.0001,
      "loss": 4.7329,
      "loss/crossentropy": 2.3316495418548584,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24601806700229645,
      "step": 7840
    },
    {
      "epoch": 0.15684,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.0918121337890625,
      "learning_rate": 0.0001,
      "loss": 4.2934,
      "loss/crossentropy": 2.140946924686432,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23532958328723907,
      "step": 7842
    },
    {
      "epoch": 0.15688,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.031525675455729166,
      "learning_rate": 0.0001,
      "loss": 4.2738,
      "loss/crossentropy": 2.372095465660095,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2615740895271301,
      "step": 7844
    },
    {
      "epoch": 0.15692,
      "grad_norm": 2.25,
      "grad_norm_var": 0.028727213541666668,
      "learning_rate": 0.0001,
      "loss": 4.1954,
      "loss/crossentropy": 1.8433185815811157,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2160269021987915,
      "step": 7846
    },
    {
      "epoch": 0.15696,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.016071573893229166,
      "learning_rate": 0.0001,
      "loss": 4.2489,
      "loss/crossentropy": 2.012324333190918,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1964300200343132,
      "step": 7848
    },
    {
      "epoch": 0.157,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.014090983072916667,
      "learning_rate": 0.0001,
      "loss": 4.486,
      "loss/crossentropy": 2.0325884222984314,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22080926597118378,
      "step": 7850
    },
    {
      "epoch": 0.15704,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.013036092122395834,
      "learning_rate": 0.0001,
      "loss": 4.3784,
      "loss/crossentropy": 2.3786104917526245,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24448946118354797,
      "step": 7852
    },
    {
      "epoch": 0.15708,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.011693318684895834,
      "learning_rate": 0.0001,
      "loss": 4.5173,
      "loss/crossentropy": 2.304913640022278,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24708375334739685,
      "step": 7854
    },
    {
      "epoch": 0.15712,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.006371053059895834,
      "learning_rate": 0.0001,
      "loss": 4.5711,
      "loss/crossentropy": 1.8640215396881104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19824489206075668,
      "step": 7856
    },
    {
      "epoch": 0.15716,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.00625,
      "learning_rate": 0.0001,
      "loss": 4.6174,
      "loss/crossentropy": 2.444548487663269,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2542492523789406,
      "step": 7858
    },
    {
      "epoch": 0.1572,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.00592041015625,
      "learning_rate": 0.0001,
      "loss": 4.0522,
      "loss/crossentropy": 1.991346299648285,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2159758359193802,
      "step": 7860
    },
    {
      "epoch": 0.15724,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.005101521809895833,
      "learning_rate": 0.0001,
      "loss": 4.2627,
      "loss/crossentropy": 2.3172048926353455,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2506742626428604,
      "step": 7862
    },
    {
      "epoch": 0.15728,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.0050201416015625,
      "learning_rate": 0.0001,
      "loss": 4.5415,
      "loss/crossentropy": 1.9302632212638855,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2104542776942253,
      "step": 7864
    },
    {
      "epoch": 0.15732,
      "grad_norm": 2.125,
      "grad_norm_var": 0.020726521809895832,
      "learning_rate": 0.0001,
      "loss": 4.7831,
      "loss/crossentropy": 2.457883358001709,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23538483679294586,
      "step": 7866
    },
    {
      "epoch": 0.15736,
      "grad_norm": 2.125,
      "grad_norm_var": 0.019677734375,
      "learning_rate": 0.0001,
      "loss": 4.1912,
      "loss/crossentropy": 1.9857566952705383,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2131117358803749,
      "step": 7868
    },
    {
      "epoch": 0.1574,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.019140625,
      "learning_rate": 0.0001,
      "loss": 4.4264,
      "loss/crossentropy": 2.0062127113342285,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22806233912706375,
      "step": 7870
    },
    {
      "epoch": 0.15744,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.019489542643229166,
      "learning_rate": 0.0001,
      "loss": 4.6181,
      "loss/crossentropy": 2.283127784729004,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23883548378944397,
      "step": 7872
    },
    {
      "epoch": 0.15748,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.019136555989583335,
      "learning_rate": 0.0001,
      "loss": 4.4442,
      "loss/crossentropy": 2.110979437828064,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.209333136677742,
      "step": 7874
    },
    {
      "epoch": 0.15752,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.019856770833333332,
      "learning_rate": 0.0001,
      "loss": 4.2107,
      "loss/crossentropy": 1.8705166578292847,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20791998505592346,
      "step": 7876
    },
    {
      "epoch": 0.15756,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.019017537434895832,
      "learning_rate": 0.0001,
      "loss": 4.2956,
      "loss/crossentropy": 2.133803129196167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23219672590494156,
      "step": 7878
    },
    {
      "epoch": 0.1576,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.019627888997395832,
      "learning_rate": 0.0001,
      "loss": 4.5028,
      "loss/crossentropy": 2.1062549352645874,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2266012579202652,
      "step": 7880
    },
    {
      "epoch": 0.15764,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.0058553059895833336,
      "learning_rate": 0.0001,
      "loss": 4.5078,
      "loss/crossentropy": 2.0088155269622803,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2079932913184166,
      "step": 7882
    },
    {
      "epoch": 0.15768,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.0065582275390625,
      "learning_rate": 0.0001,
      "loss": 3.9204,
      "loss/crossentropy": 1.6621176600456238,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19370710104703903,
      "step": 7884
    },
    {
      "epoch": 0.15772,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.007136027018229167,
      "learning_rate": 0.0001,
      "loss": 4.3462,
      "loss/crossentropy": 1.7729167938232422,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18562481552362442,
      "step": 7886
    },
    {
      "epoch": 0.15776,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.006843058268229166,
      "learning_rate": 0.0001,
      "loss": 4.3897,
      "loss/crossentropy": 2.132485508918762,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23432063311338425,
      "step": 7888
    },
    {
      "epoch": 0.1578,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.0091949462890625,
      "learning_rate": 0.0001,
      "loss": 4.3731,
      "loss/crossentropy": 2.122144937515259,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23042195290327072,
      "step": 7890
    },
    {
      "epoch": 0.15784,
      "grad_norm": 2.8125,
      "grad_norm_var": 0.034956868489583334,
      "learning_rate": 0.0001,
      "loss": 4.4025,
      "loss/crossentropy": 1.855578601360321,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20124691724777222,
      "step": 7892
    },
    {
      "epoch": 0.15788,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.035054524739583336,
      "learning_rate": 0.0001,
      "loss": 4.172,
      "loss/crossentropy": 2.02128005027771,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20766886323690414,
      "step": 7894
    },
    {
      "epoch": 0.15792,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.03681233723958333,
      "learning_rate": 0.0001,
      "loss": 4.3096,
      "loss/crossentropy": 2.112824857234955,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22250613570213318,
      "step": 7896
    },
    {
      "epoch": 0.15796,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.03662821451822917,
      "learning_rate": 0.0001,
      "loss": 4.5595,
      "loss/crossentropy": 2.2290207147598267,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2531846910715103,
      "step": 7898
    },
    {
      "epoch": 0.158,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.03882548014322917,
      "learning_rate": 0.0001,
      "loss": 4.3586,
      "loss/crossentropy": 2.135373592376709,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2266940325498581,
      "step": 7900
    },
    {
      "epoch": 0.15804,
      "grad_norm": 2.125,
      "grad_norm_var": 0.04052734375,
      "learning_rate": 0.0001,
      "loss": 4.1719,
      "loss/crossentropy": 1.7298616170883179,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1986929401755333,
      "step": 7902
    },
    {
      "epoch": 0.15808,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.04173075358072917,
      "learning_rate": 0.0001,
      "loss": 4.403,
      "loss/crossentropy": 2.18759286403656,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21622422337532043,
      "step": 7904
    },
    {
      "epoch": 0.15812,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.03889058430989583,
      "learning_rate": 0.0001,
      "loss": 4.4765,
      "loss/crossentropy": 2.0889216661453247,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21440055221319199,
      "step": 7906
    },
    {
      "epoch": 0.15816,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.010204060872395834,
      "learning_rate": 0.0001,
      "loss": 4.2231,
      "loss/crossentropy": 1.7791658639907837,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1967781037092209,
      "step": 7908
    },
    {
      "epoch": 0.1582,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.012386067708333334,
      "learning_rate": 0.0001,
      "loss": 4.2314,
      "loss/crossentropy": 2.2144845724105835,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22152486443519592,
      "step": 7910
    },
    {
      "epoch": 0.15824,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.012723795572916667,
      "learning_rate": 0.0001,
      "loss": 4.6236,
      "loss/crossentropy": 2.316117286682129,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24549901485443115,
      "step": 7912
    },
    {
      "epoch": 0.15828,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0115875244140625,
      "learning_rate": 0.0001,
      "loss": 4.0867,
      "loss/crossentropy": 1.8642511367797852,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22441796958446503,
      "step": 7914
    },
    {
      "epoch": 0.15832,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0098541259765625,
      "learning_rate": 0.0001,
      "loss": 4.1365,
      "loss/crossentropy": 1.807969868183136,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22138798981904984,
      "step": 7916
    },
    {
      "epoch": 0.15836,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.013866170247395834,
      "learning_rate": 0.0001,
      "loss": 4.5625,
      "loss/crossentropy": 2.266697645187378,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22544697672128677,
      "step": 7918
    },
    {
      "epoch": 0.1584,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.012262980143229166,
      "learning_rate": 0.0001,
      "loss": 4.5632,
      "loss/crossentropy": 2.0871587991714478,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21997228264808655,
      "step": 7920
    },
    {
      "epoch": 0.15844,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.0126861572265625,
      "learning_rate": 0.0001,
      "loss": 4.5567,
      "loss/crossentropy": 2.1993759870529175,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2416691780090332,
      "step": 7922
    },
    {
      "epoch": 0.15848,
      "grad_norm": 2.5,
      "grad_norm_var": 0.020340983072916666,
      "learning_rate": 0.0001,
      "loss": 4.2997,
      "loss/crossentropy": 1.8039653897285461,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21721573173999786,
      "step": 7924
    },
    {
      "epoch": 0.15852,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.0225982666015625,
      "learning_rate": 0.0001,
      "loss": 4.4795,
      "loss/crossentropy": 2.188117265701294,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.231883242726326,
      "step": 7926
    },
    {
      "epoch": 0.15856,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.021923828125,
      "learning_rate": 0.0001,
      "loss": 4.6694,
      "loss/crossentropy": 2.3920425176620483,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24655026197433472,
      "step": 7928
    },
    {
      "epoch": 0.1586,
      "grad_norm": 2.125,
      "grad_norm_var": 0.017561848958333334,
      "learning_rate": 0.0001,
      "loss": 4.1001,
      "loss/crossentropy": 2.286831498146057,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23398208618164062,
      "step": 7930
    },
    {
      "epoch": 0.15864,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.016825358072916668,
      "learning_rate": 0.0001,
      "loss": 4.0007,
      "loss/crossentropy": 2.075824797153473,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22423933446407318,
      "step": 7932
    },
    {
      "epoch": 0.15868,
      "grad_norm": 2.25,
      "grad_norm_var": 0.0155426025390625,
      "learning_rate": 0.0001,
      "loss": 4.5697,
      "loss/crossentropy": 2.197165012359619,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22215355187654495,
      "step": 7934
    },
    {
      "epoch": 0.15872,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.017854817708333335,
      "learning_rate": 0.0001,
      "loss": 4.2899,
      "loss/crossentropy": 1.8253535032272339,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20457974076271057,
      "step": 7936
    },
    {
      "epoch": 0.15876,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.01783447265625,
      "learning_rate": 0.0001,
      "loss": 4.4774,
      "loss/crossentropy": 1.842383086681366,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20742817968130112,
      "step": 7938
    },
    {
      "epoch": 0.1588,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.014208984375,
      "learning_rate": 0.0001,
      "loss": 4.3801,
      "loss/crossentropy": 2.2086315155029297,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2086438685655594,
      "step": 7940
    },
    {
      "epoch": 0.15884,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0067047119140625,
      "learning_rate": 0.0001,
      "loss": 4.3718,
      "loss/crossentropy": 2.3381282091140747,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2546956539154053,
      "step": 7942
    },
    {
      "epoch": 0.15888,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.005248006184895833,
      "learning_rate": 0.0001,
      "loss": 4.3439,
      "loss/crossentropy": 2.1159931421279907,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23917824029922485,
      "step": 7944
    },
    {
      "epoch": 0.15892,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0069081624348958336,
      "learning_rate": 0.0001,
      "loss": 4.2056,
      "loss/crossentropy": 1.7507159113883972,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1870383694767952,
      "step": 7946
    },
    {
      "epoch": 0.15896,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0073964436848958336,
      "learning_rate": 0.0001,
      "loss": 4.2118,
      "loss/crossentropy": 2.213807225227356,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22211921215057373,
      "step": 7948
    },
    {
      "epoch": 0.159,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009598795572916667,
      "learning_rate": 0.0001,
      "loss": 4.4316,
      "loss/crossentropy": 2.0570366978645325,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2662791311740875,
      "step": 7950
    },
    {
      "epoch": 0.15904,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.009748331705729167,
      "learning_rate": 0.0001,
      "loss": 4.2041,
      "loss/crossentropy": 1.8775206208229065,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22045740485191345,
      "step": 7952
    },
    {
      "epoch": 0.15908,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.009601847330729166,
      "learning_rate": 0.0001,
      "loss": 4.2575,
      "loss/crossentropy": 2.3483108282089233,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22626785188913345,
      "step": 7954
    },
    {
      "epoch": 0.15912,
      "grad_norm": 2.125,
      "grad_norm_var": 0.009496053059895834,
      "learning_rate": 0.0001,
      "loss": 4.193,
      "loss/crossentropy": 1.9531084895133972,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22194421291351318,
      "step": 7956
    },
    {
      "epoch": 0.15916,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.011213175455729167,
      "learning_rate": 0.0001,
      "loss": 4.4819,
      "loss/crossentropy": 2.057813823223114,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21116723865270615,
      "step": 7958
    },
    {
      "epoch": 0.1592,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.01148681640625,
      "learning_rate": 0.0001,
      "loss": 4.3478,
      "loss/crossentropy": 1.9398415088653564,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2257898524403572,
      "step": 7960
    },
    {
      "epoch": 0.15924,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0111480712890625,
      "learning_rate": 0.0001,
      "loss": 4.1942,
      "loss/crossentropy": 1.9200270175933838,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2119666188955307,
      "step": 7962
    },
    {
      "epoch": 0.15928,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.010640462239583334,
      "learning_rate": 0.0001,
      "loss": 4.4513,
      "loss/crossentropy": 2.157149076461792,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23271384835243225,
      "step": 7964
    },
    {
      "epoch": 0.15932,
      "grad_norm": 2.25,
      "grad_norm_var": 0.0083404541015625,
      "learning_rate": 0.0001,
      "loss": 4.2064,
      "loss/crossentropy": 2.0308582186698914,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23934553563594818,
      "step": 7966
    },
    {
      "epoch": 0.15936,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.015543619791666666,
      "learning_rate": 0.0001,
      "loss": 4.3977,
      "loss/crossentropy": 2.1855711936950684,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.236568883061409,
      "step": 7968
    },
    {
      "epoch": 0.1594,
      "grad_norm": 2.375,
      "grad_norm_var": 0.07629292805989583,
      "learning_rate": 0.0001,
      "loss": 4.719,
      "loss/crossentropy": 2.4479551315307617,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24693088978528976,
      "step": 7970
    },
    {
      "epoch": 0.15944,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.07333882649739583,
      "learning_rate": 0.0001,
      "loss": 4.5221,
      "loss/crossentropy": 2.439974784851074,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2502682954072952,
      "step": 7972
    },
    {
      "epoch": 0.15948,
      "grad_norm": 2.375,
      "grad_norm_var": 0.07625325520833333,
      "learning_rate": 0.0001,
      "loss": 4.4024,
      "loss/crossentropy": 1.9899010062217712,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23031124472618103,
      "step": 7974
    },
    {
      "epoch": 0.15952,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.07517903645833333,
      "learning_rate": 0.0001,
      "loss": 4.2522,
      "loss/crossentropy": 1.830255150794983,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1984459012746811,
      "step": 7976
    },
    {
      "epoch": 0.15956,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.07088114420572916,
      "learning_rate": 0.0001,
      "loss": 4.3832,
      "loss/crossentropy": 1.9675705432891846,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21336429566144943,
      "step": 7978
    },
    {
      "epoch": 0.1596,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.06965738932291667,
      "learning_rate": 0.0001,
      "loss": 4.4784,
      "loss/crossentropy": 2.030815005302429,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2102503478527069,
      "step": 7980
    },
    {
      "epoch": 0.15964,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.07285054524739583,
      "learning_rate": 0.0001,
      "loss": 4.492,
      "loss/crossentropy": 2.4932440519332886,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.246867336332798,
      "step": 7982
    },
    {
      "epoch": 0.15968,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.07330322265625,
      "learning_rate": 0.0001,
      "loss": 4.2085,
      "loss/crossentropy": 1.5839802622795105,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17231802642345428,
      "step": 7984
    },
    {
      "epoch": 0.15972,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.0091217041015625,
      "learning_rate": 0.0001,
      "loss": 4.4045,
      "loss/crossentropy": 1.821477472782135,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2239791452884674,
      "step": 7986
    },
    {
      "epoch": 0.15976,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0093414306640625,
      "learning_rate": 0.0001,
      "loss": 4.0655,
      "loss/crossentropy": 2.013838052749634,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2335103377699852,
      "step": 7988
    },
    {
      "epoch": 0.1598,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.006376139322916667,
      "learning_rate": 0.0001,
      "loss": 4.4989,
      "loss/crossentropy": 1.9412779211997986,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20336110144853592,
      "step": 7990
    },
    {
      "epoch": 0.15984,
      "grad_norm": 2.0,
      "grad_norm_var": 0.008234659830729166,
      "learning_rate": 0.0001,
      "loss": 3.9015,
      "loss/crossentropy": 1.6653677225112915,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2025909200310707,
      "step": 7992
    },
    {
      "epoch": 0.15988,
      "grad_norm": 2.125,
      "grad_norm_var": 0.00865478515625,
      "learning_rate": 0.0001,
      "loss": 4.4299,
      "loss/crossentropy": 2.0069726705551147,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21300900727510452,
      "step": 7994
    },
    {
      "epoch": 0.15992,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.009919230143229167,
      "learning_rate": 0.0001,
      "loss": 4.4395,
      "loss/crossentropy": 2.1118472814559937,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25853876769542694,
      "step": 7996
    },
    {
      "epoch": 0.15996,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.013622029622395834,
      "learning_rate": 0.0001,
      "loss": 4.1922,
      "loss/crossentropy": 1.608262836933136,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1958206593990326,
      "step": 7998
    },
    {
      "epoch": 0.16,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.0138580322265625,
      "learning_rate": 0.0001,
      "loss": 4.2896,
      "loss/crossentropy": 1.6572073101997375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19102878868579865,
      "step": 8000
    },
    {
      "epoch": 0.16004,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0113677978515625,
      "learning_rate": 0.0001,
      "loss": 4.0811,
      "loss/crossentropy": 1.9421688318252563,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19628942012786865,
      "step": 8002
    },
    {
      "epoch": 0.16008,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.0134918212890625,
      "learning_rate": 0.0001,
      "loss": 4.5344,
      "loss/crossentropy": 2.2197489738464355,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.248014435172081,
      "step": 8004
    },
    {
      "epoch": 0.16012,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.0126617431640625,
      "learning_rate": 0.0001,
      "loss": 4.4625,
      "loss/crossentropy": 2.200868308544159,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2476908192038536,
      "step": 8006
    },
    {
      "epoch": 0.16016,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.01162109375,
      "learning_rate": 0.0001,
      "loss": 4.8151,
      "loss/crossentropy": 2.3793649673461914,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2356039509177208,
      "step": 8008
    },
    {
      "epoch": 0.1602,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0112945556640625,
      "learning_rate": 0.0001,
      "loss": 4.1849,
      "loss/crossentropy": 2.130257308483124,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.224358968436718,
      "step": 8010
    },
    {
      "epoch": 0.16024,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.009566243489583333,
      "learning_rate": 0.0001,
      "loss": 4.5948,
      "loss/crossentropy": 2.370365023612976,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23811528086662292,
      "step": 8012
    },
    {
      "epoch": 0.16028,
      "grad_norm": 2.0,
      "grad_norm_var": 0.008812459309895833,
      "learning_rate": 0.0001,
      "loss": 4.4326,
      "loss/crossentropy": 1.985486626625061,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19639353454113007,
      "step": 8014
    },
    {
      "epoch": 0.16032,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.01129150390625,
      "learning_rate": 0.0001,
      "loss": 4.4335,
      "loss/crossentropy": 2.2128632068634033,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2340361252427101,
      "step": 8016
    },
    {
      "epoch": 0.16036,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009956868489583333,
      "learning_rate": 0.0001,
      "loss": 4.3688,
      "loss/crossentropy": 1.830498456954956,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19760622829198837,
      "step": 8018
    },
    {
      "epoch": 0.1604,
      "grad_norm": 4.375,
      "grad_norm_var": 0.3061757405598958,
      "learning_rate": 0.0001,
      "loss": 4.6443,
      "loss/crossentropy": 1.9595977067947388,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2138657420873642,
      "step": 8020
    },
    {
      "epoch": 0.16044,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.30684305826822916,
      "learning_rate": 0.0001,
      "loss": 4.3101,
      "loss/crossentropy": 2.400893449783325,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2615668326616287,
      "step": 8022
    },
    {
      "epoch": 0.16048,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.3080963134765625,
      "learning_rate": 0.0001,
      "loss": 4.6359,
      "loss/crossentropy": 2.5079843997955322,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.251235693693161,
      "step": 8024
    },
    {
      "epoch": 0.16052,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.30686848958333335,
      "learning_rate": 0.0001,
      "loss": 4.4973,
      "loss/crossentropy": 2.573891043663025,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2646481841802597,
      "step": 8026
    },
    {
      "epoch": 0.16056,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.3064605712890625,
      "learning_rate": 0.0001,
      "loss": 4.4177,
      "loss/crossentropy": 2.0088363885879517,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22888045758008957,
      "step": 8028
    },
    {
      "epoch": 0.1606,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.31115697224934896,
      "learning_rate": 0.0001,
      "loss": 4.214,
      "loss/crossentropy": 2.4596647024154663,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2507014721632004,
      "step": 8030
    },
    {
      "epoch": 0.16064,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.3093462626139323,
      "learning_rate": 0.0001,
      "loss": 4.6826,
      "loss/crossentropy": 2.204255223274231,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22913406044244766,
      "step": 8032
    },
    {
      "epoch": 0.16068,
      "grad_norm": 2.125,
      "grad_norm_var": 0.30677261352539065,
      "learning_rate": 0.0001,
      "loss": 4.6603,
      "loss/crossentropy": 2.285408139228821,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23185817897319794,
      "step": 8034
    },
    {
      "epoch": 0.16072,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.012520345052083333,
      "learning_rate": 0.0001,
      "loss": 3.9513,
      "loss/crossentropy": 2.0788660645484924,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20811481028795242,
      "step": 8036
    },
    {
      "epoch": 0.16076,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.01307373046875,
      "learning_rate": 0.0001,
      "loss": 4.5368,
      "loss/crossentropy": 2.398258686065674,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2411409318447113,
      "step": 8038
    },
    {
      "epoch": 0.1608,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.015579986572265624,
      "learning_rate": 0.0001,
      "loss": 4.0415,
      "loss/crossentropy": 2.3101454973220825,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22493668645620346,
      "step": 8040
    },
    {
      "epoch": 0.16084,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.015134429931640625,
      "learning_rate": 0.0001,
      "loss": 4.0612,
      "loss/crossentropy": 1.9015105962753296,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22360052913427353,
      "step": 8042
    },
    {
      "epoch": 0.16088,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.017704010009765625,
      "learning_rate": 0.0001,
      "loss": 4.216,
      "loss/crossentropy": 2.0112481117248535,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24062782526016235,
      "step": 8044
    },
    {
      "epoch": 0.16092,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.0161773681640625,
      "learning_rate": 0.0001,
      "loss": 4.3638,
      "loss/crossentropy": 2.2024285793304443,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21959003806114197,
      "step": 8046
    },
    {
      "epoch": 0.16096,
      "grad_norm": 2.125,
      "grad_norm_var": 0.014412434895833333,
      "learning_rate": 0.0001,
      "loss": 4.4825,
      "loss/crossentropy": 2.1069058775901794,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21980682760477066,
      "step": 8048
    },
    {
      "epoch": 0.161,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.012540690104166667,
      "learning_rate": 0.0001,
      "loss": 4.2106,
      "loss/crossentropy": 1.8380340337753296,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2201671525835991,
      "step": 8050
    },
    {
      "epoch": 0.16104,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.010965728759765625,
      "learning_rate": 0.0001,
      "loss": 4.3778,
      "loss/crossentropy": 2.276741087436676,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23787930607795715,
      "step": 8052
    },
    {
      "epoch": 0.16108,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.010680898030598959,
      "learning_rate": 0.0001,
      "loss": 4.0784,
      "loss/crossentropy": 1.631809651851654,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19654212146997452,
      "step": 8054
    },
    {
      "epoch": 0.16112,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.007453409830729166,
      "learning_rate": 0.0001,
      "loss": 4.5927,
      "loss/crossentropy": 2.067444145679474,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21420229971408844,
      "step": 8056
    },
    {
      "epoch": 0.16116,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.011767578125,
      "learning_rate": 0.0001,
      "loss": 4.562,
      "loss/crossentropy": 2.384890556335449,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2312106341123581,
      "step": 8058
    },
    {
      "epoch": 0.1612,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.010986328125,
      "learning_rate": 0.0001,
      "loss": 4.363,
      "loss/crossentropy": 2.3683160543441772,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2527216002345085,
      "step": 8060
    },
    {
      "epoch": 0.16124,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.011253865559895833,
      "learning_rate": 0.0001,
      "loss": 4.4576,
      "loss/crossentropy": 2.1545952558517456,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23928315192461014,
      "step": 8062
    },
    {
      "epoch": 0.16128,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.013695271809895833,
      "learning_rate": 0.0001,
      "loss": 4.2285,
      "loss/crossentropy": 2.0792208313941956,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2226722463965416,
      "step": 8064
    },
    {
      "epoch": 0.16132,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.013068644205729167,
      "learning_rate": 0.0001,
      "loss": 3.8788,
      "loss/crossentropy": 2.181519627571106,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2193506434559822,
      "step": 8066
    },
    {
      "epoch": 0.16136,
      "grad_norm": 2.375,
      "grad_norm_var": 0.015067545572916667,
      "learning_rate": 0.0001,
      "loss": 4.4445,
      "loss/crossentropy": 1.7798657417297363,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23413337767124176,
      "step": 8068
    },
    {
      "epoch": 0.1614,
      "grad_norm": 2.125,
      "grad_norm_var": 0.017235310872395833,
      "learning_rate": 0.0001,
      "loss": 4.0731,
      "loss/crossentropy": 2.1232666969299316,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23351240158081055,
      "step": 8070
    },
    {
      "epoch": 0.16144,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.017609659830729166,
      "learning_rate": 0.0001,
      "loss": 4.6975,
      "loss/crossentropy": 2.34002685546875,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2533388137817383,
      "step": 8072
    },
    {
      "epoch": 0.16148,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.011637369791666666,
      "learning_rate": 0.0001,
      "loss": 4.3487,
      "loss/crossentropy": 2.066399872303009,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21156150847673416,
      "step": 8074
    },
    {
      "epoch": 0.16152,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.010677083333333334,
      "learning_rate": 0.0001,
      "loss": 4.3646,
      "loss/crossentropy": 2.2298463582992554,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2307809516787529,
      "step": 8076
    },
    {
      "epoch": 0.16156,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0105865478515625,
      "learning_rate": 0.0001,
      "loss": 4.1946,
      "loss/crossentropy": 1.9858508110046387,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21499747782945633,
      "step": 8078
    },
    {
      "epoch": 0.1616,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.011149088541666666,
      "learning_rate": 0.0001,
      "loss": 3.9984,
      "loss/crossentropy": 1.5669215321540833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1889330968260765,
      "step": 8080
    },
    {
      "epoch": 0.16164,
      "grad_norm": 2.25,
      "grad_norm_var": 0.011881510416666666,
      "learning_rate": 0.0001,
      "loss": 4.4743,
      "loss/crossentropy": 2.296878218650818,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2405889928340912,
      "step": 8082
    },
    {
      "epoch": 0.16168,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.007307942708333333,
      "learning_rate": 0.0001,
      "loss": 4.2205,
      "loss/crossentropy": 1.954626441001892,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19809379428625107,
      "step": 8084
    },
    {
      "epoch": 0.16172,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.008687337239583334,
      "learning_rate": 0.0001,
      "loss": 4.5923,
      "loss/crossentropy": 1.8640353083610535,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.226090669631958,
      "step": 8086
    },
    {
      "epoch": 0.16176,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.009663899739583334,
      "learning_rate": 0.0001,
      "loss": 4.3363,
      "loss/crossentropy": 2.1914591789245605,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24432511627674103,
      "step": 8088
    },
    {
      "epoch": 0.1618,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.009798177083333333,
      "learning_rate": 0.0001,
      "loss": 4.7291,
      "loss/crossentropy": 2.192594051361084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23682628571987152,
      "step": 8090
    },
    {
      "epoch": 0.16184,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.010993448893229167,
      "learning_rate": 0.0001,
      "loss": 4.4592,
      "loss/crossentropy": 2.210235595703125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.230379119515419,
      "step": 8092
    },
    {
      "epoch": 0.16188,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.011115519205729167,
      "learning_rate": 0.0001,
      "loss": 4.4111,
      "loss/crossentropy": 2.214667320251465,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26643867790699005,
      "step": 8094
    },
    {
      "epoch": 0.16192,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.009586588541666666,
      "learning_rate": 0.0001,
      "loss": 4.3131,
      "loss/crossentropy": 1.9808599948883057,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20679324120283127,
      "step": 8096
    },
    {
      "epoch": 0.16196,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.008333333333333333,
      "learning_rate": 0.0001,
      "loss": 4.4683,
      "loss/crossentropy": 2.076589345932007,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23430196940898895,
      "step": 8098
    },
    {
      "epoch": 0.162,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.006883748372395833,
      "learning_rate": 0.0001,
      "loss": 4.5645,
      "loss/crossentropy": 2.364492177963257,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23023054748773575,
      "step": 8100
    },
    {
      "epoch": 0.16204,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.005890909830729167,
      "learning_rate": 0.0001,
      "loss": 4.6595,
      "loss/crossentropy": 2.2908111214637756,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21239468455314636,
      "step": 8102
    },
    {
      "epoch": 0.16208,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.008958943684895833,
      "learning_rate": 0.0001,
      "loss": 4.5503,
      "loss/crossentropy": 1.8306183218955994,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21511316299438477,
      "step": 8104
    },
    {
      "epoch": 0.16212,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.010724894205729167,
      "learning_rate": 0.0001,
      "loss": 4.1582,
      "loss/crossentropy": 2.1121758222579956,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23471853882074356,
      "step": 8106
    },
    {
      "epoch": 0.16216,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.012355295817057292,
      "learning_rate": 0.0001,
      "loss": 4.1464,
      "loss/crossentropy": 1.7613067030906677,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2025228664278984,
      "step": 8108
    },
    {
      "epoch": 0.1622,
      "grad_norm": 2.0,
      "grad_norm_var": 0.013844553629557292,
      "learning_rate": 0.0001,
      "loss": 4.0797,
      "loss/crossentropy": 2.1413058042526245,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21817607432603836,
      "step": 8110
    },
    {
      "epoch": 0.16224,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.013641103108723959,
      "learning_rate": 0.0001,
      "loss": 4.5739,
      "loss/crossentropy": 2.375948429107666,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30982857942581177,
      "step": 8112
    },
    {
      "epoch": 0.16228,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.013396962483723959,
      "learning_rate": 0.0001,
      "loss": 4.076,
      "loss/crossentropy": 1.9669193029403687,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21873307973146439,
      "step": 8114
    },
    {
      "epoch": 0.16232,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.015592193603515625,
      "learning_rate": 0.0001,
      "loss": 4.4024,
      "loss/crossentropy": 2.2028547525405884,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.229970782995224,
      "step": 8116
    },
    {
      "epoch": 0.16236,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.01622289021809896,
      "learning_rate": 0.0001,
      "loss": 4.3359,
      "loss/crossentropy": 2.082156002521515,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2262335941195488,
      "step": 8118
    },
    {
      "epoch": 0.1624,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.012379709879557292,
      "learning_rate": 0.0001,
      "loss": 4.2737,
      "loss/crossentropy": 2.0538666248321533,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22941745817661285,
      "step": 8120
    },
    {
      "epoch": 0.16244,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.015750885009765625,
      "learning_rate": 0.0001,
      "loss": 4.3615,
      "loss/crossentropy": 2.338989734649658,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2434876710176468,
      "step": 8122
    },
    {
      "epoch": 0.16248,
      "grad_norm": 2.375,
      "grad_norm_var": 2.602311197916667,
      "learning_rate": 0.0001,
      "loss": 4.5472,
      "loss/crossentropy": 2.277916193008423,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29285988211631775,
      "step": 8124
    },
    {
      "epoch": 0.16252,
      "grad_norm": 2.46875,
      "grad_norm_var": 2.567577107747396,
      "learning_rate": 0.0001,
      "loss": 4.3437,
      "loss/crossentropy": 2.1196334958076477,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20728030800819397,
      "step": 8126
    },
    {
      "epoch": 0.16256,
      "grad_norm": 2.15625,
      "grad_norm_var": 2.5655558268229166,
      "learning_rate": 0.0001,
      "loss": 4.4216,
      "loss/crossentropy": 2.0216450095176697,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.229092076420784,
      "step": 8128
    },
    {
      "epoch": 0.1626,
      "grad_norm": 2.171875,
      "grad_norm_var": 2.5546834309895834,
      "learning_rate": 0.0001,
      "loss": 4.5163,
      "loss/crossentropy": 2.098921537399292,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22539281845092773,
      "step": 8130
    },
    {
      "epoch": 0.16264,
      "grad_norm": 2.03125,
      "grad_norm_var": 2.580052693684896,
      "learning_rate": 0.0001,
      "loss": 4.1066,
      "loss/crossentropy": 2.0186068415641785,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20286529511213303,
      "step": 8132
    },
    {
      "epoch": 0.16268,
      "grad_norm": 2.28125,
      "grad_norm_var": 2.5584706624348956,
      "learning_rate": 0.0001,
      "loss": 4.7847,
      "loss/crossentropy": 2.157357335090637,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23342902958393097,
      "step": 8134
    },
    {
      "epoch": 0.16272,
      "grad_norm": 2.140625,
      "grad_norm_var": 2.5516998291015627,
      "learning_rate": 0.0001,
      "loss": 4.3489,
      "loss/crossentropy": 2.1885476112365723,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22314801812171936,
      "step": 8136
    },
    {
      "epoch": 0.16276,
      "grad_norm": 2.21875,
      "grad_norm_var": 2.5796160380045574,
      "learning_rate": 0.0001,
      "loss": 4.3403,
      "loss/crossentropy": 2.2208757400512695,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2295266091823578,
      "step": 8138
    },
    {
      "epoch": 0.1628,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.027522532145182292,
      "learning_rate": 0.0001,
      "loss": 4.0473,
      "loss/crossentropy": 1.7885233163833618,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2008970081806183,
      "step": 8140
    },
    {
      "epoch": 0.16284,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.022989654541015626,
      "learning_rate": 0.0001,
      "loss": 4.0835,
      "loss/crossentropy": 2.152435064315796,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2311881259083748,
      "step": 8142
    },
    {
      "epoch": 0.16288,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.016001129150390626,
      "learning_rate": 0.0001,
      "loss": 4.3384,
      "loss/crossentropy": 2.0818406343460083,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24117180705070496,
      "step": 8144
    },
    {
      "epoch": 0.16292,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.02513402303059896,
      "learning_rate": 0.0001,
      "loss": 4.4246,
      "loss/crossentropy": 2.1775856614112854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23246955126523972,
      "step": 8146
    },
    {
      "epoch": 0.16296,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.033607737223307295,
      "learning_rate": 0.0001,
      "loss": 3.9993,
      "loss/crossentropy": 1.7029682397842407,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19564303010702133,
      "step": 8148
    },
    {
      "epoch": 0.163,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.03468195597330729,
      "learning_rate": 0.0001,
      "loss": 4.5631,
      "loss/crossentropy": 2.070194900035858,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2382609099149704,
      "step": 8150
    },
    {
      "epoch": 0.16304,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.036043039957682294,
      "learning_rate": 0.0001,
      "loss": 4.19,
      "loss/crossentropy": 1.9214876890182495,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20894134789705276,
      "step": 8152
    },
    {
      "epoch": 0.16308,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.03243815104166667,
      "learning_rate": 0.0001,
      "loss": 4.3622,
      "loss/crossentropy": 1.7311474084854126,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1962232068181038,
      "step": 8154
    },
    {
      "epoch": 0.16312,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0323150634765625,
      "learning_rate": 0.0001,
      "loss": 4.4623,
      "loss/crossentropy": 2.2161877155303955,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22318138182163239,
      "step": 8156
    },
    {
      "epoch": 0.16316,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.03186747233072917,
      "learning_rate": 0.0001,
      "loss": 4.7343,
      "loss/crossentropy": 2.21865177154541,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.234280064702034,
      "step": 8158
    },
    {
      "epoch": 0.1632,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.032835896809895834,
      "learning_rate": 0.0001,
      "loss": 4.3277,
      "loss/crossentropy": 1.9517142176628113,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1964063122868538,
      "step": 8160
    },
    {
      "epoch": 0.16324,
      "grad_norm": 2.25,
      "grad_norm_var": 0.023713175455729166,
      "learning_rate": 0.0001,
      "loss": 4.3878,
      "loss/crossentropy": 2.261234760284424,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24549759924411774,
      "step": 8162
    },
    {
      "epoch": 0.16328,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.012751261393229166,
      "learning_rate": 0.0001,
      "loss": 4.159,
      "loss/crossentropy": 1.9791623950004578,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2018037736415863,
      "step": 8164
    },
    {
      "epoch": 0.16332,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.005125935872395833,
      "learning_rate": 0.0001,
      "loss": 4.6231,
      "loss/crossentropy": 2.3916029930114746,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24446460604667664,
      "step": 8166
    },
    {
      "epoch": 0.16336,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.007991536458333334,
      "learning_rate": 0.0001,
      "loss": 4.3251,
      "loss/crossentropy": 2.204437553882599,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22334939241409302,
      "step": 8168
    },
    {
      "epoch": 0.1634,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.007111612955729167,
      "learning_rate": 0.0001,
      "loss": 4.3236,
      "loss/crossentropy": 2.2013272047042847,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2068365067243576,
      "step": 8170
    },
    {
      "epoch": 0.16344,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0065826416015625,
      "learning_rate": 0.0001,
      "loss": 4.2597,
      "loss/crossentropy": 1.8648701310157776,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19690127670764923,
      "step": 8172
    },
    {
      "epoch": 0.16348,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.008857981363932291,
      "learning_rate": 0.0001,
      "loss": 4.1324,
      "loss/crossentropy": 2.221195936203003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1900760903954506,
      "step": 8174
    },
    {
      "epoch": 0.16352,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.009445953369140624,
      "learning_rate": 0.0001,
      "loss": 4.4874,
      "loss/crossentropy": 1.8648499846458435,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1982845515012741,
      "step": 8176
    },
    {
      "epoch": 0.16356,
      "grad_norm": 2.125,
      "grad_norm_var": 0.010762278238932292,
      "learning_rate": 0.0001,
      "loss": 4.588,
      "loss/crossentropy": 2.3032894134521484,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25082532316446304,
      "step": 8178
    },
    {
      "epoch": 0.1636,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.010931142171223958,
      "learning_rate": 0.0001,
      "loss": 4.3375,
      "loss/crossentropy": 2.0634626150131226,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23546822369098663,
      "step": 8180
    },
    {
      "epoch": 0.16364,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.012737782796223958,
      "learning_rate": 0.0001,
      "loss": 4.2402,
      "loss/crossentropy": 1.7406468391418457,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20627902448177338,
      "step": 8182
    },
    {
      "epoch": 0.16368,
      "grad_norm": 2.125,
      "grad_norm_var": 0.012668609619140625,
      "learning_rate": 0.0001,
      "loss": 4.4103,
      "loss/crossentropy": 2.3812272548675537,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25258868932724,
      "step": 8184
    },
    {
      "epoch": 0.16372,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.012499745686848958,
      "learning_rate": 0.0001,
      "loss": 4.3584,
      "loss/crossentropy": 2.20754611492157,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21015550196170807,
      "step": 8186
    },
    {
      "epoch": 0.16376,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0140777587890625,
      "learning_rate": 0.0001,
      "loss": 4.1902,
      "loss/crossentropy": 2.081672966480255,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19573034346103668,
      "step": 8188
    },
    {
      "epoch": 0.1638,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.009905751546223958,
      "learning_rate": 0.0001,
      "loss": 4.5267,
      "loss/crossentropy": 2.184974491596222,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2477174997329712,
      "step": 8190
    },
    {
      "epoch": 0.16384,
      "grad_norm": 2.125,
      "grad_norm_var": 0.010322825113932291,
      "learning_rate": 0.0001,
      "loss": 4.3804,
      "loss/crossentropy": 2.1048192977905273,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22900478541851044,
      "step": 8192
    },
    {
      "epoch": 0.16388,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.010135650634765625,
      "learning_rate": 0.0001,
      "loss": 4.1068,
      "loss/crossentropy": 1.960956335067749,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23504969477653503,
      "step": 8194
    },
    {
      "epoch": 0.16392,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.017114003499348957,
      "learning_rate": 0.0001,
      "loss": 4.7698,
      "loss/crossentropy": 2.158856213092804,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2200375646352768,
      "step": 8196
    },
    {
      "epoch": 0.16396,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.017286936442057293,
      "learning_rate": 0.0001,
      "loss": 4.0533,
      "loss/crossentropy": 2.0954058170318604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22729048877954483,
      "step": 8198
    },
    {
      "epoch": 0.164,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.01599299112955729,
      "learning_rate": 0.0001,
      "loss": 4.3492,
      "loss/crossentropy": 2.1452964544296265,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21412881463766098,
      "step": 8200
    },
    {
      "epoch": 0.16404,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.02072321573893229,
      "learning_rate": 0.0001,
      "loss": 4.0204,
      "loss/crossentropy": 1.9737866520881653,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19449186325073242,
      "step": 8202
    },
    {
      "epoch": 0.16408,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.020344034830729166,
      "learning_rate": 0.0001,
      "loss": 4.5119,
      "loss/crossentropy": 2.213072657585144,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2333887815475464,
      "step": 8204
    },
    {
      "epoch": 0.16412,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0203277587890625,
      "learning_rate": 0.0001,
      "loss": 4.2139,
      "loss/crossentropy": 1.8702161312103271,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21368274092674255,
      "step": 8206
    },
    {
      "epoch": 0.16416,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.02102635701497396,
      "learning_rate": 0.0001,
      "loss": 4.242,
      "loss/crossentropy": 2.177275776863098,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21248809248209,
      "step": 8208
    },
    {
      "epoch": 0.1642,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.021945953369140625,
      "learning_rate": 0.0001,
      "loss": 4.3151,
      "loss/crossentropy": 2.3422038555145264,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2288341298699379,
      "step": 8210
    },
    {
      "epoch": 0.16424,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.014212799072265626,
      "learning_rate": 0.0001,
      "loss": 4.2195,
      "loss/crossentropy": 2.094432234764099,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2224871665239334,
      "step": 8212
    },
    {
      "epoch": 0.16428,
      "grad_norm": 2.125,
      "grad_norm_var": 0.020072174072265626,
      "learning_rate": 0.0001,
      "loss": 4.5045,
      "loss/crossentropy": 2.3371682167053223,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2710718363523483,
      "step": 8214
    },
    {
      "epoch": 0.16432,
      "grad_norm": 2.25,
      "grad_norm_var": 0.034242502848307294,
      "learning_rate": 0.0001,
      "loss": 4.6926,
      "loss/crossentropy": 1.8700988292694092,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20795634388923645,
      "step": 8216
    },
    {
      "epoch": 0.16436,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.02575658162434896,
      "learning_rate": 0.0001,
      "loss": 4.3266,
      "loss/crossentropy": 2.080985188484192,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23497651517391205,
      "step": 8218
    },
    {
      "epoch": 0.1644,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.026364898681640624,
      "learning_rate": 0.0001,
      "loss": 4.3873,
      "loss/crossentropy": 2.3486984968185425,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25393396615982056,
      "step": 8220
    },
    {
      "epoch": 0.16444,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.028148396809895834,
      "learning_rate": 0.0001,
      "loss": 4.3446,
      "loss/crossentropy": 2.0084245800971985,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21738936007022858,
      "step": 8222
    },
    {
      "epoch": 0.16448,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.02846247355143229,
      "learning_rate": 0.0001,
      "loss": 3.8584,
      "loss/crossentropy": 1.536482572555542,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17787320166826248,
      "step": 8224
    },
    {
      "epoch": 0.16452,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.024857330322265624,
      "learning_rate": 0.0001,
      "loss": 4.494,
      "loss/crossentropy": 2.1727080941200256,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22967493534088135,
      "step": 8226
    },
    {
      "epoch": 0.16456,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.02490208943684896,
      "learning_rate": 0.0001,
      "loss": 4.1895,
      "loss/crossentropy": 1.6915069222450256,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19113170355558395,
      "step": 8228
    },
    {
      "epoch": 0.1646,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.022739410400390625,
      "learning_rate": 0.0001,
      "loss": 4.4378,
      "loss/crossentropy": 2.1645957231521606,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21431444585323334,
      "step": 8230
    },
    {
      "epoch": 0.16464,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.00858154296875,
      "learning_rate": 0.0001,
      "loss": 3.8088,
      "loss/crossentropy": 1.8797736763954163,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19344569742679596,
      "step": 8232
    },
    {
      "epoch": 0.16468,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0148345947265625,
      "learning_rate": 0.0001,
      "loss": 4.3119,
      "loss/crossentropy": 2.092893421649933,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21476523578166962,
      "step": 8234
    },
    {
      "epoch": 0.16472,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0143310546875,
      "learning_rate": 0.0001,
      "loss": 4.4145,
      "loss/crossentropy": 2.336071252822876,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25474119186401367,
      "step": 8236
    },
    {
      "epoch": 0.16476,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.014410146077473958,
      "learning_rate": 0.0001,
      "loss": 4.3054,
      "loss/crossentropy": 2.008872926235199,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22672003507614136,
      "step": 8238
    },
    {
      "epoch": 0.1648,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.011822255452473958,
      "learning_rate": 0.0001,
      "loss": 4.4997,
      "loss/crossentropy": 2.2229605317115784,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2392737865447998,
      "step": 8240
    },
    {
      "epoch": 0.16484,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.011525217692057292,
      "learning_rate": 0.0001,
      "loss": 4.5894,
      "loss/crossentropy": 2.1929808855056763,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22832375764846802,
      "step": 8242
    },
    {
      "epoch": 0.16488,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.010625966389973958,
      "learning_rate": 0.0001,
      "loss": 4.642,
      "loss/crossentropy": 2.1557860374450684,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24640469253063202,
      "step": 8244
    },
    {
      "epoch": 0.16492,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.009582265218098959,
      "learning_rate": 0.0001,
      "loss": 4.2508,
      "loss/crossentropy": 2.2462236881256104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23639583587646484,
      "step": 8246
    },
    {
      "epoch": 0.16496,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009501139322916666,
      "learning_rate": 0.0001,
      "loss": 4.1476,
      "loss/crossentropy": 2.1238350868225098,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23066531121730804,
      "step": 8248
    },
    {
      "epoch": 0.165,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0049479166666666664,
      "learning_rate": 0.0001,
      "loss": 4.1666,
      "loss/crossentropy": 1.6866248846054077,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20677632093429565,
      "step": 8250
    },
    {
      "epoch": 0.16504,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.005338541666666667,
      "learning_rate": 0.0001,
      "loss": 4.1494,
      "loss/crossentropy": 2.0640709400177,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2148914858698845,
      "step": 8252
    },
    {
      "epoch": 0.16508,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.00513916015625,
      "learning_rate": 0.0001,
      "loss": 4.4138,
      "loss/crossentropy": 2.003119468688965,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21817681193351746,
      "step": 8254
    },
    {
      "epoch": 0.16512,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.007763671875,
      "learning_rate": 0.0001,
      "loss": 4.276,
      "loss/crossentropy": 2.053581953048706,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22257455438375473,
      "step": 8256
    },
    {
      "epoch": 0.16516,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.008381144205729166,
      "learning_rate": 0.0001,
      "loss": 4.1314,
      "loss/crossentropy": 1.788454830646515,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21449437737464905,
      "step": 8258
    },
    {
      "epoch": 0.1652,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.009032185872395833,
      "learning_rate": 0.0001,
      "loss": 4.3986,
      "loss/crossentropy": 1.8791787028312683,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1877290904521942,
      "step": 8260
    },
    {
      "epoch": 0.16524,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.009761555989583334,
      "learning_rate": 0.0001,
      "loss": 4.4326,
      "loss/crossentropy": 2.2346811294555664,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.221164733171463,
      "step": 8262
    },
    {
      "epoch": 0.16528,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009227498372395834,
      "learning_rate": 0.0001,
      "loss": 4.3296,
      "loss/crossentropy": 2.1032413244247437,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21829386800527573,
      "step": 8264
    },
    {
      "epoch": 0.16532,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.008690388997395833,
      "learning_rate": 0.0001,
      "loss": 4.3188,
      "loss/crossentropy": 2.1452749967575073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24733971804380417,
      "step": 8266
    },
    {
      "epoch": 0.16536,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.008373006184895834,
      "learning_rate": 0.0001,
      "loss": 4.2093,
      "loss/crossentropy": 1.8318313956260681,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22139593213796616,
      "step": 8268
    },
    {
      "epoch": 0.1654,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0074045817057291664,
      "learning_rate": 0.0001,
      "loss": 4.3494,
      "loss/crossentropy": 1.9865980744361877,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22354473173618317,
      "step": 8270
    },
    {
      "epoch": 0.16544,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.005464680989583333,
      "learning_rate": 0.0001,
      "loss": 4.318,
      "loss/crossentropy": 2.140509843826294,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22832269966602325,
      "step": 8272
    },
    {
      "epoch": 0.16548,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.005052693684895833,
      "learning_rate": 0.0001,
      "loss": 4.1632,
      "loss/crossentropy": 2.3763319849967957,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22866757214069366,
      "step": 8274
    },
    {
      "epoch": 0.16552,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.004198201497395833,
      "learning_rate": 0.0001,
      "loss": 4.2366,
      "loss/crossentropy": 1.785762071609497,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19961480796337128,
      "step": 8276
    },
    {
      "epoch": 0.16556,
      "grad_norm": 7.34375,
      "grad_norm_var": 1.7181477864583334,
      "learning_rate": 0.0001,
      "loss": 4.5317,
      "loss/crossentropy": 2.0500977635383606,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2972872704267502,
      "step": 8278
    },
    {
      "epoch": 0.1656,
      "grad_norm": 2.265625,
      "grad_norm_var": 1.6987589518229167,
      "learning_rate": 0.0001,
      "loss": 4.4247,
      "loss/crossentropy": 2.3665153980255127,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24115745723247528,
      "step": 8280
    },
    {
      "epoch": 0.16564,
      "grad_norm": 2.25,
      "grad_norm_var": 1.6973592122395833,
      "learning_rate": 0.0001,
      "loss": 4.326,
      "loss/crossentropy": 1.9636226892471313,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2048879787325859,
      "step": 8282
    },
    {
      "epoch": 0.16568,
      "grad_norm": 2.171875,
      "grad_norm_var": 1.6948720296223958,
      "learning_rate": 0.0001,
      "loss": 4.4303,
      "loss/crossentropy": 2.2624993324279785,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22186043858528137,
      "step": 8284
    },
    {
      "epoch": 0.16572,
      "grad_norm": 2.046875,
      "grad_norm_var": 1.7048886617024739,
      "learning_rate": 0.0001,
      "loss": 4.0816,
      "loss/crossentropy": 2.098397970199585,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.219336099922657,
      "step": 8286
    },
    {
      "epoch": 0.16576,
      "grad_norm": 2.140625,
      "grad_norm_var": 1.6997393290201823,
      "learning_rate": 0.0001,
      "loss": 4.4705,
      "loss/crossentropy": 1.9795190691947937,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2145320549607277,
      "step": 8288
    },
    {
      "epoch": 0.1658,
      "grad_norm": 2.203125,
      "grad_norm_var": 1.6936927795410157,
      "learning_rate": 0.0001,
      "loss": 4.4625,
      "loss/crossentropy": 2.1028788089752197,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22656689584255219,
      "step": 8290
    },
    {
      "epoch": 0.16584,
      "grad_norm": 1.984375,
      "grad_norm_var": 1.7015398661295573,
      "learning_rate": 0.0001,
      "loss": 4.1901,
      "loss/crossentropy": 2.2936136722564697,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23688847571611404,
      "step": 8292
    },
    {
      "epoch": 0.16588,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.01862360636393229,
      "learning_rate": 0.0001,
      "loss": 4.1665,
      "loss/crossentropy": 1.870754897594452,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20914533734321594,
      "step": 8294
    },
    {
      "epoch": 0.16592,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.007999674479166666,
      "learning_rate": 0.0001,
      "loss": 3.9884,
      "loss/crossentropy": 2.034530520439148,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22234197705984116,
      "step": 8296
    },
    {
      "epoch": 0.16596,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.0073811848958333336,
      "learning_rate": 0.0001,
      "loss": 4.3674,
      "loss/crossentropy": 1.974400520324707,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20479100942611694,
      "step": 8298
    },
    {
      "epoch": 0.166,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.0078765869140625,
      "learning_rate": 0.0001,
      "loss": 4.3157,
      "loss/crossentropy": 2.180828809738159,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2229994386434555,
      "step": 8300
    },
    {
      "epoch": 0.16604,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.007624308268229167,
      "learning_rate": 0.0001,
      "loss": 4.0704,
      "loss/crossentropy": 2.098487079143524,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19709115475416183,
      "step": 8302
    },
    {
      "epoch": 0.16608,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.007673136393229167,
      "learning_rate": 0.0001,
      "loss": 4.2292,
      "loss/crossentropy": 1.999358892440796,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20891200006008148,
      "step": 8304
    },
    {
      "epoch": 0.16612,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.009022776285807292,
      "learning_rate": 0.0001,
      "loss": 4.4263,
      "loss/crossentropy": 2.40866219997406,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22355159372091293,
      "step": 8306
    },
    {
      "epoch": 0.16616,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.008156077067057291,
      "learning_rate": 0.0001,
      "loss": 4.0532,
      "loss/crossentropy": 1.774325966835022,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2058887928724289,
      "step": 8308
    },
    {
      "epoch": 0.1662,
      "grad_norm": 2.125,
      "grad_norm_var": 0.007352447509765625,
      "learning_rate": 0.0001,
      "loss": 4.4951,
      "loss/crossentropy": 1.9870773553848267,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21368569880723953,
      "step": 8310
    },
    {
      "epoch": 0.16624,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.01121826171875,
      "learning_rate": 0.0001,
      "loss": 4.3496,
      "loss/crossentropy": 1.9224175810813904,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2279355376958847,
      "step": 8312
    },
    {
      "epoch": 0.16628,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.012230428059895833,
      "learning_rate": 0.0001,
      "loss": 4.4074,
      "loss/crossentropy": 2.011419177055359,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2363004833459854,
      "step": 8314
    },
    {
      "epoch": 0.16632,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.01444091796875,
      "learning_rate": 0.0001,
      "loss": 4.6627,
      "loss/crossentropy": 1.9777795672416687,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21830055862665176,
      "step": 8316
    },
    {
      "epoch": 0.16636,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.013508860270182292,
      "learning_rate": 0.0001,
      "loss": 4.3798,
      "loss/crossentropy": 2.3334981203079224,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21089734882116318,
      "step": 8318
    },
    {
      "epoch": 0.1664,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.015148671468098958,
      "learning_rate": 0.0001,
      "loss": 4.3792,
      "loss/crossentropy": 1.9137234687805176,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21954041719436646,
      "step": 8320
    },
    {
      "epoch": 0.16644,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.013492838541666666,
      "learning_rate": 0.0001,
      "loss": 4.3826,
      "loss/crossentropy": 2.1358155608177185,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21321023255586624,
      "step": 8322
    },
    {
      "epoch": 0.16648,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.013688151041666667,
      "learning_rate": 0.0001,
      "loss": 4.3454,
      "loss/crossentropy": 2.1747822165489197,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23682481050491333,
      "step": 8324
    },
    {
      "epoch": 0.16652,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.013505045572916667,
      "learning_rate": 0.0001,
      "loss": 4.2685,
      "loss/crossentropy": 2.2818257808685303,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24226247519254684,
      "step": 8326
    },
    {
      "epoch": 0.16656,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.010155232747395833,
      "learning_rate": 0.0001,
      "loss": 4.4806,
      "loss/crossentropy": 1.9066791534423828,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20610930025577545,
      "step": 8328
    },
    {
      "epoch": 0.1666,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.008284505208333333,
      "learning_rate": 0.0001,
      "loss": 4.4799,
      "loss/crossentropy": 2.2431830763816833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22407780587673187,
      "step": 8330
    },
    {
      "epoch": 0.16664,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0067708333333333336,
      "learning_rate": 0.0001,
      "loss": 4.5894,
      "loss/crossentropy": 2.569726347923279,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2647472620010376,
      "step": 8332
    },
    {
      "epoch": 0.16668,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.006371053059895834,
      "learning_rate": 0.0001,
      "loss": 4.2307,
      "loss/crossentropy": 1.8364137411117554,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2113686427474022,
      "step": 8334
    },
    {
      "epoch": 0.16672,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.007968902587890625,
      "learning_rate": 0.0001,
      "loss": 4.0369,
      "loss/crossentropy": 2.0510441064834595,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2268069088459015,
      "step": 8336
    },
    {
      "epoch": 0.16676,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.007968902587890625,
      "learning_rate": 0.0001,
      "loss": 4.2528,
      "loss/crossentropy": 1.7306728959083557,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20341812074184418,
      "step": 8338
    },
    {
      "epoch": 0.1668,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.023361968994140624,
      "learning_rate": 0.0001,
      "loss": 4.8146,
      "loss/crossentropy": 2.398088574409485,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2426001876592636,
      "step": 8340
    },
    {
      "epoch": 0.16684,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.023128000895182292,
      "learning_rate": 0.0001,
      "loss": 4.0659,
      "loss/crossentropy": 2.1259487867355347,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22666773200035095,
      "step": 8342
    },
    {
      "epoch": 0.16688,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.022141265869140624,
      "learning_rate": 0.0001,
      "loss": 4.2857,
      "loss/crossentropy": 1.9744665026664734,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2887374758720398,
      "step": 8344
    },
    {
      "epoch": 0.16692,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.02316869099934896,
      "learning_rate": 0.0001,
      "loss": 4.1989,
      "loss/crossentropy": 1.9841225743293762,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2008478343486786,
      "step": 8346
    },
    {
      "epoch": 0.16696,
      "grad_norm": 2.125,
      "grad_norm_var": 0.1913469950358073,
      "learning_rate": 0.0001,
      "loss": 4.5208,
      "loss/crossentropy": 2.060486137866974,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23954987525939941,
      "step": 8348
    },
    {
      "epoch": 0.167,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.19230931599934895,
      "learning_rate": 0.0001,
      "loss": 4.2185,
      "loss/crossentropy": 1.9356245398521423,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1987425833940506,
      "step": 8350
    },
    {
      "epoch": 0.16704,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.1835845947265625,
      "learning_rate": 0.0001,
      "loss": 4.1875,
      "loss/crossentropy": 1.9968677163124084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21396416425704956,
      "step": 8352
    },
    {
      "epoch": 0.16708,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.1820465087890625,
      "learning_rate": 0.0001,
      "loss": 4.1696,
      "loss/crossentropy": 2.1678614616394043,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22860293090343475,
      "step": 8354
    },
    {
      "epoch": 0.16712,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.17534891764322916,
      "learning_rate": 0.0001,
      "loss": 4.1761,
      "loss/crossentropy": 1.7752264142036438,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20555391907691956,
      "step": 8356
    },
    {
      "epoch": 0.16716,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.1767578125,
      "learning_rate": 0.0001,
      "loss": 4.2391,
      "loss/crossentropy": 1.7073925137519836,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19325412809848785,
      "step": 8358
    },
    {
      "epoch": 0.1672,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.17827860514322916,
      "learning_rate": 0.0001,
      "loss": 4.7185,
      "loss/crossentropy": 2.2200660705566406,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21834757924079895,
      "step": 8360
    },
    {
      "epoch": 0.16724,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.17594401041666666,
      "learning_rate": 0.0001,
      "loss": 4.3551,
      "loss/crossentropy": 2.3598183393478394,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25592371821403503,
      "step": 8362
    },
    {
      "epoch": 0.16728,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.014644368489583334,
      "learning_rate": 0.0001,
      "loss": 4.3429,
      "loss/crossentropy": 2.342926025390625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2480776235461235,
      "step": 8364
    },
    {
      "epoch": 0.16732,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.014411417643229167,
      "learning_rate": 0.0001,
      "loss": 4.2802,
      "loss/crossentropy": 1.8505961894989014,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21376194059848785,
      "step": 8366
    },
    {
      "epoch": 0.16736,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0146636962890625,
      "learning_rate": 0.0001,
      "loss": 4.4199,
      "loss/crossentropy": 2.2064108848571777,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21334534883499146,
      "step": 8368
    },
    {
      "epoch": 0.1674,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.0338043212890625,
      "learning_rate": 0.0001,
      "loss": 4.5344,
      "loss/crossentropy": 2.1280709505081177,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24501197040081024,
      "step": 8370
    },
    {
      "epoch": 0.16744,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.03312886555989583,
      "learning_rate": 0.0001,
      "loss": 4.3382,
      "loss/crossentropy": 2.2134695053100586,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2368277609348297,
      "step": 8372
    },
    {
      "epoch": 0.16748,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.03319905598958333,
      "learning_rate": 0.0001,
      "loss": 4.1197,
      "loss/crossentropy": 1.6942040920257568,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21226602047681808,
      "step": 8374
    },
    {
      "epoch": 0.16752,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.025755818684895834,
      "learning_rate": 0.0001,
      "loss": 4.5004,
      "loss/crossentropy": 2.1554355025291443,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2655039578676224,
      "step": 8376
    },
    {
      "epoch": 0.16756,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.024800618489583332,
      "learning_rate": 0.0001,
      "loss": 4.2446,
      "loss/crossentropy": 1.8107115030288696,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1756090670824051,
      "step": 8378
    },
    {
      "epoch": 0.1676,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.026496378580729167,
      "learning_rate": 0.0001,
      "loss": 4.0745,
      "loss/crossentropy": 1.655519425868988,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1760415881872177,
      "step": 8380
    },
    {
      "epoch": 0.16764,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.026460774739583335,
      "learning_rate": 0.0001,
      "loss": 4.2369,
      "loss/crossentropy": 2.1618025302886963,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21886380016803741,
      "step": 8382
    },
    {
      "epoch": 0.16768,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.028270467122395834,
      "learning_rate": 0.0001,
      "loss": 4.0798,
      "loss/crossentropy": 2.043319880962372,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21368451416492462,
      "step": 8384
    },
    {
      "epoch": 0.16772,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.010724894205729167,
      "learning_rate": 0.0001,
      "loss": 4.4035,
      "loss/crossentropy": 2.3043102025985718,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25452760607004166,
      "step": 8386
    },
    {
      "epoch": 0.16776,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.011546834309895834,
      "learning_rate": 0.0001,
      "loss": 4.5633,
      "loss/crossentropy": 2.3627192974090576,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2438303530216217,
      "step": 8388
    },
    {
      "epoch": 0.1678,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.011335245768229167,
      "learning_rate": 0.0001,
      "loss": 4.4339,
      "loss/crossentropy": 2.0960012674331665,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20813053101301193,
      "step": 8390
    },
    {
      "epoch": 0.16784,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.013923136393229167,
      "learning_rate": 0.0001,
      "loss": 4.0401,
      "loss/crossentropy": 1.9132550358772278,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19716795533895493,
      "step": 8392
    },
    {
      "epoch": 0.16788,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.014742024739583333,
      "learning_rate": 0.0001,
      "loss": 4.2371,
      "loss/crossentropy": 2.1912059783935547,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.233934685587883,
      "step": 8394
    },
    {
      "epoch": 0.16792,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.012691243489583334,
      "learning_rate": 0.0001,
      "loss": 4.4501,
      "loss/crossentropy": 2.165616512298584,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23475389927625656,
      "step": 8396
    },
    {
      "epoch": 0.16796,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.01265869140625,
      "learning_rate": 0.0001,
      "loss": 4.4841,
      "loss/crossentropy": 2.209625542163849,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24280209839344025,
      "step": 8398
    },
    {
      "epoch": 0.168,
      "grad_norm": 2.125,
      "grad_norm_var": 0.013606516520182292,
      "learning_rate": 0.0001,
      "loss": 3.8959,
      "loss/crossentropy": 1.972103476524353,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2072778418660164,
      "step": 8400
    },
    {
      "epoch": 0.16804,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.009663645426432292,
      "learning_rate": 0.0001,
      "loss": 4.4646,
      "loss/crossentropy": 2.402593731880188,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23028723895549774,
      "step": 8402
    },
    {
      "epoch": 0.16808,
      "grad_norm": 2.125,
      "grad_norm_var": 0.008754221598307292,
      "learning_rate": 0.0001,
      "loss": 4.392,
      "loss/crossentropy": 2.218156576156616,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24104444682598114,
      "step": 8404
    },
    {
      "epoch": 0.16812,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.008722941080729166,
      "learning_rate": 0.0001,
      "loss": 4.2713,
      "loss/crossentropy": 1.9802079796791077,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2134247124195099,
      "step": 8406
    },
    {
      "epoch": 0.16816,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.008577219645182292,
      "learning_rate": 0.0001,
      "loss": 3.9765,
      "loss/crossentropy": 2.0322983264923096,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21680974960327148,
      "step": 8408
    },
    {
      "epoch": 0.1682,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008194732666015624,
      "learning_rate": 0.0001,
      "loss": 4.2457,
      "loss/crossentropy": 2.11838436126709,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2319432571530342,
      "step": 8410
    },
    {
      "epoch": 0.16824,
      "grad_norm": 2.125,
      "grad_norm_var": 0.013398996988932292,
      "learning_rate": 0.0001,
      "loss": 4.2204,
      "loss/crossentropy": 2.3801279067993164,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23981253802776337,
      "step": 8412
    },
    {
      "epoch": 0.16828,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.013042958577473958,
      "learning_rate": 0.0001,
      "loss": 4.2766,
      "loss/crossentropy": 2.0953307151794434,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22835668921470642,
      "step": 8414
    },
    {
      "epoch": 0.16832,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.011774698893229166,
      "learning_rate": 0.0001,
      "loss": 4.528,
      "loss/crossentropy": 2.0654338598251343,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2100541964173317,
      "step": 8416
    },
    {
      "epoch": 0.16836,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.018184407552083334,
      "learning_rate": 0.0001,
      "loss": 4.5734,
      "loss/crossentropy": 2.383318305015564,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.246080219745636,
      "step": 8418
    },
    {
      "epoch": 0.1684,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.018318684895833333,
      "learning_rate": 0.0001,
      "loss": 4.2764,
      "loss/crossentropy": 2.103544294834137,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20659767091274261,
      "step": 8420
    },
    {
      "epoch": 0.16844,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.017437489827473958,
      "learning_rate": 0.0001,
      "loss": 4.2737,
      "loss/crossentropy": 2.064394950866699,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23458171635866165,
      "step": 8422
    },
    {
      "epoch": 0.16848,
      "grad_norm": 2.25,
      "grad_norm_var": 0.014872233072916666,
      "learning_rate": 0.0001,
      "loss": 4.4133,
      "loss/crossentropy": 2.087044835090637,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23527196049690247,
      "step": 8424
    },
    {
      "epoch": 0.16852,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.0133941650390625,
      "learning_rate": 0.0001,
      "loss": 4.5867,
      "loss/crossentropy": 2.41584312915802,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2425323873758316,
      "step": 8426
    },
    {
      "epoch": 0.16856,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.010400390625,
      "learning_rate": 0.0001,
      "loss": 4.3292,
      "loss/crossentropy": 2.2542352080345154,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2330578714609146,
      "step": 8428
    },
    {
      "epoch": 0.1686,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.010724894205729167,
      "learning_rate": 0.0001,
      "loss": 4.6628,
      "loss/crossentropy": 2.453263282775879,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.280864879488945,
      "step": 8430
    },
    {
      "epoch": 0.16864,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.01197509765625,
      "learning_rate": 0.0001,
      "loss": 4.0443,
      "loss/crossentropy": 1.9185429811477661,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19230744242668152,
      "step": 8432
    },
    {
      "epoch": 0.16868,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.005956013997395833,
      "learning_rate": 0.0001,
      "loss": 4.1683,
      "loss/crossentropy": 2.020436644554138,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21756108105182648,
      "step": 8434
    },
    {
      "epoch": 0.16872,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.007616170247395833,
      "learning_rate": 0.0001,
      "loss": 4.7727,
      "loss/crossentropy": 2.3050636053085327,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24950820207595825,
      "step": 8436
    },
    {
      "epoch": 0.16876,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.01510009765625,
      "learning_rate": 0.0001,
      "loss": 4.4091,
      "loss/crossentropy": 2.2787232398986816,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24267014116048813,
      "step": 8438
    },
    {
      "epoch": 0.1688,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.015327962239583333,
      "learning_rate": 0.0001,
      "loss": 4.2927,
      "loss/crossentropy": 2.185176372528076,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2157401591539383,
      "step": 8440
    },
    {
      "epoch": 0.16884,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.01558837890625,
      "learning_rate": 0.0001,
      "loss": 4.2087,
      "loss/crossentropy": 2.0673694610595703,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23145683109760284,
      "step": 8442
    },
    {
      "epoch": 0.16888,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.01754150390625,
      "learning_rate": 0.0001,
      "loss": 4.0229,
      "loss/crossentropy": 1.9011740684509277,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21629629284143448,
      "step": 8444
    },
    {
      "epoch": 0.16892,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.018651326497395832,
      "learning_rate": 0.0001,
      "loss": 4.4821,
      "loss/crossentropy": 2.055977463722229,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2275683432817459,
      "step": 8446
    },
    {
      "epoch": 0.16896,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0171295166015625,
      "learning_rate": 0.0001,
      "loss": 4.2152,
      "loss/crossentropy": 1.752756416797638,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19437911361455917,
      "step": 8448
    },
    {
      "epoch": 0.169,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.015706380208333332,
      "learning_rate": 0.0001,
      "loss": 4.3631,
      "loss/crossentropy": 2.479012131690979,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23593680560588837,
      "step": 8450
    },
    {
      "epoch": 0.16904,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.015523274739583334,
      "learning_rate": 0.0001,
      "loss": 4.3245,
      "loss/crossentropy": 2.065472185611725,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2148493528366089,
      "step": 8452
    },
    {
      "epoch": 0.16908,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.015901692708333335,
      "learning_rate": 0.0001,
      "loss": 4.3955,
      "loss/crossentropy": 2.214062213897705,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25444991141557693,
      "step": 8454
    },
    {
      "epoch": 0.16912,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.0155426025390625,
      "learning_rate": 0.0001,
      "loss": 4.2059,
      "loss/crossentropy": 1.7410383224487305,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21134207397699356,
      "step": 8456
    },
    {
      "epoch": 0.16916,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.020542144775390625,
      "learning_rate": 0.0001,
      "loss": 4.196,
      "loss/crossentropy": 1.9303107857704163,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18110749125480652,
      "step": 8458
    },
    {
      "epoch": 0.1692,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.01810480753580729,
      "learning_rate": 0.0001,
      "loss": 4.3916,
      "loss/crossentropy": 1.9091919660568237,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21320972591638565,
      "step": 8460
    },
    {
      "epoch": 0.16924,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.025099436442057293,
      "learning_rate": 0.0001,
      "loss": 4.4075,
      "loss/crossentropy": 1.8021087050437927,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21294282376766205,
      "step": 8462
    },
    {
      "epoch": 0.16928,
      "grad_norm": 2.0,
      "grad_norm_var": 0.027186838785807292,
      "learning_rate": 0.0001,
      "loss": 3.97,
      "loss/crossentropy": 2.018254518508911,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21781788766384125,
      "step": 8464
    },
    {
      "epoch": 0.16932,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.02906061808268229,
      "learning_rate": 0.0001,
      "loss": 4.5665,
      "loss/crossentropy": 2.0593737959861755,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24223209917545319,
      "step": 8466
    },
    {
      "epoch": 0.16936,
      "grad_norm": 2.125,
      "grad_norm_var": 0.029504140218098957,
      "learning_rate": 0.0001,
      "loss": 4.3116,
      "loss/crossentropy": 2.218974232673645,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22679369151592255,
      "step": 8468
    },
    {
      "epoch": 0.1694,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.022332509358723957,
      "learning_rate": 0.0001,
      "loss": 4.3313,
      "loss/crossentropy": 2.182355046272278,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23690057545900345,
      "step": 8470
    },
    {
      "epoch": 0.16944,
      "grad_norm": 2.375,
      "grad_norm_var": 0.024930572509765624,
      "learning_rate": 0.0001,
      "loss": 4.6241,
      "loss/crossentropy": 2.1669063568115234,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2458687722682953,
      "step": 8472
    },
    {
      "epoch": 0.16948,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.021100870768229165,
      "learning_rate": 0.0001,
      "loss": 4.3181,
      "loss/crossentropy": 2.0656558871269226,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20421817898750305,
      "step": 8474
    },
    {
      "epoch": 0.16952,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.024625651041666665,
      "learning_rate": 0.0001,
      "loss": 4.4916,
      "loss/crossentropy": 2.1470741033554077,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20798998326063156,
      "step": 8476
    },
    {
      "epoch": 0.16956,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.014972941080729166,
      "learning_rate": 0.0001,
      "loss": 4.2616,
      "loss/crossentropy": 2.206741452217102,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2327529340982437,
      "step": 8478
    },
    {
      "epoch": 0.1696,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.012596638997395833,
      "learning_rate": 0.0001,
      "loss": 4.3338,
      "loss/crossentropy": 2.4421777725219727,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2385794073343277,
      "step": 8480
    },
    {
      "epoch": 0.16964,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0115234375,
      "learning_rate": 0.0001,
      "loss": 3.963,
      "loss/crossentropy": 1.8545736074447632,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18871797621250153,
      "step": 8482
    },
    {
      "epoch": 0.16968,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0118072509765625,
      "learning_rate": 0.0001,
      "loss": 4.1352,
      "loss/crossentropy": 1.807646930217743,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19625309854745865,
      "step": 8484
    },
    {
      "epoch": 0.16972,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.012202962239583334,
      "learning_rate": 0.0001,
      "loss": 4.5029,
      "loss/crossentropy": 1.923595905303955,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2233208492398262,
      "step": 8486
    },
    {
      "epoch": 0.16976,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0080718994140625,
      "learning_rate": 0.0001,
      "loss": 4.3251,
      "loss/crossentropy": 2.1018277406692505,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2556355446577072,
      "step": 8488
    },
    {
      "epoch": 0.1698,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.01099853515625,
      "learning_rate": 0.0001,
      "loss": 4.1127,
      "loss/crossentropy": 1.9638542532920837,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2101784572005272,
      "step": 8490
    },
    {
      "epoch": 0.16984,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.008703358968098958,
      "learning_rate": 0.0001,
      "loss": 3.9463,
      "loss/crossentropy": 1.7770507335662842,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2004244327545166,
      "step": 8492
    },
    {
      "epoch": 0.16988,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.008573150634765625,
      "learning_rate": 0.0001,
      "loss": 4.3901,
      "loss/crossentropy": 2.4064877033233643,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23509501665830612,
      "step": 8494
    },
    {
      "epoch": 0.16992,
      "grad_norm": 2.0,
      "grad_norm_var": 0.008713531494140624,
      "learning_rate": 0.0001,
      "loss": 4.3197,
      "loss/crossentropy": 2.183193802833557,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22985967248678207,
      "step": 8496
    },
    {
      "epoch": 0.16996,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.009126536051432292,
      "learning_rate": 0.0001,
      "loss": 4.4983,
      "loss/crossentropy": 2.241714060306549,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22633583843708038,
      "step": 8498
    },
    {
      "epoch": 0.17,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0097808837890625,
      "learning_rate": 0.0001,
      "loss": 4.32,
      "loss/crossentropy": 2.212075114250183,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20900271832942963,
      "step": 8500
    },
    {
      "epoch": 0.17004,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0071685791015625,
      "learning_rate": 0.0001,
      "loss": 4.1885,
      "loss/crossentropy": 2.1283940076828003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21419794112443924,
      "step": 8502
    },
    {
      "epoch": 0.17008,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.00731201171875,
      "learning_rate": 0.0001,
      "loss": 4.3498,
      "loss/crossentropy": 2.1958925127983093,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23375380039215088,
      "step": 8504
    },
    {
      "epoch": 0.17012,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.0065673828125,
      "learning_rate": 0.0001,
      "loss": 4.3396,
      "loss/crossentropy": 1.7883376479148865,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2097182646393776,
      "step": 8506
    },
    {
      "epoch": 0.17016,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.006811269124348958,
      "learning_rate": 0.0001,
      "loss": 4.5606,
      "loss/crossentropy": 2.114001750946045,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2059268057346344,
      "step": 8508
    },
    {
      "epoch": 0.1702,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.006929270426432292,
      "learning_rate": 0.0001,
      "loss": 4.3178,
      "loss/crossentropy": 2.2413129806518555,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2517316862940788,
      "step": 8510
    },
    {
      "epoch": 0.17024,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.015083567301432291,
      "learning_rate": 0.0001,
      "loss": 4.4202,
      "loss/crossentropy": 2.0968031883239746,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2454070746898651,
      "step": 8512
    },
    {
      "epoch": 0.17028,
      "grad_norm": 1.875,
      "grad_norm_var": 0.01945978800455729,
      "learning_rate": 0.0001,
      "loss": 4.407,
      "loss/crossentropy": 1.9813454151153564,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21238256990909576,
      "step": 8514
    },
    {
      "epoch": 0.17032,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.019755045572916668,
      "learning_rate": 0.0001,
      "loss": 4.4724,
      "loss/crossentropy": 2.261451005935669,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2393329069018364,
      "step": 8516
    },
    {
      "epoch": 0.17036,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.020654296875,
      "learning_rate": 0.0001,
      "loss": 4.5377,
      "loss/crossentropy": 1.9661999344825745,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2088441476225853,
      "step": 8518
    },
    {
      "epoch": 0.1704,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.022163899739583333,
      "learning_rate": 0.0001,
      "loss": 3.9288,
      "loss/crossentropy": 1.8275291323661804,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20091407746076584,
      "step": 8520
    },
    {
      "epoch": 0.17044,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.023346964518229166,
      "learning_rate": 0.0001,
      "loss": 4.3154,
      "loss/crossentropy": 2.266944646835327,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2514675632119179,
      "step": 8522
    },
    {
      "epoch": 0.17048,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.02271728515625,
      "learning_rate": 0.0001,
      "loss": 4.3571,
      "loss/crossentropy": 2.244120240211487,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25534868240356445,
      "step": 8524
    },
    {
      "epoch": 0.17052,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.025048828125,
      "learning_rate": 0.0001,
      "loss": 4.3037,
      "loss/crossentropy": 1.8628552556037903,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21675898134708405,
      "step": 8526
    },
    {
      "epoch": 0.17056,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.01597900390625,
      "learning_rate": 0.0001,
      "loss": 4.3883,
      "loss/crossentropy": 1.971808135509491,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2143295481801033,
      "step": 8528
    },
    {
      "epoch": 0.1706,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0114410400390625,
      "learning_rate": 0.0001,
      "loss": 4.3481,
      "loss/crossentropy": 1.959191381931305,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24014096707105637,
      "step": 8530
    },
    {
      "epoch": 0.17064,
      "grad_norm": 3.0,
      "grad_norm_var": 0.05729878743489583,
      "learning_rate": 0.0001,
      "loss": 4.2633,
      "loss/crossentropy": 1.836454451084137,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20957449078559875,
      "step": 8532
    },
    {
      "epoch": 0.17068,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.06005452473958333,
      "learning_rate": 0.0001,
      "loss": 4.1254,
      "loss/crossentropy": 2.003947675228119,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23538918048143387,
      "step": 8534
    },
    {
      "epoch": 0.17072,
      "grad_norm": 2.375,
      "grad_norm_var": 0.05788472493489583,
      "learning_rate": 0.0001,
      "loss": 4.4412,
      "loss/crossentropy": 2.0154194831848145,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2278473973274231,
      "step": 8536
    },
    {
      "epoch": 0.17076,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.05548502604166667,
      "learning_rate": 0.0001,
      "loss": 4.2426,
      "loss/crossentropy": 2.1560275554656982,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21601636707782745,
      "step": 8538
    },
    {
      "epoch": 0.1708,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.05689697265625,
      "learning_rate": 0.0001,
      "loss": 4.2258,
      "loss/crossentropy": 2.1494773626327515,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22883395850658417,
      "step": 8540
    },
    {
      "epoch": 0.17084,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.05607808430989583,
      "learning_rate": 0.0001,
      "loss": 4.5201,
      "loss/crossentropy": 2.20908784866333,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26363062113523483,
      "step": 8542
    },
    {
      "epoch": 0.17088,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0566802978515625,
      "learning_rate": 0.0001,
      "loss": 4.2477,
      "loss/crossentropy": 1.7932087182998657,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18538028001785278,
      "step": 8544
    },
    {
      "epoch": 0.17092,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.05625,
      "learning_rate": 0.0001,
      "loss": 4.471,
      "loss/crossentropy": 2.069899260997772,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22498925775289536,
      "step": 8546
    },
    {
      "epoch": 0.17096,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.019169108072916666,
      "learning_rate": 0.0001,
      "loss": 4.474,
      "loss/crossentropy": 2.008604884147644,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21895557641983032,
      "step": 8548
    },
    {
      "epoch": 0.171,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.018798828125,
      "learning_rate": 0.0001,
      "loss": 4.1973,
      "loss/crossentropy": 2.0652626156806946,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23341115564107895,
      "step": 8550
    },
    {
      "epoch": 0.17104,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.015607706705729167,
      "learning_rate": 0.0001,
      "loss": 4.383,
      "loss/crossentropy": 1.9780349135398865,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22891747951507568,
      "step": 8552
    },
    {
      "epoch": 0.17108,
      "grad_norm": 2.5,
      "grad_norm_var": 0.022980753580729166,
      "learning_rate": 0.0001,
      "loss": 4.4424,
      "loss/crossentropy": 2.163089871406555,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2447328343987465,
      "step": 8554
    },
    {
      "epoch": 0.17112,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.022945149739583334,
      "learning_rate": 0.0001,
      "loss": 4.1503,
      "loss/crossentropy": 2.2946064472198486,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24353116750717163,
      "step": 8556
    },
    {
      "epoch": 0.17116,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.022459920247395834,
      "learning_rate": 0.0001,
      "loss": 4.3049,
      "loss/crossentropy": 2.0238161087036133,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2153262495994568,
      "step": 8558
    },
    {
      "epoch": 0.1712,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.02604955037434896,
      "learning_rate": 0.0001,
      "loss": 3.8636,
      "loss/crossentropy": 1.5683120489120483,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17940818518400192,
      "step": 8560
    },
    {
      "epoch": 0.17124,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.027337392171223957,
      "learning_rate": 0.0001,
      "loss": 4.3867,
      "loss/crossentropy": 1.9956589937210083,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22829821705818176,
      "step": 8562
    },
    {
      "epoch": 0.17128,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.036834462483723955,
      "learning_rate": 0.0001,
      "loss": 4.7055,
      "loss/crossentropy": 2.2242285013198853,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23567892611026764,
      "step": 8564
    },
    {
      "epoch": 0.17132,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.033719635009765624,
      "learning_rate": 0.0001,
      "loss": 4.274,
      "loss/crossentropy": 2.2143776416778564,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2457498088479042,
      "step": 8566
    },
    {
      "epoch": 0.17136,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.033782704671223955,
      "learning_rate": 0.0001,
      "loss": 4.354,
      "loss/crossentropy": 1.852292537689209,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24004538357257843,
      "step": 8568
    },
    {
      "epoch": 0.1714,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.02676976521809896,
      "learning_rate": 0.0001,
      "loss": 4.4429,
      "loss/crossentropy": 2.311089515686035,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22059186547994614,
      "step": 8570
    },
    {
      "epoch": 0.17144,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.027675120035807292,
      "learning_rate": 0.0001,
      "loss": 4.1697,
      "loss/crossentropy": 1.9324169754981995,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21051711589097977,
      "step": 8572
    },
    {
      "epoch": 0.17148,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.02904052734375,
      "learning_rate": 0.0001,
      "loss": 4.2457,
      "loss/crossentropy": 1.982999861240387,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2084646075963974,
      "step": 8574
    },
    {
      "epoch": 0.17152,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.02505671183268229,
      "learning_rate": 0.0001,
      "loss": 4.2503,
      "loss/crossentropy": 2.1837204694747925,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22953644394874573,
      "step": 8576
    },
    {
      "epoch": 0.17156,
      "grad_norm": 2.25,
      "grad_norm_var": 0.024102528889973957,
      "learning_rate": 0.0001,
      "loss": 4.4448,
      "loss/crossentropy": 2.2588841319084167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22851599752902985,
      "step": 8578
    },
    {
      "epoch": 0.1716,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.011502838134765625,
      "learning_rate": 0.0001,
      "loss": 4.2033,
      "loss/crossentropy": 1.982733964920044,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21284686028957367,
      "step": 8580
    },
    {
      "epoch": 0.17164,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.012143707275390625,
      "learning_rate": 0.0001,
      "loss": 4.1238,
      "loss/crossentropy": 2.443873167037964,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23966002464294434,
      "step": 8582
    },
    {
      "epoch": 0.17168,
      "grad_norm": 5.09375,
      "grad_norm_var": 0.5580645243326823,
      "learning_rate": 0.0001,
      "loss": 4.2654,
      "loss/crossentropy": 2.462417483329773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2341725453734398,
      "step": 8584
    },
    {
      "epoch": 0.17172,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.5521705627441407,
      "learning_rate": 0.0001,
      "loss": 4.183,
      "loss/crossentropy": 1.8569464683532715,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.224809430539608,
      "step": 8586
    },
    {
      "epoch": 0.17176,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.55804443359375,
      "learning_rate": 0.0001,
      "loss": 4.2762,
      "loss/crossentropy": 2.017254650592804,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2259984165430069,
      "step": 8588
    },
    {
      "epoch": 0.1718,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.550066884358724,
      "learning_rate": 0.0001,
      "loss": 4.463,
      "loss/crossentropy": 1.9804525971412659,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20844170451164246,
      "step": 8590
    },
    {
      "epoch": 0.17184,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.544781239827474,
      "learning_rate": 0.0001,
      "loss": 4.2676,
      "loss/crossentropy": 2.2506592869758606,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22404606640338898,
      "step": 8592
    },
    {
      "epoch": 0.17188,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.5579335530598958,
      "learning_rate": 0.0001,
      "loss": 4.197,
      "loss/crossentropy": 1.9515153765678406,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.216490276157856,
      "step": 8594
    },
    {
      "epoch": 0.17192,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.5598052978515625,
      "learning_rate": 0.0001,
      "loss": 4.4037,
      "loss/crossentropy": 2.090883791446686,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22496677190065384,
      "step": 8596
    },
    {
      "epoch": 0.17196,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.5523844401041667,
      "learning_rate": 0.0001,
      "loss": 4.8429,
      "loss/crossentropy": 2.4344359636306763,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2748369127511978,
      "step": 8598
    },
    {
      "epoch": 0.172,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.05840250651041667,
      "learning_rate": 0.0001,
      "loss": 4.4359,
      "loss/crossentropy": 1.9280555844306946,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20140594244003296,
      "step": 8600
    },
    {
      "epoch": 0.17204,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.05537007649739583,
      "learning_rate": 0.0001,
      "loss": 4.2802,
      "loss/crossentropy": 2.045006573200226,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21741003543138504,
      "step": 8602
    },
    {
      "epoch": 0.17208,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.04461034138997396,
      "learning_rate": 0.0001,
      "loss": 4.1622,
      "loss/crossentropy": 2.092265546321869,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20971956849098206,
      "step": 8604
    },
    {
      "epoch": 0.17212,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.048130035400390625,
      "learning_rate": 0.0001,
      "loss": 4.16,
      "loss/crossentropy": 1.794768512248993,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20680297911167145,
      "step": 8606
    },
    {
      "epoch": 0.17216,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.05690078735351563,
      "learning_rate": 0.0001,
      "loss": 4.2213,
      "loss/crossentropy": 1.9316805601119995,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2163936346769333,
      "step": 8608
    },
    {
      "epoch": 0.1722,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.05872294108072917,
      "learning_rate": 0.0001,
      "loss": 4.5981,
      "loss/crossentropy": 2.2786675691604614,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2411205694079399,
      "step": 8610
    },
    {
      "epoch": 0.17224,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.05852457682291667,
      "learning_rate": 0.0001,
      "loss": 4.5688,
      "loss/crossentropy": 1.9211469888687134,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23195043951272964,
      "step": 8612
    },
    {
      "epoch": 0.17228,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.037393951416015626,
      "learning_rate": 0.0001,
      "loss": 4.1776,
      "loss/crossentropy": 1.900360643863678,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2339140772819519,
      "step": 8614
    },
    {
      "epoch": 0.17232,
      "grad_norm": 2.125,
      "grad_norm_var": 0.03743464152018229,
      "learning_rate": 0.0001,
      "loss": 4.4408,
      "loss/crossentropy": 1.976994514465332,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.218951515853405,
      "step": 8616
    },
    {
      "epoch": 0.17236,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.035982004801432294,
      "learning_rate": 0.0001,
      "loss": 4.1988,
      "loss/crossentropy": 2.045244038105011,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20886047929525375,
      "step": 8618
    },
    {
      "epoch": 0.1724,
      "grad_norm": 2.25,
      "grad_norm_var": 0.03240534464518229,
      "learning_rate": 0.0001,
      "loss": 4.5082,
      "loss/crossentropy": 2.21256685256958,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21809116005897522,
      "step": 8620
    },
    {
      "epoch": 0.17244,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.02641779581705729,
      "learning_rate": 0.0001,
      "loss": 4.6127,
      "loss/crossentropy": 2.300337314605713,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22060109674930573,
      "step": 8622
    },
    {
      "epoch": 0.17248,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.016932932535807292,
      "learning_rate": 0.0001,
      "loss": 4.2411,
      "loss/crossentropy": 1.8734883666038513,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20403072237968445,
      "step": 8624
    },
    {
      "epoch": 0.17252,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.01962865193684896,
      "learning_rate": 0.0001,
      "loss": 4.6907,
      "loss/crossentropy": 2.1382813453674316,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27397096157073975,
      "step": 8626
    },
    {
      "epoch": 0.17256,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.019760894775390624,
      "learning_rate": 0.0001,
      "loss": 4.0917,
      "loss/crossentropy": 1.9718505144119263,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20658842474222183,
      "step": 8628
    },
    {
      "epoch": 0.1726,
      "grad_norm": 2.125,
      "grad_norm_var": 0.016706339518229165,
      "learning_rate": 0.0001,
      "loss": 4.5233,
      "loss/crossentropy": 2.0957319736480713,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2187560573220253,
      "step": 8630
    },
    {
      "epoch": 0.17264,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.019954427083333334,
      "learning_rate": 0.0001,
      "loss": 4.0986,
      "loss/crossentropy": 2.0504234433174133,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22376062721014023,
      "step": 8632
    },
    {
      "epoch": 0.17268,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.05233968098958333,
      "learning_rate": 0.0001,
      "loss": 4.4513,
      "loss/crossentropy": 2.057171046733856,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23209689557552338,
      "step": 8634
    },
    {
      "epoch": 0.17272,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.052611287434895834,
      "learning_rate": 0.0001,
      "loss": 4.3473,
      "loss/crossentropy": 1.9635317921638489,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21854296326637268,
      "step": 8636
    },
    {
      "epoch": 0.17276,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.0546539306640625,
      "learning_rate": 0.0001,
      "loss": 4.3124,
      "loss/crossentropy": 1.8973188400268555,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24171262234449387,
      "step": 8638
    },
    {
      "epoch": 0.1728,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.05347900390625,
      "learning_rate": 0.0001,
      "loss": 4.2068,
      "loss/crossentropy": 1.6730469465255737,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17975886166095734,
      "step": 8640
    },
    {
      "epoch": 0.17284,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.04702123006184896,
      "learning_rate": 0.0001,
      "loss": 4.4222,
      "loss/crossentropy": 2.2531689405441284,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2170827016234398,
      "step": 8642
    },
    {
      "epoch": 0.17288,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.045873769124348956,
      "learning_rate": 0.0001,
      "loss": 4.0249,
      "loss/crossentropy": 2.0913639068603516,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22586838155984879,
      "step": 8644
    },
    {
      "epoch": 0.17292,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.04533869425455729,
      "learning_rate": 0.0001,
      "loss": 4.45,
      "loss/crossentropy": 2.163489580154419,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2290281057357788,
      "step": 8646
    },
    {
      "epoch": 0.17296,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.04267552693684896,
      "learning_rate": 0.0001,
      "loss": 4.3198,
      "loss/crossentropy": 2.0669034719467163,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22411519289016724,
      "step": 8648
    },
    {
      "epoch": 0.173,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.008213043212890625,
      "learning_rate": 0.0001,
      "loss": 4.0474,
      "loss/crossentropy": 1.9942336678504944,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21262076497077942,
      "step": 8650
    },
    {
      "epoch": 0.17304,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.009474436442057291,
      "learning_rate": 0.0001,
      "loss": 4.2701,
      "loss/crossentropy": 2.046514868736267,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2376151606440544,
      "step": 8652
    },
    {
      "epoch": 0.17308,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.005995432535807292,
      "learning_rate": 0.0001,
      "loss": 4.3308,
      "loss/crossentropy": 1.8385429382324219,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19642101973295212,
      "step": 8654
    },
    {
      "epoch": 0.17312,
      "grad_norm": 2.125,
      "grad_norm_var": 0.007252756754557292,
      "learning_rate": 0.0001,
      "loss": 4.446,
      "loss/crossentropy": 2.259633481502533,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24319174885749817,
      "step": 8656
    },
    {
      "epoch": 0.17316,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.005301920572916666,
      "learning_rate": 0.0001,
      "loss": 4.2067,
      "loss/crossentropy": 1.9811018109321594,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21801364421844482,
      "step": 8658
    },
    {
      "epoch": 0.1732,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.0059722900390625,
      "learning_rate": 0.0001,
      "loss": 4.2158,
      "loss/crossentropy": 2.1726362705230713,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23193368315696716,
      "step": 8660
    },
    {
      "epoch": 0.17324,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.006086222330729167,
      "learning_rate": 0.0001,
      "loss": 4.2587,
      "loss/crossentropy": 1.9915854930877686,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2111910656094551,
      "step": 8662
    },
    {
      "epoch": 0.17328,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.026851399739583334,
      "learning_rate": 0.0001,
      "loss": 4.5001,
      "loss/crossentropy": 1.9651137590408325,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.242890365421772,
      "step": 8664
    },
    {
      "epoch": 0.17332,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.026008097330729167,
      "learning_rate": 0.0001,
      "loss": 4.3848,
      "loss/crossentropy": 1.865262508392334,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21065659821033478,
      "step": 8666
    },
    {
      "epoch": 0.17336,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.02535400390625,
      "learning_rate": 0.0001,
      "loss": 4.3402,
      "loss/crossentropy": 1.9073076248168945,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1959603875875473,
      "step": 8668
    },
    {
      "epoch": 0.1734,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.024332682291666668,
      "learning_rate": 0.0001,
      "loss": 4.2965,
      "loss/crossentropy": 2.167983889579773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22176912426948547,
      "step": 8670
    },
    {
      "epoch": 0.17344,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.5460896809895833,
      "learning_rate": 0.0001,
      "loss": 4.4551,
      "loss/crossentropy": 1.7029761672019958,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.222617506980896,
      "step": 8672
    },
    {
      "epoch": 0.17348,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.5439849853515625,
      "learning_rate": 0.0001,
      "loss": 4.3414,
      "loss/crossentropy": 2.053748309612274,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23398682475090027,
      "step": 8674
    },
    {
      "epoch": 0.17352,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.5408274332682291,
      "learning_rate": 0.0001,
      "loss": 4.7882,
      "loss/crossentropy": 2.309812903404236,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2961876690387726,
      "step": 8676
    },
    {
      "epoch": 0.17356,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.5421953837076823,
      "learning_rate": 0.0001,
      "loss": 4.4416,
      "loss/crossentropy": 2.1045809984207153,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22072184830904007,
      "step": 8678
    },
    {
      "epoch": 0.1736,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.5490435282389323,
      "learning_rate": 0.0001,
      "loss": 4.4437,
      "loss/crossentropy": 2.2114070653915405,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2208957076072693,
      "step": 8680
    },
    {
      "epoch": 0.17364,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.5490435282389323,
      "learning_rate": 0.0001,
      "loss": 4.5931,
      "loss/crossentropy": 2.1773669719696045,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22639526426792145,
      "step": 8682
    },
    {
      "epoch": 0.17368,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.5398272196451823,
      "learning_rate": 0.0001,
      "loss": 3.9397,
      "loss/crossentropy": 1.4213417768478394,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17653951048851013,
      "step": 8684
    },
    {
      "epoch": 0.17372,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.5425374348958333,
      "learning_rate": 0.0001,
      "loss": 4.2711,
      "loss/crossentropy": 1.968630075454712,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2007492408156395,
      "step": 8686
    },
    {
      "epoch": 0.17376,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.030304972330729166,
      "learning_rate": 0.0001,
      "loss": 4.2297,
      "loss/crossentropy": 2.0826632976531982,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21895240992307663,
      "step": 8688
    },
    {
      "epoch": 0.1738,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.029002888997395834,
      "learning_rate": 0.0001,
      "loss": 4.4188,
      "loss/crossentropy": 2.2756701707839966,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22170037031173706,
      "step": 8690
    },
    {
      "epoch": 0.17384,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.010773722330729167,
      "learning_rate": 0.0001,
      "loss": 4.51,
      "loss/crossentropy": 2.329536557197571,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24717354029417038,
      "step": 8692
    },
    {
      "epoch": 0.17388,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.010033162434895833,
      "learning_rate": 0.0001,
      "loss": 4.0776,
      "loss/crossentropy": 2.077241063117981,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.215146966278553,
      "step": 8694
    },
    {
      "epoch": 0.17392,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0091949462890625,
      "learning_rate": 0.0001,
      "loss": 4.3482,
      "loss/crossentropy": 2.2363221645355225,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23428452014923096,
      "step": 8696
    },
    {
      "epoch": 0.17396,
      "grad_norm": 2.0,
      "grad_norm_var": 0.01024169921875,
      "learning_rate": 0.0001,
      "loss": 4.1321,
      "loss/crossentropy": 2.055815279483795,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22674524784088135,
      "step": 8698
    },
    {
      "epoch": 0.174,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0098541259765625,
      "learning_rate": 0.0001,
      "loss": 4.2789,
      "loss/crossentropy": 2.205570936203003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23787499964237213,
      "step": 8700
    },
    {
      "epoch": 0.17404,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009895579020182291,
      "learning_rate": 0.0001,
      "loss": 4.4636,
      "loss/crossentropy": 2.262540578842163,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23921719938516617,
      "step": 8702
    },
    {
      "epoch": 0.17408,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.010114542643229167,
      "learning_rate": 0.0001,
      "loss": 4.3372,
      "loss/crossentropy": 2.5464816093444824,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23302219063043594,
      "step": 8704
    },
    {
      "epoch": 0.17412,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.00897216796875,
      "learning_rate": 0.0001,
      "loss": 4.2355,
      "loss/crossentropy": 2.050383508205414,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23393510282039642,
      "step": 8706
    },
    {
      "epoch": 0.17416,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0076812744140625,
      "learning_rate": 0.0001,
      "loss": 3.9943,
      "loss/crossentropy": 1.9034642577171326,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20801686495542526,
      "step": 8708
    },
    {
      "epoch": 0.1742,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.007045237223307291,
      "learning_rate": 0.0001,
      "loss": 4.2801,
      "loss/crossentropy": 2.313044309616089,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24881915748119354,
      "step": 8710
    },
    {
      "epoch": 0.17424,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.009388987223307292,
      "learning_rate": 0.0001,
      "loss": 4.3754,
      "loss/crossentropy": 1.973829746246338,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21400053054094315,
      "step": 8712
    },
    {
      "epoch": 0.17428,
      "grad_norm": 2.25,
      "grad_norm_var": 0.010109202067057291,
      "learning_rate": 0.0001,
      "loss": 4.2936,
      "loss/crossentropy": 1.831783950328827,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20855721831321716,
      "step": 8714
    },
    {
      "epoch": 0.17432,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.009683990478515625,
      "learning_rate": 0.0001,
      "loss": 4.2681,
      "loss/crossentropy": 2.0173734426498413,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22365443408489227,
      "step": 8716
    },
    {
      "epoch": 0.17436,
      "grad_norm": 2.125,
      "grad_norm_var": 0.007155100504557292,
      "learning_rate": 0.0001,
      "loss": 4.4177,
      "loss/crossentropy": 1.6534234285354614,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20528900623321533,
      "step": 8718
    },
    {
      "epoch": 0.1744,
      "grad_norm": 2.125,
      "grad_norm_var": 0.006396484375,
      "learning_rate": 0.0001,
      "loss": 4.3658,
      "loss/crossentropy": 1.8113531470298767,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21433213353157043,
      "step": 8720
    },
    {
      "epoch": 0.17444,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0064280192057291664,
      "learning_rate": 0.0001,
      "loss": 4.5135,
      "loss/crossentropy": 2.0750836730003357,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22202756255865097,
      "step": 8722
    },
    {
      "epoch": 0.17448,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0056955973307291664,
      "learning_rate": 0.0001,
      "loss": 4.2512,
      "loss/crossentropy": 2.1388206481933594,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24137140065431595,
      "step": 8724
    },
    {
      "epoch": 0.17452,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.006745402018229167,
      "learning_rate": 0.0001,
      "loss": 4.0401,
      "loss/crossentropy": 2.068696141242981,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2215234711766243,
      "step": 8726
    },
    {
      "epoch": 0.17456,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.005280558268229167,
      "learning_rate": 0.0001,
      "loss": 4.0277,
      "loss/crossentropy": 1.6970900893211365,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21108710020780563,
      "step": 8728
    },
    {
      "epoch": 0.1746,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.005582682291666667,
      "learning_rate": 0.0001,
      "loss": 3.8555,
      "loss/crossentropy": 1.8847576975822449,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2151619866490364,
      "step": 8730
    },
    {
      "epoch": 0.17464,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.009403483072916666,
      "learning_rate": 0.0001,
      "loss": 4.4088,
      "loss/crossentropy": 2.4103721380233765,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2589537426829338,
      "step": 8732
    },
    {
      "epoch": 0.17468,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.01011962890625,
      "learning_rate": 0.0001,
      "loss": 4.1415,
      "loss/crossentropy": 1.8340824842453003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22282177209854126,
      "step": 8734
    },
    {
      "epoch": 0.17472,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.010302734375,
      "learning_rate": 0.0001,
      "loss": 4.2472,
      "loss/crossentropy": 1.88236665725708,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2193661779165268,
      "step": 8736
    },
    {
      "epoch": 0.17476,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.05056050618489583,
      "learning_rate": 0.0001,
      "loss": 4.441,
      "loss/crossentropy": 1.8121293783187866,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21855003386735916,
      "step": 8738
    },
    {
      "epoch": 0.1748,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.05090738932291667,
      "learning_rate": 0.0001,
      "loss": 4.1422,
      "loss/crossentropy": 2.215694308280945,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2405528798699379,
      "step": 8740
    },
    {
      "epoch": 0.17484,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.05090738932291667,
      "learning_rate": 0.0001,
      "loss": 4.2431,
      "loss/crossentropy": 2.124837279319763,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22565175592899323,
      "step": 8742
    },
    {
      "epoch": 0.17488,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.0476959228515625,
      "learning_rate": 0.0001,
      "loss": 4.6783,
      "loss/crossentropy": 2.2531429529190063,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.259593665599823,
      "step": 8744
    },
    {
      "epoch": 0.17492,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.04537760416666667,
      "learning_rate": 0.0001,
      "loss": 4.3546,
      "loss/crossentropy": 2.403178572654724,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2475409209728241,
      "step": 8746
    },
    {
      "epoch": 0.17496,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.04527587890625,
      "learning_rate": 0.0001,
      "loss": 4.5698,
      "loss/crossentropy": 1.7886858582496643,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21677076816558838,
      "step": 8748
    },
    {
      "epoch": 0.175,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.04397379557291667,
      "learning_rate": 0.0001,
      "loss": 4.1374,
      "loss/crossentropy": 1.9257569313049316,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2098483294248581,
      "step": 8750
    },
    {
      "epoch": 0.17504,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.04396870930989583,
      "learning_rate": 0.0001,
      "loss": 4.407,
      "loss/crossentropy": 2.1609140634536743,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22846446931362152,
      "step": 8752
    },
    {
      "epoch": 0.17508,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0081939697265625,
      "learning_rate": 0.0001,
      "loss": 4.2683,
      "loss/crossentropy": 2.529700756072998,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2515157088637352,
      "step": 8754
    },
    {
      "epoch": 0.17512,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.008548736572265625,
      "learning_rate": 0.0001,
      "loss": 4.4965,
      "loss/crossentropy": 2.1920565366744995,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24070476740598679,
      "step": 8756
    },
    {
      "epoch": 0.17516,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0097900390625,
      "learning_rate": 0.0001,
      "loss": 3.8416,
      "loss/crossentropy": 1.7714558839797974,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1787928193807602,
      "step": 8758
    },
    {
      "epoch": 0.1752,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.009666951497395833,
      "learning_rate": 0.0001,
      "loss": 4.4457,
      "loss/crossentropy": 1.986818790435791,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20736730098724365,
      "step": 8760
    },
    {
      "epoch": 0.17524,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010453287760416667,
      "learning_rate": 0.0001,
      "loss": 4.2731,
      "loss/crossentropy": 1.8152282238006592,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22459527850151062,
      "step": 8762
    },
    {
      "epoch": 0.17528,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0069163004557291664,
      "learning_rate": 0.0001,
      "loss": 4.4058,
      "loss/crossentropy": 2.2312777042388916,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23017627000808716,
      "step": 8764
    },
    {
      "epoch": 0.17532,
      "grad_norm": 2.375,
      "grad_norm_var": 0.013252766927083333,
      "learning_rate": 0.0001,
      "loss": 4.4714,
      "loss/crossentropy": 2.107849955558777,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21224269270896912,
      "step": 8766
    },
    {
      "epoch": 0.17536,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.013206990559895833,
      "learning_rate": 0.0001,
      "loss": 4.1641,
      "loss/crossentropy": 2.1588711738586426,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2200954109430313,
      "step": 8768
    },
    {
      "epoch": 0.1754,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0130126953125,
      "learning_rate": 0.0001,
      "loss": 4.4056,
      "loss/crossentropy": 2.1355313062667847,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24048195779323578,
      "step": 8770
    },
    {
      "epoch": 0.17544,
      "grad_norm": 2.125,
      "grad_norm_var": 0.011557769775390626,
      "learning_rate": 0.0001,
      "loss": 4.3505,
      "loss/crossentropy": 2.477591037750244,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24459562450647354,
      "step": 8772
    },
    {
      "epoch": 0.17548,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.00845947265625,
      "learning_rate": 0.0001,
      "loss": 4.4184,
      "loss/crossentropy": 2.2577285766601562,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23038798570632935,
      "step": 8774
    },
    {
      "epoch": 0.17552,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.009813435872395833,
      "learning_rate": 0.0001,
      "loss": 4.2681,
      "loss/crossentropy": 2.239536762237549,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22290733456611633,
      "step": 8776
    },
    {
      "epoch": 0.17556,
      "grad_norm": 2.125,
      "grad_norm_var": 0.007373046875,
      "learning_rate": 0.0001,
      "loss": 4.4351,
      "loss/crossentropy": 1.9139958024024963,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2175864800810814,
      "step": 8778
    },
    {
      "epoch": 0.1756,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.00888671875,
      "learning_rate": 0.0001,
      "loss": 4.0756,
      "loss/crossentropy": 2.0622661113739014,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2223067432641983,
      "step": 8780
    },
    {
      "epoch": 0.17564,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.004524739583333334,
      "learning_rate": 0.0001,
      "loss": 4.4552,
      "loss/crossentropy": 2.222475051879883,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21181221306324005,
      "step": 8782
    },
    {
      "epoch": 0.17568,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.007002512613932292,
      "learning_rate": 0.0001,
      "loss": 4.1696,
      "loss/crossentropy": 2.2612074613571167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24079158157110214,
      "step": 8784
    },
    {
      "epoch": 0.17572,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.005863189697265625,
      "learning_rate": 0.0001,
      "loss": 4.0889,
      "loss/crossentropy": 2.1629387736320496,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21437199413776398,
      "step": 8786
    },
    {
      "epoch": 0.17576,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.009492746988932292,
      "learning_rate": 0.0001,
      "loss": 4.6583,
      "loss/crossentropy": 2.145151972770691,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23797442018985748,
      "step": 8788
    },
    {
      "epoch": 0.1758,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.015457916259765624,
      "learning_rate": 0.0001,
      "loss": 4.5188,
      "loss/crossentropy": 2.0366984605789185,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2191104218363762,
      "step": 8790
    },
    {
      "epoch": 0.17584,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.014422353108723958,
      "learning_rate": 0.0001,
      "loss": 4.1604,
      "loss/crossentropy": 2.1049715280532837,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.230033777654171,
      "step": 8792
    },
    {
      "epoch": 0.17588,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.014338938395182292,
      "learning_rate": 0.0001,
      "loss": 4.4327,
      "loss/crossentropy": 2.2549991607666016,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23084092140197754,
      "step": 8794
    },
    {
      "epoch": 0.17592,
      "grad_norm": 2.125,
      "grad_norm_var": 0.015547688802083333,
      "learning_rate": 0.0001,
      "loss": 4.4653,
      "loss/crossentropy": 1.9873813390731812,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19879335910081863,
      "step": 8796
    },
    {
      "epoch": 0.17596,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.015677897135416667,
      "learning_rate": 0.0001,
      "loss": 4.6371,
      "loss/crossentropy": 2.0723283886909485,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20538055896759033,
      "step": 8798
    },
    {
      "epoch": 0.176,
      "grad_norm": 2.125,
      "grad_norm_var": 0.013099924723307291,
      "learning_rate": 0.0001,
      "loss": 4.0313,
      "loss/crossentropy": 2.090642750263214,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2329270914196968,
      "step": 8800
    },
    {
      "epoch": 0.17604,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.012672678629557291,
      "learning_rate": 0.0001,
      "loss": 4.4626,
      "loss/crossentropy": 2.3432271480560303,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2456662431359291,
      "step": 8802
    },
    {
      "epoch": 0.17608,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.010465240478515625,
      "learning_rate": 0.0001,
      "loss": 4.5133,
      "loss/crossentropy": 2.1210837364196777,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22580894827842712,
      "step": 8804
    },
    {
      "epoch": 0.17612,
      "grad_norm": 2.0,
      "grad_norm_var": 0.005147043863932292,
      "learning_rate": 0.0001,
      "loss": 4.024,
      "loss/crossentropy": 2.142494797706604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22426588833332062,
      "step": 8806
    },
    {
      "epoch": 0.17616,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0049435933430989586,
      "learning_rate": 0.0001,
      "loss": 4.2877,
      "loss/crossentropy": 1.9163227677345276,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21529845893383026,
      "step": 8808
    },
    {
      "epoch": 0.1762,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.004937489827473958,
      "learning_rate": 0.0001,
      "loss": 4.4776,
      "loss/crossentropy": 2.1478612422943115,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2354428842663765,
      "step": 8810
    },
    {
      "epoch": 0.17624,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.0033854166666666668,
      "learning_rate": 0.0001,
      "loss": 4.435,
      "loss/crossentropy": 2.1546601057052612,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22553270310163498,
      "step": 8812
    },
    {
      "epoch": 0.17628,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.004233551025390625,
      "learning_rate": 0.0001,
      "loss": 4.1661,
      "loss/crossentropy": 2.0559862852096558,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19714127480983734,
      "step": 8814
    },
    {
      "epoch": 0.17632,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.004078928629557292,
      "learning_rate": 0.0001,
      "loss": 4.3543,
      "loss/crossentropy": 2.1340363025665283,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22422882914543152,
      "step": 8816
    },
    {
      "epoch": 0.17636,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0038937886555989584,
      "learning_rate": 0.0001,
      "loss": 4.4464,
      "loss/crossentropy": 2.265942335128784,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23249086737632751,
      "step": 8818
    },
    {
      "epoch": 0.1764,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0031939188639322916,
      "learning_rate": 0.0001,
      "loss": 4.3187,
      "loss/crossentropy": 2.245513081550598,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22869569063186646,
      "step": 8820
    },
    {
      "epoch": 0.17644,
      "grad_norm": 2.0,
      "grad_norm_var": 0.006461334228515625,
      "learning_rate": 0.0001,
      "loss": 4.179,
      "loss/crossentropy": 1.851025104522705,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21989689767360687,
      "step": 8822
    },
    {
      "epoch": 0.17648,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.006266021728515625,
      "learning_rate": 0.0001,
      "loss": 4.276,
      "loss/crossentropy": 2.2972241640090942,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22793132066726685,
      "step": 8824
    },
    {
      "epoch": 0.17652,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.006276194254557292,
      "learning_rate": 0.0001,
      "loss": 4.3955,
      "loss/crossentropy": 2.248735189437866,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2133597657084465,
      "step": 8826
    },
    {
      "epoch": 0.17656,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.008314768473307291,
      "learning_rate": 0.0001,
      "loss": 4.4423,
      "loss/crossentropy": 2.4173099994659424,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23985996842384338,
      "step": 8828
    },
    {
      "epoch": 0.1766,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.008698527018229167,
      "learning_rate": 0.0001,
      "loss": 4.5425,
      "loss/crossentropy": 2.5017653703689575,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26119648665189743,
      "step": 8830
    },
    {
      "epoch": 0.17664,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0090484619140625,
      "learning_rate": 0.0001,
      "loss": 4.2474,
      "loss/crossentropy": 1.9006813764572144,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19938994944095612,
      "step": 8832
    },
    {
      "epoch": 0.17668,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.009496053059895834,
      "learning_rate": 0.0001,
      "loss": 4.1643,
      "loss/crossentropy": 2.101746916770935,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20544035732746124,
      "step": 8834
    },
    {
      "epoch": 0.17672,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.010542805989583333,
      "learning_rate": 0.0001,
      "loss": 4.3211,
      "loss/crossentropy": 2.1605160236358643,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23061934113502502,
      "step": 8836
    },
    {
      "epoch": 0.17676,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.008967081705729166,
      "learning_rate": 0.0001,
      "loss": 4.4357,
      "loss/crossentropy": 1.963772177696228,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20736530423164368,
      "step": 8838
    },
    {
      "epoch": 0.1768,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.009733072916666667,
      "learning_rate": 0.0001,
      "loss": 4.3572,
      "loss/crossentropy": 2.154300093650818,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22221273183822632,
      "step": 8840
    },
    {
      "epoch": 0.17684,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.009501139322916666,
      "learning_rate": 0.0001,
      "loss": 4.4876,
      "loss/crossentropy": 2.1576497554779053,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21834726631641388,
      "step": 8842
    },
    {
      "epoch": 0.17688,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0115875244140625,
      "learning_rate": 0.0001,
      "loss": 4.1192,
      "loss/crossentropy": 2.1316112279891968,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2166184037923813,
      "step": 8844
    },
    {
      "epoch": 0.17692,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.0098052978515625,
      "learning_rate": 0.0001,
      "loss": 4.2421,
      "loss/crossentropy": 2.068525493144989,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23183748871088028,
      "step": 8846
    },
    {
      "epoch": 0.17696,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.01051025390625,
      "learning_rate": 0.0001,
      "loss": 4.5128,
      "loss/crossentropy": 2.185767650604248,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25881427526474,
      "step": 8848
    },
    {
      "epoch": 0.177,
      "grad_norm": 2.125,
      "grad_norm_var": 0.010619099934895833,
      "learning_rate": 0.0001,
      "loss": 4.3073,
      "loss/crossentropy": 1.979454517364502,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20669714361429214,
      "step": 8850
    },
    {
      "epoch": 0.17704,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.009373982747395834,
      "learning_rate": 0.0001,
      "loss": 4.3356,
      "loss/crossentropy": 2.3473092317581177,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23232270777225494,
      "step": 8852
    },
    {
      "epoch": 0.17708,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.007616170247395833,
      "learning_rate": 0.0001,
      "loss": 4.3732,
      "loss/crossentropy": 2.461324691772461,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.242417573928833,
      "step": 8854
    },
    {
      "epoch": 0.17712,
      "grad_norm": 2.125,
      "grad_norm_var": 0.008385976155598959,
      "learning_rate": 0.0001,
      "loss": 4.1107,
      "loss/crossentropy": 1.5953214168548584,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17768454551696777,
      "step": 8856
    },
    {
      "epoch": 0.17716,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.008377838134765624,
      "learning_rate": 0.0001,
      "loss": 4.4289,
      "loss/crossentropy": 2.1969146728515625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21466109156608582,
      "step": 8858
    },
    {
      "epoch": 0.1772,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.008459218343098958,
      "learning_rate": 0.0001,
      "loss": 4.2154,
      "loss/crossentropy": 2.243234634399414,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22729554027318954,
      "step": 8860
    },
    {
      "epoch": 0.17724,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.010628000895182291,
      "learning_rate": 0.0001,
      "loss": 4.2062,
      "loss/crossentropy": 2.1855397820472717,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24905066192150116,
      "step": 8862
    },
    {
      "epoch": 0.17728,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.010628000895182291,
      "learning_rate": 0.0001,
      "loss": 4.4075,
      "loss/crossentropy": 2.320886254310608,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22135943174362183,
      "step": 8864
    },
    {
      "epoch": 0.17732,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.011736806233723958,
      "learning_rate": 0.0001,
      "loss": 4.1444,
      "loss/crossentropy": 2.140891909599304,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2109990492463112,
      "step": 8866
    },
    {
      "epoch": 0.17736,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.013038889567057291,
      "learning_rate": 0.0001,
      "loss": 4.1009,
      "loss/crossentropy": 2.147824764251709,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2131204530596733,
      "step": 8868
    },
    {
      "epoch": 0.1774,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.039249420166015625,
      "learning_rate": 0.0001,
      "loss": 4.7136,
      "loss/crossentropy": 2.187807321548462,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2335718423128128,
      "step": 8870
    },
    {
      "epoch": 0.17744,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.037287394205729164,
      "learning_rate": 0.0001,
      "loss": 4.5265,
      "loss/crossentropy": 2.3127458095550537,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2385788857936859,
      "step": 8872
    },
    {
      "epoch": 0.17748,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0369537353515625,
      "learning_rate": 0.0001,
      "loss": 4.1364,
      "loss/crossentropy": 1.859586775302887,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.204575777053833,
      "step": 8874
    },
    {
      "epoch": 0.17752,
      "grad_norm": 2.0,
      "grad_norm_var": 0.03585611979166667,
      "learning_rate": 0.0001,
      "loss": 4.1993,
      "loss/crossentropy": 1.9626107215881348,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21016598492860794,
      "step": 8876
    },
    {
      "epoch": 0.17756,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.034077962239583336,
      "learning_rate": 0.0001,
      "loss": 4.462,
      "loss/crossentropy": 2.1130539774894714,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21995095163583755,
      "step": 8878
    },
    {
      "epoch": 0.1776,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.035374959309895836,
      "learning_rate": 0.0001,
      "loss": 4.4677,
      "loss/crossentropy": 1.8914743065834045,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20437531173229218,
      "step": 8880
    },
    {
      "epoch": 0.17764,
      "grad_norm": 2.125,
      "grad_norm_var": 0.032763671875,
      "learning_rate": 0.0001,
      "loss": 4.4975,
      "loss/crossentropy": 2.2135708332061768,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2478664517402649,
      "step": 8882
    },
    {
      "epoch": 0.17768,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.028922526041666667,
      "learning_rate": 0.0001,
      "loss": 4.3763,
      "loss/crossentropy": 2.194110333919525,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2366471290588379,
      "step": 8884
    },
    {
      "epoch": 0.17772,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.009822336832682292,
      "learning_rate": 0.0001,
      "loss": 4.2844,
      "loss/crossentropy": 2.4365748167037964,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25842973589897156,
      "step": 8886
    },
    {
      "epoch": 0.17776,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.010001373291015626,
      "learning_rate": 0.0001,
      "loss": 4.0574,
      "loss/crossentropy": 1.9800177216529846,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21982619166374207,
      "step": 8888
    },
    {
      "epoch": 0.1778,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.009956614176432291,
      "learning_rate": 0.0001,
      "loss": 4.3465,
      "loss/crossentropy": 2.1437748670578003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21494000405073166,
      "step": 8890
    },
    {
      "epoch": 0.17784,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.008957672119140624,
      "learning_rate": 0.0001,
      "loss": 4.4051,
      "loss/crossentropy": 2.0610267519950867,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22473593801259995,
      "step": 8892
    },
    {
      "epoch": 0.17788,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.013952382405598958,
      "learning_rate": 0.0001,
      "loss": 4.4637,
      "loss/crossentropy": 2.261958599090576,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2250354364514351,
      "step": 8894
    },
    {
      "epoch": 0.17792,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.012237294514973959,
      "learning_rate": 0.0001,
      "loss": 4.3508,
      "loss/crossentropy": 2.3689773082733154,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22185539454221725,
      "step": 8896
    },
    {
      "epoch": 0.17796,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.013034820556640625,
      "learning_rate": 0.0001,
      "loss": 4.7432,
      "loss/crossentropy": 2.612341523170471,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2561237961053848,
      "step": 8898
    },
    {
      "epoch": 0.178,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.010990142822265625,
      "learning_rate": 0.0001,
      "loss": 4.5876,
      "loss/crossentropy": 2.1230576038360596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23115848749876022,
      "step": 8900
    },
    {
      "epoch": 0.17804,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.009626261393229167,
      "learning_rate": 0.0001,
      "loss": 4.2154,
      "loss/crossentropy": 2.200004458427429,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22002413868904114,
      "step": 8902
    },
    {
      "epoch": 0.17808,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009989420572916666,
      "learning_rate": 0.0001,
      "loss": 4.3438,
      "loss/crossentropy": 2.323713779449463,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22923698276281357,
      "step": 8904
    },
    {
      "epoch": 0.17812,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.00982666015625,
      "learning_rate": 0.0001,
      "loss": 4.3079,
      "loss/crossentropy": 2.038426458835602,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21465667337179184,
      "step": 8906
    },
    {
      "epoch": 0.17816,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.01217041015625,
      "learning_rate": 0.0001,
      "loss": 4.4498,
      "loss/crossentropy": 2.3639097213745117,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2378860041499138,
      "step": 8908
    },
    {
      "epoch": 0.1782,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.00699462890625,
      "learning_rate": 0.0001,
      "loss": 4.261,
      "loss/crossentropy": 1.8291080594062805,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19894887506961823,
      "step": 8910
    },
    {
      "epoch": 0.17824,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.011263020833333333,
      "learning_rate": 0.0001,
      "loss": 4.5606,
      "loss/crossentropy": 2.3113714456558228,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24526391178369522,
      "step": 8912
    },
    {
      "epoch": 0.17828,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.011449178059895834,
      "learning_rate": 0.0001,
      "loss": 4.3016,
      "loss/crossentropy": 2.114617943763733,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21305133402347565,
      "step": 8914
    },
    {
      "epoch": 0.17832,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.014975738525390626,
      "learning_rate": 0.0001,
      "loss": 3.9333,
      "loss/crossentropy": 1.6893808841705322,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19195494800806046,
      "step": 8916
    },
    {
      "epoch": 0.17836,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.016721343994140624,
      "learning_rate": 0.0001,
      "loss": 4.2501,
      "loss/crossentropy": 1.829396367073059,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21181780844926834,
      "step": 8918
    },
    {
      "epoch": 0.1784,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.01869481404622396,
      "learning_rate": 0.0001,
      "loss": 4.1542,
      "loss/crossentropy": 1.8910154104232788,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20975399017333984,
      "step": 8920
    },
    {
      "epoch": 0.17844,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.01953709920247396,
      "learning_rate": 0.0001,
      "loss": 4.3327,
      "loss/crossentropy": 2.0501255989074707,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22336408495903015,
      "step": 8922
    },
    {
      "epoch": 0.17848,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.01740086873372396,
      "learning_rate": 0.0001,
      "loss": 4.3035,
      "loss/crossentropy": 2.3023892641067505,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23251917958259583,
      "step": 8924
    },
    {
      "epoch": 0.17852,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.01822077433268229,
      "learning_rate": 0.0001,
      "loss": 4.0592,
      "loss/crossentropy": 2.0030421018600464,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19997069239616394,
      "step": 8926
    },
    {
      "epoch": 0.17856,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.013741048177083333,
      "learning_rate": 0.0001,
      "loss": 4.2568,
      "loss/crossentropy": 2.2309017181396484,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22145257890224457,
      "step": 8928
    },
    {
      "epoch": 0.1786,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.013728841145833334,
      "learning_rate": 0.0001,
      "loss": 4.4366,
      "loss/crossentropy": 2.1135157346725464,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21580957621335983,
      "step": 8930
    },
    {
      "epoch": 0.17864,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011146799723307291,
      "learning_rate": 0.0001,
      "loss": 4.3473,
      "loss/crossentropy": 2.098900556564331,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22469021379947662,
      "step": 8932
    },
    {
      "epoch": 0.17868,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.0103179931640625,
      "learning_rate": 0.0001,
      "loss": 4.2702,
      "loss/crossentropy": 2.1558337211608887,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24765773862600327,
      "step": 8934
    },
    {
      "epoch": 0.17872,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.006004842122395834,
      "learning_rate": 0.0001,
      "loss": 4.3318,
      "loss/crossentropy": 2.141040623188019,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22490675747394562,
      "step": 8936
    },
    {
      "epoch": 0.17876,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.0096099853515625,
      "learning_rate": 0.0001,
      "loss": 4.4126,
      "loss/crossentropy": 2.1036806106567383,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23260055482387543,
      "step": 8938
    },
    {
      "epoch": 0.1788,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.013036092122395834,
      "learning_rate": 0.0001,
      "loss": 4.4151,
      "loss/crossentropy": 1.9403663277626038,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21941428631544113,
      "step": 8940
    },
    {
      "epoch": 0.17884,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.014717610677083333,
      "learning_rate": 0.0001,
      "loss": 4.2,
      "loss/crossentropy": 1.8589079976081848,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21533852070569992,
      "step": 8942
    },
    {
      "epoch": 0.17888,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.013816070556640626,
      "learning_rate": 0.0001,
      "loss": 4.2476,
      "loss/crossentropy": 2.3136903643608093,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23139026761054993,
      "step": 8944
    },
    {
      "epoch": 0.17892,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.012835439046223958,
      "learning_rate": 0.0001,
      "loss": 4.2669,
      "loss/crossentropy": 2.305663585662842,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24263737350702286,
      "step": 8946
    },
    {
      "epoch": 0.17896,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.02195002237955729,
      "learning_rate": 0.0001,
      "loss": 4.3059,
      "loss/crossentropy": 2.0450612902641296,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2124529778957367,
      "step": 8948
    },
    {
      "epoch": 0.179,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.020393880208333333,
      "learning_rate": 0.0001,
      "loss": 4.4229,
      "loss/crossentropy": 2.435065984725952,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24603386223316193,
      "step": 8950
    },
    {
      "epoch": 0.17904,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.02295099894205729,
      "learning_rate": 0.0001,
      "loss": 4.2604,
      "loss/crossentropy": 2.092818021774292,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20905248820781708,
      "step": 8952
    },
    {
      "epoch": 0.17908,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.024621327718098957,
      "learning_rate": 0.0001,
      "loss": 4.2539,
      "loss/crossentropy": 2.08588969707489,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2142588049173355,
      "step": 8954
    },
    {
      "epoch": 0.17912,
      "grad_norm": 2.25,
      "grad_norm_var": 0.021144358317057292,
      "learning_rate": 0.0001,
      "loss": 4.3958,
      "loss/crossentropy": 1.9161878824234009,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20626582205295563,
      "step": 8956
    },
    {
      "epoch": 0.17916,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.020182037353515626,
      "learning_rate": 0.0001,
      "loss": 4.3726,
      "loss/crossentropy": 2.3072937726974487,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22028075903654099,
      "step": 8958
    },
    {
      "epoch": 0.1792,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.021109771728515626,
      "learning_rate": 0.0001,
      "loss": 4.4876,
      "loss/crossentropy": 2.3053938150405884,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22212930023670197,
      "step": 8960
    },
    {
      "epoch": 0.17924,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.02269261678059896,
      "learning_rate": 0.0001,
      "loss": 4.5137,
      "loss/crossentropy": 1.9130414128303528,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20036083459854126,
      "step": 8962
    },
    {
      "epoch": 0.17928,
      "grad_norm": 2.0,
      "grad_norm_var": 0.013242340087890625,
      "learning_rate": 0.0001,
      "loss": 4.1299,
      "loss/crossentropy": 2.3808066844940186,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2116570845246315,
      "step": 8964
    },
    {
      "epoch": 0.17932,
      "grad_norm": 2.125,
      "grad_norm_var": 0.009549713134765625,
      "learning_rate": 0.0001,
      "loss": 4.296,
      "loss/crossentropy": 2.180716395378113,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2275484874844551,
      "step": 8966
    },
    {
      "epoch": 0.17936,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.008837890625,
      "learning_rate": 0.0001,
      "loss": 4.2597,
      "loss/crossentropy": 2.027850866317749,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2011229619383812,
      "step": 8968
    },
    {
      "epoch": 0.1794,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.017447916666666667,
      "learning_rate": 0.0001,
      "loss": 4.3167,
      "loss/crossentropy": 2.077622890472412,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21361806988716125,
      "step": 8970
    },
    {
      "epoch": 0.17944,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.017513020833333334,
      "learning_rate": 0.0001,
      "loss": 4.1972,
      "loss/crossentropy": 2.004905104637146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21110886335372925,
      "step": 8972
    },
    {
      "epoch": 0.17948,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.019466145833333334,
      "learning_rate": 0.0001,
      "loss": 4.5026,
      "loss/crossentropy": 2.1859925389289856,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23011507838964462,
      "step": 8974
    },
    {
      "epoch": 0.17952,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.019840494791666666,
      "learning_rate": 0.0001,
      "loss": 4.2908,
      "loss/crossentropy": 1.8372295498847961,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2171105071902275,
      "step": 8976
    },
    {
      "epoch": 0.17956,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.020197550455729168,
      "learning_rate": 0.0001,
      "loss": 4.3887,
      "loss/crossentropy": 2.127313494682312,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22017831355333328,
      "step": 8978
    },
    {
      "epoch": 0.1796,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.022098795572916666,
      "learning_rate": 0.0001,
      "loss": 4.3389,
      "loss/crossentropy": 2.121580421924591,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24898843467235565,
      "step": 8980
    },
    {
      "epoch": 0.17964,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0226470947265625,
      "learning_rate": 0.0001,
      "loss": 4.1127,
      "loss/crossentropy": 2.0973563194274902,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21355029940605164,
      "step": 8982
    },
    {
      "epoch": 0.17968,
      "grad_norm": 2.125,
      "grad_norm_var": 0.018896484375,
      "learning_rate": 0.0001,
      "loss": 4.3898,
      "loss/crossentropy": 2.1109927892684937,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2096879929304123,
      "step": 8984
    },
    {
      "epoch": 0.17972,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0126617431640625,
      "learning_rate": 0.0001,
      "loss": 4.1264,
      "loss/crossentropy": 1.954129159450531,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20745816081762314,
      "step": 8986
    },
    {
      "epoch": 0.17976,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.013483683268229166,
      "learning_rate": 0.0001,
      "loss": 4.5357,
      "loss/crossentropy": 1.875806748867035,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19150983542203903,
      "step": 8988
    },
    {
      "epoch": 0.1798,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.01451416015625,
      "learning_rate": 0.0001,
      "loss": 4.3042,
      "loss/crossentropy": 2.3101617097854614,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2393202781677246,
      "step": 8990
    },
    {
      "epoch": 0.17984,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.013374837239583333,
      "learning_rate": 0.0001,
      "loss": 4.4954,
      "loss/crossentropy": 2.3156551122665405,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23346291482448578,
      "step": 8992
    },
    {
      "epoch": 0.17988,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.013966623942057292,
      "learning_rate": 0.0001,
      "loss": 4.3675,
      "loss/crossentropy": 2.4437999725341797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2172817811369896,
      "step": 8994
    },
    {
      "epoch": 0.17992,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.009437815348307291,
      "learning_rate": 0.0001,
      "loss": 4.3122,
      "loss/crossentropy": 2.0451250076293945,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22556670010089874,
      "step": 8996
    },
    {
      "epoch": 0.17996,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.009852854410807292,
      "learning_rate": 0.0001,
      "loss": 4.5654,
      "loss/crossentropy": 2.3135393857955933,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2313593551516533,
      "step": 8998
    },
    {
      "epoch": 0.18,
      "grad_norm": 2.125,
      "grad_norm_var": 0.010530344645182292,
      "learning_rate": 0.0001,
      "loss": 4.3001,
      "loss/crossentropy": 1.9934805035591125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20815817266702652,
      "step": 9000
    },
    {
      "epoch": 0.18004,
      "grad_norm": 2.125,
      "grad_norm_var": 0.011982981363932292,
      "learning_rate": 0.0001,
      "loss": 4.6193,
      "loss/crossentropy": 2.1770662665367126,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23489046096801758,
      "step": 9002
    },
    {
      "epoch": 0.18008,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.009779612223307291,
      "learning_rate": 0.0001,
      "loss": 4.3708,
      "loss/crossentropy": 1.9791364073753357,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2139936238527298,
      "step": 9004
    },
    {
      "epoch": 0.18012,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011445871988932292,
      "learning_rate": 0.0001,
      "loss": 4.298,
      "loss/crossentropy": 2.2092931270599365,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22620443254709244,
      "step": 9006
    },
    {
      "epoch": 0.18016,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.011034901936848958,
      "learning_rate": 0.0001,
      "loss": 4.3114,
      "loss/crossentropy": 2.123443365097046,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21943332999944687,
      "step": 9008
    },
    {
      "epoch": 0.1802,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.009212239583333334,
      "learning_rate": 0.0001,
      "loss": 4.4666,
      "loss/crossentropy": 2.243329405784607,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24108020961284637,
      "step": 9010
    },
    {
      "epoch": 0.18024,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.008426920572916666,
      "learning_rate": 0.0001,
      "loss": 4.3687,
      "loss/crossentropy": 2.366227388381958,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21806316077709198,
      "step": 9012
    },
    {
      "epoch": 0.18028,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009663899739583334,
      "learning_rate": 0.0001,
      "loss": 3.8255,
      "loss/crossentropy": 1.768812358379364,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1861925944685936,
      "step": 9014
    },
    {
      "epoch": 0.18032,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.012225087483723958,
      "learning_rate": 0.0001,
      "loss": 4.1236,
      "loss/crossentropy": 1.9376537799835205,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18691913783550262,
      "step": 9016
    },
    {
      "epoch": 0.18036,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.010709381103515625,
      "learning_rate": 0.0001,
      "loss": 4.2748,
      "loss/crossentropy": 2.3026299476623535,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23757526278495789,
      "step": 9018
    },
    {
      "epoch": 0.1804,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.010057576497395833,
      "learning_rate": 0.0001,
      "loss": 4.0026,
      "loss/crossentropy": 1.9697216153144836,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21453910320997238,
      "step": 9020
    },
    {
      "epoch": 0.18044,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.011058553059895834,
      "learning_rate": 0.0001,
      "loss": 4.5457,
      "loss/crossentropy": 2.257638931274414,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24236120283603668,
      "step": 9022
    },
    {
      "epoch": 0.18048,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0115142822265625,
      "learning_rate": 0.0001,
      "loss": 4.3323,
      "loss/crossentropy": 2.244320869445801,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25429578125476837,
      "step": 9024
    },
    {
      "epoch": 0.18052,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.012214152018229167,
      "learning_rate": 0.0001,
      "loss": 4.2854,
      "loss/crossentropy": 2.1105872988700867,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23081901669502258,
      "step": 9026
    },
    {
      "epoch": 0.18056,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.012839508056640626,
      "learning_rate": 0.0001,
      "loss": 4.2561,
      "loss/crossentropy": 1.9657647609710693,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22111424803733826,
      "step": 9028
    },
    {
      "epoch": 0.1806,
      "grad_norm": 2.125,
      "grad_norm_var": 0.012308502197265625,
      "learning_rate": 0.0001,
      "loss": 4.4019,
      "loss/crossentropy": 2.0759438276290894,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22763221710920334,
      "step": 9030
    },
    {
      "epoch": 0.18064,
      "grad_norm": 2.125,
      "grad_norm_var": 0.010888417561848959,
      "learning_rate": 0.0001,
      "loss": 3.9999,
      "loss/crossentropy": 2.0250572562217712,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2279352843761444,
      "step": 9032
    },
    {
      "epoch": 0.18068,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.010534413655598958,
      "learning_rate": 0.0001,
      "loss": 4.3772,
      "loss/crossentropy": 2.270031213760376,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23242096602916718,
      "step": 9034
    },
    {
      "epoch": 0.18072,
      "grad_norm": 2.25,
      "grad_norm_var": 0.011922200520833334,
      "learning_rate": 0.0001,
      "loss": 4.4304,
      "loss/crossentropy": 2.10041344165802,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23192601650953293,
      "step": 9036
    },
    {
      "epoch": 0.18076,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.009227498372395834,
      "learning_rate": 0.0001,
      "loss": 4.1041,
      "loss/crossentropy": 2.0255953073501587,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23707614094018936,
      "step": 9038
    },
    {
      "epoch": 0.1808,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.010406239827473959,
      "learning_rate": 0.0001,
      "loss": 4.1399,
      "loss/crossentropy": 1.8162729740142822,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19466694444417953,
      "step": 9040
    },
    {
      "epoch": 0.18084,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.010170237223307291,
      "learning_rate": 0.0001,
      "loss": 4.5543,
      "loss/crossentropy": 2.1271599531173706,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22750889509916306,
      "step": 9042
    },
    {
      "epoch": 0.18088,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.009447224934895833,
      "learning_rate": 0.0001,
      "loss": 4.5377,
      "loss/crossentropy": 2.3638603687286377,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2456393539905548,
      "step": 9044
    },
    {
      "epoch": 0.18092,
      "grad_norm": 2.125,
      "grad_norm_var": 0.019437662760416665,
      "learning_rate": 0.0001,
      "loss": 4.3413,
      "loss/crossentropy": 1.5851669907569885,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20994101464748383,
      "step": 9046
    },
    {
      "epoch": 0.18096,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.017789459228515624,
      "learning_rate": 0.0001,
      "loss": 4.2284,
      "loss/crossentropy": 1.7990906834602356,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18919725716114044,
      "step": 9048
    },
    {
      "epoch": 0.181,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.017490386962890625,
      "learning_rate": 0.0001,
      "loss": 4.5415,
      "loss/crossentropy": 2.0975595712661743,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23837832361459732,
      "step": 9050
    },
    {
      "epoch": 0.18104,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.018070475260416666,
      "learning_rate": 0.0001,
      "loss": 4.1304,
      "loss/crossentropy": 2.1970856189727783,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23219536244869232,
      "step": 9052
    },
    {
      "epoch": 0.18108,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.017292277018229166,
      "learning_rate": 0.0001,
      "loss": 4.211,
      "loss/crossentropy": 2.0846009850502014,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20775136351585388,
      "step": 9054
    },
    {
      "epoch": 0.18112,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.015075429280598959,
      "learning_rate": 0.0001,
      "loss": 4.0415,
      "loss/crossentropy": 1.662496030330658,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18223516643047333,
      "step": 9056
    },
    {
      "epoch": 0.18116,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.016078440348307292,
      "learning_rate": 0.0001,
      "loss": 4.2561,
      "loss/crossentropy": 2.126902401447296,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2193778082728386,
      "step": 9058
    },
    {
      "epoch": 0.1812,
      "grad_norm": 2.984375,
      "grad_norm_var": 0.062459309895833336,
      "learning_rate": 0.0001,
      "loss": 4.0343,
      "loss/crossentropy": 1.9529814720153809,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19026879966259003,
      "step": 9060
    },
    {
      "epoch": 0.18124,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.05981852213541667,
      "learning_rate": 0.0001,
      "loss": 4.1911,
      "loss/crossentropy": 2.140450179576874,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23864319920539856,
      "step": 9062
    },
    {
      "epoch": 0.18128,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.06083882649739583,
      "learning_rate": 0.0001,
      "loss": 4.3442,
      "loss/crossentropy": 2.4139195680618286,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2360195592045784,
      "step": 9064
    },
    {
      "epoch": 0.18132,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.06083882649739583,
      "learning_rate": 0.0001,
      "loss": 4.5122,
      "loss/crossentropy": 2.190356135368347,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22671552002429962,
      "step": 9066
    },
    {
      "epoch": 0.18136,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.05916315714518229,
      "learning_rate": 0.0001,
      "loss": 4.281,
      "loss/crossentropy": 1.9428812861442566,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22318705916404724,
      "step": 9068
    },
    {
      "epoch": 0.1814,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.059242502848307295,
      "learning_rate": 0.0001,
      "loss": 4.4628,
      "loss/crossentropy": 2.296473503112793,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22005227208137512,
      "step": 9070
    },
    {
      "epoch": 0.18144,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.05919774373372396,
      "learning_rate": 0.0001,
      "loss": 4.5848,
      "loss/crossentropy": 2.2118901014328003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22189343720674515,
      "step": 9072
    },
    {
      "epoch": 0.18148,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.05810114542643229,
      "learning_rate": 0.0001,
      "loss": 4.285,
      "loss/crossentropy": 1.8919037580490112,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1986190229654312,
      "step": 9074
    },
    {
      "epoch": 0.18152,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.011161295572916667,
      "learning_rate": 0.0001,
      "loss": 4.3046,
      "loss/crossentropy": 1.975312054157257,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20403321832418442,
      "step": 9076
    },
    {
      "epoch": 0.18156,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.0075032552083333336,
      "learning_rate": 0.0001,
      "loss": 4.2175,
      "loss/crossentropy": 1.8076966404914856,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19708115607500076,
      "step": 9078
    },
    {
      "epoch": 0.1816,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.007112630208333333,
      "learning_rate": 0.0001,
      "loss": 4.2166,
      "loss/crossentropy": 2.101171374320984,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22123181074857712,
      "step": 9080
    },
    {
      "epoch": 0.18164,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.007225545247395834,
      "learning_rate": 0.0001,
      "loss": 4.2292,
      "loss/crossentropy": 2.09942090511322,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21761803328990936,
      "step": 9082
    },
    {
      "epoch": 0.18168,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.008072916666666667,
      "learning_rate": 0.0001,
      "loss": 4.4802,
      "loss/crossentropy": 2.4418424367904663,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24908769130706787,
      "step": 9084
    },
    {
      "epoch": 0.18172,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0087890625,
      "learning_rate": 0.0001,
      "loss": 4.3408,
      "loss/crossentropy": 1.944950520992279,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20515284687280655,
      "step": 9086
    },
    {
      "epoch": 0.18176,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.007991536458333334,
      "learning_rate": 0.0001,
      "loss": 4.3117,
      "loss/crossentropy": 1.870418667793274,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20427027344703674,
      "step": 9088
    },
    {
      "epoch": 0.1818,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.007682291666666666,
      "learning_rate": 0.0001,
      "loss": 4.1323,
      "loss/crossentropy": 1.9338520169258118,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19928501546382904,
      "step": 9090
    },
    {
      "epoch": 0.18184,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.008876291910807292,
      "learning_rate": 0.0001,
      "loss": 3.9734,
      "loss/crossentropy": 1.7826221585273743,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19778436422348022,
      "step": 9092
    },
    {
      "epoch": 0.18188,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.007458241780598959,
      "learning_rate": 0.0001,
      "loss": 4.3537,
      "loss/crossentropy": 2.2922680377960205,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23613491654396057,
      "step": 9094
    },
    {
      "epoch": 0.18192,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.15102513631184897,
      "learning_rate": 0.0001,
      "loss": 4.6789,
      "loss/crossentropy": 2.1802788972854614,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1969536542892456,
      "step": 9096
    },
    {
      "epoch": 0.18196,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.1498308817545573,
      "learning_rate": 0.0001,
      "loss": 4.1715,
      "loss/crossentropy": 1.9129992723464966,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1942882016301155,
      "step": 9098
    },
    {
      "epoch": 0.182,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.14738337198893228,
      "learning_rate": 0.0001,
      "loss": 4.3374,
      "loss/crossentropy": 1.8288249969482422,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19879616051912308,
      "step": 9100
    },
    {
      "epoch": 0.18204,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.14580663045247397,
      "learning_rate": 0.0001,
      "loss": 4.2824,
      "loss/crossentropy": 2.039812684059143,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21374420076608658,
      "step": 9102
    },
    {
      "epoch": 0.18208,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.15449193318684895,
      "learning_rate": 0.0001,
      "loss": 4.1862,
      "loss/crossentropy": 2.1177414059638977,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2193453460931778,
      "step": 9104
    },
    {
      "epoch": 0.18212,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.16704076131184895,
      "learning_rate": 0.0001,
      "loss": 4.4869,
      "loss/crossentropy": 2.171097159385681,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21211445331573486,
      "step": 9106
    },
    {
      "epoch": 0.18216,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.16413548787434895,
      "learning_rate": 0.0001,
      "loss": 4.1422,
      "loss/crossentropy": 1.9781638383865356,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21307373046875,
      "step": 9108
    },
    {
      "epoch": 0.1822,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.17157363891601562,
      "learning_rate": 0.0001,
      "loss": 4.2454,
      "loss/crossentropy": 2.0868560075759888,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20633937418460846,
      "step": 9110
    },
    {
      "epoch": 0.18224,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.04592463175455729,
      "learning_rate": 0.0001,
      "loss": 4.1153,
      "loss/crossentropy": 2.079172134399414,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20709815621376038,
      "step": 9112
    },
    {
      "epoch": 0.18228,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.04835383097330729,
      "learning_rate": 0.0001,
      "loss": 4.1836,
      "loss/crossentropy": 2.100913643836975,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21831409633159637,
      "step": 9114
    },
    {
      "epoch": 0.18232,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.049478912353515626,
      "learning_rate": 0.0001,
      "loss": 4.4087,
      "loss/crossentropy": 2.018395781517029,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22545243054628372,
      "step": 9116
    },
    {
      "epoch": 0.18236,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.04278132120768229,
      "learning_rate": 0.0001,
      "loss": 4.3421,
      "loss/crossentropy": 1.9177632331848145,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23346271365880966,
      "step": 9118
    },
    {
      "epoch": 0.1824,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.038358306884765624,
      "learning_rate": 0.0001,
      "loss": 3.9924,
      "loss/crossentropy": 2.1122357845306396,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22593770176172256,
      "step": 9120
    },
    {
      "epoch": 0.18244,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.014021555582682291,
      "learning_rate": 0.0001,
      "loss": 4.1399,
      "loss/crossentropy": 1.7920495867729187,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2127150148153305,
      "step": 9122
    },
    {
      "epoch": 0.18248,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0096099853515625,
      "learning_rate": 0.0001,
      "loss": 4.2061,
      "loss/crossentropy": 2.2180920839309692,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25088224560022354,
      "step": 9124
    },
    {
      "epoch": 0.18252,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.008185831705729167,
      "learning_rate": 0.0001,
      "loss": 4.3287,
      "loss/crossentropy": 1.8999969959259033,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21616832166910172,
      "step": 9126
    },
    {
      "epoch": 0.18256,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0081451416015625,
      "learning_rate": 0.0001,
      "loss": 4.5351,
      "loss/crossentropy": 1.9424527287483215,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22016742825508118,
      "step": 9128
    },
    {
      "epoch": 0.1826,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0049468994140625,
      "learning_rate": 0.0001,
      "loss": 4.4432,
      "loss/crossentropy": 2.0539366006851196,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26818516850471497,
      "step": 9130
    },
    {
      "epoch": 0.18264,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.00458984375,
      "learning_rate": 0.0001,
      "loss": 4.1634,
      "loss/crossentropy": 1.9180519580841064,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2067112922668457,
      "step": 9132
    },
    {
      "epoch": 0.18268,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.005052693684895833,
      "learning_rate": 0.0001,
      "loss": 3.9913,
      "loss/crossentropy": 1.7417545318603516,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1970825269818306,
      "step": 9134
    },
    {
      "epoch": 0.18272,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.004833984375,
      "learning_rate": 0.0001,
      "loss": 4.016,
      "loss/crossentropy": 1.9498217701911926,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21281737089157104,
      "step": 9136
    },
    {
      "epoch": 0.18276,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.005615234375,
      "learning_rate": 0.0001,
      "loss": 4.3842,
      "loss/crossentropy": 2.140692949295044,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24291902035474777,
      "step": 9138
    },
    {
      "epoch": 0.1828,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0069976806640625,
      "learning_rate": 0.0001,
      "loss": 4.1785,
      "loss/crossentropy": 2.3510342836380005,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23326712846755981,
      "step": 9140
    },
    {
      "epoch": 0.18284,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0072743733723958336,
      "learning_rate": 0.0001,
      "loss": 4.0707,
      "loss/crossentropy": 2.049591898918152,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21118033677339554,
      "step": 9142
    },
    {
      "epoch": 0.18288,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.005631510416666667,
      "learning_rate": 0.0001,
      "loss": 4.263,
      "loss/crossentropy": 1.949703335762024,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22161198407411575,
      "step": 9144
    },
    {
      "epoch": 0.18292,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.005399576822916667,
      "learning_rate": 0.0001,
      "loss": 4.5602,
      "loss/crossentropy": 2.1442413330078125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21281076222658157,
      "step": 9146
    },
    {
      "epoch": 0.18296,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.007111612955729167,
      "learning_rate": 0.0001,
      "loss": 4.0848,
      "loss/crossentropy": 2.103494882583618,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20892268419265747,
      "step": 9148
    },
    {
      "epoch": 0.183,
      "grad_norm": 2.25,
      "grad_norm_var": 0.01060791015625,
      "learning_rate": 0.0001,
      "loss": 4.133,
      "loss/crossentropy": 2.1544495224952698,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2008143737912178,
      "step": 9150
    },
    {
      "epoch": 0.18304,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.01109619140625,
      "learning_rate": 0.0001,
      "loss": 4.2799,
      "loss/crossentropy": 2.021821677684784,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20975126326084137,
      "step": 9152
    },
    {
      "epoch": 0.18308,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0102203369140625,
      "learning_rate": 0.0001,
      "loss": 4.2858,
      "loss/crossentropy": 2.109215199947357,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19855067878961563,
      "step": 9154
    },
    {
      "epoch": 0.18312,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.011165364583333334,
      "learning_rate": 0.0001,
      "loss": 4.2181,
      "loss/crossentropy": 1.7631941437721252,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20584283769130707,
      "step": 9156
    },
    {
      "epoch": 0.18316,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.0116851806640625,
      "learning_rate": 0.0001,
      "loss": 4.4005,
      "loss/crossentropy": 2.131524443626404,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2151477411389351,
      "step": 9158
    },
    {
      "epoch": 0.1832,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.01334228515625,
      "learning_rate": 0.0001,
      "loss": 4.2729,
      "loss/crossentropy": 2.018375277519226,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21982873231172562,
      "step": 9160
    },
    {
      "epoch": 0.18324,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0154205322265625,
      "learning_rate": 0.0001,
      "loss": 4.3302,
      "loss/crossentropy": 2.217617154121399,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2257251739501953,
      "step": 9162
    },
    {
      "epoch": 0.18328,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.014606730143229166,
      "learning_rate": 0.0001,
      "loss": 4.227,
      "loss/crossentropy": 1.8632460832595825,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19728046655654907,
      "step": 9164
    },
    {
      "epoch": 0.18332,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.012137858072916667,
      "learning_rate": 0.0001,
      "loss": 4.5335,
      "loss/crossentropy": 2.2818111181259155,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22397568821907043,
      "step": 9166
    },
    {
      "epoch": 0.18336,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.01177978515625,
      "learning_rate": 0.0001,
      "loss": 4.4104,
      "loss/crossentropy": 2.1209938526153564,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22490206360816956,
      "step": 9168
    },
    {
      "epoch": 0.1834,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.00924072265625,
      "learning_rate": 0.0001,
      "loss": 4.244,
      "loss/crossentropy": 2.141623795032501,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21710850298404694,
      "step": 9170
    },
    {
      "epoch": 0.18344,
      "grad_norm": 2.125,
      "grad_norm_var": 0.007323201497395833,
      "learning_rate": 0.0001,
      "loss": 4.2063,
      "loss/crossentropy": 2.165239691734314,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.235738106071949,
      "step": 9172
    },
    {
      "epoch": 0.18348,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008103179931640624,
      "learning_rate": 0.0001,
      "loss": 4.0186,
      "loss/crossentropy": 1.8649475574493408,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2088451236486435,
      "step": 9174
    },
    {
      "epoch": 0.18352,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.007012685139973958,
      "learning_rate": 0.0001,
      "loss": 4.2853,
      "loss/crossentropy": 2.312218189239502,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.236283540725708,
      "step": 9176
    },
    {
      "epoch": 0.18356,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.008699544270833333,
      "learning_rate": 0.0001,
      "loss": 3.9626,
      "loss/crossentropy": 2.0149282217025757,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21434535831212997,
      "step": 9178
    },
    {
      "epoch": 0.1836,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0086822509765625,
      "learning_rate": 0.0001,
      "loss": 4.5081,
      "loss/crossentropy": 2.499966621398926,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26318275928497314,
      "step": 9180
    },
    {
      "epoch": 0.18364,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0084136962890625,
      "learning_rate": 0.0001,
      "loss": 4.4498,
      "loss/crossentropy": 2.0454984307289124,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2202616035938263,
      "step": 9182
    },
    {
      "epoch": 0.18368,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.010835520426432292,
      "learning_rate": 0.0001,
      "loss": 4.0754,
      "loss/crossentropy": 2.1708725094795227,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20958629250526428,
      "step": 9184
    },
    {
      "epoch": 0.18372,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.010792795817057292,
      "learning_rate": 0.0001,
      "loss": 4.2104,
      "loss/crossentropy": 1.8261350989341736,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1952563151717186,
      "step": 9186
    },
    {
      "epoch": 0.18376,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.010009511311848959,
      "learning_rate": 0.0001,
      "loss": 4.1247,
      "loss/crossentropy": 2.036627769470215,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22490298002958298,
      "step": 9188
    },
    {
      "epoch": 0.1838,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.010501861572265625,
      "learning_rate": 0.0001,
      "loss": 4.0088,
      "loss/crossentropy": 1.7977086305618286,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19282807409763336,
      "step": 9190
    },
    {
      "epoch": 0.18384,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.011211903889973958,
      "learning_rate": 0.0001,
      "loss": 4.1774,
      "loss/crossentropy": 2.170135021209717,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23149186372756958,
      "step": 9192
    },
    {
      "epoch": 0.18388,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.012035115559895834,
      "learning_rate": 0.0001,
      "loss": 4.4155,
      "loss/crossentropy": 2.1453020572662354,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25120319426059723,
      "step": 9194
    },
    {
      "epoch": 0.18392,
      "grad_norm": 2.125,
      "grad_norm_var": 0.01224365234375,
      "learning_rate": 0.0001,
      "loss": 4.3628,
      "loss/crossentropy": 1.8794063925743103,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20883548259735107,
      "step": 9196
    },
    {
      "epoch": 0.18396,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.0141998291015625,
      "learning_rate": 0.0001,
      "loss": 4.3384,
      "loss/crossentropy": 2.021254241466522,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21939975768327713,
      "step": 9198
    },
    {
      "epoch": 0.184,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.009557851155598958,
      "learning_rate": 0.0001,
      "loss": 4.158,
      "loss/crossentropy": 2.195580303668976,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2108924463391304,
      "step": 9200
    },
    {
      "epoch": 0.18404,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.014085896809895833,
      "learning_rate": 0.0001,
      "loss": 4.0781,
      "loss/crossentropy": 1.6260902881622314,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20428159832954407,
      "step": 9202
    },
    {
      "epoch": 0.18408,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.015363566080729167,
      "learning_rate": 0.0001,
      "loss": 4.2525,
      "loss/crossentropy": 2.138678550720215,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21641074120998383,
      "step": 9204
    },
    {
      "epoch": 0.18412,
      "grad_norm": 2.125,
      "grad_norm_var": 0.01825129191080729,
      "learning_rate": 0.0001,
      "loss": 4.4706,
      "loss/crossentropy": 2.047194480895996,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20891964435577393,
      "step": 9206
    },
    {
      "epoch": 0.18416,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.019606272379557293,
      "learning_rate": 0.0001,
      "loss": 4.2312,
      "loss/crossentropy": 2.189277768135071,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2150397077202797,
      "step": 9208
    },
    {
      "epoch": 0.1842,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.017618560791015626,
      "learning_rate": 0.0001,
      "loss": 4.2792,
      "loss/crossentropy": 2.184122920036316,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23278063535690308,
      "step": 9210
    },
    {
      "epoch": 0.18424,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.019139607747395832,
      "learning_rate": 0.0001,
      "loss": 4.3471,
      "loss/crossentropy": 2.413718104362488,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24333150684833527,
      "step": 9212
    },
    {
      "epoch": 0.18428,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.019017537434895832,
      "learning_rate": 0.0001,
      "loss": 3.8486,
      "loss/crossentropy": 1.8086814880371094,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19976364076137543,
      "step": 9214
    },
    {
      "epoch": 0.18432,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.019978841145833332,
      "learning_rate": 0.0001,
      "loss": 4.5222,
      "loss/crossentropy": 2.2418206930160522,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.229690782725811,
      "step": 9216
    },
    {
      "epoch": 0.18436,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.016355133056640624,
      "learning_rate": 0.0001,
      "loss": 4.0817,
      "loss/crossentropy": 1.8083258867263794,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20999443531036377,
      "step": 9218
    },
    {
      "epoch": 0.1844,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.01693115234375,
      "learning_rate": 0.0001,
      "loss": 3.772,
      "loss/crossentropy": 1.8117709755897522,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20436270534992218,
      "step": 9220
    },
    {
      "epoch": 0.18444,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.010550944010416667,
      "learning_rate": 0.0001,
      "loss": 4.2149,
      "loss/crossentropy": 2.024270534515381,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20097267627716064,
      "step": 9222
    },
    {
      "epoch": 0.18448,
      "grad_norm": 2.25,
      "grad_norm_var": 0.011449178059895834,
      "learning_rate": 0.0001,
      "loss": 4.4756,
      "loss/crossentropy": 2.2385981678962708,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21131044626235962,
      "step": 9224
    },
    {
      "epoch": 0.18452,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.011271158854166666,
      "learning_rate": 0.0001,
      "loss": 4.3745,
      "loss/crossentropy": 2.127749502658844,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22855369001626968,
      "step": 9226
    },
    {
      "epoch": 0.18456,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.011237589518229167,
      "learning_rate": 0.0001,
      "loss": 4.2645,
      "loss/crossentropy": 2.1877033710479736,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2339663878083229,
      "step": 9228
    },
    {
      "epoch": 0.1846,
      "grad_norm": 2.25,
      "grad_norm_var": 0.011139933268229167,
      "learning_rate": 0.0001,
      "loss": 4.2642,
      "loss/crossentropy": 1.931507408618927,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20912021398544312,
      "step": 9230
    },
    {
      "epoch": 0.18464,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.011579386393229167,
      "learning_rate": 0.0001,
      "loss": 4.3309,
      "loss/crossentropy": 1.8101251125335693,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23724676668643951,
      "step": 9232
    },
    {
      "epoch": 0.18468,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0111480712890625,
      "learning_rate": 0.0001,
      "loss": 4.201,
      "loss/crossentropy": 2.015208065509796,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23482007533311844,
      "step": 9234
    },
    {
      "epoch": 0.18472,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.010282135009765625,
      "learning_rate": 0.0001,
      "loss": 4.5357,
      "loss/crossentropy": 2.2742738723754883,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23632052540779114,
      "step": 9236
    },
    {
      "epoch": 0.18476,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.016806793212890626,
      "learning_rate": 0.0001,
      "loss": 4.3574,
      "loss/crossentropy": 1.7254774570465088,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2104022428393364,
      "step": 9238
    },
    {
      "epoch": 0.1848,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.016585032145182293,
      "learning_rate": 0.0001,
      "loss": 4.3477,
      "loss/crossentropy": 2.181770443916321,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22411521524190903,
      "step": 9240
    },
    {
      "epoch": 0.18484,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.020157877604166666,
      "learning_rate": 0.0001,
      "loss": 3.9517,
      "loss/crossentropy": 1.8449691534042358,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2091139778494835,
      "step": 9242
    },
    {
      "epoch": 0.18488,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.017195383707682293,
      "learning_rate": 0.0001,
      "loss": 4.3389,
      "loss/crossentropy": 2.0917609333992004,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2644127458333969,
      "step": 9244
    },
    {
      "epoch": 0.18492,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.01793390909830729,
      "learning_rate": 0.0001,
      "loss": 4.083,
      "loss/crossentropy": 2.060012102127075,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21294714510440826,
      "step": 9246
    },
    {
      "epoch": 0.18496,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.018387603759765624,
      "learning_rate": 0.0001,
      "loss": 4.6598,
      "loss/crossentropy": 2.059940278530121,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22648683190345764,
      "step": 9248
    },
    {
      "epoch": 0.185,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.01862360636393229,
      "learning_rate": 0.0001,
      "loss": 4.4696,
      "loss/crossentropy": 1.8423291444778442,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22509171068668365,
      "step": 9250
    },
    {
      "epoch": 0.18504,
      "grad_norm": 2.125,
      "grad_norm_var": 0.016410064697265626,
      "learning_rate": 0.0001,
      "loss": 4.4629,
      "loss/crossentropy": 2.2559698820114136,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23896963894367218,
      "step": 9252
    },
    {
      "epoch": 0.18508,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.01579767862955729,
      "learning_rate": 0.0001,
      "loss": 4.6027,
      "loss/crossentropy": 2.0583658814430237,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22763275355100632,
      "step": 9254
    },
    {
      "epoch": 0.18512,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.016658274332682292,
      "learning_rate": 0.0001,
      "loss": 4.2654,
      "loss/crossentropy": 1.816649854183197,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18608735501766205,
      "step": 9256
    },
    {
      "epoch": 0.18516,
      "grad_norm": 4.65625,
      "grad_norm_var": 0.4073150634765625,
      "learning_rate": 0.0001,
      "loss": 4.2837,
      "loss/crossentropy": 2.0920958518981934,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22859029471874237,
      "step": 9258
    },
    {
      "epoch": 0.1852,
      "grad_norm": 2.125,
      "grad_norm_var": 0.4108306884765625,
      "learning_rate": 0.0001,
      "loss": 4.2517,
      "loss/crossentropy": 2.1475982666015625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21678777784109116,
      "step": 9260
    },
    {
      "epoch": 0.18524,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.4100901285807292,
      "learning_rate": 0.0001,
      "loss": 4.4258,
      "loss/crossentropy": 2.22346031665802,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2333592176437378,
      "step": 9262
    },
    {
      "epoch": 0.18528,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.41646703084309894,
      "learning_rate": 0.0001,
      "loss": 4.2134,
      "loss/crossentropy": 1.7652028799057007,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18461769074201584,
      "step": 9264
    },
    {
      "epoch": 0.18532,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.4225006103515625,
      "learning_rate": 0.0001,
      "loss": 4.2362,
      "loss/crossentropy": 2.1549625396728516,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22428707033395767,
      "step": 9266
    },
    {
      "epoch": 0.18536,
      "grad_norm": 2.0,
      "grad_norm_var": 0.4281972249348958,
      "learning_rate": 0.0001,
      "loss": 4.3685,
      "loss/crossentropy": 2.1677842140197754,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22887174785137177,
      "step": 9268
    },
    {
      "epoch": 0.1854,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.42428792317708336,
      "learning_rate": 0.0001,
      "loss": 4.1517,
      "loss/crossentropy": 1.8352131247520447,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19628287106752396,
      "step": 9270
    },
    {
      "epoch": 0.18544,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.43038304646809894,
      "learning_rate": 0.0001,
      "loss": 4.101,
      "loss/crossentropy": 2.104279100894928,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20481227338314056,
      "step": 9272
    },
    {
      "epoch": 0.18548,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.009795888264973959,
      "learning_rate": 0.0001,
      "loss": 4.2011,
      "loss/crossentropy": 1.8447301387786865,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2227916270494461,
      "step": 9274
    },
    {
      "epoch": 0.18552,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.013651275634765625,
      "learning_rate": 0.0001,
      "loss": 4.3544,
      "loss/crossentropy": 2.303207755088806,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24895529448986053,
      "step": 9276
    },
    {
      "epoch": 0.18556,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.014115142822265624,
      "learning_rate": 0.0001,
      "loss": 4.699,
      "loss/crossentropy": 2.248077630996704,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23077847063541412,
      "step": 9278
    },
    {
      "epoch": 0.1856,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.017438761393229165,
      "learning_rate": 0.0001,
      "loss": 4.4943,
      "loss/crossentropy": 2.261389970779419,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23669905960559845,
      "step": 9280
    },
    {
      "epoch": 0.18564,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.016056060791015625,
      "learning_rate": 0.0001,
      "loss": 4.3805,
      "loss/crossentropy": 2.402338147163391,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2453952580690384,
      "step": 9282
    },
    {
      "epoch": 0.18568,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.01587702433268229,
      "learning_rate": 0.0001,
      "loss": 4.2253,
      "loss/crossentropy": 2.011174201965332,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22673364728689194,
      "step": 9284
    },
    {
      "epoch": 0.18572,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.016810862223307292,
      "learning_rate": 0.0001,
      "loss": 4.3897,
      "loss/crossentropy": 1.9700093269348145,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22711393237113953,
      "step": 9286
    },
    {
      "epoch": 0.18576,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.014922841389973959,
      "learning_rate": 0.0001,
      "loss": 3.9609,
      "loss/crossentropy": 2.0253939032554626,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21902555227279663,
      "step": 9288
    },
    {
      "epoch": 0.1858,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.016961415608723957,
      "learning_rate": 0.0001,
      "loss": 4.1799,
      "loss/crossentropy": 1.741984784603119,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1879509538412094,
      "step": 9290
    },
    {
      "epoch": 0.18584,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.022564442952473958,
      "learning_rate": 0.0001,
      "loss": 4.7983,
      "loss/crossentropy": 2.3156943321228027,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2687100023031235,
      "step": 9292
    },
    {
      "epoch": 0.18588,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.025833892822265624,
      "learning_rate": 0.0001,
      "loss": 4.2894,
      "loss/crossentropy": 2.0826371908187866,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21234872937202454,
      "step": 9294
    },
    {
      "epoch": 0.18592,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.025921376546223958,
      "learning_rate": 0.0001,
      "loss": 4.6473,
      "loss/crossentropy": 2.4080610275268555,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2670409381389618,
      "step": 9296
    },
    {
      "epoch": 0.18596,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.02958958943684896,
      "learning_rate": 0.0001,
      "loss": 4.1997,
      "loss/crossentropy": 1.722270905971527,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20094333589076996,
      "step": 9298
    },
    {
      "epoch": 0.186,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.028527577718098957,
      "learning_rate": 0.0001,
      "loss": 4.3247,
      "loss/crossentropy": 2.0514711141586304,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.209539495408535,
      "step": 9300
    },
    {
      "epoch": 0.18604,
      "grad_norm": 2.640625,
      "grad_norm_var": 2.5507850646972656,
      "learning_rate": 0.0001,
      "loss": 4.9819,
      "loss/crossentropy": 2.4808409214019775,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2367517650127411,
      "step": 9302
    },
    {
      "epoch": 0.18608,
      "grad_norm": 2.078125,
      "grad_norm_var": 2.536018880208333,
      "learning_rate": 0.0001,
      "loss": 4.1105,
      "loss/crossentropy": 2.2539944648742676,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22834083437919617,
      "step": 9304
    },
    {
      "epoch": 0.18612,
      "grad_norm": 2.078125,
      "grad_norm_var": 2.5449544270833333,
      "learning_rate": 0.0001,
      "loss": 4.2383,
      "loss/crossentropy": 1.9335210919380188,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20264607667922974,
      "step": 9306
    },
    {
      "epoch": 0.18616,
      "grad_norm": 2.109375,
      "grad_norm_var": 2.5707194010416665,
      "learning_rate": 0.0001,
      "loss": 4.2285,
      "loss/crossentropy": 1.90863037109375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20882528275251389,
      "step": 9308
    },
    {
      "epoch": 0.1862,
      "grad_norm": 2.1875,
      "grad_norm_var": 2.5589996337890626,
      "learning_rate": 0.0001,
      "loss": 4.2195,
      "loss/crossentropy": 2.1449084281921387,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2329169511795044,
      "step": 9310
    },
    {
      "epoch": 0.18624,
      "grad_norm": 2.015625,
      "grad_norm_var": 2.567341105143229,
      "learning_rate": 0.0001,
      "loss": 4.1806,
      "loss/crossentropy": 1.9571366906166077,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2148708775639534,
      "step": 9312
    },
    {
      "epoch": 0.18628,
      "grad_norm": 2.25,
      "grad_norm_var": 2.542252604166667,
      "learning_rate": 0.0001,
      "loss": 4.4691,
      "loss/crossentropy": 2.174731135368347,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2280728593468666,
      "step": 9314
    },
    {
      "epoch": 0.18632,
      "grad_norm": 1.96875,
      "grad_norm_var": 2.5416575113932294,
      "learning_rate": 0.0001,
      "loss": 4.4344,
      "loss/crossentropy": 1.9569833874702454,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20855706185102463,
      "step": 9316
    },
    {
      "epoch": 0.18636,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0090972900390625,
      "learning_rate": 0.0001,
      "loss": 4.3162,
      "loss/crossentropy": 2.153563976287842,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22807130962610245,
      "step": 9318
    },
    {
      "epoch": 0.1864,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.010741933186848959,
      "learning_rate": 0.0001,
      "loss": 4.3028,
      "loss/crossentropy": 1.9219058752059937,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2295902967453003,
      "step": 9320
    },
    {
      "epoch": 0.18644,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.011926015218098959,
      "learning_rate": 0.0001,
      "loss": 4.29,
      "loss/crossentropy": 2.1993675231933594,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.223759263753891,
      "step": 9322
    },
    {
      "epoch": 0.18648,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.011730702718098958,
      "learning_rate": 0.0001,
      "loss": 4.2254,
      "loss/crossentropy": 2.173740863800049,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23405525833368301,
      "step": 9324
    },
    {
      "epoch": 0.18652,
      "grad_norm": 2.25,
      "grad_norm_var": 0.012564849853515626,
      "learning_rate": 0.0001,
      "loss": 4.3717,
      "loss/crossentropy": 2.026577115058899,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2265823632478714,
      "step": 9326
    },
    {
      "epoch": 0.18656,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.030326080322265626,
      "learning_rate": 0.0001,
      "loss": 4.4664,
      "loss/crossentropy": 2.3629637956619263,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23659023642539978,
      "step": 9328
    },
    {
      "epoch": 0.1866,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.03050715128580729,
      "learning_rate": 0.0001,
      "loss": 4.3245,
      "loss/crossentropy": 2.100727915763855,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21468425542116165,
      "step": 9330
    },
    {
      "epoch": 0.18664,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.025923411051432293,
      "learning_rate": 0.0001,
      "loss": 4.5291,
      "loss/crossentropy": 2.163568615913391,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23402437567710876,
      "step": 9332
    },
    {
      "epoch": 0.18668,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.02934748331705729,
      "learning_rate": 0.0001,
      "loss": 3.9687,
      "loss/crossentropy": 1.9579994082450867,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19713342934846878,
      "step": 9334
    },
    {
      "epoch": 0.18672,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.029412587483723957,
      "learning_rate": 0.0001,
      "loss": 4.2586,
      "loss/crossentropy": 1.805375874042511,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22647518664598465,
      "step": 9336
    },
    {
      "epoch": 0.18676,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.02797215779622396,
      "learning_rate": 0.0001,
      "loss": 4.2025,
      "loss/crossentropy": 2.0764458775520325,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2270444855093956,
      "step": 9338
    },
    {
      "epoch": 0.1868,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.030987294514973958,
      "learning_rate": 0.0001,
      "loss": 3.8731,
      "loss/crossentropy": 1.5654467940330505,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15944529324769974,
      "step": 9340
    },
    {
      "epoch": 0.18684,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.02976048787434896,
      "learning_rate": 0.0001,
      "loss": 4.3831,
      "loss/crossentropy": 2.007612407207489,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2061111181974411,
      "step": 9342
    },
    {
      "epoch": 0.18688,
      "grad_norm": 1.875,
      "grad_norm_var": 0.010910797119140624,
      "learning_rate": 0.0001,
      "loss": 4.0835,
      "loss/crossentropy": 1.981432855129242,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19339826703071594,
      "step": 9344
    },
    {
      "epoch": 0.18692,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.00965576171875,
      "learning_rate": 0.0001,
      "loss": 4.1291,
      "loss/crossentropy": 1.804275631904602,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1968545839190483,
      "step": 9346
    },
    {
      "epoch": 0.18696,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.008763631184895834,
      "learning_rate": 0.0001,
      "loss": 4.2109,
      "loss/crossentropy": 1.9718617796897888,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21482623368501663,
      "step": 9348
    },
    {
      "epoch": 0.187,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.008397420247395834,
      "learning_rate": 0.0001,
      "loss": 4.1672,
      "loss/crossentropy": 1.8358338475227356,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20042669028043747,
      "step": 9350
    },
    {
      "epoch": 0.18704,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.009089914957682292,
      "learning_rate": 0.0001,
      "loss": 4.2045,
      "loss/crossentropy": 1.9447709321975708,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21207460761070251,
      "step": 9352
    },
    {
      "epoch": 0.18708,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.011356353759765625,
      "learning_rate": 0.0001,
      "loss": 4.3134,
      "loss/crossentropy": 1.8921163082122803,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19945629686117172,
      "step": 9354
    },
    {
      "epoch": 0.18712,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.009580230712890625,
      "learning_rate": 0.0001,
      "loss": 4.2018,
      "loss/crossentropy": 2.1323947310447693,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2258753925561905,
      "step": 9356
    },
    {
      "epoch": 0.18716,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.009277089436848959,
      "learning_rate": 0.0001,
      "loss": 4.4278,
      "loss/crossentropy": 2.064914345741272,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23450962454080582,
      "step": 9358
    },
    {
      "epoch": 0.1872,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.005783843994140625,
      "learning_rate": 0.0001,
      "loss": 4.0173,
      "loss/crossentropy": 2.1264703273773193,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2169734537601471,
      "step": 9360
    },
    {
      "epoch": 0.18724,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.004784138997395834,
      "learning_rate": 0.0001,
      "loss": 4.2926,
      "loss/crossentropy": 1.7044150233268738,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19169463962316513,
      "step": 9362
    },
    {
      "epoch": 0.18728,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.004541015625,
      "learning_rate": 0.0001,
      "loss": 4.5367,
      "loss/crossentropy": 1.9475398659706116,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22891414165496826,
      "step": 9364
    },
    {
      "epoch": 0.18732,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.008017730712890626,
      "learning_rate": 0.0001,
      "loss": 4.1416,
      "loss/crossentropy": 2.134114623069763,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2241017445921898,
      "step": 9366
    },
    {
      "epoch": 0.18736,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.007342274983723958,
      "learning_rate": 0.0001,
      "loss": 4.2185,
      "loss/crossentropy": 1.8317620158195496,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20189791917800903,
      "step": 9368
    },
    {
      "epoch": 0.1874,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.005041249593098958,
      "learning_rate": 0.0001,
      "loss": 4.3518,
      "loss/crossentropy": 2.257538855075836,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2343401461839676,
      "step": 9370
    },
    {
      "epoch": 0.18744,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.005228424072265625,
      "learning_rate": 0.0001,
      "loss": 4.0535,
      "loss/crossentropy": 2.3722634315490723,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23045828938484192,
      "step": 9372
    },
    {
      "epoch": 0.18748,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.005222320556640625,
      "learning_rate": 0.0001,
      "loss": 4.2293,
      "loss/crossentropy": 2.1880545020103455,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2345728725194931,
      "step": 9374
    },
    {
      "epoch": 0.18752,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.008255767822265624,
      "learning_rate": 0.0001,
      "loss": 4.5037,
      "loss/crossentropy": 1.8883287906646729,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25232937932014465,
      "step": 9376
    },
    {
      "epoch": 0.18756,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.009124501546223959,
      "learning_rate": 0.0001,
      "loss": 4.2963,
      "loss/crossentropy": 1.9253730773925781,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20324261486530304,
      "step": 9378
    },
    {
      "epoch": 0.1876,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.009211985270182292,
      "learning_rate": 0.0001,
      "loss": 4.2745,
      "loss/crossentropy": 1.961540937423706,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22623597085475922,
      "step": 9380
    },
    {
      "epoch": 0.18764,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.007835896809895833,
      "learning_rate": 0.0001,
      "loss": 4.347,
      "loss/crossentropy": 2.190120279788971,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24154195934534073,
      "step": 9382
    },
    {
      "epoch": 0.18768,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.007445271809895833,
      "learning_rate": 0.0001,
      "loss": 4.6159,
      "loss/crossentropy": 2.0888350009918213,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22750811278820038,
      "step": 9384
    },
    {
      "epoch": 0.18772,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.008128865559895834,
      "learning_rate": 0.0001,
      "loss": 4.3171,
      "loss/crossentropy": 1.724816918373108,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21245518326759338,
      "step": 9386
    },
    {
      "epoch": 0.18776,
      "grad_norm": 2.0,
      "grad_norm_var": 0.009307607014973959,
      "learning_rate": 0.0001,
      "loss": 4.0868,
      "loss/crossentropy": 1.6542762517929077,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1961463838815689,
      "step": 9388
    },
    {
      "epoch": 0.1878,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.009714508056640625,
      "learning_rate": 0.0001,
      "loss": 4.4748,
      "loss/crossentropy": 2.3528761863708496,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23583710938692093,
      "step": 9390
    },
    {
      "epoch": 0.18784,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009012603759765625,
      "learning_rate": 0.0001,
      "loss": 4.394,
      "loss/crossentropy": 2.181188702583313,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24201467633247375,
      "step": 9392
    },
    {
      "epoch": 0.18788,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.008414459228515626,
      "learning_rate": 0.0001,
      "loss": 4.3476,
      "loss/crossentropy": 2.280683398246765,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22065220028162003,
      "step": 9394
    },
    {
      "epoch": 0.18792,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.009275054931640625,
      "learning_rate": 0.0001,
      "loss": 4.1603,
      "loss/crossentropy": 1.9839438199996948,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22708184272050858,
      "step": 9396
    },
    {
      "epoch": 0.18796,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0070879618326822914,
      "learning_rate": 0.0001,
      "loss": 4.4322,
      "loss/crossentropy": 2.311874270439148,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22249652445316315,
      "step": 9398
    },
    {
      "epoch": 0.188,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.008503214518229166,
      "learning_rate": 0.0001,
      "loss": 4.3808,
      "loss/crossentropy": 2.430112838745117,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23544297367334366,
      "step": 9400
    },
    {
      "epoch": 0.18804,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.00892333984375,
      "learning_rate": 0.0001,
      "loss": 4.2393,
      "loss/crossentropy": 2.146397888660431,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23211465775966644,
      "step": 9402
    },
    {
      "epoch": 0.18808,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.008318837483723958,
      "learning_rate": 0.0001,
      "loss": 4.5071,
      "loss/crossentropy": 2.4402170181274414,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25594406574964523,
      "step": 9404
    },
    {
      "epoch": 0.18812,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.008294423421223959,
      "learning_rate": 0.0001,
      "loss": 4.3848,
      "loss/crossentropy": 2.141621232032776,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2156461626291275,
      "step": 9406
    },
    {
      "epoch": 0.18816,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.006290435791015625,
      "learning_rate": 0.0001,
      "loss": 4.2511,
      "loss/crossentropy": 2.0266456604003906,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20194754749536514,
      "step": 9408
    },
    {
      "epoch": 0.1882,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.007062784830729167,
      "learning_rate": 0.0001,
      "loss": 4.2831,
      "loss/crossentropy": 2.304496645927429,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22345459461212158,
      "step": 9410
    },
    {
      "epoch": 0.18824,
      "grad_norm": 2.25,
      "grad_norm_var": 0.007523600260416667,
      "learning_rate": 0.0001,
      "loss": 4.4085,
      "loss/crossentropy": 1.8486470580101013,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20493299514055252,
      "step": 9412
    },
    {
      "epoch": 0.18828,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0075927734375,
      "learning_rate": 0.0001,
      "loss": 4.4562,
      "loss/crossentropy": 2.0593990683555603,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22942744940519333,
      "step": 9414
    },
    {
      "epoch": 0.18832,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.006528472900390625,
      "learning_rate": 0.0001,
      "loss": 4.1551,
      "loss/crossentropy": 1.7936111688613892,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1776513010263443,
      "step": 9416
    },
    {
      "epoch": 0.18836,
      "grad_norm": 2.0,
      "grad_norm_var": 0.005997467041015625,
      "learning_rate": 0.0001,
      "loss": 4.4484,
      "loss/crossentropy": 2.0676616430282593,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21748381853103638,
      "step": 9418
    },
    {
      "epoch": 0.1884,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.005236562093098958,
      "learning_rate": 0.0001,
      "loss": 4.3415,
      "loss/crossentropy": 2.0317665934562683,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2215682566165924,
      "step": 9420
    },
    {
      "epoch": 0.18844,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.006359608968098959,
      "learning_rate": 0.0001,
      "loss": 4.5623,
      "loss/crossentropy": 2.3345483541488647,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24850556254386902,
      "step": 9422
    },
    {
      "epoch": 0.18848,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.007043202718098958,
      "learning_rate": 0.0001,
      "loss": 4.3526,
      "loss/crossentropy": 2.0603779554367065,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20997442305088043,
      "step": 9424
    },
    {
      "epoch": 0.18852,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.006591796875,
      "learning_rate": 0.0001,
      "loss": 4.4944,
      "loss/crossentropy": 1.9450209140777588,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21465806663036346,
      "step": 9426
    },
    {
      "epoch": 0.18856,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0065266927083333336,
      "learning_rate": 0.0001,
      "loss": 4.2452,
      "loss/crossentropy": 2.3568087816238403,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24619507789611816,
      "step": 9428
    },
    {
      "epoch": 0.1886,
      "grad_norm": 2.25,
      "grad_norm_var": 0.007828776041666667,
      "learning_rate": 0.0001,
      "loss": 4.1976,
      "loss/crossentropy": 2.2047020196914673,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2519628629088402,
      "step": 9430
    },
    {
      "epoch": 0.18864,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.010188547770182292,
      "learning_rate": 0.0001,
      "loss": 4.3168,
      "loss/crossentropy": 2.1594117879867554,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21274243295192719,
      "step": 9432
    },
    {
      "epoch": 0.18868,
      "grad_norm": 2.25,
      "grad_norm_var": 0.009936269124348958,
      "learning_rate": 0.0001,
      "loss": 4.3847,
      "loss/crossentropy": 2.120336890220642,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2092270776629448,
      "step": 9434
    },
    {
      "epoch": 0.18872,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011120351155598958,
      "learning_rate": 0.0001,
      "loss": 4.2725,
      "loss/crossentropy": 2.13198459148407,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.238224595785141,
      "step": 9436
    },
    {
      "epoch": 0.18876,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.010009511311848959,
      "learning_rate": 0.0001,
      "loss": 4.3706,
      "loss/crossentropy": 1.9252395629882812,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22254322469234467,
      "step": 9438
    },
    {
      "epoch": 0.1888,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0123291015625,
      "learning_rate": 0.0001,
      "loss": 4.0574,
      "loss/crossentropy": 2.123266577720642,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22276867926120758,
      "step": 9440
    },
    {
      "epoch": 0.18884,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.010741170247395833,
      "learning_rate": 0.0001,
      "loss": 4.2195,
      "loss/crossentropy": 1.9219747185707092,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18884174525737762,
      "step": 9442
    },
    {
      "epoch": 0.18888,
      "grad_norm": 2.125,
      "grad_norm_var": 0.010587565104166667,
      "learning_rate": 0.0001,
      "loss": 4.2603,
      "loss/crossentropy": 2.0122207403182983,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20913395285606384,
      "step": 9444
    },
    {
      "epoch": 0.18892,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.010632069905598958,
      "learning_rate": 0.0001,
      "loss": 4.0871,
      "loss/crossentropy": 2.0255361199378967,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1982431635260582,
      "step": 9446
    },
    {
      "epoch": 0.18896,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.008770497639973958,
      "learning_rate": 0.0001,
      "loss": 4.4085,
      "loss/crossentropy": 1.8060500025749207,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19787351042032242,
      "step": 9448
    },
    {
      "epoch": 0.189,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.007389068603515625,
      "learning_rate": 0.0001,
      "loss": 4.3799,
      "loss/crossentropy": 2.340656042098999,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22321298718452454,
      "step": 9450
    },
    {
      "epoch": 0.18904,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.00966796875,
      "learning_rate": 0.0001,
      "loss": 4.3973,
      "loss/crossentropy": 2.35786235332489,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2356308028101921,
      "step": 9452
    },
    {
      "epoch": 0.18908,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.01002197265625,
      "learning_rate": 0.0001,
      "loss": 4.2622,
      "loss/crossentropy": 2.344806671142578,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21194154769182205,
      "step": 9454
    },
    {
      "epoch": 0.18912,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.008786773681640625,
      "learning_rate": 0.0001,
      "loss": 4.4861,
      "loss/crossentropy": 2.2449493408203125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21038557589054108,
      "step": 9456
    },
    {
      "epoch": 0.18916,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.014518229166666667,
      "learning_rate": 0.0001,
      "loss": 4.0343,
      "loss/crossentropy": 1.977162778377533,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20586547255516052,
      "step": 9458
    },
    {
      "epoch": 0.1892,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.015860748291015626,
      "learning_rate": 0.0001,
      "loss": 4.3453,
      "loss/crossentropy": 2.0941065549850464,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2197253629565239,
      "step": 9460
    },
    {
      "epoch": 0.18924,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0143463134765625,
      "learning_rate": 0.0001,
      "loss": 4.3479,
      "loss/crossentropy": 2.5145565271377563,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25138507783412933,
      "step": 9462
    },
    {
      "epoch": 0.18928,
      "grad_norm": 2.0,
      "grad_norm_var": 0.014289347330729167,
      "learning_rate": 0.0001,
      "loss": 4.4076,
      "loss/crossentropy": 2.2870718240737915,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2298167496919632,
      "step": 9464
    },
    {
      "epoch": 0.18932,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.013678995768229167,
      "learning_rate": 0.0001,
      "loss": 4.3588,
      "loss/crossentropy": 2.2095978260040283,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.217079259455204,
      "step": 9466
    },
    {
      "epoch": 0.18936,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.011726633707682291,
      "learning_rate": 0.0001,
      "loss": 4.3735,
      "loss/crossentropy": 1.9591819047927856,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21825183182954788,
      "step": 9468
    },
    {
      "epoch": 0.1894,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.011352284749348959,
      "learning_rate": 0.0001,
      "loss": 4.5072,
      "loss/crossentropy": 2.266845226287842,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24172072112560272,
      "step": 9470
    },
    {
      "epoch": 0.18944,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.011437733968098959,
      "learning_rate": 0.0001,
      "loss": 4.3448,
      "loss/crossentropy": 2.0732688903808594,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23074156790971756,
      "step": 9472
    },
    {
      "epoch": 0.18948,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.0090728759765625,
      "learning_rate": 0.0001,
      "loss": 4.6697,
      "loss/crossentropy": 1.935340702533722,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2194545865058899,
      "step": 9474
    },
    {
      "epoch": 0.18952,
      "grad_norm": 2.125,
      "grad_norm_var": 0.007594553629557291,
      "learning_rate": 0.0001,
      "loss": 4.1518,
      "loss/crossentropy": 1.795669674873352,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18794939666986465,
      "step": 9476
    },
    {
      "epoch": 0.18956,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.008841705322265626,
      "learning_rate": 0.0001,
      "loss": 3.7745,
      "loss/crossentropy": 1.9292446970939636,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20147182047367096,
      "step": 9478
    },
    {
      "epoch": 0.1896,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0067779541015625,
      "learning_rate": 0.0001,
      "loss": 4.1955,
      "loss/crossentropy": 2.0110061168670654,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2096811756491661,
      "step": 9480
    },
    {
      "epoch": 0.18964,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.008034006754557291,
      "learning_rate": 0.0001,
      "loss": 4.2937,
      "loss/crossentropy": 2.1543468236923218,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2142205834388733,
      "step": 9482
    },
    {
      "epoch": 0.18968,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.009723917643229166,
      "learning_rate": 0.0001,
      "loss": 4.0126,
      "loss/crossentropy": 2.0860520601272583,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21716032177209854,
      "step": 9484
    },
    {
      "epoch": 0.18972,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.010241444905598958,
      "learning_rate": 0.0001,
      "loss": 3.9412,
      "loss/crossentropy": 1.7190409302711487,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19397014379501343,
      "step": 9486
    },
    {
      "epoch": 0.18976,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.017116038004557292,
      "learning_rate": 0.0001,
      "loss": 4.4711,
      "loss/crossentropy": 2.178081512451172,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2280048429965973,
      "step": 9488
    },
    {
      "epoch": 0.1898,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.013063303629557292,
      "learning_rate": 0.0001,
      "loss": 4.267,
      "loss/crossentropy": 2.0749863982200623,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22755203396081924,
      "step": 9490
    },
    {
      "epoch": 0.18984,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.014098866780598959,
      "learning_rate": 0.0001,
      "loss": 4.3147,
      "loss/crossentropy": 2.214204430580139,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23071825504302979,
      "step": 9492
    },
    {
      "epoch": 0.18988,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.015794881184895835,
      "learning_rate": 0.0001,
      "loss": 4.0041,
      "loss/crossentropy": 1.6625414490699768,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1857389286160469,
      "step": 9494
    },
    {
      "epoch": 0.18992,
      "grad_norm": 2.375,
      "grad_norm_var": 0.020539347330729166,
      "learning_rate": 0.0001,
      "loss": 4.4474,
      "loss/crossentropy": 1.8537201285362244,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21718977391719818,
      "step": 9496
    },
    {
      "epoch": 0.18996,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.024074045817057292,
      "learning_rate": 0.0001,
      "loss": 4.0488,
      "loss/crossentropy": 1.716725468635559,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18544895946979523,
      "step": 9498
    },
    {
      "epoch": 0.19,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.024192047119140626,
      "learning_rate": 0.0001,
      "loss": 4.1035,
      "loss/crossentropy": 1.94467431306839,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.210828959941864,
      "step": 9500
    },
    {
      "epoch": 0.19004,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.023298136393229165,
      "learning_rate": 0.0001,
      "loss": 4.2383,
      "loss/crossentropy": 1.9377062320709229,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22141354531049728,
      "step": 9502
    },
    {
      "epoch": 0.19008,
      "grad_norm": 2.0,
      "grad_norm_var": 0.016950480143229165,
      "learning_rate": 0.0001,
      "loss": 4.3717,
      "loss/crossentropy": 2.2015358209609985,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22363336384296417,
      "step": 9504
    },
    {
      "epoch": 0.19012,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.017235310872395833,
      "learning_rate": 0.0001,
      "loss": 4.2332,
      "loss/crossentropy": 2.373024582862854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2402234748005867,
      "step": 9506
    },
    {
      "epoch": 0.19016,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.016649373372395835,
      "learning_rate": 0.0001,
      "loss": 4.0519,
      "loss/crossentropy": 2.1573110222816467,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24207139760255814,
      "step": 9508
    },
    {
      "epoch": 0.1902,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.014788564046223958,
      "learning_rate": 0.0001,
      "loss": 4.4041,
      "loss/crossentropy": 2.324455976486206,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24618541449308395,
      "step": 9510
    },
    {
      "epoch": 0.19024,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.010603586832682291,
      "learning_rate": 0.0001,
      "loss": 4.0065,
      "loss/crossentropy": 1.7480111718177795,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2049119919538498,
      "step": 9512
    },
    {
      "epoch": 0.19028,
      "grad_norm": 2.125,
      "grad_norm_var": 0.006300608317057292,
      "learning_rate": 0.0001,
      "loss": 4.3644,
      "loss/crossentropy": 1.9925439953804016,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2314591035246849,
      "step": 9514
    },
    {
      "epoch": 0.19032,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.007094065348307292,
      "learning_rate": 0.0001,
      "loss": 4.0702,
      "loss/crossentropy": 2.133601188659668,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2329796403646469,
      "step": 9516
    },
    {
      "epoch": 0.19036,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.010198720296223958,
      "learning_rate": 0.0001,
      "loss": 4.246,
      "loss/crossentropy": 2.093464970588684,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22636859863996506,
      "step": 9518
    },
    {
      "epoch": 0.1904,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009757232666015626,
      "learning_rate": 0.0001,
      "loss": 4.4766,
      "loss/crossentropy": 2.6137614250183105,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27037859708070755,
      "step": 9520
    },
    {
      "epoch": 0.19044,
      "grad_norm": 2.125,
      "grad_norm_var": 0.012300364176432292,
      "learning_rate": 0.0001,
      "loss": 4.5499,
      "loss/crossentropy": 2.008640229701996,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21647297590970993,
      "step": 9522
    },
    {
      "epoch": 0.19048,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.012286122639973958,
      "learning_rate": 0.0001,
      "loss": 4.2467,
      "loss/crossentropy": 2.12644362449646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23370585590600967,
      "step": 9524
    },
    {
      "epoch": 0.19052,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.011425526936848958,
      "learning_rate": 0.0001,
      "loss": 4.5409,
      "loss/crossentropy": 2.2338638305664062,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23637598007917404,
      "step": 9526
    },
    {
      "epoch": 0.19056,
      "grad_norm": 2.0,
      "grad_norm_var": 0.011785634358723958,
      "learning_rate": 0.0001,
      "loss": 4.1405,
      "loss/crossentropy": 2.0632832646369934,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20682457089424133,
      "step": 9528
    },
    {
      "epoch": 0.1906,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.012726847330729167,
      "learning_rate": 0.0001,
      "loss": 4.4058,
      "loss/crossentropy": 2.219120740890503,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22551076859235764,
      "step": 9530
    },
    {
      "epoch": 0.19064,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.011860911051432292,
      "learning_rate": 0.0001,
      "loss": 4.2985,
      "loss/crossentropy": 2.4633371829986572,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24491076171398163,
      "step": 9532
    },
    {
      "epoch": 0.19068,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008211008707682292,
      "learning_rate": 0.0001,
      "loss": 4.2152,
      "loss/crossentropy": 2.2408339977264404,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22265981882810593,
      "step": 9534
    },
    {
      "epoch": 0.19072,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.009544881184895833,
      "learning_rate": 0.0001,
      "loss": 4.326,
      "loss/crossentropy": 1.9779353141784668,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2159332111477852,
      "step": 9536
    },
    {
      "epoch": 0.19076,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.0082275390625,
      "learning_rate": 0.0001,
      "loss": 4.3522,
      "loss/crossentropy": 2.0315812826156616,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20710154622793198,
      "step": 9538
    },
    {
      "epoch": 0.1908,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.011787923177083333,
      "learning_rate": 0.0001,
      "loss": 4.18,
      "loss/crossentropy": 2.085337817668915,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22174393385648727,
      "step": 9540
    },
    {
      "epoch": 0.19084,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.016507975260416665,
      "learning_rate": 0.0001,
      "loss": 4.4386,
      "loss/crossentropy": 2.3449169397354126,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3135230466723442,
      "step": 9542
    },
    {
      "epoch": 0.19088,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.016169230143229168,
      "learning_rate": 0.0001,
      "loss": 4.3943,
      "loss/crossentropy": 2.1083431243896484,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21395261585712433,
      "step": 9544
    },
    {
      "epoch": 0.19092,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.015730539957682293,
      "learning_rate": 0.0001,
      "loss": 4.2091,
      "loss/crossentropy": 2.0386710166931152,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22487390786409378,
      "step": 9546
    },
    {
      "epoch": 0.19096,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.015240224202473958,
      "learning_rate": 0.0001,
      "loss": 4.2377,
      "loss/crossentropy": 1.9533087611198425,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2223353162407875,
      "step": 9548
    },
    {
      "epoch": 0.191,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.015317535400390625,
      "learning_rate": 0.0001,
      "loss": 4.4183,
      "loss/crossentropy": 2.35861599445343,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26162558794021606,
      "step": 9550
    },
    {
      "epoch": 0.19104,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.014388020833333333,
      "learning_rate": 0.0001,
      "loss": 4.2507,
      "loss/crossentropy": 1.9529705047607422,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2125585675239563,
      "step": 9552
    },
    {
      "epoch": 0.19108,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.011844889322916666,
      "learning_rate": 0.0001,
      "loss": 4.2041,
      "loss/crossentropy": 2.0364453196525574,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21163037419319153,
      "step": 9554
    },
    {
      "epoch": 0.19112,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.009749348958333333,
      "learning_rate": 0.0001,
      "loss": 4.2758,
      "loss/crossentropy": 2.1321409940719604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24250106513500214,
      "step": 9556
    },
    {
      "epoch": 0.19116,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0073931376139322914,
      "learning_rate": 0.0001,
      "loss": 4.1848,
      "loss/crossentropy": 2.024011969566345,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20132286846637726,
      "step": 9558
    },
    {
      "epoch": 0.1912,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.006980133056640625,
      "learning_rate": 0.0001,
      "loss": 4.3726,
      "loss/crossentropy": 2.106776535511017,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23114337772130966,
      "step": 9560
    },
    {
      "epoch": 0.19124,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.007645416259765625,
      "learning_rate": 0.0001,
      "loss": 4.4642,
      "loss/crossentropy": 1.9228236079216003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22426098585128784,
      "step": 9562
    },
    {
      "epoch": 0.19128,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.008090972900390625,
      "learning_rate": 0.0001,
      "loss": 4.2384,
      "loss/crossentropy": 2.3033370971679688,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20113499462604523,
      "step": 9564
    },
    {
      "epoch": 0.19132,
      "grad_norm": 2.125,
      "grad_norm_var": 0.008973948160807292,
      "learning_rate": 0.0001,
      "loss": 4.0736,
      "loss/crossentropy": 1.6983963251113892,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19003060460090637,
      "step": 9566
    },
    {
      "epoch": 0.19136,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.011295318603515625,
      "learning_rate": 0.0001,
      "loss": 4.5283,
      "loss/crossentropy": 2.502004861831665,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24949797987937927,
      "step": 9568
    },
    {
      "epoch": 0.1914,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.011793772379557291,
      "learning_rate": 0.0001,
      "loss": 4.2004,
      "loss/crossentropy": 1.9981504678726196,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2152494639158249,
      "step": 9570
    },
    {
      "epoch": 0.19144,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.012931060791015626,
      "learning_rate": 0.0001,
      "loss": 4.1211,
      "loss/crossentropy": 2.1489784717559814,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22105325013399124,
      "step": 9572
    },
    {
      "epoch": 0.19148,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.013016510009765624,
      "learning_rate": 0.0001,
      "loss": 4.2234,
      "loss/crossentropy": 1.974421203136444,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23960395902395248,
      "step": 9574
    },
    {
      "epoch": 0.19152,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.012670644124348958,
      "learning_rate": 0.0001,
      "loss": 4.4119,
      "loss/crossentropy": 2.379599928855896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26731471717357635,
      "step": 9576
    },
    {
      "epoch": 0.19156,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.012444814046223959,
      "learning_rate": 0.0001,
      "loss": 4.3718,
      "loss/crossentropy": 2.26211154460907,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22957566380500793,
      "step": 9578
    },
    {
      "epoch": 0.1916,
      "grad_norm": 2.125,
      "grad_norm_var": 0.011871083577473959,
      "learning_rate": 0.0001,
      "loss": 4.4594,
      "loss/crossentropy": 2.282869577407837,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2404445931315422,
      "step": 9580
    },
    {
      "epoch": 0.19164,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.010457102457682292,
      "learning_rate": 0.0001,
      "loss": 4.1592,
      "loss/crossentropy": 1.9183810949325562,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18672315031290054,
      "step": 9582
    },
    {
      "epoch": 0.19168,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0076812744140625,
      "learning_rate": 0.0001,
      "loss": 4.1801,
      "loss/crossentropy": 2.2722173929214478,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23262400180101395,
      "step": 9584
    },
    {
      "epoch": 0.19172,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0075266520182291664,
      "learning_rate": 0.0001,
      "loss": 4.4032,
      "loss/crossentropy": 2.426178455352783,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23461100459098816,
      "step": 9586
    },
    {
      "epoch": 0.19176,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.00687255859375,
      "learning_rate": 0.0001,
      "loss": 4.344,
      "loss/crossentropy": 2.2266165018081665,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22530251741409302,
      "step": 9588
    },
    {
      "epoch": 0.1918,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.005541737874348958,
      "learning_rate": 0.0001,
      "loss": 4.1633,
      "loss/crossentropy": 1.786275327205658,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19257958233356476,
      "step": 9590
    },
    {
      "epoch": 0.19184,
      "grad_norm": 2.0,
      "grad_norm_var": 0.008868153889973958,
      "learning_rate": 0.0001,
      "loss": 4.2367,
      "loss/crossentropy": 1.8497431874275208,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21965742111206055,
      "step": 9592
    },
    {
      "epoch": 0.19188,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.008143870035807292,
      "learning_rate": 0.0001,
      "loss": 4.5612,
      "loss/crossentropy": 2.492846131324768,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21851783990859985,
      "step": 9594
    },
    {
      "epoch": 0.19192,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.010027821858723958,
      "learning_rate": 0.0001,
      "loss": 4.6838,
      "loss/crossentropy": 2.3447986841201782,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26567359268665314,
      "step": 9596
    },
    {
      "epoch": 0.19196,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.010654449462890625,
      "learning_rate": 0.0001,
      "loss": 4.1513,
      "loss/crossentropy": 1.656063199043274,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1903422325849533,
      "step": 9598
    },
    {
      "epoch": 0.192,
      "grad_norm": 2.125,
      "grad_norm_var": 0.008329264322916667,
      "learning_rate": 0.0001,
      "loss": 4.3252,
      "loss/crossentropy": 2.1816134452819824,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23314762860536575,
      "step": 9600
    },
    {
      "epoch": 0.19204,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010007476806640625,
      "learning_rate": 0.0001,
      "loss": 3.8537,
      "loss/crossentropy": 1.787261426448822,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20938758552074432,
      "step": 9602
    },
    {
      "epoch": 0.19208,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010526275634765625,
      "learning_rate": 0.0001,
      "loss": 4.2374,
      "loss/crossentropy": 1.9722678065299988,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23085469752550125,
      "step": 9604
    },
    {
      "epoch": 0.19212,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.012113444010416667,
      "learning_rate": 0.0001,
      "loss": 4.0954,
      "loss/crossentropy": 2.04559987783432,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21647220849990845,
      "step": 9606
    },
    {
      "epoch": 0.19216,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.008934529622395833,
      "learning_rate": 0.0001,
      "loss": 4.5397,
      "loss/crossentropy": 2.5426105260849,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23680832237005234,
      "step": 9608
    },
    {
      "epoch": 0.1922,
      "grad_norm": 2.0,
      "grad_norm_var": 0.008890787760416666,
      "learning_rate": 0.0001,
      "loss": 4.245,
      "loss/crossentropy": 2.1339075565338135,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22501112520694733,
      "step": 9610
    },
    {
      "epoch": 0.19224,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.006550852457682292,
      "learning_rate": 0.0001,
      "loss": 4.0924,
      "loss/crossentropy": 1.9554831981658936,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20187357813119888,
      "step": 9612
    },
    {
      "epoch": 0.19228,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.009065500895182292,
      "learning_rate": 0.0001,
      "loss": 4.5578,
      "loss/crossentropy": 2.2996249198913574,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24954287707805634,
      "step": 9614
    },
    {
      "epoch": 0.19232,
      "grad_norm": 2.125,
      "grad_norm_var": 0.009065500895182292,
      "learning_rate": 0.0001,
      "loss": 4.3753,
      "loss/crossentropy": 2.2439414262771606,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22833245247602463,
      "step": 9616
    },
    {
      "epoch": 0.19236,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.007933553059895833,
      "learning_rate": 0.0001,
      "loss": 4.2297,
      "loss/crossentropy": 1.8900890946388245,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21077623218297958,
      "step": 9618
    },
    {
      "epoch": 0.1924,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.007804361979166666,
      "learning_rate": 0.0001,
      "loss": 4.287,
      "loss/crossentropy": 2.110726058483124,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21897974610328674,
      "step": 9620
    },
    {
      "epoch": 0.19244,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0080474853515625,
      "learning_rate": 0.0001,
      "loss": 4.0135,
      "loss/crossentropy": 1.7363090515136719,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19939633458852768,
      "step": 9622
    },
    {
      "epoch": 0.19248,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.01014404296875,
      "learning_rate": 0.0001,
      "loss": 4.3643,
      "loss/crossentropy": 1.8148014545440674,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21637701243162155,
      "step": 9624
    },
    {
      "epoch": 0.19252,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0092041015625,
      "learning_rate": 0.0001,
      "loss": 4.2451,
      "loss/crossentropy": 2.2895134687423706,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2297188639640808,
      "step": 9626
    },
    {
      "epoch": 0.19256,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.006296539306640625,
      "learning_rate": 0.0001,
      "loss": 4.4778,
      "loss/crossentropy": 2.117924213409424,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24057473242282867,
      "step": 9628
    },
    {
      "epoch": 0.1926,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.005421702067057292,
      "learning_rate": 0.0001,
      "loss": 4.2791,
      "loss/crossentropy": 1.8709319829940796,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2106107696890831,
      "step": 9630
    },
    {
      "epoch": 0.19264,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0079742431640625,
      "learning_rate": 0.0001,
      "loss": 4.0174,
      "loss/crossentropy": 1.5156871676445007,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17487400770187378,
      "step": 9632
    },
    {
      "epoch": 0.19268,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0183837890625,
      "learning_rate": 0.0001,
      "loss": 4.6281,
      "loss/crossentropy": 2.153126537799835,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22009101510047913,
      "step": 9634
    },
    {
      "epoch": 0.19272,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.017853800455729166,
      "learning_rate": 0.0001,
      "loss": 4.443,
      "loss/crossentropy": 2.0890414714813232,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2233077436685562,
      "step": 9636
    },
    {
      "epoch": 0.19276,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.01858495076497396,
      "learning_rate": 0.0001,
      "loss": 4.2572,
      "loss/crossentropy": 2.1218496561050415,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22576671838760376,
      "step": 9638
    },
    {
      "epoch": 0.1928,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.020918528238932293,
      "learning_rate": 0.0001,
      "loss": 4.2522,
      "loss/crossentropy": 2.1131649017333984,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21846124529838562,
      "step": 9640
    },
    {
      "epoch": 0.19284,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.02067845662434896,
      "learning_rate": 0.0001,
      "loss": 4.6643,
      "loss/crossentropy": 2.3941714763641357,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.246211439371109,
      "step": 9642
    },
    {
      "epoch": 0.19288,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.02687352498372396,
      "learning_rate": 0.0001,
      "loss": 4.0641,
      "loss/crossentropy": 1.9579638838768005,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.198220357298851,
      "step": 9644
    },
    {
      "epoch": 0.19292,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.02754491170247396,
      "learning_rate": 0.0001,
      "loss": 4.564,
      "loss/crossentropy": 2.388027787208557,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23349716514348984,
      "step": 9646
    },
    {
      "epoch": 0.19296,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.02520726521809896,
      "learning_rate": 0.0001,
      "loss": 4.3678,
      "loss/crossentropy": 1.8203087449073792,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18923642486333847,
      "step": 9648
    },
    {
      "epoch": 0.193,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.015476226806640625,
      "learning_rate": 0.0001,
      "loss": 4.2119,
      "loss/crossentropy": 1.996269702911377,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2096453383564949,
      "step": 9650
    },
    {
      "epoch": 0.19304,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.01572240193684896,
      "learning_rate": 0.0001,
      "loss": 4.2921,
      "loss/crossentropy": 1.806606113910675,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19824761897325516,
      "step": 9652
    },
    {
      "epoch": 0.19308,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.01651178995768229,
      "learning_rate": 0.0001,
      "loss": 4.1681,
      "loss/crossentropy": 2.190830111503601,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22319861501455307,
      "step": 9654
    },
    {
      "epoch": 0.19312,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.014048004150390625,
      "learning_rate": 0.0001,
      "loss": 4.4579,
      "loss/crossentropy": 1.9721493124961853,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2011292800307274,
      "step": 9656
    },
    {
      "epoch": 0.19316,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.014277903238932292,
      "learning_rate": 0.0001,
      "loss": 4.5673,
      "loss/crossentropy": 2.1256929636001587,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2728194147348404,
      "step": 9658
    },
    {
      "epoch": 0.1932,
      "grad_norm": 2.125,
      "grad_norm_var": 0.008345286051432291,
      "learning_rate": 0.0001,
      "loss": 4.3206,
      "loss/crossentropy": 2.090156316757202,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2240355908870697,
      "step": 9660
    },
    {
      "epoch": 0.19324,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0108795166015625,
      "learning_rate": 0.0001,
      "loss": 4.0231,
      "loss/crossentropy": 2.2930272817611694,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23188824206590652,
      "step": 9662
    },
    {
      "epoch": 0.19328,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0122222900390625,
      "learning_rate": 0.0001,
      "loss": 4.0075,
      "loss/crossentropy": 1.9754068851470947,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18698863685131073,
      "step": 9664
    },
    {
      "epoch": 0.19332,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0118560791015625,
      "learning_rate": 0.0001,
      "loss": 4.1668,
      "loss/crossentropy": 1.8987788558006287,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22531752288341522,
      "step": 9666
    },
    {
      "epoch": 0.19336,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.011678059895833334,
      "learning_rate": 0.0001,
      "loss": 4.3185,
      "loss/crossentropy": 2.2699583768844604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2572309076786041,
      "step": 9668
    },
    {
      "epoch": 0.1934,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0072509765625,
      "learning_rate": 0.0001,
      "loss": 4.3267,
      "loss/crossentropy": 1.6649349927902222,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23388104140758514,
      "step": 9670
    },
    {
      "epoch": 0.19344,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.007258097330729167,
      "learning_rate": 0.0001,
      "loss": 4.3846,
      "loss/crossentropy": 2.173617362976074,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20848755538463593,
      "step": 9672
    },
    {
      "epoch": 0.19348,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.005890909830729167,
      "learning_rate": 0.0001,
      "loss": 4.3252,
      "loss/crossentropy": 2.2690787315368652,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23493453860282898,
      "step": 9674
    },
    {
      "epoch": 0.19352,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.007079060872395833,
      "learning_rate": 0.0001,
      "loss": 4.3621,
      "loss/crossentropy": 2.00560861825943,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22538188099861145,
      "step": 9676
    },
    {
      "epoch": 0.19356,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.004369862874348958,
      "learning_rate": 0.0001,
      "loss": 4.1264,
      "loss/crossentropy": 1.960120975971222,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21056914329528809,
      "step": 9678
    },
    {
      "epoch": 0.1936,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.004689280192057292,
      "learning_rate": 0.0001,
      "loss": 3.9868,
      "loss/crossentropy": 1.8921862840652466,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22250327467918396,
      "step": 9680
    },
    {
      "epoch": 0.19364,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.004839833577473958,
      "learning_rate": 0.0001,
      "loss": 4.296,
      "loss/crossentropy": 1.9474233984947205,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19719959795475006,
      "step": 9682
    },
    {
      "epoch": 0.19368,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.0072100321451822914,
      "learning_rate": 0.0001,
      "loss": 4.2985,
      "loss/crossentropy": 2.3391844034194946,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22263485193252563,
      "step": 9684
    },
    {
      "epoch": 0.19372,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.008715565999348958,
      "learning_rate": 0.0001,
      "loss": 4.3641,
      "loss/crossentropy": 2.190012037754059,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22531607002019882,
      "step": 9686
    },
    {
      "epoch": 0.19376,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.008283487955729167,
      "learning_rate": 0.0001,
      "loss": 4.2601,
      "loss/crossentropy": 1.9935640096664429,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21438511461019516,
      "step": 9688
    },
    {
      "epoch": 0.1938,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.009186808268229167,
      "learning_rate": 0.0001,
      "loss": 4.1756,
      "loss/crossentropy": 1.7482191324234009,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20972990244627,
      "step": 9690
    },
    {
      "epoch": 0.19384,
      "grad_norm": 2.125,
      "grad_norm_var": 0.008353678385416667,
      "learning_rate": 0.0001,
      "loss": 4.475,
      "loss/crossentropy": 2.2413275241851807,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24103231728076935,
      "step": 9692
    },
    {
      "epoch": 0.19388,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.008519490559895834,
      "learning_rate": 0.0001,
      "loss": 4.2886,
      "loss/crossentropy": 2.2846572399139404,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2420315518975258,
      "step": 9694
    },
    {
      "epoch": 0.19392,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.007968902587890625,
      "learning_rate": 0.0001,
      "loss": 4.1665,
      "loss/crossentropy": 1.7977504134178162,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20145538449287415,
      "step": 9696
    },
    {
      "epoch": 0.19396,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.007535552978515625,
      "learning_rate": 0.0001,
      "loss": 4.205,
      "loss/crossentropy": 2.0343902111053467,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21595563739538193,
      "step": 9698
    },
    {
      "epoch": 0.194,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0052487691243489586,
      "learning_rate": 0.0001,
      "loss": 3.9524,
      "loss/crossentropy": 1.8828233480453491,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19624938070774078,
      "step": 9700
    },
    {
      "epoch": 0.19404,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.005602773030598958,
      "learning_rate": 0.0001,
      "loss": 4.1569,
      "loss/crossentropy": 1.9177573323249817,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19467243552207947,
      "step": 9702
    },
    {
      "epoch": 0.19408,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.006613922119140625,
      "learning_rate": 0.0001,
      "loss": 3.8881,
      "loss/crossentropy": 1.8025588393211365,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18757501989603043,
      "step": 9704
    },
    {
      "epoch": 0.19412,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.006723785400390625,
      "learning_rate": 0.0001,
      "loss": 4.3195,
      "loss/crossentropy": 2.025633454322815,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21623078733682632,
      "step": 9706
    },
    {
      "epoch": 0.19416,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.009129842122395834,
      "learning_rate": 0.0001,
      "loss": 3.8244,
      "loss/crossentropy": 1.9421055316925049,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20563311874866486,
      "step": 9708
    },
    {
      "epoch": 0.1942,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.009191640218098958,
      "learning_rate": 0.0001,
      "loss": 4.0977,
      "loss/crossentropy": 1.9820671081542969,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22029083967208862,
      "step": 9710
    },
    {
      "epoch": 0.19424,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.010341135660807292,
      "learning_rate": 0.0001,
      "loss": 4.2037,
      "loss/crossentropy": 1.9491158723831177,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19559209793806076,
      "step": 9712
    },
    {
      "epoch": 0.19428,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.010416412353515625,
      "learning_rate": 0.0001,
      "loss": 4.1592,
      "loss/crossentropy": 1.8653306365013123,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18629660457372665,
      "step": 9714
    },
    {
      "epoch": 0.19432,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.01950251261393229,
      "learning_rate": 0.0001,
      "loss": 4.2567,
      "loss/crossentropy": 1.6897491812705994,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1948539912700653,
      "step": 9716
    },
    {
      "epoch": 0.19436,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.020776112874348957,
      "learning_rate": 0.0001,
      "loss": 4.4387,
      "loss/crossentropy": 2.07690966129303,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2147115021944046,
      "step": 9718
    },
    {
      "epoch": 0.1944,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.020189412434895835,
      "learning_rate": 0.0001,
      "loss": 4.269,
      "loss/crossentropy": 2.0356597304344177,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20626354217529297,
      "step": 9720
    },
    {
      "epoch": 0.19444,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0204254150390625,
      "learning_rate": 0.0001,
      "loss": 4.0548,
      "loss/crossentropy": 1.7709991931915283,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18573015183210373,
      "step": 9722
    },
    {
      "epoch": 0.19448,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.01830012003580729,
      "learning_rate": 0.0001,
      "loss": 4.117,
      "loss/crossentropy": 2.0255925059318542,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22114800661802292,
      "step": 9724
    },
    {
      "epoch": 0.19452,
      "grad_norm": 2.25,
      "grad_norm_var": 0.01871337890625,
      "learning_rate": 0.0001,
      "loss": 4.4655,
      "loss/crossentropy": 2.0046772956848145,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22649522870779037,
      "step": 9726
    },
    {
      "epoch": 0.19456,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.015827433268229166,
      "learning_rate": 0.0001,
      "loss": 4.179,
      "loss/crossentropy": 2.2746634483337402,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23017344623804092,
      "step": 9728
    },
    {
      "epoch": 0.1946,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.015184529622395833,
      "learning_rate": 0.0001,
      "loss": 4.203,
      "loss/crossentropy": 2.0191025137901306,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20670472085475922,
      "step": 9730
    },
    {
      "epoch": 0.19464,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008177693684895833,
      "learning_rate": 0.0001,
      "loss": 4.1657,
      "loss/crossentropy": 2.3198455572128296,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24985665082931519,
      "step": 9732
    },
    {
      "epoch": 0.19468,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0073150634765625,
      "learning_rate": 0.0001,
      "loss": 4.3886,
      "loss/crossentropy": 2.5229711532592773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22920701652765274,
      "step": 9734
    },
    {
      "epoch": 0.19472,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0072662353515625,
      "learning_rate": 0.0001,
      "loss": 3.9991,
      "loss/crossentropy": 2.081319808959961,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21542686223983765,
      "step": 9736
    },
    {
      "epoch": 0.19476,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.008373006184895834,
      "learning_rate": 0.0001,
      "loss": 4.0698,
      "loss/crossentropy": 2.2170007824897766,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21852095425128937,
      "step": 9738
    },
    {
      "epoch": 0.1948,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.008265940348307292,
      "learning_rate": 0.0001,
      "loss": 4.1528,
      "loss/crossentropy": 1.830683708190918,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1910252571105957,
      "step": 9740
    },
    {
      "epoch": 0.19484,
      "grad_norm": 2.25,
      "grad_norm_var": 0.008420562744140625,
      "learning_rate": 0.0001,
      "loss": 4.3543,
      "loss/crossentropy": 2.1303864121437073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.228670135140419,
      "step": 9742
    },
    {
      "epoch": 0.19488,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.007486724853515625,
      "learning_rate": 0.0001,
      "loss": 4.1504,
      "loss/crossentropy": 2.2621915340423584,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22027206420898438,
      "step": 9744
    },
    {
      "epoch": 0.19492,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0110260009765625,
      "learning_rate": 0.0001,
      "loss": 3.9564,
      "loss/crossentropy": 2.044555902481079,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20469766855239868,
      "step": 9746
    },
    {
      "epoch": 0.19496,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0115142822265625,
      "learning_rate": 0.0001,
      "loss": 4.1677,
      "loss/crossentropy": 1.911176860332489,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22036674618721008,
      "step": 9748
    },
    {
      "epoch": 0.195,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.010545857747395833,
      "learning_rate": 0.0001,
      "loss": 4.0904,
      "loss/crossentropy": 1.8650219440460205,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19878911972045898,
      "step": 9750
    },
    {
      "epoch": 0.19504,
      "grad_norm": 2.125,
      "grad_norm_var": 0.02088623046875,
      "learning_rate": 0.0001,
      "loss": 4.3648,
      "loss/crossentropy": 2.196391463279724,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20845064520835876,
      "step": 9752
    },
    {
      "epoch": 0.19508,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.021201324462890626,
      "learning_rate": 0.0001,
      "loss": 4.0132,
      "loss/crossentropy": 2.222484588623047,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2144618257880211,
      "step": 9754
    },
    {
      "epoch": 0.19512,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.020685831705729168,
      "learning_rate": 0.0001,
      "loss": 4.4078,
      "loss/crossentropy": 2.416514754295349,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2370949387550354,
      "step": 9756
    },
    {
      "epoch": 0.19516,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.020566558837890624,
      "learning_rate": 0.0001,
      "loss": 4.0119,
      "loss/crossentropy": 2.22190260887146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23128122836351395,
      "step": 9758
    },
    {
      "epoch": 0.1952,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.024008941650390626,
      "learning_rate": 0.0001,
      "loss": 4.4253,
      "loss/crossentropy": 2.446492910385132,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23962965607643127,
      "step": 9760
    },
    {
      "epoch": 0.19524,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.019245402018229166,
      "learning_rate": 0.0001,
      "loss": 4.1676,
      "loss/crossentropy": 2.1458136439323425,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19657295942306519,
      "step": 9762
    },
    {
      "epoch": 0.19528,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.0197418212890625,
      "learning_rate": 0.0001,
      "loss": 4.6029,
      "loss/crossentropy": 1.9773340225219727,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22887050360441208,
      "step": 9764
    },
    {
      "epoch": 0.19532,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.020334625244140626,
      "learning_rate": 0.0001,
      "loss": 4.1583,
      "loss/crossentropy": 2.0938061475753784,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19823112338781357,
      "step": 9766
    },
    {
      "epoch": 0.19536,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.015636952718098958,
      "learning_rate": 0.0001,
      "loss": 4.2628,
      "loss/crossentropy": 1.9068174958229065,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19924984872341156,
      "step": 9768
    },
    {
      "epoch": 0.1954,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.012422688802083333,
      "learning_rate": 0.0001,
      "loss": 4.3567,
      "loss/crossentropy": 2.045413613319397,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19175175577402115,
      "step": 9770
    },
    {
      "epoch": 0.19544,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.013106282552083333,
      "learning_rate": 0.0001,
      "loss": 4.1972,
      "loss/crossentropy": 2.0262961983680725,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22137057781219482,
      "step": 9772
    },
    {
      "epoch": 0.19548,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.012992350260416667,
      "learning_rate": 0.0001,
      "loss": 4.0624,
      "loss/crossentropy": 2.146475672721863,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2284562587738037,
      "step": 9774
    },
    {
      "epoch": 0.19552,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011885579427083333,
      "learning_rate": 0.0001,
      "loss": 4.154,
      "loss/crossentropy": 2.0316100120544434,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26275022327899933,
      "step": 9776
    },
    {
      "epoch": 0.19556,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.011359659830729167,
      "learning_rate": 0.0001,
      "loss": 4.5424,
      "loss/crossentropy": 2.1386696100234985,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22695952653884888,
      "step": 9778
    },
    {
      "epoch": 0.1956,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.024405924479166667,
      "learning_rate": 0.0001,
      "loss": 4.2391,
      "loss/crossentropy": 2.3000820875167847,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24045731872320175,
      "step": 9780
    },
    {
      "epoch": 0.19564,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.02415949503580729,
      "learning_rate": 0.0001,
      "loss": 4.7778,
      "loss/crossentropy": 2.1272310614585876,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2318389192223549,
      "step": 9782
    },
    {
      "epoch": 0.19568,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.025655110677083332,
      "learning_rate": 0.0001,
      "loss": 4.1232,
      "loss/crossentropy": 1.8953965306282043,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.213637076318264,
      "step": 9784
    },
    {
      "epoch": 0.19572,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0261871337890625,
      "learning_rate": 0.0001,
      "loss": 4.3459,
      "loss/crossentropy": 2.0738734006881714,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2100028172135353,
      "step": 9786
    },
    {
      "epoch": 0.19576,
      "grad_norm": 2.0,
      "grad_norm_var": 0.027378082275390625,
      "learning_rate": 0.0001,
      "loss": 4.2599,
      "loss/crossentropy": 1.9454593658447266,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18557409197092056,
      "step": 9788
    },
    {
      "epoch": 0.1958,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.026569620768229166,
      "learning_rate": 0.0001,
      "loss": 4.1655,
      "loss/crossentropy": 2.101949095726013,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22479471564292908,
      "step": 9790
    },
    {
      "epoch": 0.19584,
      "grad_norm": 2.25,
      "grad_norm_var": 0.025935872395833334,
      "learning_rate": 0.0001,
      "loss": 4.3841,
      "loss/crossentropy": 1.9524416327476501,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20574256777763367,
      "step": 9792
    },
    {
      "epoch": 0.19588,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.06084391276041667,
      "learning_rate": 0.0001,
      "loss": 4.1596,
      "loss/crossentropy": 1.9490719437599182,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19411193579435349,
      "step": 9794
    },
    {
      "epoch": 0.19592,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.050675455729166666,
      "learning_rate": 0.0001,
      "loss": 4.3438,
      "loss/crossentropy": 2.095793664455414,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21078093349933624,
      "step": 9796
    },
    {
      "epoch": 0.19596,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0508056640625,
      "learning_rate": 0.0001,
      "loss": 4.0828,
      "loss/crossentropy": 2.032066822052002,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21181444078683853,
      "step": 9798
    },
    {
      "epoch": 0.196,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.048378245035807295,
      "learning_rate": 0.0001,
      "loss": 4.0019,
      "loss/crossentropy": 2.1378380060195923,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.224056214094162,
      "step": 9800
    },
    {
      "epoch": 0.19604,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.04784520467122396,
      "learning_rate": 0.0001,
      "loss": 4.2932,
      "loss/crossentropy": 2.120614767074585,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2104548141360283,
      "step": 9802
    },
    {
      "epoch": 0.19608,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.048954010009765625,
      "learning_rate": 0.0001,
      "loss": 4.1975,
      "loss/crossentropy": 1.8837141394615173,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19755827635526657,
      "step": 9804
    },
    {
      "epoch": 0.19612,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.04793675740559896,
      "learning_rate": 0.0001,
      "loss": 4.1645,
      "loss/crossentropy": 1.9325945973396301,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2005266472697258,
      "step": 9806
    },
    {
      "epoch": 0.19616,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.04839045206705729,
      "learning_rate": 0.0001,
      "loss": 4.2211,
      "loss/crossentropy": 1.789370834827423,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20060084015130997,
      "step": 9808
    },
    {
      "epoch": 0.1962,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.01709162394205729,
      "learning_rate": 0.0001,
      "loss": 4.2463,
      "loss/crossentropy": 1.9807876348495483,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21805762499570847,
      "step": 9810
    },
    {
      "epoch": 0.19624,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.018888092041015624,
      "learning_rate": 0.0001,
      "loss": 4.1339,
      "loss/crossentropy": 2.210070848464966,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23082506656646729,
      "step": 9812
    },
    {
      "epoch": 0.19628,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.01693903605143229,
      "learning_rate": 0.0001,
      "loss": 4.1633,
      "loss/crossentropy": 1.8644117712974548,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20300551503896713,
      "step": 9814
    },
    {
      "epoch": 0.19632,
      "grad_norm": 2.125,
      "grad_norm_var": 0.01634496053059896,
      "learning_rate": 0.0001,
      "loss": 4.2064,
      "loss/crossentropy": 1.6804233193397522,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20438820868730545,
      "step": 9816
    },
    {
      "epoch": 0.19636,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.01757990519205729,
      "learning_rate": 0.0001,
      "loss": 4.4339,
      "loss/crossentropy": 2.2513452768325806,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24517202377319336,
      "step": 9818
    },
    {
      "epoch": 0.1964,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.01683527628580729,
      "learning_rate": 0.0001,
      "loss": 4.2238,
      "loss/crossentropy": 2.3133161067962646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22146832942962646,
      "step": 9820
    },
    {
      "epoch": 0.19644,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.016949208577473958,
      "learning_rate": 0.0001,
      "loss": 4.5003,
      "loss/crossentropy": 2.3226611614227295,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2327018678188324,
      "step": 9822
    },
    {
      "epoch": 0.19648,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.016228993733723957,
      "learning_rate": 0.0001,
      "loss": 4.1832,
      "loss/crossentropy": 2.234626054763794,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23657850921154022,
      "step": 9824
    },
    {
      "epoch": 0.19652,
      "grad_norm": 2.25,
      "grad_norm_var": 0.008941396077473959,
      "learning_rate": 0.0001,
      "loss": 4.4576,
      "loss/crossentropy": 2.3478844165802,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24796272069215775,
      "step": 9826
    },
    {
      "epoch": 0.19656,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.006982167561848958,
      "learning_rate": 0.0001,
      "loss": 4.3678,
      "loss/crossentropy": 2.094748795032501,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21118396520614624,
      "step": 9828
    },
    {
      "epoch": 0.1966,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.007289377848307291,
      "learning_rate": 0.0001,
      "loss": 4.2916,
      "loss/crossentropy": 2.166300058364868,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21773284673690796,
      "step": 9830
    },
    {
      "epoch": 0.19664,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.007212066650390625,
      "learning_rate": 0.0001,
      "loss": 4.5253,
      "loss/crossentropy": 2.198317289352417,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22636590898036957,
      "step": 9832
    },
    {
      "epoch": 0.19668,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.006278228759765625,
      "learning_rate": 0.0001,
      "loss": 4.3405,
      "loss/crossentropy": 2.3081597089767456,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.231675922870636,
      "step": 9834
    },
    {
      "epoch": 0.19672,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.005206044514973958,
      "learning_rate": 0.0001,
      "loss": 4.1181,
      "loss/crossentropy": 2.048017203807831,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20089948922395706,
      "step": 9836
    },
    {
      "epoch": 0.19676,
      "grad_norm": 2.125,
      "grad_norm_var": 0.01768773396809896,
      "learning_rate": 0.0001,
      "loss": 4.0515,
      "loss/crossentropy": 1.886117160320282,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21702590584754944,
      "step": 9838
    },
    {
      "epoch": 0.1968,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.017895253499348958,
      "learning_rate": 0.0001,
      "loss": 3.9827,
      "loss/crossentropy": 1.902605414390564,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19976041465997696,
      "step": 9840
    },
    {
      "epoch": 0.19684,
      "grad_norm": 4.28125,
      "grad_norm_var": 0.309179433186849,
      "learning_rate": 0.0001,
      "loss": 4.0874,
      "loss/crossentropy": 1.7959995865821838,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2023170217871666,
      "step": 9842
    },
    {
      "epoch": 0.19688,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.3066993713378906,
      "learning_rate": 0.0001,
      "loss": 4.5535,
      "loss/crossentropy": 2.1481738090515137,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22133169323205948,
      "step": 9844
    },
    {
      "epoch": 0.19692,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.30752741495768227,
      "learning_rate": 0.0001,
      "loss": 4.0269,
      "loss/crossentropy": 1.9328197240829468,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20346572250127792,
      "step": 9846
    },
    {
      "epoch": 0.19696,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.31202367146809895,
      "learning_rate": 0.0001,
      "loss": 4.0299,
      "loss/crossentropy": 2.096457004547119,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21341010928153992,
      "step": 9848
    },
    {
      "epoch": 0.197,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.3146522521972656,
      "learning_rate": 0.0001,
      "loss": 4.3105,
      "loss/crossentropy": 1.9698969721794128,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1855439990758896,
      "step": 9850
    },
    {
      "epoch": 0.19704,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.31038004557291665,
      "learning_rate": 0.0001,
      "loss": 4.2165,
      "loss/crossentropy": 1.987346351146698,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23657751083374023,
      "step": 9852
    },
    {
      "epoch": 0.19708,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.3109169006347656,
      "learning_rate": 0.0001,
      "loss": 4.0346,
      "loss/crossentropy": 1.9535572528839111,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21131790429353714,
      "step": 9854
    },
    {
      "epoch": 0.19712,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.3090349833170573,
      "learning_rate": 0.0001,
      "loss": 4.5828,
      "loss/crossentropy": 2.08541601896286,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2290833741426468,
      "step": 9856
    },
    {
      "epoch": 0.19716,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.016155751546223958,
      "learning_rate": 0.0001,
      "loss": 4.2774,
      "loss/crossentropy": 2.1930192708969116,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23072391748428345,
      "step": 9858
    },
    {
      "epoch": 0.1972,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.016658274332682292,
      "learning_rate": 0.0001,
      "loss": 4.3923,
      "loss/crossentropy": 1.8369358777999878,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20687467604875565,
      "step": 9860
    },
    {
      "epoch": 0.19724,
      "grad_norm": 2.0,
      "grad_norm_var": 0.017380523681640624,
      "learning_rate": 0.0001,
      "loss": 4.0287,
      "loss/crossentropy": 1.6707186102867126,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1808091625571251,
      "step": 9862
    },
    {
      "epoch": 0.19728,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.017651112874348958,
      "learning_rate": 0.0001,
      "loss": 4.1184,
      "loss/crossentropy": 1.8352991342544556,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19169726222753525,
      "step": 9864
    },
    {
      "epoch": 0.19732,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.042909495035807294,
      "learning_rate": 0.0001,
      "loss": 4.2857,
      "loss/crossentropy": 1.9427489638328552,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20076656341552734,
      "step": 9866
    },
    {
      "epoch": 0.19736,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.04237035115559896,
      "learning_rate": 0.0001,
      "loss": 4.3725,
      "loss/crossentropy": 2.2190250158309937,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24385111033916473,
      "step": 9868
    },
    {
      "epoch": 0.1974,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.04182510375976563,
      "learning_rate": 0.0001,
      "loss": 4.2192,
      "loss/crossentropy": 2.0958147644996643,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23264098167419434,
      "step": 9870
    },
    {
      "epoch": 0.19744,
      "grad_norm": 2.25,
      "grad_norm_var": 0.04228897094726562,
      "learning_rate": 0.0001,
      "loss": 4.3571,
      "loss/crossentropy": 2.542472720146179,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23458892852067947,
      "step": 9872
    },
    {
      "epoch": 0.19748,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.03728408813476562,
      "learning_rate": 0.0001,
      "loss": 4.2767,
      "loss/crossentropy": 2.1562893390655518,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2243279591202736,
      "step": 9874
    },
    {
      "epoch": 0.19752,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.034395090738932294,
      "learning_rate": 0.0001,
      "loss": 4.0114,
      "loss/crossentropy": 1.9851951599121094,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2369420975446701,
      "step": 9876
    },
    {
      "epoch": 0.19756,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.032714589436848955,
      "learning_rate": 0.0001,
      "loss": 4.3017,
      "loss/crossentropy": 1.8751549124717712,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20777788758277893,
      "step": 9878
    },
    {
      "epoch": 0.1976,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0318267822265625,
      "learning_rate": 0.0001,
      "loss": 4.1328,
      "loss/crossentropy": 2.010055720806122,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21204733848571777,
      "step": 9880
    },
    {
      "epoch": 0.19764,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.008942667643229167,
      "learning_rate": 0.0001,
      "loss": 4.1122,
      "loss/crossentropy": 2.04589307308197,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21330490708351135,
      "step": 9882
    },
    {
      "epoch": 0.19768,
      "grad_norm": 2.125,
      "grad_norm_var": 0.010138956705729167,
      "learning_rate": 0.0001,
      "loss": 4.1663,
      "loss/crossentropy": 1.9441133737564087,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22589778900146484,
      "step": 9884
    },
    {
      "epoch": 0.19772,
      "grad_norm": 2.25,
      "grad_norm_var": 0.008296457926432292,
      "learning_rate": 0.0001,
      "loss": 4.4239,
      "loss/crossentropy": 2.305395483970642,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24267761409282684,
      "step": 9886
    },
    {
      "epoch": 0.19776,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.007269032796223958,
      "learning_rate": 0.0001,
      "loss": 4.386,
      "loss/crossentropy": 2.096014082431793,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21358592063188553,
      "step": 9888
    },
    {
      "epoch": 0.1978,
      "grad_norm": 2.25,
      "grad_norm_var": 0.008149973551432292,
      "learning_rate": 0.0001,
      "loss": 4.2315,
      "loss/crossentropy": 2.1115033626556396,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22994756698608398,
      "step": 9890
    },
    {
      "epoch": 0.19784,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.008654530843098958,
      "learning_rate": 0.0001,
      "loss": 4.1419,
      "loss/crossentropy": 1.6122692227363586,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1897209882736206,
      "step": 9892
    },
    {
      "epoch": 0.19788,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.012933095296223959,
      "learning_rate": 0.0001,
      "loss": 4.1679,
      "loss/crossentropy": 1.4790136218070984,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17421242594718933,
      "step": 9894
    },
    {
      "epoch": 0.19792,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.013700358072916667,
      "learning_rate": 0.0001,
      "loss": 4.2921,
      "loss/crossentropy": 1.8394885063171387,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20840360969305038,
      "step": 9896
    },
    {
      "epoch": 0.19796,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0135894775390625,
      "learning_rate": 0.0001,
      "loss": 4.3993,
      "loss/crossentropy": 2.152444541454315,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23834071308374405,
      "step": 9898
    },
    {
      "epoch": 0.198,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.012189737955729167,
      "learning_rate": 0.0001,
      "loss": 4.5268,
      "loss/crossentropy": 2.367082357406616,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23409543931484222,
      "step": 9900
    },
    {
      "epoch": 0.19804,
      "grad_norm": 2.125,
      "grad_norm_var": 0.011799112955729166,
      "learning_rate": 0.0001,
      "loss": 4.4599,
      "loss/crossentropy": 2.136048436164856,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21797242760658264,
      "step": 9902
    },
    {
      "epoch": 0.19808,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.011945597330729167,
      "learning_rate": 0.0001,
      "loss": 4.2262,
      "loss/crossentropy": 2.035883128643036,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.215388223528862,
      "step": 9904
    },
    {
      "epoch": 0.19812,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.011970774332682291,
      "learning_rate": 0.0001,
      "loss": 4.2024,
      "loss/crossentropy": 2.0339369773864746,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2085876688361168,
      "step": 9906
    },
    {
      "epoch": 0.19816,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009655507405598958,
      "learning_rate": 0.0001,
      "loss": 4.4441,
      "loss/crossentropy": 1.829429566860199,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20085029304027557,
      "step": 9908
    },
    {
      "epoch": 0.1982,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.008861287434895834,
      "learning_rate": 0.0001,
      "loss": 4.1108,
      "loss/crossentropy": 2.1721774339675903,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2118668630719185,
      "step": 9910
    },
    {
      "epoch": 0.19824,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.008861287434895834,
      "learning_rate": 0.0001,
      "loss": 4.3511,
      "loss/crossentropy": 1.958261251449585,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22304313629865646,
      "step": 9912
    },
    {
      "epoch": 0.19828,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.009870402018229167,
      "learning_rate": 0.0001,
      "loss": 4.5819,
      "loss/crossentropy": 2.4651763439178467,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2435118407011032,
      "step": 9914
    },
    {
      "epoch": 0.19832,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.012532297770182292,
      "learning_rate": 0.0001,
      "loss": 4.0973,
      "loss/crossentropy": 2.173910617828369,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21142201125621796,
      "step": 9916
    },
    {
      "epoch": 0.19836,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.013239542643229166,
      "learning_rate": 0.0001,
      "loss": 3.9283,
      "loss/crossentropy": 1.9559763073921204,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19051063805818558,
      "step": 9918
    },
    {
      "epoch": 0.1984,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.015895334879557292,
      "learning_rate": 0.0001,
      "loss": 4.2516,
      "loss/crossentropy": 2.238003969192505,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2338731735944748,
      "step": 9920
    },
    {
      "epoch": 0.19844,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.015233357747395834,
      "learning_rate": 0.0001,
      "loss": 4.4624,
      "loss/crossentropy": 2.269726276397705,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19430368393659592,
      "step": 9922
    },
    {
      "epoch": 0.19848,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.015697224934895834,
      "learning_rate": 0.0001,
      "loss": 4.436,
      "loss/crossentropy": 2.351833701133728,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24469508230686188,
      "step": 9924
    },
    {
      "epoch": 0.19852,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.013398996988932292,
      "learning_rate": 0.0001,
      "loss": 4.3833,
      "loss/crossentropy": 2.0600146055221558,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21375566720962524,
      "step": 9926
    },
    {
      "epoch": 0.19856,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.012562815348307292,
      "learning_rate": 0.0001,
      "loss": 4.2957,
      "loss/crossentropy": 2.1543694734573364,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20916736871004105,
      "step": 9928
    },
    {
      "epoch": 0.1986,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010782623291015625,
      "learning_rate": 0.0001,
      "loss": 4.3111,
      "loss/crossentropy": 1.8952317833900452,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2216978445649147,
      "step": 9930
    },
    {
      "epoch": 0.19864,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008695475260416667,
      "learning_rate": 0.0001,
      "loss": 4.0542,
      "loss/crossentropy": 2.102243661880493,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22314947098493576,
      "step": 9932
    },
    {
      "epoch": 0.19868,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0072021484375,
      "learning_rate": 0.0001,
      "loss": 4.1653,
      "loss/crossentropy": 1.9036884307861328,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19729873538017273,
      "step": 9934
    },
    {
      "epoch": 0.19872,
      "grad_norm": 2.0,
      "grad_norm_var": 0.006459299723307292,
      "learning_rate": 0.0001,
      "loss": 4.17,
      "loss/crossentropy": 1.823796033859253,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18089265376329422,
      "step": 9936
    },
    {
      "epoch": 0.19876,
      "grad_norm": 2.0,
      "grad_norm_var": 0.006302642822265625,
      "learning_rate": 0.0001,
      "loss": 3.8742,
      "loss/crossentropy": 2.055707633495331,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20620816200971603,
      "step": 9938
    },
    {
      "epoch": 0.1988,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0055010477701822914,
      "learning_rate": 0.0001,
      "loss": 4.256,
      "loss/crossentropy": 2.0490049719810486,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21709006279706955,
      "step": 9940
    },
    {
      "epoch": 0.19884,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.005891672770182292,
      "learning_rate": 0.0001,
      "loss": 4.2733,
      "loss/crossentropy": 2.164198637008667,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2335330694913864,
      "step": 9942
    },
    {
      "epoch": 0.19888,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.006941731770833333,
      "learning_rate": 0.0001,
      "loss": 4.1463,
      "loss/crossentropy": 1.9218478202819824,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1945241093635559,
      "step": 9944
    },
    {
      "epoch": 0.19892,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.007328033447265625,
      "learning_rate": 0.0001,
      "loss": 3.8591,
      "loss/crossentropy": 1.9456552267074585,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20599794387817383,
      "step": 9946
    },
    {
      "epoch": 0.19896,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.007749176025390625,
      "learning_rate": 0.0001,
      "loss": 4.0385,
      "loss/crossentropy": 2.0472273230552673,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20920252054929733,
      "step": 9948
    },
    {
      "epoch": 0.199,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.007860310872395833,
      "learning_rate": 0.0001,
      "loss": 4.308,
      "loss/crossentropy": 2.2252047061920166,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2360471710562706,
      "step": 9950
    },
    {
      "epoch": 0.19904,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.005537923177083333,
      "learning_rate": 0.0001,
      "loss": 4.0828,
      "loss/crossentropy": 1.646431565284729,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1692601889371872,
      "step": 9952
    },
    {
      "epoch": 0.19908,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0054443359375,
      "learning_rate": 0.0001,
      "loss": 3.8818,
      "loss/crossentropy": 2.0114784836769104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20070793479681015,
      "step": 9954
    },
    {
      "epoch": 0.19912,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.00677490234375,
      "learning_rate": 0.0001,
      "loss": 4.324,
      "loss/crossentropy": 2.001778781414032,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21009314060211182,
      "step": 9956
    },
    {
      "epoch": 0.19916,
      "grad_norm": 2.0,
      "grad_norm_var": 0.006037394205729167,
      "learning_rate": 0.0001,
      "loss": 4.3493,
      "loss/crossentropy": 2.1330565214157104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22442802786827087,
      "step": 9958
    },
    {
      "epoch": 0.1992,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.005635579427083333,
      "learning_rate": 0.0001,
      "loss": 4.193,
      "loss/crossentropy": 1.9146793484687805,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21145610511302948,
      "step": 9960
    },
    {
      "epoch": 0.19924,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.004705556233723958,
      "learning_rate": 0.0001,
      "loss": 4.3262,
      "loss/crossentropy": 2.5224483013153076,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23354032635688782,
      "step": 9962
    },
    {
      "epoch": 0.19928,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.004552968343098958,
      "learning_rate": 0.0001,
      "loss": 4.3663,
      "loss/crossentropy": 2.245160937309265,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21848157793283463,
      "step": 9964
    },
    {
      "epoch": 0.19932,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.004622141520182292,
      "learning_rate": 0.0001,
      "loss": 4.3506,
      "loss/crossentropy": 2.122299015522003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21853189170360565,
      "step": 9966
    },
    {
      "epoch": 0.19936,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.005716705322265625,
      "learning_rate": 0.0001,
      "loss": 4.4524,
      "loss/crossentropy": 2.31084668636322,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23691194504499435,
      "step": 9968
    },
    {
      "epoch": 0.1994,
      "grad_norm": 2.25,
      "grad_norm_var": 0.007834625244140626,
      "learning_rate": 0.0001,
      "loss": 4.274,
      "loss/crossentropy": 2.2242307662963867,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21957046538591385,
      "step": 9970
    },
    {
      "epoch": 0.19944,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.007063547770182292,
      "learning_rate": 0.0001,
      "loss": 4.2264,
      "loss/crossentropy": 1.792852759361267,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19996580481529236,
      "step": 9972
    },
    {
      "epoch": 0.19948,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0063250223795572914,
      "learning_rate": 0.0001,
      "loss": 4.3029,
      "loss/crossentropy": 1.9593411087989807,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21858739852905273,
      "step": 9974
    },
    {
      "epoch": 0.19952,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0067827860514322914,
      "learning_rate": 0.0001,
      "loss": 4.3067,
      "loss/crossentropy": 2.259618401527405,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22094043344259262,
      "step": 9976
    },
    {
      "epoch": 0.19956,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0063168843587239586,
      "learning_rate": 0.0001,
      "loss": 4.373,
      "loss/crossentropy": 2.31876802444458,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22025877982378006,
      "step": 9978
    },
    {
      "epoch": 0.1996,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.006109364827473958,
      "learning_rate": 0.0001,
      "loss": 4.15,
      "loss/crossentropy": 1.7625555396080017,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21313250809907913,
      "step": 9980
    },
    {
      "epoch": 0.19964,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.007342274983723958,
      "learning_rate": 0.0001,
      "loss": 4.4628,
      "loss/crossentropy": 2.19295072555542,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23694515973329544,
      "step": 9982
    },
    {
      "epoch": 0.19968,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.008739217122395834,
      "learning_rate": 0.0001,
      "loss": 4.187,
      "loss/crossentropy": 2.4073877334594727,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22179137915372849,
      "step": 9984
    },
    {
      "epoch": 0.19972,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008125813802083333,
      "learning_rate": 0.0001,
      "loss": 3.9804,
      "loss/crossentropy": 1.8942558765411377,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20588286221027374,
      "step": 9986
    },
    {
      "epoch": 0.19976,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.012188466389973958,
      "learning_rate": 0.0001,
      "loss": 4.109,
      "loss/crossentropy": 2.220746397972107,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2144280970096588,
      "step": 9988
    },
    {
      "epoch": 0.1998,
      "grad_norm": 2.125,
      "grad_norm_var": 0.01260986328125,
      "learning_rate": 0.0001,
      "loss": 4.2279,
      "loss/crossentropy": 1.9511706233024597,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1998429372906685,
      "step": 9990
    },
    {
      "epoch": 0.19984,
      "grad_norm": 2.125,
      "grad_norm_var": 0.012247467041015625,
      "learning_rate": 0.0001,
      "loss": 4.143,
      "loss/crossentropy": 2.249726891517639,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24011528491973877,
      "step": 9992
    },
    {
      "epoch": 0.19988,
      "grad_norm": 2.6875,
      "grad_norm_var": 0.03792292277018229,
      "learning_rate": 0.0001,
      "loss": 4.3104,
      "loss/crossentropy": 1.957375943660736,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24464774131774902,
      "step": 9994
    },
    {
      "epoch": 0.19992,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.039033762613932294,
      "learning_rate": 0.0001,
      "loss": 4.5492,
      "loss/crossentropy": 2.265984058380127,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24291887879371643,
      "step": 9996
    },
    {
      "epoch": 0.19996,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.03920873006184896,
      "learning_rate": 0.0001,
      "loss": 4.1432,
      "loss/crossentropy": 1.8064668774604797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21571090072393417,
      "step": 9998
    },
    {
      "epoch": 0.2,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.03875732421875,
      "learning_rate": 0.0001,
      "loss": 4.2295,
      "loss/crossentropy": 1.9072380661964417,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19980185478925705,
      "step": 10000
    },
    {
      "epoch": 0.20004,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.036641438802083336,
      "learning_rate": 0.0001,
      "loss": 4.3919,
      "loss/crossentropy": 2.074933707714081,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22161328792572021,
      "step": 10002
    },
    {
      "epoch": 0.20008,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.027581532796223957,
      "learning_rate": 0.0001,
      "loss": 4.0447,
      "loss/crossentropy": 1.9344687461853027,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1825423538684845,
      "step": 10004
    },
    {
      "epoch": 0.20012,
      "grad_norm": 3.640625,
      "grad_norm_var": 0.162158203125,
      "learning_rate": 0.0001,
      "loss": 4.1116,
      "loss/crossentropy": 1.866003930568695,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22012518346309662,
      "step": 10006
    },
    {
      "epoch": 0.20016,
      "grad_norm": 2.0,
      "grad_norm_var": 0.162158203125,
      "learning_rate": 0.0001,
      "loss": 4.3517,
      "loss/crossentropy": 2.145058751106262,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22737383097410202,
      "step": 10008
    },
    {
      "epoch": 0.2002,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.15038960774739582,
      "learning_rate": 0.0001,
      "loss": 3.9755,
      "loss/crossentropy": 1.7828176617622375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22082456946372986,
      "step": 10010
    },
    {
      "epoch": 0.20024,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.1537994384765625,
      "learning_rate": 0.0001,
      "loss": 4.3143,
      "loss/crossentropy": 2.1222537755966187,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2176828756928444,
      "step": 10012
    },
    {
      "epoch": 0.20028,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.15230712890625,
      "learning_rate": 0.0001,
      "loss": 4.53,
      "loss/crossentropy": 2.119267463684082,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24002012610435486,
      "step": 10014
    },
    {
      "epoch": 0.20032,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.15458577473958332,
      "learning_rate": 0.0001,
      "loss": 4.3528,
      "loss/crossentropy": 2.198129415512085,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23767977952957153,
      "step": 10016
    },
    {
      "epoch": 0.20036,
      "grad_norm": 2.125,
      "grad_norm_var": 0.15608317057291668,
      "learning_rate": 0.0001,
      "loss": 4.0228,
      "loss/crossentropy": 1.7466872334480286,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19916100800037384,
      "step": 10018
    },
    {
      "epoch": 0.2004,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.15812352498372395,
      "learning_rate": 0.0001,
      "loss": 4.1989,
      "loss/crossentropy": 1.9834936261177063,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1891954466700554,
      "step": 10020
    },
    {
      "epoch": 0.20044,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.005840810139973959,
      "learning_rate": 0.0001,
      "loss": 4.0987,
      "loss/crossentropy": 1.8997412323951721,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19624030590057373,
      "step": 10022
    },
    {
      "epoch": 0.20048,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0065915425618489586,
      "learning_rate": 0.0001,
      "loss": 4.2531,
      "loss/crossentropy": 2.0051563382148743,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2069488987326622,
      "step": 10024
    },
    {
      "epoch": 0.20052,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.008107248942057292,
      "learning_rate": 0.0001,
      "loss": 4.4215,
      "loss/crossentropy": 1.8662462830543518,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2053019255399704,
      "step": 10026
    },
    {
      "epoch": 0.20056,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.02585627237955729,
      "learning_rate": 0.0001,
      "loss": 4.1444,
      "loss/crossentropy": 2.1043936014175415,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22167576104402542,
      "step": 10028
    },
    {
      "epoch": 0.2006,
      "grad_norm": 5.78125,
      "grad_norm_var": 0.859185536702474,
      "learning_rate": 0.0001,
      "loss": 4.3685,
      "loss/crossentropy": 2.1248152256011963,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22400003671646118,
      "step": 10030
    },
    {
      "epoch": 0.20064,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.852441151936849,
      "learning_rate": 0.0001,
      "loss": 4.1971,
      "loss/crossentropy": 2.326894521713257,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24597454071044922,
      "step": 10032
    },
    {
      "epoch": 0.20068,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.8413164774576823,
      "learning_rate": 0.0001,
      "loss": 4.1801,
      "loss/crossentropy": 1.8590435981750488,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20270948112010956,
      "step": 10034
    },
    {
      "epoch": 0.20072,
      "grad_norm": 2.0,
      "grad_norm_var": 0.82972412109375,
      "learning_rate": 0.0001,
      "loss": 4.5098,
      "loss/crossentropy": 2.0383604168891907,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21504472196102142,
      "step": 10036
    },
    {
      "epoch": 0.20076,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.81461181640625,
      "learning_rate": 0.0001,
      "loss": 4.599,
      "loss/crossentropy": 2.42622447013855,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2143569439649582,
      "step": 10038
    },
    {
      "epoch": 0.2008,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.825640614827474,
      "learning_rate": 0.0001,
      "loss": 4.3216,
      "loss/crossentropy": 2.0825703144073486,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21215181052684784,
      "step": 10040
    },
    {
      "epoch": 0.20084,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.8326515197753906,
      "learning_rate": 0.0001,
      "loss": 3.9615,
      "loss/crossentropy": 2.1682112216949463,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23538567870855331,
      "step": 10042
    },
    {
      "epoch": 0.20088,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.8466957092285157,
      "learning_rate": 0.0001,
      "loss": 4.2244,
      "loss/crossentropy": 2.0161439180374146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21078374981880188,
      "step": 10044
    },
    {
      "epoch": 0.20092,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.018143463134765624,
      "learning_rate": 0.0001,
      "loss": 4.5491,
      "loss/crossentropy": 2.2262184619903564,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2855340391397476,
      "step": 10046
    },
    {
      "epoch": 0.20096,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.011321767171223959,
      "learning_rate": 0.0001,
      "loss": 4.5078,
      "loss/crossentropy": 2.454360246658325,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2287754938006401,
      "step": 10048
    },
    {
      "epoch": 0.201,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.007940419514973958,
      "learning_rate": 0.0001,
      "loss": 4.1146,
      "loss/crossentropy": 1.970844566822052,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20197579264640808,
      "step": 10050
    },
    {
      "epoch": 0.20104,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.006211090087890625,
      "learning_rate": 0.0001,
      "loss": 4.3573,
      "loss/crossentropy": 2.0942559242248535,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21927295625209808,
      "step": 10052
    },
    {
      "epoch": 0.20108,
      "grad_norm": 2.0,
      "grad_norm_var": 0.005771636962890625,
      "learning_rate": 0.0001,
      "loss": 4.036,
      "loss/crossentropy": 1.868508517742157,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1838754191994667,
      "step": 10054
    },
    {
      "epoch": 0.20112,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.0058095296223958336,
      "learning_rate": 0.0001,
      "loss": 4.2101,
      "loss/crossentropy": 2.028561532497406,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21768346428871155,
      "step": 10056
    },
    {
      "epoch": 0.20116,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.006669108072916667,
      "learning_rate": 0.0001,
      "loss": 3.9803,
      "loss/crossentropy": 2.3005030155181885,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23332027345895767,
      "step": 10058
    },
    {
      "epoch": 0.2012,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0065826416015625,
      "learning_rate": 0.0001,
      "loss": 4.089,
      "loss/crossentropy": 1.7793474793434143,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18733646720647812,
      "step": 10060
    },
    {
      "epoch": 0.20124,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.005125935872395833,
      "learning_rate": 0.0001,
      "loss": 4.0197,
      "loss/crossentropy": 2.0612844228744507,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21674886345863342,
      "step": 10062
    },
    {
      "epoch": 0.20128,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.004500071207682292,
      "learning_rate": 0.0001,
      "loss": 4.237,
      "loss/crossentropy": 2.3219568729400635,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20989079773426056,
      "step": 10064
    },
    {
      "epoch": 0.20132,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.004659016927083333,
      "learning_rate": 0.0001,
      "loss": 4.0959,
      "loss/crossentropy": 2.0941001176834106,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19992788136005402,
      "step": 10066
    },
    {
      "epoch": 0.20136,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0038330078125,
      "learning_rate": 0.0001,
      "loss": 4.053,
      "loss/crossentropy": 1.7448341250419617,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2108384072780609,
      "step": 10068
    },
    {
      "epoch": 0.2014,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0059397379557291664,
      "learning_rate": 0.0001,
      "loss": 4.263,
      "loss/crossentropy": 2.0475903749465942,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19875742495059967,
      "step": 10070
    },
    {
      "epoch": 0.20144,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0499755859375,
      "learning_rate": 0.0001,
      "loss": 4.1014,
      "loss/crossentropy": 2.0624433755874634,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20577051490545273,
      "step": 10072
    },
    {
      "epoch": 0.20148,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.04807840983072917,
      "learning_rate": 0.0001,
      "loss": 4.3612,
      "loss/crossentropy": 2.194978952407837,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19964048266410828,
      "step": 10074
    },
    {
      "epoch": 0.20152,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.04691162109375,
      "learning_rate": 0.0001,
      "loss": 4.4025,
      "loss/crossentropy": 2.011984169483185,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22259068489074707,
      "step": 10076
    },
    {
      "epoch": 0.20156,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.0458740234375,
      "learning_rate": 0.0001,
      "loss": 4.3539,
      "loss/crossentropy": 2.098285675048828,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2232050970196724,
      "step": 10078
    },
    {
      "epoch": 0.2016,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.04423726399739583,
      "learning_rate": 0.0001,
      "loss": 4.2772,
      "loss/crossentropy": 2.313677191734314,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2224883735179901,
      "step": 10080
    },
    {
      "epoch": 0.20164,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.04454523722330729,
      "learning_rate": 0.0001,
      "loss": 4.1523,
      "loss/crossentropy": 2.3712470531463623,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23282987624406815,
      "step": 10082
    },
    {
      "epoch": 0.20168,
      "grad_norm": 2.125,
      "grad_norm_var": 0.043342844645182295,
      "learning_rate": 0.0001,
      "loss": 4.3173,
      "loss/crossentropy": 2.204525947570801,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2384341061115265,
      "step": 10084
    },
    {
      "epoch": 0.20172,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.04528401692708333,
      "learning_rate": 0.0001,
      "loss": 3.9497,
      "loss/crossentropy": 1.8102558851242065,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17472535371780396,
      "step": 10086
    },
    {
      "epoch": 0.20176,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.005711873372395833,
      "learning_rate": 0.0001,
      "loss": 4.2119,
      "loss/crossentropy": 2.041890263557434,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21201669424772263,
      "step": 10088
    },
    {
      "epoch": 0.2018,
      "grad_norm": 2.125,
      "grad_norm_var": 0.006026204427083333,
      "learning_rate": 0.0001,
      "loss": 4.3869,
      "loss/crossentropy": 1.9904854893684387,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23453570157289505,
      "step": 10090
    },
    {
      "epoch": 0.20184,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.00716552734375,
      "learning_rate": 0.0001,
      "loss": 4.4324,
      "loss/crossentropy": 2.199320912361145,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22087062150239944,
      "step": 10092
    },
    {
      "epoch": 0.20188,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.0073150634765625,
      "learning_rate": 0.0001,
      "loss": 4.392,
      "loss/crossentropy": 2.321953535079956,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2463374137878418,
      "step": 10094
    },
    {
      "epoch": 0.20192,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006076812744140625,
      "learning_rate": 0.0001,
      "loss": 4.3276,
      "loss/crossentropy": 2.1109840869903564,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23713566362857819,
      "step": 10096
    },
    {
      "epoch": 0.20196,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.006322987874348958,
      "learning_rate": 0.0001,
      "loss": 3.9339,
      "loss/crossentropy": 1.9126858711242676,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20392102003097534,
      "step": 10098
    },
    {
      "epoch": 0.202,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.006268056233723959,
      "learning_rate": 0.0001,
      "loss": 4.2902,
      "loss/crossentropy": 2.073318660259247,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20465004444122314,
      "step": 10100
    },
    {
      "epoch": 0.20204,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.009837849934895834,
      "learning_rate": 0.0001,
      "loss": 4.1883,
      "loss/crossentropy": 1.7532709836959839,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22591036558151245,
      "step": 10102
    },
    {
      "epoch": 0.20208,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.010660807291666666,
      "learning_rate": 0.0001,
      "loss": 4.3088,
      "loss/crossentropy": 2.1874141693115234,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.227812297642231,
      "step": 10104
    },
    {
      "epoch": 0.20212,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.014095052083333334,
      "learning_rate": 0.0001,
      "loss": 4.4738,
      "loss/crossentropy": 2.522923469543457,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24104881286621094,
      "step": 10106
    },
    {
      "epoch": 0.20216,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.013963826497395833,
      "learning_rate": 0.0001,
      "loss": 4.2823,
      "loss/crossentropy": 1.9359918236732483,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19563085585832596,
      "step": 10108
    },
    {
      "epoch": 0.2022,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.015315755208333334,
      "learning_rate": 0.0001,
      "loss": 4.3582,
      "loss/crossentropy": 2.390757203102112,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24748887866735458,
      "step": 10110
    },
    {
      "epoch": 0.20224,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.015013631184895833,
      "learning_rate": 0.0001,
      "loss": 4.5576,
      "loss/crossentropy": 2.419153571128845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2228071466088295,
      "step": 10112
    },
    {
      "epoch": 0.20228,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.015404256184895833,
      "learning_rate": 0.0001,
      "loss": 4.0235,
      "loss/crossentropy": 1.7460771799087524,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19431117177009583,
      "step": 10114
    },
    {
      "epoch": 0.20232,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.01754150390625,
      "learning_rate": 0.0001,
      "loss": 4.601,
      "loss/crossentropy": 2.308094024658203,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24104833602905273,
      "step": 10116
    },
    {
      "epoch": 0.20236,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.013374837239583333,
      "learning_rate": 0.0001,
      "loss": 4.1696,
      "loss/crossentropy": 2.0273314118385315,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2325449138879776,
      "step": 10118
    },
    {
      "epoch": 0.2024,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.013792928059895833,
      "learning_rate": 0.0001,
      "loss": 4.016,
      "loss/crossentropy": 2.1261476278305054,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21481933444738388,
      "step": 10120
    },
    {
      "epoch": 0.20244,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0107818603515625,
      "learning_rate": 0.0001,
      "loss": 4.478,
      "loss/crossentropy": 2.5006214380264282,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25569023191928864,
      "step": 10122
    },
    {
      "epoch": 0.20248,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.011775461832682292,
      "learning_rate": 0.0001,
      "loss": 4.3208,
      "loss/crossentropy": 2.2033116817474365,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2601129561662674,
      "step": 10124
    },
    {
      "epoch": 0.20252,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.011572011311848958,
      "learning_rate": 0.0001,
      "loss": 4.1098,
      "loss/crossentropy": 2.2208765745162964,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20958464592695236,
      "step": 10126
    },
    {
      "epoch": 0.20256,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.011848958333333333,
      "learning_rate": 0.0001,
      "loss": 3.943,
      "loss/crossentropy": 1.7752392888069153,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22697371244430542,
      "step": 10128
    },
    {
      "epoch": 0.2026,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.011937459309895834,
      "learning_rate": 0.0001,
      "loss": 4.0529,
      "loss/crossentropy": 1.6444379687309265,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18721824884414673,
      "step": 10130
    },
    {
      "epoch": 0.20264,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.007271321614583334,
      "learning_rate": 0.0001,
      "loss": 4.1796,
      "loss/crossentropy": 1.7681297659873962,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24296899884939194,
      "step": 10132
    },
    {
      "epoch": 0.20268,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.007614898681640625,
      "learning_rate": 0.0001,
      "loss": 4.1593,
      "loss/crossentropy": 1.8710024952888489,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1891346201300621,
      "step": 10134
    },
    {
      "epoch": 0.20272,
      "grad_norm": 2.125,
      "grad_norm_var": 0.007515462239583334,
      "learning_rate": 0.0001,
      "loss": 4.0822,
      "loss/crossentropy": 2.0258530974388123,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22128118574619293,
      "step": 10136
    },
    {
      "epoch": 0.20276,
      "grad_norm": 2.015625,
      "grad_norm_var": 1.7479237874348958,
      "learning_rate": 0.0001,
      "loss": 4.3091,
      "loss/crossentropy": 2.3366141319274902,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2612725794315338,
      "step": 10138
    },
    {
      "epoch": 0.2028,
      "grad_norm": 2.0625,
      "grad_norm_var": 1.739208730061849,
      "learning_rate": 0.0001,
      "loss": 4.357,
      "loss/crossentropy": 2.152353823184967,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2239176332950592,
      "step": 10140
    },
    {
      "epoch": 0.20284,
      "grad_norm": 2.15625,
      "grad_norm_var": 1.7346433003743489,
      "learning_rate": 0.0001,
      "loss": 4.2552,
      "loss/crossentropy": 1.8189843893051147,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21529949456453323,
      "step": 10142
    },
    {
      "epoch": 0.20288,
      "grad_norm": 2.03125,
      "grad_norm_var": 1.7219970703125,
      "learning_rate": 0.0001,
      "loss": 4.3179,
      "loss/crossentropy": 2.308253049850464,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21748851984739304,
      "step": 10144
    },
    {
      "epoch": 0.20292,
      "grad_norm": 2.015625,
      "grad_norm_var": 1.7289377848307292,
      "learning_rate": 0.0001,
      "loss": 4.1762,
      "loss/crossentropy": 1.953293800354004,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1948333978652954,
      "step": 10146
    },
    {
      "epoch": 0.20296,
      "grad_norm": 1.953125,
      "grad_norm_var": 1.726512654622396,
      "learning_rate": 0.0001,
      "loss": 3.9573,
      "loss/crossentropy": 2.1122325658798218,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2085581198334694,
      "step": 10148
    },
    {
      "epoch": 0.203,
      "grad_norm": 2.0625,
      "grad_norm_var": 1.716387685139974,
      "learning_rate": 0.0001,
      "loss": 4.4209,
      "loss/crossentropy": 2.450512409210205,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23670841753482819,
      "step": 10150
    },
    {
      "epoch": 0.20304,
      "grad_norm": 1.984375,
      "grad_norm_var": 1.7187327067057292,
      "learning_rate": 0.0001,
      "loss": 4.1131,
      "loss/crossentropy": 2.3858957290649414,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23381955921649933,
      "step": 10152
    },
    {
      "epoch": 0.20308,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0060618082682291664,
      "learning_rate": 0.0001,
      "loss": 4.2154,
      "loss/crossentropy": 2.1214572191238403,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22399096935987473,
      "step": 10154
    },
    {
      "epoch": 0.20312,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0082916259765625,
      "learning_rate": 0.0001,
      "loss": 4.0487,
      "loss/crossentropy": 1.9299064874649048,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19886164367198944,
      "step": 10156
    },
    {
      "epoch": 0.20316,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.009877268473307292,
      "learning_rate": 0.0001,
      "loss": 4.108,
      "loss/crossentropy": 1.9582993388175964,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1859385445713997,
      "step": 10158
    },
    {
      "epoch": 0.2032,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.008790842692057292,
      "learning_rate": 0.0001,
      "loss": 4.408,
      "loss/crossentropy": 2.0555814504623413,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25231435894966125,
      "step": 10160
    },
    {
      "epoch": 0.20324,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.009090169270833334,
      "learning_rate": 0.0001,
      "loss": 4.2159,
      "loss/crossentropy": 1.9849395155906677,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20529592037200928,
      "step": 10162
    },
    {
      "epoch": 0.20328,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0071441650390625,
      "learning_rate": 0.0001,
      "loss": 4.5619,
      "loss/crossentropy": 1.9529971480369568,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1997845619916916,
      "step": 10164
    },
    {
      "epoch": 0.20332,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.009330240885416667,
      "learning_rate": 0.0001,
      "loss": 4.3117,
      "loss/crossentropy": 1.9885541200637817,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19847019761800766,
      "step": 10166
    },
    {
      "epoch": 0.20336,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.010013834635416666,
      "learning_rate": 0.0001,
      "loss": 4.3431,
      "loss/crossentropy": 1.8039852380752563,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19332807511091232,
      "step": 10168
    },
    {
      "epoch": 0.2034,
      "grad_norm": 2.125,
      "grad_norm_var": 0.01004638671875,
      "learning_rate": 0.0001,
      "loss": 4.4287,
      "loss/crossentropy": 2.1633352041244507,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22904948145151138,
      "step": 10170
    },
    {
      "epoch": 0.20344,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0070953369140625,
      "learning_rate": 0.0001,
      "loss": 4.1148,
      "loss/crossentropy": 2.2646039724349976,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22390951961278915,
      "step": 10172
    },
    {
      "epoch": 0.20348,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.005741119384765625,
      "learning_rate": 0.0001,
      "loss": 4.4114,
      "loss/crossentropy": 2.604608416557312,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25521157681941986,
      "step": 10174
    },
    {
      "epoch": 0.20352,
      "grad_norm": 2.125,
      "grad_norm_var": 0.016257476806640626,
      "learning_rate": 0.0001,
      "loss": 4.3319,
      "loss/crossentropy": 2.286113977432251,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23809745162725449,
      "step": 10176
    },
    {
      "epoch": 0.20356,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.01777928670247396,
      "learning_rate": 0.0001,
      "loss": 3.9951,
      "loss/crossentropy": 2.0746694207191467,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21103999018669128,
      "step": 10178
    },
    {
      "epoch": 0.2036,
      "grad_norm": 2.0,
      "grad_norm_var": 0.018381500244140626,
      "learning_rate": 0.0001,
      "loss": 3.9892,
      "loss/crossentropy": 1.9076440930366516,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2048335075378418,
      "step": 10180
    },
    {
      "epoch": 0.20364,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0178863525390625,
      "learning_rate": 0.0001,
      "loss": 4.1768,
      "loss/crossentropy": 1.8896904587745667,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20643991231918335,
      "step": 10182
    },
    {
      "epoch": 0.20368,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.016747029622395833,
      "learning_rate": 0.0001,
      "loss": 4.215,
      "loss/crossentropy": 2.156657338142395,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22853681445121765,
      "step": 10184
    },
    {
      "epoch": 0.20372,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.017772420247395834,
      "learning_rate": 0.0001,
      "loss": 4.6307,
      "loss/crossentropy": 2.3767203092575073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2422076091170311,
      "step": 10186
    },
    {
      "epoch": 0.20376,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.01834894816080729,
      "learning_rate": 0.0001,
      "loss": 4.3402,
      "loss/crossentropy": 2.5728834867477417,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22912005335092545,
      "step": 10188
    },
    {
      "epoch": 0.2038,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.02005182902018229,
      "learning_rate": 0.0001,
      "loss": 4.11,
      "loss/crossentropy": 2.1542125940322876,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19575025141239166,
      "step": 10190
    },
    {
      "epoch": 0.20384,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.004610188802083333,
      "learning_rate": 0.0001,
      "loss": 3.9114,
      "loss/crossentropy": 1.5845852494239807,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1845107451081276,
      "step": 10192
    },
    {
      "epoch": 0.20388,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.010957590738932292,
      "learning_rate": 0.0001,
      "loss": 4.2193,
      "loss/crossentropy": 2.1166247129440308,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2352868989109993,
      "step": 10194
    },
    {
      "epoch": 0.20392,
      "grad_norm": 2.0,
      "grad_norm_var": 0.010941314697265624,
      "learning_rate": 0.0001,
      "loss": 4.3374,
      "loss/crossentropy": 2.0129401683807373,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22231722623109818,
      "step": 10196
    },
    {
      "epoch": 0.20396,
      "grad_norm": 2.125,
      "grad_norm_var": 0.010595703125,
      "learning_rate": 0.0001,
      "loss": 3.9709,
      "loss/crossentropy": 1.7132073044776917,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18660317361354828,
      "step": 10198
    },
    {
      "epoch": 0.204,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.016743977864583332,
      "learning_rate": 0.0001,
      "loss": 4.3959,
      "loss/crossentropy": 2.0064845085144043,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20879874378442764,
      "step": 10200
    },
    {
      "epoch": 0.20404,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.01803766886393229,
      "learning_rate": 0.0001,
      "loss": 4.3558,
      "loss/crossentropy": 2.4205944538116455,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24201631546020508,
      "step": 10202
    },
    {
      "epoch": 0.20408,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.019694010416666668,
      "learning_rate": 0.0001,
      "loss": 3.8977,
      "loss/crossentropy": 2.0392255187034607,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21578273177146912,
      "step": 10204
    },
    {
      "epoch": 0.20412,
      "grad_norm": 2.25,
      "grad_norm_var": 0.019230143229166666,
      "learning_rate": 0.0001,
      "loss": 4.1414,
      "loss/crossentropy": 1.8652849197387695,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1896674558520317,
      "step": 10206
    },
    {
      "epoch": 0.20416,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.02332331339518229,
      "learning_rate": 0.0001,
      "loss": 4.7698,
      "loss/crossentropy": 2.016683042049408,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23144973814487457,
      "step": 10208
    },
    {
      "epoch": 0.2042,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.020401763916015624,
      "learning_rate": 0.0001,
      "loss": 4.3691,
      "loss/crossentropy": 1.9395010471343994,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19717292487621307,
      "step": 10210
    },
    {
      "epoch": 0.20424,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.0210845947265625,
      "learning_rate": 0.0001,
      "loss": 4.0907,
      "loss/crossentropy": 1.7626919150352478,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.200415201485157,
      "step": 10212
    },
    {
      "epoch": 0.20428,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.021955362955729165,
      "learning_rate": 0.0001,
      "loss": 4.3411,
      "loss/crossentropy": 2.3014339208602905,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22324562072753906,
      "step": 10214
    },
    {
      "epoch": 0.20432,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.019636027018229165,
      "learning_rate": 0.0001,
      "loss": 4.1237,
      "loss/crossentropy": 1.906779408454895,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2223847657442093,
      "step": 10216
    },
    {
      "epoch": 0.20436,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.01789118448893229,
      "learning_rate": 0.0001,
      "loss": 4.4555,
      "loss/crossentropy": 2.085246205329895,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21039249747991562,
      "step": 10218
    },
    {
      "epoch": 0.2044,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.014806874593098958,
      "learning_rate": 0.0001,
      "loss": 4.4477,
      "loss/crossentropy": 2.213107645511627,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22880114614963531,
      "step": 10220
    },
    {
      "epoch": 0.20444,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.013392893473307292,
      "learning_rate": 0.0001,
      "loss": 3.9015,
      "loss/crossentropy": 1.9510034322738647,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20935232937335968,
      "step": 10222
    },
    {
      "epoch": 0.20448,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.008906809488932292,
      "learning_rate": 0.0001,
      "loss": 4.1237,
      "loss/crossentropy": 1.8595823645591736,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20829308032989502,
      "step": 10224
    },
    {
      "epoch": 0.20452,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008990224202473958,
      "learning_rate": 0.0001,
      "loss": 4.1287,
      "loss/crossentropy": 1.8250519037246704,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18982955813407898,
      "step": 10226
    },
    {
      "epoch": 0.20456,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.0074859619140625,
      "learning_rate": 0.0001,
      "loss": 4.3656,
      "loss/crossentropy": 2.410372495651245,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22655458748340607,
      "step": 10228
    },
    {
      "epoch": 0.2046,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.007533518473307291,
      "learning_rate": 0.0001,
      "loss": 4.2581,
      "loss/crossentropy": 2.321051836013794,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2280098795890808,
      "step": 10230
    },
    {
      "epoch": 0.20464,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.007380167643229167,
      "learning_rate": 0.0001,
      "loss": 3.9927,
      "loss/crossentropy": 2.266388177871704,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2341095432639122,
      "step": 10232
    },
    {
      "epoch": 0.20468,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.007478841145833333,
      "learning_rate": 0.0001,
      "loss": 4.5263,
      "loss/crossentropy": 2.390430450439453,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23684432357549667,
      "step": 10234
    },
    {
      "epoch": 0.20472,
      "grad_norm": 3.390625,
      "grad_norm_var": 0.11719563802083334,
      "learning_rate": 0.0001,
      "loss": 4.4269,
      "loss/crossentropy": 2.07179594039917,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21902770549058914,
      "step": 10236
    },
    {
      "epoch": 0.20476,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.12704264322916667,
      "learning_rate": 0.0001,
      "loss": 4.3473,
      "loss/crossentropy": 1.686942458152771,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20007195323705673,
      "step": 10238
    },
    {
      "epoch": 0.2048,
      "grad_norm": 2.0,
      "grad_norm_var": 0.26913248697916664,
      "learning_rate": 0.0001,
      "loss": 4.2375,
      "loss/crossentropy": 2.0421791076660156,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2187313288450241,
      "step": 10240
    },
    {
      "epoch": 0.20484,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.26201985677083334,
      "learning_rate": 0.0001,
      "loss": 4.1641,
      "loss/crossentropy": 1.9503712058067322,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19879397749900818,
      "step": 10242
    },
    {
      "epoch": 0.20488,
      "grad_norm": 3.21875,
      "grad_norm_var": 0.82620849609375,
      "learning_rate": 0.0001,
      "loss": 4.413,
      "loss/crossentropy": 2.1453936100006104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22767861932516098,
      "step": 10244
    },
    {
      "epoch": 0.20492,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.808221181233724,
      "learning_rate": 0.0001,
      "loss": 4.1924,
      "loss/crossentropy": 1.9007731080055237,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20069654285907745,
      "step": 10246
    },
    {
      "epoch": 0.20496,
      "grad_norm": 1.875,
      "grad_norm_var": 0.8184832255045573,
      "learning_rate": 0.0001,
      "loss": 4.0685,
      "loss/crossentropy": 2.0545393228530884,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1983654722571373,
      "step": 10248
    },
    {
      "epoch": 0.205,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.804272206624349,
      "learning_rate": 0.0001,
      "loss": 4.3391,
      "loss/crossentropy": 2.158636450767517,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24239712953567505,
      "step": 10250
    },
    {
      "epoch": 0.20504,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.7762794494628906,
      "learning_rate": 0.0001,
      "loss": 4.1608,
      "loss/crossentropy": 2.1119120121002197,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2136205956339836,
      "step": 10252
    },
    {
      "epoch": 0.20508,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.7838417053222656,
      "learning_rate": 0.0001,
      "loss": 4.3184,
      "loss/crossentropy": 2.0690027475357056,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20616184175014496,
      "step": 10254
    },
    {
      "epoch": 0.20512,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.6926798502604167,
      "learning_rate": 0.0001,
      "loss": 4.0635,
      "loss/crossentropy": 2.178507924079895,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21580064296722412,
      "step": 10256
    },
    {
      "epoch": 0.20516,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.6889719645182292,
      "learning_rate": 0.0001,
      "loss": 4.1433,
      "loss/crossentropy": 2.292190670967102,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22163032740354538,
      "step": 10258
    },
    {
      "epoch": 0.2052,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.03568115234375,
      "learning_rate": 0.0001,
      "loss": 4.0138,
      "loss/crossentropy": 2.066649317741394,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2081037238240242,
      "step": 10260
    },
    {
      "epoch": 0.20524,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.05111490885416667,
      "learning_rate": 0.0001,
      "loss": 4.4315,
      "loss/crossentropy": 1.9017595052719116,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19284649938344955,
      "step": 10262
    },
    {
      "epoch": 0.20528,
      "grad_norm": 2.125,
      "grad_norm_var": 0.04326960245768229,
      "learning_rate": 0.0001,
      "loss": 4.0974,
      "loss/crossentropy": 2.1215697526931763,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2244422286748886,
      "step": 10264
    },
    {
      "epoch": 0.20532,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.02539647420247396,
      "learning_rate": 0.0001,
      "loss": 4.3231,
      "loss/crossentropy": 2.170191764831543,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2308938354253769,
      "step": 10266
    },
    {
      "epoch": 0.20536,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.02535985310872396,
      "learning_rate": 0.0001,
      "loss": 4.3472,
      "loss/crossentropy": 2.0430655479431152,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2435612976551056,
      "step": 10268
    },
    {
      "epoch": 0.2054,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.025394439697265625,
      "learning_rate": 0.0001,
      "loss": 4.536,
      "loss/crossentropy": 2.3141634464263916,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2234780564904213,
      "step": 10270
    },
    {
      "epoch": 0.20544,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.023371378580729168,
      "learning_rate": 0.0001,
      "loss": 4.1944,
      "loss/crossentropy": 2.310709834098816,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2505229711532593,
      "step": 10272
    },
    {
      "epoch": 0.20548,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.02434056599934896,
      "learning_rate": 0.0001,
      "loss": 4.0664,
      "loss/crossentropy": 1.9158611297607422,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2062816470861435,
      "step": 10274
    },
    {
      "epoch": 0.20552,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.02697728474934896,
      "learning_rate": 0.0001,
      "loss": 4.0545,
      "loss/crossentropy": 2.0835859179496765,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21051569283008575,
      "step": 10276
    },
    {
      "epoch": 0.20556,
      "grad_norm": 2.125,
      "grad_norm_var": 0.006951649983723958,
      "learning_rate": 0.0001,
      "loss": 4.4047,
      "loss/crossentropy": 1.9533037543296814,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20007294416427612,
      "step": 10278
    },
    {
      "epoch": 0.2056,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.010701497395833334,
      "learning_rate": 0.0001,
      "loss": 4.4499,
      "loss/crossentropy": 2.3090076446533203,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2295464426279068,
      "step": 10280
    },
    {
      "epoch": 0.20564,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.014989217122395834,
      "learning_rate": 0.0001,
      "loss": 4.2708,
      "loss/crossentropy": 2.2951393127441406,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22297964990139008,
      "step": 10282
    },
    {
      "epoch": 0.20568,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.014574178059895833,
      "learning_rate": 0.0001,
      "loss": 4.1153,
      "loss/crossentropy": 2.311514675617218,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2287898138165474,
      "step": 10284
    },
    {
      "epoch": 0.20572,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.014989217122395834,
      "learning_rate": 0.0001,
      "loss": 4.2287,
      "loss/crossentropy": 2.277890205383301,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21169421076774597,
      "step": 10286
    },
    {
      "epoch": 0.20576,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.015143839518229167,
      "learning_rate": 0.0001,
      "loss": 4.4448,
      "loss/crossentropy": 2.070693612098694,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21682647615671158,
      "step": 10288
    },
    {
      "epoch": 0.2058,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.014788564046223958,
      "learning_rate": 0.0001,
      "loss": 4.205,
      "loss/crossentropy": 2.223360061645508,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24135399609804153,
      "step": 10290
    },
    {
      "epoch": 0.20584,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.018155670166015624,
      "learning_rate": 0.0001,
      "loss": 4.3304,
      "loss/crossentropy": 2.430101752281189,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21579021215438843,
      "step": 10292
    },
    {
      "epoch": 0.20588,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.018173980712890624,
      "learning_rate": 0.0001,
      "loss": 4.1615,
      "loss/crossentropy": 1.960309624671936,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22619223594665527,
      "step": 10294
    },
    {
      "epoch": 0.20592,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.024217732747395835,
      "learning_rate": 0.0001,
      "loss": 4.8862,
      "loss/crossentropy": 2.0035970211029053,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20148956030607224,
      "step": 10296
    },
    {
      "epoch": 0.20596,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.021968587239583334,
      "learning_rate": 0.0001,
      "loss": 4.175,
      "loss/crossentropy": 1.976987361907959,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21035870164632797,
      "step": 10298
    },
    {
      "epoch": 0.206,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.021675618489583333,
      "learning_rate": 0.0001,
      "loss": 4.1596,
      "loss/crossentropy": 2.0631470680236816,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21725613623857498,
      "step": 10300
    },
    {
      "epoch": 0.20604,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.02693456013997396,
      "learning_rate": 0.0001,
      "loss": 4.4734,
      "loss/crossentropy": 2.2747987508773804,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2295055389404297,
      "step": 10302
    },
    {
      "epoch": 0.20608,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.029255167643229166,
      "learning_rate": 0.0001,
      "loss": 4.1255,
      "loss/crossentropy": 2.0811264514923096,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21890189498662949,
      "step": 10304
    },
    {
      "epoch": 0.20612,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.028319295247395834,
      "learning_rate": 0.0001,
      "loss": 4.4391,
      "loss/crossentropy": 2.2474766969680786,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2430667206645012,
      "step": 10306
    },
    {
      "epoch": 0.20616,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.024933878580729166,
      "learning_rate": 0.0001,
      "loss": 4.3016,
      "loss/crossentropy": 1.899698257446289,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21101202815771103,
      "step": 10308
    },
    {
      "epoch": 0.2062,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.023851521809895835,
      "learning_rate": 0.0001,
      "loss": 3.9769,
      "loss/crossentropy": 1.6432967782020569,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19004638493061066,
      "step": 10310
    },
    {
      "epoch": 0.20624,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0149658203125,
      "learning_rate": 0.0001,
      "loss": 4.3901,
      "loss/crossentropy": 1.9606398940086365,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24546240270137787,
      "step": 10312
    },
    {
      "epoch": 0.20628,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.015021769205729167,
      "learning_rate": 0.0001,
      "loss": 4.0988,
      "loss/crossentropy": 2.140414595603943,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2428836077451706,
      "step": 10314
    },
    {
      "epoch": 0.20632,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.018393707275390626,
      "learning_rate": 0.0001,
      "loss": 4.3768,
      "loss/crossentropy": 2.0886260271072388,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2201090231537819,
      "step": 10316
    },
    {
      "epoch": 0.20636,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.1039947509765625,
      "learning_rate": 0.0001,
      "loss": 4.5696,
      "loss/crossentropy": 2.3409098386764526,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24670831114053726,
      "step": 10318
    },
    {
      "epoch": 0.2064,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.10114313761393229,
      "learning_rate": 0.0001,
      "loss": 4.4153,
      "loss/crossentropy": 2.233125150203705,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21548831462860107,
      "step": 10320
    },
    {
      "epoch": 0.20644,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.1031206766764323,
      "learning_rate": 0.0001,
      "loss": 4.0617,
      "loss/crossentropy": 2.131038188934326,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20244313776493073,
      "step": 10322
    },
    {
      "epoch": 0.20648,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.10423965454101562,
      "learning_rate": 0.0001,
      "loss": 4.3428,
      "loss/crossentropy": 2.1683152318000793,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21852879226207733,
      "step": 10324
    },
    {
      "epoch": 0.20652,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.10465672810872396,
      "learning_rate": 0.0001,
      "loss": 4.4117,
      "loss/crossentropy": 2.2986634969711304,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24344030022621155,
      "step": 10326
    },
    {
      "epoch": 0.20656,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.10876057942708334,
      "learning_rate": 0.0001,
      "loss": 4.063,
      "loss/crossentropy": 2.2356297969818115,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21090354025363922,
      "step": 10328
    },
    {
      "epoch": 0.2066,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.10851949055989583,
      "learning_rate": 0.0001,
      "loss": 4.4235,
      "loss/crossentropy": 2.6431000232696533,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26503315567970276,
      "step": 10330
    },
    {
      "epoch": 0.20664,
      "grad_norm": 2.125,
      "grad_norm_var": 0.10668919881184896,
      "learning_rate": 0.0001,
      "loss": 4.2277,
      "loss/crossentropy": 2.0261669754981995,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21528497338294983,
      "step": 10332
    },
    {
      "epoch": 0.20668,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.007755279541015625,
      "learning_rate": 0.0001,
      "loss": 4.1092,
      "loss/crossentropy": 1.5593605041503906,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18069667369127274,
      "step": 10334
    },
    {
      "epoch": 0.20672,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0087646484375,
      "learning_rate": 0.0001,
      "loss": 4.1836,
      "loss/crossentropy": 1.953243374824524,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19982123374938965,
      "step": 10336
    },
    {
      "epoch": 0.20676,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0088531494140625,
      "learning_rate": 0.0001,
      "loss": 4.4369,
      "loss/crossentropy": 2.067806303501129,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2209654077887535,
      "step": 10338
    },
    {
      "epoch": 0.2068,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0088043212890625,
      "learning_rate": 0.0001,
      "loss": 4.2458,
      "loss/crossentropy": 1.9948397874832153,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21432363241910934,
      "step": 10340
    },
    {
      "epoch": 0.20684,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.010231272379557291,
      "learning_rate": 0.0001,
      "loss": 3.7643,
      "loss/crossentropy": 1.7932568788528442,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19161275029182434,
      "step": 10342
    },
    {
      "epoch": 0.20688,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.0162017822265625,
      "learning_rate": 0.0001,
      "loss": 4.5325,
      "loss/crossentropy": 2.021036922931671,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21150880306959152,
      "step": 10344
    },
    {
      "epoch": 0.20692,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.014623769124348958,
      "learning_rate": 0.0001,
      "loss": 3.9351,
      "loss/crossentropy": 2.0004186630249023,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18971875309944153,
      "step": 10346
    },
    {
      "epoch": 0.20696,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.014898427327473958,
      "learning_rate": 0.0001,
      "loss": 4.1234,
      "loss/crossentropy": 2.2949434518814087,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2356676608324051,
      "step": 10348
    },
    {
      "epoch": 0.207,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.013108062744140624,
      "learning_rate": 0.0001,
      "loss": 4.1755,
      "loss/crossentropy": 2.161319613456726,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24161628633737564,
      "step": 10350
    },
    {
      "epoch": 0.20704,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.013285319010416666,
      "learning_rate": 0.0001,
      "loss": 4.1432,
      "loss/crossentropy": 2.027602195739746,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19196761399507523,
      "step": 10352
    },
    {
      "epoch": 0.20708,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.013703409830729167,
      "learning_rate": 0.0001,
      "loss": 4.3658,
      "loss/crossentropy": 2.163583278656006,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21044812351465225,
      "step": 10354
    },
    {
      "epoch": 0.20712,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.014045969645182291,
      "learning_rate": 0.0001,
      "loss": 4.0419,
      "loss/crossentropy": 2.055150866508484,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21587203443050385,
      "step": 10356
    },
    {
      "epoch": 0.20716,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.0192047119140625,
      "learning_rate": 0.0001,
      "loss": 4.2947,
      "loss/crossentropy": 2.368631362915039,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2406565323472023,
      "step": 10358
    },
    {
      "epoch": 0.2072,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.016486612955729167,
      "learning_rate": 0.0001,
      "loss": 4.6473,
      "loss/crossentropy": 2.5399086475372314,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2554238885641098,
      "step": 10360
    },
    {
      "epoch": 0.20724,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.015366363525390624,
      "learning_rate": 0.0001,
      "loss": 4.2346,
      "loss/crossentropy": 2.0829185843467712,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2278829663991928,
      "step": 10362
    },
    {
      "epoch": 0.20728,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.014371490478515625,
      "learning_rate": 0.0001,
      "loss": 4.2583,
      "loss/crossentropy": 1.9829052090644836,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19677383452653885,
      "step": 10364
    },
    {
      "epoch": 0.20732,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.014385732014973958,
      "learning_rate": 0.0001,
      "loss": 4.3801,
      "loss/crossentropy": 2.2335458993911743,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23407185822725296,
      "step": 10366
    },
    {
      "epoch": 0.20736,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.012149810791015625,
      "learning_rate": 0.0001,
      "loss": 4.2074,
      "loss/crossentropy": 1.824280858039856,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.203867107629776,
      "step": 10368
    },
    {
      "epoch": 0.2074,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.013898722330729167,
      "learning_rate": 0.0001,
      "loss": 4.0666,
      "loss/crossentropy": 2.1007773876190186,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20312541723251343,
      "step": 10370
    },
    {
      "epoch": 0.20744,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.013350168863932291,
      "learning_rate": 0.0001,
      "loss": 4.3692,
      "loss/crossentropy": 1.953888475894928,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20697196573019028,
      "step": 10372
    },
    {
      "epoch": 0.20748,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.03509089152018229,
      "learning_rate": 0.0001,
      "loss": 4.2786,
      "loss/crossentropy": 2.186478853225708,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24410755187273026,
      "step": 10374
    },
    {
      "epoch": 0.20752,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.03706232706705729,
      "learning_rate": 0.0001,
      "loss": 3.9974,
      "loss/crossentropy": 1.9011998772621155,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1983819529414177,
      "step": 10376
    },
    {
      "epoch": 0.20756,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.03695246378580729,
      "learning_rate": 0.0001,
      "loss": 4.5884,
      "loss/crossentropy": 2.599787950515747,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25155550241470337,
      "step": 10378
    },
    {
      "epoch": 0.2076,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.037021636962890625,
      "learning_rate": 0.0001,
      "loss": 4.2702,
      "loss/crossentropy": 2.0120421648025513,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21281170845031738,
      "step": 10380
    },
    {
      "epoch": 0.20764,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.036710357666015624,
      "learning_rate": 0.0001,
      "loss": 4.4915,
      "loss/crossentropy": 2.0685967803001404,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23026303946971893,
      "step": 10382
    },
    {
      "epoch": 0.20768,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0445068359375,
      "learning_rate": 0.0001,
      "loss": 4.1171,
      "loss/crossentropy": 1.8392394185066223,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19240978360176086,
      "step": 10384
    },
    {
      "epoch": 0.20772,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.04523518880208333,
      "learning_rate": 0.0001,
      "loss": 4.2428,
      "loss/crossentropy": 1.786954402923584,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20547360181808472,
      "step": 10386
    },
    {
      "epoch": 0.20776,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.0452301025390625,
      "learning_rate": 0.0001,
      "loss": 4.2084,
      "loss/crossentropy": 1.6960806250572205,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1748766005039215,
      "step": 10388
    },
    {
      "epoch": 0.2078,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.018693033854166666,
      "learning_rate": 0.0001,
      "loss": 4.2491,
      "loss/crossentropy": 2.271879196166992,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25173740088939667,
      "step": 10390
    },
    {
      "epoch": 0.20784,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.015778605143229166,
      "learning_rate": 0.0001,
      "loss": 4.5199,
      "loss/crossentropy": 2.2860567569732666,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23291928321123123,
      "step": 10392
    },
    {
      "epoch": 0.20788,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0148193359375,
      "learning_rate": 0.0001,
      "loss": 4.1871,
      "loss/crossentropy": 1.9925037026405334,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.212161086499691,
      "step": 10394
    },
    {
      "epoch": 0.20792,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.016306304931640626,
      "learning_rate": 0.0001,
      "loss": 4.3728,
      "loss/crossentropy": 2.1189831495285034,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19486143440008163,
      "step": 10396
    },
    {
      "epoch": 0.20796,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0168121337890625,
      "learning_rate": 0.0001,
      "loss": 4.2325,
      "loss/crossentropy": 2.170132279396057,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23083829134702682,
      "step": 10398
    },
    {
      "epoch": 0.208,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.012129465738932291,
      "learning_rate": 0.0001,
      "loss": 4.0773,
      "loss/crossentropy": 1.8486035466194153,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18999971449375153,
      "step": 10400
    },
    {
      "epoch": 0.20804,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.011240386962890625,
      "learning_rate": 0.0001,
      "loss": 4.138,
      "loss/crossentropy": 1.727788269519806,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19234652817249298,
      "step": 10402
    },
    {
      "epoch": 0.20808,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.01046142578125,
      "learning_rate": 0.0001,
      "loss": 4.4102,
      "loss/crossentropy": 2.196265935897827,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21071433275938034,
      "step": 10404
    },
    {
      "epoch": 0.20812,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.007111612955729167,
      "learning_rate": 0.0001,
      "loss": 4.3378,
      "loss/crossentropy": 1.7230273485183716,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19570616632699966,
      "step": 10406
    },
    {
      "epoch": 0.20816,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.00567626953125,
      "learning_rate": 0.0001,
      "loss": 3.9693,
      "loss/crossentropy": 1.8815646767616272,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2055530995130539,
      "step": 10408
    },
    {
      "epoch": 0.2082,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.006400299072265625,
      "learning_rate": 0.0001,
      "loss": 4.1974,
      "loss/crossentropy": 2.2131329774856567,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22710958123207092,
      "step": 10410
    },
    {
      "epoch": 0.20824,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006103515625,
      "learning_rate": 0.0001,
      "loss": 4.1904,
      "loss/crossentropy": 1.5199981927871704,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16881641000509262,
      "step": 10412
    },
    {
      "epoch": 0.20828,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.005356597900390625,
      "learning_rate": 0.0001,
      "loss": 4.2456,
      "loss/crossentropy": 2.305867075920105,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2523300349712372,
      "step": 10414
    },
    {
      "epoch": 0.20832,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.004571278889973958,
      "learning_rate": 0.0001,
      "loss": 4.2727,
      "loss/crossentropy": 1.989980161190033,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19124356657266617,
      "step": 10416
    },
    {
      "epoch": 0.20836,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.004255167643229167,
      "learning_rate": 0.0001,
      "loss": 4.4745,
      "loss/crossentropy": 2.165328025817871,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23673278093338013,
      "step": 10418
    },
    {
      "epoch": 0.2084,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.008074696858723958,
      "learning_rate": 0.0001,
      "loss": 4.5429,
      "loss/crossentropy": 2.2451056241989136,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23130114376544952,
      "step": 10420
    },
    {
      "epoch": 0.20844,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0106689453125,
      "learning_rate": 0.0001,
      "loss": 3.9259,
      "loss/crossentropy": 2.1694064140319824,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21424317359924316,
      "step": 10422
    },
    {
      "epoch": 0.20848,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.012059529622395834,
      "learning_rate": 0.0001,
      "loss": 4.3558,
      "loss/crossentropy": 2.140601873397827,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22452331334352493,
      "step": 10424
    },
    {
      "epoch": 0.20852,
      "grad_norm": 2.125,
      "grad_norm_var": 0.011476389567057292,
      "learning_rate": 0.0001,
      "loss": 4.3058,
      "loss/crossentropy": 2.2076770067214966,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21608063578605652,
      "step": 10426
    },
    {
      "epoch": 0.20856,
      "grad_norm": 2.125,
      "grad_norm_var": 0.011579386393229167,
      "learning_rate": 0.0001,
      "loss": 4.151,
      "loss/crossentropy": 2.1738568544387817,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2288571298122406,
      "step": 10428
    },
    {
      "epoch": 0.2086,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.012165323893229166,
      "learning_rate": 0.0001,
      "loss": 4.2576,
      "loss/crossentropy": 2.2586612701416016,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22107571363449097,
      "step": 10430
    },
    {
      "epoch": 0.20864,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.014511871337890624,
      "learning_rate": 0.0001,
      "loss": 4.5441,
      "loss/crossentropy": 2.336306095123291,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23462744057178497,
      "step": 10432
    },
    {
      "epoch": 0.20868,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0148590087890625,
      "learning_rate": 0.0001,
      "loss": 4.1625,
      "loss/crossentropy": 2.3164994716644287,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21541497856378555,
      "step": 10434
    },
    {
      "epoch": 0.20872,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0121734619140625,
      "learning_rate": 0.0001,
      "loss": 4.2497,
      "loss/crossentropy": 1.848636507987976,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20993927866220474,
      "step": 10436
    },
    {
      "epoch": 0.20876,
      "grad_norm": 2.25,
      "grad_norm_var": 0.010509999593098958,
      "learning_rate": 0.0001,
      "loss": 4.4985,
      "loss/crossentropy": 2.234964370727539,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21675845235586166,
      "step": 10438
    },
    {
      "epoch": 0.2088,
      "grad_norm": 2.125,
      "grad_norm_var": 0.009159088134765625,
      "learning_rate": 0.0001,
      "loss": 4.2389,
      "loss/crossentropy": 1.9301238656044006,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21708600223064423,
      "step": 10440
    },
    {
      "epoch": 0.20884,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.010721842447916666,
      "learning_rate": 0.0001,
      "loss": 4.1706,
      "loss/crossentropy": 2.231620192527771,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23156532645225525,
      "step": 10442
    },
    {
      "epoch": 0.20888,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010237375895182291,
      "learning_rate": 0.0001,
      "loss": 4.18,
      "loss/crossentropy": 1.8612747192382812,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1914522647857666,
      "step": 10444
    },
    {
      "epoch": 0.20892,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.009907786051432292,
      "learning_rate": 0.0001,
      "loss": 4.1713,
      "loss/crossentropy": 2.2229456305503845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23199696838855743,
      "step": 10446
    },
    {
      "epoch": 0.20896,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.009708658854166666,
      "learning_rate": 0.0001,
      "loss": 4.0556,
      "loss/crossentropy": 1.9055940508842468,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2158679962158203,
      "step": 10448
    },
    {
      "epoch": 0.209,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.011774698893229166,
      "learning_rate": 0.0001,
      "loss": 3.8876,
      "loss/crossentropy": 1.5537404417991638,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17176809161901474,
      "step": 10450
    },
    {
      "epoch": 0.20904,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0123931884765625,
      "learning_rate": 0.0001,
      "loss": 3.9222,
      "loss/crossentropy": 2.319318413734436,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22001250088214874,
      "step": 10452
    },
    {
      "epoch": 0.20908,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.010383097330729167,
      "learning_rate": 0.0001,
      "loss": 4.421,
      "loss/crossentropy": 2.2334396839141846,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.222968190908432,
      "step": 10454
    },
    {
      "epoch": 0.20912,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0110260009765625,
      "learning_rate": 0.0001,
      "loss": 4.5172,
      "loss/crossentropy": 2.5762773752212524,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26605312526226044,
      "step": 10456
    },
    {
      "epoch": 0.20916,
      "grad_norm": 2.0,
      "grad_norm_var": 0.010503896077473958,
      "learning_rate": 0.0001,
      "loss": 4.1109,
      "loss/crossentropy": 1.7634761333465576,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18641388416290283,
      "step": 10458
    },
    {
      "epoch": 0.2092,
      "grad_norm": 1.875,
      "grad_norm_var": 0.013224029541015625,
      "learning_rate": 0.0001,
      "loss": 4.126,
      "loss/crossentropy": 2.28191876411438,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22916647791862488,
      "step": 10460
    },
    {
      "epoch": 0.20924,
      "grad_norm": 2.125,
      "grad_norm_var": 0.013903554280598958,
      "learning_rate": 0.0001,
      "loss": 4.1984,
      "loss/crossentropy": 1.9712265729904175,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2293689101934433,
      "step": 10462
    },
    {
      "epoch": 0.20928,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.012292226155598959,
      "learning_rate": 0.0001,
      "loss": 4.3826,
      "loss/crossentropy": 2.425857424736023,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.255269430577755,
      "step": 10464
    },
    {
      "epoch": 0.20932,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.011031087239583333,
      "learning_rate": 0.0001,
      "loss": 4.1919,
      "loss/crossentropy": 2.0697131752967834,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22233501076698303,
      "step": 10466
    },
    {
      "epoch": 0.20936,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.008854166666666666,
      "learning_rate": 0.0001,
      "loss": 4.4017,
      "loss/crossentropy": 1.8367178440093994,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2063748985528946,
      "step": 10468
    },
    {
      "epoch": 0.2094,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.018192545572916666,
      "learning_rate": 0.0001,
      "loss": 4.1864,
      "loss/crossentropy": 1.977232813835144,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20568673312664032,
      "step": 10470
    },
    {
      "epoch": 0.20944,
      "grad_norm": 2.0,
      "grad_norm_var": 0.018195597330729167,
      "learning_rate": 0.0001,
      "loss": 4.1622,
      "loss/crossentropy": 2.1934465169906616,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2175101339817047,
      "step": 10472
    },
    {
      "epoch": 0.20948,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.016047159830729168,
      "learning_rate": 0.0001,
      "loss": 4.3005,
      "loss/crossentropy": 1.7997339367866516,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20166774094104767,
      "step": 10474
    },
    {
      "epoch": 0.20952,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.014671834309895833,
      "learning_rate": 0.0001,
      "loss": 4.1167,
      "loss/crossentropy": 1.9334582090377808,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19190441817045212,
      "step": 10476
    },
    {
      "epoch": 0.20956,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.015148671468098958,
      "learning_rate": 0.0001,
      "loss": 3.9434,
      "loss/crossentropy": 1.7263885140419006,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17968116700649261,
      "step": 10478
    },
    {
      "epoch": 0.2096,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.01661961873372396,
      "learning_rate": 0.0001,
      "loss": 4.1153,
      "loss/crossentropy": 2.1710296869277954,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19717055559158325,
      "step": 10480
    },
    {
      "epoch": 0.20964,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.016947174072265626,
      "learning_rate": 0.0001,
      "loss": 4.2089,
      "loss/crossentropy": 2.043896973133087,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21403591334819794,
      "step": 10482
    },
    {
      "epoch": 0.20968,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.02459691365559896,
      "learning_rate": 0.0001,
      "loss": 3.9237,
      "loss/crossentropy": 1.7813313603401184,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20273292809724808,
      "step": 10484
    },
    {
      "epoch": 0.20972,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.013155110677083333,
      "learning_rate": 0.0001,
      "loss": 4.0985,
      "loss/crossentropy": 1.8628470301628113,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1945454627275467,
      "step": 10486
    },
    {
      "epoch": 0.20976,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.013044230143229167,
      "learning_rate": 0.0001,
      "loss": 4.2279,
      "loss/crossentropy": 2.2047882080078125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23798923939466476,
      "step": 10488
    },
    {
      "epoch": 0.2098,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.012697092692057292,
      "learning_rate": 0.0001,
      "loss": 4.1327,
      "loss/crossentropy": 1.8838441967964172,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2224583625793457,
      "step": 10490
    },
    {
      "epoch": 0.20984,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011201731363932292,
      "learning_rate": 0.0001,
      "loss": 4.1985,
      "loss/crossentropy": 1.9326539039611816,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2116800993680954,
      "step": 10492
    },
    {
      "epoch": 0.20988,
      "grad_norm": 2.375,
      "grad_norm_var": 0.017829386393229167,
      "learning_rate": 0.0001,
      "loss": 4.2023,
      "loss/crossentropy": 1.7767577171325684,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20496132969856262,
      "step": 10494
    },
    {
      "epoch": 0.20992,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.016996256510416665,
      "learning_rate": 0.0001,
      "loss": 4.1665,
      "loss/crossentropy": 1.996176838874817,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1979241669178009,
      "step": 10496
    },
    {
      "epoch": 0.20996,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.016136678059895833,
      "learning_rate": 0.0001,
      "loss": 4.1467,
      "loss/crossentropy": 2.5282262563705444,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23003733158111572,
      "step": 10498
    },
    {
      "epoch": 0.21,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.008760579427083333,
      "learning_rate": 0.0001,
      "loss": 4.1981,
      "loss/crossentropy": 2.3120675086975098,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22543538361787796,
      "step": 10500
    },
    {
      "epoch": 0.21004,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.008766428629557291,
      "learning_rate": 0.0001,
      "loss": 4.1216,
      "loss/crossentropy": 2.2110280990600586,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23066550493240356,
      "step": 10502
    },
    {
      "epoch": 0.21008,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.009242502848307292,
      "learning_rate": 0.0001,
      "loss": 4.0804,
      "loss/crossentropy": 1.6792908906936646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20689093321561813,
      "step": 10504
    },
    {
      "epoch": 0.21012,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.011116536458333333,
      "learning_rate": 0.0001,
      "loss": 4.137,
      "loss/crossentropy": 1.9797767400741577,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2185518741607666,
      "step": 10506
    },
    {
      "epoch": 0.21016,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.011533355712890625,
      "learning_rate": 0.0001,
      "loss": 4.0243,
      "loss/crossentropy": 2.0001984238624573,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22058366239070892,
      "step": 10508
    },
    {
      "epoch": 0.2102,
      "grad_norm": 2.125,
      "grad_norm_var": 0.006414540608723958,
      "learning_rate": 0.0001,
      "loss": 4.3707,
      "loss/crossentropy": 2.1622806787490845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21733585745096207,
      "step": 10510
    },
    {
      "epoch": 0.21024,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.00740966796875,
      "learning_rate": 0.0001,
      "loss": 4.0392,
      "loss/crossentropy": 2.0224735736846924,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20596059411764145,
      "step": 10512
    },
    {
      "epoch": 0.21028,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.011244455973307291,
      "learning_rate": 0.0001,
      "loss": 4.0712,
      "loss/crossentropy": 2.099945902824402,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19777391105890274,
      "step": 10514
    },
    {
      "epoch": 0.21032,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.011823527018229167,
      "learning_rate": 0.0001,
      "loss": 4.1246,
      "loss/crossentropy": 1.9806578159332275,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19959519058465958,
      "step": 10516
    },
    {
      "epoch": 0.21036,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.012239329020182292,
      "learning_rate": 0.0001,
      "loss": 4.1182,
      "loss/crossentropy": 2.0056468844413757,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21810457110404968,
      "step": 10518
    },
    {
      "epoch": 0.2104,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.01785456339518229,
      "learning_rate": 0.0001,
      "loss": 4.1085,
      "loss/crossentropy": 1.9525137543678284,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19195258617401123,
      "step": 10520
    },
    {
      "epoch": 0.21044,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.01580988566080729,
      "learning_rate": 0.0001,
      "loss": 4.4648,
      "loss/crossentropy": 2.2391252517700195,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23115740716457367,
      "step": 10522
    },
    {
      "epoch": 0.21048,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.016727701822916666,
      "learning_rate": 0.0001,
      "loss": 4.6467,
      "loss/crossentropy": 2.550819158554077,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2846238762140274,
      "step": 10524
    },
    {
      "epoch": 0.21052,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.0176177978515625,
      "learning_rate": 0.0001,
      "loss": 4.2787,
      "loss/crossentropy": 2.181081712245941,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23952369391918182,
      "step": 10526
    },
    {
      "epoch": 0.21056,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.01784032185872396,
      "learning_rate": 0.0001,
      "loss": 4.1627,
      "loss/crossentropy": 2.3070446848869324,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21255015581846237,
      "step": 10528
    },
    {
      "epoch": 0.2106,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.014235178629557291,
      "learning_rate": 0.0001,
      "loss": 4.3184,
      "loss/crossentropy": 1.9560331106185913,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19652864336967468,
      "step": 10530
    },
    {
      "epoch": 0.21064,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.0127593994140625,
      "learning_rate": 0.0001,
      "loss": 4.5728,
      "loss/crossentropy": 2.2470709085464478,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25076867640018463,
      "step": 10532
    },
    {
      "epoch": 0.21068,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.011767323811848958,
      "learning_rate": 0.0001,
      "loss": 4.2652,
      "loss/crossentropy": 2.0689820051193237,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21815705299377441,
      "step": 10534
    },
    {
      "epoch": 0.21072,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.013499959309895834,
      "learning_rate": 0.0001,
      "loss": 3.9524,
      "loss/crossentropy": 2.0633797645568848,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20865381509065628,
      "step": 10536
    },
    {
      "epoch": 0.21076,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.014029947916666667,
      "learning_rate": 0.0001,
      "loss": 4.2644,
      "loss/crossentropy": 2.3332748413085938,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21530094742774963,
      "step": 10538
    },
    {
      "epoch": 0.2108,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.011942545572916666,
      "learning_rate": 0.0001,
      "loss": 4.1452,
      "loss/crossentropy": 2.093048572540283,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23568043112754822,
      "step": 10540
    },
    {
      "epoch": 0.21084,
      "grad_norm": 2.125,
      "grad_norm_var": 0.009992472330729167,
      "learning_rate": 0.0001,
      "loss": 4.5464,
      "loss/crossentropy": 2.0059397220611572,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.253838449716568,
      "step": 10542
    },
    {
      "epoch": 0.21088,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009284464518229167,
      "learning_rate": 0.0001,
      "loss": 4.1417,
      "loss/crossentropy": 2.278248429298401,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23320979624986649,
      "step": 10544
    },
    {
      "epoch": 0.21092,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.008540598551432292,
      "learning_rate": 0.0001,
      "loss": 4.0951,
      "loss/crossentropy": 2.063527822494507,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21290308982133865,
      "step": 10546
    },
    {
      "epoch": 0.21096,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.007511138916015625,
      "learning_rate": 0.0001,
      "loss": 4.1456,
      "loss/crossentropy": 2.092045545578003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21721098572015762,
      "step": 10548
    },
    {
      "epoch": 0.211,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.06824111938476562,
      "learning_rate": 0.0001,
      "loss": 4.0586,
      "loss/crossentropy": 1.9876770973205566,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21775592118501663,
      "step": 10550
    },
    {
      "epoch": 0.21104,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.06347249348958334,
      "learning_rate": 0.0001,
      "loss": 4.2372,
      "loss/crossentropy": 2.292428970336914,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24829304218292236,
      "step": 10552
    },
    {
      "epoch": 0.21108,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.06552632649739583,
      "learning_rate": 0.0001,
      "loss": 4.3412,
      "loss/crossentropy": 2.257239043712616,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2195405438542366,
      "step": 10554
    },
    {
      "epoch": 0.21112,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.06616923014322916,
      "learning_rate": 0.0001,
      "loss": 4.404,
      "loss/crossentropy": 2.1424754858016968,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21044345945119858,
      "step": 10556
    },
    {
      "epoch": 0.21116,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.06642964680989584,
      "learning_rate": 0.0001,
      "loss": 4.2812,
      "loss/crossentropy": 1.777747094631195,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19684316962957382,
      "step": 10558
    },
    {
      "epoch": 0.2112,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.06655171712239584,
      "learning_rate": 0.0001,
      "loss": 4.1239,
      "loss/crossentropy": 2.0860772728919983,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20560266077518463,
      "step": 10560
    },
    {
      "epoch": 0.21124,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.06787007649739583,
      "learning_rate": 0.0001,
      "loss": 4.2409,
      "loss/crossentropy": 1.9594369530677795,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2015877440571785,
      "step": 10562
    },
    {
      "epoch": 0.21128,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.06467692057291667,
      "learning_rate": 0.0001,
      "loss": 4.2539,
      "loss/crossentropy": 2.069046676158905,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21409741044044495,
      "step": 10564
    },
    {
      "epoch": 0.21132,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.009464263916015625,
      "learning_rate": 0.0001,
      "loss": 3.822,
      "loss/crossentropy": 1.7300589084625244,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18024658411741257,
      "step": 10566
    },
    {
      "epoch": 0.21136,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0077288309733072914,
      "learning_rate": 0.0001,
      "loss": 4.2608,
      "loss/crossentropy": 2.048615336418152,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.228649340569973,
      "step": 10568
    },
    {
      "epoch": 0.2114,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.007085927327473958,
      "learning_rate": 0.0001,
      "loss": 4.4622,
      "loss/crossentropy": 1.9678268432617188,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20117972791194916,
      "step": 10570
    },
    {
      "epoch": 0.21144,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.006534576416015625,
      "learning_rate": 0.0001,
      "loss": 4.3919,
      "loss/crossentropy": 2.595113754272461,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24565115571022034,
      "step": 10572
    },
    {
      "epoch": 0.21148,
      "grad_norm": 2.25,
      "grad_norm_var": 0.008658599853515626,
      "learning_rate": 0.0001,
      "loss": 4.0873,
      "loss/crossentropy": 2.096100628376007,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20913005620241165,
      "step": 10574
    },
    {
      "epoch": 0.21152,
      "grad_norm": 2.25,
      "grad_norm_var": 0.012223052978515624,
      "learning_rate": 0.0001,
      "loss": 4.181,
      "loss/crossentropy": 2.0096259713172913,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21125900745391846,
      "step": 10576
    },
    {
      "epoch": 0.21156,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.011572011311848958,
      "learning_rate": 0.0001,
      "loss": 3.9738,
      "loss/crossentropy": 1.9990533590316772,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20894166082143784,
      "step": 10578
    },
    {
      "epoch": 0.2116,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.013606516520182292,
      "learning_rate": 0.0001,
      "loss": 4.0404,
      "loss/crossentropy": 2.1385116577148438,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19753456860780716,
      "step": 10580
    },
    {
      "epoch": 0.21164,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.012482706705729167,
      "learning_rate": 0.0001,
      "loss": 4.5056,
      "loss/crossentropy": 2.2081239819526672,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23155531287193298,
      "step": 10582
    },
    {
      "epoch": 0.21168,
      "grad_norm": 2.125,
      "grad_norm_var": 0.014339192708333334,
      "learning_rate": 0.0001,
      "loss": 4.161,
      "loss/crossentropy": 1.6915860772132874,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18060292303562164,
      "step": 10584
    },
    {
      "epoch": 0.21172,
      "grad_norm": 1.875,
      "grad_norm_var": 0.019245402018229166,
      "learning_rate": 0.0001,
      "loss": 4.083,
      "loss/crossentropy": 1.5825872421264648,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19387374818325043,
      "step": 10586
    },
    {
      "epoch": 0.21176,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0253082275390625,
      "learning_rate": 0.0001,
      "loss": 4.6102,
      "loss/crossentropy": 2.386876940727234,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23594971746206284,
      "step": 10588
    },
    {
      "epoch": 0.2118,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.02759577433268229,
      "learning_rate": 0.0001,
      "loss": 4.106,
      "loss/crossentropy": 2.202653169631958,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22666631639003754,
      "step": 10590
    },
    {
      "epoch": 0.21184,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.026041412353515626,
      "learning_rate": 0.0001,
      "loss": 4.2285,
      "loss/crossentropy": 2.013135075569153,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22155165672302246,
      "step": 10592
    },
    {
      "epoch": 0.21188,
      "grad_norm": 2.25,
      "grad_norm_var": 2.762861887613932,
      "learning_rate": 0.0001,
      "loss": 4.1353,
      "loss/crossentropy": 1.6701499223709106,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21808167546987534,
      "step": 10594
    },
    {
      "epoch": 0.21192,
      "grad_norm": 2.015625,
      "grad_norm_var": 2.7677996317545572,
      "learning_rate": 0.0001,
      "loss": 4.1438,
      "loss/crossentropy": 2.074462592601776,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22254129499197006,
      "step": 10596
    },
    {
      "epoch": 0.21196,
      "grad_norm": 2.203125,
      "grad_norm_var": 2.787275950113932,
      "learning_rate": 0.0001,
      "loss": 4.2329,
      "loss/crossentropy": 2.182308316230774,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21201200038194656,
      "step": 10598
    },
    {
      "epoch": 0.212,
      "grad_norm": 2.0625,
      "grad_norm_var": 2.784148915608724,
      "learning_rate": 0.0001,
      "loss": 4.0688,
      "loss/crossentropy": 1.9777602553367615,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21454624831676483,
      "step": 10600
    },
    {
      "epoch": 0.21204,
      "grad_norm": 2.03125,
      "grad_norm_var": 2.778930409749349,
      "learning_rate": 0.0001,
      "loss": 4.2707,
      "loss/crossentropy": 2.272592306137085,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22290532290935516,
      "step": 10602
    },
    {
      "epoch": 0.21208,
      "grad_norm": 2.078125,
      "grad_norm_var": 2.784010569254557,
      "learning_rate": 0.0001,
      "loss": 4.3406,
      "loss/crossentropy": 2.110643744468689,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22552277147769928,
      "step": 10604
    },
    {
      "epoch": 0.21212,
      "grad_norm": 2.125,
      "grad_norm_var": 2.7753326416015627,
      "learning_rate": 0.0001,
      "loss": 4.3485,
      "loss/crossentropy": 2.279823422431946,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22356732934713364,
      "step": 10606
    },
    {
      "epoch": 0.21216,
      "grad_norm": 2.109375,
      "grad_norm_var": 2.768701171875,
      "learning_rate": 0.0001,
      "loss": 4.3203,
      "loss/crossentropy": 2.0985517501831055,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22894078493118286,
      "step": 10608
    },
    {
      "epoch": 0.2122,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006884765625,
      "learning_rate": 0.0001,
      "loss": 4.2295,
      "loss/crossentropy": 2.251029133796692,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24473578482866287,
      "step": 10610
    },
    {
      "epoch": 0.21224,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.005785115559895833,
      "learning_rate": 0.0001,
      "loss": 4.2451,
      "loss/crossentropy": 2.1706892251968384,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21725767105817795,
      "step": 10612
    },
    {
      "epoch": 0.21228,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.004378255208333333,
      "learning_rate": 0.0001,
      "loss": 4.3319,
      "loss/crossentropy": 2.0709590315818787,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22771050035953522,
      "step": 10614
    },
    {
      "epoch": 0.21232,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.005163319905598958,
      "learning_rate": 0.0001,
      "loss": 4.2507,
      "loss/crossentropy": 2.0437510013580322,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2055979147553444,
      "step": 10616
    },
    {
      "epoch": 0.21236,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.005204010009765625,
      "learning_rate": 0.0001,
      "loss": 4.2492,
      "loss/crossentropy": 2.023369252681732,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23950359225273132,
      "step": 10618
    },
    {
      "epoch": 0.2124,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.004325103759765625,
      "learning_rate": 0.0001,
      "loss": 4.3522,
      "loss/crossentropy": 2.051850199699402,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24145089089870453,
      "step": 10620
    },
    {
      "epoch": 0.21244,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0044247945149739586,
      "learning_rate": 0.0001,
      "loss": 4.1743,
      "loss/crossentropy": 1.9617546796798706,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2095404863357544,
      "step": 10622
    },
    {
      "epoch": 0.21248,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.004662831624348958,
      "learning_rate": 0.0001,
      "loss": 4.2115,
      "loss/crossentropy": 1.9792284965515137,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21156969666481018,
      "step": 10624
    },
    {
      "epoch": 0.21252,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.004662831624348958,
      "learning_rate": 0.0001,
      "loss": 4.4402,
      "loss/crossentropy": 1.936375379562378,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21360263973474503,
      "step": 10626
    },
    {
      "epoch": 0.21256,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.0071408589680989586,
      "learning_rate": 0.0001,
      "loss": 4.0695,
      "loss/crossentropy": 2.11286723613739,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20090097934007645,
      "step": 10628
    },
    {
      "epoch": 0.2126,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.007120513916015625,
      "learning_rate": 0.0001,
      "loss": 4.2555,
      "loss/crossentropy": 2.1020379066467285,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21139420568943024,
      "step": 10630
    },
    {
      "epoch": 0.21264,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0064453125,
      "learning_rate": 0.0001,
      "loss": 4.2027,
      "loss/crossentropy": 2.220509111881256,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22286564111709595,
      "step": 10632
    },
    {
      "epoch": 0.21268,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0080078125,
      "learning_rate": 0.0001,
      "loss": 4.0012,
      "loss/crossentropy": 2.058075189590454,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22242067009210587,
      "step": 10634
    },
    {
      "epoch": 0.21272,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.005269368489583333,
      "learning_rate": 0.0001,
      "loss": 4.1825,
      "loss/crossentropy": 1.9641217589378357,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20702090859413147,
      "step": 10636
    },
    {
      "epoch": 0.21276,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.005231730143229167,
      "learning_rate": 0.0001,
      "loss": 4.1309,
      "loss/crossentropy": 1.9234120845794678,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20697829127311707,
      "step": 10638
    },
    {
      "epoch": 0.2128,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0067047119140625,
      "learning_rate": 0.0001,
      "loss": 4.2551,
      "loss/crossentropy": 2.1050453782081604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21321691572666168,
      "step": 10640
    },
    {
      "epoch": 0.21284,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.009323883056640624,
      "learning_rate": 0.0001,
      "loss": 4.0852,
      "loss/crossentropy": 2.21867573261261,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21468330919742584,
      "step": 10642
    },
    {
      "epoch": 0.21288,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.007319895426432291,
      "learning_rate": 0.0001,
      "loss": 4.4209,
      "loss/crossentropy": 2.213089942932129,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21172764152288437,
      "step": 10644
    },
    {
      "epoch": 0.21292,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.012556711832682291,
      "learning_rate": 0.0001,
      "loss": 4.3261,
      "loss/crossentropy": 2.0370752811431885,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23427169024944305,
      "step": 10646
    },
    {
      "epoch": 0.21296,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.016035715738932293,
      "learning_rate": 0.0001,
      "loss": 4.2622,
      "loss/crossentropy": 2.080373227596283,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2148713618516922,
      "step": 10648
    },
    {
      "epoch": 0.213,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.014134724934895834,
      "learning_rate": 0.0001,
      "loss": 4.5408,
      "loss/crossentropy": 2.3023130893707275,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22768481075763702,
      "step": 10650
    },
    {
      "epoch": 0.21304,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.016185506184895834,
      "learning_rate": 0.0001,
      "loss": 4.372,
      "loss/crossentropy": 2.14642870426178,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22463608533143997,
      "step": 10652
    },
    {
      "epoch": 0.21308,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.016377766927083332,
      "learning_rate": 0.0001,
      "loss": 4.2164,
      "loss/crossentropy": 2.2469639778137207,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2505262568593025,
      "step": 10654
    },
    {
      "epoch": 0.21312,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.015915679931640624,
      "learning_rate": 0.0001,
      "loss": 4.243,
      "loss/crossentropy": 2.0431448221206665,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21013544499874115,
      "step": 10656
    },
    {
      "epoch": 0.21316,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.013631184895833334,
      "learning_rate": 0.0001,
      "loss": 3.9872,
      "loss/crossentropy": 1.6768526434898376,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18652133643627167,
      "step": 10658
    },
    {
      "epoch": 0.2132,
      "grad_norm": 2.0,
      "grad_norm_var": 0.017175038655598957,
      "learning_rate": 0.0001,
      "loss": 3.9442,
      "loss/crossentropy": 1.3748261332511902,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15284860879182816,
      "step": 10660
    },
    {
      "epoch": 0.21324,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.012282053629557291,
      "learning_rate": 0.0001,
      "loss": 4.1451,
      "loss/crossentropy": 1.9126732349395752,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21141140908002853,
      "step": 10662
    },
    {
      "epoch": 0.21328,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.010908762613932291,
      "learning_rate": 0.0001,
      "loss": 4.2392,
      "loss/crossentropy": 1.97357976436615,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21326088905334473,
      "step": 10664
    },
    {
      "epoch": 0.21332,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.011201985677083333,
      "learning_rate": 0.0001,
      "loss": 4.2825,
      "loss/crossentropy": 2.1782814860343933,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21183249354362488,
      "step": 10666
    },
    {
      "epoch": 0.21336,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.009098307291666666,
      "learning_rate": 0.0001,
      "loss": 4.1251,
      "loss/crossentropy": 2.1700649857521057,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2310282066464424,
      "step": 10668
    },
    {
      "epoch": 0.2134,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0087554931640625,
      "learning_rate": 0.0001,
      "loss": 4.479,
      "loss/crossentropy": 2.249666213989258,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21025578677654266,
      "step": 10670
    },
    {
      "epoch": 0.21344,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.009287261962890625,
      "learning_rate": 0.0001,
      "loss": 4.3445,
      "loss/crossentropy": 1.993752121925354,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20829375833272934,
      "step": 10672
    },
    {
      "epoch": 0.21348,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.009244791666666667,
      "learning_rate": 0.0001,
      "loss": 4.0814,
      "loss/crossentropy": 2.0472013354301453,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21456149220466614,
      "step": 10674
    },
    {
      "epoch": 0.21352,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.007222239176432292,
      "learning_rate": 0.0001,
      "loss": 3.9497,
      "loss/crossentropy": 2.0347819328308105,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2133965790271759,
      "step": 10676
    },
    {
      "epoch": 0.21356,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.008168284098307292,
      "learning_rate": 0.0001,
      "loss": 4.1528,
      "loss/crossentropy": 1.86410254240036,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22139018028974533,
      "step": 10678
    },
    {
      "epoch": 0.2136,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.005641428629557291,
      "learning_rate": 0.0001,
      "loss": 4.3134,
      "loss/crossentropy": 2.3244482278823853,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24854417145252228,
      "step": 10680
    },
    {
      "epoch": 0.21364,
      "grad_norm": 2.0,
      "grad_norm_var": 0.006304677327473958,
      "learning_rate": 0.0001,
      "loss": 4.7182,
      "loss/crossentropy": 2.518718123435974,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.274397537112236,
      "step": 10682
    },
    {
      "epoch": 0.21368,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.005928548177083334,
      "learning_rate": 0.0001,
      "loss": 4.1289,
      "loss/crossentropy": 1.8111079931259155,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2039540931582451,
      "step": 10684
    },
    {
      "epoch": 0.21372,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.008129628499348958,
      "learning_rate": 0.0001,
      "loss": 4.3446,
      "loss/crossentropy": 2.186485230922699,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22497782111167908,
      "step": 10686
    },
    {
      "epoch": 0.21376,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.009877268473307292,
      "learning_rate": 0.0001,
      "loss": 4.4708,
      "loss/crossentropy": 2.1850993633270264,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22541093826293945,
      "step": 10688
    },
    {
      "epoch": 0.2138,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0090576171875,
      "learning_rate": 0.0001,
      "loss": 4.3991,
      "loss/crossentropy": 1.9756001830101013,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23400776088237762,
      "step": 10690
    },
    {
      "epoch": 0.21384,
      "grad_norm": 2.125,
      "grad_norm_var": 0.010652669270833333,
      "learning_rate": 0.0001,
      "loss": 4.0283,
      "loss/crossentropy": 1.7454423904418945,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18263398110866547,
      "step": 10692
    },
    {
      "epoch": 0.21388,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010066731770833334,
      "learning_rate": 0.0001,
      "loss": 3.9364,
      "loss/crossentropy": 1.5824024081230164,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18881092965602875,
      "step": 10694
    },
    {
      "epoch": 0.21392,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.009383138020833333,
      "learning_rate": 0.0001,
      "loss": 4.3587,
      "loss/crossentropy": 2.1171644926071167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2173115462064743,
      "step": 10696
    },
    {
      "epoch": 0.21396,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.008698527018229167,
      "learning_rate": 0.0001,
      "loss": 4.2102,
      "loss/crossentropy": 1.9327389001846313,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22025491297245026,
      "step": 10698
    },
    {
      "epoch": 0.214,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.009952799479166666,
      "learning_rate": 0.0001,
      "loss": 3.9622,
      "loss/crossentropy": 1.9806901216506958,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2089216560125351,
      "step": 10700
    },
    {
      "epoch": 0.21404,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.007503000895182291,
      "learning_rate": 0.0001,
      "loss": 4.2503,
      "loss/crossentropy": 2.216805338859558,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21268506348133087,
      "step": 10702
    },
    {
      "epoch": 0.21408,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0055084228515625,
      "learning_rate": 0.0001,
      "loss": 4.5121,
      "loss/crossentropy": 2.3998383283615112,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26092807948589325,
      "step": 10704
    },
    {
      "epoch": 0.21412,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.007869466145833334,
      "learning_rate": 0.0001,
      "loss": 4.4273,
      "loss/crossentropy": 2.0581844449043274,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2316882163286209,
      "step": 10706
    },
    {
      "epoch": 0.21416,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.006258138020833333,
      "learning_rate": 0.0001,
      "loss": 4.2684,
      "loss/crossentropy": 2.3091371059417725,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23538943380117416,
      "step": 10708
    },
    {
      "epoch": 0.2142,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0067291259765625,
      "learning_rate": 0.0001,
      "loss": 3.9567,
      "loss/crossentropy": 1.7134324312210083,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1941806674003601,
      "step": 10710
    },
    {
      "epoch": 0.21424,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0060699462890625,
      "learning_rate": 0.0001,
      "loss": 4.3165,
      "loss/crossentropy": 2.2040648460388184,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20669078081846237,
      "step": 10712
    },
    {
      "epoch": 0.21428,
      "grad_norm": 2.25,
      "grad_norm_var": 0.020612589518229165,
      "learning_rate": 0.0001,
      "loss": 4.3207,
      "loss/crossentropy": 2.3040376901626587,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20685256272554398,
      "step": 10714
    },
    {
      "epoch": 0.21432,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.020318349202473957,
      "learning_rate": 0.0001,
      "loss": 4.2674,
      "loss/crossentropy": 2.189309239387512,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21499691903591156,
      "step": 10716
    },
    {
      "epoch": 0.21436,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.020216623942057293,
      "learning_rate": 0.0001,
      "loss": 4.4144,
      "loss/crossentropy": 2.1955957412719727,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20932748913764954,
      "step": 10718
    },
    {
      "epoch": 0.2144,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.020271809895833333,
      "learning_rate": 0.0001,
      "loss": 4.2135,
      "loss/crossentropy": 1.7800896763801575,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1939619928598404,
      "step": 10720
    },
    {
      "epoch": 0.21444,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.020335896809895834,
      "learning_rate": 0.0001,
      "loss": 4.2757,
      "loss/crossentropy": 1.8974847197532654,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19434937089681625,
      "step": 10722
    },
    {
      "epoch": 0.21448,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.021455891927083335,
      "learning_rate": 0.0001,
      "loss": 4.2688,
      "loss/crossentropy": 2.1631242632865906,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21765826642513275,
      "step": 10724
    },
    {
      "epoch": 0.21452,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0203521728515625,
      "learning_rate": 0.0001,
      "loss": 4.6421,
      "loss/crossentropy": 2.241260290145874,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2334459200501442,
      "step": 10726
    },
    {
      "epoch": 0.21456,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0226470947265625,
      "learning_rate": 0.0001,
      "loss": 3.9478,
      "loss/crossentropy": 1.9063156247138977,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22118454426527023,
      "step": 10728
    },
    {
      "epoch": 0.2146,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.011310831705729166,
      "learning_rate": 0.0001,
      "loss": 4.0446,
      "loss/crossentropy": 1.8109349012374878,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18581955134868622,
      "step": 10730
    },
    {
      "epoch": 0.21464,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.009749348958333333,
      "learning_rate": 0.0001,
      "loss": 4.3262,
      "loss/crossentropy": 2.1113163232803345,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21339301764965057,
      "step": 10732
    },
    {
      "epoch": 0.21468,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0108551025390625,
      "learning_rate": 0.0001,
      "loss": 4.5308,
      "loss/crossentropy": 2.1731717586517334,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22267260402441025,
      "step": 10734
    },
    {
      "epoch": 0.21472,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.010087076822916667,
      "learning_rate": 0.0001,
      "loss": 4.2363,
      "loss/crossentropy": 2.322808027267456,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21375543624162674,
      "step": 10736
    },
    {
      "epoch": 0.21476,
      "grad_norm": 2.0,
      "grad_norm_var": 0.010358683268229167,
      "learning_rate": 0.0001,
      "loss": 4.109,
      "loss/crossentropy": 1.8200489282608032,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19796901941299438,
      "step": 10738
    },
    {
      "epoch": 0.2148,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009455362955729166,
      "learning_rate": 0.0001,
      "loss": 4.0918,
      "loss/crossentropy": 2.353461265563965,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22755976021289825,
      "step": 10740
    },
    {
      "epoch": 0.21484,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0071451822916666664,
      "learning_rate": 0.0001,
      "loss": 4.2925,
      "loss/crossentropy": 2.3200663328170776,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23450587689876556,
      "step": 10742
    },
    {
      "epoch": 0.21488,
      "grad_norm": 2.0,
      "grad_norm_var": 0.006403605143229167,
      "learning_rate": 0.0001,
      "loss": 4.187,
      "loss/crossentropy": 1.7293912768363953,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17938270419836044,
      "step": 10744
    },
    {
      "epoch": 0.21492,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.008656565348307292,
      "learning_rate": 0.0001,
      "loss": 4.3213,
      "loss/crossentropy": 1.9759944081306458,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20631389319896698,
      "step": 10746
    },
    {
      "epoch": 0.21496,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.006870269775390625,
      "learning_rate": 0.0001,
      "loss": 4.3658,
      "loss/crossentropy": 2.2232764959335327,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21919700503349304,
      "step": 10748
    },
    {
      "epoch": 0.215,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.004898834228515625,
      "learning_rate": 0.0001,
      "loss": 4.2466,
      "loss/crossentropy": 2.0827722549438477,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20569747686386108,
      "step": 10750
    },
    {
      "epoch": 0.21504,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.005509440104166667,
      "learning_rate": 0.0001,
      "loss": 4.2373,
      "loss/crossentropy": 2.0712032318115234,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20504066348075867,
      "step": 10752
    },
    {
      "epoch": 0.21508,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.007963053385416667,
      "learning_rate": 0.0001,
      "loss": 4.3466,
      "loss/crossentropy": 2.2717082500457764,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22957370430231094,
      "step": 10754
    },
    {
      "epoch": 0.21512,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.0133056640625,
      "learning_rate": 0.0001,
      "loss": 4.2821,
      "loss/crossentropy": 2.263810157775879,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2291572391986847,
      "step": 10756
    },
    {
      "epoch": 0.21516,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.013724517822265626,
      "learning_rate": 0.0001,
      "loss": 4.2157,
      "loss/crossentropy": 2.103231191635132,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21839886158704758,
      "step": 10758
    },
    {
      "epoch": 0.2152,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.013962554931640624,
      "learning_rate": 0.0001,
      "loss": 4.4772,
      "loss/crossentropy": 2.1272148489952087,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2280830293893814,
      "step": 10760
    },
    {
      "epoch": 0.21524,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.012962849934895833,
      "learning_rate": 0.0001,
      "loss": 4.1192,
      "loss/crossentropy": 2.133938789367676,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2162996605038643,
      "step": 10762
    },
    {
      "epoch": 0.21528,
      "grad_norm": 2.828125,
      "grad_norm_var": 0.04528401692708333,
      "learning_rate": 0.0001,
      "loss": 4.5351,
      "loss/crossentropy": 2.305683732032776,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23797836154699326,
      "step": 10764
    },
    {
      "epoch": 0.21532,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.04370829264322917,
      "learning_rate": 0.0001,
      "loss": 3.9886,
      "loss/crossentropy": 1.7248413562774658,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17226764559745789,
      "step": 10766
    },
    {
      "epoch": 0.21536,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.041715240478515624,
      "learning_rate": 0.0001,
      "loss": 4.6422,
      "loss/crossentropy": 2.275644540786743,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2433001920580864,
      "step": 10768
    },
    {
      "epoch": 0.2154,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.043338775634765625,
      "learning_rate": 0.0001,
      "loss": 4.2836,
      "loss/crossentropy": 1.908457100391388,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19758973270654678,
      "step": 10770
    },
    {
      "epoch": 0.21544,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.07423909505208333,
      "learning_rate": 0.0001,
      "loss": 4.2317,
      "loss/crossentropy": 1.9401238560676575,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22177566587924957,
      "step": 10772
    },
    {
      "epoch": 0.21548,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.07476170857747395,
      "learning_rate": 0.0001,
      "loss": 4.295,
      "loss/crossentropy": 2.247257351875305,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21560464799404144,
      "step": 10774
    },
    {
      "epoch": 0.21552,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.07644424438476563,
      "learning_rate": 0.0001,
      "loss": 4.1482,
      "loss/crossentropy": 1.9191248416900635,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.214834064245224,
      "step": 10776
    },
    {
      "epoch": 0.21556,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0785888671875,
      "learning_rate": 0.0001,
      "loss": 4.0644,
      "loss/crossentropy": 1.8963102102279663,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20552606880664825,
      "step": 10778
    },
    {
      "epoch": 0.2156,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.047078450520833336,
      "learning_rate": 0.0001,
      "loss": 4.265,
      "loss/crossentropy": 2.0517951250076294,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1963544860482216,
      "step": 10780
    },
    {
      "epoch": 0.21564,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.046727498372395836,
      "learning_rate": 0.0001,
      "loss": 4.3571,
      "loss/crossentropy": 2.1484888792037964,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23963741213083267,
      "step": 10782
    },
    {
      "epoch": 0.21568,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.06716206868489584,
      "learning_rate": 0.0001,
      "loss": 4.6827,
      "loss/crossentropy": 2.012593388557434,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2115195021033287,
      "step": 10784
    },
    {
      "epoch": 0.21572,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.06651102701822917,
      "learning_rate": 0.0001,
      "loss": 4.018,
      "loss/crossentropy": 2.139856696128845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23609659075737,
      "step": 10786
    },
    {
      "epoch": 0.21576,
      "grad_norm": 2.25,
      "grad_norm_var": 0.03216120402018229,
      "learning_rate": 0.0001,
      "loss": 4.5007,
      "loss/crossentropy": 2.0139951705932617,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22799161076545715,
      "step": 10788
    },
    {
      "epoch": 0.2158,
      "grad_norm": 2.125,
      "grad_norm_var": 0.035676829020182294,
      "learning_rate": 0.0001,
      "loss": 4.227,
      "loss/crossentropy": 1.8325074315071106,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18156912177801132,
      "step": 10790
    },
    {
      "epoch": 0.21584,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.04155654907226562,
      "learning_rate": 0.0001,
      "loss": 3.8042,
      "loss/crossentropy": 1.7531892657279968,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18498936295509338,
      "step": 10792
    },
    {
      "epoch": 0.21588,
      "grad_norm": 2.25,
      "grad_norm_var": 0.03681233723958333,
      "learning_rate": 0.0001,
      "loss": 4.3955,
      "loss/crossentropy": 2.09742671251297,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21543975174427032,
      "step": 10794
    },
    {
      "epoch": 0.21592,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.03532613118489583,
      "learning_rate": 0.0001,
      "loss": 4.4752,
      "loss/crossentropy": 2.4303117990493774,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24365779757499695,
      "step": 10796
    },
    {
      "epoch": 0.21596,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0371002197265625,
      "learning_rate": 0.0001,
      "loss": 4.1973,
      "loss/crossentropy": 2.007950007915497,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20470578223466873,
      "step": 10798
    },
    {
      "epoch": 0.216,
      "grad_norm": 2.5,
      "grad_norm_var": 0.026949055989583335,
      "learning_rate": 0.0001,
      "loss": 3.9975,
      "loss/crossentropy": 1.895507276058197,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2005787119269371,
      "step": 10800
    },
    {
      "epoch": 0.21604,
      "grad_norm": 2.25,
      "grad_norm_var": 0.026691691080729166,
      "learning_rate": 0.0001,
      "loss": 4.2543,
      "loss/crossentropy": 2.174374043941498,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2282470539212227,
      "step": 10802
    },
    {
      "epoch": 0.21608,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.02789484659830729,
      "learning_rate": 0.0001,
      "loss": 4.0642,
      "loss/crossentropy": 1.8878389596939087,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19730794429779053,
      "step": 10804
    },
    {
      "epoch": 0.21612,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.023884073893229166,
      "learning_rate": 0.0001,
      "loss": 4.1733,
      "loss/crossentropy": 1.9874022006988525,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21521702408790588,
      "step": 10806
    },
    {
      "epoch": 0.21616,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.01962458292643229,
      "learning_rate": 0.0001,
      "loss": 4.0783,
      "loss/crossentropy": 2.0173474550247192,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21552105993032455,
      "step": 10808
    },
    {
      "epoch": 0.2162,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.018344879150390625,
      "learning_rate": 0.0001,
      "loss": 4.0583,
      "loss/crossentropy": 2.175139367580414,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20279064774513245,
      "step": 10810
    },
    {
      "epoch": 0.21624,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.022849273681640626,
      "learning_rate": 0.0001,
      "loss": 4.5125,
      "loss/crossentropy": 2.6407864093780518,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2939087897539139,
      "step": 10812
    },
    {
      "epoch": 0.21628,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.024930826822916665,
      "learning_rate": 0.0001,
      "loss": 4.0869,
      "loss/crossentropy": 1.947974681854248,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2102196291089058,
      "step": 10814
    },
    {
      "epoch": 0.21632,
      "grad_norm": 9.875,
      "grad_norm_var": 3.7874745686848956,
      "learning_rate": 0.0001,
      "loss": 4.3404,
      "loss/crossentropy": 1.9040276408195496,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20579400658607483,
      "step": 10816
    },
    {
      "epoch": 0.21636,
      "grad_norm": 2.21875,
      "grad_norm_var": 3.7853190104166665,
      "learning_rate": 0.0001,
      "loss": 3.7119,
      "loss/crossentropy": 1.7941421270370483,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19792088121175766,
      "step": 10818
    },
    {
      "epoch": 0.2164,
      "grad_norm": 1.953125,
      "grad_norm_var": 3.7839088439941406,
      "learning_rate": 0.0001,
      "loss": 4.2573,
      "loss/crossentropy": 2.2164441347122192,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2903618812561035,
      "step": 10820
    },
    {
      "epoch": 0.21644,
      "grad_norm": 2.109375,
      "grad_norm_var": 3.77445068359375,
      "learning_rate": 0.0001,
      "loss": 4.2537,
      "loss/crossentropy": 1.9888933897018433,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20800678431987762,
      "step": 10822
    },
    {
      "epoch": 0.21648,
      "grad_norm": 1.875,
      "grad_norm_var": 3.79569091796875,
      "learning_rate": 0.0001,
      "loss": 3.9218,
      "loss/crossentropy": 1.9026559591293335,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18179579824209213,
      "step": 10824
    },
    {
      "epoch": 0.21652,
      "grad_norm": 2.078125,
      "grad_norm_var": 3.7974202473958334,
      "learning_rate": 0.0001,
      "loss": 4.1267,
      "loss/crossentropy": 1.9020920991897583,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20755057036876678,
      "step": 10826
    },
    {
      "epoch": 0.21656,
      "grad_norm": 2.03125,
      "grad_norm_var": 3.8177286783854165,
      "learning_rate": 0.0001,
      "loss": 4.4689,
      "loss/crossentropy": 2.3464537858963013,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23209182918071747,
      "step": 10828
    },
    {
      "epoch": 0.2166,
      "grad_norm": 2.3125,
      "grad_norm_var": 3.7860877990722654,
      "learning_rate": 0.0001,
      "loss": 4.4486,
      "loss/crossentropy": 2.305312991142273,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2548810988664627,
      "step": 10830
    },
    {
      "epoch": 0.21664,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.026775868733723958,
      "learning_rate": 0.0001,
      "loss": 3.9818,
      "loss/crossentropy": 1.4548576474189758,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1718049794435501,
      "step": 10832
    },
    {
      "epoch": 0.21668,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.02504247029622396,
      "learning_rate": 0.0001,
      "loss": 4.4909,
      "loss/crossentropy": 2.2954181432724,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22977370768785477,
      "step": 10834
    },
    {
      "epoch": 0.21672,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.022930653889973958,
      "learning_rate": 0.0001,
      "loss": 4.0433,
      "loss/crossentropy": 2.1974023580551147,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21474600583314896,
      "step": 10836
    },
    {
      "epoch": 0.21676,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.02304865519205729,
      "learning_rate": 0.0001,
      "loss": 4.5905,
      "loss/crossentropy": 2.1211158633232117,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19471213221549988,
      "step": 10838
    },
    {
      "epoch": 0.2168,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0204010009765625,
      "learning_rate": 0.0001,
      "loss": 4.0868,
      "loss/crossentropy": 1.9942908883094788,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1899253949522972,
      "step": 10840
    },
    {
      "epoch": 0.21684,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.023656209309895832,
      "learning_rate": 0.0001,
      "loss": 4.0787,
      "loss/crossentropy": 1.9848283529281616,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1777423620223999,
      "step": 10842
    },
    {
      "epoch": 0.21688,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.02545140584309896,
      "learning_rate": 0.0001,
      "loss": 4.1365,
      "loss/crossentropy": 1.84994775056839,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19193057715892792,
      "step": 10844
    },
    {
      "epoch": 0.21692,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.02163670857747396,
      "learning_rate": 0.0001,
      "loss": 4.2869,
      "loss/crossentropy": 1.8300130367279053,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19747256487607956,
      "step": 10846
    },
    {
      "epoch": 0.21696,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.010676829020182292,
      "learning_rate": 0.0001,
      "loss": 4.1624,
      "loss/crossentropy": 1.9316250681877136,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22893256694078445,
      "step": 10848
    },
    {
      "epoch": 0.217,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.009085845947265626,
      "learning_rate": 0.0001,
      "loss": 4.0294,
      "loss/crossentropy": 1.9862067103385925,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22470611333847046,
      "step": 10850
    },
    {
      "epoch": 0.21704,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.009372711181640625,
      "learning_rate": 0.0001,
      "loss": 4.3979,
      "loss/crossentropy": 2.196234107017517,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22592387348413467,
      "step": 10852
    },
    {
      "epoch": 0.21708,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.2918291727701823,
      "learning_rate": 0.0001,
      "loss": 4.5516,
      "loss/crossentropy": 2.3097496032714844,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29694322496652603,
      "step": 10854
    },
    {
      "epoch": 0.21712,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.2865191141764323,
      "learning_rate": 0.0001,
      "loss": 3.8687,
      "loss/crossentropy": 1.883777916431427,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21657781302928925,
      "step": 10856
    },
    {
      "epoch": 0.21716,
      "grad_norm": 2.25,
      "grad_norm_var": 0.2916338602701823,
      "learning_rate": 0.0001,
      "loss": 4.1915,
      "loss/crossentropy": 2.0483964681625366,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22432015091180801,
      "step": 10858
    },
    {
      "epoch": 0.2172,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.2870839436848958,
      "learning_rate": 0.0001,
      "loss": 4.1503,
      "loss/crossentropy": 2.2700769901275635,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21760191768407822,
      "step": 10860
    },
    {
      "epoch": 0.21724,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.2878214518229167,
      "learning_rate": 0.0001,
      "loss": 4.574,
      "loss/crossentropy": 2.3758704662323,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21792180836200714,
      "step": 10862
    },
    {
      "epoch": 0.21728,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.2934641520182292,
      "learning_rate": 0.0001,
      "loss": 4.2895,
      "loss/crossentropy": 2.1972473859786987,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22460343688726425,
      "step": 10864
    },
    {
      "epoch": 0.21732,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.29321187337239585,
      "learning_rate": 0.0001,
      "loss": 4.3693,
      "loss/crossentropy": 1.9811919331550598,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23008134216070175,
      "step": 10866
    },
    {
      "epoch": 0.21736,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.2992327372233073,
      "learning_rate": 0.0001,
      "loss": 4.2237,
      "loss/crossentropy": 2.032214403152466,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1958027482032776,
      "step": 10868
    },
    {
      "epoch": 0.2174,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.035162099202473956,
      "learning_rate": 0.0001,
      "loss": 4.0956,
      "loss/crossentropy": 2.08488667011261,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21948565542697906,
      "step": 10870
    },
    {
      "epoch": 0.21744,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.03401260375976563,
      "learning_rate": 0.0001,
      "loss": 4.2923,
      "loss/crossentropy": 2.304922103881836,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1977907121181488,
      "step": 10872
    },
    {
      "epoch": 0.21748,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.005582427978515625,
      "learning_rate": 0.0001,
      "loss": 4.1762,
      "loss/crossentropy": 1.991280436515808,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2196018323302269,
      "step": 10874
    },
    {
      "epoch": 0.21752,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.004965972900390625,
      "learning_rate": 0.0001,
      "loss": 4.4116,
      "loss/crossentropy": 1.955183207988739,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21059879660606384,
      "step": 10876
    },
    {
      "epoch": 0.21756,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0045562744140625,
      "learning_rate": 0.0001,
      "loss": 4.2105,
      "loss/crossentropy": 2.1443604230880737,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23160522431135178,
      "step": 10878
    },
    {
      "epoch": 0.2176,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.004889933268229166,
      "learning_rate": 0.0001,
      "loss": 4.2374,
      "loss/crossentropy": 2.0859211683273315,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2296902909874916,
      "step": 10880
    },
    {
      "epoch": 0.21764,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0059234619140625,
      "learning_rate": 0.0001,
      "loss": 4.3303,
      "loss/crossentropy": 2.220987915992737,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22863281518220901,
      "step": 10882
    },
    {
      "epoch": 0.21768,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.005625152587890625,
      "learning_rate": 0.0001,
      "loss": 4.2195,
      "loss/crossentropy": 2.1172574758529663,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.197882778942585,
      "step": 10884
    },
    {
      "epoch": 0.21772,
      "grad_norm": 2.25,
      "grad_norm_var": 0.02047704060872396,
      "learning_rate": 0.0001,
      "loss": 4.3907,
      "loss/crossentropy": 1.7861940264701843,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1984492540359497,
      "step": 10886
    },
    {
      "epoch": 0.21776,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0218414306640625,
      "learning_rate": 0.0001,
      "loss": 4.0445,
      "loss/crossentropy": 2.1119033098220825,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20427606999874115,
      "step": 10888
    },
    {
      "epoch": 0.2178,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.02235692342122396,
      "learning_rate": 0.0001,
      "loss": 3.923,
      "loss/crossentropy": 1.7849717140197754,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20817570388317108,
      "step": 10890
    },
    {
      "epoch": 0.21784,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.025131988525390624,
      "learning_rate": 0.0001,
      "loss": 4.2015,
      "loss/crossentropy": 2.0543535351753235,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20579595863819122,
      "step": 10892
    },
    {
      "epoch": 0.21788,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.02643000284830729,
      "learning_rate": 0.0001,
      "loss": 4.1527,
      "loss/crossentropy": 2.085465431213379,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23541826009750366,
      "step": 10894
    },
    {
      "epoch": 0.21792,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.03343073527018229,
      "learning_rate": 0.0001,
      "loss": 4.3288,
      "loss/crossentropy": 1.9521069526672363,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21404746174812317,
      "step": 10896
    },
    {
      "epoch": 0.21796,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.03172378540039063,
      "learning_rate": 0.0001,
      "loss": 4.039,
      "loss/crossentropy": 1.8710272908210754,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20500759035348892,
      "step": 10898
    },
    {
      "epoch": 0.218,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.032083892822265626,
      "learning_rate": 0.0001,
      "loss": 4.0695,
      "loss/crossentropy": 2.275243639945984,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24073244631290436,
      "step": 10900
    },
    {
      "epoch": 0.21804,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0181060791015625,
      "learning_rate": 0.0001,
      "loss": 4.3508,
      "loss/crossentropy": 2.2093106508255005,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23254899680614471,
      "step": 10902
    },
    {
      "epoch": 0.21808,
      "grad_norm": 2.0,
      "grad_norm_var": 0.01702855428059896,
      "learning_rate": 0.0001,
      "loss": 4.2805,
      "loss/crossentropy": 2.0315810441970825,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20614907145500183,
      "step": 10904
    },
    {
      "epoch": 0.21812,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.016478474934895834,
      "learning_rate": 0.0001,
      "loss": 4.3461,
      "loss/crossentropy": 2.0397735834121704,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2093086987733841,
      "step": 10906
    },
    {
      "epoch": 0.21816,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.013508097330729166,
      "learning_rate": 0.0001,
      "loss": 4.0286,
      "loss/crossentropy": 1.9616519808769226,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19433944672346115,
      "step": 10908
    },
    {
      "epoch": 0.2182,
      "grad_norm": 2.25,
      "grad_norm_var": 0.01749445597330729,
      "learning_rate": 0.0001,
      "loss": 4.2157,
      "loss/crossentropy": 2.1907248497009277,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23218485713005066,
      "step": 10910
    },
    {
      "epoch": 0.21824,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009981282552083333,
      "learning_rate": 0.0001,
      "loss": 4.163,
      "loss/crossentropy": 2.2785152196884155,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21416624635457993,
      "step": 10912
    },
    {
      "epoch": 0.21828,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.00955810546875,
      "learning_rate": 0.0001,
      "loss": 4.2528,
      "loss/crossentropy": 2.151498794555664,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23632780462503433,
      "step": 10914
    },
    {
      "epoch": 0.21832,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008829752604166666,
      "learning_rate": 0.0001,
      "loss": 4.3937,
      "loss/crossentropy": 2.816411852836609,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2243700549006462,
      "step": 10916
    },
    {
      "epoch": 0.21836,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.014662424723307291,
      "learning_rate": 0.0001,
      "loss": 4.4091,
      "loss/crossentropy": 2.2203346490859985,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22302204370498657,
      "step": 10918
    },
    {
      "epoch": 0.2184,
      "grad_norm": 2.0,
      "grad_norm_var": 0.014662424723307291,
      "learning_rate": 0.0001,
      "loss": 4.3534,
      "loss/crossentropy": 2.0416316390037537,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20798437297344208,
      "step": 10920
    },
    {
      "epoch": 0.21844,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.014426422119140626,
      "learning_rate": 0.0001,
      "loss": 4.1014,
      "loss/crossentropy": 2.2251007556915283,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2149331197142601,
      "step": 10922
    },
    {
      "epoch": 0.21848,
      "grad_norm": 2.0,
      "grad_norm_var": 0.014631907145182291,
      "learning_rate": 0.0001,
      "loss": 4.2346,
      "loss/crossentropy": 1.976640522480011,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21704353392124176,
      "step": 10924
    },
    {
      "epoch": 0.21852,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.011006418863932292,
      "learning_rate": 0.0001,
      "loss": 4.3284,
      "loss/crossentropy": 2.041864037513733,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19621634483337402,
      "step": 10926
    },
    {
      "epoch": 0.21856,
      "grad_norm": 2.125,
      "grad_norm_var": 0.009696451822916667,
      "learning_rate": 0.0001,
      "loss": 4.241,
      "loss/crossentropy": 1.7188060879707336,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1992729976773262,
      "step": 10928
    },
    {
      "epoch": 0.2186,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.010512034098307291,
      "learning_rate": 0.0001,
      "loss": 4.0333,
      "loss/crossentropy": 1.664458692073822,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20099520683288574,
      "step": 10930
    },
    {
      "epoch": 0.21864,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011161041259765626,
      "learning_rate": 0.0001,
      "loss": 4.1952,
      "loss/crossentropy": 2.199326276779175,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23000852018594742,
      "step": 10932
    },
    {
      "epoch": 0.21868,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.005694325764973958,
      "learning_rate": 0.0001,
      "loss": 4.1399,
      "loss/crossentropy": 1.980049967765808,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2211422473192215,
      "step": 10934
    },
    {
      "epoch": 0.21872,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.005411529541015625,
      "learning_rate": 0.0001,
      "loss": 4.0876,
      "loss/crossentropy": 1.7884072661399841,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.203270323574543,
      "step": 10936
    },
    {
      "epoch": 0.21876,
      "grad_norm": 2.0,
      "grad_norm_var": 0.005527496337890625,
      "learning_rate": 0.0001,
      "loss": 4.1923,
      "loss/crossentropy": 2.1758522987365723,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21960537880659103,
      "step": 10938
    },
    {
      "epoch": 0.2188,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.004416656494140625,
      "learning_rate": 0.0001,
      "loss": 4.1708,
      "loss/crossentropy": 2.255508303642273,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2171120047569275,
      "step": 10940
    },
    {
      "epoch": 0.21884,
      "grad_norm": 2.0,
      "grad_norm_var": 0.004308827718098958,
      "learning_rate": 0.0001,
      "loss": 4.1294,
      "loss/crossentropy": 1.9758012890815735,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19945884495973587,
      "step": 10942
    },
    {
      "epoch": 0.21888,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0055582682291666664,
      "learning_rate": 0.0001,
      "loss": 4.2122,
      "loss/crossentropy": 2.2980172634124756,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20864219218492508,
      "step": 10944
    },
    {
      "epoch": 0.21892,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.006666819254557292,
      "learning_rate": 0.0001,
      "loss": 3.919,
      "loss/crossentropy": 1.5307916402816772,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1799665167927742,
      "step": 10946
    },
    {
      "epoch": 0.21896,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.006648508707682291,
      "learning_rate": 0.0001,
      "loss": 4.3796,
      "loss/crossentropy": 2.3332602977752686,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22964681684970856,
      "step": 10948
    },
    {
      "epoch": 0.219,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.007380930582682291,
      "learning_rate": 0.0001,
      "loss": 4.1207,
      "loss/crossentropy": 2.2509007453918457,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22141733020544052,
      "step": 10950
    },
    {
      "epoch": 0.21904,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.008949534098307291,
      "learning_rate": 0.0001,
      "loss": 4.4879,
      "loss/crossentropy": 2.3677018880844116,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25676336884498596,
      "step": 10952
    },
    {
      "epoch": 0.21908,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.009934234619140624,
      "learning_rate": 0.0001,
      "loss": 4.4283,
      "loss/crossentropy": 2.2807843685150146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21486903727054596,
      "step": 10954
    },
    {
      "epoch": 0.21912,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.009232330322265624,
      "learning_rate": 0.0001,
      "loss": 4.3126,
      "loss/crossentropy": 2.3883347511291504,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21362561732530594,
      "step": 10956
    },
    {
      "epoch": 0.21916,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009405263264973958,
      "learning_rate": 0.0001,
      "loss": 4.0692,
      "loss/crossentropy": 1.8598107695579529,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19806977361440659,
      "step": 10958
    },
    {
      "epoch": 0.2192,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.007738240559895833,
      "learning_rate": 0.0001,
      "loss": 4.3259,
      "loss/crossentropy": 2.1962249875068665,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20640095323324203,
      "step": 10960
    },
    {
      "epoch": 0.21924,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.007958984375,
      "learning_rate": 0.0001,
      "loss": 4.1774,
      "loss/crossentropy": 2.061887502670288,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22463876008987427,
      "step": 10962
    },
    {
      "epoch": 0.21928,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.010237375895182291,
      "learning_rate": 0.0001,
      "loss": 3.9749,
      "loss/crossentropy": 1.9112628102302551,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19327937066555023,
      "step": 10964
    },
    {
      "epoch": 0.21932,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.010640207926432292,
      "learning_rate": 0.0001,
      "loss": 4.3078,
      "loss/crossentropy": 2.1655001640319824,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22729264944791794,
      "step": 10966
    },
    {
      "epoch": 0.21936,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.010400136311848959,
      "learning_rate": 0.0001,
      "loss": 4.1641,
      "loss/crossentropy": 2.199634552001953,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23565081506967545,
      "step": 10968
    },
    {
      "epoch": 0.2194,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.010155995686848959,
      "learning_rate": 0.0001,
      "loss": 4.4298,
      "loss/crossentropy": 2.4142041206359863,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23020224273204803,
      "step": 10970
    },
    {
      "epoch": 0.21944,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.011193593343098959,
      "learning_rate": 0.0001,
      "loss": 4.3856,
      "loss/crossentropy": 2.2480571269989014,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.224863201379776,
      "step": 10972
    },
    {
      "epoch": 0.21948,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.012238566080729167,
      "learning_rate": 0.0001,
      "loss": 4.099,
      "loss/crossentropy": 2.1786953806877136,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21182173490524292,
      "step": 10974
    },
    {
      "epoch": 0.21952,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.013337961832682292,
      "learning_rate": 0.0001,
      "loss": 4.2826,
      "loss/crossentropy": 2.3767744302749634,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22259585559368134,
      "step": 10976
    },
    {
      "epoch": 0.21956,
      "grad_norm": 2.25,
      "grad_norm_var": 0.010794830322265626,
      "learning_rate": 0.0001,
      "loss": 4.405,
      "loss/crossentropy": 1.9838140606880188,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21830464899539948,
      "step": 10978
    },
    {
      "epoch": 0.2196,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0092193603515625,
      "learning_rate": 0.0001,
      "loss": 4.3811,
      "loss/crossentropy": 2.0167009234428406,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21239794790744781,
      "step": 10980
    },
    {
      "epoch": 0.21964,
      "grad_norm": 2.5,
      "grad_norm_var": 0.019269816080729165,
      "learning_rate": 0.0001,
      "loss": 4.3117,
      "loss/crossentropy": 2.001616358757019,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21801268309354782,
      "step": 10982
    },
    {
      "epoch": 0.21968,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.019432576497395833,
      "learning_rate": 0.0001,
      "loss": 4.6266,
      "loss/crossentropy": 2.259532332420349,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20131191611289978,
      "step": 10984
    },
    {
      "epoch": 0.21972,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0215728759765625,
      "learning_rate": 0.0001,
      "loss": 4.1159,
      "loss/crossentropy": 1.9223415851593018,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21203956007957458,
      "step": 10986
    },
    {
      "epoch": 0.21976,
      "grad_norm": 2.0,
      "grad_norm_var": 0.020295206705729166,
      "learning_rate": 0.0001,
      "loss": 4.0177,
      "loss/crossentropy": 2.3779542446136475,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23436614871025085,
      "step": 10988
    },
    {
      "epoch": 0.2198,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.02195002237955729,
      "learning_rate": 0.0001,
      "loss": 4.0412,
      "loss/crossentropy": 1.8552000522613525,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20212603360414505,
      "step": 10990
    },
    {
      "epoch": 0.21984,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0206451416015625,
      "learning_rate": 0.0001,
      "loss": 4.2264,
      "loss/crossentropy": 1.79928320646286,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19676420837640762,
      "step": 10992
    },
    {
      "epoch": 0.21988,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.018619791666666666,
      "learning_rate": 0.0001,
      "loss": 4.2426,
      "loss/crossentropy": 1.886910319328308,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21270643919706345,
      "step": 10994
    },
    {
      "epoch": 0.21992,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.019066365559895833,
      "learning_rate": 0.0001,
      "loss": 4.0775,
      "loss/crossentropy": 1.5682110786437988,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19159042835235596,
      "step": 10996
    },
    {
      "epoch": 0.21996,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.008104451497395833,
      "learning_rate": 0.0001,
      "loss": 4.2933,
      "loss/crossentropy": 2.3335143327713013,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23225059360265732,
      "step": 10998
    },
    {
      "epoch": 0.22,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.007222493489583333,
      "learning_rate": 0.0001,
      "loss": 4.3146,
      "loss/crossentropy": 2.2234359979629517,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21465980261564255,
      "step": 11000
    },
    {
      "epoch": 0.22004,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.00601806640625,
      "learning_rate": 0.0001,
      "loss": 4.6023,
      "loss/crossentropy": 2.3676271438598633,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2674448639154434,
      "step": 11002
    },
    {
      "epoch": 0.22008,
      "grad_norm": 2.0,
      "grad_norm_var": 0.006180826822916667,
      "learning_rate": 0.0001,
      "loss": 4.1064,
      "loss/crossentropy": 1.9216612577438354,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2036839798092842,
      "step": 11004
    },
    {
      "epoch": 0.22012,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.004369099934895833,
      "learning_rate": 0.0001,
      "loss": 4.0427,
      "loss/crossentropy": 1.6732578873634338,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18962469696998596,
      "step": 11006
    },
    {
      "epoch": 0.22016,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.009761555989583334,
      "learning_rate": 0.0001,
      "loss": 4.2704,
      "loss/crossentropy": 2.0608668327331543,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23521222919225693,
      "step": 11008
    },
    {
      "epoch": 0.2202,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010872395833333333,
      "learning_rate": 0.0001,
      "loss": 4.1808,
      "loss/crossentropy": 1.8038227558135986,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1811496466398239,
      "step": 11010
    },
    {
      "epoch": 0.22024,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.010789998372395833,
      "learning_rate": 0.0001,
      "loss": 4.3407,
      "loss/crossentropy": 1.8687627911567688,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20196181535720825,
      "step": 11012
    },
    {
      "epoch": 0.22028,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.01138916015625,
      "learning_rate": 0.0001,
      "loss": 4.0608,
      "loss/crossentropy": 1.8011687397956848,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18384280055761337,
      "step": 11014
    },
    {
      "epoch": 0.22032,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.012727610270182292,
      "learning_rate": 0.0001,
      "loss": 4.0229,
      "loss/crossentropy": 1.8108493089675903,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19042345136404037,
      "step": 11016
    },
    {
      "epoch": 0.22036,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.012611643473307291,
      "learning_rate": 0.0001,
      "loss": 4.2755,
      "loss/crossentropy": 2.4056872129440308,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22949732840061188,
      "step": 11018
    },
    {
      "epoch": 0.2204,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.011451975504557291,
      "learning_rate": 0.0001,
      "loss": 4.2768,
      "loss/crossentropy": 2.0417627692222595,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21793337166309357,
      "step": 11020
    },
    {
      "epoch": 0.22044,
      "grad_norm": 2.125,
      "grad_norm_var": 0.011549631754557291,
      "learning_rate": 0.0001,
      "loss": 4.2767,
      "loss/crossentropy": 1.8312503099441528,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19038879871368408,
      "step": 11022
    },
    {
      "epoch": 0.22048,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.01941095987955729,
      "learning_rate": 0.0001,
      "loss": 4.1846,
      "loss/crossentropy": 2.0615866780281067,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.224861241877079,
      "step": 11024
    },
    {
      "epoch": 0.22052,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.02702611287434896,
      "learning_rate": 0.0001,
      "loss": 4.0216,
      "loss/crossentropy": 1.8578996062278748,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20426958799362183,
      "step": 11026
    },
    {
      "epoch": 0.22056,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.028364817301432293,
      "learning_rate": 0.0001,
      "loss": 3.9303,
      "loss/crossentropy": 2.01333224773407,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2140725627541542,
      "step": 11028
    },
    {
      "epoch": 0.2206,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0296142578125,
      "learning_rate": 0.0001,
      "loss": 3.8142,
      "loss/crossentropy": 1.9607431292533875,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19605513662099838,
      "step": 11030
    },
    {
      "epoch": 0.22064,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.026712799072265626,
      "learning_rate": 0.0001,
      "loss": 4.1336,
      "loss/crossentropy": 1.8091335892677307,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18755877763032913,
      "step": 11032
    },
    {
      "epoch": 0.22068,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0282958984375,
      "learning_rate": 0.0001,
      "loss": 4.0206,
      "loss/crossentropy": 1.5547168254852295,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18891388177871704,
      "step": 11034
    },
    {
      "epoch": 0.22072,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.029605865478515625,
      "learning_rate": 0.0001,
      "loss": 4.1658,
      "loss/crossentropy": 2.0881760120391846,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21140296012163162,
      "step": 11036
    },
    {
      "epoch": 0.22076,
      "grad_norm": 2.953125,
      "grad_norm_var": 0.07485936482747396,
      "learning_rate": 0.0001,
      "loss": 4.0865,
      "loss/crossentropy": 2.0874351263046265,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2202945575118065,
      "step": 11038
    },
    {
      "epoch": 0.2208,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.06611302693684896,
      "learning_rate": 0.0001,
      "loss": 4.3998,
      "loss/crossentropy": 2.2349069118499756,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2247784063220024,
      "step": 11040
    },
    {
      "epoch": 0.22084,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.060373687744140626,
      "learning_rate": 0.0001,
      "loss": 4.4679,
      "loss/crossentropy": 2.0404593348503113,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23377884924411774,
      "step": 11042
    },
    {
      "epoch": 0.22088,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.05997314453125,
      "learning_rate": 0.0001,
      "loss": 4.115,
      "loss/crossentropy": 2.418761968612671,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2288268655538559,
      "step": 11044
    },
    {
      "epoch": 0.22092,
      "grad_norm": 2.25,
      "grad_norm_var": 0.05608495076497396,
      "learning_rate": 0.0001,
      "loss": 4.514,
      "loss/crossentropy": 2.3263691663742065,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23381420969963074,
      "step": 11046
    },
    {
      "epoch": 0.22096,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.058166249593098955,
      "learning_rate": 0.0001,
      "loss": 4.3659,
      "loss/crossentropy": 2.0020187497138977,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1993579939007759,
      "step": 11048
    },
    {
      "epoch": 0.221,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.05726318359375,
      "learning_rate": 0.0001,
      "loss": 4.1869,
      "loss/crossentropy": 2.3417880535125732,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21232012659311295,
      "step": 11050
    },
    {
      "epoch": 0.22104,
      "grad_norm": 2.375,
      "grad_norm_var": 0.06346817016601562,
      "learning_rate": 0.0001,
      "loss": 4.9882,
      "loss/crossentropy": 2.2952964305877686,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22096505016088486,
      "step": 11052
    },
    {
      "epoch": 0.22108,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.028562164306640624,
      "learning_rate": 0.0001,
      "loss": 4.2431,
      "loss/crossentropy": 1.7963152527809143,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18996400386095047,
      "step": 11054
    },
    {
      "epoch": 0.22112,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.028433990478515626,
      "learning_rate": 0.0001,
      "loss": 4.4276,
      "loss/crossentropy": 2.1328593492507935,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21772069483995438,
      "step": 11056
    },
    {
      "epoch": 0.22116,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.0318267822265625,
      "learning_rate": 0.0001,
      "loss": 4.317,
      "loss/crossentropy": 1.8048993349075317,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17359213531017303,
      "step": 11058
    },
    {
      "epoch": 0.2212,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0309234619140625,
      "learning_rate": 0.0001,
      "loss": 4.0173,
      "loss/crossentropy": 1.9539333581924438,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20828361809253693,
      "step": 11060
    },
    {
      "epoch": 0.22124,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.032291412353515625,
      "learning_rate": 0.0001,
      "loss": 4.0288,
      "loss/crossentropy": 1.8231340050697327,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20730753242969513,
      "step": 11062
    },
    {
      "epoch": 0.22128,
      "grad_norm": 2.25,
      "grad_norm_var": 0.033878326416015625,
      "learning_rate": 0.0001,
      "loss": 4.2756,
      "loss/crossentropy": 1.9315263032913208,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20077265799045563,
      "step": 11064
    },
    {
      "epoch": 0.22132,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.034795888264973956,
      "learning_rate": 0.0001,
      "loss": 4.2137,
      "loss/crossentropy": 1.9278368949890137,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2203744500875473,
      "step": 11066
    },
    {
      "epoch": 0.22136,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.016812896728515624,
      "learning_rate": 0.0001,
      "loss": 4.38,
      "loss/crossentropy": 1.9314138889312744,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2024018093943596,
      "step": 11068
    },
    {
      "epoch": 0.2214,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.019618479410807292,
      "learning_rate": 0.0001,
      "loss": 3.9142,
      "loss/crossentropy": 1.822297751903534,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20530518889427185,
      "step": 11070
    },
    {
      "epoch": 0.22144,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.02112401326497396,
      "learning_rate": 0.0001,
      "loss": 4.1412,
      "loss/crossentropy": 2.0021358132362366,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1963522955775261,
      "step": 11072
    },
    {
      "epoch": 0.22148,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.018293253580729165,
      "learning_rate": 0.0001,
      "loss": 4.5595,
      "loss/crossentropy": 2.3780601024627686,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24999547004699707,
      "step": 11074
    },
    {
      "epoch": 0.22152,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.01727472941080729,
      "learning_rate": 0.0001,
      "loss": 4.2572,
      "loss/crossentropy": 1.8128371238708496,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1940363049507141,
      "step": 11076
    },
    {
      "epoch": 0.22156,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0147857666015625,
      "learning_rate": 0.0001,
      "loss": 4.1388,
      "loss/crossentropy": 1.836561381816864,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19289480894804,
      "step": 11078
    },
    {
      "epoch": 0.2216,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0125640869140625,
      "learning_rate": 0.0001,
      "loss": 3.9891,
      "loss/crossentropy": 2.203549385070801,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23240803182125092,
      "step": 11080
    },
    {
      "epoch": 0.22164,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.0147857666015625,
      "learning_rate": 0.0001,
      "loss": 3.9827,
      "loss/crossentropy": 2.126620829105377,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21564993262290955,
      "step": 11082
    },
    {
      "epoch": 0.22168,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.015672810872395835,
      "learning_rate": 0.0001,
      "loss": 4.6468,
      "loss/crossentropy": 2.5914783477783203,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23293393850326538,
      "step": 11084
    },
    {
      "epoch": 0.22172,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.016752115885416665,
      "learning_rate": 0.0001,
      "loss": 4.0881,
      "loss/crossentropy": 2.097515106201172,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21570277214050293,
      "step": 11086
    },
    {
      "epoch": 0.22176,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.01565526326497396,
      "learning_rate": 0.0001,
      "loss": 4.2469,
      "loss/crossentropy": 1.9408356547355652,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22177009284496307,
      "step": 11088
    },
    {
      "epoch": 0.2218,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.014713287353515625,
      "learning_rate": 0.0001,
      "loss": 4.378,
      "loss/crossentropy": 1.9851223826408386,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20538930594921112,
      "step": 11090
    },
    {
      "epoch": 0.22184,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.014694976806640624,
      "learning_rate": 0.0001,
      "loss": 4.2424,
      "loss/crossentropy": 2.261754631996155,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21294979751110077,
      "step": 11092
    },
    {
      "epoch": 0.22188,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.014924875895182292,
      "learning_rate": 0.0001,
      "loss": 4.1847,
      "loss/crossentropy": 1.978752851486206,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21184594929218292,
      "step": 11094
    },
    {
      "epoch": 0.22192,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.014918772379557292,
      "learning_rate": 0.0001,
      "loss": 4.515,
      "loss/crossentropy": 2.1992926597595215,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2428576499223709,
      "step": 11096
    },
    {
      "epoch": 0.22196,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.010131581624348959,
      "learning_rate": 0.0001,
      "loss": 4.1475,
      "loss/crossentropy": 1.9407767057418823,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25351718813180923,
      "step": 11098
    },
    {
      "epoch": 0.222,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.009993235270182291,
      "learning_rate": 0.0001,
      "loss": 4.2485,
      "loss/crossentropy": 2.2973347902297974,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2211253046989441,
      "step": 11100
    },
    {
      "epoch": 0.22204,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.007380930582682291,
      "learning_rate": 0.0001,
      "loss": 4.466,
      "loss/crossentropy": 2.323284387588501,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23907601833343506,
      "step": 11102
    },
    {
      "epoch": 0.22208,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.00631103515625,
      "learning_rate": 0.0001,
      "loss": 4.3745,
      "loss/crossentropy": 1.9557610750198364,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20206698775291443,
      "step": 11104
    },
    {
      "epoch": 0.22212,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.006083170572916667,
      "learning_rate": 0.0001,
      "loss": 4.1432,
      "loss/crossentropy": 2.130094528198242,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22658853232860565,
      "step": 11106
    },
    {
      "epoch": 0.22216,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.006078084309895833,
      "learning_rate": 0.0001,
      "loss": 4.0207,
      "loss/crossentropy": 1.6461073160171509,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.183306485414505,
      "step": 11108
    },
    {
      "epoch": 0.2222,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.008310699462890625,
      "learning_rate": 0.0001,
      "loss": 4.2901,
      "loss/crossentropy": 1.9869291186332703,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20724371075630188,
      "step": 11110
    },
    {
      "epoch": 0.22224,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.007696278889973958,
      "learning_rate": 0.0001,
      "loss": 4.2361,
      "loss/crossentropy": 2.314267873764038,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22680803388357162,
      "step": 11112
    },
    {
      "epoch": 0.22228,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0104888916015625,
      "learning_rate": 0.0001,
      "loss": 4.37,
      "loss/crossentropy": 2.061935067176819,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1931627094745636,
      "step": 11114
    },
    {
      "epoch": 0.22232,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009601847330729166,
      "learning_rate": 0.0001,
      "loss": 4.1174,
      "loss/crossentropy": 2.157355546951294,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2328970953822136,
      "step": 11116
    },
    {
      "epoch": 0.22236,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.012214914957682291,
      "learning_rate": 0.0001,
      "loss": 4.2092,
      "loss/crossentropy": 1.8968737125396729,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19509851932525635,
      "step": 11118
    },
    {
      "epoch": 0.2224,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.013516998291015625,
      "learning_rate": 0.0001,
      "loss": 4.3433,
      "loss/crossentropy": 2.1147825717926025,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22639968246221542,
      "step": 11120
    },
    {
      "epoch": 0.22244,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.01519775390625,
      "learning_rate": 0.0001,
      "loss": 3.9818,
      "loss/crossentropy": 1.9084222316741943,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16847409307956696,
      "step": 11122
    },
    {
      "epoch": 0.22248,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.015028635660807291,
      "learning_rate": 0.0001,
      "loss": 4.0188,
      "loss/crossentropy": 1.8558747172355652,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2013881430029869,
      "step": 11124
    },
    {
      "epoch": 0.22252,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.015755208333333333,
      "learning_rate": 0.0001,
      "loss": 4.3236,
      "loss/crossentropy": 2.2021052837371826,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2129388153553009,
      "step": 11126
    },
    {
      "epoch": 0.22256,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.019212849934895835,
      "learning_rate": 0.0001,
      "loss": 4.2892,
      "loss/crossentropy": 2.267111897468567,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22087457031011581,
      "step": 11128
    },
    {
      "epoch": 0.2226,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.02021052042643229,
      "learning_rate": 0.0001,
      "loss": 4.3781,
      "loss/crossentropy": 1.860244870185852,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21286892145872116,
      "step": 11130
    },
    {
      "epoch": 0.22264,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.021142578125,
      "learning_rate": 0.0001,
      "loss": 3.7832,
      "loss/crossentropy": 1.8312670588493347,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20548796653747559,
      "step": 11132
    },
    {
      "epoch": 0.22268,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.019087473551432293,
      "learning_rate": 0.0001,
      "loss": 4.1916,
      "loss/crossentropy": 2.2309051752090454,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2244933694601059,
      "step": 11134
    },
    {
      "epoch": 0.22272,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.01761449178059896,
      "learning_rate": 0.0001,
      "loss": 4.1125,
      "loss/crossentropy": 2.1003236770629883,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22222542017698288,
      "step": 11136
    },
    {
      "epoch": 0.22276,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.015965779622395832,
      "learning_rate": 0.0001,
      "loss": 3.9391,
      "loss/crossentropy": 1.95048588514328,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20614230632781982,
      "step": 11138
    },
    {
      "epoch": 0.2228,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.016364542643229167,
      "learning_rate": 0.0001,
      "loss": 4.1608,
      "loss/crossentropy": 2.2856240272521973,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23748096823692322,
      "step": 11140
    },
    {
      "epoch": 0.22284,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.012455240885416666,
      "learning_rate": 0.0001,
      "loss": 4.2187,
      "loss/crossentropy": 2.3873090744018555,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23743586987257004,
      "step": 11142
    },
    {
      "epoch": 0.22288,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.008234659830729166,
      "learning_rate": 0.0001,
      "loss": 4.1283,
      "loss/crossentropy": 2.248707115650177,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22035827487707138,
      "step": 11144
    },
    {
      "epoch": 0.22292,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0067860921223958336,
      "learning_rate": 0.0001,
      "loss": 4.3561,
      "loss/crossentropy": 1.9792630672454834,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2102261707186699,
      "step": 11146
    },
    {
      "epoch": 0.22296,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.006156158447265625,
      "learning_rate": 0.0001,
      "loss": 4.0479,
      "loss/crossentropy": 2.284587323665619,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20941253006458282,
      "step": 11148
    },
    {
      "epoch": 0.223,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.006361643473307292,
      "learning_rate": 0.0001,
      "loss": 4.1792,
      "loss/crossentropy": 1.9800288677215576,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20009202510118484,
      "step": 11150
    },
    {
      "epoch": 0.22304,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0064165751139322914,
      "learning_rate": 0.0001,
      "loss": 4.1859,
      "loss/crossentropy": 1.925826370716095,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18616148829460144,
      "step": 11152
    },
    {
      "epoch": 0.22308,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0053708394368489586,
      "learning_rate": 0.0001,
      "loss": 3.9281,
      "loss/crossentropy": 2.158196806907654,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21486472338438034,
      "step": 11154
    },
    {
      "epoch": 0.22312,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.004393513997395833,
      "learning_rate": 0.0001,
      "loss": 4.3342,
      "loss/crossentropy": 2.0873841047286987,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22101643681526184,
      "step": 11156
    },
    {
      "epoch": 0.22316,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0044748942057291664,
      "learning_rate": 0.0001,
      "loss": 4.3772,
      "loss/crossentropy": 1.9485042691230774,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19757962226867676,
      "step": 11158
    },
    {
      "epoch": 0.2232,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.004279581705729166,
      "learning_rate": 0.0001,
      "loss": 4.331,
      "loss/crossentropy": 2.055552661418915,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19271192699670792,
      "step": 11160
    },
    {
      "epoch": 0.22324,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0033599853515625,
      "learning_rate": 0.0001,
      "loss": 4.0958,
      "loss/crossentropy": 1.9115247130393982,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1922249048948288,
      "step": 11162
    },
    {
      "epoch": 0.22328,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0043609619140625,
      "learning_rate": 0.0001,
      "loss": 4.2582,
      "loss/crossentropy": 2.1461042761802673,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2185581848025322,
      "step": 11164
    },
    {
      "epoch": 0.22332,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.004808553059895833,
      "learning_rate": 0.0001,
      "loss": 4.1212,
      "loss/crossentropy": 2.1625128984451294,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20505116879940033,
      "step": 11166
    },
    {
      "epoch": 0.22336,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.003692372639973958,
      "learning_rate": 0.0001,
      "loss": 4.0317,
      "loss/crossentropy": 1.9978403449058533,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2015106976032257,
      "step": 11168
    },
    {
      "epoch": 0.2234,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.003794097900390625,
      "learning_rate": 0.0001,
      "loss": 4.0108,
      "loss/crossentropy": 1.9462909698486328,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20814144611358643,
      "step": 11170
    },
    {
      "epoch": 0.22344,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.004423014322916667,
      "learning_rate": 0.0001,
      "loss": 4.2147,
      "loss/crossentropy": 2.262490153312683,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21692577749490738,
      "step": 11172
    },
    {
      "epoch": 0.22348,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0052734375,
      "learning_rate": 0.0001,
      "loss": 4.4079,
      "loss/crossentropy": 2.092001974582672,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22943469882011414,
      "step": 11174
    },
    {
      "epoch": 0.22352,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0052734375,
      "learning_rate": 0.0001,
      "loss": 4.273,
      "loss/crossentropy": 2.302329421043396,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21188674122095108,
      "step": 11176
    },
    {
      "epoch": 0.22356,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.008701324462890625,
      "learning_rate": 0.0001,
      "loss": 4.0762,
      "loss/crossentropy": 1.7181463837623596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19460663199424744,
      "step": 11178
    },
    {
      "epoch": 0.2236,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.007895660400390626,
      "learning_rate": 0.0001,
      "loss": 4.0061,
      "loss/crossentropy": 1.8402328491210938,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1928708627820015,
      "step": 11180
    },
    {
      "epoch": 0.22364,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.008103179931640624,
      "learning_rate": 0.0001,
      "loss": 4.2968,
      "loss/crossentropy": 1.9490735530853271,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2492925077676773,
      "step": 11182
    },
    {
      "epoch": 0.22368,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.00867919921875,
      "learning_rate": 0.0001,
      "loss": 4.113,
      "loss/crossentropy": 2.367197036743164,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25882700830698013,
      "step": 11184
    },
    {
      "epoch": 0.22372,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0090972900390625,
      "learning_rate": 0.0001,
      "loss": 4.0782,
      "loss/crossentropy": 2.1192296743392944,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2208312824368477,
      "step": 11186
    },
    {
      "epoch": 0.22376,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.010772450764973959,
      "learning_rate": 0.0001,
      "loss": 4.7239,
      "loss/crossentropy": 2.2150460481643677,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21440500020980835,
      "step": 11188
    },
    {
      "epoch": 0.2238,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.0121490478515625,
      "learning_rate": 0.0001,
      "loss": 4.4245,
      "loss/crossentropy": 2.0837016105651855,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2777148336172104,
      "step": 11190
    },
    {
      "epoch": 0.22384,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.014240519205729166,
      "learning_rate": 0.0001,
      "loss": 3.918,
      "loss/crossentropy": 1.7995057106018066,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18295922130346298,
      "step": 11192
    },
    {
      "epoch": 0.22388,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.014357248942057291,
      "learning_rate": 0.0001,
      "loss": 4.3361,
      "loss/crossentropy": 2.2150347232818604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24010418355464935,
      "step": 11194
    },
    {
      "epoch": 0.22392,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.014357248942057291,
      "learning_rate": 0.0001,
      "loss": 4.1711,
      "loss/crossentropy": 2.1161770820617676,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2124250829219818,
      "step": 11196
    },
    {
      "epoch": 0.22396,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.013578033447265625,
      "learning_rate": 0.0001,
      "loss": 4.4875,
      "loss/crossentropy": 2.224100112915039,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23966734111309052,
      "step": 11198
    },
    {
      "epoch": 0.224,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.011921946207682292,
      "learning_rate": 0.0001,
      "loss": 4.377,
      "loss/crossentropy": 2.196989417076111,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21709956228733063,
      "step": 11200
    },
    {
      "epoch": 0.22404,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.011574045817057291,
      "learning_rate": 0.0001,
      "loss": 4.0457,
      "loss/crossentropy": 1.6592280864715576,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1827787384390831,
      "step": 11202
    },
    {
      "epoch": 0.22408,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.008790842692057292,
      "learning_rate": 0.0001,
      "loss": 4.0564,
      "loss/crossentropy": 1.9689467549324036,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20387520641088486,
      "step": 11204
    },
    {
      "epoch": 0.22412,
      "grad_norm": 2.8125,
      "grad_norm_var": 0.03997802734375,
      "learning_rate": 0.0001,
      "loss": 4.3494,
      "loss/crossentropy": 2.1846379041671753,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2068176046013832,
      "step": 11206
    },
    {
      "epoch": 0.22416,
      "grad_norm": 2.0,
      "grad_norm_var": 0.037353515625,
      "learning_rate": 0.0001,
      "loss": 4.3948,
      "loss/crossentropy": 1.8361602425575256,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22452392429113388,
      "step": 11208
    },
    {
      "epoch": 0.2242,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0390625,
      "learning_rate": 0.0001,
      "loss": 4.3597,
      "loss/crossentropy": 2.5658172369003296,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2602368891239166,
      "step": 11210
    },
    {
      "epoch": 0.22424,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.04119440714518229,
      "learning_rate": 0.0001,
      "loss": 4.0555,
      "loss/crossentropy": 1.8167916536331177,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20330622047185898,
      "step": 11212
    },
    {
      "epoch": 0.22428,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.04070612589518229,
      "learning_rate": 0.0001,
      "loss": 4.2374,
      "loss/crossentropy": 1.9151161313056946,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20695462822914124,
      "step": 11214
    },
    {
      "epoch": 0.22432,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.04528401692708333,
      "learning_rate": 0.0001,
      "loss": 3.9807,
      "loss/crossentropy": 1.9590752720832825,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1945583075284958,
      "step": 11216
    },
    {
      "epoch": 0.22436,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.04522298177083333,
      "learning_rate": 0.0001,
      "loss": 4.2803,
      "loss/crossentropy": 1.9777710437774658,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20104076713323593,
      "step": 11218
    },
    {
      "epoch": 0.2244,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.04744847615559896,
      "learning_rate": 0.0001,
      "loss": 4.0279,
      "loss/crossentropy": 1.725940465927124,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19038556516170502,
      "step": 11220
    },
    {
      "epoch": 0.22444,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.012963612874348959,
      "learning_rate": 0.0001,
      "loss": 4.4067,
      "loss/crossentropy": 2.505728602409363,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2434321641921997,
      "step": 11222
    },
    {
      "epoch": 0.22448,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.010518137613932292,
      "learning_rate": 0.0001,
      "loss": 4.2902,
      "loss/crossentropy": 2.0733951330184937,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21223794668912888,
      "step": 11224
    },
    {
      "epoch": 0.22452,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.010628255208333333,
      "learning_rate": 0.0001,
      "loss": 4.1548,
      "loss/crossentropy": 2.105073928833008,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22444891929626465,
      "step": 11226
    },
    {
      "epoch": 0.22456,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.009745025634765625,
      "learning_rate": 0.0001,
      "loss": 4.2361,
      "loss/crossentropy": 1.9164994359016418,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20893585681915283,
      "step": 11228
    },
    {
      "epoch": 0.2246,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.014902496337890625,
      "learning_rate": 0.0001,
      "loss": 4.3753,
      "loss/crossentropy": 2.0922536849975586,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22757402062416077,
      "step": 11230
    },
    {
      "epoch": 0.22464,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.01190185546875,
      "learning_rate": 0.0001,
      "loss": 3.9791,
      "loss/crossentropy": 1.886056363582611,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2054021805524826,
      "step": 11232
    },
    {
      "epoch": 0.22468,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.014300282796223958,
      "learning_rate": 0.0001,
      "loss": 3.9194,
      "loss/crossentropy": 2.028389871120453,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2109459862112999,
      "step": 11234
    },
    {
      "epoch": 0.22472,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.012813313802083334,
      "learning_rate": 0.0001,
      "loss": 4.2084,
      "loss/crossentropy": 1.9590765833854675,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19774210453033447,
      "step": 11236
    },
    {
      "epoch": 0.22476,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.012800852457682291,
      "learning_rate": 0.0001,
      "loss": 4.0153,
      "loss/crossentropy": 1.8284756541252136,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17485490441322327,
      "step": 11238
    },
    {
      "epoch": 0.2248,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.014625803629557291,
      "learning_rate": 0.0001,
      "loss": 4.3494,
      "loss/crossentropy": 2.1119225025177,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22235971689224243,
      "step": 11240
    },
    {
      "epoch": 0.22484,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.015209706624348958,
      "learning_rate": 0.0001,
      "loss": 4.3303,
      "loss/crossentropy": 2.3050538301467896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2249491587281227,
      "step": 11242
    },
    {
      "epoch": 0.22488,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.015360514322916666,
      "learning_rate": 0.0001,
      "loss": 4.1235,
      "loss/crossentropy": 1.9806398153305054,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20141054689884186,
      "step": 11244
    },
    {
      "epoch": 0.22492,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.007879384358723958,
      "learning_rate": 0.0001,
      "loss": 4.0793,
      "loss/crossentropy": 2.0702012181282043,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1980852484703064,
      "step": 11246
    },
    {
      "epoch": 0.22496,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.008113606770833334,
      "learning_rate": 0.0001,
      "loss": 3.9997,
      "loss/crossentropy": 2.205365300178528,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22107956558465958,
      "step": 11248
    },
    {
      "epoch": 0.225,
      "grad_norm": 2.890625,
      "grad_norm_var": 0.05272598266601562,
      "learning_rate": 0.0001,
      "loss": 4.0748,
      "loss/crossentropy": 2.196021556854248,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.199070006608963,
      "step": 11250
    },
    {
      "epoch": 0.22504,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.05241673787434896,
      "learning_rate": 0.0001,
      "loss": 4.2062,
      "loss/crossentropy": 2.0661864280700684,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22117173671722412,
      "step": 11252
    },
    {
      "epoch": 0.22508,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.050687408447265624,
      "learning_rate": 0.0001,
      "loss": 3.9176,
      "loss/crossentropy": 1.825901210308075,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.174439437687397,
      "step": 11254
    },
    {
      "epoch": 0.22512,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.061470286051432295,
      "learning_rate": 0.0001,
      "loss": 4.7052,
      "loss/crossentropy": 2.3710498809814453,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2357504665851593,
      "step": 11256
    },
    {
      "epoch": 0.22516,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.06520182291666667,
      "learning_rate": 0.0001,
      "loss": 3.9419,
      "loss/crossentropy": 2.1523157358169556,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2179568186402321,
      "step": 11258
    },
    {
      "epoch": 0.2252,
      "grad_norm": 2.0,
      "grad_norm_var": 0.06559015909830729,
      "learning_rate": 0.0001,
      "loss": 4.1953,
      "loss/crossentropy": 2.0974292755126953,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1980036199092865,
      "step": 11260
    },
    {
      "epoch": 0.22524,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.06339925130208333,
      "learning_rate": 0.0001,
      "loss": 4.4409,
      "loss/crossentropy": 2.2679883241653442,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22085320204496384,
      "step": 11262
    },
    {
      "epoch": 0.22528,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.06108373006184896,
      "learning_rate": 0.0001,
      "loss": 4.2972,
      "loss/crossentropy": 2.0697352290153503,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21976519376039505,
      "step": 11264
    },
    {
      "epoch": 0.22532,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.025465647379557293,
      "learning_rate": 0.0001,
      "loss": 4.4149,
      "loss/crossentropy": 2.1807767748832703,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21304857730865479,
      "step": 11266
    },
    {
      "epoch": 0.22536,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.02541071573893229,
      "learning_rate": 0.0001,
      "loss": 4.4395,
      "loss/crossentropy": 2.178891122341156,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22129195928573608,
      "step": 11268
    },
    {
      "epoch": 0.2254,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.023607381184895835,
      "learning_rate": 0.0001,
      "loss": 4.2908,
      "loss/crossentropy": 2.1025387048721313,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20581847429275513,
      "step": 11270
    },
    {
      "epoch": 0.22544,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.013055165608723959,
      "learning_rate": 0.0001,
      "loss": 4.2192,
      "loss/crossentropy": 1.8710424900054932,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1900918111205101,
      "step": 11272
    },
    {
      "epoch": 0.22548,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.012373606363932291,
      "learning_rate": 0.0001,
      "loss": 4.498,
      "loss/crossentropy": 1.8837561011314392,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21019018441438675,
      "step": 11274
    },
    {
      "epoch": 0.22552,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.013132476806640625,
      "learning_rate": 0.0001,
      "loss": 4.3139,
      "loss/crossentropy": 2.2795380353927612,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23858627676963806,
      "step": 11276
    },
    {
      "epoch": 0.22556,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0158111572265625,
      "learning_rate": 0.0001,
      "loss": 4.0532,
      "loss/crossentropy": 2.0241262316703796,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1990864798426628,
      "step": 11278
    },
    {
      "epoch": 0.2256,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.022541300455729166,
      "learning_rate": 0.0001,
      "loss": 4.2935,
      "loss/crossentropy": 1.9307058453559875,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2100597321987152,
      "step": 11280
    },
    {
      "epoch": 0.22564,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.020414225260416665,
      "learning_rate": 0.0001,
      "loss": 4.2523,
      "loss/crossentropy": 2.08352792263031,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21930356323719025,
      "step": 11282
    },
    {
      "epoch": 0.22568,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.021361287434895834,
      "learning_rate": 0.0001,
      "loss": 4.4277,
      "loss/crossentropy": 2.393290877342224,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2560981214046478,
      "step": 11284
    },
    {
      "epoch": 0.22572,
      "grad_norm": 2.0,
      "grad_norm_var": 0.023738606770833334,
      "learning_rate": 0.0001,
      "loss": 4.1612,
      "loss/crossentropy": 2.20097017288208,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20318082720041275,
      "step": 11286
    },
    {
      "epoch": 0.22576,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0221099853515625,
      "learning_rate": 0.0001,
      "loss": 4.2307,
      "loss/crossentropy": 2.031981647014618,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20313747972249985,
      "step": 11288
    },
    {
      "epoch": 0.2258,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0187255859375,
      "learning_rate": 0.0001,
      "loss": 4.3513,
      "loss/crossentropy": 2.197006046772003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2301759570837021,
      "step": 11290
    },
    {
      "epoch": 0.22584,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0197021484375,
      "learning_rate": 0.0001,
      "loss": 4.2903,
      "loss/crossentropy": 2.294014096260071,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21280069649219513,
      "step": 11292
    },
    {
      "epoch": 0.22588,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.018089803059895833,
      "learning_rate": 0.0001,
      "loss": 4.0985,
      "loss/crossentropy": 2.0551719665527344,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20910422503948212,
      "step": 11294
    },
    {
      "epoch": 0.22592,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.010636393229166667,
      "learning_rate": 0.0001,
      "loss": 4.2109,
      "loss/crossentropy": 2.1044358015060425,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20463456213474274,
      "step": 11296
    },
    {
      "epoch": 0.22596,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.012631988525390625,
      "learning_rate": 0.0001,
      "loss": 4.198,
      "loss/crossentropy": 2.1013529300689697,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21202504634857178,
      "step": 11298
    },
    {
      "epoch": 0.226,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.009329986572265626,
      "learning_rate": 0.0001,
      "loss": 4.4227,
      "loss/crossentropy": 2.1887649297714233,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2048492729663849,
      "step": 11300
    },
    {
      "epoch": 0.22604,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.009525299072265625,
      "learning_rate": 0.0001,
      "loss": 4.0952,
      "loss/crossentropy": 1.8243364691734314,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2013952136039734,
      "step": 11302
    },
    {
      "epoch": 0.22608,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0137451171875,
      "learning_rate": 0.0001,
      "loss": 4.4636,
      "loss/crossentropy": 2.2550541162490845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2493698000907898,
      "step": 11304
    },
    {
      "epoch": 0.22612,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0149169921875,
      "learning_rate": 0.0001,
      "loss": 3.9407,
      "loss/crossentropy": 1.8744492530822754,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19957631081342697,
      "step": 11306
    },
    {
      "epoch": 0.22616,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.013681793212890625,
      "learning_rate": 0.0001,
      "loss": 4.0717,
      "loss/crossentropy": 2.026526629924774,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20474696904420853,
      "step": 11308
    },
    {
      "epoch": 0.2262,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.015881093343098958,
      "learning_rate": 0.0001,
      "loss": 4.3898,
      "loss/crossentropy": 2.3802725076675415,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.247590571641922,
      "step": 11310
    },
    {
      "epoch": 0.22624,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.016123199462890626,
      "learning_rate": 0.0001,
      "loss": 4.171,
      "loss/crossentropy": 1.9359605312347412,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.207523413002491,
      "step": 11312
    },
    {
      "epoch": 0.22628,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0129547119140625,
      "learning_rate": 0.0001,
      "loss": 4.1634,
      "loss/crossentropy": 2.3035311698913574,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21815945208072662,
      "step": 11314
    },
    {
      "epoch": 0.22632,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.0225341796875,
      "learning_rate": 0.0001,
      "loss": 4.3189,
      "loss/crossentropy": 1.8083779215812683,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20705801248550415,
      "step": 11316
    },
    {
      "epoch": 0.22636,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.0205718994140625,
      "learning_rate": 0.0001,
      "loss": 4.2912,
      "loss/crossentropy": 2.306682825088501,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22666794806718826,
      "step": 11318
    },
    {
      "epoch": 0.2264,
      "grad_norm": 2.0,
      "grad_norm_var": 0.01737060546875,
      "learning_rate": 0.0001,
      "loss": 4.0519,
      "loss/crossentropy": 1.809365153312683,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20744331926107407,
      "step": 11320
    },
    {
      "epoch": 0.22644,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.017894490559895834,
      "learning_rate": 0.0001,
      "loss": 4.2428,
      "loss/crossentropy": 2.289853572845459,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23791835457086563,
      "step": 11322
    },
    {
      "epoch": 0.22648,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.027341461181640624,
      "learning_rate": 0.0001,
      "loss": 4.5668,
      "loss/crossentropy": 2.0609280467033386,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2166810780763626,
      "step": 11324
    },
    {
      "epoch": 0.22652,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.02535400390625,
      "learning_rate": 0.0001,
      "loss": 4.0567,
      "loss/crossentropy": 2.1174912452697754,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20515341311693192,
      "step": 11326
    },
    {
      "epoch": 0.22656,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.029319000244140626,
      "learning_rate": 0.0001,
      "loss": 4.0318,
      "loss/crossentropy": 1.99192476272583,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20127686113119125,
      "step": 11328
    },
    {
      "epoch": 0.2266,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.031060536702473957,
      "learning_rate": 0.0001,
      "loss": 3.9552,
      "loss/crossentropy": 1.7805609107017517,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17500489950180054,
      "step": 11330
    },
    {
      "epoch": 0.22664,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.02398656209309896,
      "learning_rate": 0.0001,
      "loss": 4.1417,
      "loss/crossentropy": 2.024085283279419,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20659280568361282,
      "step": 11332
    },
    {
      "epoch": 0.22668,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.023631795247395834,
      "learning_rate": 0.0001,
      "loss": 4.2674,
      "loss/crossentropy": 2.174618899822235,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23250284045934677,
      "step": 11334
    },
    {
      "epoch": 0.22672,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.02451171875,
      "learning_rate": 0.0001,
      "loss": 4.213,
      "loss/crossentropy": 1.886943757534027,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18855806440114975,
      "step": 11336
    },
    {
      "epoch": 0.22676,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.02182184855143229,
      "learning_rate": 0.0001,
      "loss": 4.1281,
      "loss/crossentropy": 2.060371160507202,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1951771154999733,
      "step": 11338
    },
    {
      "epoch": 0.2268,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.005576324462890625,
      "learning_rate": 0.0001,
      "loss": 4.1692,
      "loss/crossentropy": 2.15705668926239,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23738030344247818,
      "step": 11340
    },
    {
      "epoch": 0.22684,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.005576324462890625,
      "learning_rate": 0.0001,
      "loss": 4.2096,
      "loss/crossentropy": 2.1922959089279175,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24581187963485718,
      "step": 11342
    },
    {
      "epoch": 0.22688,
      "grad_norm": 2.0,
      "grad_norm_var": 0.007303873697916667,
      "learning_rate": 0.0001,
      "loss": 3.9786,
      "loss/crossentropy": 2.1590365171432495,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20936457812786102,
      "step": 11344
    },
    {
      "epoch": 0.22692,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.020783487955729166,
      "learning_rate": 0.0001,
      "loss": 4.2382,
      "loss/crossentropy": 1.8087702989578247,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18874070048332214,
      "step": 11346
    },
    {
      "epoch": 0.22696,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.021732584635416666,
      "learning_rate": 0.0001,
      "loss": 4.4981,
      "loss/crossentropy": 2.5205971002578735,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24057865887880325,
      "step": 11348
    },
    {
      "epoch": 0.227,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.02061945597330729,
      "learning_rate": 0.0001,
      "loss": 4.16,
      "loss/crossentropy": 2.114508092403412,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20631257444620132,
      "step": 11350
    },
    {
      "epoch": 0.22704,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.020881144205729167,
      "learning_rate": 0.0001,
      "loss": 4.4344,
      "loss/crossentropy": 2.385537028312683,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.233476921916008,
      "step": 11352
    },
    {
      "epoch": 0.22708,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.0217926025390625,
      "learning_rate": 0.0001,
      "loss": 4.3335,
      "loss/crossentropy": 2.4251039028167725,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24502182751893997,
      "step": 11354
    },
    {
      "epoch": 0.22712,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.025690714518229168,
      "learning_rate": 0.0001,
      "loss": 4.345,
      "loss/crossentropy": 2.278030514717102,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22328510880470276,
      "step": 11356
    },
    {
      "epoch": 0.22716,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.025031534830729167,
      "learning_rate": 0.0001,
      "loss": 4.2835,
      "loss/crossentropy": 1.7373265027999878,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17635879665613174,
      "step": 11358
    },
    {
      "epoch": 0.2272,
      "grad_norm": 2.25,
      "grad_norm_var": 0.024461873372395835,
      "learning_rate": 0.0001,
      "loss": 4.3644,
      "loss/crossentropy": 2.1546722650527954,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22285740077495575,
      "step": 11360
    },
    {
      "epoch": 0.22724,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0143218994140625,
      "learning_rate": 0.0001,
      "loss": 4.5415,
      "loss/crossentropy": 2.341711401939392,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23693612217903137,
      "step": 11362
    },
    {
      "epoch": 0.22728,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.0461090087890625,
      "learning_rate": 0.0001,
      "loss": 4.3131,
      "loss/crossentropy": 1.8764930367469788,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19202134013175964,
      "step": 11364
    },
    {
      "epoch": 0.22732,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0450439453125,
      "learning_rate": 0.0001,
      "loss": 4.1172,
      "loss/crossentropy": 2.035600185394287,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2188200280070305,
      "step": 11366
    },
    {
      "epoch": 0.22736,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.045481109619140626,
      "learning_rate": 0.0001,
      "loss": 4.0469,
      "loss/crossentropy": 2.120850682258606,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22664117813110352,
      "step": 11368
    },
    {
      "epoch": 0.2274,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.050455729166666664,
      "learning_rate": 0.0001,
      "loss": 3.9552,
      "loss/crossentropy": 1.9753262996673584,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20000861585140228,
      "step": 11370
    },
    {
      "epoch": 0.22744,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.06503499348958333,
      "learning_rate": 0.0001,
      "loss": 4.2474,
      "loss/crossentropy": 2.0264564156532288,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2026129513978958,
      "step": 11372
    },
    {
      "epoch": 0.22748,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.07055562337239583,
      "learning_rate": 0.0001,
      "loss": 3.8546,
      "loss/crossentropy": 1.9269734025001526,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22436296939849854,
      "step": 11374
    },
    {
      "epoch": 0.22752,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.07017822265625,
      "learning_rate": 0.0001,
      "loss": 4.4164,
      "loss/crossentropy": 2.1881991624832153,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22483092546463013,
      "step": 11376
    },
    {
      "epoch": 0.22756,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.07446187337239583,
      "learning_rate": 0.0001,
      "loss": 4.0277,
      "loss/crossentropy": 2.0142401456832886,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21913451701402664,
      "step": 11378
    },
    {
      "epoch": 0.2276,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.05676676432291667,
      "learning_rate": 0.0001,
      "loss": 4.3374,
      "loss/crossentropy": 2.0229859352111816,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.238722562789917,
      "step": 11380
    },
    {
      "epoch": 0.22764,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.05724054972330729,
      "learning_rate": 0.0001,
      "loss": 4.0223,
      "loss/crossentropy": 2.043630540370941,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21258120238780975,
      "step": 11382
    },
    {
      "epoch": 0.22768,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.06020685831705729,
      "learning_rate": 0.0001,
      "loss": 4.0752,
      "loss/crossentropy": 2.702946662902832,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24362115561962128,
      "step": 11384
    },
    {
      "epoch": 0.22772,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.05449600219726562,
      "learning_rate": 0.0001,
      "loss": 3.9806,
      "loss/crossentropy": 2.243411898612976,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22256402671337128,
      "step": 11386
    },
    {
      "epoch": 0.22776,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.033699289957682295,
      "learning_rate": 0.0001,
      "loss": 3.7808,
      "loss/crossentropy": 2.1451956033706665,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21471337974071503,
      "step": 11388
    },
    {
      "epoch": 0.2278,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.030775705973307293,
      "learning_rate": 0.0001,
      "loss": 4.2865,
      "loss/crossentropy": 2.426279664039612,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22846391052007675,
      "step": 11390
    },
    {
      "epoch": 0.22784,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.033455149332682295,
      "learning_rate": 0.0001,
      "loss": 4.2878,
      "loss/crossentropy": 2.0890655517578125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23008010536432266,
      "step": 11392
    },
    {
      "epoch": 0.22788,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.030452219645182292,
      "learning_rate": 0.0001,
      "loss": 4.2124,
      "loss/crossentropy": 2.3496296405792236,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23718956112861633,
      "step": 11394
    },
    {
      "epoch": 0.22792,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.013152821858723959,
      "learning_rate": 0.0001,
      "loss": 4.0025,
      "loss/crossentropy": 2.02141535282135,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21102941036224365,
      "step": 11396
    },
    {
      "epoch": 0.22796,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.01275634765625,
      "learning_rate": 0.0001,
      "loss": 4.4817,
      "loss/crossentropy": 2.213461995124817,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21838735044002533,
      "step": 11398
    },
    {
      "epoch": 0.228,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.11743062337239583,
      "learning_rate": 0.0001,
      "loss": 4.2313,
      "loss/crossentropy": 2.161481499671936,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24635545909404755,
      "step": 11400
    },
    {
      "epoch": 0.22804,
      "grad_norm": 2.125,
      "grad_norm_var": 0.1158843994140625,
      "learning_rate": 0.0001,
      "loss": 4.5372,
      "loss/crossentropy": 2.0998951196670532,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2205812931060791,
      "step": 11402
    },
    {
      "epoch": 0.22808,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.11588109334309896,
      "learning_rate": 0.0001,
      "loss": 4.1606,
      "loss/crossentropy": 1.8785207867622375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20330313593149185,
      "step": 11404
    },
    {
      "epoch": 0.22812,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.11588109334309896,
      "learning_rate": 0.0001,
      "loss": 4.2408,
      "loss/crossentropy": 2.254656672477722,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.220963753759861,
      "step": 11406
    },
    {
      "epoch": 0.22816,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.11553726196289063,
      "learning_rate": 0.0001,
      "loss": 4.1356,
      "loss/crossentropy": 1.7936404347419739,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1878006011247635,
      "step": 11408
    },
    {
      "epoch": 0.2282,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.12290445963541667,
      "learning_rate": 0.0001,
      "loss": 4.2199,
      "loss/crossentropy": 1.8537682890892029,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1893099844455719,
      "step": 11410
    },
    {
      "epoch": 0.22824,
      "grad_norm": 3.234375,
      "grad_norm_var": 0.1904205322265625,
      "learning_rate": 0.0001,
      "loss": 4.1628,
      "loss/crossentropy": 2.0987170338630676,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2072230949997902,
      "step": 11412
    },
    {
      "epoch": 0.22828,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.1907958984375,
      "learning_rate": 0.0001,
      "loss": 4.2473,
      "loss/crossentropy": 2.049329698085785,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2142597660422325,
      "step": 11414
    },
    {
      "epoch": 0.22832,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.09533284505208334,
      "learning_rate": 0.0001,
      "loss": 4.283,
      "loss/crossentropy": 2.066833019256592,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22086824476718903,
      "step": 11416
    },
    {
      "epoch": 0.22836,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.09519755045572917,
      "learning_rate": 0.0001,
      "loss": 4.1837,
      "loss/crossentropy": 2.01213002204895,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23191991448402405,
      "step": 11418
    },
    {
      "epoch": 0.2284,
      "grad_norm": 5.6875,
      "grad_norm_var": 0.8556495666503906,
      "learning_rate": 0.0001,
      "loss": 4.1977,
      "loss/crossentropy": 2.1349334716796875,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22979671508073807,
      "step": 11420
    },
    {
      "epoch": 0.22844,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.8487709045410157,
      "learning_rate": 0.0001,
      "loss": 4.2101,
      "loss/crossentropy": 1.9201850295066833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21903745830059052,
      "step": 11422
    },
    {
      "epoch": 0.22848,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.8546376546223958,
      "learning_rate": 0.0001,
      "loss": 4.1927,
      "loss/crossentropy": 2.011150360107422,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2059009075164795,
      "step": 11424
    },
    {
      "epoch": 0.22852,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.8533322652180989,
      "learning_rate": 0.0001,
      "loss": 4.3762,
      "loss/crossentropy": 2.1529648303985596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23028475046157837,
      "step": 11426
    },
    {
      "epoch": 0.22856,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.8176177978515625,
      "learning_rate": 0.0001,
      "loss": 3.9657,
      "loss/crossentropy": 2.1602566838264465,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2262946516275406,
      "step": 11428
    },
    {
      "epoch": 0.2286,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.829766591389974,
      "learning_rate": 0.0001,
      "loss": 3.9369,
      "loss/crossentropy": 1.7230631113052368,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19715693593025208,
      "step": 11430
    },
    {
      "epoch": 0.22864,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.8287737528483073,
      "learning_rate": 0.0001,
      "loss": 4.3791,
      "loss/crossentropy": 2.4396276473999023,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24283458292484283,
      "step": 11432
    },
    {
      "epoch": 0.22868,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.8311480204264323,
      "learning_rate": 0.0001,
      "loss": 4.2692,
      "loss/crossentropy": 1.8615645170211792,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21288780122995377,
      "step": 11434
    },
    {
      "epoch": 0.22872,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.012941233317057292,
      "learning_rate": 0.0001,
      "loss": 4.394,
      "loss/crossentropy": 1.8972707390785217,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18510619550943375,
      "step": 11436
    },
    {
      "epoch": 0.22876,
      "grad_norm": 2.0,
      "grad_norm_var": 0.011980946858723958,
      "learning_rate": 0.0001,
      "loss": 4.1459,
      "loss/crossentropy": 2.217754364013672,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21217356622219086,
      "step": 11438
    },
    {
      "epoch": 0.2288,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.014818318684895833,
      "learning_rate": 0.0001,
      "loss": 3.7955,
      "loss/crossentropy": 1.6481398940086365,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17804963141679764,
      "step": 11440
    },
    {
      "epoch": 0.22884,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.01375732421875,
      "learning_rate": 0.0001,
      "loss": 4.3291,
      "loss/crossentropy": 2.026508390903473,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20739784091711044,
      "step": 11442
    },
    {
      "epoch": 0.22888,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.012334950764973958,
      "learning_rate": 0.0001,
      "loss": 4.2086,
      "loss/crossentropy": 2.05685293674469,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22639908641576767,
      "step": 11444
    },
    {
      "epoch": 0.22892,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.01197509765625,
      "learning_rate": 0.0001,
      "loss": 4.4689,
      "loss/crossentropy": 2.073192059993744,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21727359294891357,
      "step": 11446
    },
    {
      "epoch": 0.22896,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.0092193603515625,
      "learning_rate": 0.0001,
      "loss": 4.4312,
      "loss/crossentropy": 2.1311055421829224,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19499187916517258,
      "step": 11448
    },
    {
      "epoch": 0.229,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.01021728515625,
      "learning_rate": 0.0001,
      "loss": 4.2043,
      "loss/crossentropy": 2.169051766395569,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22266974300146103,
      "step": 11450
    },
    {
      "epoch": 0.22904,
      "grad_norm": 1.875,
      "grad_norm_var": 0.013727823893229166,
      "learning_rate": 0.0001,
      "loss": 4.1784,
      "loss/crossentropy": 2.022417426109314,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20541484653949738,
      "step": 11452
    },
    {
      "epoch": 0.22908,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.015458170572916667,
      "learning_rate": 0.0001,
      "loss": 4.2921,
      "loss/crossentropy": 2.2652071714401245,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21639510244131088,
      "step": 11454
    },
    {
      "epoch": 0.22912,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.01343994140625,
      "learning_rate": 0.0001,
      "loss": 4.1575,
      "loss/crossentropy": 1.912036418914795,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2204061597585678,
      "step": 11456
    },
    {
      "epoch": 0.22916,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.014188639322916667,
      "learning_rate": 0.0001,
      "loss": 4.4497,
      "loss/crossentropy": 2.029780328273773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2032444253563881,
      "step": 11458
    },
    {
      "epoch": 0.2292,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.016039021809895835,
      "learning_rate": 0.0001,
      "loss": 4.0627,
      "loss/crossentropy": 1.954946756362915,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2028508484363556,
      "step": 11460
    },
    {
      "epoch": 0.22924,
      "grad_norm": 2.8125,
      "grad_norm_var": 0.049494425455729164,
      "learning_rate": 0.0001,
      "loss": 4.5787,
      "loss/crossentropy": 2.3707855939865112,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2460392713546753,
      "step": 11462
    },
    {
      "epoch": 0.22928,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.047972615559895834,
      "learning_rate": 0.0001,
      "loss": 4.1509,
      "loss/crossentropy": 2.410157322883606,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24578960239887238,
      "step": 11464
    },
    {
      "epoch": 0.22932,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.049117024739583334,
      "learning_rate": 0.0001,
      "loss": 4.0935,
      "loss/crossentropy": 2.1403380036354065,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20563968271017075,
      "step": 11466
    },
    {
      "epoch": 0.22936,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0471588134765625,
      "learning_rate": 0.0001,
      "loss": 4.0991,
      "loss/crossentropy": 2.1596190333366394,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24405791610479355,
      "step": 11468
    },
    {
      "epoch": 0.2294,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.0506744384765625,
      "learning_rate": 0.0001,
      "loss": 4.3973,
      "loss/crossentropy": 1.9427857398986816,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22412577271461487,
      "step": 11470
    },
    {
      "epoch": 0.22944,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.05306574503580729,
      "learning_rate": 0.0001,
      "loss": 4.1093,
      "loss/crossentropy": 2.078735053539276,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20844466239213943,
      "step": 11472
    },
    {
      "epoch": 0.22948,
      "grad_norm": 2.0,
      "grad_norm_var": 0.054323069254557294,
      "learning_rate": 0.0001,
      "loss": 4.0749,
      "loss/crossentropy": 2.351656198501587,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24430027604103088,
      "step": 11474
    },
    {
      "epoch": 0.22952,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.05269139607747396,
      "learning_rate": 0.0001,
      "loss": 3.9647,
      "loss/crossentropy": 2.057854652404785,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21224451810121536,
      "step": 11476
    },
    {
      "epoch": 0.22956,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.018534088134765626,
      "learning_rate": 0.0001,
      "loss": 4.1451,
      "loss/crossentropy": 2.207979917526245,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22855235636234283,
      "step": 11478
    },
    {
      "epoch": 0.2296,
      "grad_norm": 2.125,
      "grad_norm_var": 0.01870905558268229,
      "learning_rate": 0.0001,
      "loss": 4.3748,
      "loss/crossentropy": 2.08840012550354,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21988992393016815,
      "step": 11480
    },
    {
      "epoch": 0.22964,
      "grad_norm": 2.125,
      "grad_norm_var": 0.017144521077473957,
      "learning_rate": 0.0001,
      "loss": 4.2314,
      "loss/crossentropy": 2.1339274644851685,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2278972491621971,
      "step": 11482
    },
    {
      "epoch": 0.22968,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.015386708577473958,
      "learning_rate": 0.0001,
      "loss": 4.2013,
      "loss/crossentropy": 1.970679223537445,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20286859571933746,
      "step": 11484
    },
    {
      "epoch": 0.22972,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.005295562744140625,
      "learning_rate": 0.0001,
      "loss": 4.2443,
      "loss/crossentropy": 2.1023008823394775,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21906304359436035,
      "step": 11486
    },
    {
      "epoch": 0.22976,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.004870351155598958,
      "learning_rate": 0.0001,
      "loss": 4.2774,
      "loss/crossentropy": 2.1415608525276184,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21857701241970062,
      "step": 11488
    },
    {
      "epoch": 0.2298,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.005191802978515625,
      "learning_rate": 0.0001,
      "loss": 4.0697,
      "loss/crossentropy": 1.9523325562477112,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19859597831964493,
      "step": 11490
    },
    {
      "epoch": 0.22984,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0046770731608072914,
      "learning_rate": 0.0001,
      "loss": 4.2433,
      "loss/crossentropy": 2.1532927751541138,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2239024043083191,
      "step": 11492
    },
    {
      "epoch": 0.22988,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.006705474853515625,
      "learning_rate": 0.0001,
      "loss": 4.3556,
      "loss/crossentropy": 2.160528779029846,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23207154870033264,
      "step": 11494
    },
    {
      "epoch": 0.22992,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.006528472900390625,
      "learning_rate": 0.0001,
      "loss": 4.277,
      "loss/crossentropy": 2.07854962348938,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2189328968524933,
      "step": 11496
    },
    {
      "epoch": 0.22996,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.009069569905598958,
      "learning_rate": 0.0001,
      "loss": 4.4237,
      "loss/crossentropy": 2.2270501852035522,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23237968981266022,
      "step": 11498
    },
    {
      "epoch": 0.23,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0125396728515625,
      "learning_rate": 0.0001,
      "loss": 3.96,
      "loss/crossentropy": 1.9641701579093933,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2063433900475502,
      "step": 11500
    },
    {
      "epoch": 0.23004,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.016866048177083332,
      "learning_rate": 0.0001,
      "loss": 4.3002,
      "loss/crossentropy": 1.9243032932281494,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20478814095258713,
      "step": 11502
    },
    {
      "epoch": 0.23008,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.01587092081705729,
      "learning_rate": 0.0001,
      "loss": 4.329,
      "loss/crossentropy": 2.295292854309082,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23184886574745178,
      "step": 11504
    },
    {
      "epoch": 0.23012,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.01654052734375,
      "learning_rate": 0.0001,
      "loss": 4.0697,
      "loss/crossentropy": 2.134859561920166,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21923956274986267,
      "step": 11506
    },
    {
      "epoch": 0.23016,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0158355712890625,
      "learning_rate": 0.0001,
      "loss": 4.2523,
      "loss/crossentropy": 1.905085265636444,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22443564236164093,
      "step": 11508
    },
    {
      "epoch": 0.2302,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.014892323811848959,
      "learning_rate": 0.0001,
      "loss": 4.0296,
      "loss/crossentropy": 1.6967324614524841,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19518497586250305,
      "step": 11510
    },
    {
      "epoch": 0.23024,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.015364583333333333,
      "learning_rate": 0.0001,
      "loss": 4.225,
      "loss/crossentropy": 1.638957679271698,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19098830223083496,
      "step": 11512
    },
    {
      "epoch": 0.23028,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.013732655843098959,
      "learning_rate": 0.0001,
      "loss": 4.0479,
      "loss/crossentropy": 1.8760477900505066,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.202397421002388,
      "step": 11514
    },
    {
      "epoch": 0.23032,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.012189737955729167,
      "learning_rate": 0.0001,
      "loss": 4.2265,
      "loss/crossentropy": 2.201690912246704,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22841450572013855,
      "step": 11516
    },
    {
      "epoch": 0.23036,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.007987467447916667,
      "learning_rate": 0.0001,
      "loss": 4.5451,
      "loss/crossentropy": 2.5022183656692505,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25180216133594513,
      "step": 11518
    },
    {
      "epoch": 0.2304,
      "grad_norm": 2.0,
      "grad_norm_var": 0.007682291666666666,
      "learning_rate": 0.0001,
      "loss": 4.2848,
      "loss/crossentropy": 2.4634610414505005,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23911123722791672,
      "step": 11520
    },
    {
      "epoch": 0.23044,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.007503000895182291,
      "learning_rate": 0.0001,
      "loss": 4.2919,
      "loss/crossentropy": 2.1176512241363525,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20835164189338684,
      "step": 11522
    },
    {
      "epoch": 0.23048,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.007252756754557292,
      "learning_rate": 0.0001,
      "loss": 4.0803,
      "loss/crossentropy": 1.7708171606063843,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18085652589797974,
      "step": 11524
    },
    {
      "epoch": 0.23052,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.008504231770833334,
      "learning_rate": 0.0001,
      "loss": 4.1159,
      "loss/crossentropy": 1.6740695238113403,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17464587092399597,
      "step": 11526
    },
    {
      "epoch": 0.23056,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.008135732014973958,
      "learning_rate": 0.0001,
      "loss": 4.1829,
      "loss/crossentropy": 1.8153178691864014,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19107046723365784,
      "step": 11528
    },
    {
      "epoch": 0.2306,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.006231435139973958,
      "learning_rate": 0.0001,
      "loss": 4.2074,
      "loss/crossentropy": 2.2833873629570007,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19900934398174286,
      "step": 11530
    },
    {
      "epoch": 0.23064,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008593495686848958,
      "learning_rate": 0.0001,
      "loss": 4.3154,
      "loss/crossentropy": 1.9106029272079468,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1996043100953102,
      "step": 11532
    },
    {
      "epoch": 0.23068,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.006030019124348958,
      "learning_rate": 0.0001,
      "loss": 4.3849,
      "loss/crossentropy": 2.049258530139923,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21438845992088318,
      "step": 11534
    },
    {
      "epoch": 0.23072,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.006029256184895833,
      "learning_rate": 0.0001,
      "loss": 3.9236,
      "loss/crossentropy": 1.9124351739883423,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20111311972141266,
      "step": 11536
    },
    {
      "epoch": 0.23076,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0059397379557291664,
      "learning_rate": 0.0001,
      "loss": 4.0781,
      "loss/crossentropy": 1.8045400381088257,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20846854895353317,
      "step": 11538
    },
    {
      "epoch": 0.2308,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.007297515869140625,
      "learning_rate": 0.0001,
      "loss": 4.1804,
      "loss/crossentropy": 1.9970109462738037,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20165999233722687,
      "step": 11540
    },
    {
      "epoch": 0.23084,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.013065338134765625,
      "learning_rate": 0.0001,
      "loss": 4.1756,
      "loss/crossentropy": 2.0300097465515137,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24561113119125366,
      "step": 11542
    },
    {
      "epoch": 0.23088,
      "grad_norm": 2.5,
      "grad_norm_var": 0.025402577718098958,
      "learning_rate": 0.0001,
      "loss": 4.4841,
      "loss/crossentropy": 2.194391131401062,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24075081944465637,
      "step": 11544
    },
    {
      "epoch": 0.23092,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0239898681640625,
      "learning_rate": 0.0001,
      "loss": 3.9177,
      "loss/crossentropy": 2.0229761600494385,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21907109022140503,
      "step": 11546
    },
    {
      "epoch": 0.23096,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.022606404622395833,
      "learning_rate": 0.0001,
      "loss": 4.3089,
      "loss/crossentropy": 1.831793487071991,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18943443894386292,
      "step": 11548
    },
    {
      "epoch": 0.231,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.023395792643229166,
      "learning_rate": 0.0001,
      "loss": 4.19,
      "loss/crossentropy": 2.4099985361099243,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22612392157316208,
      "step": 11550
    },
    {
      "epoch": 0.23104,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.022946929931640624,
      "learning_rate": 0.0001,
      "loss": 4.2512,
      "loss/crossentropy": 2.0496281385421753,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20928414165973663,
      "step": 11552
    },
    {
      "epoch": 0.23108,
      "grad_norm": 1.875,
      "grad_norm_var": 0.025986480712890624,
      "learning_rate": 0.0001,
      "loss": 4.196,
      "loss/crossentropy": 1.9366755485534668,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19255827367305756,
      "step": 11554
    },
    {
      "epoch": 0.23112,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.02516454060872396,
      "learning_rate": 0.0001,
      "loss": 4.3758,
      "loss/crossentropy": 1.9077317714691162,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20340882241725922,
      "step": 11556
    },
    {
      "epoch": 0.23116,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.02020848592122396,
      "learning_rate": 0.0001,
      "loss": 4.091,
      "loss/crossentropy": 1.9167283773422241,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21566946804523468,
      "step": 11558
    },
    {
      "epoch": 0.2312,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0059506734212239586,
      "learning_rate": 0.0001,
      "loss": 4.5758,
      "loss/crossentropy": 2.4604564905166626,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22104239463806152,
      "step": 11560
    },
    {
      "epoch": 0.23124,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.009124501546223959,
      "learning_rate": 0.0001,
      "loss": 4.0398,
      "loss/crossentropy": 2.359586775302887,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2064266949892044,
      "step": 11562
    },
    {
      "epoch": 0.23128,
      "grad_norm": 1.875,
      "grad_norm_var": 0.009456125895182292,
      "learning_rate": 0.0001,
      "loss": 4.0065,
      "loss/crossentropy": 1.9757064580917358,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19224581122398376,
      "step": 11564
    },
    {
      "epoch": 0.23132,
      "grad_norm": 2.125,
      "grad_norm_var": 0.03843561808268229,
      "learning_rate": 0.0001,
      "loss": 4.5544,
      "loss/crossentropy": 1.9888432025909424,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20799735933542252,
      "step": 11566
    },
    {
      "epoch": 0.23136,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.03802057902018229,
      "learning_rate": 0.0001,
      "loss": 4.2281,
      "loss/crossentropy": 2.0829046964645386,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22345586121082306,
      "step": 11568
    },
    {
      "epoch": 0.2314,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.036649322509765624,
      "learning_rate": 0.0001,
      "loss": 4.3421,
      "loss/crossentropy": 1.964626431465149,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21068184822797775,
      "step": 11570
    },
    {
      "epoch": 0.23144,
      "grad_norm": 2.125,
      "grad_norm_var": 0.034993489583333336,
      "learning_rate": 0.0001,
      "loss": 4.4206,
      "loss/crossentropy": 2.313928008079529,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23291154205799103,
      "step": 11572
    },
    {
      "epoch": 0.23148,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.038386027018229164,
      "learning_rate": 0.0001,
      "loss": 3.9609,
      "loss/crossentropy": 2.221343159675598,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20674917846918106,
      "step": 11574
    },
    {
      "epoch": 0.23152,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.03853759765625,
      "learning_rate": 0.0001,
      "loss": 4.4097,
      "loss/crossentropy": 2.070296823978424,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22503886371850967,
      "step": 11576
    },
    {
      "epoch": 0.23156,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.03483784993489583,
      "learning_rate": 0.0001,
      "loss": 3.9417,
      "loss/crossentropy": 2.21540367603302,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2334480583667755,
      "step": 11578
    },
    {
      "epoch": 0.2316,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.03614679972330729,
      "learning_rate": 0.0001,
      "loss": 4.2491,
      "loss/crossentropy": 1.9438464641571045,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20312584936618805,
      "step": 11580
    },
    {
      "epoch": 0.23164,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.010994211832682291,
      "learning_rate": 0.0001,
      "loss": 4.0622,
      "loss/crossentropy": 1.4970324039459229,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17415452748537064,
      "step": 11582
    },
    {
      "epoch": 0.23168,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.026244099934895834,
      "learning_rate": 0.0001,
      "loss": 4.2293,
      "loss/crossentropy": 2.2618579864501953,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23066900670528412,
      "step": 11584
    },
    {
      "epoch": 0.23172,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.02603123982747396,
      "learning_rate": 0.0001,
      "loss": 4.0974,
      "loss/crossentropy": 2.2342761754989624,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20567959547042847,
      "step": 11586
    },
    {
      "epoch": 0.23176,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.026151275634765624,
      "learning_rate": 0.0001,
      "loss": 4.0417,
      "loss/crossentropy": 2.144785463809967,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19473369419574738,
      "step": 11588
    },
    {
      "epoch": 0.2318,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.02851130167643229,
      "learning_rate": 0.0001,
      "loss": 4.1996,
      "loss/crossentropy": 1.9032491445541382,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1956048086285591,
      "step": 11590
    },
    {
      "epoch": 0.23184,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.03227717081705729,
      "learning_rate": 0.0001,
      "loss": 4.4689,
      "loss/crossentropy": 1.934233546257019,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22426530718803406,
      "step": 11592
    },
    {
      "epoch": 0.23188,
      "grad_norm": 2.0,
      "grad_norm_var": 0.03166071573893229,
      "learning_rate": 0.0001,
      "loss": 4.2321,
      "loss/crossentropy": 2.1792030930519104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22469744831323624,
      "step": 11594
    },
    {
      "epoch": 0.23192,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.027705891927083334,
      "learning_rate": 0.0001,
      "loss": 4.3779,
      "loss/crossentropy": 2.3242534399032593,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23015478998422623,
      "step": 11596
    },
    {
      "epoch": 0.23196,
      "grad_norm": 2.5,
      "grad_norm_var": 0.036622873942057294,
      "learning_rate": 0.0001,
      "loss": 4.3553,
      "loss/crossentropy": 2.2343804836273193,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2392263486981392,
      "step": 11598
    },
    {
      "epoch": 0.232,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.024559529622395833,
      "learning_rate": 0.0001,
      "loss": 4.3019,
      "loss/crossentropy": 1.9622855186462402,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22550886124372482,
      "step": 11600
    },
    {
      "epoch": 0.23204,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.023981730143229168,
      "learning_rate": 0.0001,
      "loss": 4.4955,
      "loss/crossentropy": 2.2274144887924194,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21090717613697052,
      "step": 11602
    },
    {
      "epoch": 0.23208,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.021683756510416666,
      "learning_rate": 0.0001,
      "loss": 4.3011,
      "loss/crossentropy": 2.092648506164551,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21386967599391937,
      "step": 11604
    },
    {
      "epoch": 0.23212,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.020643870035807293,
      "learning_rate": 0.0001,
      "loss": 4.2739,
      "loss/crossentropy": 2.3358936309814453,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22385042905807495,
      "step": 11606
    },
    {
      "epoch": 0.23216,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.01883112589518229,
      "learning_rate": 0.0001,
      "loss": 4.1968,
      "loss/crossentropy": 1.9800407886505127,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20614364743232727,
      "step": 11608
    },
    {
      "epoch": 0.2322,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.018536122639973958,
      "learning_rate": 0.0001,
      "loss": 4.2385,
      "loss/crossentropy": 1.9646947979927063,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21902073919773102,
      "step": 11610
    },
    {
      "epoch": 0.23224,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.018930816650390626,
      "learning_rate": 0.0001,
      "loss": 4.1564,
      "loss/crossentropy": 2.0113691687583923,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21262314170598984,
      "step": 11612
    },
    {
      "epoch": 0.23228,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.0254791259765625,
      "learning_rate": 0.0001,
      "loss": 4.3179,
      "loss/crossentropy": 1.8359833359718323,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2125518098473549,
      "step": 11614
    },
    {
      "epoch": 0.23232,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.02577489217122396,
      "learning_rate": 0.0001,
      "loss": 4.0954,
      "loss/crossentropy": 1.934333622455597,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2148296758532524,
      "step": 11616
    },
    {
      "epoch": 0.23236,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.026192220052083333,
      "learning_rate": 0.0001,
      "loss": 4.3556,
      "loss/crossentropy": 1.9486380815505981,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.201569102704525,
      "step": 11618
    },
    {
      "epoch": 0.2324,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0251708984375,
      "learning_rate": 0.0001,
      "loss": 4.3761,
      "loss/crossentropy": 1.9379103183746338,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2109408900141716,
      "step": 11620
    },
    {
      "epoch": 0.23244,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.02165705362955729,
      "learning_rate": 0.0001,
      "loss": 4.3957,
      "loss/crossentropy": 2.1635884046554565,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23330900818109512,
      "step": 11622
    },
    {
      "epoch": 0.23248,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.02851130167643229,
      "learning_rate": 0.0001,
      "loss": 4.1035,
      "loss/crossentropy": 2.3183244466781616,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21685285866260529,
      "step": 11624
    },
    {
      "epoch": 0.23252,
      "grad_norm": 2.0,
      "grad_norm_var": 0.03022028605143229,
      "learning_rate": 0.0001,
      "loss": 3.9465,
      "loss/crossentropy": 2.2932451367378235,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22147410362958908,
      "step": 11626
    },
    {
      "epoch": 0.23256,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.03144505818684896,
      "learning_rate": 0.0001,
      "loss": 4.0598,
      "loss/crossentropy": 2.1384140253067017,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22141631692647934,
      "step": 11628
    },
    {
      "epoch": 0.2326,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0146728515625,
      "learning_rate": 0.0001,
      "loss": 3.9313,
      "loss/crossentropy": 1.7338963747024536,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18959754705429077,
      "step": 11630
    },
    {
      "epoch": 0.23264,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.014943186442057292,
      "learning_rate": 0.0001,
      "loss": 4.3293,
      "loss/crossentropy": 2.186310887336731,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19735413044691086,
      "step": 11632
    },
    {
      "epoch": 0.23268,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.01092529296875,
      "learning_rate": 0.0001,
      "loss": 3.9636,
      "loss/crossentropy": 2.0566734075546265,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21930715441703796,
      "step": 11634
    },
    {
      "epoch": 0.23272,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.011631011962890625,
      "learning_rate": 0.0001,
      "loss": 3.8382,
      "loss/crossentropy": 1.9993655681610107,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18999667465686798,
      "step": 11636
    },
    {
      "epoch": 0.23276,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008316802978515624,
      "learning_rate": 0.0001,
      "loss": 4.2309,
      "loss/crossentropy": 2.2788418531417847,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21202587336301804,
      "step": 11638
    },
    {
      "epoch": 0.2328,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.00792236328125,
      "learning_rate": 0.0001,
      "loss": 3.7837,
      "loss/crossentropy": 1.7248046398162842,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21424376964569092,
      "step": 11640
    },
    {
      "epoch": 0.23284,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.008235422770182292,
      "learning_rate": 0.0001,
      "loss": 4.0023,
      "loss/crossentropy": 1.9053270816802979,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19046200066804886,
      "step": 11642
    },
    {
      "epoch": 0.23288,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0118408203125,
      "learning_rate": 0.0001,
      "loss": 4.3561,
      "loss/crossentropy": 2.3588117361068726,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23569310456514359,
      "step": 11644
    },
    {
      "epoch": 0.23292,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.012271881103515625,
      "learning_rate": 0.0001,
      "loss": 4.4722,
      "loss/crossentropy": 2.15751576423645,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22107885777950287,
      "step": 11646
    },
    {
      "epoch": 0.23296,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009720611572265624,
      "learning_rate": 0.0001,
      "loss": 4.4075,
      "loss/crossentropy": 2.094591200351715,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20681703090667725,
      "step": 11648
    },
    {
      "epoch": 0.233,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009523264567057292,
      "learning_rate": 0.0001,
      "loss": 3.7832,
      "loss/crossentropy": 1.7536470890045166,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18886109441518784,
      "step": 11650
    },
    {
      "epoch": 0.23304,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0082427978515625,
      "learning_rate": 0.0001,
      "loss": 3.9905,
      "loss/crossentropy": 1.8172362446784973,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18359588831663132,
      "step": 11652
    },
    {
      "epoch": 0.23308,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.00841064453125,
      "learning_rate": 0.0001,
      "loss": 4.2404,
      "loss/crossentropy": 2.2346678376197815,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2154158428311348,
      "step": 11654
    },
    {
      "epoch": 0.23312,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008040110270182291,
      "learning_rate": 0.0001,
      "loss": 4.1559,
      "loss/crossentropy": 2.0573307275772095,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2160855233669281,
      "step": 11656
    },
    {
      "epoch": 0.23316,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.007621256510416666,
      "learning_rate": 0.0001,
      "loss": 4.0669,
      "loss/crossentropy": 2.4358904361724854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22637036442756653,
      "step": 11658
    },
    {
      "epoch": 0.2332,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.004644521077473958,
      "learning_rate": 0.0001,
      "loss": 3.9938,
      "loss/crossentropy": 2.387966513633728,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2385111078619957,
      "step": 11660
    },
    {
      "epoch": 0.23324,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.004709625244140625,
      "learning_rate": 0.0001,
      "loss": 4.3371,
      "loss/crossentropy": 2.30223548412323,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2149546965956688,
      "step": 11662
    },
    {
      "epoch": 0.23328,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.005494944254557292,
      "learning_rate": 0.0001,
      "loss": 4.3242,
      "loss/crossentropy": 2.001839280128479,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2001148834824562,
      "step": 11664
    },
    {
      "epoch": 0.23332,
      "grad_norm": 2.25,
      "grad_norm_var": 0.00693359375,
      "learning_rate": 0.0001,
      "loss": 4.4932,
      "loss/crossentropy": 2.387674927711487,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22714952379465103,
      "step": 11666
    },
    {
      "epoch": 0.23336,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.007470448811848958,
      "learning_rate": 0.0001,
      "loss": 4.1333,
      "loss/crossentropy": 2.133235454559326,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22263716161251068,
      "step": 11668
    },
    {
      "epoch": 0.2334,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.008874257405598959,
      "learning_rate": 0.0001,
      "loss": 4.1392,
      "loss/crossentropy": 2.1041141748428345,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20474642515182495,
      "step": 11670
    },
    {
      "epoch": 0.23344,
      "grad_norm": 2.25,
      "grad_norm_var": 0.010796864827473959,
      "learning_rate": 0.0001,
      "loss": 4.4075,
      "loss/crossentropy": 2.0425861477851868,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23144961893558502,
      "step": 11672
    },
    {
      "epoch": 0.23348,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.009883626302083334,
      "learning_rate": 0.0001,
      "loss": 4.0878,
      "loss/crossentropy": 2.0170212388038635,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2134404480457306,
      "step": 11674
    },
    {
      "epoch": 0.23352,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010489908854166667,
      "learning_rate": 0.0001,
      "loss": 4.2275,
      "loss/crossentropy": 2.205981433391571,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.220575213432312,
      "step": 11676
    },
    {
      "epoch": 0.23356,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.011429595947265624,
      "learning_rate": 0.0001,
      "loss": 4.1845,
      "loss/crossentropy": 1.9121403694152832,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23312295228242874,
      "step": 11678
    },
    {
      "epoch": 0.2336,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.0156890869140625,
      "learning_rate": 0.0001,
      "loss": 4.1112,
      "loss/crossentropy": 1.6793898940086365,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19598360359668732,
      "step": 11680
    },
    {
      "epoch": 0.23364,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.0222564697265625,
      "learning_rate": 0.0001,
      "loss": 4.4814,
      "loss/crossentropy": 2.1657907962799072,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2281472533941269,
      "step": 11682
    },
    {
      "epoch": 0.23368,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.022027333577473957,
      "learning_rate": 0.0001,
      "loss": 4.0869,
      "loss/crossentropy": 2.1918715238571167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22837525606155396,
      "step": 11684
    },
    {
      "epoch": 0.23372,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.020699055989583333,
      "learning_rate": 0.0001,
      "loss": 4.27,
      "loss/crossentropy": 2.296495795249939,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23491691797971725,
      "step": 11686
    },
    {
      "epoch": 0.23376,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.019559733072916665,
      "learning_rate": 0.0001,
      "loss": 4.1422,
      "loss/crossentropy": 1.9278987646102905,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1960090771317482,
      "step": 11688
    },
    {
      "epoch": 0.2338,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.019461822509765626,
      "learning_rate": 0.0001,
      "loss": 4.3393,
      "loss/crossentropy": 2.0491825938224792,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21345915645360947,
      "step": 11690
    },
    {
      "epoch": 0.23384,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.02127863566080729,
      "learning_rate": 0.0001,
      "loss": 3.9209,
      "loss/crossentropy": 1.7594041228294373,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19004464149475098,
      "step": 11692
    },
    {
      "epoch": 0.23388,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.020961252848307292,
      "learning_rate": 0.0001,
      "loss": 4.3645,
      "loss/crossentropy": 2.0313411951065063,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20986144989728928,
      "step": 11694
    },
    {
      "epoch": 0.23392,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.016068522135416666,
      "learning_rate": 0.0001,
      "loss": 4.4644,
      "loss/crossentropy": 2.3580493927001953,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2253532111644745,
      "step": 11696
    },
    {
      "epoch": 0.23396,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.008548990885416666,
      "learning_rate": 0.0001,
      "loss": 4.3954,
      "loss/crossentropy": 2.2036253213882446,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22294757515192032,
      "step": 11698
    },
    {
      "epoch": 0.234,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.007972971598307291,
      "learning_rate": 0.0001,
      "loss": 4.2602,
      "loss/crossentropy": 1.8575093150138855,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19865535199642181,
      "step": 11700
    },
    {
      "epoch": 0.23404,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.008571116129557292,
      "learning_rate": 0.0001,
      "loss": 4.2619,
      "loss/crossentropy": 2.2169028520584106,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22326484322547913,
      "step": 11702
    },
    {
      "epoch": 0.23408,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.007500966389973958,
      "learning_rate": 0.0001,
      "loss": 4.4045,
      "loss/crossentropy": 2.1897542476654053,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2097000628709793,
      "step": 11704
    },
    {
      "epoch": 0.23412,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0063168843587239586,
      "learning_rate": 0.0001,
      "loss": 4.271,
      "loss/crossentropy": 2.214607834815979,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24112706631422043,
      "step": 11706
    },
    {
      "epoch": 0.23416,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.004780832926432292,
      "learning_rate": 0.0001,
      "loss": 4.1278,
      "loss/crossentropy": 2.124355912208557,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2222435548901558,
      "step": 11708
    },
    {
      "epoch": 0.2342,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.0051025390625,
      "learning_rate": 0.0001,
      "loss": 4.3118,
      "loss/crossentropy": 2.35421621799469,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26120802760124207,
      "step": 11710
    },
    {
      "epoch": 0.23424,
      "grad_norm": 2.0,
      "grad_norm_var": 0.00521240234375,
      "learning_rate": 0.0001,
      "loss": 4.1781,
      "loss/crossentropy": 1.9134620428085327,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21830002963542938,
      "step": 11712
    },
    {
      "epoch": 0.23428,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.003763580322265625,
      "learning_rate": 0.0001,
      "loss": 4.2445,
      "loss/crossentropy": 1.9336887001991272,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2025599479675293,
      "step": 11714
    },
    {
      "epoch": 0.23432,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.004133097330729167,
      "learning_rate": 0.0001,
      "loss": 4.0775,
      "loss/crossentropy": 1.9964489936828613,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2020488828420639,
      "step": 11716
    },
    {
      "epoch": 0.23436,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.00423583984375,
      "learning_rate": 0.0001,
      "loss": 4.264,
      "loss/crossentropy": 2.108368992805481,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20985107123851776,
      "step": 11718
    },
    {
      "epoch": 0.2344,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.005008697509765625,
      "learning_rate": 0.0001,
      "loss": 3.9458,
      "loss/crossentropy": 2.0314669013023376,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20942936092615128,
      "step": 11720
    },
    {
      "epoch": 0.23444,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0054107666015625,
      "learning_rate": 0.0001,
      "loss": 4.1098,
      "loss/crossentropy": 2.343130350112915,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.237278014421463,
      "step": 11722
    },
    {
      "epoch": 0.23448,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.010658518473307291,
      "learning_rate": 0.0001,
      "loss": 4.2497,
      "loss/crossentropy": 2.1542173624038696,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22762110829353333,
      "step": 11724
    },
    {
      "epoch": 0.23452,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.009395090738932292,
      "learning_rate": 0.0001,
      "loss": 3.9056,
      "loss/crossentropy": 1.638447105884552,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1707247570157051,
      "step": 11726
    },
    {
      "epoch": 0.23456,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.010573069254557291,
      "learning_rate": 0.0001,
      "loss": 4.0939,
      "loss/crossentropy": 2.1509228944778442,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2091827318072319,
      "step": 11728
    },
    {
      "epoch": 0.2346,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.011677805582682292,
      "learning_rate": 0.0001,
      "loss": 4.325,
      "loss/crossentropy": 2.193585455417633,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2697590962052345,
      "step": 11730
    },
    {
      "epoch": 0.23464,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.012648264567057291,
      "learning_rate": 0.0001,
      "loss": 4.2239,
      "loss/crossentropy": 2.2058298587799072,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21958298981189728,
      "step": 11732
    },
    {
      "epoch": 0.23468,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.012894439697265624,
      "learning_rate": 0.0001,
      "loss": 4.3269,
      "loss/crossentropy": 2.0816246271133423,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20916947722434998,
      "step": 11734
    },
    {
      "epoch": 0.23472,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0150543212890625,
      "learning_rate": 0.0001,
      "loss": 4.1495,
      "loss/crossentropy": 1.838355541229248,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20002590864896774,
      "step": 11736
    },
    {
      "epoch": 0.23476,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.016795857747395834,
      "learning_rate": 0.0001,
      "loss": 4.0559,
      "loss/crossentropy": 2.1602721214294434,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21018436551094055,
      "step": 11738
    },
    {
      "epoch": 0.2348,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.013181304931640625,
      "learning_rate": 0.0001,
      "loss": 4.5376,
      "loss/crossentropy": 2.6967735290527344,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23233920335769653,
      "step": 11740
    },
    {
      "epoch": 0.23484,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.012715403238932292,
      "learning_rate": 0.0001,
      "loss": 3.9772,
      "loss/crossentropy": 2.033313810825348,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21240226924419403,
      "step": 11742
    },
    {
      "epoch": 0.23488,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.012245432535807291,
      "learning_rate": 0.0001,
      "loss": 4.21,
      "loss/crossentropy": 2.123607873916626,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.207147017121315,
      "step": 11744
    },
    {
      "epoch": 0.23492,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.013719685872395833,
      "learning_rate": 0.0001,
      "loss": 4.1121,
      "loss/crossentropy": 2.1173813343048096,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21387682110071182,
      "step": 11746
    },
    {
      "epoch": 0.23496,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.011456044514973958,
      "learning_rate": 0.0001,
      "loss": 4.1219,
      "loss/crossentropy": 2.27209734916687,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2254154533147812,
      "step": 11748
    },
    {
      "epoch": 0.235,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.011156209309895833,
      "learning_rate": 0.0001,
      "loss": 4.3138,
      "loss/crossentropy": 1.844546616077423,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1740722581744194,
      "step": 11750
    },
    {
      "epoch": 0.23504,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.009346516927083333,
      "learning_rate": 0.0001,
      "loss": 4.4615,
      "loss/crossentropy": 2.341967821121216,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23638595640659332,
      "step": 11752
    },
    {
      "epoch": 0.23508,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.009354400634765624,
      "learning_rate": 0.0001,
      "loss": 4.4113,
      "loss/crossentropy": 2.2561213970184326,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23169096559286118,
      "step": 11754
    },
    {
      "epoch": 0.23512,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.007972971598307291,
      "learning_rate": 0.0001,
      "loss": 4.1603,
      "loss/crossentropy": 1.8693158030509949,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19997497648000717,
      "step": 11756
    },
    {
      "epoch": 0.23516,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0085845947265625,
      "learning_rate": 0.0001,
      "loss": 3.9721,
      "loss/crossentropy": 2.2228434085845947,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19779722392559052,
      "step": 11758
    },
    {
      "epoch": 0.2352,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.007899729410807292,
      "learning_rate": 0.0001,
      "loss": 4.2572,
      "loss/crossentropy": 2.177064299583435,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21091558039188385,
      "step": 11760
    },
    {
      "epoch": 0.23524,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.005020904541015625,
      "learning_rate": 0.0001,
      "loss": 4.398,
      "loss/crossentropy": 2.2667617797851562,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2120228409767151,
      "step": 11762
    },
    {
      "epoch": 0.23528,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.004713694254557292,
      "learning_rate": 0.0001,
      "loss": 4.198,
      "loss/crossentropy": 2.0310307145118713,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2194480448961258,
      "step": 11764
    },
    {
      "epoch": 0.23532,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.005033111572265625,
      "learning_rate": 0.0001,
      "loss": 3.9906,
      "loss/crossentropy": 1.8481100797653198,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20181456953287125,
      "step": 11766
    },
    {
      "epoch": 0.23536,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.008642323811848958,
      "learning_rate": 0.0001,
      "loss": 4.3486,
      "loss/crossentropy": 2.254691958427429,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26685942709445953,
      "step": 11768
    },
    {
      "epoch": 0.2354,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.007993316650390625,
      "learning_rate": 0.0001,
      "loss": 4.144,
      "loss/crossentropy": 2.112728714942932,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19724663347005844,
      "step": 11770
    },
    {
      "epoch": 0.23544,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.011195627848307292,
      "learning_rate": 0.0001,
      "loss": 4.089,
      "loss/crossentropy": 2.0724143981933594,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20518633723258972,
      "step": 11772
    },
    {
      "epoch": 0.23548,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010064442952473959,
      "learning_rate": 0.0001,
      "loss": 4.1882,
      "loss/crossentropy": 2.0743810534477234,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2181471511721611,
      "step": 11774
    },
    {
      "epoch": 0.23552,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.011740875244140626,
      "learning_rate": 0.0001,
      "loss": 4.1121,
      "loss/crossentropy": 2.02871835231781,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22081361711025238,
      "step": 11776
    },
    {
      "epoch": 0.23556,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.013185373942057292,
      "learning_rate": 0.0001,
      "loss": 4.5607,
      "loss/crossentropy": 2.1898428201675415,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22084421664476395,
      "step": 11778
    },
    {
      "epoch": 0.2356,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.012748209635416667,
      "learning_rate": 0.0001,
      "loss": 4.2404,
      "loss/crossentropy": 2.0893847346305847,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21616562455892563,
      "step": 11780
    },
    {
      "epoch": 0.23564,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.011769358317057292,
      "learning_rate": 0.0001,
      "loss": 4.215,
      "loss/crossentropy": 1.8590916991233826,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18936381489038467,
      "step": 11782
    },
    {
      "epoch": 0.23568,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.012741851806640624,
      "learning_rate": 0.0001,
      "loss": 3.9544,
      "loss/crossentropy": 2.105339765548706,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22275932878255844,
      "step": 11784
    },
    {
      "epoch": 0.23572,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.011987050374348959,
      "learning_rate": 0.0001,
      "loss": 4.2585,
      "loss/crossentropy": 2.1720080375671387,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20734921097755432,
      "step": 11786
    },
    {
      "epoch": 0.23576,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.012640126546223958,
      "learning_rate": 0.0001,
      "loss": 4.2715,
      "loss/crossentropy": 2.2301958799362183,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21865415573120117,
      "step": 11788
    },
    {
      "epoch": 0.2358,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.013626861572265624,
      "learning_rate": 0.0001,
      "loss": 3.7518,
      "loss/crossentropy": 1.588155210018158,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19334131479263306,
      "step": 11790
    },
    {
      "epoch": 0.23584,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.025248209635416668,
      "learning_rate": 0.0001,
      "loss": 4.4052,
      "loss/crossentropy": 2.208239734172821,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21055688709020615,
      "step": 11792
    },
    {
      "epoch": 0.23588,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.030368804931640625,
      "learning_rate": 0.0001,
      "loss": 4.1996,
      "loss/crossentropy": 2.46909761428833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.225817009806633,
      "step": 11794
    },
    {
      "epoch": 0.23592,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.03367691040039063,
      "learning_rate": 0.0001,
      "loss": 4.2792,
      "loss/crossentropy": 2.022938549518585,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21232923865318298,
      "step": 11796
    },
    {
      "epoch": 0.23596,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.03394953409830729,
      "learning_rate": 0.0001,
      "loss": 4.3949,
      "loss/crossentropy": 2.333058714866638,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23842789232730865,
      "step": 11798
    },
    {
      "epoch": 0.236,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0366363525390625,
      "learning_rate": 0.0001,
      "loss": 3.8698,
      "loss/crossentropy": 1.9570570588111877,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20660093426704407,
      "step": 11800
    },
    {
      "epoch": 0.23604,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0335601806640625,
      "learning_rate": 0.0001,
      "loss": 4.3649,
      "loss/crossentropy": 2.2074697017669678,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23766764998435974,
      "step": 11802
    },
    {
      "epoch": 0.23608,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.03026301066080729,
      "learning_rate": 0.0001,
      "loss": 4.4127,
      "loss/crossentropy": 2.327863335609436,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22967173904180527,
      "step": 11804
    },
    {
      "epoch": 0.23612,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.02802734375,
      "learning_rate": 0.0001,
      "loss": 4.3367,
      "loss/crossentropy": 2.135041356086731,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21708428114652634,
      "step": 11806
    },
    {
      "epoch": 0.23616,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.00665283203125,
      "learning_rate": 0.0001,
      "loss": 3.9805,
      "loss/crossentropy": 1.9111011624336243,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20122328400611877,
      "step": 11808
    },
    {
      "epoch": 0.2362,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.006436920166015625,
      "learning_rate": 0.0001,
      "loss": 4.19,
      "loss/crossentropy": 2.0823878049850464,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23593349009752274,
      "step": 11810
    },
    {
      "epoch": 0.23624,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0069000244140625,
      "learning_rate": 0.0001,
      "loss": 3.9554,
      "loss/crossentropy": 2.298948645591736,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22076356410980225,
      "step": 11812
    },
    {
      "epoch": 0.23628,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.006483713785807292,
      "learning_rate": 0.0001,
      "loss": 4.1119,
      "loss/crossentropy": 2.383033037185669,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20850034803152084,
      "step": 11814
    },
    {
      "epoch": 0.23632,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.10857645670572917,
      "learning_rate": 0.0001,
      "loss": 4.3648,
      "loss/crossentropy": 2.004193425178528,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27853623032569885,
      "step": 11816
    },
    {
      "epoch": 0.23636,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.11877339680989583,
      "learning_rate": 0.0001,
      "loss": 4.4454,
      "loss/crossentropy": 1.9110660552978516,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21773407608270645,
      "step": 11818
    },
    {
      "epoch": 0.2364,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.11894505818684896,
      "learning_rate": 0.0001,
      "loss": 4.5428,
      "loss/crossentropy": 2.300672471523285,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23710736632347107,
      "step": 11820
    },
    {
      "epoch": 0.23644,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.11943333943684896,
      "learning_rate": 0.0001,
      "loss": 4.1413,
      "loss/crossentropy": 2.270769238471985,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2182588130235672,
      "step": 11822
    },
    {
      "epoch": 0.23648,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.11885960896809895,
      "learning_rate": 0.0001,
      "loss": 4.2744,
      "loss/crossentropy": 2.085016667842865,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19494660943746567,
      "step": 11824
    },
    {
      "epoch": 0.23652,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.11644261678059896,
      "learning_rate": 0.0001,
      "loss": 4.5554,
      "loss/crossentropy": 2.384607672691345,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22075054794549942,
      "step": 11826
    },
    {
      "epoch": 0.23656,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.11926167805989583,
      "learning_rate": 0.0001,
      "loss": 3.9076,
      "loss/crossentropy": 2.1217936277389526,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2126227766275406,
      "step": 11828
    },
    {
      "epoch": 0.2366,
      "grad_norm": 1.875,
      "grad_norm_var": 0.12241185506184896,
      "learning_rate": 0.0001,
      "loss": 3.8285,
      "loss/crossentropy": 2.1690168380737305,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20450890809297562,
      "step": 11830
    },
    {
      "epoch": 0.23664,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.031998697916666666,
      "learning_rate": 0.0001,
      "loss": 4.2817,
      "loss/crossentropy": 2.2614429593086243,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22817185521125793,
      "step": 11832
    },
    {
      "epoch": 0.23668,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.023545074462890624,
      "learning_rate": 0.0001,
      "loss": 4.3074,
      "loss/crossentropy": 2.177332043647766,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.226592555642128,
      "step": 11834
    },
    {
      "epoch": 0.23672,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.01765925089518229,
      "learning_rate": 0.0001,
      "loss": 4.0471,
      "loss/crossentropy": 1.947661578655243,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1762884557247162,
      "step": 11836
    },
    {
      "epoch": 0.23676,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.018184153238932292,
      "learning_rate": 0.0001,
      "loss": 3.9014,
      "loss/crossentropy": 1.7112661004066467,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18696465343236923,
      "step": 11838
    },
    {
      "epoch": 0.2368,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.017439524332682293,
      "learning_rate": 0.0001,
      "loss": 4.1761,
      "loss/crossentropy": 2.0478790402412415,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21436551213264465,
      "step": 11840
    },
    {
      "epoch": 0.23684,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.01718724568684896,
      "learning_rate": 0.0001,
      "loss": 4.1702,
      "loss/crossentropy": 2.0541738867759705,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23121927678585052,
      "step": 11842
    },
    {
      "epoch": 0.23688,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.01876805623372396,
      "learning_rate": 0.0001,
      "loss": 4.208,
      "loss/crossentropy": 2.04353004693985,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22237974405288696,
      "step": 11844
    },
    {
      "epoch": 0.23692,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.017488606770833335,
      "learning_rate": 0.0001,
      "loss": 4.4641,
      "loss/crossentropy": 2.110148549079895,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23324476927518845,
      "step": 11846
    },
    {
      "epoch": 0.23696,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.018623860677083333,
      "learning_rate": 0.0001,
      "loss": 4.0831,
      "loss/crossentropy": 2.207589864730835,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.209608756005764,
      "step": 11848
    },
    {
      "epoch": 0.237,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.012180328369140625,
      "learning_rate": 0.0001,
      "loss": 4.1896,
      "loss/crossentropy": 2.2447429895401,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2016080766916275,
      "step": 11850
    },
    {
      "epoch": 0.23704,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009771474202473958,
      "learning_rate": 0.0001,
      "loss": 4.3133,
      "loss/crossentropy": 2.3349474668502808,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22105249762535095,
      "step": 11852
    },
    {
      "epoch": 0.23708,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.008024088541666667,
      "learning_rate": 0.0001,
      "loss": 4.1091,
      "loss/crossentropy": 1.8444748520851135,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1993526816368103,
      "step": 11854
    },
    {
      "epoch": 0.23712,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.009323883056640624,
      "learning_rate": 0.0001,
      "loss": 4.0758,
      "loss/crossentropy": 1.9181615710258484,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18698236346244812,
      "step": 11856
    },
    {
      "epoch": 0.23716,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.009627024332682291,
      "learning_rate": 0.0001,
      "loss": 4.1454,
      "loss/crossentropy": 2.1930960416793823,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2122444212436676,
      "step": 11858
    },
    {
      "epoch": 0.2372,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.009439849853515625,
      "learning_rate": 0.0001,
      "loss": 4.3177,
      "loss/crossentropy": 1.7835432887077332,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20900961011648178,
      "step": 11860
    },
    {
      "epoch": 0.23724,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.011775461832682292,
      "learning_rate": 0.0001,
      "loss": 4.6362,
      "loss/crossentropy": 2.1839439868927,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21574077755212784,
      "step": 11862
    },
    {
      "epoch": 0.23728,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.010791015625,
      "learning_rate": 0.0001,
      "loss": 3.8916,
      "loss/crossentropy": 1.9817007184028625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21773921698331833,
      "step": 11864
    },
    {
      "epoch": 0.23732,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.008923085530598958,
      "learning_rate": 0.0001,
      "loss": 4.2065,
      "loss/crossentropy": 2.017501652240753,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2247873619198799,
      "step": 11866
    },
    {
      "epoch": 0.23736,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.010992177327473958,
      "learning_rate": 0.0001,
      "loss": 3.8813,
      "loss/crossentropy": 2.0779114961624146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21136770397424698,
      "step": 11868
    },
    {
      "epoch": 0.2374,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.010497029622395833,
      "learning_rate": 0.0001,
      "loss": 4.341,
      "loss/crossentropy": 2.3987231254577637,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2326417714357376,
      "step": 11870
    },
    {
      "epoch": 0.23744,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.008975982666015625,
      "learning_rate": 0.0001,
      "loss": 4.0466,
      "loss/crossentropy": 1.7145346999168396,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19222228974103928,
      "step": 11872
    },
    {
      "epoch": 0.23748,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008829498291015625,
      "learning_rate": 0.0001,
      "loss": 4.2683,
      "loss/crossentropy": 2.378560423851013,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21374844759702682,
      "step": 11874
    },
    {
      "epoch": 0.23752,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0073626200358072914,
      "learning_rate": 0.0001,
      "loss": 4.2548,
      "loss/crossentropy": 1.951455295085907,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2217455804347992,
      "step": 11876
    },
    {
      "epoch": 0.23756,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.00513916015625,
      "learning_rate": 0.0001,
      "loss": 3.8867,
      "loss/crossentropy": 1.9309074878692627,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2100483626127243,
      "step": 11878
    },
    {
      "epoch": 0.2376,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.00445556640625,
      "learning_rate": 0.0001,
      "loss": 4.102,
      "loss/crossentropy": 2.2295292615890503,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2196313813328743,
      "step": 11880
    },
    {
      "epoch": 0.23764,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.004964192708333333,
      "learning_rate": 0.0001,
      "loss": 4.0514,
      "loss/crossentropy": 2.0665449500083923,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20752248913049698,
      "step": 11882
    },
    {
      "epoch": 0.23768,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.00560302734375,
      "learning_rate": 0.0001,
      "loss": 4.4022,
      "loss/crossentropy": 1.9645958542823792,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20391641557216644,
      "step": 11884
    },
    {
      "epoch": 0.23772,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.005956013997395833,
      "learning_rate": 0.0001,
      "loss": 4.1698,
      "loss/crossentropy": 2.1280174255371094,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21947231888771057,
      "step": 11886
    },
    {
      "epoch": 0.23776,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0073964436848958336,
      "learning_rate": 0.0001,
      "loss": 4.2043,
      "loss/crossentropy": 1.9462851285934448,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1961178332567215,
      "step": 11888
    },
    {
      "epoch": 0.2378,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.009422810872395833,
      "learning_rate": 0.0001,
      "loss": 4.4846,
      "loss/crossentropy": 2.249086618423462,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22473402321338654,
      "step": 11890
    },
    {
      "epoch": 0.23784,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.010472615559895834,
      "learning_rate": 0.0001,
      "loss": 4.1368,
      "loss/crossentropy": 2.1972378492355347,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2164871245622635,
      "step": 11892
    },
    {
      "epoch": 0.23788,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.009409332275390625,
      "learning_rate": 0.0001,
      "loss": 4.228,
      "loss/crossentropy": 2.0067209601402283,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20195162296295166,
      "step": 11894
    },
    {
      "epoch": 0.23792,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009673817952473959,
      "learning_rate": 0.0001,
      "loss": 4.1767,
      "loss/crossentropy": 1.9103696942329407,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19616805016994476,
      "step": 11896
    },
    {
      "epoch": 0.23796,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.011595662434895833,
      "learning_rate": 0.0001,
      "loss": 4.228,
      "loss/crossentropy": 2.1524049639701843,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21107815951108932,
      "step": 11898
    },
    {
      "epoch": 0.238,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0110260009765625,
      "learning_rate": 0.0001,
      "loss": 4.183,
      "loss/crossentropy": 2.2112287878990173,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2099093720316887,
      "step": 11900
    },
    {
      "epoch": 0.23804,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.8914347330729167,
      "learning_rate": 0.0001,
      "loss": 4.647,
      "loss/crossentropy": 2.2945204973220825,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2713918164372444,
      "step": 11902
    },
    {
      "epoch": 0.23808,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.8781575520833333,
      "learning_rate": 0.0001,
      "loss": 4.4699,
      "loss/crossentropy": 2.1838968992233276,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22640438377857208,
      "step": 11904
    },
    {
      "epoch": 0.23812,
      "grad_norm": 2.0,
      "grad_norm_var": 0.89010009765625,
      "learning_rate": 0.0001,
      "loss": 4.0843,
      "loss/crossentropy": 1.9089699983596802,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19553899765014648,
      "step": 11906
    },
    {
      "epoch": 0.23816,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.8873443603515625,
      "learning_rate": 0.0001,
      "loss": 4.2573,
      "loss/crossentropy": 2.105385661125183,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21528839319944382,
      "step": 11908
    },
    {
      "epoch": 0.2382,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.8921946207682292,
      "learning_rate": 0.0001,
      "loss": 4.0381,
      "loss/crossentropy": 1.565223515033722,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17765602469444275,
      "step": 11910
    },
    {
      "epoch": 0.23824,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.8919016520182291,
      "learning_rate": 0.0001,
      "loss": 4.3014,
      "loss/crossentropy": 2.220748543739319,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21904101967811584,
      "step": 11912
    },
    {
      "epoch": 0.23828,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.889306640625,
      "learning_rate": 0.0001,
      "loss": 4.2965,
      "loss/crossentropy": 2.294031500816345,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21700909733772278,
      "step": 11914
    },
    {
      "epoch": 0.23832,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.8795237223307292,
      "learning_rate": 0.0001,
      "loss": 4.3439,
      "loss/crossentropy": 2.072917103767395,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20462248474359512,
      "step": 11916
    },
    {
      "epoch": 0.23836,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.00533447265625,
      "learning_rate": 0.0001,
      "loss": 4.1843,
      "loss/crossentropy": 2.0541720390319824,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2127074897289276,
      "step": 11918
    },
    {
      "epoch": 0.2384,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0046539306640625,
      "learning_rate": 0.0001,
      "loss": 3.8922,
      "loss/crossentropy": 1.9873629808425903,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19215020537376404,
      "step": 11920
    },
    {
      "epoch": 0.23844,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0032135009765625,
      "learning_rate": 0.0001,
      "loss": 4.1943,
      "loss/crossentropy": 2.447067141532898,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2318919375538826,
      "step": 11922
    },
    {
      "epoch": 0.23848,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0038157145182291666,
      "learning_rate": 0.0001,
      "loss": 3.8756,
      "loss/crossentropy": 1.9918025732040405,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20865648984909058,
      "step": 11924
    },
    {
      "epoch": 0.23852,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0026041666666666665,
      "learning_rate": 0.0001,
      "loss": 4.3243,
      "loss/crossentropy": 2.0803070068359375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21753622591495514,
      "step": 11926
    },
    {
      "epoch": 0.23856,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.002976226806640625,
      "learning_rate": 0.0001,
      "loss": 4.1828,
      "loss/crossentropy": 2.118411421775818,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21741919964551926,
      "step": 11928
    },
    {
      "epoch": 0.2386,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0034075419108072916,
      "learning_rate": 0.0001,
      "loss": 4.0056,
      "loss/crossentropy": 1.8924900889396667,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20609601587057114,
      "step": 11930
    },
    {
      "epoch": 0.23864,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.002418772379557292,
      "learning_rate": 0.0001,
      "loss": 4.0012,
      "loss/crossentropy": 1.743731439113617,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19873473048210144,
      "step": 11932
    },
    {
      "epoch": 0.23868,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.002929433186848958,
      "learning_rate": 0.0001,
      "loss": 4.2564,
      "loss/crossentropy": 2.291381061077118,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21248165518045425,
      "step": 11934
    },
    {
      "epoch": 0.23872,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.007452138264973958,
      "learning_rate": 0.0001,
      "loss": 4.3938,
      "loss/crossentropy": 1.7672501802444458,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19201595336198807,
      "step": 11936
    },
    {
      "epoch": 0.23876,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.008162434895833333,
      "learning_rate": 0.0001,
      "loss": 3.8693,
      "loss/crossentropy": 1.8719280362129211,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1960783526301384,
      "step": 11938
    },
    {
      "epoch": 0.2388,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.010978190104166667,
      "learning_rate": 0.0001,
      "loss": 4.3345,
      "loss/crossentropy": 1.8103876113891602,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1964937001466751,
      "step": 11940
    },
    {
      "epoch": 0.23884,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.011901601155598959,
      "learning_rate": 0.0001,
      "loss": 4.2134,
      "loss/crossentropy": 1.8743855953216553,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19404225796461105,
      "step": 11942
    },
    {
      "epoch": 0.23888,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.012562815348307292,
      "learning_rate": 0.0001,
      "loss": 4.2379,
      "loss/crossentropy": 2.3096803426742554,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23256508260965347,
      "step": 11944
    },
    {
      "epoch": 0.23892,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.012123362223307291,
      "learning_rate": 0.0001,
      "loss": 4.5089,
      "loss/crossentropy": 2.207027554512024,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20778407156467438,
      "step": 11946
    },
    {
      "epoch": 0.23896,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011671702067057291,
      "learning_rate": 0.0001,
      "loss": 4.0423,
      "loss/crossentropy": 1.845999002456665,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19381015002727509,
      "step": 11948
    },
    {
      "epoch": 0.239,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.011628977457682292,
      "learning_rate": 0.0001,
      "loss": 4.1054,
      "loss/crossentropy": 1.677983045578003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17493421584367752,
      "step": 11950
    },
    {
      "epoch": 0.23904,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.008107248942057292,
      "learning_rate": 0.0001,
      "loss": 4.1729,
      "loss/crossentropy": 2.1798466444015503,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22395183145999908,
      "step": 11952
    },
    {
      "epoch": 0.23908,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.013525390625,
      "learning_rate": 0.0001,
      "loss": 4.0162,
      "loss/crossentropy": 2.008498191833496,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21069375425577164,
      "step": 11954
    },
    {
      "epoch": 0.23912,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.014546712239583334,
      "learning_rate": 0.0001,
      "loss": 4.3736,
      "loss/crossentropy": 2.1712071895599365,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21842175722122192,
      "step": 11956
    },
    {
      "epoch": 0.23916,
      "grad_norm": 2.578125,
      "grad_norm_var": 0.03242365519205729,
      "learning_rate": 0.0001,
      "loss": 4.1338,
      "loss/crossentropy": 2.2769562005996704,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22064895182847977,
      "step": 11958
    },
    {
      "epoch": 0.2392,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.032572428385416664,
      "learning_rate": 0.0001,
      "loss": 4.3401,
      "loss/crossentropy": 2.127632260322571,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2139139398932457,
      "step": 11960
    },
    {
      "epoch": 0.23924,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.03223368326822917,
      "learning_rate": 0.0001,
      "loss": 4.3547,
      "loss/crossentropy": 2.2687970399856567,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23024404793977737,
      "step": 11962
    },
    {
      "epoch": 0.23928,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.032136027018229166,
      "learning_rate": 0.0001,
      "loss": 4.3532,
      "loss/crossentropy": 2.2271196246147156,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22528529912233353,
      "step": 11964
    },
    {
      "epoch": 0.23932,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.032136027018229166,
      "learning_rate": 0.0001,
      "loss": 4.1898,
      "loss/crossentropy": 2.3209941387176514,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21150042116641998,
      "step": 11966
    },
    {
      "epoch": 0.23936,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.03208719889322917,
      "learning_rate": 0.0001,
      "loss": 4.197,
      "loss/crossentropy": 1.9936136603355408,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22582116723060608,
      "step": 11968
    },
    {
      "epoch": 0.2394,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.024055989583333333,
      "learning_rate": 0.0001,
      "loss": 3.9045,
      "loss/crossentropy": 1.7509311437606812,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19028881192207336,
      "step": 11970
    },
    {
      "epoch": 0.23944,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.02213134765625,
      "learning_rate": 0.0001,
      "loss": 4.2365,
      "loss/crossentropy": 2.2120620012283325,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23357221484184265,
      "step": 11972
    },
    {
      "epoch": 0.23948,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.005686187744140625,
      "learning_rate": 0.0001,
      "loss": 4.0129,
      "loss/crossentropy": 1.794329285621643,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17914240062236786,
      "step": 11974
    },
    {
      "epoch": 0.23952,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.010550689697265626,
      "learning_rate": 0.0001,
      "loss": 4.1347,
      "loss/crossentropy": 2.1647136211395264,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21009670197963715,
      "step": 11976
    },
    {
      "epoch": 0.23956,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.010660552978515625,
      "learning_rate": 0.0001,
      "loss": 4.2365,
      "loss/crossentropy": 1.674091637134552,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19617585837841034,
      "step": 11978
    },
    {
      "epoch": 0.2396,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011739095052083334,
      "learning_rate": 0.0001,
      "loss": 4.2474,
      "loss/crossentropy": 1.9591755867004395,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18309858441352844,
      "step": 11980
    },
    {
      "epoch": 0.23964,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.017862955729166668,
      "learning_rate": 0.0001,
      "loss": 4.3642,
      "loss/crossentropy": 2.0550093054771423,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22483228147029877,
      "step": 11982
    },
    {
      "epoch": 0.23968,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.019291178385416666,
      "learning_rate": 0.0001,
      "loss": 4.3937,
      "loss/crossentropy": 2.264032781124115,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2101321816444397,
      "step": 11984
    },
    {
      "epoch": 0.23972,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.020213826497395834,
      "learning_rate": 0.0001,
      "loss": 4.4341,
      "loss/crossentropy": 2.06082820892334,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2090815082192421,
      "step": 11986
    },
    {
      "epoch": 0.23976,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0197662353515625,
      "learning_rate": 0.0001,
      "loss": 4.4159,
      "loss/crossentropy": 2.248009443283081,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22642739117145538,
      "step": 11988
    },
    {
      "epoch": 0.2398,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.02211278279622396,
      "learning_rate": 0.0001,
      "loss": 3.9601,
      "loss/crossentropy": 1.797426462173462,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18600185215473175,
      "step": 11990
    },
    {
      "epoch": 0.23984,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.01573460896809896,
      "learning_rate": 0.0001,
      "loss": 4.1666,
      "loss/crossentropy": 2.5529314279556274,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23957456648349762,
      "step": 11992
    },
    {
      "epoch": 0.23988,
      "grad_norm": 2.125,
      "grad_norm_var": 0.015933990478515625,
      "learning_rate": 0.0001,
      "loss": 4.2971,
      "loss/crossentropy": 1.9974916577339172,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23130090534687042,
      "step": 11994
    },
    {
      "epoch": 0.23992,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.014679972330729167,
      "learning_rate": 0.0001,
      "loss": 4.0358,
      "loss/crossentropy": 1.9714577794075012,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1969093456864357,
      "step": 11996
    },
    {
      "epoch": 0.23996,
      "grad_norm": 2.25,
      "grad_norm_var": 0.011055501302083333,
      "learning_rate": 0.0001,
      "loss": 4.282,
      "loss/crossentropy": 1.8047285079956055,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19191773235797882,
      "step": 11998
    },
    {
      "epoch": 0.24,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.009187825520833333,
      "learning_rate": 0.0001,
      "loss": 4.4078,
      "loss/crossentropy": 1.9240365028381348,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19021467864513397,
      "step": 12000
    },
    {
      "epoch": 0.24004,
      "grad_norm": 2.125,
      "grad_norm_var": 0.009041086832682291,
      "learning_rate": 0.0001,
      "loss": 4.1657,
      "loss/crossentropy": 1.6503748297691345,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18838192522525787,
      "step": 12002
    },
    {
      "epoch": 0.24008,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.009368642171223959,
      "learning_rate": 0.0001,
      "loss": 4.3451,
      "loss/crossentropy": 2.238506555557251,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2139800265431404,
      "step": 12004
    },
    {
      "epoch": 0.24012,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.008695220947265625,
      "learning_rate": 0.0001,
      "loss": 4.0978,
      "loss/crossentropy": 1.8623422384262085,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19039735198020935,
      "step": 12006
    },
    {
      "epoch": 0.24016,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0096588134765625,
      "learning_rate": 0.0001,
      "loss": 3.9741,
      "loss/crossentropy": 1.975899577140808,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20808710902929306,
      "step": 12008
    },
    {
      "epoch": 0.2402,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.0108642578125,
      "learning_rate": 0.0001,
      "loss": 4.4332,
      "loss/crossentropy": 2.2959831953048706,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23725779354572296,
      "step": 12010
    },
    {
      "epoch": 0.24024,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.015130360921223959,
      "learning_rate": 0.0001,
      "loss": 3.9117,
      "loss/crossentropy": 2.1626380681991577,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2068287953734398,
      "step": 12012
    },
    {
      "epoch": 0.24028,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.013474273681640624,
      "learning_rate": 0.0001,
      "loss": 4.1614,
      "loss/crossentropy": 1.925924837589264,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2097126841545105,
      "step": 12014
    },
    {
      "epoch": 0.24032,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.018302154541015626,
      "learning_rate": 0.0001,
      "loss": 4.5545,
      "loss/crossentropy": 2.147459626197815,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2345646619796753,
      "step": 12016
    },
    {
      "epoch": 0.24036,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.021418253580729168,
      "learning_rate": 0.0001,
      "loss": 4.0219,
      "loss/crossentropy": 2.052124857902527,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2058626338839531,
      "step": 12018
    },
    {
      "epoch": 0.2404,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.021329752604166665,
      "learning_rate": 0.0001,
      "loss": 3.8619,
      "loss/crossentropy": 2.377007842063904,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2124926745891571,
      "step": 12020
    },
    {
      "epoch": 0.24044,
      "grad_norm": 2.0,
      "grad_norm_var": 0.020173136393229166,
      "learning_rate": 0.0001,
      "loss": 3.9542,
      "loss/crossentropy": 1.9117819666862488,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1976870447397232,
      "step": 12022
    },
    {
      "epoch": 0.24048,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.021478017171223957,
      "learning_rate": 0.0001,
      "loss": 4.5771,
      "loss/crossentropy": 2.439339756965637,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22178436815738678,
      "step": 12024
    },
    {
      "epoch": 0.24052,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.02029596964518229,
      "learning_rate": 0.0001,
      "loss": 4.1182,
      "loss/crossentropy": 1.6923209428787231,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19288206100463867,
      "step": 12026
    },
    {
      "epoch": 0.24056,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.016434478759765624,
      "learning_rate": 0.0001,
      "loss": 4.1733,
      "loss/crossentropy": 2.071919083595276,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21092405915260315,
      "step": 12028
    },
    {
      "epoch": 0.2406,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.016993967692057292,
      "learning_rate": 0.0001,
      "loss": 3.9688,
      "loss/crossentropy": 1.994953691959381,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19953829050064087,
      "step": 12030
    },
    {
      "epoch": 0.24064,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.011201731363932292,
      "learning_rate": 0.0001,
      "loss": 4.3252,
      "loss/crossentropy": 2.0545560121536255,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20861412584781647,
      "step": 12032
    },
    {
      "epoch": 0.24068,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.0103424072265625,
      "learning_rate": 0.0001,
      "loss": 4.0859,
      "loss/crossentropy": 2.0211291909217834,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2146567404270172,
      "step": 12034
    },
    {
      "epoch": 0.24072,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.012312825520833333,
      "learning_rate": 0.0001,
      "loss": 4.3156,
      "loss/crossentropy": 2.165773868560791,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21522508561611176,
      "step": 12036
    },
    {
      "epoch": 0.24076,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.013216145833333333,
      "learning_rate": 0.0001,
      "loss": 4.2253,
      "loss/crossentropy": 2.0272024273872375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20012138038873672,
      "step": 12038
    },
    {
      "epoch": 0.2408,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.011946360270182291,
      "learning_rate": 0.0001,
      "loss": 4.2419,
      "loss/crossentropy": 1.9311429262161255,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19755128771066666,
      "step": 12040
    },
    {
      "epoch": 0.24084,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.012086741129557292,
      "learning_rate": 0.0001,
      "loss": 4.1857,
      "loss/crossentropy": 2.0740894079208374,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20651141554117203,
      "step": 12042
    },
    {
      "epoch": 0.24088,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.012876129150390625,
      "learning_rate": 0.0001,
      "loss": 3.9882,
      "loss/crossentropy": 2.215203881263733,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21735452860593796,
      "step": 12044
    },
    {
      "epoch": 0.24092,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.02371190388997396,
      "learning_rate": 0.0001,
      "loss": 4.3755,
      "loss/crossentropy": 1.9994693994522095,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22001128643751144,
      "step": 12046
    },
    {
      "epoch": 0.24096,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.025187174479166668,
      "learning_rate": 0.0001,
      "loss": 4.1145,
      "loss/crossentropy": 2.195701003074646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20777013897895813,
      "step": 12048
    },
    {
      "epoch": 0.241,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.02474950154622396,
      "learning_rate": 0.0001,
      "loss": 4.3486,
      "loss/crossentropy": 2.1975715160369873,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21987678855657578,
      "step": 12050
    },
    {
      "epoch": 0.24104,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.02188695271809896,
      "learning_rate": 0.0001,
      "loss": 4.1337,
      "loss/crossentropy": 2.225023865699768,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24454358220100403,
      "step": 12052
    },
    {
      "epoch": 0.24108,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.022200520833333334,
      "learning_rate": 0.0001,
      "loss": 4.0404,
      "loss/crossentropy": 1.892149806022644,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2110423520207405,
      "step": 12054
    },
    {
      "epoch": 0.24112,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.023579915364583332,
      "learning_rate": 0.0001,
      "loss": 4.1604,
      "loss/crossentropy": 1.9996158480644226,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1821269765496254,
      "step": 12056
    },
    {
      "epoch": 0.24116,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.02569580078125,
      "learning_rate": 0.0001,
      "loss": 4.0591,
      "loss/crossentropy": 2.0965115427970886,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2140883505344391,
      "step": 12058
    },
    {
      "epoch": 0.2412,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0256500244140625,
      "learning_rate": 0.0001,
      "loss": 3.9494,
      "loss/crossentropy": 1.9169449210166931,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2034926936030388,
      "step": 12060
    },
    {
      "epoch": 0.24124,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.015135701497395833,
      "learning_rate": 0.0001,
      "loss": 4.5,
      "loss/crossentropy": 2.3266680240631104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24314038455486298,
      "step": 12062
    },
    {
      "epoch": 0.24128,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.013509114583333334,
      "learning_rate": 0.0001,
      "loss": 4.1009,
      "loss/crossentropy": 2.0337759256362915,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20842785388231277,
      "step": 12064
    },
    {
      "epoch": 0.24132,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.013618977864583333,
      "learning_rate": 0.0001,
      "loss": 3.9186,
      "loss/crossentropy": 1.9502894878387451,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1812899112701416,
      "step": 12066
    },
    {
      "epoch": 0.24136,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.013822174072265625,
      "learning_rate": 0.0001,
      "loss": 4.269,
      "loss/crossentropy": 2.13326895236969,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21872679144144058,
      "step": 12068
    },
    {
      "epoch": 0.2414,
      "grad_norm": 2.125,
      "grad_norm_var": 0.013304646809895833,
      "learning_rate": 0.0001,
      "loss": 4.3594,
      "loss/crossentropy": 2.0836809873580933,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.212929405272007,
      "step": 12070
    },
    {
      "epoch": 0.24144,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.012516276041666666,
      "learning_rate": 0.0001,
      "loss": 4.0467,
      "loss/crossentropy": 2.207913398742676,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20640508085489273,
      "step": 12072
    },
    {
      "epoch": 0.24148,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.01080322265625,
      "learning_rate": 0.0001,
      "loss": 4.1877,
      "loss/crossentropy": 2.0770451426506042,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20633003860712051,
      "step": 12074
    },
    {
      "epoch": 0.24152,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.010786946614583333,
      "learning_rate": 0.0001,
      "loss": 4.2237,
      "loss/crossentropy": 2.0042858719825745,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18523608148097992,
      "step": 12076
    },
    {
      "epoch": 0.24156,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.010141754150390625,
      "learning_rate": 0.0001,
      "loss": 4.2755,
      "loss/crossentropy": 1.8673237562179565,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19267796725034714,
      "step": 12078
    },
    {
      "epoch": 0.2416,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.013793690999348959,
      "learning_rate": 0.0001,
      "loss": 3.9766,
      "loss/crossentropy": 2.0686238408088684,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19344990700483322,
      "step": 12080
    },
    {
      "epoch": 0.24164,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009504954020182291,
      "learning_rate": 0.0001,
      "loss": 4.0366,
      "loss/crossentropy": 2.0069685578346252,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1971682757139206,
      "step": 12082
    },
    {
      "epoch": 0.24168,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009468587239583333,
      "learning_rate": 0.0001,
      "loss": 4.1975,
      "loss/crossentropy": 2.2052754163742065,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20646511763334274,
      "step": 12084
    },
    {
      "epoch": 0.24172,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.008983357747395834,
      "learning_rate": 0.0001,
      "loss": 4.1362,
      "loss/crossentropy": 2.0343621373176575,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19575881958007812,
      "step": 12086
    },
    {
      "epoch": 0.24176,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.009894816080729167,
      "learning_rate": 0.0001,
      "loss": 3.9695,
      "loss/crossentropy": 2.2500641345977783,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23151995986700058,
      "step": 12088
    },
    {
      "epoch": 0.2418,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.0375640869140625,
      "learning_rate": 0.0001,
      "loss": 4.4361,
      "loss/crossentropy": 2.2596821784973145,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22305716574192047,
      "step": 12090
    },
    {
      "epoch": 0.24184,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.03715184529622396,
      "learning_rate": 0.0001,
      "loss": 4.0381,
      "loss/crossentropy": 2.0940088033676147,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23409207165241241,
      "step": 12092
    },
    {
      "epoch": 0.24188,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.03586018880208333,
      "learning_rate": 0.0001,
      "loss": 3.9715,
      "loss/crossentropy": 1.6980834603309631,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19339510053396225,
      "step": 12094
    },
    {
      "epoch": 0.24192,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.03047459920247396,
      "learning_rate": 0.0001,
      "loss": 4.192,
      "loss/crossentropy": 2.1921679973602295,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21085387468338013,
      "step": 12096
    },
    {
      "epoch": 0.24196,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.03551839192708333,
      "learning_rate": 0.0001,
      "loss": 4.3551,
      "loss/crossentropy": 2.18610817193985,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2332247570157051,
      "step": 12098
    },
    {
      "epoch": 0.242,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.038852691650390625,
      "learning_rate": 0.0001,
      "loss": 4.1216,
      "loss/crossentropy": 2.08358097076416,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21169763058423996,
      "step": 12100
    },
    {
      "epoch": 0.24204,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.038913726806640625,
      "learning_rate": 0.0001,
      "loss": 4.2339,
      "loss/crossentropy": 2.1001542806625366,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2132650762796402,
      "step": 12102
    },
    {
      "epoch": 0.24208,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.033614095052083334,
      "learning_rate": 0.0001,
      "loss": 4.2051,
      "loss/crossentropy": 1.9031851887702942,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2042345404624939,
      "step": 12104
    },
    {
      "epoch": 0.24212,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0127838134765625,
      "learning_rate": 0.0001,
      "loss": 4.0341,
      "loss/crossentropy": 1.670085072517395,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1878880336880684,
      "step": 12106
    },
    {
      "epoch": 0.24216,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.012914784749348958,
      "learning_rate": 0.0001,
      "loss": 4.3395,
      "loss/crossentropy": 1.9557109475135803,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2211412936449051,
      "step": 12108
    },
    {
      "epoch": 0.2422,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.013637034098307292,
      "learning_rate": 0.0001,
      "loss": 4.0107,
      "loss/crossentropy": 2.085852086544037,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21336784213781357,
      "step": 12110
    },
    {
      "epoch": 0.24224,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.015721638997395832,
      "learning_rate": 0.0001,
      "loss": 4.0321,
      "loss/crossentropy": 2.1064602732658386,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2096766158938408,
      "step": 12112
    },
    {
      "epoch": 0.24228,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.009325917561848958,
      "learning_rate": 0.0001,
      "loss": 4.2998,
      "loss/crossentropy": 2.381577968597412,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23788201808929443,
      "step": 12114
    },
    {
      "epoch": 0.24232,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.006758626302083333,
      "learning_rate": 0.0001,
      "loss": 4.0687,
      "loss/crossentropy": 1.7200234532356262,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18620850890874863,
      "step": 12116
    },
    {
      "epoch": 0.24236,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.007697550455729166,
      "learning_rate": 0.0001,
      "loss": 4.2672,
      "loss/crossentropy": 1.9106165170669556,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21456392109394073,
      "step": 12118
    },
    {
      "epoch": 0.2424,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.007106272379557291,
      "learning_rate": 0.0001,
      "loss": 4.04,
      "loss/crossentropy": 1.9852410554885864,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1927378550171852,
      "step": 12120
    },
    {
      "epoch": 0.24244,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.007991282145182292,
      "learning_rate": 0.0001,
      "loss": 4.1278,
      "loss/crossentropy": 1.4948370456695557,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1698828637599945,
      "step": 12122
    },
    {
      "epoch": 0.24248,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0078857421875,
      "learning_rate": 0.0001,
      "loss": 4.2874,
      "loss/crossentropy": 1.956885814666748,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19851476699113846,
      "step": 12124
    },
    {
      "epoch": 0.24252,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.009422810872395833,
      "learning_rate": 0.0001,
      "loss": 3.8709,
      "loss/crossentropy": 1.9052257537841797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18438522517681122,
      "step": 12126
    },
    {
      "epoch": 0.24256,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.008430735270182291,
      "learning_rate": 0.0001,
      "loss": 4.4347,
      "loss/crossentropy": 2.353670358657837,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22648276388645172,
      "step": 12128
    },
    {
      "epoch": 0.2426,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.01263427734375,
      "learning_rate": 0.0001,
      "loss": 3.9896,
      "loss/crossentropy": 1.714030683040619,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19275055080652237,
      "step": 12130
    },
    {
      "epoch": 0.24264,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.012898763020833334,
      "learning_rate": 0.0001,
      "loss": 4.2981,
      "loss/crossentropy": 2.0048200488090515,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20478381216526031,
      "step": 12132
    },
    {
      "epoch": 0.24268,
      "grad_norm": 2.125,
      "grad_norm_var": 0.011766560872395833,
      "learning_rate": 0.0001,
      "loss": 4.0497,
      "loss/crossentropy": 1.7151115536689758,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19872941821813583,
      "step": 12134
    },
    {
      "epoch": 0.24272,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.011818186442057291,
      "learning_rate": 0.0001,
      "loss": 4.2753,
      "loss/crossentropy": 2.1646838188171387,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22345983982086182,
      "step": 12136
    },
    {
      "epoch": 0.24276,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.010807037353515625,
      "learning_rate": 0.0001,
      "loss": 4.157,
      "loss/crossentropy": 2.053311765193939,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2143322378396988,
      "step": 12138
    },
    {
      "epoch": 0.2428,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.009993235270182291,
      "learning_rate": 0.0001,
      "loss": 3.8805,
      "loss/crossentropy": 2.0471617579460144,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20450014621019363,
      "step": 12140
    },
    {
      "epoch": 0.24284,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.008737945556640625,
      "learning_rate": 0.0001,
      "loss": 4.2966,
      "loss/crossentropy": 2.1727033853530884,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22905820608139038,
      "step": 12142
    },
    {
      "epoch": 0.24288,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.008245595296223958,
      "learning_rate": 0.0001,
      "loss": 4.3861,
      "loss/crossentropy": 2.197450280189514,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22059273719787598,
      "step": 12144
    },
    {
      "epoch": 0.24292,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.003413899739583333,
      "learning_rate": 0.0001,
      "loss": 4.4841,
      "loss/crossentropy": 1.9954137206077576,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22441548854112625,
      "step": 12146
    },
    {
      "epoch": 0.24296,
      "grad_norm": 2.0,
      "grad_norm_var": 0.005110677083333333,
      "learning_rate": 0.0001,
      "loss": 4.2113,
      "loss/crossentropy": 2.3393132090568542,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22949577867984772,
      "step": 12148
    },
    {
      "epoch": 0.243,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0067291259765625,
      "learning_rate": 0.0001,
      "loss": 4.146,
      "loss/crossentropy": 2.11602646112442,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19181709736585617,
      "step": 12150
    },
    {
      "epoch": 0.24304,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.005597941080729167,
      "learning_rate": 0.0001,
      "loss": 4.1381,
      "loss/crossentropy": 2.130657136440277,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20904043316841125,
      "step": 12152
    },
    {
      "epoch": 0.24308,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0053293863932291664,
      "learning_rate": 0.0001,
      "loss": 4.2983,
      "loss/crossentropy": 2.3923556804656982,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23029828071594238,
      "step": 12154
    },
    {
      "epoch": 0.24312,
      "grad_norm": 1.875,
      "grad_norm_var": 0.007209269205729166,
      "learning_rate": 0.0001,
      "loss": 3.9697,
      "loss/crossentropy": 2.024892747402191,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19879258424043655,
      "step": 12156
    },
    {
      "epoch": 0.24316,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.007826487223307291,
      "learning_rate": 0.0001,
      "loss": 3.8761,
      "loss/crossentropy": 1.7833393812179565,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18542324006557465,
      "step": 12158
    },
    {
      "epoch": 0.2432,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.007328033447265625,
      "learning_rate": 0.0001,
      "loss": 4.1896,
      "loss/crossentropy": 2.253599762916565,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2148476019501686,
      "step": 12160
    },
    {
      "epoch": 0.24324,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.005177561442057292,
      "learning_rate": 0.0001,
      "loss": 4.0253,
      "loss/crossentropy": 1.7218471765518188,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1923334002494812,
      "step": 12162
    },
    {
      "epoch": 0.24328,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.007575480143229166,
      "learning_rate": 0.0001,
      "loss": 4.2947,
      "loss/crossentropy": 2.210463523864746,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2203536182641983,
      "step": 12164
    },
    {
      "epoch": 0.24332,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0076487223307291664,
      "learning_rate": 0.0001,
      "loss": 4.4164,
      "loss/crossentropy": 2.452837347984314,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22888235747814178,
      "step": 12166
    },
    {
      "epoch": 0.24336,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.007657877604166667,
      "learning_rate": 0.0001,
      "loss": 4.2157,
      "loss/crossentropy": 2.3511279821395874,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2431751787662506,
      "step": 12168
    },
    {
      "epoch": 0.2434,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.007819620768229167,
      "learning_rate": 0.0001,
      "loss": 4.0911,
      "loss/crossentropy": 2.0734334588050842,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21137738972902298,
      "step": 12170
    },
    {
      "epoch": 0.24344,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0065582275390625,
      "learning_rate": 0.0001,
      "loss": 4.0987,
      "loss/crossentropy": 1.9492397904396057,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22894760966300964,
      "step": 12172
    },
    {
      "epoch": 0.24348,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.005936431884765625,
      "learning_rate": 0.0001,
      "loss": 4.2316,
      "loss/crossentropy": 2.03458708524704,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20981666445732117,
      "step": 12174
    },
    {
      "epoch": 0.24352,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.007059478759765625,
      "learning_rate": 0.0001,
      "loss": 4.3396,
      "loss/crossentropy": 2.1464394330978394,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20741496980190277,
      "step": 12176
    },
    {
      "epoch": 0.24356,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0064999898274739586,
      "learning_rate": 0.0001,
      "loss": 3.9902,
      "loss/crossentropy": 1.8449034094810486,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21950078010559082,
      "step": 12178
    },
    {
      "epoch": 0.2436,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.0054433186848958336,
      "learning_rate": 0.0001,
      "loss": 4.0129,
      "loss/crossentropy": 1.8646993041038513,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2117064744234085,
      "step": 12180
    },
    {
      "epoch": 0.24364,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.008003743489583333,
      "learning_rate": 0.0001,
      "loss": 3.9347,
      "loss/crossentropy": 1.9643146991729736,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1946149319410324,
      "step": 12182
    },
    {
      "epoch": 0.24368,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.008141835530598959,
      "learning_rate": 0.0001,
      "loss": 4.1468,
      "loss/crossentropy": 1.867002248764038,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19111115485429764,
      "step": 12184
    },
    {
      "epoch": 0.24372,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.008129628499348958,
      "learning_rate": 0.0001,
      "loss": 4.0443,
      "loss/crossentropy": 1.9431232810020447,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19612180441617966,
      "step": 12186
    },
    {
      "epoch": 0.24376,
      "grad_norm": 2.25,
      "grad_norm_var": 0.0105224609375,
      "learning_rate": 0.0001,
      "loss": 4.2931,
      "loss/crossentropy": 2.197216033935547,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22340577840805054,
      "step": 12188
    },
    {
      "epoch": 0.2438,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.010469563802083333,
      "learning_rate": 0.0001,
      "loss": 4.2499,
      "loss/crossentropy": 1.916576623916626,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20536810904741287,
      "step": 12190
    },
    {
      "epoch": 0.24384,
      "grad_norm": 2.25,
      "grad_norm_var": 0.010497029622395833,
      "learning_rate": 0.0001,
      "loss": 4.3838,
      "loss/crossentropy": 2.0369369983673096,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26386965811252594,
      "step": 12192
    },
    {
      "epoch": 0.24388,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.01639404296875,
      "learning_rate": 0.0001,
      "loss": 4.2817,
      "loss/crossentropy": 2.086443066596985,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21351207792758942,
      "step": 12194
    },
    {
      "epoch": 0.24392,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.01619873046875,
      "learning_rate": 0.0001,
      "loss": 4.3649,
      "loss/crossentropy": 2.0969839096069336,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21489758789539337,
      "step": 12196
    },
    {
      "epoch": 0.24396,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.016463216145833334,
      "learning_rate": 0.0001,
      "loss": 4.2481,
      "loss/crossentropy": 2.086832642555237,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20651830732822418,
      "step": 12198
    },
    {
      "epoch": 0.244,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.017146809895833334,
      "learning_rate": 0.0001,
      "loss": 4.1397,
      "loss/crossentropy": 1.9104391932487488,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1963956654071808,
      "step": 12200
    },
    {
      "epoch": 0.24404,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0191314697265625,
      "learning_rate": 0.0001,
      "loss": 3.8558,
      "loss/crossentropy": 1.9419523477554321,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19279541075229645,
      "step": 12202
    },
    {
      "epoch": 0.24408,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.01697998046875,
      "learning_rate": 0.0001,
      "loss": 4.3239,
      "loss/crossentropy": 2.2867754697799683,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23261378705501556,
      "step": 12204
    },
    {
      "epoch": 0.24412,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.017634073893229168,
      "learning_rate": 0.0001,
      "loss": 4.0786,
      "loss/crossentropy": 1.941315233707428,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20962880551815033,
      "step": 12206
    },
    {
      "epoch": 0.24416,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.01541748046875,
      "learning_rate": 0.0001,
      "loss": 4.2044,
      "loss/crossentropy": 2.0638798475265503,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21499283611774445,
      "step": 12208
    },
    {
      "epoch": 0.2442,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.008454386393229167,
      "learning_rate": 0.0001,
      "loss": 4.1889,
      "loss/crossentropy": 2.3167499899864197,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21805942803621292,
      "step": 12210
    },
    {
      "epoch": 0.24424,
      "grad_norm": 2.0,
      "grad_norm_var": 0.00848388671875,
      "learning_rate": 0.0001,
      "loss": 4.0956,
      "loss/crossentropy": 2.1785646080970764,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21259736269712448,
      "step": 12212
    },
    {
      "epoch": 0.24428,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.006615193684895834,
      "learning_rate": 0.0001,
      "loss": 4.2054,
      "loss/crossentropy": 1.983469545841217,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21091710776090622,
      "step": 12214
    },
    {
      "epoch": 0.24432,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.006937408447265625,
      "learning_rate": 0.0001,
      "loss": 4.2283,
      "loss/crossentropy": 1.8501896858215332,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20537292212247849,
      "step": 12216
    },
    {
      "epoch": 0.24436,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0072934468587239586,
      "learning_rate": 0.0001,
      "loss": 4.182,
      "loss/crossentropy": 1.7293490767478943,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1916719302535057,
      "step": 12218
    },
    {
      "epoch": 0.2444,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0069488525390625,
      "learning_rate": 0.0001,
      "loss": 4.3247,
      "loss/crossentropy": 2.0931429862976074,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22081860899925232,
      "step": 12220
    },
    {
      "epoch": 0.24444,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.007657623291015625,
      "learning_rate": 0.0001,
      "loss": 4.0619,
      "loss/crossentropy": 2.157875657081604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20259525626897812,
      "step": 12222
    },
    {
      "epoch": 0.24448,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.007248687744140625,
      "learning_rate": 0.0001,
      "loss": 4.2712,
      "loss/crossentropy": 2.2028552889823914,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2431233748793602,
      "step": 12224
    },
    {
      "epoch": 0.24452,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006473541259765625,
      "learning_rate": 0.0001,
      "loss": 4.1184,
      "loss/crossentropy": 2.0906929969787598,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2051575481891632,
      "step": 12226
    },
    {
      "epoch": 0.24456,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.006091054280598958,
      "learning_rate": 0.0001,
      "loss": 4.5601,
      "loss/crossentropy": 2.5020272731781006,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2485819309949875,
      "step": 12228
    },
    {
      "epoch": 0.2446,
      "grad_norm": 8.0625,
      "grad_norm_var": 2.244887034098307,
      "learning_rate": 0.0001,
      "loss": 4.1465,
      "loss/crossentropy": 1.502736508846283,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1839340552687645,
      "step": 12230
    },
    {
      "epoch": 0.24464,
      "grad_norm": 2.484375,
      "grad_norm_var": 2.2357358296712238,
      "learning_rate": 0.0001,
      "loss": 4.3882,
      "loss/crossentropy": 2.1888676285743713,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.226266011595726,
      "step": 12232
    },
    {
      "epoch": 0.24468,
      "grad_norm": 2.125,
      "grad_norm_var": 2.243033599853516,
      "learning_rate": 0.0001,
      "loss": 4.3887,
      "loss/crossentropy": 2.377061367034912,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23195213079452515,
      "step": 12234
    },
    {
      "epoch": 0.24472,
      "grad_norm": 2.03125,
      "grad_norm_var": 2.257559967041016,
      "learning_rate": 0.0001,
      "loss": 3.9354,
      "loss/crossentropy": 2.0567076206207275,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20983586460351944,
      "step": 12236
    },
    {
      "epoch": 0.24476,
      "grad_norm": 2.015625,
      "grad_norm_var": 2.246906534830729,
      "learning_rate": 0.0001,
      "loss": 4.1978,
      "loss/crossentropy": 2.007763922214508,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21983467787504196,
      "step": 12238
    },
    {
      "epoch": 0.2448,
      "grad_norm": 2.078125,
      "grad_norm_var": 2.247749837239583,
      "learning_rate": 0.0001,
      "loss": 4.1411,
      "loss/crossentropy": 2.3309481143951416,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23099908232688904,
      "step": 12240
    },
    {
      "epoch": 0.24484,
      "grad_norm": 2.5,
      "grad_norm_var": 2.23668212890625,
      "learning_rate": 0.0001,
      "loss": 4.2159,
      "loss/crossentropy": 1.8231184482574463,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22009263187646866,
      "step": 12242
    },
    {
      "epoch": 0.24488,
      "grad_norm": 1.9609375,
      "grad_norm_var": 2.2464637756347656,
      "learning_rate": 0.0001,
      "loss": 4.1131,
      "loss/crossentropy": 2.140045642852783,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22202204167842865,
      "step": 12244
    },
    {
      "epoch": 0.24492,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.02835057576497396,
      "learning_rate": 0.0001,
      "loss": 4.2209,
      "loss/crossentropy": 1.8495931029319763,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19803623855113983,
      "step": 12246
    },
    {
      "epoch": 0.24496,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.0185943603515625,
      "learning_rate": 0.0001,
      "loss": 4.1624,
      "loss/crossentropy": 2.2141982913017273,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23418182879686356,
      "step": 12248
    },
    {
      "epoch": 0.245,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.01844482421875,
      "learning_rate": 0.0001,
      "loss": 4.4104,
      "loss/crossentropy": 2.14465594291687,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22683896869421005,
      "step": 12250
    },
    {
      "epoch": 0.24504,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.01610107421875,
      "learning_rate": 0.0001,
      "loss": 4.1913,
      "loss/crossentropy": 2.0804443359375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2265845239162445,
      "step": 12252
    },
    {
      "epoch": 0.24508,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0160308837890625,
      "learning_rate": 0.0001,
      "loss": 4.485,
      "loss/crossentropy": 2.2451776266098022,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22881492972373962,
      "step": 12254
    },
    {
      "epoch": 0.24512,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.016947428385416668,
      "learning_rate": 0.0001,
      "loss": 4.0869,
      "loss/crossentropy": 1.6861794590950012,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18126793205738068,
      "step": 12256
    },
    {
      "epoch": 0.24516,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0059773763020833336,
      "learning_rate": 0.0001,
      "loss": 4.2343,
      "loss/crossentropy": 1.87660551071167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20990663766860962,
      "step": 12258
    },
    {
      "epoch": 0.2452,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.006799062093098958,
      "learning_rate": 0.0001,
      "loss": 4.2019,
      "loss/crossentropy": 2.103124976158142,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19615671038627625,
      "step": 12260
    },
    {
      "epoch": 0.24524,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0069048563639322914,
      "learning_rate": 0.0001,
      "loss": 4.2509,
      "loss/crossentropy": 2.139270842075348,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21861301362514496,
      "step": 12262
    },
    {
      "epoch": 0.24528,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.00791015625,
      "learning_rate": 0.0001,
      "loss": 4.4081,
      "loss/crossentropy": 2.108114778995514,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20996354520320892,
      "step": 12264
    },
    {
      "epoch": 0.24532,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.008847808837890625,
      "learning_rate": 0.0001,
      "loss": 3.9495,
      "loss/crossentropy": 2.1205111145973206,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21352755278348923,
      "step": 12266
    },
    {
      "epoch": 0.24536,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.009248860677083333,
      "learning_rate": 0.0001,
      "loss": 3.9946,
      "loss/crossentropy": 2.20136821269989,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2277800738811493,
      "step": 12268
    },
    {
      "epoch": 0.2454,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.011336008707682291,
      "learning_rate": 0.0001,
      "loss": 4.1116,
      "loss/crossentropy": 2.1149147748947144,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2062554806470871,
      "step": 12270
    },
    {
      "epoch": 0.24544,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.01932347615559896,
      "learning_rate": 0.0001,
      "loss": 4.5226,
      "loss/crossentropy": 2.1119648218154907,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2922344133257866,
      "step": 12272
    },
    {
      "epoch": 0.24548,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.017964680989583332,
      "learning_rate": 0.0001,
      "loss": 4.1766,
      "loss/crossentropy": 2.2631434202194214,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2044210433959961,
      "step": 12274
    },
    {
      "epoch": 0.24552,
      "grad_norm": 2.125,
      "grad_norm_var": 0.06928609212239584,
      "learning_rate": 0.0001,
      "loss": 4.0956,
      "loss/crossentropy": 2.264181971549988,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23541904985904694,
      "step": 12276
    },
    {
      "epoch": 0.24556,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.06836649576822916,
      "learning_rate": 0.0001,
      "loss": 4.2083,
      "loss/crossentropy": 2.256329298019409,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21784210950136185,
      "step": 12278
    },
    {
      "epoch": 0.2456,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.06968765258789063,
      "learning_rate": 0.0001,
      "loss": 4.267,
      "loss/crossentropy": 2.0962833166122437,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.211236834526062,
      "step": 12280
    },
    {
      "epoch": 0.24564,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0683990478515625,
      "learning_rate": 0.0001,
      "loss": 4.4767,
      "loss/crossentropy": 2.3430999517440796,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25170228630304337,
      "step": 12282
    },
    {
      "epoch": 0.24568,
      "grad_norm": 2.125,
      "grad_norm_var": 0.06921361287434896,
      "learning_rate": 0.0001,
      "loss": 4.1492,
      "loss/crossentropy": 1.8256065845489502,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20632392168045044,
      "step": 12284
    },
    {
      "epoch": 0.24572,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.06643778483072917,
      "learning_rate": 0.0001,
      "loss": 4.294,
      "loss/crossentropy": 2.481971561908722,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2121504619717598,
      "step": 12286
    },
    {
      "epoch": 0.24576,
      "grad_norm": 2.125,
      "grad_norm_var": 0.060469563802083334,
      "learning_rate": 0.0001,
      "loss": 4.2136,
      "loss/crossentropy": 2.2815581560134888,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22004567831754684,
      "step": 12288
    },
    {
      "epoch": 0.2458,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.060323079427083336,
      "learning_rate": 0.0001,
      "loss": 4.2426,
      "loss/crossentropy": 1.7125394940376282,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19618325680494308,
      "step": 12290
    },
    {
      "epoch": 0.24584,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.008347320556640624,
      "learning_rate": 0.0001,
      "loss": 3.9676,
      "loss/crossentropy": 2.0333253145217896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20251524448394775,
      "step": 12292
    },
    {
      "epoch": 0.24588,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.008939361572265625,
      "learning_rate": 0.0001,
      "loss": 4.0958,
      "loss/crossentropy": 2.359953284263611,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23697812855243683,
      "step": 12294
    },
    {
      "epoch": 0.24592,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0065348307291666664,
      "learning_rate": 0.0001,
      "loss": 3.9914,
      "loss/crossentropy": 1.8465049266815186,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18830078095197678,
      "step": 12296
    },
    {
      "epoch": 0.24596,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.00662841796875,
      "learning_rate": 0.0001,
      "loss": 4.1669,
      "loss/crossentropy": 2.407967984676361,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20054074376821518,
      "step": 12298
    },
    {
      "epoch": 0.246,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.007005818684895833,
      "learning_rate": 0.0001,
      "loss": 4.1943,
      "loss/crossentropy": 2.306265115737915,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2286146879196167,
      "step": 12300
    },
    {
      "epoch": 0.24604,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0076416015625,
      "learning_rate": 0.0001,
      "loss": 4.108,
      "loss/crossentropy": 1.9776363968849182,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2015514224767685,
      "step": 12302
    },
    {
      "epoch": 0.24608,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.010041300455729167,
      "learning_rate": 0.0001,
      "loss": 3.7023,
      "loss/crossentropy": 1.6589386463165283,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16657201945781708,
      "step": 12304
    },
    {
      "epoch": 0.24612,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.009968058268229166,
      "learning_rate": 0.0001,
      "loss": 4.2749,
      "loss/crossentropy": 2.263510227203369,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21220041066408157,
      "step": 12306
    },
    {
      "epoch": 0.24616,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008583323160807291,
      "learning_rate": 0.0001,
      "loss": 4.1735,
      "loss/crossentropy": 2.0802704095840454,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20156628638505936,
      "step": 12308
    },
    {
      "epoch": 0.2462,
      "grad_norm": 2.0,
      "grad_norm_var": 0.015193430582682292,
      "learning_rate": 0.0001,
      "loss": 4.2597,
      "loss/crossentropy": 2.0921266674995422,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22044362872838974,
      "step": 12310
    },
    {
      "epoch": 0.24624,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.0164703369140625,
      "learning_rate": 0.0001,
      "loss": 3.8587,
      "loss/crossentropy": 1.7366862297058105,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1722203940153122,
      "step": 12312
    },
    {
      "epoch": 0.24628,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.016377766927083332,
      "learning_rate": 0.0001,
      "loss": 4.1408,
      "loss/crossentropy": 2.0741729140281677,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1942591667175293,
      "step": 12314
    },
    {
      "epoch": 0.24632,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0145751953125,
      "learning_rate": 0.0001,
      "loss": 4.1343,
      "loss/crossentropy": 2.0611414909362793,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22244945168495178,
      "step": 12316
    },
    {
      "epoch": 0.24636,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.014196523030598958,
      "learning_rate": 0.0001,
      "loss": 3.8586,
      "loss/crossentropy": 1.9337337017059326,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20655318349599838,
      "step": 12318
    },
    {
      "epoch": 0.2464,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.012141672770182292,
      "learning_rate": 0.0001,
      "loss": 3.9197,
      "loss/crossentropy": 1.8766502737998962,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1851816102862358,
      "step": 12320
    },
    {
      "epoch": 0.24644,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.012552897135416666,
      "learning_rate": 0.0001,
      "loss": 4.2407,
      "loss/crossentropy": 2.121203899383545,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2177201583981514,
      "step": 12322
    },
    {
      "epoch": 0.24648,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0136138916015625,
      "learning_rate": 0.0001,
      "loss": 4.1441,
      "loss/crossentropy": 1.8102782368659973,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19694476574659348,
      "step": 12324
    },
    {
      "epoch": 0.24652,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0062164306640625,
      "learning_rate": 0.0001,
      "loss": 4.2526,
      "loss/crossentropy": 2.323423147201538,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20612536370754242,
      "step": 12326
    },
    {
      "epoch": 0.24656,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.007814280192057292,
      "learning_rate": 0.0001,
      "loss": 4.4862,
      "loss/crossentropy": 2.2835768461227417,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22697383165359497,
      "step": 12328
    },
    {
      "epoch": 0.2466,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.006192779541015625,
      "learning_rate": 0.0001,
      "loss": 4.2868,
      "loss/crossentropy": 2.197639048099518,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22446971386671066,
      "step": 12330
    },
    {
      "epoch": 0.24664,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.006761678059895833,
      "learning_rate": 0.0001,
      "loss": 4.0758,
      "loss/crossentropy": 2.1731618642807007,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21908622235059738,
      "step": 12332
    },
    {
      "epoch": 0.24668,
      "grad_norm": 1.875,
      "grad_norm_var": 0.009291330973307291,
      "learning_rate": 0.0001,
      "loss": 3.9108,
      "loss/crossentropy": 1.9236284494400024,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1810280755162239,
      "step": 12334
    },
    {
      "epoch": 0.24672,
      "grad_norm": 2.0,
      "grad_norm_var": 0.008318837483723958,
      "learning_rate": 0.0001,
      "loss": 4.3975,
      "loss/crossentropy": 2.2902809381484985,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23192713409662247,
      "step": 12336
    },
    {
      "epoch": 0.24676,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008367665608723958,
      "learning_rate": 0.0001,
      "loss": 4.1681,
      "loss/crossentropy": 1.9808775186538696,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19344759732484818,
      "step": 12338
    },
    {
      "epoch": 0.2468,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.008337148030598958,
      "learning_rate": 0.0001,
      "loss": 4.1581,
      "loss/crossentropy": 2.148995041847229,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19994106143712997,
      "step": 12340
    },
    {
      "epoch": 0.24684,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.008528391520182291,
      "learning_rate": 0.0001,
      "loss": 4.3927,
      "loss/crossentropy": 2.374568462371826,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24350540339946747,
      "step": 12342
    },
    {
      "epoch": 0.24688,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.006959788004557292,
      "learning_rate": 0.0001,
      "loss": 4.258,
      "loss/crossentropy": 1.9702014923095703,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2038111537694931,
      "step": 12344
    },
    {
      "epoch": 0.24692,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.007008616129557292,
      "learning_rate": 0.0001,
      "loss": 4.0126,
      "loss/crossentropy": 1.8438855409622192,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1822836771607399,
      "step": 12346
    },
    {
      "epoch": 0.24696,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.010754140218098958,
      "learning_rate": 0.0001,
      "loss": 4.1352,
      "loss/crossentropy": 2.005755662918091,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21399007737636566,
      "step": 12348
    },
    {
      "epoch": 0.247,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008819325764973959,
      "learning_rate": 0.0001,
      "loss": 4.0988,
      "loss/crossentropy": 1.8640305399894714,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19055305421352386,
      "step": 12350
    },
    {
      "epoch": 0.24704,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.010453033447265624,
      "learning_rate": 0.0001,
      "loss": 4.1981,
      "loss/crossentropy": 2.1293725967407227,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2117801234126091,
      "step": 12352
    },
    {
      "epoch": 0.24708,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.02337621053059896,
      "learning_rate": 0.0001,
      "loss": 4.2666,
      "loss/crossentropy": 1.7655459642410278,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2085946872830391,
      "step": 12354
    },
    {
      "epoch": 0.24712,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.021897125244140624,
      "learning_rate": 0.0001,
      "loss": 4.3175,
      "loss/crossentropy": 2.184986114501953,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22597889602184296,
      "step": 12356
    },
    {
      "epoch": 0.24716,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.021897125244140624,
      "learning_rate": 0.0001,
      "loss": 4.1295,
      "loss/crossentropy": 2.368021607398987,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21957845985889435,
      "step": 12358
    },
    {
      "epoch": 0.2472,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.020645904541015624,
      "learning_rate": 0.0001,
      "loss": 4.16,
      "loss/crossentropy": 2.297884225845337,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2155354768037796,
      "step": 12360
    },
    {
      "epoch": 0.24724,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.023273722330729166,
      "learning_rate": 0.0001,
      "loss": 4.2181,
      "loss/crossentropy": 2.121878147125244,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22192668169736862,
      "step": 12362
    },
    {
      "epoch": 0.24728,
      "grad_norm": 2.125,
      "grad_norm_var": 0.020442454020182292,
      "learning_rate": 0.0001,
      "loss": 4.3233,
      "loss/crossentropy": 2.130228877067566,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20712029188871384,
      "step": 12364
    },
    {
      "epoch": 0.24732,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.02367121378580729,
      "learning_rate": 0.0001,
      "loss": 3.8542,
      "loss/crossentropy": 1.6520383954048157,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16511157900094986,
      "step": 12366
    },
    {
      "epoch": 0.24736,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.02220637003580729,
      "learning_rate": 0.0001,
      "loss": 4.1894,
      "loss/crossentropy": 1.9993118047714233,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2112603336572647,
      "step": 12368
    },
    {
      "epoch": 0.2474,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.011494700113932292,
      "learning_rate": 0.0001,
      "loss": 4.0522,
      "loss/crossentropy": 1.9502257108688354,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1944574937224388,
      "step": 12370
    },
    {
      "epoch": 0.24744,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.011872355143229167,
      "learning_rate": 0.0001,
      "loss": 4.1098,
      "loss/crossentropy": 2.1219520568847656,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22872696816921234,
      "step": 12372
    },
    {
      "epoch": 0.24748,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0502593994140625,
      "learning_rate": 0.0001,
      "loss": 4.172,
      "loss/crossentropy": 2.0668599605560303,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21508124470710754,
      "step": 12374
    },
    {
      "epoch": 0.24752,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.05098241170247396,
      "learning_rate": 0.0001,
      "loss": 4.1901,
      "loss/crossentropy": 2.242877721786499,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2308686003088951,
      "step": 12376
    },
    {
      "epoch": 0.24756,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.047304026285807294,
      "learning_rate": 0.0001,
      "loss": 3.8779,
      "loss/crossentropy": 1.8258161544799805,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1852153167128563,
      "step": 12378
    },
    {
      "epoch": 0.2476,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.04918390909830729,
      "learning_rate": 0.0001,
      "loss": 4.1241,
      "loss/crossentropy": 2.0654167532920837,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20526036620140076,
      "step": 12380
    },
    {
      "epoch": 0.24764,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.04582087198893229,
      "learning_rate": 0.0001,
      "loss": 4.1601,
      "loss/crossentropy": 2.125216484069824,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.216343455016613,
      "step": 12382
    },
    {
      "epoch": 0.24768,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0473052978515625,
      "learning_rate": 0.0001,
      "loss": 4.1756,
      "loss/crossentropy": 2.285245180130005,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22941745072603226,
      "step": 12384
    },
    {
      "epoch": 0.24772,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.04720637003580729,
      "learning_rate": 0.0001,
      "loss": 4.1097,
      "loss/crossentropy": 2.0148350596427917,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1874203458428383,
      "step": 12386
    },
    {
      "epoch": 0.24776,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.04812825520833333,
      "learning_rate": 0.0001,
      "loss": 4.3127,
      "loss/crossentropy": 2.2349741458892822,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23187313228845596,
      "step": 12388
    },
    {
      "epoch": 0.2478,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.009178670247395833,
      "learning_rate": 0.0001,
      "loss": 4.1875,
      "loss/crossentropy": 2.195721983909607,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2119324654340744,
      "step": 12390
    },
    {
      "epoch": 0.24784,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.012676747639973958,
      "learning_rate": 0.0001,
      "loss": 4.5732,
      "loss/crossentropy": 2.1261669397354126,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21143332868814468,
      "step": 12392
    },
    {
      "epoch": 0.24788,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.012798817952473958,
      "learning_rate": 0.0001,
      "loss": 3.9475,
      "loss/crossentropy": 2.1640073657035828,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2192412167787552,
      "step": 12394
    },
    {
      "epoch": 0.24792,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.011230214436848959,
      "learning_rate": 0.0001,
      "loss": 4.2435,
      "loss/crossentropy": 1.9555792808532715,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2110436111688614,
      "step": 12396
    },
    {
      "epoch": 0.24796,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011156972249348958,
      "learning_rate": 0.0001,
      "loss": 4.0616,
      "loss/crossentropy": 1.948053002357483,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19957780838012695,
      "step": 12398
    },
    {
      "epoch": 0.248,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.011156972249348958,
      "learning_rate": 0.0001,
      "loss": 4.0865,
      "loss/crossentropy": 2.355304718017578,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20301833003759384,
      "step": 12400
    },
    {
      "epoch": 0.24804,
      "grad_norm": 2.0,
      "grad_norm_var": 0.010944620768229166,
      "learning_rate": 0.0001,
      "loss": 4.2537,
      "loss/crossentropy": 2.172769784927368,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2157917320728302,
      "step": 12402
    },
    {
      "epoch": 0.24808,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.012109120686848959,
      "learning_rate": 0.0001,
      "loss": 4.2817,
      "loss/crossentropy": 2.2186524868011475,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.221016563475132,
      "step": 12404
    },
    {
      "epoch": 0.24812,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.012113189697265625,
      "learning_rate": 0.0001,
      "loss": 3.9624,
      "loss/crossentropy": 2.011409044265747,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19150879979133606,
      "step": 12406
    },
    {
      "epoch": 0.24816,
      "grad_norm": 2.125,
      "grad_norm_var": 0.010984039306640625,
      "learning_rate": 0.0001,
      "loss": 4.5039,
      "loss/crossentropy": 1.9245591163635254,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2296570986509323,
      "step": 12408
    },
    {
      "epoch": 0.2482,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.016434733072916666,
      "learning_rate": 0.0001,
      "loss": 4.5765,
      "loss/crossentropy": 1.973130702972412,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20532477647066116,
      "step": 12410
    },
    {
      "epoch": 0.24824,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.01617609659830729,
      "learning_rate": 0.0001,
      "loss": 3.914,
      "loss/crossentropy": 1.655932605266571,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1609746441245079,
      "step": 12412
    },
    {
      "epoch": 0.24828,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.02411677042643229,
      "learning_rate": 0.0001,
      "loss": 4.2172,
      "loss/crossentropy": 2.104023277759552,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19898276031017303,
      "step": 12414
    },
    {
      "epoch": 0.24832,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.02210261027018229,
      "learning_rate": 0.0001,
      "loss": 4.0597,
      "loss/crossentropy": 1.9362882375717163,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20107803493738174,
      "step": 12416
    },
    {
      "epoch": 0.24836,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.021996815999348957,
      "learning_rate": 0.0001,
      "loss": 4.4181,
      "loss/crossentropy": 2.1508368253707886,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2056911736726761,
      "step": 12418
    },
    {
      "epoch": 0.2484,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.02194391886393229,
      "learning_rate": 0.0001,
      "loss": 3.802,
      "loss/crossentropy": 1.7712991833686829,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1908227875828743,
      "step": 12420
    },
    {
      "epoch": 0.24844,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.02060114542643229,
      "learning_rate": 0.0001,
      "loss": 4.3557,
      "loss/crossentropy": 2.2705806493759155,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23912303894758224,
      "step": 12422
    },
    {
      "epoch": 0.24848,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.019791666666666666,
      "learning_rate": 0.0001,
      "loss": 4.2696,
      "loss/crossentropy": 2.4222676753997803,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21739919483661652,
      "step": 12424
    },
    {
      "epoch": 0.24852,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0145172119140625,
      "learning_rate": 0.0001,
      "loss": 4.1935,
      "loss/crossentropy": 1.945872962474823,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19402413070201874,
      "step": 12426
    },
    {
      "epoch": 0.24856,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.01646728515625,
      "learning_rate": 0.0001,
      "loss": 4.0908,
      "loss/crossentropy": 1.7793864011764526,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19452669471502304,
      "step": 12428
    },
    {
      "epoch": 0.2486,
      "grad_norm": 2.0,
      "grad_norm_var": 0.015941365559895834,
      "learning_rate": 0.0001,
      "loss": 4.1045,
      "loss/crossentropy": 2.2827813625335693,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21572840213775635,
      "step": 12430
    },
    {
      "epoch": 0.24864,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.016544342041015625,
      "learning_rate": 0.0001,
      "loss": 3.749,
      "loss/crossentropy": 1.9543398022651672,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20825360715389252,
      "step": 12432
    },
    {
      "epoch": 0.24868,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.017276763916015625,
      "learning_rate": 0.0001,
      "loss": 4.3727,
      "loss/crossentropy": 2.223303198814392,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21834088116884232,
      "step": 12434
    },
    {
      "epoch": 0.24872,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.017071278889973958,
      "learning_rate": 0.0001,
      "loss": 3.9993,
      "loss/crossentropy": 1.6987267136573792,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1834847405552864,
      "step": 12436
    },
    {
      "epoch": 0.24876,
      "grad_norm": 2.125,
      "grad_norm_var": 0.014522043863932292,
      "learning_rate": 0.0001,
      "loss": 4.4309,
      "loss/crossentropy": 2.3947317600250244,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22872642427682877,
      "step": 12438
    },
    {
      "epoch": 0.2488,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.014435831705729167,
      "learning_rate": 0.0001,
      "loss": 4.4858,
      "loss/crossentropy": 2.5481021404266357,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23773372173309326,
      "step": 12440
    },
    {
      "epoch": 0.24884,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0145904541015625,
      "learning_rate": 0.0001,
      "loss": 4.1621,
      "loss/crossentropy": 2.2349241971969604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21584660559892654,
      "step": 12442
    },
    {
      "epoch": 0.24888,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.012988026936848958,
      "learning_rate": 0.0001,
      "loss": 4.2827,
      "loss/crossentropy": 1.9308242201805115,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19295598566532135,
      "step": 12444
    },
    {
      "epoch": 0.24892,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.008536783854166667,
      "learning_rate": 0.0001,
      "loss": 4.0778,
      "loss/crossentropy": 2.3508042097091675,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22008787840604782,
      "step": 12446
    },
    {
      "epoch": 0.24896,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.012593332926432292,
      "learning_rate": 0.0001,
      "loss": 4.0865,
      "loss/crossentropy": 2.0870128870010376,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20749760419130325,
      "step": 12448
    },
    {
      "epoch": 0.249,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.012971750895182292,
      "learning_rate": 0.0001,
      "loss": 4.2758,
      "loss/crossentropy": 2.087821125984192,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21110248565673828,
      "step": 12450
    },
    {
      "epoch": 0.24904,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.011494954427083334,
      "learning_rate": 0.0001,
      "loss": 4.3075,
      "loss/crossentropy": 2.131770372390747,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21426154673099518,
      "step": 12452
    },
    {
      "epoch": 0.24908,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.011092122395833333,
      "learning_rate": 0.0001,
      "loss": 4.2373,
      "loss/crossentropy": 2.175555467605591,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20418058335781097,
      "step": 12454
    },
    {
      "epoch": 0.24912,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.012198893229166667,
      "learning_rate": 0.0001,
      "loss": 4.2788,
      "loss/crossentropy": 2.205165147781372,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23265192657709122,
      "step": 12456
    },
    {
      "epoch": 0.24916,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.014534250895182291,
      "learning_rate": 0.0001,
      "loss": 3.9345,
      "loss/crossentropy": 1.6410180926322937,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18058068305253983,
      "step": 12458
    },
    {
      "epoch": 0.2492,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.014345041910807292,
      "learning_rate": 0.0001,
      "loss": 3.962,
      "loss/crossentropy": 1.8249012231826782,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19164791703224182,
      "step": 12460
    },
    {
      "epoch": 0.24924,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.0142242431640625,
      "learning_rate": 0.0001,
      "loss": 4.4945,
      "loss/crossentropy": 2.0906582474708557,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23415963351726532,
      "step": 12462
    },
    {
      "epoch": 0.24928,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.009285227457682291,
      "learning_rate": 0.0001,
      "loss": 4.4702,
      "loss/crossentropy": 2.314555048942566,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24343416839838028,
      "step": 12464
    },
    {
      "epoch": 0.24932,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.009105428059895834,
      "learning_rate": 0.0001,
      "loss": 4.3059,
      "loss/crossentropy": 2.1258983612060547,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21851430088281631,
      "step": 12466
    },
    {
      "epoch": 0.24936,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009325917561848958,
      "learning_rate": 0.0001,
      "loss": 4.1642,
      "loss/crossentropy": 1.977954626083374,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20545368641614914,
      "step": 12468
    },
    {
      "epoch": 0.2494,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.011258951822916667,
      "learning_rate": 0.0001,
      "loss": 4.2224,
      "loss/crossentropy": 2.0212838649749756,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21720624715089798,
      "step": 12470
    },
    {
      "epoch": 0.24944,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.010205078125,
      "learning_rate": 0.0001,
      "loss": 4.3526,
      "loss/crossentropy": 1.964124321937561,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19991052150726318,
      "step": 12472
    },
    {
      "epoch": 0.24948,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.008567047119140626,
      "learning_rate": 0.0001,
      "loss": 4.2633,
      "loss/crossentropy": 1.9038777947425842,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19013714790344238,
      "step": 12474
    },
    {
      "epoch": 0.24952,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.008449045817057292,
      "learning_rate": 0.0001,
      "loss": 4.4785,
      "loss/crossentropy": 2.1772372722625732,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21470360457897186,
      "step": 12476
    },
    {
      "epoch": 0.24956,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.006605784098307292,
      "learning_rate": 0.0001,
      "loss": 4.3636,
      "loss/crossentropy": 2.153541684150696,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23221681267023087,
      "step": 12478
    },
    {
      "epoch": 0.2496,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.006668853759765625,
      "learning_rate": 0.0001,
      "loss": 4.4578,
      "loss/crossentropy": 2.421363592147827,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20812640339136124,
      "step": 12480
    },
    {
      "epoch": 0.24964,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.008188629150390625,
      "learning_rate": 0.0001,
      "loss": 4.0628,
      "loss/crossentropy": 1.8497061133384705,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18766893446445465,
      "step": 12482
    },
    {
      "epoch": 0.24968,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.008894602457682291,
      "learning_rate": 0.0001,
      "loss": 4.1011,
      "loss/crossentropy": 1.9876007437705994,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1932743340730667,
      "step": 12484
    },
    {
      "epoch": 0.24972,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.007616170247395833,
      "learning_rate": 0.0001,
      "loss": 4.2722,
      "loss/crossentropy": 2.4254921674728394,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23315788805484772,
      "step": 12486
    },
    {
      "epoch": 0.24976,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0079010009765625,
      "learning_rate": 0.0001,
      "loss": 4.1012,
      "loss/crossentropy": 1.983458697795868,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2190450206398964,
      "step": 12488
    },
    {
      "epoch": 0.2498,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.007111612955729167,
      "learning_rate": 0.0001,
      "loss": 4.0284,
      "loss/crossentropy": 1.8051987886428833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19166412204504013,
      "step": 12490
    },
    {
      "epoch": 0.24984,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0052398681640625,
      "learning_rate": 0.0001,
      "loss": 4.2385,
      "loss/crossentropy": 2.1552056670188904,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2053346112370491,
      "step": 12492
    },
    {
      "epoch": 0.24988,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.004992421468098958,
      "learning_rate": 0.0001,
      "loss": 4.0157,
      "loss/crossentropy": 1.9644648432731628,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17631876468658447,
      "step": 12494
    },
    {
      "epoch": 0.24992,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.006089019775390625,
      "learning_rate": 0.0001,
      "loss": 4.1626,
      "loss/crossentropy": 2.027154862880707,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21671167761087418,
      "step": 12496
    },
    {
      "epoch": 0.24996,
      "grad_norm": 2.125,
      "grad_norm_var": 0.00628662109375,
      "learning_rate": 0.0001,
      "loss": 4.2485,
      "loss/crossentropy": 2.2398791313171387,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22507991641759872,
      "step": 12498
    },
    {
      "epoch": 0.25,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.006232706705729166,
      "learning_rate": 0.0001,
      "loss": 4.13,
      "loss/crossentropy": 2.2342909574508667,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2192724049091339,
      "step": 12500
    },
    {
      "epoch": 0.25004,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0054443359375,
      "learning_rate": 0.0001,
      "loss": 4.2779,
      "loss/crossentropy": 1.8621744513511658,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1882321536540985,
      "step": 12502
    },
    {
      "epoch": 0.25008,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.004988606770833333,
      "learning_rate": 0.0001,
      "loss": 4.4516,
      "loss/crossentropy": 2.2012354135513306,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2128412127494812,
      "step": 12504
    },
    {
      "epoch": 0.25012,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.005873362223307292,
      "learning_rate": 0.0001,
      "loss": 4.0122,
      "loss/crossentropy": 1.7629758715629578,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1928938776254654,
      "step": 12506
    },
    {
      "epoch": 0.25016,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.005730946858723958,
      "learning_rate": 0.0001,
      "loss": 4.1563,
      "loss/crossentropy": 2.076514720916748,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20530561357736588,
      "step": 12508
    },
    {
      "epoch": 0.2502,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.006207021077473959,
      "learning_rate": 0.0001,
      "loss": 4.2005,
      "loss/crossentropy": 2.004107654094696,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21081873774528503,
      "step": 12510
    },
    {
      "epoch": 0.25024,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.008013661702473958,
      "learning_rate": 0.0001,
      "loss": 4.3323,
      "loss/crossentropy": 2.232061505317688,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21327269077301025,
      "step": 12512
    },
    {
      "epoch": 0.25028,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0068662007649739586,
      "learning_rate": 0.0001,
      "loss": 4.1507,
      "loss/crossentropy": 2.1506210565567017,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20797627419233322,
      "step": 12514
    },
    {
      "epoch": 0.25032,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008207194010416667,
      "learning_rate": 0.0001,
      "loss": 4.1999,
      "loss/crossentropy": 2.084704279899597,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25306878983974457,
      "step": 12516
    },
    {
      "epoch": 0.25036,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.008137003580729166,
      "learning_rate": 0.0001,
      "loss": 4.3286,
      "loss/crossentropy": 2.000536620616913,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20802763104438782,
      "step": 12518
    },
    {
      "epoch": 0.2504,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0088531494140625,
      "learning_rate": 0.0001,
      "loss": 4.1324,
      "loss/crossentropy": 2.2391778230667114,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2171899899840355,
      "step": 12520
    },
    {
      "epoch": 0.25044,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.011375935872395833,
      "learning_rate": 0.0001,
      "loss": 4.458,
      "loss/crossentropy": 2.1465210914611816,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2026364952325821,
      "step": 12522
    },
    {
      "epoch": 0.25048,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.011226145426432292,
      "learning_rate": 0.0001,
      "loss": 4.0804,
      "loss/crossentropy": 1.8129625916481018,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17917531728744507,
      "step": 12524
    },
    {
      "epoch": 0.25052,
      "grad_norm": 2.0,
      "grad_norm_var": 0.010029856363932292,
      "learning_rate": 0.0001,
      "loss": 4.1832,
      "loss/crossentropy": 1.98322331905365,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20269985496997833,
      "step": 12526
    },
    {
      "epoch": 0.25056,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.008955637613932291,
      "learning_rate": 0.0001,
      "loss": 4.373,
      "loss/crossentropy": 2.196588397026062,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23827942460775375,
      "step": 12528
    },
    {
      "epoch": 0.2506,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.009209950764973959,
      "learning_rate": 0.0001,
      "loss": 4.3811,
      "loss/crossentropy": 2.183007001876831,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2071165144443512,
      "step": 12530
    },
    {
      "epoch": 0.25064,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0106353759765625,
      "learning_rate": 0.0001,
      "loss": 4.4205,
      "loss/crossentropy": 2.2728021144866943,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21004344522953033,
      "step": 12532
    },
    {
      "epoch": 0.25068,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0111083984375,
      "learning_rate": 0.0001,
      "loss": 4.5478,
      "loss/crossentropy": 2.280518889427185,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22367073595523834,
      "step": 12534
    },
    {
      "epoch": 0.25072,
      "grad_norm": 2.25,
      "grad_norm_var": 0.015038045247395833,
      "learning_rate": 0.0001,
      "loss": 4.135,
      "loss/crossentropy": 2.088103711605072,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2038320079445839,
      "step": 12536
    },
    {
      "epoch": 0.25076,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.012837727864583334,
      "learning_rate": 0.0001,
      "loss": 4.1798,
      "loss/crossentropy": 1.646530568599701,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18982253968715668,
      "step": 12538
    },
    {
      "epoch": 0.2508,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.013622792561848958,
      "learning_rate": 0.0001,
      "loss": 4.2246,
      "loss/crossentropy": 2.1743921041488647,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22835461795330048,
      "step": 12540
    },
    {
      "epoch": 0.25084,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0151123046875,
      "learning_rate": 0.0001,
      "loss": 4.3138,
      "loss/crossentropy": 2.2216947078704834,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21978579461574554,
      "step": 12542
    },
    {
      "epoch": 0.25088,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.014890289306640625,
      "learning_rate": 0.0001,
      "loss": 4.298,
      "loss/crossentropy": 2.141101062297821,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21347828209400177,
      "step": 12544
    },
    {
      "epoch": 0.25092,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.014861806233723959,
      "learning_rate": 0.0001,
      "loss": 4.1534,
      "loss/crossentropy": 2.1961969137191772,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2192479968070984,
      "step": 12546
    },
    {
      "epoch": 0.25096,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.014274088541666667,
      "learning_rate": 0.0001,
      "loss": 3.9219,
      "loss/crossentropy": 1.9632240533828735,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19182069599628448,
      "step": 12548
    },
    {
      "epoch": 0.251,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.02173639933268229,
      "learning_rate": 0.0001,
      "loss": 4.0594,
      "loss/crossentropy": 1.7821694612503052,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24946660548448563,
      "step": 12550
    },
    {
      "epoch": 0.25104,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.016290028889973957,
      "learning_rate": 0.0001,
      "loss": 4.2423,
      "loss/crossentropy": 2.261335611343384,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23269569873809814,
      "step": 12552
    },
    {
      "epoch": 0.25108,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.01762669881184896,
      "learning_rate": 0.0001,
      "loss": 3.9858,
      "loss/crossentropy": 2.2365309596061707,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2119913324713707,
      "step": 12554
    },
    {
      "epoch": 0.25112,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0176025390625,
      "learning_rate": 0.0001,
      "loss": 3.9798,
      "loss/crossentropy": 2.0685681104660034,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19985978305339813,
      "step": 12556
    },
    {
      "epoch": 0.25116,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0144683837890625,
      "learning_rate": 0.0001,
      "loss": 4.0231,
      "loss/crossentropy": 2.0773105025291443,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2006489858031273,
      "step": 12558
    },
    {
      "epoch": 0.2512,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.01561279296875,
      "learning_rate": 0.0001,
      "loss": 4.1948,
      "loss/crossentropy": 2.1145309805870056,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20449287444353104,
      "step": 12560
    },
    {
      "epoch": 0.25124,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.017097981770833333,
      "learning_rate": 0.0001,
      "loss": 4.0543,
      "loss/crossentropy": 2.109993577003479,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20761344581842422,
      "step": 12562
    },
    {
      "epoch": 0.25128,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.016755167643229166,
      "learning_rate": 0.0001,
      "loss": 4.1661,
      "loss/crossentropy": 1.9653338193893433,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2069985270500183,
      "step": 12564
    },
    {
      "epoch": 0.25132,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.007045237223307291,
      "learning_rate": 0.0001,
      "loss": 4.101,
      "loss/crossentropy": 1.7084832191467285,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18922459334135056,
      "step": 12566
    },
    {
      "epoch": 0.25136,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006494140625,
      "learning_rate": 0.0001,
      "loss": 3.9288,
      "loss/crossentropy": 1.8007041215896606,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2122679501771927,
      "step": 12568
    },
    {
      "epoch": 0.2514,
      "grad_norm": 2.125,
      "grad_norm_var": 0.007972971598307291,
      "learning_rate": 0.0001,
      "loss": 4.3199,
      "loss/crossentropy": 2.041845440864563,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2219957411289215,
      "step": 12570
    },
    {
      "epoch": 0.25144,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.008577219645182292,
      "learning_rate": 0.0001,
      "loss": 4.3703,
      "loss/crossentropy": 2.2668861150741577,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2117021307349205,
      "step": 12572
    },
    {
      "epoch": 0.25148,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.010235341389973958,
      "learning_rate": 0.0001,
      "loss": 3.8462,
      "loss/crossentropy": 1.8246251940727234,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1822521835565567,
      "step": 12574
    },
    {
      "epoch": 0.25152,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.010553995768229166,
      "learning_rate": 0.0001,
      "loss": 3.8137,
      "loss/crossentropy": 2.0739742517471313,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2114427089691162,
      "step": 12576
    },
    {
      "epoch": 0.25156,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008906809488932292,
      "learning_rate": 0.0001,
      "loss": 3.9261,
      "loss/crossentropy": 1.6614344120025635,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18680214881896973,
      "step": 12578
    },
    {
      "epoch": 0.2516,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.01279296875,
      "learning_rate": 0.0001,
      "loss": 4.4042,
      "loss/crossentropy": 2.5629080533981323,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26097629219293594,
      "step": 12580
    },
    {
      "epoch": 0.25164,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.01337890625,
      "learning_rate": 0.0001,
      "loss": 4.1583,
      "loss/crossentropy": 2.058123230934143,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2273598164319992,
      "step": 12582
    },
    {
      "epoch": 0.25168,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.014314524332682292,
      "learning_rate": 0.0001,
      "loss": 4.2802,
      "loss/crossentropy": 1.9959335327148438,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1991373971104622,
      "step": 12584
    },
    {
      "epoch": 0.25172,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.018293253580729165,
      "learning_rate": 0.0001,
      "loss": 3.9794,
      "loss/crossentropy": 1.8165839314460754,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1888013407588005,
      "step": 12586
    },
    {
      "epoch": 0.25176,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.015221913655598959,
      "learning_rate": 0.0001,
      "loss": 4.0687,
      "loss/crossentropy": 1.9918802976608276,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18641646206378937,
      "step": 12588
    },
    {
      "epoch": 0.2518,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.013378651936848958,
      "learning_rate": 0.0001,
      "loss": 4.3443,
      "loss/crossentropy": 2.0256036520004272,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19628985226154327,
      "step": 12590
    },
    {
      "epoch": 0.25184,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.01236572265625,
      "learning_rate": 0.0001,
      "loss": 4.1317,
      "loss/crossentropy": 1.948347806930542,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19971590489149094,
      "step": 12592
    },
    {
      "epoch": 0.25188,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.012544759114583333,
      "learning_rate": 0.0001,
      "loss": 3.8528,
      "loss/crossentropy": 2.2025747299194336,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2437238171696663,
      "step": 12594
    },
    {
      "epoch": 0.25192,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.010872141520182291,
      "learning_rate": 0.0001,
      "loss": 4.2102,
      "loss/crossentropy": 2.3209575414657593,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2366873100399971,
      "step": 12596
    },
    {
      "epoch": 0.25196,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.010846964518229167,
      "learning_rate": 0.0001,
      "loss": 3.8028,
      "loss/crossentropy": 1.7978705763816833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19252559542655945,
      "step": 12598
    },
    {
      "epoch": 0.252,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010359446207682291,
      "learning_rate": 0.0001,
      "loss": 3.9269,
      "loss/crossentropy": 1.964760184288025,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1943562552332878,
      "step": 12600
    },
    {
      "epoch": 0.25204,
      "grad_norm": 2.0,
      "grad_norm_var": 0.005399322509765625,
      "learning_rate": 0.0001,
      "loss": 4.2205,
      "loss/crossentropy": 2.1420929431915283,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.206790953874588,
      "step": 12602
    },
    {
      "epoch": 0.25208,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.008084869384765625,
      "learning_rate": 0.0001,
      "loss": 4.5232,
      "loss/crossentropy": 2.2422659397125244,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22513067722320557,
      "step": 12604
    },
    {
      "epoch": 0.25212,
      "grad_norm": 2.75,
      "grad_norm_var": 0.04026667277018229,
      "learning_rate": 0.0001,
      "loss": 3.8394,
      "loss/crossentropy": 1.587006688117981,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1833646520972252,
      "step": 12606
    },
    {
      "epoch": 0.25216,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.03873291015625,
      "learning_rate": 0.0001,
      "loss": 4.4461,
      "loss/crossentropy": 2.426867365837097,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2435208261013031,
      "step": 12608
    },
    {
      "epoch": 0.2522,
      "grad_norm": 2.0,
      "grad_norm_var": 0.040185546875,
      "learning_rate": 0.0001,
      "loss": 4.0039,
      "loss/crossentropy": 1.9733251333236694,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21772438287734985,
      "step": 12610
    },
    {
      "epoch": 0.25224,
      "grad_norm": 2.84375,
      "grad_norm_var": 0.0764312744140625,
      "learning_rate": 0.0001,
      "loss": 4.2712,
      "loss/crossentropy": 1.4835429191589355,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17013566195964813,
      "step": 12612
    },
    {
      "epoch": 0.25228,
      "grad_norm": 2.0,
      "grad_norm_var": 0.07500178019205729,
      "learning_rate": 0.0001,
      "loss": 3.8985,
      "loss/crossentropy": 2.165693759918213,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21315700560808182,
      "step": 12614
    },
    {
      "epoch": 0.25232,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.07396647135416666,
      "learning_rate": 0.0001,
      "loss": 3.9856,
      "loss/crossentropy": 2.206323266029358,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19913922995328903,
      "step": 12616
    },
    {
      "epoch": 0.25236,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.07515360514322916,
      "learning_rate": 0.0001,
      "loss": 4.0788,
      "loss/crossentropy": 2.2520995140075684,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22997721284627914,
      "step": 12618
    },
    {
      "epoch": 0.2524,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.07444559733072917,
      "learning_rate": 0.0001,
      "loss": 4.365,
      "loss/crossentropy": 1.9322227239608765,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20600315928459167,
      "step": 12620
    },
    {
      "epoch": 0.25244,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.047459920247395836,
      "learning_rate": 0.0001,
      "loss": 4.0578,
      "loss/crossentropy": 2.140220284461975,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21072514355182648,
      "step": 12622
    },
    {
      "epoch": 0.25248,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.04942118326822917,
      "learning_rate": 0.0001,
      "loss": 4.2316,
      "loss/crossentropy": 2.061431884765625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2259845808148384,
      "step": 12624
    },
    {
      "epoch": 0.25252,
      "grad_norm": 2.125,
      "grad_norm_var": 0.047055816650390624,
      "learning_rate": 0.0001,
      "loss": 4.14,
      "loss/crossentropy": 2.323551654815674,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23168490827083588,
      "step": 12626
    },
    {
      "epoch": 0.25256,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.008335113525390625,
      "learning_rate": 0.0001,
      "loss": 4.0036,
      "loss/crossentropy": 1.93999582529068,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21139459311962128,
      "step": 12628
    },
    {
      "epoch": 0.2526,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.011173248291015625,
      "learning_rate": 0.0001,
      "loss": 4.207,
      "loss/crossentropy": 1.9334313869476318,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21537292003631592,
      "step": 12630
    },
    {
      "epoch": 0.25264,
      "grad_norm": 2.0,
      "grad_norm_var": 0.010477447509765625,
      "learning_rate": 0.0001,
      "loss": 4.1674,
      "loss/crossentropy": 2.145465135574341,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22080931067466736,
      "step": 12632
    },
    {
      "epoch": 0.25268,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.011668904622395834,
      "learning_rate": 0.0001,
      "loss": 4.0058,
      "loss/crossentropy": 1.9708059430122375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2007170245051384,
      "step": 12634
    },
    {
      "epoch": 0.25272,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0115631103515625,
      "learning_rate": 0.0001,
      "loss": 3.9649,
      "loss/crossentropy": 1.9522746801376343,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20030274242162704,
      "step": 12636
    },
    {
      "epoch": 0.25276,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.012035878499348958,
      "learning_rate": 0.0001,
      "loss": 4.0416,
      "loss/crossentropy": 1.856387436389923,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19685833156108856,
      "step": 12638
    },
    {
      "epoch": 0.2528,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.013787587483723959,
      "learning_rate": 0.0001,
      "loss": 4.475,
      "loss/crossentropy": 1.9449518322944641,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20622673630714417,
      "step": 12640
    },
    {
      "epoch": 0.25284,
      "grad_norm": 2.25,
      "grad_norm_var": 0.017533365885416666,
      "learning_rate": 0.0001,
      "loss": 4.4683,
      "loss/crossentropy": 2.4559473991394043,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21476459503173828,
      "step": 12642
    },
    {
      "epoch": 0.25288,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.016778310139973957,
      "learning_rate": 0.0001,
      "loss": 4.3164,
      "loss/crossentropy": 2.238897919654846,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22538839280605316,
      "step": 12644
    },
    {
      "epoch": 0.25292,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.013944498697916667,
      "learning_rate": 0.0001,
      "loss": 3.9801,
      "loss/crossentropy": 1.7506417036056519,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18603645265102386,
      "step": 12646
    },
    {
      "epoch": 0.25296,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.014359283447265624,
      "learning_rate": 0.0001,
      "loss": 4.1238,
      "loss/crossentropy": 2.162258505821228,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21168682724237442,
      "step": 12648
    },
    {
      "epoch": 0.253,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.015154774983723958,
      "learning_rate": 0.0001,
      "loss": 4.0441,
      "loss/crossentropy": 1.9122841954231262,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20636393874883652,
      "step": 12650
    },
    {
      "epoch": 0.25304,
      "grad_norm": 2.125,
      "grad_norm_var": 0.015187327067057292,
      "learning_rate": 0.0001,
      "loss": 4.3566,
      "loss/crossentropy": 2.1501541137695312,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22346170246601105,
      "step": 12652
    },
    {
      "epoch": 0.25308,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.01456298828125,
      "learning_rate": 0.0001,
      "loss": 4.0914,
      "loss/crossentropy": 1.797228217124939,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17510685324668884,
      "step": 12654
    },
    {
      "epoch": 0.25312,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.010422515869140624,
      "learning_rate": 0.0001,
      "loss": 3.9232,
      "loss/crossentropy": 1.7334046363830566,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19618064910173416,
      "step": 12656
    },
    {
      "epoch": 0.25316,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0066912333170572914,
      "learning_rate": 0.0001,
      "loss": 4.3287,
      "loss/crossentropy": 1.99091237783432,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21426931023597717,
      "step": 12658
    },
    {
      "epoch": 0.2532,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0069244384765625,
      "learning_rate": 0.0001,
      "loss": 4.5036,
      "loss/crossentropy": 2.1908310651779175,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20360572636127472,
      "step": 12660
    },
    {
      "epoch": 0.25324,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.00787353515625,
      "learning_rate": 0.0001,
      "loss": 3.9746,
      "loss/crossentropy": 1.7925593852996826,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18203188478946686,
      "step": 12662
    },
    {
      "epoch": 0.25328,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.009006500244140625,
      "learning_rate": 0.0001,
      "loss": 3.8955,
      "loss/crossentropy": 1.9464862942695618,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1896686628460884,
      "step": 12664
    },
    {
      "epoch": 0.25332,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006379954020182292,
      "learning_rate": 0.0001,
      "loss": 3.9922,
      "loss/crossentropy": 2.0207581520080566,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20436276495456696,
      "step": 12666
    },
    {
      "epoch": 0.25336,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.005322011311848959,
      "learning_rate": 0.0001,
      "loss": 4.141,
      "loss/crossentropy": 1.9915068745613098,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21265029907226562,
      "step": 12668
    },
    {
      "epoch": 0.2534,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.005204010009765625,
      "learning_rate": 0.0001,
      "loss": 4.2602,
      "loss/crossentropy": 1.787190020084381,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20195094496011734,
      "step": 12670
    },
    {
      "epoch": 0.25344,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.005036417643229167,
      "learning_rate": 0.0001,
      "loss": 4.3554,
      "loss/crossentropy": 2.2403881549835205,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2313152700662613,
      "step": 12672
    },
    {
      "epoch": 0.25348,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.00540771484375,
      "learning_rate": 0.0001,
      "loss": 3.9895,
      "loss/crossentropy": 1.6122660636901855,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19166506081819534,
      "step": 12674
    },
    {
      "epoch": 0.25352,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.005402628580729167,
      "learning_rate": 0.0001,
      "loss": 4.1966,
      "loss/crossentropy": 2.0758888125419617,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19904160499572754,
      "step": 12676
    },
    {
      "epoch": 0.25356,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0064656575520833336,
      "learning_rate": 0.0001,
      "loss": 4.2466,
      "loss/crossentropy": 1.8765565156936646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17384758591651917,
      "step": 12678
    },
    {
      "epoch": 0.2536,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0054705301920572914,
      "learning_rate": 0.0001,
      "loss": 4.0977,
      "loss/crossentropy": 2.1322853565216064,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21987880766391754,
      "step": 12680
    },
    {
      "epoch": 0.25364,
      "grad_norm": 2.0,
      "grad_norm_var": 0.006196848551432292,
      "learning_rate": 0.0001,
      "loss": 3.9862,
      "loss/crossentropy": 2.0303893089294434,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20912615954875946,
      "step": 12682
    },
    {
      "epoch": 0.25368,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0065826416015625,
      "learning_rate": 0.0001,
      "loss": 4.0022,
      "loss/crossentropy": 2.2314319610595703,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22904963046312332,
      "step": 12684
    },
    {
      "epoch": 0.25372,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.007438151041666666,
      "learning_rate": 0.0001,
      "loss": 4.1854,
      "loss/crossentropy": 1.9751350283622742,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20683745294809341,
      "step": 12686
    },
    {
      "epoch": 0.25376,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0088043212890625,
      "learning_rate": 0.0001,
      "loss": 4.112,
      "loss/crossentropy": 1.9198334217071533,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19134660065174103,
      "step": 12688
    },
    {
      "epoch": 0.2538,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.009214019775390625,
      "learning_rate": 0.0001,
      "loss": 3.8783,
      "loss/crossentropy": 1.945135474205017,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20948659628629684,
      "step": 12690
    },
    {
      "epoch": 0.25384,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0094879150390625,
      "learning_rate": 0.0001,
      "loss": 4.1018,
      "loss/crossentropy": 2.146402955055237,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21819791197776794,
      "step": 12692
    },
    {
      "epoch": 0.25388,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.007806142171223958,
      "learning_rate": 0.0001,
      "loss": 4.3696,
      "loss/crossentropy": 2.309106230735779,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2160014659166336,
      "step": 12694
    },
    {
      "epoch": 0.25392,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.007616170247395833,
      "learning_rate": 0.0001,
      "loss": 4.0637,
      "loss/crossentropy": 2.1726107597351074,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20791994035243988,
      "step": 12696
    },
    {
      "epoch": 0.25396,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.007370758056640625,
      "learning_rate": 0.0001,
      "loss": 4.252,
      "loss/crossentropy": 2.0905996561050415,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2083379328250885,
      "step": 12698
    },
    {
      "epoch": 0.254,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.008185831705729167,
      "learning_rate": 0.0001,
      "loss": 4.197,
      "loss/crossentropy": 1.9190585017204285,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1856430396437645,
      "step": 12700
    },
    {
      "epoch": 0.25404,
      "grad_norm": 2.125,
      "grad_norm_var": 0.007877604166666666,
      "learning_rate": 0.0001,
      "loss": 4.3906,
      "loss/crossentropy": 2.2493419647216797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20197859406471252,
      "step": 12702
    },
    {
      "epoch": 0.25408,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.007731119791666667,
      "learning_rate": 0.0001,
      "loss": 4.1935,
      "loss/crossentropy": 2.0205613374710083,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2155362293124199,
      "step": 12704
    },
    {
      "epoch": 0.25412,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.006257120768229167,
      "learning_rate": 0.0001,
      "loss": 4.188,
      "loss/crossentropy": 1.8853323459625244,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20137012749910355,
      "step": 12706
    },
    {
      "epoch": 0.25416,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0061431884765625,
      "learning_rate": 0.0001,
      "loss": 4.098,
      "loss/crossentropy": 2.109869122505188,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2036309316754341,
      "step": 12708
    },
    {
      "epoch": 0.2542,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.005890909830729167,
      "learning_rate": 0.0001,
      "loss": 4.1782,
      "loss/crossentropy": 2.0598954558372498,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22793744504451752,
      "step": 12710
    },
    {
      "epoch": 0.25424,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.0055735270182291664,
      "learning_rate": 0.0001,
      "loss": 4.2984,
      "loss/crossentropy": 2.169256567955017,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20931441336870193,
      "step": 12712
    },
    {
      "epoch": 0.25428,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.005197906494140625,
      "learning_rate": 0.0001,
      "loss": 4.2715,
      "loss/crossentropy": 1.8178632855415344,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20344894379377365,
      "step": 12714
    },
    {
      "epoch": 0.25432,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.004327138264973958,
      "learning_rate": 0.0001,
      "loss": 4.0897,
      "loss/crossentropy": 1.6468743085861206,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16709627211093903,
      "step": 12716
    },
    {
      "epoch": 0.25436,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.004748280843098958,
      "learning_rate": 0.0001,
      "loss": 4.1702,
      "loss/crossentropy": 1.8353837728500366,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20837673544883728,
      "step": 12718
    },
    {
      "epoch": 0.2544,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.005492146809895833,
      "learning_rate": 0.0001,
      "loss": 3.8056,
      "loss/crossentropy": 1.7881666421890259,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18411727994680405,
      "step": 12720
    },
    {
      "epoch": 0.25444,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0065958658854166664,
      "learning_rate": 0.0001,
      "loss": 4.5369,
      "loss/crossentropy": 2.19934618473053,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22607161849737167,
      "step": 12722
    },
    {
      "epoch": 0.25448,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.008821360270182292,
      "learning_rate": 0.0001,
      "loss": 4.5237,
      "loss/crossentropy": 1.9789779782295227,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2156950756907463,
      "step": 12724
    },
    {
      "epoch": 0.25452,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.010778554280598958,
      "learning_rate": 0.0001,
      "loss": 4.1059,
      "loss/crossentropy": 2.0605525970458984,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20622699707746506,
      "step": 12726
    },
    {
      "epoch": 0.25456,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.011336263020833333,
      "learning_rate": 0.0001,
      "loss": 4.2862,
      "loss/crossentropy": 2.1131407022476196,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21070127934217453,
      "step": 12728
    },
    {
      "epoch": 0.2546,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.011336263020833333,
      "learning_rate": 0.0001,
      "loss": 4.3789,
      "loss/crossentropy": 1.9708096981048584,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22219926118850708,
      "step": 12730
    },
    {
      "epoch": 0.25464,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.011918131510416667,
      "learning_rate": 0.0001,
      "loss": 4.2985,
      "loss/crossentropy": 2.220608353614807,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2271113097667694,
      "step": 12732
    },
    {
      "epoch": 0.25468,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.01146240234375,
      "learning_rate": 0.0001,
      "loss": 4.2097,
      "loss/crossentropy": 2.535509705543518,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24400310218334198,
      "step": 12734
    },
    {
      "epoch": 0.25472,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.015952301025390626,
      "learning_rate": 0.0001,
      "loss": 4.6837,
      "loss/crossentropy": 2.3619974851608276,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2268039956688881,
      "step": 12736
    },
    {
      "epoch": 0.25476,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0168212890625,
      "learning_rate": 0.0001,
      "loss": 3.9929,
      "loss/crossentropy": 2.0095953941345215,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20850248634815216,
      "step": 12738
    },
    {
      "epoch": 0.2548,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.012276204427083333,
      "learning_rate": 0.0001,
      "loss": 4.1466,
      "loss/crossentropy": 1.934277892112732,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2116999626159668,
      "step": 12740
    },
    {
      "epoch": 0.25484,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.011701456705729167,
      "learning_rate": 0.0001,
      "loss": 4.336,
      "loss/crossentropy": 2.237201452255249,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22638197988271713,
      "step": 12742
    },
    {
      "epoch": 0.25488,
      "grad_norm": 2.0,
      "grad_norm_var": 0.011464182535807292,
      "learning_rate": 0.0001,
      "loss": 4.0354,
      "loss/crossentropy": 2.302059292793274,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22285569459199905,
      "step": 12744
    },
    {
      "epoch": 0.25492,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.013944244384765625,
      "learning_rate": 0.0001,
      "loss": 3.7097,
      "loss/crossentropy": 1.7690886855125427,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19209770113229752,
      "step": 12746
    },
    {
      "epoch": 0.25496,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.013944244384765625,
      "learning_rate": 0.0001,
      "loss": 4.2324,
      "loss/crossentropy": 2.4372475147247314,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2192872166633606,
      "step": 12748
    },
    {
      "epoch": 0.255,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.016405232747395835,
      "learning_rate": 0.0001,
      "loss": 4.2627,
      "loss/crossentropy": 2.12698757648468,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21698793768882751,
      "step": 12750
    },
    {
      "epoch": 0.25504,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.007710520426432292,
      "learning_rate": 0.0001,
      "loss": 3.9991,
      "loss/crossentropy": 1.989953875541687,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1969183310866356,
      "step": 12752
    },
    {
      "epoch": 0.25508,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.034501139322916666,
      "learning_rate": 0.0001,
      "loss": 4.5505,
      "loss/crossentropy": 2.2911970615386963,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21863195300102234,
      "step": 12754
    },
    {
      "epoch": 0.25512,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.03460286458333333,
      "learning_rate": 0.0001,
      "loss": 4.0915,
      "loss/crossentropy": 1.8447460532188416,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20167672634124756,
      "step": 12756
    },
    {
      "epoch": 0.25516,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.035471343994140626,
      "learning_rate": 0.0001,
      "loss": 3.8488,
      "loss/crossentropy": 2.043856382369995,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2075670212507248,
      "step": 12758
    },
    {
      "epoch": 0.2552,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.037536366780598955,
      "learning_rate": 0.0001,
      "loss": 4.0893,
      "loss/crossentropy": 1.735123872756958,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19347237050533295,
      "step": 12760
    },
    {
      "epoch": 0.25524,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.034407552083333334,
      "learning_rate": 0.0001,
      "loss": 4.0586,
      "loss/crossentropy": 2.1058340072631836,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21859320253133774,
      "step": 12762
    },
    {
      "epoch": 0.25528,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.03444722493489583,
      "learning_rate": 0.0001,
      "loss": 4.2518,
      "loss/crossentropy": 2.0583395957946777,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.224592886865139,
      "step": 12764
    },
    {
      "epoch": 0.25532,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.032206217447916664,
      "learning_rate": 0.0001,
      "loss": 4.3562,
      "loss/crossentropy": 2.2550116777420044,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23644013702869415,
      "step": 12766
    },
    {
      "epoch": 0.25536,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.032293446858723956,
      "learning_rate": 0.0001,
      "loss": 4.2338,
      "loss/crossentropy": 2.463193655014038,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2229352444410324,
      "step": 12768
    },
    {
      "epoch": 0.2554,
      "grad_norm": 2.0,
      "grad_norm_var": 0.006400299072265625,
      "learning_rate": 0.0001,
      "loss": 4.016,
      "loss/crossentropy": 2.010310709476471,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20725534856319427,
      "step": 12770
    },
    {
      "epoch": 0.25544,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.006219228108723958,
      "learning_rate": 0.0001,
      "loss": 4.1988,
      "loss/crossentropy": 2.2002042531967163,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21686428785324097,
      "step": 12772
    },
    {
      "epoch": 0.25548,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.009690093994140624,
      "learning_rate": 0.0001,
      "loss": 4.2974,
      "loss/crossentropy": 2.072916865348816,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2210928201675415,
      "step": 12774
    },
    {
      "epoch": 0.25552,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010149892171223958,
      "learning_rate": 0.0001,
      "loss": 4.3168,
      "loss/crossentropy": 2.068341016769409,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20776809751987457,
      "step": 12776
    },
    {
      "epoch": 0.25556,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.010949452718098959,
      "learning_rate": 0.0001,
      "loss": 3.8552,
      "loss/crossentropy": 1.6145030856132507,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2056947946548462,
      "step": 12778
    },
    {
      "epoch": 0.2556,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.011311848958333334,
      "learning_rate": 0.0001,
      "loss": 4.0132,
      "loss/crossentropy": 2.1529496908187866,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21802741289138794,
      "step": 12780
    },
    {
      "epoch": 0.25564,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.011226399739583334,
      "learning_rate": 0.0001,
      "loss": 4.1708,
      "loss/crossentropy": 2.255640387535095,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21794818341732025,
      "step": 12782
    },
    {
      "epoch": 0.25568,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.011546834309895834,
      "learning_rate": 0.0001,
      "loss": 4.1102,
      "loss/crossentropy": 1.9934669137001038,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21103205531835556,
      "step": 12784
    },
    {
      "epoch": 0.25572,
      "grad_norm": 2.125,
      "grad_norm_var": 0.011864217122395833,
      "learning_rate": 0.0001,
      "loss": 4.2924,
      "loss/crossentropy": 1.9404807090759277,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22771108150482178,
      "step": 12786
    },
    {
      "epoch": 0.25576,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.012084706624348959,
      "learning_rate": 0.0001,
      "loss": 4.0444,
      "loss/crossentropy": 1.7803818583488464,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19245180487632751,
      "step": 12788
    },
    {
      "epoch": 0.2558,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.009423828125,
      "learning_rate": 0.0001,
      "loss": 4.2033,
      "loss/crossentropy": 2.108216881752014,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2239757925271988,
      "step": 12790
    },
    {
      "epoch": 0.25584,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0059478759765625,
      "learning_rate": 0.0001,
      "loss": 4.1999,
      "loss/crossentropy": 2.056196451187134,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20573420077562332,
      "step": 12792
    },
    {
      "epoch": 0.25588,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.007275390625,
      "learning_rate": 0.0001,
      "loss": 3.9749,
      "loss/crossentropy": 2.2215099334716797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21107079833745956,
      "step": 12794
    },
    {
      "epoch": 0.25592,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.009281158447265625,
      "learning_rate": 0.0001,
      "loss": 3.9527,
      "loss/crossentropy": 1.947695553302765,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19738437235355377,
      "step": 12796
    },
    {
      "epoch": 0.25596,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.010204823811848958,
      "learning_rate": 0.0001,
      "loss": 4.1072,
      "loss/crossentropy": 2.3351560831069946,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21714143455028534,
      "step": 12798
    },
    {
      "epoch": 0.256,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009798177083333333,
      "learning_rate": 0.0001,
      "loss": 4.3055,
      "loss/crossentropy": 2.399898648262024,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25758983194828033,
      "step": 12800
    },
    {
      "epoch": 0.25604,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.03814697265625,
      "learning_rate": 0.0001,
      "loss": 4.5632,
      "loss/crossentropy": 2.3537880182266235,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2976065129041672,
      "step": 12802
    },
    {
      "epoch": 0.25608,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.03857421875,
      "learning_rate": 0.0001,
      "loss": 3.8394,
      "loss/crossentropy": 1.7765586972236633,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1823679357767105,
      "step": 12804
    },
    {
      "epoch": 0.25612,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.039793904622395834,
      "learning_rate": 0.0001,
      "loss": 4.4023,
      "loss/crossentropy": 2.119332432746887,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23052836954593658,
      "step": 12806
    },
    {
      "epoch": 0.25616,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0414947509765625,
      "learning_rate": 0.0001,
      "loss": 4.0251,
      "loss/crossentropy": 1.8325288891792297,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19811426103115082,
      "step": 12808
    },
    {
      "epoch": 0.2562,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.039896392822265626,
      "learning_rate": 0.0001,
      "loss": 4.1089,
      "loss/crossentropy": 2.079294800758362,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21122989803552628,
      "step": 12810
    },
    {
      "epoch": 0.25624,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.03581110636393229,
      "learning_rate": 0.0001,
      "loss": 4.0105,
      "loss/crossentropy": 2.029142141342163,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2008090615272522,
      "step": 12812
    },
    {
      "epoch": 0.25628,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.03578058878580729,
      "learning_rate": 0.0001,
      "loss": 4.1743,
      "loss/crossentropy": 1.8697097301483154,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19589021801948547,
      "step": 12814
    },
    {
      "epoch": 0.25632,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.038331858317057294,
      "learning_rate": 0.0001,
      "loss": 3.8837,
      "loss/crossentropy": 1.8488793969154358,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19999422132968903,
      "step": 12816
    },
    {
      "epoch": 0.25636,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.011946360270182291,
      "learning_rate": 0.0001,
      "loss": 4.236,
      "loss/crossentropy": 2.352132201194763,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23569129407405853,
      "step": 12818
    },
    {
      "epoch": 0.2564,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010155232747395833,
      "learning_rate": 0.0001,
      "loss": 4.2247,
      "loss/crossentropy": 2.213895559310913,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19782276451587677,
      "step": 12820
    },
    {
      "epoch": 0.25644,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.008876291910807292,
      "learning_rate": 0.0001,
      "loss": 4.205,
      "loss/crossentropy": 2.328965425491333,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2138378769159317,
      "step": 12822
    },
    {
      "epoch": 0.25648,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.007407379150390625,
      "learning_rate": 0.0001,
      "loss": 3.9178,
      "loss/crossentropy": 2.148880124092102,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20276722311973572,
      "step": 12824
    },
    {
      "epoch": 0.25652,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.007124582926432292,
      "learning_rate": 0.0001,
      "loss": 4.2742,
      "loss/crossentropy": 2.1533660888671875,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22211932390928268,
      "step": 12826
    },
    {
      "epoch": 0.25656,
      "grad_norm": 2.75,
      "grad_norm_var": 0.04098078409830729,
      "learning_rate": 0.0001,
      "loss": 4.0928,
      "loss/crossentropy": 1.7518101930618286,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17689460515975952,
      "step": 12828
    },
    {
      "epoch": 0.2566,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0395660400390625,
      "learning_rate": 0.0001,
      "loss": 3.9745,
      "loss/crossentropy": 2.346145749092102,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22157911956310272,
      "step": 12830
    },
    {
      "epoch": 0.25664,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0358062744140625,
      "learning_rate": 0.0001,
      "loss": 4.1689,
      "loss/crossentropy": 1.8355774283409119,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20288754999637604,
      "step": 12832
    },
    {
      "epoch": 0.25668,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.03906962076822917,
      "learning_rate": 0.0001,
      "loss": 3.9947,
      "loss/crossentropy": 1.949233889579773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19860410690307617,
      "step": 12834
    },
    {
      "epoch": 0.25672,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.04248046875,
      "learning_rate": 0.0001,
      "loss": 4.0903,
      "loss/crossentropy": 2.0141645669937134,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19531898200511932,
      "step": 12836
    },
    {
      "epoch": 0.25676,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.04277725219726562,
      "learning_rate": 0.0001,
      "loss": 3.918,
      "loss/crossentropy": 1.6573863625526428,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.163545623421669,
      "step": 12838
    },
    {
      "epoch": 0.2568,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.04273656209309896,
      "learning_rate": 0.0001,
      "loss": 4.3252,
      "loss/crossentropy": 2.2174651622772217,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23083333671092987,
      "step": 12840
    },
    {
      "epoch": 0.25684,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.045967356363932295,
      "learning_rate": 0.0001,
      "loss": 3.7593,
      "loss/crossentropy": 1.8067168593406677,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1963481903076172,
      "step": 12842
    },
    {
      "epoch": 0.25688,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.010536448160807291,
      "learning_rate": 0.0001,
      "loss": 3.936,
      "loss/crossentropy": 1.928274691104889,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21783485263586044,
      "step": 12844
    },
    {
      "epoch": 0.25692,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.013498687744140625,
      "learning_rate": 0.0001,
      "loss": 4.0984,
      "loss/crossentropy": 2.183669090270996,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2040836587548256,
      "step": 12846
    },
    {
      "epoch": 0.25696,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.012894439697265624,
      "learning_rate": 0.0001,
      "loss": 4.0624,
      "loss/crossentropy": 1.7395422458648682,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19434035569429398,
      "step": 12848
    },
    {
      "epoch": 0.257,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.012772623697916667,
      "learning_rate": 0.0001,
      "loss": 3.9914,
      "loss/crossentropy": 1.9124428629875183,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19825156033039093,
      "step": 12850
    },
    {
      "epoch": 0.25704,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.011606597900390625,
      "learning_rate": 0.0001,
      "loss": 3.9325,
      "loss/crossentropy": 1.8064388036727905,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19374487549066544,
      "step": 12852
    },
    {
      "epoch": 0.25708,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0111724853515625,
      "learning_rate": 0.0001,
      "loss": 3.8722,
      "loss/crossentropy": 2.0943931341171265,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20997963845729828,
      "step": 12854
    },
    {
      "epoch": 0.25712,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.00948486328125,
      "learning_rate": 0.0001,
      "loss": 3.9974,
      "loss/crossentropy": 1.9252876043319702,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20608004182577133,
      "step": 12856
    },
    {
      "epoch": 0.25716,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.007575480143229166,
      "learning_rate": 0.0001,
      "loss": 3.9536,
      "loss/crossentropy": 1.813286304473877,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2107577547430992,
      "step": 12858
    },
    {
      "epoch": 0.2572,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.007405598958333333,
      "learning_rate": 0.0001,
      "loss": 4.1571,
      "loss/crossentropy": 1.985919713973999,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18724270164966583,
      "step": 12860
    },
    {
      "epoch": 0.25724,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.004654693603515625,
      "learning_rate": 0.0001,
      "loss": 3.9377,
      "loss/crossentropy": 1.912338137626648,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19602636992931366,
      "step": 12862
    },
    {
      "epoch": 0.25728,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.004743448893229167,
      "learning_rate": 0.0001,
      "loss": 4.0823,
      "loss/crossentropy": 1.962310791015625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1975855752825737,
      "step": 12864
    },
    {
      "epoch": 0.25732,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0050046284993489586,
      "learning_rate": 0.0001,
      "loss": 4.285,
      "loss/crossentropy": 2.063133656978607,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20012972503900528,
      "step": 12866
    },
    {
      "epoch": 0.25736,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.005751291910807292,
      "learning_rate": 0.0001,
      "loss": 4.1443,
      "loss/crossentropy": 2.1755728721618652,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21089357137680054,
      "step": 12868
    },
    {
      "epoch": 0.2574,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.006148274739583333,
      "learning_rate": 0.0001,
      "loss": 4.3854,
      "loss/crossentropy": 2.21665620803833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22525488585233688,
      "step": 12870
    },
    {
      "epoch": 0.25744,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.006959788004557292,
      "learning_rate": 0.0001,
      "loss": 4.2465,
      "loss/crossentropy": 2.2826790809631348,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22989312559366226,
      "step": 12872
    },
    {
      "epoch": 0.25748,
      "grad_norm": 1.875,
      "grad_norm_var": 0.008432769775390625,
      "learning_rate": 0.0001,
      "loss": 4.0753,
      "loss/crossentropy": 2.3187366724014282,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22048770636320114,
      "step": 12874
    },
    {
      "epoch": 0.25752,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0093017578125,
      "learning_rate": 0.0001,
      "loss": 4.2469,
      "loss/crossentropy": 2.234878957271576,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19875259697437286,
      "step": 12876
    },
    {
      "epoch": 0.25756,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.007897694905598959,
      "learning_rate": 0.0001,
      "loss": 4.0981,
      "loss/crossentropy": 1.8919751644134521,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20682695508003235,
      "step": 12878
    },
    {
      "epoch": 0.2576,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.008861287434895834,
      "learning_rate": 0.0001,
      "loss": 3.9564,
      "loss/crossentropy": 2.135833740234375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2100691795349121,
      "step": 12880
    },
    {
      "epoch": 0.25764,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.008421834309895833,
      "learning_rate": 0.0001,
      "loss": 4.163,
      "loss/crossentropy": 2.043276846408844,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20327290892601013,
      "step": 12882
    },
    {
      "epoch": 0.25768,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0077512105305989586,
      "learning_rate": 0.0001,
      "loss": 4.2439,
      "loss/crossentropy": 1.996088445186615,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20499806851148605,
      "step": 12884
    },
    {
      "epoch": 0.25772,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.006648508707682291,
      "learning_rate": 0.0001,
      "loss": 3.9821,
      "loss/crossentropy": 1.7864345908164978,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19354674220085144,
      "step": 12886
    },
    {
      "epoch": 0.25776,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0051310221354166664,
      "learning_rate": 0.0001,
      "loss": 3.961,
      "loss/crossentropy": 1.7890866994857788,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1806325614452362,
      "step": 12888
    },
    {
      "epoch": 0.2578,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.004938761393229167,
      "learning_rate": 0.0001,
      "loss": 4.1836,
      "loss/crossentropy": 1.9287649989128113,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19555142521858215,
      "step": 12890
    },
    {
      "epoch": 0.25784,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.005081939697265625,
      "learning_rate": 0.0001,
      "loss": 3.9151,
      "loss/crossentropy": 1.930963397026062,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19455663859844208,
      "step": 12892
    },
    {
      "epoch": 0.25788,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.007551829020182292,
      "learning_rate": 0.0001,
      "loss": 4.3216,
      "loss/crossentropy": 2.1105018854141235,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22564146667718887,
      "step": 12894
    },
    {
      "epoch": 0.25792,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008365885416666666,
      "learning_rate": 0.0001,
      "loss": 4.1924,
      "loss/crossentropy": 2.0535677671432495,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2083517387509346,
      "step": 12896
    },
    {
      "epoch": 0.25796,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.007966105143229167,
      "learning_rate": 0.0001,
      "loss": 4.2951,
      "loss/crossentropy": 2.062114655971527,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20135939121246338,
      "step": 12898
    },
    {
      "epoch": 0.258,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.012482706705729167,
      "learning_rate": 0.0001,
      "loss": 3.8187,
      "loss/crossentropy": 1.968269407749176,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20238329470157623,
      "step": 12900
    },
    {
      "epoch": 0.25804,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0598541259765625,
      "learning_rate": 0.0001,
      "loss": 4.1899,
      "loss/crossentropy": 2.0783804655075073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23015306890010834,
      "step": 12902
    },
    {
      "epoch": 0.25808,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.05815836588541667,
      "learning_rate": 0.0001,
      "loss": 4.1245,
      "loss/crossentropy": 1.8616145253181458,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19802623987197876,
      "step": 12904
    },
    {
      "epoch": 0.25812,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.05834147135416667,
      "learning_rate": 0.0001,
      "loss": 4.2293,
      "loss/crossentropy": 2.0005027651786804,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21621856093406677,
      "step": 12906
    },
    {
      "epoch": 0.25816,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.06181208292643229,
      "learning_rate": 0.0001,
      "loss": 4.0563,
      "loss/crossentropy": 1.796087920665741,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18594232201576233,
      "step": 12908
    },
    {
      "epoch": 0.2582,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.06258316040039062,
      "learning_rate": 0.0001,
      "loss": 4.3161,
      "loss/crossentropy": 2.003196358680725,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19296320527791977,
      "step": 12910
    },
    {
      "epoch": 0.25824,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.06313654581705729,
      "learning_rate": 0.0001,
      "loss": 4.0489,
      "loss/crossentropy": 1.8252119421958923,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1879737451672554,
      "step": 12912
    },
    {
      "epoch": 0.25828,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0629900614420573,
      "learning_rate": 0.0001,
      "loss": 4.2053,
      "loss/crossentropy": 1.9801989793777466,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20867998152971268,
      "step": 12914
    },
    {
      "epoch": 0.25832,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.05583902994791667,
      "learning_rate": 0.0001,
      "loss": 3.9754,
      "loss/crossentropy": 2.04214608669281,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21134068816900253,
      "step": 12916
    },
    {
      "epoch": 0.25836,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.009159342447916666,
      "learning_rate": 0.0001,
      "loss": 4.2202,
      "loss/crossentropy": 2.0669824481010437,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20436694473028183,
      "step": 12918
    },
    {
      "epoch": 0.2584,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.008771769205729167,
      "learning_rate": 0.0001,
      "loss": 4.3616,
      "loss/crossentropy": 1.951616883277893,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19572293758392334,
      "step": 12920
    },
    {
      "epoch": 0.25844,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.010074869791666666,
      "learning_rate": 0.0001,
      "loss": 3.8476,
      "loss/crossentropy": 1.7768954634666443,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1930573582649231,
      "step": 12922
    },
    {
      "epoch": 0.25848,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.020458984375,
      "learning_rate": 0.0001,
      "loss": 3.8775,
      "loss/crossentropy": 1.8527125716209412,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18682826310396194,
      "step": 12924
    },
    {
      "epoch": 0.25852,
      "grad_norm": 2.0,
      "grad_norm_var": 0.018990071614583333,
      "learning_rate": 0.0001,
      "loss": 4.3814,
      "loss/crossentropy": 2.3325828313827515,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21535523980855942,
      "step": 12926
    },
    {
      "epoch": 0.25856,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.02063166300455729,
      "learning_rate": 0.0001,
      "loss": 3.7701,
      "loss/crossentropy": 1.759222686290741,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17470692098140717,
      "step": 12928
    },
    {
      "epoch": 0.2586,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.021247355143229167,
      "learning_rate": 0.0001,
      "loss": 3.8065,
      "loss/crossentropy": 1.944493055343628,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20259656757116318,
      "step": 12930
    },
    {
      "epoch": 0.25864,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.05238825480143229,
      "learning_rate": 0.0001,
      "loss": 4.133,
      "loss/crossentropy": 2.086555302143097,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2180738002061844,
      "step": 12932
    },
    {
      "epoch": 0.25868,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.052779134114583334,
      "learning_rate": 0.0001,
      "loss": 4.1414,
      "loss/crossentropy": 1.9723011255264282,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22814874351024628,
      "step": 12934
    },
    {
      "epoch": 0.25872,
      "grad_norm": 2.125,
      "grad_norm_var": 0.05356038411458333,
      "learning_rate": 0.0001,
      "loss": 4.418,
      "loss/crossentropy": 1.5646896958351135,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18182098120450974,
      "step": 12936
    },
    {
      "epoch": 0.25876,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.054870351155598955,
      "learning_rate": 0.0001,
      "loss": 3.8356,
      "loss/crossentropy": 1.8396940231323242,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19390598684549332,
      "step": 12938
    },
    {
      "epoch": 0.2588,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.041112263997395836,
      "learning_rate": 0.0001,
      "loss": 4.5902,
      "loss/crossentropy": 2.086738705635071,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.209588885307312,
      "step": 12940
    },
    {
      "epoch": 0.25884,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.04136530558268229,
      "learning_rate": 0.0001,
      "loss": 4.2611,
      "loss/crossentropy": 2.0965282917022705,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21021046489477158,
      "step": 12942
    },
    {
      "epoch": 0.25888,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.04311421712239583,
      "learning_rate": 0.0001,
      "loss": 4.0702,
      "loss/crossentropy": 2.2878576517105103,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2203715667128563,
      "step": 12944
    },
    {
      "epoch": 0.25892,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.045446523030598956,
      "learning_rate": 0.0001,
      "loss": 4.419,
      "loss/crossentropy": 2.1753373742103577,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21585986018180847,
      "step": 12946
    },
    {
      "epoch": 0.25896,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.036232248942057295,
      "learning_rate": 0.0001,
      "loss": 4.3735,
      "loss/crossentropy": 1.963489055633545,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20374132692813873,
      "step": 12948
    },
    {
      "epoch": 0.259,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.037751261393229166,
      "learning_rate": 0.0001,
      "loss": 4.4073,
      "loss/crossentropy": 2.3734259605407715,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2375379502773285,
      "step": 12950
    },
    {
      "epoch": 0.25904,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.040710194905598955,
      "learning_rate": 0.0001,
      "loss": 4.184,
      "loss/crossentropy": 2.266390085220337,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22274205088615417,
      "step": 12952
    },
    {
      "epoch": 0.25908,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.03729654947916667,
      "learning_rate": 0.0001,
      "loss": 3.9735,
      "loss/crossentropy": 1.9015297293663025,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20370282977819443,
      "step": 12954
    },
    {
      "epoch": 0.25912,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.03886617024739583,
      "learning_rate": 0.0001,
      "loss": 4.3269,
      "loss/crossentropy": 2.1192378997802734,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26288172602653503,
      "step": 12956
    },
    {
      "epoch": 0.25916,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.03792088826497396,
      "learning_rate": 0.0001,
      "loss": 4.0982,
      "loss/crossentropy": 1.713826835155487,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18557386100292206,
      "step": 12958
    },
    {
      "epoch": 0.2592,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.029545084635416666,
      "learning_rate": 0.0001,
      "loss": 4.2002,
      "loss/crossentropy": 2.0365039706230164,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21086978912353516,
      "step": 12960
    },
    {
      "epoch": 0.25924,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.027852376302083332,
      "learning_rate": 0.0001,
      "loss": 4.2422,
      "loss/crossentropy": 1.6235400438308716,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18112218379974365,
      "step": 12962
    },
    {
      "epoch": 0.25928,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0116455078125,
      "learning_rate": 0.0001,
      "loss": 4.2996,
      "loss/crossentropy": 2.042620003223419,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21463964879512787,
      "step": 12964
    },
    {
      "epoch": 0.25932,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.009611002604166667,
      "learning_rate": 0.0001,
      "loss": 4.0222,
      "loss/crossentropy": 1.6238983273506165,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2171783745288849,
      "step": 12966
    },
    {
      "epoch": 0.25936,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.014070383707682292,
      "learning_rate": 0.0001,
      "loss": 4.0098,
      "loss/crossentropy": 2.3150339126586914,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22038520872592926,
      "step": 12968
    },
    {
      "epoch": 0.2594,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.013301595052083334,
      "learning_rate": 0.0001,
      "loss": 4.2202,
      "loss/crossentropy": 2.3907227516174316,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2269512265920639,
      "step": 12970
    },
    {
      "epoch": 0.25944,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.014378865559895834,
      "learning_rate": 0.0001,
      "loss": 4.1032,
      "loss/crossentropy": 2.1069165468215942,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21007738262414932,
      "step": 12972
    },
    {
      "epoch": 0.25948,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.017942047119140624,
      "learning_rate": 0.0001,
      "loss": 3.9475,
      "loss/crossentropy": 2.277982234954834,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21795185655355453,
      "step": 12974
    },
    {
      "epoch": 0.25952,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.017775217692057293,
      "learning_rate": 0.0001,
      "loss": 4.0159,
      "loss/crossentropy": 1.7227254509925842,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19050486385822296,
      "step": 12976
    },
    {
      "epoch": 0.25956,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0184722900390625,
      "learning_rate": 0.0001,
      "loss": 4.0281,
      "loss/crossentropy": 2.0112340450286865,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21032612770795822,
      "step": 12978
    },
    {
      "epoch": 0.2596,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.019530232747395834,
      "learning_rate": 0.0001,
      "loss": 4.3041,
      "loss/crossentropy": 2.4241796731948853,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22474834322929382,
      "step": 12980
    },
    {
      "epoch": 0.25964,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.019710286458333334,
      "learning_rate": 0.0001,
      "loss": 4.3814,
      "loss/crossentropy": 1.893187701702118,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20330534130334854,
      "step": 12982
    },
    {
      "epoch": 0.25968,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.008467356363932291,
      "learning_rate": 0.0001,
      "loss": 4.1527,
      "loss/crossentropy": 2.045100212097168,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20636773109436035,
      "step": 12984
    },
    {
      "epoch": 0.25972,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.010035959879557292,
      "learning_rate": 0.0001,
      "loss": 3.864,
      "loss/crossentropy": 1.819455087184906,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1775035709142685,
      "step": 12986
    },
    {
      "epoch": 0.25976,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.00985107421875,
      "learning_rate": 0.0001,
      "loss": 4.0553,
      "loss/crossentropy": 2.1358631253242493,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2027900367975235,
      "step": 12988
    },
    {
      "epoch": 0.2598,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.010902659098307291,
      "learning_rate": 0.0001,
      "loss": 4.1552,
      "loss/crossentropy": 1.7847145199775696,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21203257888555527,
      "step": 12990
    },
    {
      "epoch": 0.25984,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.011979166666666667,
      "learning_rate": 0.0001,
      "loss": 4.325,
      "loss/crossentropy": 2.2848687171936035,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22337154299020767,
      "step": 12992
    },
    {
      "epoch": 0.25988,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.012889607747395834,
      "learning_rate": 0.0001,
      "loss": 3.975,
      "loss/crossentropy": 1.9629716277122498,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20660604536533356,
      "step": 12994
    },
    {
      "epoch": 0.25992,
      "grad_norm": 2.0,
      "grad_norm_var": 0.010461171468098959,
      "learning_rate": 0.0001,
      "loss": 4.1521,
      "loss/crossentropy": 2.0321004390716553,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21755626797676086,
      "step": 12996
    },
    {
      "epoch": 0.25996,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0105712890625,
      "learning_rate": 0.0001,
      "loss": 4.1208,
      "loss/crossentropy": 1.950038492679596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1987697035074234,
      "step": 12998
    },
    {
      "epoch": 0.26,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.010544586181640624,
      "learning_rate": 0.0001,
      "loss": 4.3383,
      "loss/crossentropy": 2.2752585411071777,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20959648489952087,
      "step": 13000
    },
    {
      "epoch": 0.26004,
      "grad_norm": 2.125,
      "grad_norm_var": 0.008138020833333334,
      "learning_rate": 0.0001,
      "loss": 4.3132,
      "loss/crossentropy": 2.162104368209839,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21731911599636078,
      "step": 13002
    },
    {
      "epoch": 0.26008,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.009102121988932291,
      "learning_rate": 0.0001,
      "loss": 4.0138,
      "loss/crossentropy": 2.3739081025123596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22912397980690002,
      "step": 13004
    },
    {
      "epoch": 0.26012,
      "grad_norm": 3.0625,
      "grad_norm_var": 0.07608413696289062,
      "learning_rate": 0.0001,
      "loss": 4.3052,
      "loss/crossentropy": 2.161367416381836,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22413842380046844,
      "step": 13006
    },
    {
      "epoch": 0.26016,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.07587661743164062,
      "learning_rate": 0.0001,
      "loss": 4.1435,
      "loss/crossentropy": 1.9589285850524902,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21598373353481293,
      "step": 13008
    },
    {
      "epoch": 0.2602,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0731842041015625,
      "learning_rate": 0.0001,
      "loss": 4.3268,
      "loss/crossentropy": 2.3472602367401123,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22626091539859772,
      "step": 13010
    },
    {
      "epoch": 0.26024,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.07296727498372396,
      "learning_rate": 0.0001,
      "loss": 3.8455,
      "loss/crossentropy": 1.9968576431274414,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20777586847543716,
      "step": 13012
    },
    {
      "epoch": 0.26028,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0729400634765625,
      "learning_rate": 0.0001,
      "loss": 3.8219,
      "loss/crossentropy": 1.7597955465316772,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19565816968679428,
      "step": 13014
    },
    {
      "epoch": 0.26032,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0735992431640625,
      "learning_rate": 0.0001,
      "loss": 4.1446,
      "loss/crossentropy": 1.8115187287330627,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19675451517105103,
      "step": 13016
    },
    {
      "epoch": 0.26036,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.07507909138997396,
      "learning_rate": 0.0001,
      "loss": 4.2018,
      "loss/crossentropy": 1.9728147983551025,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19999082386493683,
      "step": 13018
    },
    {
      "epoch": 0.2604,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.07516988118489583,
      "learning_rate": 0.0001,
      "loss": 4.1434,
      "loss/crossentropy": 2.155986785888672,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20160435140132904,
      "step": 13020
    },
    {
      "epoch": 0.26044,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0081298828125,
      "learning_rate": 0.0001,
      "loss": 4.0647,
      "loss/crossentropy": 2.119171440601349,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2231176346540451,
      "step": 13022
    },
    {
      "epoch": 0.26048,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.010884348551432292,
      "learning_rate": 0.0001,
      "loss": 3.8185,
      "loss/crossentropy": 2.0395994186401367,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21075501292943954,
      "step": 13024
    },
    {
      "epoch": 0.26052,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.014808909098307291,
      "learning_rate": 0.0001,
      "loss": 4.4988,
      "loss/crossentropy": 2.42897891998291,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24882060289382935,
      "step": 13026
    },
    {
      "epoch": 0.26056,
      "grad_norm": 2.25,
      "grad_norm_var": 0.018202463785807293,
      "learning_rate": 0.0001,
      "loss": 4.4995,
      "loss/crossentropy": 2.367077350616455,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21996428817510605,
      "step": 13028
    },
    {
      "epoch": 0.2606,
      "grad_norm": 1.875,
      "grad_norm_var": 0.017207590738932292,
      "learning_rate": 0.0001,
      "loss": 4.0349,
      "loss/crossentropy": 2.071397542953491,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20548687875270844,
      "step": 13030
    },
    {
      "epoch": 0.26064,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.016943359375,
      "learning_rate": 0.0001,
      "loss": 4.0425,
      "loss/crossentropy": 1.764617681503296,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19541263580322266,
      "step": 13032
    },
    {
      "epoch": 0.26068,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.016916656494140626,
      "learning_rate": 0.0001,
      "loss": 4.2657,
      "loss/crossentropy": 2.2966678142547607,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2187207117676735,
      "step": 13034
    },
    {
      "epoch": 0.26072,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0159820556640625,
      "learning_rate": 0.0001,
      "loss": 4.2959,
      "loss/crossentropy": 1.923275649547577,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20180264115333557,
      "step": 13036
    },
    {
      "epoch": 0.26076,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.017207845052083334,
      "learning_rate": 0.0001,
      "loss": 4.3138,
      "loss/crossentropy": 2.246683716773987,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22133737802505493,
      "step": 13038
    },
    {
      "epoch": 0.2608,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.011755116780598958,
      "learning_rate": 0.0001,
      "loss": 4.3685,
      "loss/crossentropy": 2.271313190460205,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23316439241170883,
      "step": 13040
    },
    {
      "epoch": 0.26084,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.008652496337890624,
      "learning_rate": 0.0001,
      "loss": 4.4451,
      "loss/crossentropy": 2.2649654150009155,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.211005300283432,
      "step": 13042
    },
    {
      "epoch": 0.26088,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.008006795247395834,
      "learning_rate": 0.0001,
      "loss": 3.9505,
      "loss/crossentropy": 1.7872197031974792,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17795034497976303,
      "step": 13044
    },
    {
      "epoch": 0.26092,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.007875315348307292,
      "learning_rate": 0.0001,
      "loss": 3.8693,
      "loss/crossentropy": 1.793078064918518,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18739672750234604,
      "step": 13046
    },
    {
      "epoch": 0.26096,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0111572265625,
      "learning_rate": 0.0001,
      "loss": 4.4178,
      "loss/crossentropy": 2.0676932334899902,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2156887650489807,
      "step": 13048
    },
    {
      "epoch": 0.261,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.013484700520833334,
      "learning_rate": 0.0001,
      "loss": 4.2851,
      "loss/crossentropy": 2.2497498989105225,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21906063705682755,
      "step": 13050
    },
    {
      "epoch": 0.26104,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.014994303385416666,
      "learning_rate": 0.0001,
      "loss": 4.0203,
      "loss/crossentropy": 1.901672899723053,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2070387825369835,
      "step": 13052
    },
    {
      "epoch": 0.26108,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.013887532552083333,
      "learning_rate": 0.0001,
      "loss": 4.3772,
      "loss/crossentropy": 2.2334693670272827,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21539244055747986,
      "step": 13054
    },
    {
      "epoch": 0.26112,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.014288075764973958,
      "learning_rate": 0.0001,
      "loss": 4.2916,
      "loss/crossentropy": 2.0297399759292603,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18854249268770218,
      "step": 13056
    },
    {
      "epoch": 0.26116,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.014452107747395833,
      "learning_rate": 0.0001,
      "loss": 4.1115,
      "loss/crossentropy": 2.0964863896369934,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21821416169404984,
      "step": 13058
    },
    {
      "epoch": 0.2612,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.013702138264973959,
      "learning_rate": 0.0001,
      "loss": 4.4918,
      "loss/crossentropy": 2.1526511907577515,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22606930136680603,
      "step": 13060
    },
    {
      "epoch": 0.26124,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0101470947265625,
      "learning_rate": 0.0001,
      "loss": 4.2079,
      "loss/crossentropy": 2.2275509238243103,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21959830820560455,
      "step": 13062
    },
    {
      "epoch": 0.26128,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0086181640625,
      "learning_rate": 0.0001,
      "loss": 4.4246,
      "loss/crossentropy": 2.542263627052307,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22889885306358337,
      "step": 13064
    },
    {
      "epoch": 0.26132,
      "grad_norm": 1.875,
      "grad_norm_var": 0.008585611979166666,
      "learning_rate": 0.0001,
      "loss": 4.0292,
      "loss/crossentropy": 1.7915868163108826,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20242593437433243,
      "step": 13066
    },
    {
      "epoch": 0.26136,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006453450520833333,
      "learning_rate": 0.0001,
      "loss": 4.1415,
      "loss/crossentropy": 2.183286130428314,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.227259561419487,
      "step": 13068
    },
    {
      "epoch": 0.2614,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0063720703125,
      "learning_rate": 0.0001,
      "loss": 4.3211,
      "loss/crossentropy": 2.2707515954971313,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2300432324409485,
      "step": 13070
    },
    {
      "epoch": 0.26144,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0066487630208333336,
      "learning_rate": 0.0001,
      "loss": 4.0237,
      "loss/crossentropy": 1.822485864162445,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19380664825439453,
      "step": 13072
    },
    {
      "epoch": 0.26148,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.008349355061848958,
      "learning_rate": 0.0001,
      "loss": 4.0915,
      "loss/crossentropy": 2.089439034461975,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22531338036060333,
      "step": 13074
    },
    {
      "epoch": 0.26152,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.008318837483723958,
      "learning_rate": 0.0001,
      "loss": 4.3872,
      "loss/crossentropy": 2.0921221375465393,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22256288677453995,
      "step": 13076
    },
    {
      "epoch": 0.26156,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.009065500895182292,
      "learning_rate": 0.0001,
      "loss": 4.4644,
      "loss/crossentropy": 1.8978914022445679,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2677152305841446,
      "step": 13078
    },
    {
      "epoch": 0.2616,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.011643218994140624,
      "learning_rate": 0.0001,
      "loss": 4.405,
      "loss/crossentropy": 2.2540348768234253,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24195441603660583,
      "step": 13080
    },
    {
      "epoch": 0.26164,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.01092529296875,
      "learning_rate": 0.0001,
      "loss": 4.0965,
      "loss/crossentropy": 1.9546288847923279,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19464369863271713,
      "step": 13082
    },
    {
      "epoch": 0.26168,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.012457021077473958,
      "learning_rate": 0.0001,
      "loss": 4.0103,
      "loss/crossentropy": 2.1527568101882935,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22238580137491226,
      "step": 13084
    },
    {
      "epoch": 0.26172,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0128814697265625,
      "learning_rate": 0.0001,
      "loss": 4.3957,
      "loss/crossentropy": 2.111591637134552,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21360620856285095,
      "step": 13086
    },
    {
      "epoch": 0.26176,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.01339111328125,
      "learning_rate": 0.0001,
      "loss": 3.99,
      "loss/crossentropy": 2.006071925163269,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19746285676956177,
      "step": 13088
    },
    {
      "epoch": 0.2618,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.0128814697265625,
      "learning_rate": 0.0001,
      "loss": 4.2073,
      "loss/crossentropy": 2.276778221130371,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21222709119319916,
      "step": 13090
    },
    {
      "epoch": 0.26184,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.013346354166666666,
      "learning_rate": 0.0001,
      "loss": 4.0679,
      "loss/crossentropy": 2.102404534816742,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2049730271100998,
      "step": 13092
    },
    {
      "epoch": 0.26188,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.013719685872395833,
      "learning_rate": 0.0001,
      "loss": 4.439,
      "loss/crossentropy": 2.204727053642273,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21260947734117508,
      "step": 13094
    },
    {
      "epoch": 0.26192,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.010184478759765626,
      "learning_rate": 0.0001,
      "loss": 4.2801,
      "loss/crossentropy": 2.3762032985687256,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22091203182935715,
      "step": 13096
    },
    {
      "epoch": 0.26196,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.010286458333333333,
      "learning_rate": 0.0001,
      "loss": 4.3472,
      "loss/crossentropy": 2.0710391998291016,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20782632380723953,
      "step": 13098
    },
    {
      "epoch": 0.262,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.009319814046223958,
      "learning_rate": 0.0001,
      "loss": 4.3257,
      "loss/crossentropy": 2.193024158477783,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21109049022197723,
      "step": 13100
    },
    {
      "epoch": 0.26204,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.009308878580729167,
      "learning_rate": 0.0001,
      "loss": 4.1863,
      "loss/crossentropy": 1.9668392539024353,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19249404221773148,
      "step": 13102
    },
    {
      "epoch": 0.26208,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.00858154296875,
      "learning_rate": 0.0001,
      "loss": 4.3122,
      "loss/crossentropy": 2.2675795555114746,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26278799772262573,
      "step": 13104
    },
    {
      "epoch": 0.26212,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.007857004801432291,
      "learning_rate": 0.0001,
      "loss": 3.9962,
      "loss/crossentropy": 2.0119330883026123,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20821017771959305,
      "step": 13106
    },
    {
      "epoch": 0.26216,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.006237538655598959,
      "learning_rate": 0.0001,
      "loss": 4.329,
      "loss/crossentropy": 2.1259734630584717,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22454438358545303,
      "step": 13108
    },
    {
      "epoch": 0.2622,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0043413798014322914,
      "learning_rate": 0.0001,
      "loss": 4.2558,
      "loss/crossentropy": 1.8091979622840881,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19353152066469193,
      "step": 13110
    },
    {
      "epoch": 0.26224,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0038655598958333335,
      "learning_rate": 0.0001,
      "loss": 4.1847,
      "loss/crossentropy": 1.9463382363319397,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20932665467262268,
      "step": 13112
    },
    {
      "epoch": 0.26228,
      "grad_norm": 2.125,
      "grad_norm_var": 0.004366048177083333,
      "learning_rate": 0.0001,
      "loss": 4.3975,
      "loss/crossentropy": 2.475601077079773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23451963067054749,
      "step": 13114
    },
    {
      "epoch": 0.26232,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.004412587483723958,
      "learning_rate": 0.0001,
      "loss": 4.4397,
      "loss/crossentropy": 2.0833849906921387,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22380392253398895,
      "step": 13116
    },
    {
      "epoch": 0.26236,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.005301920572916666,
      "learning_rate": 0.0001,
      "loss": 4.0408,
      "loss/crossentropy": 1.9876453876495361,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20823375135660172,
      "step": 13118
    },
    {
      "epoch": 0.2624,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.006750233968098958,
      "learning_rate": 0.0001,
      "loss": 4.5321,
      "loss/crossentropy": 2.184763789176941,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24357233941555023,
      "step": 13120
    },
    {
      "epoch": 0.26244,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0061187744140625,
      "learning_rate": 0.0001,
      "loss": 4.3552,
      "loss/crossentropy": 2.217886805534363,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21944888681173325,
      "step": 13122
    },
    {
      "epoch": 0.26248,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0052886962890625,
      "learning_rate": 0.0001,
      "loss": 4.2443,
      "loss/crossentropy": 2.3299126625061035,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2158309668302536,
      "step": 13124
    },
    {
      "epoch": 0.26252,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0058349609375,
      "learning_rate": 0.0001,
      "loss": 4.1303,
      "loss/crossentropy": 2.0423209071159363,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20835570991039276,
      "step": 13126
    },
    {
      "epoch": 0.26256,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.012325032552083334,
      "learning_rate": 0.0001,
      "loss": 4.31,
      "loss/crossentropy": 2.227811813354492,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2043633684515953,
      "step": 13128
    },
    {
      "epoch": 0.2626,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.011153157552083333,
      "learning_rate": 0.0001,
      "loss": 4.1312,
      "loss/crossentropy": 2.105396568775177,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2212589681148529,
      "step": 13130
    },
    {
      "epoch": 0.26264,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.012737782796223958,
      "learning_rate": 0.0001,
      "loss": 4.2244,
      "loss/crossentropy": 2.1711814999580383,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21528291702270508,
      "step": 13132
    },
    {
      "epoch": 0.26268,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.011937459309895834,
      "learning_rate": 0.0001,
      "loss": 4.1288,
      "loss/crossentropy": 2.0086284279823303,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1974012851715088,
      "step": 13134
    },
    {
      "epoch": 0.26272,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.011205037434895834,
      "learning_rate": 0.0001,
      "loss": 3.895,
      "loss/crossentropy": 1.7025322914123535,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1878160759806633,
      "step": 13136
    },
    {
      "epoch": 0.26276,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.015950520833333332,
      "learning_rate": 0.0001,
      "loss": 4.0134,
      "loss/crossentropy": 1.821410596370697,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.197972871363163,
      "step": 13138
    },
    {
      "epoch": 0.2628,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.018993886311848958,
      "learning_rate": 0.0001,
      "loss": 4.0875,
      "loss/crossentropy": 2.0485053658485413,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20401400327682495,
      "step": 13140
    },
    {
      "epoch": 0.26284,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.018808746337890626,
      "learning_rate": 0.0001,
      "loss": 4.4344,
      "loss/crossentropy": 2.350088357925415,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2638252303004265,
      "step": 13142
    },
    {
      "epoch": 0.26288,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.012308502197265625,
      "learning_rate": 0.0001,
      "loss": 4.5756,
      "loss/crossentropy": 2.3423168659210205,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22735413163900375,
      "step": 13144
    },
    {
      "epoch": 0.26292,
      "grad_norm": 2.0,
      "grad_norm_var": 0.012033843994140625,
      "learning_rate": 0.0001,
      "loss": 4.3848,
      "loss/crossentropy": 2.134036064147949,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20407382398843765,
      "step": 13146
    },
    {
      "epoch": 0.26296,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.009748331705729167,
      "learning_rate": 0.0001,
      "loss": 4.105,
      "loss/crossentropy": 2.0744638442993164,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2107524871826172,
      "step": 13148
    },
    {
      "epoch": 0.263,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.012604777018229167,
      "learning_rate": 0.0001,
      "loss": 4.0507,
      "loss/crossentropy": 2.1522003412246704,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2046833261847496,
      "step": 13150
    },
    {
      "epoch": 0.26304,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.013224283854166666,
      "learning_rate": 0.0001,
      "loss": 4.2754,
      "loss/crossentropy": 2.328645348548889,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24340055882930756,
      "step": 13152
    },
    {
      "epoch": 0.26308,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.009161122639973958,
      "learning_rate": 0.0001,
      "loss": 4.0502,
      "loss/crossentropy": 2.0878008008003235,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2110893502831459,
      "step": 13154
    },
    {
      "epoch": 0.26312,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0074155171712239586,
      "learning_rate": 0.0001,
      "loss": 4.3065,
      "loss/crossentropy": 1.9381126761436462,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18893951922655106,
      "step": 13156
    },
    {
      "epoch": 0.26316,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.009364573160807292,
      "learning_rate": 0.0001,
      "loss": 4.0254,
      "loss/crossentropy": 2.153562545776367,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1973574310541153,
      "step": 13158
    },
    {
      "epoch": 0.2632,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.011370595296223958,
      "learning_rate": 0.0001,
      "loss": 3.998,
      "loss/crossentropy": 2.0583202242851257,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19830025732517242,
      "step": 13160
    },
    {
      "epoch": 0.26324,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.012117258707682292,
      "learning_rate": 0.0001,
      "loss": 4.4843,
      "loss/crossentropy": 2.1951998472213745,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21916336566209793,
      "step": 13162
    },
    {
      "epoch": 0.26328,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.011888631184895833,
      "learning_rate": 0.0001,
      "loss": 4.4944,
      "loss/crossentropy": 2.1612678170204163,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21040990203619003,
      "step": 13164
    },
    {
      "epoch": 0.26332,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.009415690104166667,
      "learning_rate": 0.0001,
      "loss": 4.2012,
      "loss/crossentropy": 1.9372112154960632,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24868668615818024,
      "step": 13166
    },
    {
      "epoch": 0.26336,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.009968058268229166,
      "learning_rate": 0.0001,
      "loss": 4.3605,
      "loss/crossentropy": 2.0220513939857483,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22029200196266174,
      "step": 13168
    },
    {
      "epoch": 0.2634,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.010355631510416666,
      "learning_rate": 0.0001,
      "loss": 3.7251,
      "loss/crossentropy": 1.6749334335327148,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18502884358167648,
      "step": 13170
    },
    {
      "epoch": 0.26344,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.010096995035807292,
      "learning_rate": 0.0001,
      "loss": 4.1394,
      "loss/crossentropy": 2.1463050842285156,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21098337322473526,
      "step": 13172
    },
    {
      "epoch": 0.26348,
      "grad_norm": 2.125,
      "grad_norm_var": 0.009089914957682292,
      "learning_rate": 0.0001,
      "loss": 4.3261,
      "loss/crossentropy": 2.20473051071167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23006527870893478,
      "step": 13174
    },
    {
      "epoch": 0.26352,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.012068684895833333,
      "learning_rate": 0.0001,
      "loss": 4.3289,
      "loss/crossentropy": 2.097456157207489,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2077179104089737,
      "step": 13176
    },
    {
      "epoch": 0.26356,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0124176025390625,
      "learning_rate": 0.0001,
      "loss": 4.402,
      "loss/crossentropy": 2.136400580406189,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2152952179312706,
      "step": 13178
    },
    {
      "epoch": 0.2636,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.013451131184895833,
      "learning_rate": 0.0001,
      "loss": 4.23,
      "loss/crossentropy": 2.0791231393814087,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19688992202281952,
      "step": 13180
    },
    {
      "epoch": 0.26364,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0126617431640625,
      "learning_rate": 0.0001,
      "loss": 4.1233,
      "loss/crossentropy": 2.135149598121643,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22163429111242294,
      "step": 13182
    },
    {
      "epoch": 0.26368,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.013936360677083334,
      "learning_rate": 0.0001,
      "loss": 4.4215,
      "loss/crossentropy": 2.3207671642303467,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24321655184030533,
      "step": 13184
    },
    {
      "epoch": 0.26372,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.013069407145182291,
      "learning_rate": 0.0001,
      "loss": 4.0995,
      "loss/crossentropy": 2.0136974453926086,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19274070858955383,
      "step": 13186
    },
    {
      "epoch": 0.26376,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.013610585530598959,
      "learning_rate": 0.0001,
      "loss": 4.5094,
      "loss/crossentropy": 2.3542726039886475,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24547156691551208,
      "step": 13188
    },
    {
      "epoch": 0.2638,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.016007486979166666,
      "learning_rate": 0.0001,
      "loss": 4.068,
      "loss/crossentropy": 1.905708134174347,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18261709809303284,
      "step": 13190
    },
    {
      "epoch": 0.26384,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.012147776285807292,
      "learning_rate": 0.0001,
      "loss": 4.1426,
      "loss/crossentropy": 2.137321710586548,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2052912563085556,
      "step": 13192
    },
    {
      "epoch": 0.26388,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.010827382405598959,
      "learning_rate": 0.0001,
      "loss": 4.2252,
      "loss/crossentropy": 2.083053410053253,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2038491740822792,
      "step": 13194
    },
    {
      "epoch": 0.26392,
      "grad_norm": 2.0,
      "grad_norm_var": 0.011346181233723959,
      "learning_rate": 0.0001,
      "loss": 3.8349,
      "loss/crossentropy": 1.8006438612937927,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1910744607448578,
      "step": 13196
    },
    {
      "epoch": 0.26396,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.012902577718098959,
      "learning_rate": 0.0001,
      "loss": 4.4104,
      "loss/crossentropy": 2.323628544807434,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2280128449201584,
      "step": 13198
    },
    {
      "epoch": 0.264,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.010137685139973958,
      "learning_rate": 0.0001,
      "loss": 4.1231,
      "loss/crossentropy": 1.9763594269752502,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19776062667369843,
      "step": 13200
    },
    {
      "epoch": 0.26404,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.010601552327473958,
      "learning_rate": 0.0001,
      "loss": 4.0175,
      "loss/crossentropy": 1.8776894807815552,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19732315093278885,
      "step": 13202
    },
    {
      "epoch": 0.26408,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.007478841145833333,
      "learning_rate": 0.0001,
      "loss": 4.0713,
      "loss/crossentropy": 1.826314091682434,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19153253734111786,
      "step": 13204
    },
    {
      "epoch": 0.26412,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.007673136393229167,
      "learning_rate": 0.0001,
      "loss": 4.1503,
      "loss/crossentropy": 2.435014486312866,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22149913012981415,
      "step": 13206
    },
    {
      "epoch": 0.26416,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.01866633097330729,
      "learning_rate": 0.0001,
      "loss": 4.0829,
      "loss/crossentropy": 2.055279493331909,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21625792980194092,
      "step": 13208
    },
    {
      "epoch": 0.2642,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.018641916910807292,
      "learning_rate": 0.0001,
      "loss": 4.216,
      "loss/crossentropy": 2.4019049406051636,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22626115381717682,
      "step": 13210
    },
    {
      "epoch": 0.26424,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.017508697509765626,
      "learning_rate": 0.0001,
      "loss": 3.9822,
      "loss/crossentropy": 2.336918354034424,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22773776203393936,
      "step": 13212
    },
    {
      "epoch": 0.26428,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.01608454386393229,
      "learning_rate": 0.0001,
      "loss": 4.3929,
      "loss/crossentropy": 2.0398528575897217,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20827088505029678,
      "step": 13214
    },
    {
      "epoch": 0.26432,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.015419260660807291,
      "learning_rate": 0.0001,
      "loss": 4.026,
      "loss/crossentropy": 2.166857123374939,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2136707603931427,
      "step": 13216
    },
    {
      "epoch": 0.26436,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.015095011393229166,
      "learning_rate": 0.0001,
      "loss": 4.2282,
      "loss/crossentropy": 1.8454258441925049,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20678912848234177,
      "step": 13218
    },
    {
      "epoch": 0.2644,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.014642079671223959,
      "learning_rate": 0.0001,
      "loss": 4.2966,
      "loss/crossentropy": 2.249597668647766,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22785719484090805,
      "step": 13220
    },
    {
      "epoch": 0.26444,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.014134724934895834,
      "learning_rate": 0.0001,
      "loss": 3.875,
      "loss/crossentropy": 1.7729946374893188,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2014644593000412,
      "step": 13222
    },
    {
      "epoch": 0.26448,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.004976145426432292,
      "learning_rate": 0.0001,
      "loss": 3.9144,
      "loss/crossentropy": 1.7968116998672485,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1889382004737854,
      "step": 13224
    },
    {
      "epoch": 0.26452,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.006510162353515625,
      "learning_rate": 0.0001,
      "loss": 4.2015,
      "loss/crossentropy": 2.0765512585639954,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20551903545856476,
      "step": 13226
    },
    {
      "epoch": 0.26456,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.007933553059895833,
      "learning_rate": 0.0001,
      "loss": 4.2603,
      "loss/crossentropy": 2.2007554173469543,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2250915989279747,
      "step": 13228
    },
    {
      "epoch": 0.2646,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.008184560139973958,
      "learning_rate": 0.0001,
      "loss": 4.2014,
      "loss/crossentropy": 2.319425046443939,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22913866490125656,
      "step": 13230
    },
    {
      "epoch": 0.26464,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.008365631103515625,
      "learning_rate": 0.0001,
      "loss": 4.1395,
      "loss/crossentropy": 1.9845139980316162,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20769847929477692,
      "step": 13232
    },
    {
      "epoch": 0.26468,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.008674112955729167,
      "learning_rate": 0.0001,
      "loss": 4.0985,
      "loss/crossentropy": 1.94157075881958,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2314443141222,
      "step": 13234
    },
    {
      "epoch": 0.26472,
      "grad_norm": 2.0,
      "grad_norm_var": 0.007470703125,
      "learning_rate": 0.0001,
      "loss": 4.3428,
      "loss/crossentropy": 2.0699294209480286,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22432812303304672,
      "step": 13236
    },
    {
      "epoch": 0.26476,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.006841786702473958,
      "learning_rate": 0.0001,
      "loss": 3.983,
      "loss/crossentropy": 1.7656533122062683,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20837824791669846,
      "step": 13238
    },
    {
      "epoch": 0.2648,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0060618082682291664,
      "learning_rate": 0.0001,
      "loss": 4.1799,
      "loss/crossentropy": 1.9473227858543396,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20151428878307343,
      "step": 13240
    },
    {
      "epoch": 0.26484,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.005387369791666667,
      "learning_rate": 0.0001,
      "loss": 4.284,
      "loss/crossentropy": 2.060440719127655,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2041771411895752,
      "step": 13242
    },
    {
      "epoch": 0.26488,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.003922526041666667,
      "learning_rate": 0.0001,
      "loss": 3.7503,
      "loss/crossentropy": 2.0440456867218018,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2041776031255722,
      "step": 13244
    },
    {
      "epoch": 0.26492,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.004622395833333333,
      "learning_rate": 0.0001,
      "loss": 4.1921,
      "loss/crossentropy": 1.9192551374435425,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18886462599039078,
      "step": 13246
    },
    {
      "epoch": 0.26496,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0072100321451822914,
      "learning_rate": 0.0001,
      "loss": 4.1673,
      "loss/crossentropy": 1.7112281918525696,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18532422930002213,
      "step": 13248
    },
    {
      "epoch": 0.265,
      "grad_norm": 2.0,
      "grad_norm_var": 0.007637532552083334,
      "learning_rate": 0.0001,
      "loss": 4.0561,
      "loss/crossentropy": 1.9225355386734009,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19994951784610748,
      "step": 13250
    },
    {
      "epoch": 0.26504,
      "grad_norm": 2.0,
      "grad_norm_var": 0.007950846354166667,
      "learning_rate": 0.0001,
      "loss": 3.6825,
      "loss/crossentropy": 2.0243517756462097,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19785288721323013,
      "step": 13252
    },
    {
      "epoch": 0.26508,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.007413482666015625,
      "learning_rate": 0.0001,
      "loss": 4.2929,
      "loss/crossentropy": 2.255920171737671,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21220286190509796,
      "step": 13254
    },
    {
      "epoch": 0.26512,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.008475748697916667,
      "learning_rate": 0.0001,
      "loss": 4.4145,
      "loss/crossentropy": 2.3024203777313232,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22255335003137589,
      "step": 13256
    },
    {
      "epoch": 0.26516,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.00733642578125,
      "learning_rate": 0.0001,
      "loss": 4.1386,
      "loss/crossentropy": 2.0467708110809326,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20193494111299515,
      "step": 13258
    },
    {
      "epoch": 0.2652,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.006064605712890625,
      "learning_rate": 0.0001,
      "loss": 4.2188,
      "loss/crossentropy": 2.1543976068496704,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2257647141814232,
      "step": 13260
    },
    {
      "epoch": 0.26524,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.006373850504557291,
      "learning_rate": 0.0001,
      "loss": 4.2683,
      "loss/crossentropy": 2.077743351459503,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21964918076992035,
      "step": 13262
    },
    {
      "epoch": 0.26528,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0044921875,
      "learning_rate": 0.0001,
      "loss": 3.963,
      "loss/crossentropy": 1.6356236338615417,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1847509667277336,
      "step": 13264
    },
    {
      "epoch": 0.26532,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.006925201416015625,
      "learning_rate": 0.0001,
      "loss": 4.3559,
      "loss/crossentropy": 2.0876659750938416,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22080402821302414,
      "step": 13266
    },
    {
      "epoch": 0.26536,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.005747222900390625,
      "learning_rate": 0.0001,
      "loss": 4.0452,
      "loss/crossentropy": 1.7057855129241943,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18772340565919876,
      "step": 13268
    },
    {
      "epoch": 0.2654,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0078277587890625,
      "learning_rate": 0.0001,
      "loss": 4.0032,
      "loss/crossentropy": 2.0550594329833984,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18500665575265884,
      "step": 13270
    },
    {
      "epoch": 0.26544,
      "grad_norm": 2.0,
      "grad_norm_var": 0.011131795247395833,
      "learning_rate": 0.0001,
      "loss": 4.2837,
      "loss/crossentropy": 2.027509331703186,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21377252787351608,
      "step": 13272
    },
    {
      "epoch": 0.26548,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.012393951416015625,
      "learning_rate": 0.0001,
      "loss": 3.9739,
      "loss/crossentropy": 2.0517578125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21172185242176056,
      "step": 13274
    },
    {
      "epoch": 0.26552,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.01395263671875,
      "learning_rate": 0.0001,
      "loss": 4.1628,
      "loss/crossentropy": 2.0550750494003296,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22374523431062698,
      "step": 13276
    },
    {
      "epoch": 0.26556,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.013399251302083333,
      "learning_rate": 0.0001,
      "loss": 4.1829,
      "loss/crossentropy": 1.8055049777030945,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17029780894517899,
      "step": 13278
    },
    {
      "epoch": 0.2656,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.013114166259765626,
      "learning_rate": 0.0001,
      "loss": 4.3999,
      "loss/crossentropy": 1.9650321006774902,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20327500253915787,
      "step": 13280
    },
    {
      "epoch": 0.26564,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.010209147135416667,
      "learning_rate": 0.0001,
      "loss": 4.1395,
      "loss/crossentropy": 2.164494276046753,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21579623967409134,
      "step": 13282
    },
    {
      "epoch": 0.26568,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.009810129801432291,
      "learning_rate": 0.0001,
      "loss": 4.0676,
      "loss/crossentropy": 2.0488376021385193,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20213299989700317,
      "step": 13284
    },
    {
      "epoch": 0.26572,
      "grad_norm": 2.0,
      "grad_norm_var": 0.010308583577473959,
      "learning_rate": 0.0001,
      "loss": 3.7442,
      "loss/crossentropy": 1.948447048664093,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19875720143318176,
      "step": 13286
    },
    {
      "epoch": 0.26576,
      "grad_norm": 1.875,
      "grad_norm_var": 0.008487701416015625,
      "learning_rate": 0.0001,
      "loss": 4.2312,
      "loss/crossentropy": 2.115865170955658,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19675103574991226,
      "step": 13288
    },
    {
      "epoch": 0.2658,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0105377197265625,
      "learning_rate": 0.0001,
      "loss": 4.2289,
      "loss/crossentropy": 1.9720736145973206,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20975899696350098,
      "step": 13290
    },
    {
      "epoch": 0.26584,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.009154256184895833,
      "learning_rate": 0.0001,
      "loss": 3.9258,
      "loss/crossentropy": 1.9105132222175598,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20582708716392517,
      "step": 13292
    },
    {
      "epoch": 0.26588,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.018369293212890624,
      "learning_rate": 0.0001,
      "loss": 4.2629,
      "loss/crossentropy": 2.089789867401123,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21264012902975082,
      "step": 13294
    },
    {
      "epoch": 0.26592,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.018358357747395835,
      "learning_rate": 0.0001,
      "loss": 4.41,
      "loss/crossentropy": 2.271156430244446,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21146492660045624,
      "step": 13296
    },
    {
      "epoch": 0.26596,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0196929931640625,
      "learning_rate": 0.0001,
      "loss": 3.8554,
      "loss/crossentropy": 1.930766224861145,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1959189623594284,
      "step": 13298
    },
    {
      "epoch": 0.266,
      "grad_norm": 2.25,
      "grad_norm_var": 0.023374176025390624,
      "learning_rate": 0.0001,
      "loss": 4.3551,
      "loss/crossentropy": 1.9945995807647705,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20959854125976562,
      "step": 13300
    },
    {
      "epoch": 0.26604,
      "grad_norm": 2.125,
      "grad_norm_var": 0.020637003580729167,
      "learning_rate": 0.0001,
      "loss": 4.5445,
      "loss/crossentropy": 2.2128443717956543,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21266601234674454,
      "step": 13302
    },
    {
      "epoch": 0.26608,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.018049112955729165,
      "learning_rate": 0.0001,
      "loss": 4.2578,
      "loss/crossentropy": 2.076206088066101,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2037106677889824,
      "step": 13304
    },
    {
      "epoch": 0.26612,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.018724568684895835,
      "learning_rate": 0.0001,
      "loss": 4.0512,
      "loss/crossentropy": 2.077200174331665,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.228537917137146,
      "step": 13306
    },
    {
      "epoch": 0.26616,
      "grad_norm": 2.0,
      "grad_norm_var": 0.01874567667643229,
      "learning_rate": 0.0001,
      "loss": 3.9641,
      "loss/crossentropy": 2.0368794202804565,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2150697112083435,
      "step": 13308
    },
    {
      "epoch": 0.2662,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.016556803385416666,
      "learning_rate": 0.0001,
      "loss": 3.8516,
      "loss/crossentropy": 1.9108307361602783,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1903306469321251,
      "step": 13310
    },
    {
      "epoch": 0.26624,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.016218058268229165,
      "learning_rate": 0.0001,
      "loss": 3.9928,
      "loss/crossentropy": 1.8118465542793274,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18557358533143997,
      "step": 13312
    },
    {
      "epoch": 0.26628,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.01736424763997396,
      "learning_rate": 0.0001,
      "loss": 3.897,
      "loss/crossentropy": 1.7392275929450989,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19214990735054016,
      "step": 13314
    },
    {
      "epoch": 0.26632,
      "grad_norm": 3.359375,
      "grad_norm_var": 0.12773412068684895,
      "learning_rate": 0.0001,
      "loss": 3.9728,
      "loss/crossentropy": 2.1361005306243896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22042304277420044,
      "step": 13316
    },
    {
      "epoch": 0.26636,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.12786026000976564,
      "learning_rate": 0.0001,
      "loss": 4.0055,
      "loss/crossentropy": 2.0087279677391052,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20278790593147278,
      "step": 13318
    },
    {
      "epoch": 0.2664,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.12776692708333334,
      "learning_rate": 0.0001,
      "loss": 4.0781,
      "loss/crossentropy": 2.307933807373047,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24929606914520264,
      "step": 13320
    },
    {
      "epoch": 0.26644,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.12690022786458333,
      "learning_rate": 0.0001,
      "loss": 4.3451,
      "loss/crossentropy": 2.5539438724517822,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2219071164727211,
      "step": 13322
    },
    {
      "epoch": 0.26648,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.12910334269205728,
      "learning_rate": 0.0001,
      "loss": 3.9984,
      "loss/crossentropy": 1.9069242477416992,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19276316463947296,
      "step": 13324
    },
    {
      "epoch": 0.26652,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.12194010416666666,
      "learning_rate": 0.0001,
      "loss": 4.535,
      "loss/crossentropy": 2.0715484619140625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20079267024993896,
      "step": 13326
    },
    {
      "epoch": 0.26656,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.12164713541666666,
      "learning_rate": 0.0001,
      "loss": 4.46,
      "loss/crossentropy": 2.1134172677993774,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20564979314804077,
      "step": 13328
    },
    {
      "epoch": 0.2666,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.11581929524739583,
      "learning_rate": 0.0001,
      "loss": 4.3992,
      "loss/crossentropy": 2.103445053100586,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19361437857151031,
      "step": 13330
    },
    {
      "epoch": 0.26664,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.014499664306640625,
      "learning_rate": 0.0001,
      "loss": 4.0169,
      "loss/crossentropy": 2.1882529258728027,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21402588486671448,
      "step": 13332
    },
    {
      "epoch": 0.26668,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.015498606363932292,
      "learning_rate": 0.0001,
      "loss": 4.3247,
      "loss/crossentropy": 2.2531981468200684,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2019234597682953,
      "step": 13334
    },
    {
      "epoch": 0.26672,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.021954091389973958,
      "learning_rate": 0.0001,
      "loss": 3.8596,
      "loss/crossentropy": 1.8077877759933472,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19121356308460236,
      "step": 13336
    },
    {
      "epoch": 0.26676,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.02600072224934896,
      "learning_rate": 0.0001,
      "loss": 4.3195,
      "loss/crossentropy": 1.7387139797210693,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20326050370931625,
      "step": 13338
    },
    {
      "epoch": 0.2668,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.022712198893229167,
      "learning_rate": 0.0001,
      "loss": 4.065,
      "loss/crossentropy": 2.2061930894851685,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22271326184272766,
      "step": 13340
    },
    {
      "epoch": 0.26684,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.02191162109375,
      "learning_rate": 0.0001,
      "loss": 4.4018,
      "loss/crossentropy": 2.096329092979431,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2201632410287857,
      "step": 13342
    },
    {
      "epoch": 0.26688,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0232574462890625,
      "learning_rate": 0.0001,
      "loss": 4.0783,
      "loss/crossentropy": 2.2015284299850464,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22004209458827972,
      "step": 13344
    },
    {
      "epoch": 0.26692,
      "grad_norm": 2.125,
      "grad_norm_var": 0.023957316080729166,
      "learning_rate": 0.0001,
      "loss": 4.5227,
      "loss/crossentropy": 2.256517231464386,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21762817353010178,
      "step": 13346
    },
    {
      "epoch": 0.26696,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.022874959309895835,
      "learning_rate": 0.0001,
      "loss": 3.8578,
      "loss/crossentropy": 1.8551252484321594,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18950944393873215,
      "step": 13348
    },
    {
      "epoch": 0.267,
      "grad_norm": 2.0,
      "grad_norm_var": 0.020967610677083335,
      "learning_rate": 0.0001,
      "loss": 4.3051,
      "loss/crossentropy": 1.8834841847419739,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17950639128684998,
      "step": 13350
    },
    {
      "epoch": 0.26704,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.012872060139973959,
      "learning_rate": 0.0001,
      "loss": 4.0234,
      "loss/crossentropy": 2.0059815645217896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2279331535100937,
      "step": 13352
    },
    {
      "epoch": 0.26708,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.00594482421875,
      "learning_rate": 0.0001,
      "loss": 4.0102,
      "loss/crossentropy": 2.2327537536621094,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22790935635566711,
      "step": 13354
    },
    {
      "epoch": 0.26712,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0066650390625,
      "learning_rate": 0.0001,
      "loss": 4.0535,
      "loss/crossentropy": 1.7623894214630127,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17889687418937683,
      "step": 13356
    },
    {
      "epoch": 0.26716,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.006205240885416667,
      "learning_rate": 0.0001,
      "loss": 4.1595,
      "loss/crossentropy": 1.9240076541900635,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18552126735448837,
      "step": 13358
    },
    {
      "epoch": 0.2672,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.007387034098307292,
      "learning_rate": 0.0001,
      "loss": 3.9162,
      "loss/crossentropy": 2.301589012145996,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20251981914043427,
      "step": 13360
    },
    {
      "epoch": 0.26724,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.007671864827473959,
      "learning_rate": 0.0001,
      "loss": 4.1575,
      "loss/crossentropy": 2.144057512283325,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22049692273139954,
      "step": 13362
    },
    {
      "epoch": 0.26728,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.006209309895833333,
      "learning_rate": 0.0001,
      "loss": 4.2381,
      "loss/crossentropy": 1.9329981207847595,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2015545517206192,
      "step": 13364
    },
    {
      "epoch": 0.26732,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0062896728515625,
      "learning_rate": 0.0001,
      "loss": 4.2544,
      "loss/crossentropy": 2.057462990283966,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19826926290988922,
      "step": 13366
    },
    {
      "epoch": 0.26736,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.006670888264973958,
      "learning_rate": 0.0001,
      "loss": 4.085,
      "loss/crossentropy": 2.097291588783264,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18807460367679596,
      "step": 13368
    },
    {
      "epoch": 0.2674,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.007281239827473958,
      "learning_rate": 0.0001,
      "loss": 4.1204,
      "loss/crossentropy": 2.154082179069519,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22133710980415344,
      "step": 13370
    },
    {
      "epoch": 0.26744,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.017561848958333334,
      "learning_rate": 0.0001,
      "loss": 4.1412,
      "loss/crossentropy": 2.208779454231262,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2139016017317772,
      "step": 13372
    },
    {
      "epoch": 0.26748,
      "grad_norm": 2.0,
      "grad_norm_var": 0.018355305989583334,
      "learning_rate": 0.0001,
      "loss": 3.8929,
      "loss/crossentropy": 2.251901865005493,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22264431416988373,
      "step": 13374
    },
    {
      "epoch": 0.26752,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.016857655843098958,
      "learning_rate": 0.0001,
      "loss": 4.3243,
      "loss/crossentropy": 2.4881285429000854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.250653475522995,
      "step": 13376
    },
    {
      "epoch": 0.26756,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.017427571614583335,
      "learning_rate": 0.0001,
      "loss": 4.2806,
      "loss/crossentropy": 2.104843556880951,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19004888087511063,
      "step": 13378
    },
    {
      "epoch": 0.2676,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.022899373372395834,
      "learning_rate": 0.0001,
      "loss": 4.0708,
      "loss/crossentropy": 1.913047194480896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20572267472743988,
      "step": 13380
    },
    {
      "epoch": 0.26764,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.02463353474934896,
      "learning_rate": 0.0001,
      "loss": 4.3632,
      "loss/crossentropy": 2.1043838262557983,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21029697358608246,
      "step": 13382
    },
    {
      "epoch": 0.26768,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.02617162068684896,
      "learning_rate": 0.0001,
      "loss": 4.2429,
      "loss/crossentropy": 2.055271625518799,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2113131284713745,
      "step": 13384
    },
    {
      "epoch": 0.26772,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.0488433837890625,
      "learning_rate": 0.0001,
      "loss": 4.4956,
      "loss/crossentropy": 1.8407886624336243,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2729990780353546,
      "step": 13386
    },
    {
      "epoch": 0.26776,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.04177017211914062,
      "learning_rate": 0.0001,
      "loss": 4.0097,
      "loss/crossentropy": 2.0368717908859253,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21415862441062927,
      "step": 13388
    },
    {
      "epoch": 0.2678,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.040897369384765625,
      "learning_rate": 0.0001,
      "loss": 4.1836,
      "loss/crossentropy": 1.8235292434692383,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2062123641371727,
      "step": 13390
    },
    {
      "epoch": 0.26784,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.04222183227539063,
      "learning_rate": 0.0001,
      "loss": 3.8392,
      "loss/crossentropy": 1.733048439025879,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20651907473802567,
      "step": 13392
    },
    {
      "epoch": 0.26788,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.043454742431640624,
      "learning_rate": 0.0001,
      "loss": 3.9004,
      "loss/crossentropy": 2.1339367628097534,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21730080246925354,
      "step": 13394
    },
    {
      "epoch": 0.26792,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0452789306640625,
      "learning_rate": 0.0001,
      "loss": 3.8421,
      "loss/crossentropy": 1.885707974433899,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20833835750818253,
      "step": 13396
    },
    {
      "epoch": 0.26796,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.04429423014322917,
      "learning_rate": 0.0001,
      "loss": 3.9714,
      "loss/crossentropy": 1.9896257519721985,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1938292756676674,
      "step": 13398
    },
    {
      "epoch": 0.268,
      "grad_norm": 2.25,
      "grad_norm_var": 0.04457575480143229,
      "learning_rate": 0.0001,
      "loss": 4.2962,
      "loss/crossentropy": 2.0359573364257812,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21572312712669373,
      "step": 13400
    },
    {
      "epoch": 0.26804,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.014095052083333334,
      "learning_rate": 0.0001,
      "loss": 4.3483,
      "loss/crossentropy": 2.241006851196289,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2361859679222107,
      "step": 13402
    },
    {
      "epoch": 0.26808,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.013852691650390625,
      "learning_rate": 0.0001,
      "loss": 4.0518,
      "loss/crossentropy": 2.0988917350769043,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20220966637134552,
      "step": 13404
    },
    {
      "epoch": 0.26812,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.014802805582682292,
      "learning_rate": 0.0001,
      "loss": 4.3664,
      "loss/crossentropy": 2.0803651213645935,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22251859307289124,
      "step": 13406
    },
    {
      "epoch": 0.26816,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.015827433268229166,
      "learning_rate": 0.0001,
      "loss": 4.0473,
      "loss/crossentropy": 1.9151242971420288,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19792036712169647,
      "step": 13408
    },
    {
      "epoch": 0.2682,
      "grad_norm": 2.96875,
      "grad_norm_var": 0.07619196573893229,
      "learning_rate": 0.0001,
      "loss": 4.0321,
      "loss/crossentropy": 1.908549964427948,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21568355709314346,
      "step": 13410
    },
    {
      "epoch": 0.26824,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.06887919108072917,
      "learning_rate": 0.0001,
      "loss": 4.3844,
      "loss/crossentropy": 2.3226535320281982,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2332221046090126,
      "step": 13412
    },
    {
      "epoch": 0.26828,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.06696548461914062,
      "learning_rate": 0.0001,
      "loss": 4.24,
      "loss/crossentropy": 1.9767839312553406,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2038547545671463,
      "step": 13414
    },
    {
      "epoch": 0.26832,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.06800028483072916,
      "learning_rate": 0.0001,
      "loss": 3.9592,
      "loss/crossentropy": 2.2013003826141357,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2076154574751854,
      "step": 13416
    },
    {
      "epoch": 0.26836,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.06597671508789063,
      "learning_rate": 0.0001,
      "loss": 4.1128,
      "loss/crossentropy": 2.0941338539123535,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20538556575775146,
      "step": 13418
    },
    {
      "epoch": 0.2684,
      "grad_norm": 2.125,
      "grad_norm_var": 0.06494954427083334,
      "learning_rate": 0.0001,
      "loss": 4.3656,
      "loss/crossentropy": 2.0206560492515564,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20901528000831604,
      "step": 13420
    },
    {
      "epoch": 0.26844,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.06690648396809896,
      "learning_rate": 0.0001,
      "loss": 4.2606,
      "loss/crossentropy": 2.1317169070243835,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21660470962524414,
      "step": 13422
    },
    {
      "epoch": 0.26848,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.07109273274739583,
      "learning_rate": 0.0001,
      "loss": 3.6026,
      "loss/crossentropy": 1.631429135799408,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1729787141084671,
      "step": 13424
    },
    {
      "epoch": 0.26852,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.012491607666015625,
      "learning_rate": 0.0001,
      "loss": 3.923,
      "loss/crossentropy": 1.8428707122802734,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18231894075870514,
      "step": 13426
    },
    {
      "epoch": 0.26856,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.011104075113932292,
      "learning_rate": 0.0001,
      "loss": 4.2351,
      "loss/crossentropy": 1.860277235507965,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18769516795873642,
      "step": 13428
    },
    {
      "epoch": 0.2686,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.012542470296223959,
      "learning_rate": 0.0001,
      "loss": 4.0199,
      "loss/crossentropy": 2.02046799659729,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2160683423280716,
      "step": 13430
    },
    {
      "epoch": 0.26864,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.021491495768229167,
      "learning_rate": 0.0001,
      "loss": 4.5105,
      "loss/crossentropy": 2.094564437866211,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22155898064374924,
      "step": 13432
    },
    {
      "epoch": 0.26868,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.02215550740559896,
      "learning_rate": 0.0001,
      "loss": 4.0382,
      "loss/crossentropy": 2.013366401195526,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2126912623643875,
      "step": 13434
    },
    {
      "epoch": 0.26872,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.02212702433268229,
      "learning_rate": 0.0001,
      "loss": 4.5826,
      "loss/crossentropy": 2.360998272895813,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2355523630976677,
      "step": 13436
    },
    {
      "epoch": 0.26876,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.02289606730143229,
      "learning_rate": 0.0001,
      "loss": 3.7129,
      "loss/crossentropy": 1.754651427268982,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18666986376047134,
      "step": 13438
    },
    {
      "epoch": 0.2688,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.01812922159830729,
      "learning_rate": 0.0001,
      "loss": 4.1024,
      "loss/crossentropy": 1.9176424741744995,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19241741299629211,
      "step": 13440
    },
    {
      "epoch": 0.26884,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0154449462890625,
      "learning_rate": 0.0001,
      "loss": 4.2548,
      "loss/crossentropy": 2.324304223060608,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22302107512950897,
      "step": 13442
    },
    {
      "epoch": 0.26888,
      "grad_norm": 2.5,
      "grad_norm_var": 0.028758748372395834,
      "learning_rate": 0.0001,
      "loss": 4.0693,
      "loss/crossentropy": 1.8527624011039734,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18644367158412933,
      "step": 13444
    },
    {
      "epoch": 0.26892,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.027570597330729165,
      "learning_rate": 0.0001,
      "loss": 4.4437,
      "loss/crossentropy": 2.0999260544776917,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21172572672367096,
      "step": 13446
    },
    {
      "epoch": 0.26896,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.026627349853515624,
      "learning_rate": 0.0001,
      "loss": 3.8125,
      "loss/crossentropy": 1.8577081561088562,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17302225530147552,
      "step": 13448
    },
    {
      "epoch": 0.269,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.026956939697265626,
      "learning_rate": 0.0001,
      "loss": 4.2182,
      "loss/crossentropy": 2.11561119556427,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20249811559915543,
      "step": 13450
    },
    {
      "epoch": 0.26904,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.026656087239583334,
      "learning_rate": 0.0001,
      "loss": 4.2854,
      "loss/crossentropy": 2.3036913871765137,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20101947337388992,
      "step": 13452
    },
    {
      "epoch": 0.26908,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.02341283162434896,
      "learning_rate": 0.0001,
      "loss": 3.9013,
      "loss/crossentropy": 2.0498175621032715,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19935546070337296,
      "step": 13454
    },
    {
      "epoch": 0.26912,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.08842137654622396,
      "learning_rate": 0.0001,
      "loss": 4.2601,
      "loss/crossentropy": 2.27053964138031,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31231266260147095,
      "step": 13456
    },
    {
      "epoch": 0.26916,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0908953348795573,
      "learning_rate": 0.0001,
      "loss": 4.0035,
      "loss/crossentropy": 2.191486120223999,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2058379054069519,
      "step": 13458
    },
    {
      "epoch": 0.2692,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.07780939737955729,
      "learning_rate": 0.0001,
      "loss": 4.1193,
      "loss/crossentropy": 2.008154332637787,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20862725377082825,
      "step": 13460
    },
    {
      "epoch": 0.26924,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.07779947916666667,
      "learning_rate": 0.0001,
      "loss": 4.3191,
      "loss/crossentropy": 2.3374814987182617,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24534277617931366,
      "step": 13462
    },
    {
      "epoch": 0.26928,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.07355855305989584,
      "learning_rate": 0.0001,
      "loss": 4.0558,
      "loss/crossentropy": 2.2654502391815186,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21406329423189163,
      "step": 13464
    },
    {
      "epoch": 0.26932,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.07412821451822917,
      "learning_rate": 0.0001,
      "loss": 4.2078,
      "loss/crossentropy": 2.1021856665611267,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19653601199388504,
      "step": 13466
    },
    {
      "epoch": 0.26936,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0727068583170573,
      "learning_rate": 0.0001,
      "loss": 4.4466,
      "loss/crossentropy": 1.9482329487800598,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20822357386350632,
      "step": 13468
    },
    {
      "epoch": 0.2694,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.07814915974934895,
      "learning_rate": 0.0001,
      "loss": 4.1036,
      "loss/crossentropy": 2.122725486755371,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2132444903254509,
      "step": 13470
    },
    {
      "epoch": 0.26944,
      "grad_norm": 2.0,
      "grad_norm_var": 0.015044911702473959,
      "learning_rate": 0.0001,
      "loss": 4.0337,
      "loss/crossentropy": 1.760430932044983,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1835266426205635,
      "step": 13472
    },
    {
      "epoch": 0.26948,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.014847564697265624,
      "learning_rate": 0.0001,
      "loss": 4.0338,
      "loss/crossentropy": 1.610491931438446,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19731693714857101,
      "step": 13474
    },
    {
      "epoch": 0.26952,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.015313466389973959,
      "learning_rate": 0.0001,
      "loss": 4.213,
      "loss/crossentropy": 2.1966941356658936,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22074176371097565,
      "step": 13476
    },
    {
      "epoch": 0.26956,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.029904937744140624,
      "learning_rate": 0.0001,
      "loss": 4.1291,
      "loss/crossentropy": 2.045714259147644,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2193753719329834,
      "step": 13478
    },
    {
      "epoch": 0.2696,
      "grad_norm": 2.0,
      "grad_norm_var": 0.029886881510416668,
      "learning_rate": 0.0001,
      "loss": 4.0127,
      "loss/crossentropy": 1.860603928565979,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1900448128581047,
      "step": 13480
    },
    {
      "epoch": 0.26964,
      "grad_norm": 2.125,
      "grad_norm_var": 0.028449503580729167,
      "learning_rate": 0.0001,
      "loss": 4.0643,
      "loss/crossentropy": 2.171591639518738,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21908972412347794,
      "step": 13482
    },
    {
      "epoch": 0.26968,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.030265299479166667,
      "learning_rate": 0.0001,
      "loss": 4.1351,
      "loss/crossentropy": 2.271330237388611,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21823390573263168,
      "step": 13484
    },
    {
      "epoch": 0.26972,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.026875813802083332,
      "learning_rate": 0.0001,
      "loss": 4.192,
      "loss/crossentropy": 2.155557870864868,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20512574911117554,
      "step": 13486
    },
    {
      "epoch": 0.26976,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.027164459228515625,
      "learning_rate": 0.0001,
      "loss": 4.2561,
      "loss/crossentropy": 2.3827039003372192,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22485698014497757,
      "step": 13488
    },
    {
      "epoch": 0.2698,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.028636678059895834,
      "learning_rate": 0.0001,
      "loss": 3.9937,
      "loss/crossentropy": 2.225351929664612,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22235903143882751,
      "step": 13490
    },
    {
      "epoch": 0.26984,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.027717081705729167,
      "learning_rate": 0.0001,
      "loss": 4.0144,
      "loss/crossentropy": 2.1125447750091553,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20206717401742935,
      "step": 13492
    },
    {
      "epoch": 0.26988,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.013639068603515625,
      "learning_rate": 0.0001,
      "loss": 4.0318,
      "loss/crossentropy": 2.30017626285553,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22321031987667084,
      "step": 13494
    },
    {
      "epoch": 0.26992,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0145904541015625,
      "learning_rate": 0.0001,
      "loss": 4.3979,
      "loss/crossentropy": 2.416603446006775,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2413351908326149,
      "step": 13496
    },
    {
      "epoch": 0.26996,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.013185373942057292,
      "learning_rate": 0.0001,
      "loss": 4.0436,
      "loss/crossentropy": 2.06734299659729,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23090296238660812,
      "step": 13498
    },
    {
      "epoch": 0.27,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0050961812337239586,
      "learning_rate": 0.0001,
      "loss": 4.0,
      "loss/crossentropy": 1.8088473677635193,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1988995596766472,
      "step": 13500
    },
    {
      "epoch": 0.27004,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.005092112223307291,
      "learning_rate": 0.0001,
      "loss": 4.0205,
      "loss/crossentropy": 1.7075524926185608,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18463176488876343,
      "step": 13502
    },
    {
      "epoch": 0.27008,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.0083160400390625,
      "learning_rate": 0.0001,
      "loss": 4.2291,
      "loss/crossentropy": 2.1184898018836975,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20941460877656937,
      "step": 13504
    },
    {
      "epoch": 0.27012,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.007736968994140625,
      "learning_rate": 0.0001,
      "loss": 4.0139,
      "loss/crossentropy": 1.8351567387580872,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18913155794143677,
      "step": 13506
    },
    {
      "epoch": 0.27016,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.00914306640625,
      "learning_rate": 0.0001,
      "loss": 3.935,
      "loss/crossentropy": 2.090391516685486,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21020452678203583,
      "step": 13508
    },
    {
      "epoch": 0.2702,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0078521728515625,
      "learning_rate": 0.0001,
      "loss": 4.1805,
      "loss/crossentropy": 2.0680218935012817,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20998934656381607,
      "step": 13510
    },
    {
      "epoch": 0.27024,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.007236480712890625,
      "learning_rate": 0.0001,
      "loss": 4.0074,
      "loss/crossentropy": 2.258071780204773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22634688764810562,
      "step": 13512
    },
    {
      "epoch": 0.27028,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.03350397745768229,
      "learning_rate": 0.0001,
      "loss": 4.775,
      "loss/crossentropy": 1.931319773197174,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.29265259206295013,
      "step": 13514
    },
    {
      "epoch": 0.27032,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.03398844401041667,
      "learning_rate": 0.0001,
      "loss": 4.3408,
      "loss/crossentropy": 2.45454478263855,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22465243190526962,
      "step": 13516
    },
    {
      "epoch": 0.27036,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.034063466389973956,
      "learning_rate": 0.0001,
      "loss": 4.2279,
      "loss/crossentropy": 2.005887746810913,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19818827509880066,
      "step": 13518
    },
    {
      "epoch": 0.2704,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.031107330322265626,
      "learning_rate": 0.0001,
      "loss": 4.318,
      "loss/crossentropy": 2.2329577207565308,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21015001833438873,
      "step": 13520
    },
    {
      "epoch": 0.27044,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.032373046875,
      "learning_rate": 0.0001,
      "loss": 4.1026,
      "loss/crossentropy": 2.1716688871383667,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20930524915456772,
      "step": 13522
    },
    {
      "epoch": 0.27048,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.030651601155598958,
      "learning_rate": 0.0001,
      "loss": 3.957,
      "loss/crossentropy": 1.659675419330597,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1963459849357605,
      "step": 13524
    },
    {
      "epoch": 0.27052,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.035359700520833336,
      "learning_rate": 0.0001,
      "loss": 4.2214,
      "loss/crossentropy": 1.9285706877708435,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1708521470427513,
      "step": 13526
    },
    {
      "epoch": 0.27056,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.03286031087239583,
      "learning_rate": 0.0001,
      "loss": 4.1771,
      "loss/crossentropy": 2.0428953170776367,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20893662422895432,
      "step": 13528
    },
    {
      "epoch": 0.2706,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.011655426025390625,
      "learning_rate": 0.0001,
      "loss": 3.9885,
      "loss/crossentropy": 1.979922592639923,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19091727584600449,
      "step": 13530
    },
    {
      "epoch": 0.27064,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.015922037760416667,
      "learning_rate": 0.0001,
      "loss": 4.3176,
      "loss/crossentropy": 2.1879321336746216,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24813223630189896,
      "step": 13532
    },
    {
      "epoch": 0.27068,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.01631647745768229,
      "learning_rate": 0.0001,
      "loss": 4.1852,
      "loss/crossentropy": 2.0450875759124756,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21059003472328186,
      "step": 13534
    },
    {
      "epoch": 0.27072,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.01673151652018229,
      "learning_rate": 0.0001,
      "loss": 4.3549,
      "loss/crossentropy": 2.050394892692566,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21235206723213196,
      "step": 13536
    },
    {
      "epoch": 0.27076,
      "grad_norm": 2.0,
      "grad_norm_var": 0.014793904622395833,
      "learning_rate": 0.0001,
      "loss": 4.0128,
      "loss/crossentropy": 2.0687233805656433,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22125782817602158,
      "step": 13538
    },
    {
      "epoch": 0.2708,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.014647420247395833,
      "learning_rate": 0.0001,
      "loss": 4.1247,
      "loss/crossentropy": 1.8587198853492737,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19718395173549652,
      "step": 13540
    },
    {
      "epoch": 0.27084,
      "grad_norm": 2.125,
      "grad_norm_var": 0.010223134358723959,
      "learning_rate": 0.0001,
      "loss": 4.2992,
      "loss/crossentropy": 2.2416387796401978,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23797060549259186,
      "step": 13542
    },
    {
      "epoch": 0.27088,
      "grad_norm": 4.8125,
      "grad_norm_var": 0.495751953125,
      "learning_rate": 0.0001,
      "loss": 4.1875,
      "loss/crossentropy": 2.251328468322754,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23915010690689087,
      "step": 13544
    },
    {
      "epoch": 0.27092,
      "grad_norm": 2.125,
      "grad_norm_var": 0.4896074930826823,
      "learning_rate": 0.0001,
      "loss": 4.0601,
      "loss/crossentropy": 1.990403652191162,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21710015833377838,
      "step": 13546
    },
    {
      "epoch": 0.27096,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.4871070861816406,
      "learning_rate": 0.0001,
      "loss": 4.38,
      "loss/crossentropy": 2.176861047744751,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2153458073735237,
      "step": 13548
    },
    {
      "epoch": 0.271,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.4867286682128906,
      "learning_rate": 0.0001,
      "loss": 4.0311,
      "loss/crossentropy": 1.581967830657959,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1787950098514557,
      "step": 13550
    },
    {
      "epoch": 0.27104,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.4920183817545573,
      "learning_rate": 0.0001,
      "loss": 3.9803,
      "loss/crossentropy": 1.827072560787201,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18847094476222992,
      "step": 13552
    },
    {
      "epoch": 0.27108,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.49279683430989585,
      "learning_rate": 0.0001,
      "loss": 3.9381,
      "loss/crossentropy": 1.5748514533042908,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18339257687330246,
      "step": 13554
    },
    {
      "epoch": 0.27112,
      "grad_norm": 2.125,
      "grad_norm_var": 0.4897989908854167,
      "learning_rate": 0.0001,
      "loss": 4.2545,
      "loss/crossentropy": 2.3187586069107056,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2535529136657715,
      "step": 13556
    },
    {
      "epoch": 0.27116,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.4885660807291667,
      "learning_rate": 0.0001,
      "loss": 4.4848,
      "loss/crossentropy": 2.3675668239593506,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23080945760011673,
      "step": 13558
    },
    {
      "epoch": 0.2712,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0066912333170572914,
      "learning_rate": 0.0001,
      "loss": 4.2032,
      "loss/crossentropy": 2.3976542949676514,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2418106645345688,
      "step": 13560
    },
    {
      "epoch": 0.27124,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.006221262613932291,
      "learning_rate": 0.0001,
      "loss": 4.1877,
      "loss/crossentropy": 2.1341328024864197,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21845046430826187,
      "step": 13562
    },
    {
      "epoch": 0.27128,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006154123942057292,
      "learning_rate": 0.0001,
      "loss": 4.3067,
      "loss/crossentropy": 2.0519689321517944,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21991144120693207,
      "step": 13564
    },
    {
      "epoch": 0.27132,
      "grad_norm": 2.0,
      "grad_norm_var": 0.005812327067057292,
      "learning_rate": 0.0001,
      "loss": 4.106,
      "loss/crossentropy": 1.9690070748329163,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20945511013269424,
      "step": 13566
    },
    {
      "epoch": 0.27136,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.004609934488932292,
      "learning_rate": 0.0001,
      "loss": 4.0929,
      "loss/crossentropy": 1.9553492069244385,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2299523577094078,
      "step": 13568
    },
    {
      "epoch": 0.2714,
      "grad_norm": 2.25,
      "grad_norm_var": 0.0055501302083333336,
      "learning_rate": 0.0001,
      "loss": 4.158,
      "loss/crossentropy": 1.8214278817176819,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1801338642835617,
      "step": 13570
    },
    {
      "epoch": 0.27144,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.011822255452473958,
      "learning_rate": 0.0001,
      "loss": 3.8658,
      "loss/crossentropy": 1.7834638953208923,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18421506136655807,
      "step": 13572
    },
    {
      "epoch": 0.27148,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.013097890218098958,
      "learning_rate": 0.0001,
      "loss": 3.8404,
      "loss/crossentropy": 1.7703429460525513,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18263264745473862,
      "step": 13574
    },
    {
      "epoch": 0.27152,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.0155181884765625,
      "learning_rate": 0.0001,
      "loss": 4.1489,
      "loss/crossentropy": 2.228934168815613,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21034922450780869,
      "step": 13576
    },
    {
      "epoch": 0.27156,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.015379842122395833,
      "learning_rate": 0.0001,
      "loss": 4.4665,
      "loss/crossentropy": 2.160655975341797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22300932556390762,
      "step": 13578
    },
    {
      "epoch": 0.2716,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0150634765625,
      "learning_rate": 0.0001,
      "loss": 4.2042,
      "loss/crossentropy": 2.1582624912261963,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21843481063842773,
      "step": 13580
    },
    {
      "epoch": 0.27164,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0150146484375,
      "learning_rate": 0.0001,
      "loss": 4.2513,
      "loss/crossentropy": 2.0660162568092346,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1985679790377617,
      "step": 13582
    },
    {
      "epoch": 0.27168,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.015433502197265626,
      "learning_rate": 0.0001,
      "loss": 3.873,
      "loss/crossentropy": 1.839695692062378,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1772506907582283,
      "step": 13584
    },
    {
      "epoch": 0.27172,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.012345123291015624,
      "learning_rate": 0.0001,
      "loss": 3.9649,
      "loss/crossentropy": 2.0837132930755615,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19461007416248322,
      "step": 13586
    },
    {
      "epoch": 0.27176,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.00750732421875,
      "learning_rate": 0.0001,
      "loss": 4.1986,
      "loss/crossentropy": 1.910473346710205,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20272061973810196,
      "step": 13588
    },
    {
      "epoch": 0.2718,
      "grad_norm": 2.921875,
      "grad_norm_var": 0.054402669270833336,
      "learning_rate": 0.0001,
      "loss": 4.2259,
      "loss/crossentropy": 2.2427467107772827,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21372541785240173,
      "step": 13590
    },
    {
      "epoch": 0.27184,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.055272420247395836,
      "learning_rate": 0.0001,
      "loss": 4.1123,
      "loss/crossentropy": 2.165425181388855,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2552504763007164,
      "step": 13592
    },
    {
      "epoch": 0.27188,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.05620930989583333,
      "learning_rate": 0.0001,
      "loss": 4.0993,
      "loss/crossentropy": 1.9684009552001953,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19894393533468246,
      "step": 13594
    },
    {
      "epoch": 0.27192,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.055863444010416666,
      "learning_rate": 0.0001,
      "loss": 4.2421,
      "loss/crossentropy": 2.0572606325149536,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20954644680023193,
      "step": 13596
    },
    {
      "epoch": 0.27196,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.0611724853515625,
      "learning_rate": 0.0001,
      "loss": 3.5511,
      "loss/crossentropy": 1.8215171694755554,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1905388981103897,
      "step": 13598
    },
    {
      "epoch": 0.272,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.06299209594726562,
      "learning_rate": 0.0001,
      "loss": 4.2407,
      "loss/crossentropy": 2.114215850830078,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21045731008052826,
      "step": 13600
    },
    {
      "epoch": 0.27204,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.06243057250976562,
      "learning_rate": 0.0001,
      "loss": 4.1409,
      "loss/crossentropy": 2.301407814025879,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2148626372218132,
      "step": 13602
    },
    {
      "epoch": 0.27208,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.06350809733072917,
      "learning_rate": 0.0001,
      "loss": 4.4367,
      "loss/crossentropy": 2.0508424639701843,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21292225271463394,
      "step": 13604
    },
    {
      "epoch": 0.27212,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.014826456705729166,
      "learning_rate": 0.0001,
      "loss": 4.0024,
      "loss/crossentropy": 2.095071792602539,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23258864879608154,
      "step": 13606
    },
    {
      "epoch": 0.27216,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.013802083333333333,
      "learning_rate": 0.0001,
      "loss": 4.1722,
      "loss/crossentropy": 2.3613405227661133,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20384693890810013,
      "step": 13608
    },
    {
      "epoch": 0.2722,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0142578125,
      "learning_rate": 0.0001,
      "loss": 4.1397,
      "loss/crossentropy": 2.1618664264678955,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21691302955150604,
      "step": 13610
    },
    {
      "epoch": 0.27224,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.015620676676432292,
      "learning_rate": 0.0001,
      "loss": 3.8335,
      "loss/crossentropy": 1.6352717280387878,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18613358587026596,
      "step": 13612
    },
    {
      "epoch": 0.27228,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.012851715087890625,
      "learning_rate": 0.0001,
      "loss": 4.2405,
      "loss/crossentropy": 2.304627537727356,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21715758740901947,
      "step": 13614
    },
    {
      "epoch": 0.27232,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.009357706705729166,
      "learning_rate": 0.0001,
      "loss": 4.1845,
      "loss/crossentropy": 2.232232451438904,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20997462421655655,
      "step": 13616
    },
    {
      "epoch": 0.27236,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.011503092447916667,
      "learning_rate": 0.0001,
      "loss": 4.1273,
      "loss/crossentropy": 2.0350120663642883,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20753738284111023,
      "step": 13618
    },
    {
      "epoch": 0.2724,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.015746053059895834,
      "learning_rate": 0.0001,
      "loss": 4.1955,
      "loss/crossentropy": 1.892416536808014,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1920388638973236,
      "step": 13620
    },
    {
      "epoch": 0.27244,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0128814697265625,
      "learning_rate": 0.0001,
      "loss": 4.3283,
      "loss/crossentropy": 2.1694198846817017,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.217724971473217,
      "step": 13622
    },
    {
      "epoch": 0.27248,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.013444010416666667,
      "learning_rate": 0.0001,
      "loss": 3.9786,
      "loss/crossentropy": 2.1418001651763916,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2016705498099327,
      "step": 13624
    },
    {
      "epoch": 0.27252,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.014240519205729166,
      "learning_rate": 0.0001,
      "loss": 4.3992,
      "loss/crossentropy": 2.32223117351532,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21616849303245544,
      "step": 13626
    },
    {
      "epoch": 0.27256,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.013044230143229167,
      "learning_rate": 0.0001,
      "loss": 4.1847,
      "loss/crossentropy": 1.9991823434829712,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2077011615037918,
      "step": 13628
    },
    {
      "epoch": 0.2726,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.012434641520182291,
      "learning_rate": 0.0001,
      "loss": 4.1331,
      "loss/crossentropy": 2.1381043195724487,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20748750865459442,
      "step": 13630
    },
    {
      "epoch": 0.27264,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.012303670247395834,
      "learning_rate": 0.0001,
      "loss": 4.2215,
      "loss/crossentropy": 2.018588602542877,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20897169411182404,
      "step": 13632
    },
    {
      "epoch": 0.27268,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.01114501953125,
      "learning_rate": 0.0001,
      "loss": 4.3528,
      "loss/crossentropy": 2.1859233379364014,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21695572137832642,
      "step": 13634
    },
    {
      "epoch": 0.27272,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0060198465983072914,
      "learning_rate": 0.0001,
      "loss": 4.2106,
      "loss/crossentropy": 2.17002010345459,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21605493128299713,
      "step": 13636
    },
    {
      "epoch": 0.27276,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006058502197265625,
      "learning_rate": 0.0001,
      "loss": 4.261,
      "loss/crossentropy": 2.186649441719055,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2172158733010292,
      "step": 13638
    },
    {
      "epoch": 0.2728,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.007306925455729167,
      "learning_rate": 0.0001,
      "loss": 3.8394,
      "loss/crossentropy": 1.7324808835983276,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17334696650505066,
      "step": 13640
    },
    {
      "epoch": 0.27284,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0049550374348958336,
      "learning_rate": 0.0001,
      "loss": 3.9805,
      "loss/crossentropy": 1.7768054008483887,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1750154346227646,
      "step": 13642
    },
    {
      "epoch": 0.27288,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0047515869140625,
      "learning_rate": 0.0001,
      "loss": 4.2292,
      "loss/crossentropy": 1.9629738330841064,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19489944726228714,
      "step": 13644
    },
    {
      "epoch": 0.27292,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.006882476806640625,
      "learning_rate": 0.0001,
      "loss": 4.4096,
      "loss/crossentropy": 2.144750416278839,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19367601722478867,
      "step": 13646
    },
    {
      "epoch": 0.27296,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0067291259765625,
      "learning_rate": 0.0001,
      "loss": 4.1301,
      "loss/crossentropy": 1.9281827211380005,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19894402474164963,
      "step": 13648
    },
    {
      "epoch": 0.273,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.007132720947265625,
      "learning_rate": 0.0001,
      "loss": 4.1998,
      "loss/crossentropy": 1.8092535138130188,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18536384403705597,
      "step": 13650
    },
    {
      "epoch": 0.27304,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.007726796468098958,
      "learning_rate": 0.0001,
      "loss": 4.1697,
      "loss/crossentropy": 2.5707184076309204,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23067744076251984,
      "step": 13652
    },
    {
      "epoch": 0.27308,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008434804280598958,
      "learning_rate": 0.0001,
      "loss": 4.0189,
      "loss/crossentropy": 2.108555316925049,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2052021473646164,
      "step": 13654
    },
    {
      "epoch": 0.27312,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.007228342692057291,
      "learning_rate": 0.0001,
      "loss": 4.3412,
      "loss/crossentropy": 1.836738109588623,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20734501630067825,
      "step": 13656
    },
    {
      "epoch": 0.27316,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.007990519205729166,
      "learning_rate": 0.0001,
      "loss": 4.0783,
      "loss/crossentropy": 2.223781406879425,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20618071407079697,
      "step": 13658
    },
    {
      "epoch": 0.2732,
      "grad_norm": 2.0,
      "grad_norm_var": 0.007207997639973958,
      "learning_rate": 0.0001,
      "loss": 4.1722,
      "loss/crossentropy": 2.028991222381592,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20579702407121658,
      "step": 13660
    },
    {
      "epoch": 0.27324,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.006915028889973958,
      "learning_rate": 0.0001,
      "loss": 4.0294,
      "loss/crossentropy": 2.1003119349479675,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21900994330644608,
      "step": 13662
    },
    {
      "epoch": 0.27328,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.007289377848307291,
      "learning_rate": 0.0001,
      "loss": 4.1695,
      "loss/crossentropy": 2.1813069581985474,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2117675319314003,
      "step": 13664
    },
    {
      "epoch": 0.27332,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0105621337890625,
      "learning_rate": 0.0001,
      "loss": 3.8666,
      "loss/crossentropy": 2.2203832864761353,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20088640600442886,
      "step": 13666
    },
    {
      "epoch": 0.27336,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0103759765625,
      "learning_rate": 0.0001,
      "loss": 4.1084,
      "loss/crossentropy": 1.8810867071151733,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20646882802248,
      "step": 13668
    },
    {
      "epoch": 0.2734,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.010060373942057292,
      "learning_rate": 0.0001,
      "loss": 4.0746,
      "loss/crossentropy": 2.2147200107574463,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2038613259792328,
      "step": 13670
    },
    {
      "epoch": 0.27344,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.008408355712890624,
      "learning_rate": 0.0001,
      "loss": 4.2151,
      "loss/crossentropy": 2.3654199838638306,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22163932025432587,
      "step": 13672
    },
    {
      "epoch": 0.27348,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0086669921875,
      "learning_rate": 0.0001,
      "loss": 4.1584,
      "loss/crossentropy": 2.422420859336853,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23172692954540253,
      "step": 13674
    },
    {
      "epoch": 0.27352,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008634440104166667,
      "learning_rate": 0.0001,
      "loss": 4.391,
      "loss/crossentropy": 2.0776702165603638,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2030162662267685,
      "step": 13676
    },
    {
      "epoch": 0.27356,
      "grad_norm": 2.125,
      "grad_norm_var": 0.009611002604166667,
      "learning_rate": 0.0001,
      "loss": 3.9108,
      "loss/crossentropy": 1.7956212162971497,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1932467818260193,
      "step": 13678
    },
    {
      "epoch": 0.2736,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.010573069254557291,
      "learning_rate": 0.0001,
      "loss": 4.1768,
      "loss/crossentropy": 2.176727533340454,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20450271666049957,
      "step": 13680
    },
    {
      "epoch": 0.27364,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0062255859375,
      "learning_rate": 0.0001,
      "loss": 4.061,
      "loss/crossentropy": 2.0534290075302124,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21150045841932297,
      "step": 13682
    },
    {
      "epoch": 0.27368,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0061920166015625,
      "learning_rate": 0.0001,
      "loss": 4.1682,
      "loss/crossentropy": 1.726151466369629,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19256682693958282,
      "step": 13684
    },
    {
      "epoch": 0.27372,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.005956013997395833,
      "learning_rate": 0.0001,
      "loss": 4.2666,
      "loss/crossentropy": 2.2024354934692383,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2222209945321083,
      "step": 13686
    },
    {
      "epoch": 0.27376,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.007347615559895834,
      "learning_rate": 0.0001,
      "loss": 4.3252,
      "loss/crossentropy": 2.095518469810486,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2092512547969818,
      "step": 13688
    },
    {
      "epoch": 0.2738,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0074460347493489586,
      "learning_rate": 0.0001,
      "loss": 4.0389,
      "loss/crossentropy": 1.9978103637695312,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20576970279216766,
      "step": 13690
    },
    {
      "epoch": 0.27384,
      "grad_norm": 1.875,
      "grad_norm_var": 0.008599599202473959,
      "learning_rate": 0.0001,
      "loss": 4.1023,
      "loss/crossentropy": 2.233831286430359,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20514021068811417,
      "step": 13692
    },
    {
      "epoch": 0.27388,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.010251617431640625,
      "learning_rate": 0.0001,
      "loss": 4.0479,
      "loss/crossentropy": 1.6729156970977783,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18054980039596558,
      "step": 13694
    },
    {
      "epoch": 0.27392,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.010910797119140624,
      "learning_rate": 0.0001,
      "loss": 4.1911,
      "loss/crossentropy": 2.0113388895988464,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21645111590623856,
      "step": 13696
    },
    {
      "epoch": 0.27396,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.010469563802083333,
      "learning_rate": 0.0001,
      "loss": 4.3487,
      "loss/crossentropy": 1.9311461448669434,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23587358742952347,
      "step": 13698
    },
    {
      "epoch": 0.274,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.01889012654622396,
      "learning_rate": 0.0001,
      "loss": 4.1099,
      "loss/crossentropy": 2.1205984354019165,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20603877305984497,
      "step": 13700
    },
    {
      "epoch": 0.27404,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.01889012654622396,
      "learning_rate": 0.0001,
      "loss": 4.2947,
      "loss/crossentropy": 2.1700201630592346,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21855031698942184,
      "step": 13702
    },
    {
      "epoch": 0.27408,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.022484334309895833,
      "learning_rate": 0.0001,
      "loss": 3.7717,
      "loss/crossentropy": 1.5643411874771118,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.161862351000309,
      "step": 13704
    },
    {
      "epoch": 0.27412,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.024881998697916668,
      "learning_rate": 0.0001,
      "loss": 3.7292,
      "loss/crossentropy": 2.032666802406311,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20749235153198242,
      "step": 13706
    },
    {
      "epoch": 0.27416,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.024738566080729166,
      "learning_rate": 0.0001,
      "loss": 4.0899,
      "loss/crossentropy": 1.9313859939575195,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19065556675195694,
      "step": 13708
    },
    {
      "epoch": 0.2742,
      "grad_norm": 2.0,
      "grad_norm_var": 0.02225519816080729,
      "learning_rate": 0.0001,
      "loss": 4.3295,
      "loss/crossentropy": 2.1200226545333862,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1967960000038147,
      "step": 13710
    },
    {
      "epoch": 0.27424,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.020765940348307293,
      "learning_rate": 0.0001,
      "loss": 4.0655,
      "loss/crossentropy": 2.144526958465576,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20019693672657013,
      "step": 13712
    },
    {
      "epoch": 0.27428,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.01809666951497396,
      "learning_rate": 0.0001,
      "loss": 4.2039,
      "loss/crossentropy": 1.9802407622337341,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20123689621686935,
      "step": 13714
    },
    {
      "epoch": 0.27432,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0062334696451822914,
      "learning_rate": 0.0001,
      "loss": 3.9415,
      "loss/crossentropy": 2.096527099609375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20282836258411407,
      "step": 13716
    },
    {
      "epoch": 0.27436,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.005558013916015625,
      "learning_rate": 0.0001,
      "loss": 4.1829,
      "loss/crossentropy": 2.1037757992744446,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2039530873298645,
      "step": 13718
    },
    {
      "epoch": 0.2744,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.004752349853515625,
      "learning_rate": 0.0001,
      "loss": 3.8416,
      "loss/crossentropy": 1.5415751934051514,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18478797376155853,
      "step": 13720
    },
    {
      "epoch": 0.27444,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.001859283447265625,
      "learning_rate": 0.0001,
      "loss": 4.071,
      "loss/crossentropy": 2.0357913970947266,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20176240801811218,
      "step": 13722
    },
    {
      "epoch": 0.27448,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.0034543355305989582,
      "learning_rate": 0.0001,
      "loss": 3.8472,
      "loss/crossentropy": 1.9013367891311646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20012690126895905,
      "step": 13724
    },
    {
      "epoch": 0.27452,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.00343017578125,
      "learning_rate": 0.0001,
      "loss": 3.8072,
      "loss/crossentropy": 1.5798682570457458,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18169572949409485,
      "step": 13726
    },
    {
      "epoch": 0.27456,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0034624735514322915,
      "learning_rate": 0.0001,
      "loss": 4.163,
      "loss/crossentropy": 2.196288228034973,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22311393916606903,
      "step": 13728
    },
    {
      "epoch": 0.2746,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.003928375244140625,
      "learning_rate": 0.0001,
      "loss": 4.0172,
      "loss/crossentropy": 1.714774489402771,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17624646425247192,
      "step": 13730
    },
    {
      "epoch": 0.27464,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.0053484598795572914,
      "learning_rate": 0.0001,
      "loss": 4.3264,
      "loss/crossentropy": 1.9853646159172058,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18827050924301147,
      "step": 13732
    },
    {
      "epoch": 0.27468,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.0074045817057291664,
      "learning_rate": 0.0001,
      "loss": 4.0074,
      "loss/crossentropy": 1.7830212116241455,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1801598072052002,
      "step": 13734
    },
    {
      "epoch": 0.27472,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.008046213785807292,
      "learning_rate": 0.0001,
      "loss": 3.7923,
      "loss/crossentropy": 1.7400763034820557,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1930072009563446,
      "step": 13736
    },
    {
      "epoch": 0.27476,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.008421834309895833,
      "learning_rate": 0.0001,
      "loss": 4.0992,
      "loss/crossentropy": 2.2292455434799194,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20573973655700684,
      "step": 13738
    },
    {
      "epoch": 0.2748,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.009056599934895833,
      "learning_rate": 0.0001,
      "loss": 4.4305,
      "loss/crossentropy": 2.1399097442626953,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2265445813536644,
      "step": 13740
    },
    {
      "epoch": 0.27484,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.009366861979166667,
      "learning_rate": 0.0001,
      "loss": 4.0596,
      "loss/crossentropy": 2.049125075340271,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20623435080051422,
      "step": 13742
    },
    {
      "epoch": 0.27488,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.010689036051432291,
      "learning_rate": 0.0001,
      "loss": 4.2678,
      "loss/crossentropy": 2.388680577278137,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23650053143501282,
      "step": 13744
    },
    {
      "epoch": 0.27492,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.019577789306640624,
      "learning_rate": 0.0001,
      "loss": 4.5091,
      "loss/crossentropy": 2.05421245098114,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19932237267494202,
      "step": 13746
    },
    {
      "epoch": 0.27496,
      "grad_norm": 2.0,
      "grad_norm_var": 0.01969172159830729,
      "learning_rate": 0.0001,
      "loss": 3.8274,
      "loss/crossentropy": 1.892092227935791,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1907898187637329,
      "step": 13748
    },
    {
      "epoch": 0.275,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.018507639567057293,
      "learning_rate": 0.0001,
      "loss": 4.2196,
      "loss/crossentropy": 2.160037875175476,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24229361861944199,
      "step": 13750
    },
    {
      "epoch": 0.27504,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.017451985677083334,
      "learning_rate": 0.0001,
      "loss": 4.1278,
      "loss/crossentropy": 2.5185710191726685,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2394627332687378,
      "step": 13752
    },
    {
      "epoch": 0.27508,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.01754735310872396,
      "learning_rate": 0.0001,
      "loss": 4.1551,
      "loss/crossentropy": 2.0613635778427124,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19758722186088562,
      "step": 13754
    },
    {
      "epoch": 0.27512,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.019496409098307292,
      "learning_rate": 0.0001,
      "loss": 4.1905,
      "loss/crossentropy": 1.9780999422073364,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21773972362279892,
      "step": 13756
    },
    {
      "epoch": 0.27516,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.01987482706705729,
      "learning_rate": 0.0001,
      "loss": 4.3297,
      "loss/crossentropy": 2.1116772890090942,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20809265226125717,
      "step": 13758
    },
    {
      "epoch": 0.2752,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.021201324462890626,
      "learning_rate": 0.0001,
      "loss": 4.2079,
      "loss/crossentropy": 2.1072784662246704,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21189533174037933,
      "step": 13760
    },
    {
      "epoch": 0.27524,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.014902496337890625,
      "learning_rate": 0.0001,
      "loss": 3.9102,
      "loss/crossentropy": 2.261967897415161,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2058071494102478,
      "step": 13762
    },
    {
      "epoch": 0.27528,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.01571044921875,
      "learning_rate": 0.0001,
      "loss": 4.0684,
      "loss/crossentropy": 2.240622043609619,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2305934578180313,
      "step": 13764
    },
    {
      "epoch": 0.27532,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.013874308268229166,
      "learning_rate": 0.0001,
      "loss": 3.8268,
      "loss/crossentropy": 1.6203233003616333,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17164954543113708,
      "step": 13766
    },
    {
      "epoch": 0.27536,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.013521067301432292,
      "learning_rate": 0.0001,
      "loss": 3.928,
      "loss/crossentropy": 2.171602725982666,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20083081722259521,
      "step": 13768
    },
    {
      "epoch": 0.2754,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.012645467122395834,
      "learning_rate": 0.0001,
      "loss": 4.2738,
      "loss/crossentropy": 2.2198195457458496,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21113458275794983,
      "step": 13770
    },
    {
      "epoch": 0.27544,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.012296549479166667,
      "learning_rate": 0.0001,
      "loss": 4.2833,
      "loss/crossentropy": 2.1037912368774414,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21217171102762222,
      "step": 13772
    },
    {
      "epoch": 0.27548,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.012015533447265626,
      "learning_rate": 0.0001,
      "loss": 3.9824,
      "loss/crossentropy": 2.252090096473694,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20365406572818756,
      "step": 13774
    },
    {
      "epoch": 0.27552,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.011701456705729167,
      "learning_rate": 0.0001,
      "loss": 3.9513,
      "loss/crossentropy": 1.8044906258583069,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1854797750711441,
      "step": 13776
    },
    {
      "epoch": 0.27556,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.012284342447916667,
      "learning_rate": 0.0001,
      "loss": 4.5229,
      "loss/crossentropy": 2.2397952675819397,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.214948832988739,
      "step": 13778
    },
    {
      "epoch": 0.2756,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.0164215087890625,
      "learning_rate": 0.0001,
      "loss": 4.3569,
      "loss/crossentropy": 1.776510238647461,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19995201379060745,
      "step": 13780
    },
    {
      "epoch": 0.27564,
      "grad_norm": 4.0625,
      "grad_norm_var": 0.26768290201822914,
      "learning_rate": 0.0001,
      "loss": 4.3061,
      "loss/crossentropy": 2.470343589782715,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2194378450512886,
      "step": 13782
    },
    {
      "epoch": 0.27568,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.26253433227539064,
      "learning_rate": 0.0001,
      "loss": 4.0708,
      "loss/crossentropy": 2.1340490579605103,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20280296355485916,
      "step": 13784
    },
    {
      "epoch": 0.27572,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.26368179321289065,
      "learning_rate": 0.0001,
      "loss": 4.1072,
      "loss/crossentropy": 1.6801128387451172,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16750500351190567,
      "step": 13786
    },
    {
      "epoch": 0.27576,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.26387939453125,
      "learning_rate": 0.0001,
      "loss": 3.9902,
      "loss/crossentropy": 1.9632562398910522,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19115211814641953,
      "step": 13788
    },
    {
      "epoch": 0.2758,
      "grad_norm": 2.125,
      "grad_norm_var": 0.25916519165039065,
      "learning_rate": 0.0001,
      "loss": 4.2139,
      "loss/crossentropy": 2.0817149877548218,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21430402994155884,
      "step": 13790
    },
    {
      "epoch": 0.27584,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.253088124593099,
      "learning_rate": 0.0001,
      "loss": 4.3244,
      "loss/crossentropy": 2.2736687660217285,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22024155408143997,
      "step": 13792
    },
    {
      "epoch": 0.27588,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.25625712076822915,
      "learning_rate": 0.0001,
      "loss": 4.2548,
      "loss/crossentropy": 2.2632944583892822,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.210355743765831,
      "step": 13794
    },
    {
      "epoch": 0.27592,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.26166966756184895,
      "learning_rate": 0.0001,
      "loss": 4.2799,
      "loss/crossentropy": 2.5457879304885864,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22530706226825714,
      "step": 13796
    },
    {
      "epoch": 0.27596,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.008410390218098958,
      "learning_rate": 0.0001,
      "loss": 3.8969,
      "loss/crossentropy": 2.0878910422325134,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19901857525110245,
      "step": 13798
    },
    {
      "epoch": 0.276,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008348592122395833,
      "learning_rate": 0.0001,
      "loss": 4.3213,
      "loss/crossentropy": 1.803489863872528,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19378191232681274,
      "step": 13800
    },
    {
      "epoch": 0.27604,
      "grad_norm": 2.0,
      "grad_norm_var": 0.009596506754557291,
      "learning_rate": 0.0001,
      "loss": 4.1916,
      "loss/crossentropy": 2.156951904296875,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2091423124074936,
      "step": 13802
    },
    {
      "epoch": 0.27608,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.010493977864583334,
      "learning_rate": 0.0001,
      "loss": 3.9346,
      "loss/crossentropy": 1.7714723944664001,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20779500901699066,
      "step": 13804
    },
    {
      "epoch": 0.27612,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.009590403238932291,
      "learning_rate": 0.0001,
      "loss": 4.0499,
      "loss/crossentropy": 1.8808923363685608,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19429123401641846,
      "step": 13806
    },
    {
      "epoch": 0.27616,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.005516560872395834,
      "learning_rate": 0.0001,
      "loss": 3.706,
      "loss/crossentropy": 1.6658846735954285,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17177510261535645,
      "step": 13808
    },
    {
      "epoch": 0.2762,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.018697102864583332,
      "learning_rate": 0.0001,
      "loss": 4.0909,
      "loss/crossentropy": 2.052769422531128,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20889723300933838,
      "step": 13810
    },
    {
      "epoch": 0.27624,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.18883031209309895,
      "learning_rate": 0.0001,
      "loss": 4.0225,
      "loss/crossentropy": 1.5800148844718933,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24654322117567062,
      "step": 13812
    },
    {
      "epoch": 0.27628,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.5619504292805989,
      "learning_rate": 0.0001,
      "loss": 4.1059,
      "loss/crossentropy": 1.836085557937622,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19563084840774536,
      "step": 13814
    },
    {
      "epoch": 0.27632,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.5601722717285156,
      "learning_rate": 0.0001,
      "loss": 4.0077,
      "loss/crossentropy": 1.627321183681488,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17698387801647186,
      "step": 13816
    },
    {
      "epoch": 0.27636,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.55164794921875,
      "learning_rate": 0.0001,
      "loss": 4.2038,
      "loss/crossentropy": 2.0546024441719055,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19832595437765121,
      "step": 13818
    },
    {
      "epoch": 0.2764,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.5499501546223958,
      "learning_rate": 0.0001,
      "loss": 4.064,
      "loss/crossentropy": 2.112219452857971,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19605965912342072,
      "step": 13820
    },
    {
      "epoch": 0.27644,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.5405006408691406,
      "learning_rate": 0.0001,
      "loss": 4.3825,
      "loss/crossentropy": 1.9582284688949585,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2015855312347412,
      "step": 13822
    },
    {
      "epoch": 0.27648,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.5307573954264323,
      "learning_rate": 0.0001,
      "loss": 4.2699,
      "loss/crossentropy": 2.54653799533844,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23546704649925232,
      "step": 13824
    },
    {
      "epoch": 0.27652,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.5336090087890625,
      "learning_rate": 0.0001,
      "loss": 4.0729,
      "loss/crossentropy": 1.630593478679657,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16478916257619858,
      "step": 13826
    },
    {
      "epoch": 0.27656,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.40966796875,
      "learning_rate": 0.0001,
      "loss": 4.0123,
      "loss/crossentropy": 1.7995671033859253,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19886638224124908,
      "step": 13828
    },
    {
      "epoch": 0.2766,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.0177886962890625,
      "learning_rate": 0.0001,
      "loss": 4.4547,
      "loss/crossentropy": 2.5423338413238525,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.28057335317134857,
      "step": 13830
    },
    {
      "epoch": 0.27664,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.018631744384765624,
      "learning_rate": 0.0001,
      "loss": 4.0213,
      "loss/crossentropy": 1.8524783849716187,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20141787081956863,
      "step": 13832
    },
    {
      "epoch": 0.27668,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.019440714518229166,
      "learning_rate": 0.0001,
      "loss": 4.2155,
      "loss/crossentropy": 2.242451548576355,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21003998070955276,
      "step": 13834
    },
    {
      "epoch": 0.27672,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0192047119140625,
      "learning_rate": 0.0001,
      "loss": 4.2665,
      "loss/crossentropy": 2.2434345483779907,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21679577976465225,
      "step": 13836
    },
    {
      "epoch": 0.27676,
      "grad_norm": 2.0,
      "grad_norm_var": 0.019551595052083332,
      "learning_rate": 0.0001,
      "loss": 4.014,
      "loss/crossentropy": 1.6854392290115356,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19326400756835938,
      "step": 13838
    },
    {
      "epoch": 0.2768,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.026668294270833334,
      "learning_rate": 0.0001,
      "loss": 4.2837,
      "loss/crossentropy": 2.0702155232429504,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2043909877538681,
      "step": 13840
    },
    {
      "epoch": 0.27684,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.024738566080729166,
      "learning_rate": 0.0001,
      "loss": 4.0732,
      "loss/crossentropy": 1.964758813381195,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19540227204561234,
      "step": 13842
    },
    {
      "epoch": 0.27688,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0151611328125,
      "learning_rate": 0.0001,
      "loss": 4.1681,
      "loss/crossentropy": 2.0383411645889282,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21456278860569,
      "step": 13844
    },
    {
      "epoch": 0.27692,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.015461222330729166,
      "learning_rate": 0.0001,
      "loss": 4.1407,
      "loss/crossentropy": 2.020030975341797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20521440356969833,
      "step": 13846
    },
    {
      "epoch": 0.27696,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.016110992431640624,
      "learning_rate": 0.0001,
      "loss": 4.1142,
      "loss/crossentropy": 1.7514970302581787,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19116375595331192,
      "step": 13848
    },
    {
      "epoch": 0.277,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.016437784830729166,
      "learning_rate": 0.0001,
      "loss": 3.9209,
      "loss/crossentropy": 1.8194025754928589,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18198709934949875,
      "step": 13850
    },
    {
      "epoch": 0.27704,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.017223866780598958,
      "learning_rate": 0.0001,
      "loss": 3.7911,
      "loss/crossentropy": 1.878225862979889,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18159592151641846,
      "step": 13852
    },
    {
      "epoch": 0.27708,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0178619384765625,
      "learning_rate": 0.0001,
      "loss": 4.008,
      "loss/crossentropy": 1.8619664311408997,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18098927289247513,
      "step": 13854
    },
    {
      "epoch": 0.27712,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.003474934895833333,
      "learning_rate": 0.0001,
      "loss": 4.2336,
      "loss/crossentropy": 2.0475016832351685,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19999271631240845,
      "step": 13856
    },
    {
      "epoch": 0.27716,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.004443105061848958,
      "learning_rate": 0.0001,
      "loss": 4.2412,
      "loss/crossentropy": 2.2279679775238037,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21767108887434006,
      "step": 13858
    },
    {
      "epoch": 0.2772,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0039670308430989586,
      "learning_rate": 0.0001,
      "loss": 3.8087,
      "loss/crossentropy": 1.801176130771637,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18873406946659088,
      "step": 13860
    },
    {
      "epoch": 0.27724,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.005303700764973958,
      "learning_rate": 0.0001,
      "loss": 3.7958,
      "loss/crossentropy": 1.948195457458496,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19519966840744019,
      "step": 13862
    },
    {
      "epoch": 0.27728,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0050432840983072914,
      "learning_rate": 0.0001,
      "loss": 4.0754,
      "loss/crossentropy": 2.1044042110443115,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20454693585634232,
      "step": 13864
    },
    {
      "epoch": 0.27732,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0051910400390625,
      "learning_rate": 0.0001,
      "loss": 4.1338,
      "loss/crossentropy": 2.383382737636566,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2179596871137619,
      "step": 13866
    },
    {
      "epoch": 0.27736,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.004621378580729167,
      "learning_rate": 0.0001,
      "loss": 3.9475,
      "loss/crossentropy": 1.7798078656196594,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17871373891830444,
      "step": 13868
    },
    {
      "epoch": 0.2774,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.004756418863932291,
      "learning_rate": 0.0001,
      "loss": 3.9937,
      "loss/crossentropy": 2.1839531660079956,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20030800253152847,
      "step": 13870
    },
    {
      "epoch": 0.27744,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.005028279622395834,
      "learning_rate": 0.0001,
      "loss": 4.1637,
      "loss/crossentropy": 1.990351676940918,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1951240971684456,
      "step": 13872
    },
    {
      "epoch": 0.27748,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.006573232014973959,
      "learning_rate": 0.0001,
      "loss": 4.2919,
      "loss/crossentropy": 2.191170334815979,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.230004720389843,
      "step": 13874
    },
    {
      "epoch": 0.27752,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.015726470947265626,
      "learning_rate": 0.0001,
      "loss": 4.1432,
      "loss/crossentropy": 1.8893607258796692,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2138589546084404,
      "step": 13876
    },
    {
      "epoch": 0.27756,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.012483723958333333,
      "learning_rate": 0.0001,
      "loss": 4.2749,
      "loss/crossentropy": 2.059523820877075,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20147833973169327,
      "step": 13878
    },
    {
      "epoch": 0.2776,
      "grad_norm": 2.125,
      "grad_norm_var": 0.013206990559895833,
      "learning_rate": 0.0001,
      "loss": 4.4959,
      "loss/crossentropy": 2.0867209434509277,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23967817425727844,
      "step": 13880
    },
    {
      "epoch": 0.27764,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.028393300374348958,
      "learning_rate": 0.0001,
      "loss": 4.5704,
      "loss/crossentropy": 2.3017314672470093,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23674342036247253,
      "step": 13882
    },
    {
      "epoch": 0.27768,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0261138916015625,
      "learning_rate": 0.0001,
      "loss": 4.3316,
      "loss/crossentropy": 2.241922974586487,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21731412410736084,
      "step": 13884
    },
    {
      "epoch": 0.27772,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.025724283854166665,
      "learning_rate": 0.0001,
      "loss": 4.3317,
      "loss/crossentropy": 2.3800796270370483,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21916473656892776,
      "step": 13886
    },
    {
      "epoch": 0.27776,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.02467625935872396,
      "learning_rate": 0.0001,
      "loss": 4.2972,
      "loss/crossentropy": 1.966018259525299,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21862812340259552,
      "step": 13888
    },
    {
      "epoch": 0.2778,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.030104319254557293,
      "learning_rate": 0.0001,
      "loss": 4.1955,
      "loss/crossentropy": 1.920684039592743,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2026856392621994,
      "step": 13890
    },
    {
      "epoch": 0.27784,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.02601318359375,
      "learning_rate": 0.0001,
      "loss": 3.9662,
      "loss/crossentropy": 2.1939095854759216,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20429420471191406,
      "step": 13892
    },
    {
      "epoch": 0.27788,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.026292928059895835,
      "learning_rate": 0.0001,
      "loss": 4.2335,
      "loss/crossentropy": 2.1600992679595947,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20445890724658966,
      "step": 13894
    },
    {
      "epoch": 0.27792,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.026805623372395834,
      "learning_rate": 0.0001,
      "loss": 3.9973,
      "loss/crossentropy": 1.9898765683174133,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19056615233421326,
      "step": 13896
    },
    {
      "epoch": 0.27796,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.008861287434895834,
      "learning_rate": 0.0001,
      "loss": 4.3037,
      "loss/crossentropy": 2.0497539043426514,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1878463476896286,
      "step": 13898
    },
    {
      "epoch": 0.278,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.007373046875,
      "learning_rate": 0.0001,
      "loss": 4.2299,
      "loss/crossentropy": 1.894934356212616,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.183290496468544,
      "step": 13900
    },
    {
      "epoch": 0.27804,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.010094960530598959,
      "learning_rate": 0.0001,
      "loss": 3.858,
      "loss/crossentropy": 1.808231770992279,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19447006285190582,
      "step": 13902
    },
    {
      "epoch": 0.27808,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009688059488932291,
      "learning_rate": 0.0001,
      "loss": 4.0553,
      "loss/crossentropy": 1.9384279251098633,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20293861627578735,
      "step": 13904
    },
    {
      "epoch": 0.27812,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.012835439046223958,
      "learning_rate": 0.0001,
      "loss": 4.3092,
      "loss/crossentropy": 2.4291017055511475,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24332769960165024,
      "step": 13906
    },
    {
      "epoch": 0.27816,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.012669881184895834,
      "learning_rate": 0.0001,
      "loss": 4.331,
      "loss/crossentropy": 2.3018234968185425,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21657106280326843,
      "step": 13908
    },
    {
      "epoch": 0.2782,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.0204742431640625,
      "learning_rate": 0.0001,
      "loss": 4.1756,
      "loss/crossentropy": 2.211503267288208,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19193635880947113,
      "step": 13910
    },
    {
      "epoch": 0.27824,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.02191162109375,
      "learning_rate": 0.0001,
      "loss": 4.372,
      "loss/crossentropy": 2.17062246799469,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21989908069372177,
      "step": 13912
    },
    {
      "epoch": 0.27828,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.018839518229166668,
      "learning_rate": 0.0001,
      "loss": 4.0895,
      "loss/crossentropy": 2.1446024775505066,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2036902904510498,
      "step": 13914
    },
    {
      "epoch": 0.27832,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0177642822265625,
      "learning_rate": 0.0001,
      "loss": 4.0729,
      "loss/crossentropy": 1.8176022171974182,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19358078390359879,
      "step": 13916
    },
    {
      "epoch": 0.27836,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.01946996053059896,
      "learning_rate": 0.0001,
      "loss": 4.2376,
      "loss/crossentropy": 2.341936469078064,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22822833061218262,
      "step": 13918
    },
    {
      "epoch": 0.2784,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.01697565714518229,
      "learning_rate": 0.0001,
      "loss": 4.0943,
      "loss/crossentropy": 1.9315840601921082,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17218464612960815,
      "step": 13920
    },
    {
      "epoch": 0.27844,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.01615168253580729,
      "learning_rate": 0.0001,
      "loss": 4.1507,
      "loss/crossentropy": 1.9442673921585083,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19567463546991348,
      "step": 13922
    },
    {
      "epoch": 0.27848,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.016673787434895834,
      "learning_rate": 0.0001,
      "loss": 4.1611,
      "loss/crossentropy": 2.3023592233657837,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21824805438518524,
      "step": 13924
    },
    {
      "epoch": 0.27852,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.01170654296875,
      "learning_rate": 0.0001,
      "loss": 3.7557,
      "loss/crossentropy": 2.116863250732422,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2087356001138687,
      "step": 13926
    },
    {
      "epoch": 0.27856,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.010936482747395834,
      "learning_rate": 0.0001,
      "loss": 4.3746,
      "loss/crossentropy": 2.0485963821411133,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18910974264144897,
      "step": 13928
    },
    {
      "epoch": 0.2786,
      "grad_norm": 2.0,
      "grad_norm_var": 0.024857584635416666,
      "learning_rate": 0.0001,
      "loss": 4.1309,
      "loss/crossentropy": 2.031753659248352,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23851437866687775,
      "step": 13930
    },
    {
      "epoch": 0.27864,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.025585683186848958,
      "learning_rate": 0.0001,
      "loss": 4.0688,
      "loss/crossentropy": 1.8019705414772034,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18659329414367676,
      "step": 13932
    },
    {
      "epoch": 0.27868,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.025243123372395832,
      "learning_rate": 0.0001,
      "loss": 3.9609,
      "loss/crossentropy": 1.9191861152648926,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19393808394670486,
      "step": 13934
    },
    {
      "epoch": 0.27872,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.025763956705729167,
      "learning_rate": 0.0001,
      "loss": 4.1765,
      "loss/crossentropy": 2.242986798286438,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.213445246219635,
      "step": 13936
    },
    {
      "epoch": 0.27876,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0300689697265625,
      "learning_rate": 0.0001,
      "loss": 4.1767,
      "loss/crossentropy": 2.1652570962905884,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2070690169930458,
      "step": 13938
    },
    {
      "epoch": 0.2788,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.031583404541015624,
      "learning_rate": 0.0001,
      "loss": 3.9643,
      "loss/crossentropy": 2.0970187187194824,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20118873566389084,
      "step": 13940
    },
    {
      "epoch": 0.27884,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.02823053995768229,
      "learning_rate": 0.0001,
      "loss": 4.1116,
      "loss/crossentropy": 1.9957542419433594,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2377041131258011,
      "step": 13942
    },
    {
      "epoch": 0.27888,
      "grad_norm": 2.0,
      "grad_norm_var": 0.02522761027018229,
      "learning_rate": 0.0001,
      "loss": 4.0718,
      "loss/crossentropy": 1.8680259585380554,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1919926255941391,
      "step": 13944
    },
    {
      "epoch": 0.27892,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.015516916910807291,
      "learning_rate": 0.0001,
      "loss": 4.301,
      "loss/crossentropy": 2.369240164756775,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24805612862110138,
      "step": 13946
    },
    {
      "epoch": 0.27896,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.014530436197916666,
      "learning_rate": 0.0001,
      "loss": 4.3082,
      "loss/crossentropy": 2.0923795104026794,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20823675394058228,
      "step": 13948
    },
    {
      "epoch": 0.279,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.013197580973307291,
      "learning_rate": 0.0001,
      "loss": 4.3049,
      "loss/crossentropy": 2.1777398586273193,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21319355070590973,
      "step": 13950
    },
    {
      "epoch": 0.27904,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.014869944254557291,
      "learning_rate": 0.0001,
      "loss": 4.0918,
      "loss/crossentropy": 2.2155595421791077,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2029823139309883,
      "step": 13952
    },
    {
      "epoch": 0.27908,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.011435699462890626,
      "learning_rate": 0.0001,
      "loss": 3.7195,
      "loss/crossentropy": 1.5916873812675476,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18207873404026031,
      "step": 13954
    },
    {
      "epoch": 0.27912,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.009791819254557292,
      "learning_rate": 0.0001,
      "loss": 4.3336,
      "loss/crossentropy": 2.165639281272888,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22917281091213226,
      "step": 13956
    },
    {
      "epoch": 0.27916,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.012851715087890625,
      "learning_rate": 0.0001,
      "loss": 3.9586,
      "loss/crossentropy": 2.0245776772499084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19862860441207886,
      "step": 13958
    },
    {
      "epoch": 0.2792,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.013646443684895834,
      "learning_rate": 0.0001,
      "loss": 4.1593,
      "loss/crossentropy": 2.0179941654205322,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20749768614768982,
      "step": 13960
    },
    {
      "epoch": 0.27924,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.011966705322265625,
      "learning_rate": 0.0001,
      "loss": 4.3978,
      "loss/crossentropy": 1.998136818408966,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1992153376340866,
      "step": 13962
    },
    {
      "epoch": 0.27928,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.015346018473307292,
      "learning_rate": 0.0001,
      "loss": 4.1801,
      "loss/crossentropy": 2.041996657848358,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22224828600883484,
      "step": 13964
    },
    {
      "epoch": 0.27932,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.019406890869140624,
      "learning_rate": 0.0001,
      "loss": 4.4133,
      "loss/crossentropy": 2.0259117484092712,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2001984491944313,
      "step": 13966
    },
    {
      "epoch": 0.27936,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.018436686197916666,
      "learning_rate": 0.0001,
      "loss": 4.0439,
      "loss/crossentropy": 1.8219285607337952,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19637393951416016,
      "step": 13968
    },
    {
      "epoch": 0.2794,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.017073567708333334,
      "learning_rate": 0.0001,
      "loss": 4.0294,
      "loss/crossentropy": 2.1451289653778076,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21846695989370346,
      "step": 13970
    },
    {
      "epoch": 0.27944,
      "grad_norm": 2.125,
      "grad_norm_var": 0.016877237955729166,
      "learning_rate": 0.0001,
      "loss": 4.349,
      "loss/crossentropy": 2.0527132749557495,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2147705778479576,
      "step": 13972
    },
    {
      "epoch": 0.27948,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.015143839518229167,
      "learning_rate": 0.0001,
      "loss": 4.0188,
      "loss/crossentropy": 1.9238123893737793,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21083690226078033,
      "step": 13974
    },
    {
      "epoch": 0.27952,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.015057118733723958,
      "learning_rate": 0.0001,
      "loss": 4.0072,
      "loss/crossentropy": 1.894473671913147,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20077680051326752,
      "step": 13976
    },
    {
      "epoch": 0.27956,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.013871256510416667,
      "learning_rate": 0.0001,
      "loss": 4.2344,
      "loss/crossentropy": 2.139513611793518,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2138260081410408,
      "step": 13978
    },
    {
      "epoch": 0.2796,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.013468424479166666,
      "learning_rate": 0.0001,
      "loss": 4.4519,
      "loss/crossentropy": 2.4868407249450684,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23635651916265488,
      "step": 13980
    },
    {
      "epoch": 0.27964,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.007940419514973958,
      "learning_rate": 0.0001,
      "loss": 3.7428,
      "loss/crossentropy": 1.426945686340332,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16342981159687042,
      "step": 13982
    },
    {
      "epoch": 0.27968,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.007885487874348958,
      "learning_rate": 0.0001,
      "loss": 4.0457,
      "loss/crossentropy": 2.0577683448791504,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20394782721996307,
      "step": 13984
    },
    {
      "epoch": 0.27972,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.008063761393229167,
      "learning_rate": 0.0001,
      "loss": 4.0045,
      "loss/crossentropy": 1.884658396244049,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18946699798107147,
      "step": 13986
    },
    {
      "epoch": 0.27976,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0070383707682291664,
      "learning_rate": 0.0001,
      "loss": 3.9423,
      "loss/crossentropy": 1.7344964742660522,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1923774629831314,
      "step": 13988
    },
    {
      "epoch": 0.2798,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.00888671875,
      "learning_rate": 0.0001,
      "loss": 3.9541,
      "loss/crossentropy": 1.9660141468048096,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18560642004013062,
      "step": 13990
    },
    {
      "epoch": 0.27984,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.008998362223307292,
      "learning_rate": 0.0001,
      "loss": 3.9308,
      "loss/crossentropy": 1.7832527160644531,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19922740012407303,
      "step": 13992
    },
    {
      "epoch": 0.27988,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.009987131754557291,
      "learning_rate": 0.0001,
      "loss": 4.3919,
      "loss/crossentropy": 1.9763087630271912,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2111397087574005,
      "step": 13994
    },
    {
      "epoch": 0.27992,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.005252838134765625,
      "learning_rate": 0.0001,
      "loss": 4.1331,
      "loss/crossentropy": 2.262092709541321,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21212221682071686,
      "step": 13996
    },
    {
      "epoch": 0.27996,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0054595947265625,
      "learning_rate": 0.0001,
      "loss": 4.1946,
      "loss/crossentropy": 2.237086296081543,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20895393192768097,
      "step": 13998
    },
    {
      "epoch": 0.28,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0061337788899739586,
      "learning_rate": 0.0001,
      "loss": 4.063,
      "loss/crossentropy": 1.9579973816871643,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18024132400751114,
      "step": 14000
    },
    {
      "epoch": 0.28004,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0055844624837239586,
      "learning_rate": 0.0001,
      "loss": 4.1463,
      "loss/crossentropy": 2.2264128923416138,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2181634083390236,
      "step": 14002
    },
    {
      "epoch": 0.28008,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0058013916015625,
      "learning_rate": 0.0001,
      "loss": 4.0081,
      "loss/crossentropy": 1.836763322353363,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1983085498213768,
      "step": 14004
    },
    {
      "epoch": 0.28012,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.005448404947916667,
      "learning_rate": 0.0001,
      "loss": 4.1192,
      "loss/crossentropy": 2.0163257718086243,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.214015431702137,
      "step": 14006
    },
    {
      "epoch": 0.28016,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0070383707682291664,
      "learning_rate": 0.0001,
      "loss": 4.139,
      "loss/crossentropy": 2.31567645072937,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22832776606082916,
      "step": 14008
    },
    {
      "epoch": 0.2802,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.006685129801432292,
      "learning_rate": 0.0001,
      "loss": 4.0753,
      "loss/crossentropy": 2.3515210151672363,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21946918964385986,
      "step": 14010
    },
    {
      "epoch": 0.28024,
      "grad_norm": 2.0,
      "grad_norm_var": 0.006845855712890625,
      "learning_rate": 0.0001,
      "loss": 3.9761,
      "loss/crossentropy": 2.0987448692321777,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2180740088224411,
      "step": 14012
    },
    {
      "epoch": 0.28028,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.00662841796875,
      "learning_rate": 0.0001,
      "loss": 4.0909,
      "loss/crossentropy": 2.2203436493873596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2157250940799713,
      "step": 14014
    },
    {
      "epoch": 0.28032,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.006241861979166667,
      "learning_rate": 0.0001,
      "loss": 4.3152,
      "loss/crossentropy": 2.083233594894409,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2187238186597824,
      "step": 14016
    },
    {
      "epoch": 0.28036,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0061075846354166664,
      "learning_rate": 0.0001,
      "loss": 4.1311,
      "loss/crossentropy": 2.1261327266693115,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2063112109899521,
      "step": 14018
    },
    {
      "epoch": 0.2804,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.006648508707682291,
      "learning_rate": 0.0001,
      "loss": 4.0901,
      "loss/crossentropy": 2.012390434741974,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20225800573825836,
      "step": 14020
    },
    {
      "epoch": 0.28044,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0047271728515625,
      "learning_rate": 0.0001,
      "loss": 3.9985,
      "loss/crossentropy": 1.8605966567993164,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2003621682524681,
      "step": 14022
    },
    {
      "epoch": 0.28048,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0028157552083333333,
      "learning_rate": 0.0001,
      "loss": 3.9774,
      "loss/crossentropy": 1.9208670258522034,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20056191086769104,
      "step": 14024
    },
    {
      "epoch": 0.28052,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.003951009114583333,
      "learning_rate": 0.0001,
      "loss": 3.7462,
      "loss/crossentropy": 1.924518644809723,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19410791993141174,
      "step": 14026
    },
    {
      "epoch": 0.28056,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0035113016764322918,
      "learning_rate": 0.0001,
      "loss": 4.0319,
      "loss/crossentropy": 2.103038251399994,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2054024636745453,
      "step": 14028
    },
    {
      "epoch": 0.2806,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.004303995768229167,
      "learning_rate": 0.0001,
      "loss": 4.0526,
      "loss/crossentropy": 1.8974853157997131,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18592579662799835,
      "step": 14030
    },
    {
      "epoch": 0.28064,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.004447428385416666,
      "learning_rate": 0.0001,
      "loss": 4.1735,
      "loss/crossentropy": 2.0207908749580383,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.216909758746624,
      "step": 14032
    },
    {
      "epoch": 0.28068,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.006119537353515625,
      "learning_rate": 0.0001,
      "loss": 4.5759,
      "loss/crossentropy": 2.0010873079299927,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20815415680408478,
      "step": 14034
    },
    {
      "epoch": 0.28072,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.006961822509765625,
      "learning_rate": 0.0001,
      "loss": 3.9162,
      "loss/crossentropy": 2.0840484499931335,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21481210738420486,
      "step": 14036
    },
    {
      "epoch": 0.28076,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.006880442301432292,
      "learning_rate": 0.0001,
      "loss": 4.1298,
      "loss/crossentropy": 2.0026179552078247,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1904347836971283,
      "step": 14038
    },
    {
      "epoch": 0.2808,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.006880442301432292,
      "learning_rate": 0.0001,
      "loss": 4.0618,
      "loss/crossentropy": 2.1660486459732056,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21236423403024673,
      "step": 14040
    },
    {
      "epoch": 0.28084,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.007669830322265625,
      "learning_rate": 0.0001,
      "loss": 4.0474,
      "loss/crossentropy": 2.312765598297119,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20535308122634888,
      "step": 14042
    },
    {
      "epoch": 0.28088,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0077789306640625,
      "learning_rate": 0.0001,
      "loss": 4.1959,
      "loss/crossentropy": 1.8449203372001648,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1986582651734352,
      "step": 14044
    },
    {
      "epoch": 0.28092,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.008894602457682291,
      "learning_rate": 0.0001,
      "loss": 4.016,
      "loss/crossentropy": 2.0159433484077454,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19724585115909576,
      "step": 14046
    },
    {
      "epoch": 0.28096,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0086090087890625,
      "learning_rate": 0.0001,
      "loss": 4.0691,
      "loss/crossentropy": 2.138873815536499,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21438196301460266,
      "step": 14048
    },
    {
      "epoch": 0.281,
      "grad_norm": 2.25,
      "grad_norm_var": 0.011372629801432292,
      "learning_rate": 0.0001,
      "loss": 4.0985,
      "loss/crossentropy": 1.6896708607673645,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19260118901729584,
      "step": 14050
    },
    {
      "epoch": 0.28104,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.014656321207682291,
      "learning_rate": 0.0001,
      "loss": 3.8383,
      "loss/crossentropy": 1.8495243191719055,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18072611093521118,
      "step": 14052
    },
    {
      "epoch": 0.28108,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.017354329427083332,
      "learning_rate": 0.0001,
      "loss": 4.3581,
      "loss/crossentropy": 2.193492293357849,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2247578352689743,
      "step": 14054
    },
    {
      "epoch": 0.28112,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.01761652628580729,
      "learning_rate": 0.0001,
      "loss": 4.1252,
      "loss/crossentropy": 2.154744803905487,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20149191468954086,
      "step": 14056
    },
    {
      "epoch": 0.28116,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.016022745768229166,
      "learning_rate": 0.0001,
      "loss": 4.0832,
      "loss/crossentropy": 2.137023687362671,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20409516990184784,
      "step": 14058
    },
    {
      "epoch": 0.2812,
      "grad_norm": 2.96875,
      "grad_norm_var": 0.0767242431640625,
      "learning_rate": 0.0001,
      "loss": 4.6582,
      "loss/crossentropy": 2.4530670642852783,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24031317979097366,
      "step": 14060
    },
    {
      "epoch": 0.28124,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.07713394165039063,
      "learning_rate": 0.0001,
      "loss": 4.2973,
      "loss/crossentropy": 2.3285664319992065,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.223993182182312,
      "step": 14062
    },
    {
      "epoch": 0.28128,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0784088134765625,
      "learning_rate": 0.0001,
      "loss": 4.0899,
      "loss/crossentropy": 2.126828193664551,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21428050100803375,
      "step": 14064
    },
    {
      "epoch": 0.28132,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.08012288411458333,
      "learning_rate": 0.0001,
      "loss": 3.8083,
      "loss/crossentropy": 2.0194268226623535,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20124298334121704,
      "step": 14066
    },
    {
      "epoch": 0.28136,
      "grad_norm": 2.125,
      "grad_norm_var": 0.07445882161458334,
      "learning_rate": 0.0001,
      "loss": 4.0002,
      "loss/crossentropy": 1.796358585357666,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1902359575033188,
      "step": 14068
    },
    {
      "epoch": 0.2814,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.07318700154622396,
      "learning_rate": 0.0001,
      "loss": 4.2,
      "loss/crossentropy": 1.8975006341934204,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19528520107269287,
      "step": 14070
    },
    {
      "epoch": 0.28144,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.07691141764322916,
      "learning_rate": 0.0001,
      "loss": 3.9885,
      "loss/crossentropy": 1.628948986530304,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1823427528142929,
      "step": 14072
    },
    {
      "epoch": 0.28148,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.07718480428059896,
      "learning_rate": 0.0001,
      "loss": 4.0404,
      "loss/crossentropy": 1.9894697070121765,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1945328414440155,
      "step": 14074
    },
    {
      "epoch": 0.28152,
      "grad_norm": 2.125,
      "grad_norm_var": 0.022489166259765624,
      "learning_rate": 0.0001,
      "loss": 4.0927,
      "loss/crossentropy": 1.9978017210960388,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21055076271295547,
      "step": 14076
    },
    {
      "epoch": 0.28156,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.01697565714518229,
      "learning_rate": 0.0001,
      "loss": 3.9224,
      "loss/crossentropy": 2.087389588356018,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19179877638816833,
      "step": 14078
    },
    {
      "epoch": 0.2816,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.018308258056640624,
      "learning_rate": 0.0001,
      "loss": 4.3356,
      "loss/crossentropy": 2.3281190395355225,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2292996421456337,
      "step": 14080
    },
    {
      "epoch": 0.28164,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.015215810139973958,
      "learning_rate": 0.0001,
      "loss": 3.8584,
      "loss/crossentropy": 1.8850311040878296,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19652675837278366,
      "step": 14082
    },
    {
      "epoch": 0.28168,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.015604654947916666,
      "learning_rate": 0.0001,
      "loss": 3.9013,
      "loss/crossentropy": 1.757595181465149,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17358998954296112,
      "step": 14084
    },
    {
      "epoch": 0.28172,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.01387939453125,
      "learning_rate": 0.0001,
      "loss": 4.1367,
      "loss/crossentropy": 2.173685908317566,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22043757140636444,
      "step": 14086
    },
    {
      "epoch": 0.28176,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.010758463541666667,
      "learning_rate": 0.0001,
      "loss": 4.0381,
      "loss/crossentropy": 2.0290868282318115,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19695669412612915,
      "step": 14088
    },
    {
      "epoch": 0.2818,
      "grad_norm": 2.125,
      "grad_norm_var": 0.010609690348307292,
      "learning_rate": 0.0001,
      "loss": 4.3497,
      "loss/crossentropy": 2.320943236351013,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21747629344463348,
      "step": 14090
    },
    {
      "epoch": 0.28184,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.011205037434895834,
      "learning_rate": 0.0001,
      "loss": 4.0918,
      "loss/crossentropy": 2.0311750173568726,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19410807639360428,
      "step": 14092
    },
    {
      "epoch": 0.28188,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.011207834879557291,
      "learning_rate": 0.0001,
      "loss": 4.1303,
      "loss/crossentropy": 2.164630174636841,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1962611824274063,
      "step": 14094
    },
    {
      "epoch": 0.28192,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0093658447265625,
      "learning_rate": 0.0001,
      "loss": 3.9681,
      "loss/crossentropy": 2.170566439628601,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24082274734973907,
      "step": 14096
    },
    {
      "epoch": 0.28196,
      "grad_norm": 2.0,
      "grad_norm_var": 0.009285227457682291,
      "learning_rate": 0.0001,
      "loss": 4.209,
      "loss/crossentropy": 2.3137707710266113,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22475039958953857,
      "step": 14098
    },
    {
      "epoch": 0.282,
      "grad_norm": 1.7421875,
      "grad_norm_var": 0.010794830322265626,
      "learning_rate": 0.0001,
      "loss": 3.7432,
      "loss/crossentropy": 2.0448675751686096,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19814980030059814,
      "step": 14100
    },
    {
      "epoch": 0.28204,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.011423492431640625,
      "learning_rate": 0.0001,
      "loss": 4.1729,
      "loss/crossentropy": 2.095334231853485,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19743670523166656,
      "step": 14102
    },
    {
      "epoch": 0.28208,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.008737945556640625,
      "learning_rate": 0.0001,
      "loss": 4.0362,
      "loss/crossentropy": 2.280429720878601,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21895557641983032,
      "step": 14104
    },
    {
      "epoch": 0.28212,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.008292388916015626,
      "learning_rate": 0.0001,
      "loss": 4.1166,
      "loss/crossentropy": 1.8548901677131653,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19965855032205582,
      "step": 14106
    },
    {
      "epoch": 0.28216,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.01163330078125,
      "learning_rate": 0.0001,
      "loss": 4.2589,
      "loss/crossentropy": 2.3574371337890625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23160286247730255,
      "step": 14108
    },
    {
      "epoch": 0.2822,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.012800089518229167,
      "learning_rate": 0.0001,
      "loss": 3.932,
      "loss/crossentropy": 2.112724542617798,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19597014784812927,
      "step": 14110
    },
    {
      "epoch": 0.28224,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.013068644205729167,
      "learning_rate": 0.0001,
      "loss": 3.8792,
      "loss/crossentropy": 1.9877798557281494,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21120092272758484,
      "step": 14112
    },
    {
      "epoch": 0.28228,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.013199869791666667,
      "learning_rate": 0.0001,
      "loss": 4.1761,
      "loss/crossentropy": 2.253028154373169,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2046835869550705,
      "step": 14114
    },
    {
      "epoch": 0.28232,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.008561197916666667,
      "learning_rate": 0.0001,
      "loss": 4.1711,
      "loss/crossentropy": 2.322582960128784,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22178008407354355,
      "step": 14116
    },
    {
      "epoch": 0.28236,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008204905192057292,
      "learning_rate": 0.0001,
      "loss": 4.1584,
      "loss/crossentropy": 2.470620036125183,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23360955715179443,
      "step": 14118
    },
    {
      "epoch": 0.2824,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0086334228515625,
      "learning_rate": 0.0001,
      "loss": 4.0206,
      "loss/crossentropy": 1.9218884110450745,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1978505551815033,
      "step": 14120
    },
    {
      "epoch": 0.28244,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0092437744140625,
      "learning_rate": 0.0001,
      "loss": 4.2045,
      "loss/crossentropy": 1.9114753007888794,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19295217841863632,
      "step": 14122
    },
    {
      "epoch": 0.28248,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.498583984375,
      "learning_rate": 0.0001,
      "loss": 4.2259,
      "loss/crossentropy": 2.1196082830429077,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2068253606557846,
      "step": 14124
    },
    {
      "epoch": 0.28252,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.492419179280599,
      "learning_rate": 0.0001,
      "loss": 4.2616,
      "loss/crossentropy": 1.9869969487190247,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1950489580631256,
      "step": 14126
    },
    {
      "epoch": 0.28256,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.488177235921224,
      "learning_rate": 0.0001,
      "loss": 4.1635,
      "loss/crossentropy": 2.09599232673645,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22184181958436966,
      "step": 14128
    },
    {
      "epoch": 0.2826,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.489172108968099,
      "learning_rate": 0.0001,
      "loss": 4.2238,
      "loss/crossentropy": 2.1891895532608032,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2229236587882042,
      "step": 14130
    },
    {
      "epoch": 0.28264,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.48787816365559894,
      "learning_rate": 0.0001,
      "loss": 4.12,
      "loss/crossentropy": 2.5070972442626953,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21862925589084625,
      "step": 14132
    },
    {
      "epoch": 0.28268,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.48812662760416664,
      "learning_rate": 0.0001,
      "loss": 4.0993,
      "loss/crossentropy": 2.100346863269806,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19969436526298523,
      "step": 14134
    },
    {
      "epoch": 0.28272,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.48713150024414065,
      "learning_rate": 0.0001,
      "loss": 4.2607,
      "loss/crossentropy": 2.0019801259040833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21319210529327393,
      "step": 14136
    },
    {
      "epoch": 0.28276,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.4855323791503906,
      "learning_rate": 0.0001,
      "loss": 4.7069,
      "loss/crossentropy": 2.2734099626541138,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24008513987064362,
      "step": 14138
    },
    {
      "epoch": 0.2828,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.02581965128580729,
      "learning_rate": 0.0001,
      "loss": 4.1085,
      "loss/crossentropy": 2.2589277029037476,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2224387675523758,
      "step": 14140
    },
    {
      "epoch": 0.28284,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.02667236328125,
      "learning_rate": 0.0001,
      "loss": 4.2749,
      "loss/crossentropy": 1.713346004486084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19133105129003525,
      "step": 14142
    },
    {
      "epoch": 0.28288,
      "grad_norm": 2.0,
      "grad_norm_var": 0.02616144816080729,
      "learning_rate": 0.0001,
      "loss": 4.1792,
      "loss/crossentropy": 2.212652564048767,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21326624602079391,
      "step": 14144
    },
    {
      "epoch": 0.28292,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.02902399698893229,
      "learning_rate": 0.0001,
      "loss": 3.901,
      "loss/crossentropy": 1.8900847434997559,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1882227212190628,
      "step": 14146
    },
    {
      "epoch": 0.28296,
      "grad_norm": 2.25,
      "grad_norm_var": 0.028913370768229165,
      "learning_rate": 0.0001,
      "loss": 4.3775,
      "loss/crossentropy": 2.171455979347229,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20516864210367203,
      "step": 14148
    },
    {
      "epoch": 0.283,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.027608235677083332,
      "learning_rate": 0.0001,
      "loss": 4.3466,
      "loss/crossentropy": 2.0112149715423584,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20043446868658066,
      "step": 14150
    },
    {
      "epoch": 0.28304,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.026596832275390624,
      "learning_rate": 0.0001,
      "loss": 4.2879,
      "loss/crossentropy": 2.0091559886932373,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21023409068584442,
      "step": 14152
    },
    {
      "epoch": 0.28308,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.023045857747395832,
      "learning_rate": 0.0001,
      "loss": 3.9234,
      "loss/crossentropy": 2.122502863407135,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2023375853896141,
      "step": 14154
    },
    {
      "epoch": 0.28312,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.011462148030598958,
      "learning_rate": 0.0001,
      "loss": 4.3187,
      "loss/crossentropy": 2.2842462062835693,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24532422423362732,
      "step": 14156
    },
    {
      "epoch": 0.28316,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.010990397135416666,
      "learning_rate": 0.0001,
      "loss": 4.5248,
      "loss/crossentropy": 2.247039318084717,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24580712616443634,
      "step": 14158
    },
    {
      "epoch": 0.2832,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.011432902018229166,
      "learning_rate": 0.0001,
      "loss": 3.9486,
      "loss/crossentropy": 2.182216167449951,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21095345169305801,
      "step": 14160
    },
    {
      "epoch": 0.28324,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.008648427327473958,
      "learning_rate": 0.0001,
      "loss": 4.4582,
      "loss/crossentropy": 2.219786763191223,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21747471392154694,
      "step": 14162
    },
    {
      "epoch": 0.28328,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.005830637613932292,
      "learning_rate": 0.0001,
      "loss": 3.9483,
      "loss/crossentropy": 1.8589028716087341,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19629907608032227,
      "step": 14164
    },
    {
      "epoch": 0.28332,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0138092041015625,
      "learning_rate": 0.0001,
      "loss": 4.0217,
      "loss/crossentropy": 1.9820821285247803,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20433218777179718,
      "step": 14166
    },
    {
      "epoch": 0.28336,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.014216868082682292,
      "learning_rate": 0.0001,
      "loss": 4.0081,
      "loss/crossentropy": 1.8423896431922913,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18210452795028687,
      "step": 14168
    },
    {
      "epoch": 0.2834,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.0158203125,
      "learning_rate": 0.0001,
      "loss": 4.1244,
      "loss/crossentropy": 2.0365665555000305,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21229179203510284,
      "step": 14170
    },
    {
      "epoch": 0.28344,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.019212849934895835,
      "learning_rate": 0.0001,
      "loss": 3.7894,
      "loss/crossentropy": 1.6819360256195068,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20171043276786804,
      "step": 14172
    },
    {
      "epoch": 0.28348,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.019846343994140626,
      "learning_rate": 0.0001,
      "loss": 4.1247,
      "loss/crossentropy": 1.9932513236999512,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21374855190515518,
      "step": 14174
    },
    {
      "epoch": 0.28352,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.01975072224934896,
      "learning_rate": 0.0001,
      "loss": 3.8874,
      "loss/crossentropy": 1.7137236595153809,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17538649588823318,
      "step": 14176
    },
    {
      "epoch": 0.28356,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.01977717081705729,
      "learning_rate": 0.0001,
      "loss": 3.8762,
      "loss/crossentropy": 1.7120450735092163,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19030355662107468,
      "step": 14178
    },
    {
      "epoch": 0.2836,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.019962565104166666,
      "learning_rate": 0.0001,
      "loss": 4.4603,
      "loss/crossentropy": 2.346954107284546,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24070723354816437,
      "step": 14180
    },
    {
      "epoch": 0.28364,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.013755035400390626,
      "learning_rate": 0.0001,
      "loss": 4.2238,
      "loss/crossentropy": 2.241390824317932,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2247970700263977,
      "step": 14182
    },
    {
      "epoch": 0.28368,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.02080078125,
      "learning_rate": 0.0001,
      "loss": 4.206,
      "loss/crossentropy": 2.234646439552307,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21947349607944489,
      "step": 14184
    },
    {
      "epoch": 0.28372,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.019010416666666665,
      "learning_rate": 0.0001,
      "loss": 4.0446,
      "loss/crossentropy": 1.9331231117248535,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20428168773651123,
      "step": 14186
    },
    {
      "epoch": 0.28376,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.015315500895182292,
      "learning_rate": 0.0001,
      "loss": 4.0184,
      "loss/crossentropy": 1.8166351318359375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21285146474838257,
      "step": 14188
    },
    {
      "epoch": 0.2838,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.015077463785807292,
      "learning_rate": 0.0001,
      "loss": 4.313,
      "loss/crossentropy": 2.181701421737671,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20924244076013565,
      "step": 14190
    },
    {
      "epoch": 0.28384,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.015558878580729166,
      "learning_rate": 0.0001,
      "loss": 4.0866,
      "loss/crossentropy": 2.1605955958366394,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24021095037460327,
      "step": 14192
    },
    {
      "epoch": 0.28388,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.015455881754557291,
      "learning_rate": 0.0001,
      "loss": 4.1617,
      "loss/crossentropy": 2.3546024560928345,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24228987097740173,
      "step": 14194
    },
    {
      "epoch": 0.28392,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.015314737955729166,
      "learning_rate": 0.0001,
      "loss": 4.0791,
      "loss/crossentropy": 2.2335511445999146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21917300671339035,
      "step": 14196
    },
    {
      "epoch": 0.28396,
      "grad_norm": 2.0,
      "grad_norm_var": 0.012760416666666666,
      "learning_rate": 0.0001,
      "loss": 3.9915,
      "loss/crossentropy": 1.8429189324378967,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18830660730600357,
      "step": 14198
    },
    {
      "epoch": 0.284,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0058746337890625,
      "learning_rate": 0.0001,
      "loss": 4.2671,
      "loss/crossentropy": 2.4578241109848022,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23527230322360992,
      "step": 14200
    },
    {
      "epoch": 0.28404,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.00592041015625,
      "learning_rate": 0.0001,
      "loss": 4.3027,
      "loss/crossentropy": 2.0944234132766724,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2211083546280861,
      "step": 14202
    },
    {
      "epoch": 0.28408,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0066070556640625,
      "learning_rate": 0.0001,
      "loss": 3.9341,
      "loss/crossentropy": 1.8876588344573975,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1908133327960968,
      "step": 14204
    },
    {
      "epoch": 0.28412,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.006190745035807291,
      "learning_rate": 0.0001,
      "loss": 4.368,
      "loss/crossentropy": 2.0882590413093567,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21159439533948898,
      "step": 14206
    },
    {
      "epoch": 0.28416,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.005296834309895833,
      "learning_rate": 0.0001,
      "loss": 4.3005,
      "loss/crossentropy": 2.219560742378235,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2282530516386032,
      "step": 14208
    },
    {
      "epoch": 0.2842,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.004759724934895833,
      "learning_rate": 0.0001,
      "loss": 3.9308,
      "loss/crossentropy": 1.8383984565734863,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1868659406900406,
      "step": 14210
    },
    {
      "epoch": 0.28424,
      "grad_norm": 2.125,
      "grad_norm_var": 0.006892649332682291,
      "learning_rate": 0.0001,
      "loss": 3.9613,
      "loss/crossentropy": 1.7509828209877014,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1862550526857376,
      "step": 14212
    },
    {
      "epoch": 0.28428,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.007513173421223958,
      "learning_rate": 0.0001,
      "loss": 4.1301,
      "loss/crossentropy": 2.1544927954673767,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19378253817558289,
      "step": 14214
    },
    {
      "epoch": 0.28432,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.007533518473307291,
      "learning_rate": 0.0001,
      "loss": 4.1261,
      "loss/crossentropy": 2.498751997947693,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22881630063056946,
      "step": 14216
    },
    {
      "epoch": 0.28436,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.006780751546223958,
      "learning_rate": 0.0001,
      "loss": 4.3414,
      "loss/crossentropy": 2.2638392448425293,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21407127380371094,
      "step": 14218
    },
    {
      "epoch": 0.2844,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.006380208333333333,
      "learning_rate": 0.0001,
      "loss": 4.0469,
      "loss/crossentropy": 2.097207546234131,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21202068030834198,
      "step": 14220
    },
    {
      "epoch": 0.28444,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.007661946614583333,
      "learning_rate": 0.0001,
      "loss": 4.211,
      "loss/crossentropy": 2.0577695965766907,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21462847292423248,
      "step": 14222
    },
    {
      "epoch": 0.28448,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.008373769124348958,
      "learning_rate": 0.0001,
      "loss": 4.2646,
      "loss/crossentropy": 2.3061472177505493,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23122046887874603,
      "step": 14224
    },
    {
      "epoch": 0.28452,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0068511962890625,
      "learning_rate": 0.0001,
      "loss": 3.947,
      "loss/crossentropy": 1.850829005241394,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19229399412870407,
      "step": 14226
    },
    {
      "epoch": 0.28456,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0051348368326822914,
      "learning_rate": 0.0001,
      "loss": 4.0891,
      "loss/crossentropy": 1.9520751237869263,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19159140437841415,
      "step": 14228
    },
    {
      "epoch": 0.2846,
      "grad_norm": 1.875,
      "grad_norm_var": 0.005145009358723958,
      "learning_rate": 0.0001,
      "loss": 4.2161,
      "loss/crossentropy": 2.507380247116089,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23365024477243423,
      "step": 14230
    },
    {
      "epoch": 0.28464,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006261952718098958,
      "learning_rate": 0.0001,
      "loss": 4.0787,
      "loss/crossentropy": 1.9148982763290405,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19785508513450623,
      "step": 14232
    },
    {
      "epoch": 0.28468,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.007769521077473958,
      "learning_rate": 0.0001,
      "loss": 4.1882,
      "loss/crossentropy": 2.004162549972534,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.192252054810524,
      "step": 14234
    },
    {
      "epoch": 0.28472,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0078277587890625,
      "learning_rate": 0.0001,
      "loss": 4.1318,
      "loss/crossentropy": 2.1455901861190796,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19151784479618073,
      "step": 14236
    },
    {
      "epoch": 0.28476,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.005580393473307291,
      "learning_rate": 0.0001,
      "loss": 3.9548,
      "loss/crossentropy": 2.2556833028793335,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20640526711940765,
      "step": 14238
    },
    {
      "epoch": 0.2848,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.013698069254557292,
      "learning_rate": 0.0001,
      "loss": 4.3805,
      "loss/crossentropy": 2.1121991872787476,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21154547482728958,
      "step": 14240
    },
    {
      "epoch": 0.28484,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.01586888631184896,
      "learning_rate": 0.0001,
      "loss": 3.7502,
      "loss/crossentropy": 1.9072884321212769,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19220874458551407,
      "step": 14242
    },
    {
      "epoch": 0.28488,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0154052734375,
      "learning_rate": 0.0001,
      "loss": 4.1925,
      "loss/crossentropy": 1.907566249370575,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20355048030614853,
      "step": 14244
    },
    {
      "epoch": 0.28492,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.0149169921875,
      "learning_rate": 0.0001,
      "loss": 3.8873,
      "loss/crossentropy": 2.1140421628952026,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20535241812467575,
      "step": 14246
    },
    {
      "epoch": 0.28496,
      "grad_norm": 2.125,
      "grad_norm_var": 0.016022745768229166,
      "learning_rate": 0.0001,
      "loss": 4.1704,
      "loss/crossentropy": 2.192967176437378,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20353975892066956,
      "step": 14248
    },
    {
      "epoch": 0.285,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.015213775634765624,
      "learning_rate": 0.0001,
      "loss": 3.9064,
      "loss/crossentropy": 1.6658200025558472,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17274750024080276,
      "step": 14250
    },
    {
      "epoch": 0.28504,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.017411041259765624,
      "learning_rate": 0.0001,
      "loss": 4.2015,
      "loss/crossentropy": 2.014496326446533,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.203952856361866,
      "step": 14252
    },
    {
      "epoch": 0.28508,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.0249908447265625,
      "learning_rate": 0.0001,
      "loss": 4.4607,
      "loss/crossentropy": 2.1534151434898376,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23526588827371597,
      "step": 14254
    },
    {
      "epoch": 0.28512,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.01701838175455729,
      "learning_rate": 0.0001,
      "loss": 4.1263,
      "loss/crossentropy": 2.08686763048172,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20443058758974075,
      "step": 14256
    },
    {
      "epoch": 0.28516,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0135406494140625,
      "learning_rate": 0.0001,
      "loss": 4.314,
      "loss/crossentropy": 1.9428821802139282,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20955658704042435,
      "step": 14258
    },
    {
      "epoch": 0.2852,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.015710194905598957,
      "learning_rate": 0.0001,
      "loss": 3.9157,
      "loss/crossentropy": 1.8959371447563171,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20648592710494995,
      "step": 14260
    },
    {
      "epoch": 0.28524,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.017081705729166667,
      "learning_rate": 0.0001,
      "loss": 3.8246,
      "loss/crossentropy": 1.7842467427253723,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17766441404819489,
      "step": 14262
    },
    {
      "epoch": 0.28528,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.018805948893229167,
      "learning_rate": 0.0001,
      "loss": 4.0581,
      "loss/crossentropy": 1.989040195941925,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20939016342163086,
      "step": 14264
    },
    {
      "epoch": 0.28532,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.018656158447265626,
      "learning_rate": 0.0001,
      "loss": 3.9216,
      "loss/crossentropy": 2.0441301465034485,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20979472994804382,
      "step": 14266
    },
    {
      "epoch": 0.28536,
      "grad_norm": 1.875,
      "grad_norm_var": 0.017970530192057292,
      "learning_rate": 0.0001,
      "loss": 4.2181,
      "loss/crossentropy": 1.9348166584968567,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1930733323097229,
      "step": 14268
    },
    {
      "epoch": 0.2854,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.009650675455729167,
      "learning_rate": 0.0001,
      "loss": 4.2039,
      "loss/crossentropy": 2.056081175804138,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20383870601654053,
      "step": 14270
    },
    {
      "epoch": 0.28544,
      "grad_norm": 2.125,
      "grad_norm_var": 0.011271158854166666,
      "learning_rate": 0.0001,
      "loss": 4.4046,
      "loss/crossentropy": 2.2752946615219116,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2458929866552353,
      "step": 14272
    },
    {
      "epoch": 0.28548,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.010456339518229166,
      "learning_rate": 0.0001,
      "loss": 4.1693,
      "loss/crossentropy": 2.093926787376404,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19211938977241516,
      "step": 14274
    },
    {
      "epoch": 0.28552,
      "grad_norm": 2.0,
      "grad_norm_var": 0.009913889567057292,
      "learning_rate": 0.0001,
      "loss": 4.1841,
      "loss/crossentropy": 2.448614239692688,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23177827894687653,
      "step": 14276
    },
    {
      "epoch": 0.28556,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.012401326497395834,
      "learning_rate": 0.0001,
      "loss": 4.0002,
      "loss/crossentropy": 2.0047106742858887,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2162417769432068,
      "step": 14278
    },
    {
      "epoch": 0.2856,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.017986806233723958,
      "learning_rate": 0.0001,
      "loss": 4.1669,
      "loss/crossentropy": 1.982455313205719,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22751964628696442,
      "step": 14280
    },
    {
      "epoch": 0.28564,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.017411295572916666,
      "learning_rate": 0.0001,
      "loss": 4.0355,
      "loss/crossentropy": 1.872545838356018,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19046122580766678,
      "step": 14282
    },
    {
      "epoch": 0.28568,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.016388956705729166,
      "learning_rate": 0.0001,
      "loss": 4.1266,
      "loss/crossentropy": 1.8999648690223694,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2155570462346077,
      "step": 14284
    },
    {
      "epoch": 0.28572,
      "grad_norm": 2.0,
      "grad_norm_var": 0.014129384358723959,
      "learning_rate": 0.0001,
      "loss": 3.9649,
      "loss/crossentropy": 1.8433392643928528,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18906012177467346,
      "step": 14286
    },
    {
      "epoch": 0.28576,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.014697011311848958,
      "learning_rate": 0.0001,
      "loss": 4.085,
      "loss/crossentropy": 1.817060947418213,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20473239570856094,
      "step": 14288
    },
    {
      "epoch": 0.2858,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.01796239217122396,
      "learning_rate": 0.0001,
      "loss": 4.1856,
      "loss/crossentropy": 2.0344348549842834,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19991667568683624,
      "step": 14290
    },
    {
      "epoch": 0.28584,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.01784032185872396,
      "learning_rate": 0.0001,
      "loss": 4.2334,
      "loss/crossentropy": 2.3413926362991333,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2365436628460884,
      "step": 14292
    },
    {
      "epoch": 0.28588,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.015464019775390626,
      "learning_rate": 0.0001,
      "loss": 3.9437,
      "loss/crossentropy": 2.0959852933883667,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19703902304172516,
      "step": 14294
    },
    {
      "epoch": 0.28592,
      "grad_norm": 7.625,
      "grad_norm_var": 1.9304239908854166,
      "learning_rate": 0.0001,
      "loss": 4.2144,
      "loss/crossentropy": 2.220101237297058,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21996742486953735,
      "step": 14296
    },
    {
      "epoch": 0.28596,
      "grad_norm": 2.109375,
      "grad_norm_var": 1.915691884358724,
      "learning_rate": 0.0001,
      "loss": 4.1806,
      "loss/crossentropy": 2.073485493659973,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2020936980843544,
      "step": 14298
    },
    {
      "epoch": 0.286,
      "grad_norm": 1.96875,
      "grad_norm_var": 1.9216265360514322,
      "learning_rate": 0.0001,
      "loss": 3.9176,
      "loss/crossentropy": 2.228920817375183,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22304438799619675,
      "step": 14300
    },
    {
      "epoch": 0.28604,
      "grad_norm": 1.890625,
      "grad_norm_var": 1.933642323811849,
      "learning_rate": 0.0001,
      "loss": 4.0831,
      "loss/crossentropy": 1.724998950958252,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1592913344502449,
      "step": 14302
    },
    {
      "epoch": 0.28608,
      "grad_norm": 2.0,
      "grad_norm_var": 1.9375221252441406,
      "learning_rate": 0.0001,
      "loss": 4.1738,
      "loss/crossentropy": 1.8970724940299988,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18843455612659454,
      "step": 14304
    },
    {
      "epoch": 0.28612,
      "grad_norm": 2.09375,
      "grad_norm_var": 1.9384295145670574,
      "learning_rate": 0.0001,
      "loss": 4.0808,
      "loss/crossentropy": 1.9957273602485657,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18272704631090164,
      "step": 14306
    },
    {
      "epoch": 0.28616,
      "grad_norm": 2.015625,
      "grad_norm_var": 1.9298177083333334,
      "learning_rate": 0.0001,
      "loss": 4.3861,
      "loss/crossentropy": 2.1424412727355957,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21485400944948196,
      "step": 14308
    },
    {
      "epoch": 0.2862,
      "grad_norm": 1.9765625,
      "grad_norm_var": 1.9411211649576823,
      "learning_rate": 0.0001,
      "loss": 3.9622,
      "loss/crossentropy": 1.7729334235191345,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20436155050992966,
      "step": 14310
    },
    {
      "epoch": 0.28624,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.013952382405598958,
      "learning_rate": 0.0001,
      "loss": 4.2495,
      "loss/crossentropy": 2.5863327980041504,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24598564952611923,
      "step": 14312
    },
    {
      "epoch": 0.28628,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.01014404296875,
      "learning_rate": 0.0001,
      "loss": 3.962,
      "loss/crossentropy": 1.9089832305908203,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1905580461025238,
      "step": 14314
    },
    {
      "epoch": 0.28632,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.010228474934895834,
      "learning_rate": 0.0001,
      "loss": 4.2167,
      "loss/crossentropy": 2.0153337121009827,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21556401252746582,
      "step": 14316
    },
    {
      "epoch": 0.28636,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008776601155598958,
      "learning_rate": 0.0001,
      "loss": 4.4037,
      "loss/crossentropy": 1.9692147970199585,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17990782111883163,
      "step": 14318
    },
    {
      "epoch": 0.2864,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.010969034830729167,
      "learning_rate": 0.0001,
      "loss": 4.1049,
      "loss/crossentropy": 2.0404393076896667,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2108697146177292,
      "step": 14320
    },
    {
      "epoch": 0.28644,
      "grad_norm": 2.0,
      "grad_norm_var": 0.009488932291666667,
      "learning_rate": 0.0001,
      "loss": 4.2809,
      "loss/crossentropy": 2.133711099624634,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1942032128572464,
      "step": 14322
    },
    {
      "epoch": 0.28648,
      "grad_norm": 2.0,
      "grad_norm_var": 0.004878489176432291,
      "learning_rate": 0.0001,
      "loss": 4.1605,
      "loss/crossentropy": 1.9240726232528687,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18174728006124496,
      "step": 14324
    },
    {
      "epoch": 0.28652,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.00850830078125,
      "learning_rate": 0.0001,
      "loss": 4.2671,
      "loss/crossentropy": 2.0952632427215576,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20621410757303238,
      "step": 14326
    },
    {
      "epoch": 0.28656,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.008326975504557292,
      "learning_rate": 0.0001,
      "loss": 4.0683,
      "loss/crossentropy": 2.2604600191116333,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2310045212507248,
      "step": 14328
    },
    {
      "epoch": 0.2866,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.00777587890625,
      "learning_rate": 0.0001,
      "loss": 4.2551,
      "loss/crossentropy": 2.0915167331695557,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23344147205352783,
      "step": 14330
    },
    {
      "epoch": 0.28664,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.00804443359375,
      "learning_rate": 0.0001,
      "loss": 3.9575,
      "loss/crossentropy": 2.2928651571273804,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23840509355068207,
      "step": 14332
    },
    {
      "epoch": 0.28668,
      "grad_norm": 2.0,
      "grad_norm_var": 0.008685048421223958,
      "learning_rate": 0.0001,
      "loss": 4.4469,
      "loss/crossentropy": 2.261072874069214,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22056513279676437,
      "step": 14334
    },
    {
      "epoch": 0.28672,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.007883453369140625,
      "learning_rate": 0.0001,
      "loss": 3.9218,
      "loss/crossentropy": 1.7591394186019897,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1900082305073738,
      "step": 14336
    },
    {
      "epoch": 0.28676,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.008426666259765625,
      "learning_rate": 0.0001,
      "loss": 4.0457,
      "loss/crossentropy": 2.129893183708191,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19578705728054047,
      "step": 14338
    },
    {
      "epoch": 0.2868,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.008451334635416667,
      "learning_rate": 0.0001,
      "loss": 4.4884,
      "loss/crossentropy": 2.0376795530319214,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2088106870651245,
      "step": 14340
    },
    {
      "epoch": 0.28684,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.005018870035807292,
      "learning_rate": 0.0001,
      "loss": 4.282,
      "loss/crossentropy": 2.1484848260879517,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2113661915063858,
      "step": 14342
    },
    {
      "epoch": 0.28688,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.00496826171875,
      "learning_rate": 0.0001,
      "loss": 4.2713,
      "loss/crossentropy": 2.204781651496887,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20908530056476593,
      "step": 14344
    },
    {
      "epoch": 0.28692,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0052154541015625,
      "learning_rate": 0.0001,
      "loss": 4.1075,
      "loss/crossentropy": 1.8096813559532166,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18912722170352936,
      "step": 14346
    },
    {
      "epoch": 0.28696,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.005580393473307291,
      "learning_rate": 0.0001,
      "loss": 4.0057,
      "loss/crossentropy": 1.8014967441558838,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18832575529813766,
      "step": 14348
    },
    {
      "epoch": 0.287,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.00789794921875,
      "learning_rate": 0.0001,
      "loss": 4.3211,
      "loss/crossentropy": 2.1443156003952026,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.232828289270401,
      "step": 14350
    },
    {
      "epoch": 0.28704,
      "grad_norm": 2.125,
      "grad_norm_var": 0.008259073893229166,
      "learning_rate": 0.0001,
      "loss": 4.0984,
      "loss/crossentropy": 2.1773927211761475,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19964880496263504,
      "step": 14352
    },
    {
      "epoch": 0.28708,
      "grad_norm": 2.0,
      "grad_norm_var": 0.007795206705729167,
      "learning_rate": 0.0001,
      "loss": 4.1432,
      "loss/crossentropy": 2.157313585281372,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2149301916360855,
      "step": 14354
    },
    {
      "epoch": 0.28712,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.007184855143229167,
      "learning_rate": 0.0001,
      "loss": 4.2938,
      "loss/crossentropy": 2.3431146144866943,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2297225296497345,
      "step": 14356
    },
    {
      "epoch": 0.28716,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.007940419514973958,
      "learning_rate": 0.0001,
      "loss": 4.2026,
      "loss/crossentropy": 1.894954264163971,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19812744110822678,
      "step": 14358
    },
    {
      "epoch": 0.2872,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.008784993489583334,
      "learning_rate": 0.0001,
      "loss": 4.0654,
      "loss/crossentropy": 2.1701435446739197,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21183644235134125,
      "step": 14360
    },
    {
      "epoch": 0.28724,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.007975006103515625,
      "learning_rate": 0.0001,
      "loss": 4.2457,
      "loss/crossentropy": 2.202796459197998,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21229087561368942,
      "step": 14362
    },
    {
      "epoch": 0.28728,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.009789784749348959,
      "learning_rate": 0.0001,
      "loss": 4.3692,
      "loss/crossentropy": 2.225682258605957,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22512775659561157,
      "step": 14364
    },
    {
      "epoch": 0.28732,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.007088216145833334,
      "learning_rate": 0.0001,
      "loss": 4.0938,
      "loss/crossentropy": 1.9562655687332153,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19557978957891464,
      "step": 14366
    },
    {
      "epoch": 0.28736,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0062945048014322914,
      "learning_rate": 0.0001,
      "loss": 4.2922,
      "loss/crossentropy": 1.794309377670288,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1851721778512001,
      "step": 14368
    },
    {
      "epoch": 0.2874,
      "grad_norm": 2.0,
      "grad_norm_var": 0.006660970052083334,
      "learning_rate": 0.0001,
      "loss": 4.1198,
      "loss/crossentropy": 2.276759445667267,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2071758210659027,
      "step": 14370
    },
    {
      "epoch": 0.28744,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.006696573893229167,
      "learning_rate": 0.0001,
      "loss": 4.1162,
      "loss/crossentropy": 2.332028031349182,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22525647282600403,
      "step": 14372
    },
    {
      "epoch": 0.28748,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.005997721354166667,
      "learning_rate": 0.0001,
      "loss": 4.2594,
      "loss/crossentropy": 1.975858986377716,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18405582010746002,
      "step": 14374
    },
    {
      "epoch": 0.28752,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.006453196207682292,
      "learning_rate": 0.0001,
      "loss": 4.0615,
      "loss/crossentropy": 2.002572774887085,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19543466717004776,
      "step": 14376
    },
    {
      "epoch": 0.28756,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.006703440348307292,
      "learning_rate": 0.0001,
      "loss": 4.1837,
      "loss/crossentropy": 2.400219678878784,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23009777069091797,
      "step": 14378
    },
    {
      "epoch": 0.2876,
      "grad_norm": 2.0,
      "grad_norm_var": 0.003352864583333333,
      "learning_rate": 0.0001,
      "loss": 4.0071,
      "loss/crossentropy": 1.6437376737594604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17785440385341644,
      "step": 14380
    },
    {
      "epoch": 0.28764,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.007225545247395834,
      "learning_rate": 0.0001,
      "loss": 4.1494,
      "loss/crossentropy": 1.8833640813827515,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22255511581897736,
      "step": 14382
    },
    {
      "epoch": 0.28768,
      "grad_norm": 2.0,
      "grad_norm_var": 0.007100423177083333,
      "learning_rate": 0.0001,
      "loss": 4.0881,
      "loss/crossentropy": 1.9113904237747192,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2050003483891487,
      "step": 14384
    },
    {
      "epoch": 0.28772,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.007765452067057292,
      "learning_rate": 0.0001,
      "loss": 4.1114,
      "loss/crossentropy": 2.236249327659607,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21349424868822098,
      "step": 14386
    },
    {
      "epoch": 0.28776,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.008090972900390625,
      "learning_rate": 0.0001,
      "loss": 3.9282,
      "loss/crossentropy": 1.8709489703178406,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19309207051992416,
      "step": 14388
    },
    {
      "epoch": 0.2878,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.007933553059895833,
      "learning_rate": 0.0001,
      "loss": 4.0871,
      "loss/crossentropy": 1.926324725151062,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19902244210243225,
      "step": 14390
    },
    {
      "epoch": 0.28784,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.00810546875,
      "learning_rate": 0.0001,
      "loss": 4.085,
      "loss/crossentropy": 1.898009479045868,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18698234856128693,
      "step": 14392
    },
    {
      "epoch": 0.28788,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0122711181640625,
      "learning_rate": 0.0001,
      "loss": 4.2367,
      "loss/crossentropy": 1.9728660583496094,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2033083364367485,
      "step": 14394
    },
    {
      "epoch": 0.28792,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.01282958984375,
      "learning_rate": 0.0001,
      "loss": 3.7965,
      "loss/crossentropy": 1.957255482673645,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20039385557174683,
      "step": 14396
    },
    {
      "epoch": 0.28796,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009897613525390625,
      "learning_rate": 0.0001,
      "loss": 3.8525,
      "loss/crossentropy": 1.8001562356948853,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2059212401509285,
      "step": 14398
    },
    {
      "epoch": 0.288,
      "grad_norm": 2.125,
      "grad_norm_var": 0.011177571614583333,
      "learning_rate": 0.0001,
      "loss": 4.1749,
      "loss/crossentropy": 1.9166946411132812,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19456689059734344,
      "step": 14400
    },
    {
      "epoch": 0.28804,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.01221923828125,
      "learning_rate": 0.0001,
      "loss": 4.3302,
      "loss/crossentropy": 2.033670485019684,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24448300898075104,
      "step": 14402
    },
    {
      "epoch": 0.28808,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.012580362955729167,
      "learning_rate": 0.0001,
      "loss": 3.8679,
      "loss/crossentropy": 2.2668023705482483,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20903322845697403,
      "step": 14404
    },
    {
      "epoch": 0.28812,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.012532297770182292,
      "learning_rate": 0.0001,
      "loss": 4.0638,
      "loss/crossentropy": 2.1076024770736694,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20488093793392181,
      "step": 14406
    },
    {
      "epoch": 0.28816,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.012117258707682292,
      "learning_rate": 0.0001,
      "loss": 4.0707,
      "loss/crossentropy": 1.996046781539917,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20725766569375992,
      "step": 14408
    },
    {
      "epoch": 0.2882,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.008365885416666666,
      "learning_rate": 0.0001,
      "loss": 4.2302,
      "loss/crossentropy": 2.2031015157699585,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19491394609212875,
      "step": 14410
    },
    {
      "epoch": 0.28824,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.008714803059895833,
      "learning_rate": 0.0001,
      "loss": 4.3962,
      "loss/crossentropy": 2.0339081287384033,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21979261189699173,
      "step": 14412
    },
    {
      "epoch": 0.28828,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.011641438802083333,
      "learning_rate": 0.0001,
      "loss": 4.0042,
      "loss/crossentropy": 1.9283286929130554,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19970305263996124,
      "step": 14414
    },
    {
      "epoch": 0.28832,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.011156972249348958,
      "learning_rate": 0.0001,
      "loss": 4.3879,
      "loss/crossentropy": 2.1092429161071777,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2114599123597145,
      "step": 14416
    },
    {
      "epoch": 0.28836,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.01004638671875,
      "learning_rate": 0.0001,
      "loss": 4.1234,
      "loss/crossentropy": 1.8102002143859863,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18407940864562988,
      "step": 14418
    },
    {
      "epoch": 0.2884,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.008906809488932292,
      "learning_rate": 0.0001,
      "loss": 4.0943,
      "loss/crossentropy": 2.3192719221115112,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22208110988140106,
      "step": 14420
    },
    {
      "epoch": 0.28844,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0082183837890625,
      "learning_rate": 0.0001,
      "loss": 4.2268,
      "loss/crossentropy": 2.2302430868148804,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21011962741613388,
      "step": 14422
    },
    {
      "epoch": 0.28848,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.008542633056640625,
      "learning_rate": 0.0001,
      "loss": 4.1235,
      "loss/crossentropy": 2.3015987873077393,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2012081891298294,
      "step": 14424
    },
    {
      "epoch": 0.28852,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0077056884765625,
      "learning_rate": 0.0001,
      "loss": 4.3094,
      "loss/crossentropy": 2.060473084449768,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20174618810415268,
      "step": 14426
    },
    {
      "epoch": 0.28856,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.014082845052083333,
      "learning_rate": 0.0001,
      "loss": 3.9594,
      "loss/crossentropy": 1.8146299719810486,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18100766092538834,
      "step": 14428
    },
    {
      "epoch": 0.2886,
      "grad_norm": 2.0,
      "grad_norm_var": 0.011277008056640624,
      "learning_rate": 0.0001,
      "loss": 3.8444,
      "loss/crossentropy": 2.1504935026168823,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20493116229772568,
      "step": 14430
    },
    {
      "epoch": 0.28864,
      "grad_norm": 2.0,
      "grad_norm_var": 0.010796864827473959,
      "learning_rate": 0.0001,
      "loss": 4.1153,
      "loss/crossentropy": 2.4947997331619263,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2339334562420845,
      "step": 14432
    },
    {
      "epoch": 0.28868,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.01612116495768229,
      "learning_rate": 0.0001,
      "loss": 4.3011,
      "loss/crossentropy": 2.421883702278137,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23815617710351944,
      "step": 14434
    },
    {
      "epoch": 0.28872,
      "grad_norm": 2.0,
      "grad_norm_var": 0.016600545247395834,
      "learning_rate": 0.0001,
      "loss": 4.2152,
      "loss/crossentropy": 2.093776822090149,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20425530523061752,
      "step": 14436
    },
    {
      "epoch": 0.28876,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.01817601521809896,
      "learning_rate": 0.0001,
      "loss": 3.9126,
      "loss/crossentropy": 2.02763295173645,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1972591131925583,
      "step": 14438
    },
    {
      "epoch": 0.2888,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0176422119140625,
      "learning_rate": 0.0001,
      "loss": 4.1348,
      "loss/crossentropy": 1.9769265055656433,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.186862014234066,
      "step": 14440
    },
    {
      "epoch": 0.28884,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.019024403889973958,
      "learning_rate": 0.0001,
      "loss": 4.4997,
      "loss/crossentropy": 2.34747314453125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22067507356405258,
      "step": 14442
    },
    {
      "epoch": 0.28888,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0128570556640625,
      "learning_rate": 0.0001,
      "loss": 4.1497,
      "loss/crossentropy": 2.1036205887794495,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2231181040406227,
      "step": 14444
    },
    {
      "epoch": 0.28892,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.01638768513997396,
      "learning_rate": 0.0001,
      "loss": 4.251,
      "loss/crossentropy": 2.2350860834121704,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23475627601146698,
      "step": 14446
    },
    {
      "epoch": 0.28896,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.01778132120768229,
      "learning_rate": 0.0001,
      "loss": 4.2095,
      "loss/crossentropy": 2.2336645126342773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20643477141857147,
      "step": 14448
    },
    {
      "epoch": 0.289,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.012923177083333333,
      "learning_rate": 0.0001,
      "loss": 4.0793,
      "loss/crossentropy": 2.345365524291992,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20650553703308105,
      "step": 14450
    },
    {
      "epoch": 0.28904,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0120025634765625,
      "learning_rate": 0.0001,
      "loss": 4.131,
      "loss/crossentropy": 2.307543635368347,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21354079991579056,
      "step": 14452
    },
    {
      "epoch": 0.28908,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.012001291910807291,
      "learning_rate": 0.0001,
      "loss": 4.1006,
      "loss/crossentropy": 1.8094390034675598,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1850133240222931,
      "step": 14454
    },
    {
      "epoch": 0.28912,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.011424763997395834,
      "learning_rate": 0.0001,
      "loss": 4.1326,
      "loss/crossentropy": 1.859872817993164,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.226328507065773,
      "step": 14456
    },
    {
      "epoch": 0.28916,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.009340159098307292,
      "learning_rate": 0.0001,
      "loss": 3.9997,
      "loss/crossentropy": 1.9507490396499634,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18718113750219345,
      "step": 14458
    },
    {
      "epoch": 0.2892,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.01126708984375,
      "learning_rate": 0.0001,
      "loss": 3.7868,
      "loss/crossentropy": 1.6642532348632812,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1674812287092209,
      "step": 14460
    },
    {
      "epoch": 0.28924,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.007039133707682292,
      "learning_rate": 0.0001,
      "loss": 3.9523,
      "loss/crossentropy": 1.9071928262710571,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18480891734361649,
      "step": 14462
    },
    {
      "epoch": 0.28928,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.005718739827473959,
      "learning_rate": 0.0001,
      "loss": 4.2876,
      "loss/crossentropy": 2.0758568048477173,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2002219259738922,
      "step": 14464
    },
    {
      "epoch": 0.28932,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.006769816080729167,
      "learning_rate": 0.0001,
      "loss": 3.9438,
      "loss/crossentropy": 1.9777529835700989,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19538866728544235,
      "step": 14466
    },
    {
      "epoch": 0.28936,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0062652587890625,
      "learning_rate": 0.0001,
      "loss": 4.1262,
      "loss/crossentropy": 2.008077323436737,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20578377693891525,
      "step": 14468
    },
    {
      "epoch": 0.2894,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.006673177083333333,
      "learning_rate": 0.0001,
      "loss": 3.9341,
      "loss/crossentropy": 2.011266529560089,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20298370718955994,
      "step": 14470
    },
    {
      "epoch": 0.28944,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0057769775390625,
      "learning_rate": 0.0001,
      "loss": 4.201,
      "loss/crossentropy": 2.231672167778015,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21847998350858688,
      "step": 14472
    },
    {
      "epoch": 0.28948,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.007279205322265625,
      "learning_rate": 0.0001,
      "loss": 4.382,
      "loss/crossentropy": 2.2268325090408325,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22168640047311783,
      "step": 14474
    },
    {
      "epoch": 0.28952,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.012105305989583334,
      "learning_rate": 0.0001,
      "loss": 4.2193,
      "loss/crossentropy": 2.2705942392349243,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2110700011253357,
      "step": 14476
    },
    {
      "epoch": 0.28956,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.012566884358723959,
      "learning_rate": 0.0001,
      "loss": 3.8754,
      "loss/crossentropy": 1.893052339553833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18865809589624405,
      "step": 14478
    },
    {
      "epoch": 0.2896,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.013732655843098959,
      "learning_rate": 0.0001,
      "loss": 4.0989,
      "loss/crossentropy": 2.020704984664917,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18822921812534332,
      "step": 14480
    },
    {
      "epoch": 0.28964,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.014632161458333333,
      "learning_rate": 0.0001,
      "loss": 4.1651,
      "loss/crossentropy": 1.762313961982727,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.204545259475708,
      "step": 14482
    },
    {
      "epoch": 0.28968,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.013158162434895834,
      "learning_rate": 0.0001,
      "loss": 4.0261,
      "loss/crossentropy": 1.8793032765388489,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17913836985826492,
      "step": 14484
    },
    {
      "epoch": 0.28972,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.013372548421223958,
      "learning_rate": 0.0001,
      "loss": 4.2731,
      "loss/crossentropy": 2.428762197494507,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23766332119703293,
      "step": 14486
    },
    {
      "epoch": 0.28976,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.014969635009765624,
      "learning_rate": 0.0001,
      "loss": 4.1134,
      "loss/crossentropy": 2.2160138487815857,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21619082987308502,
      "step": 14488
    },
    {
      "epoch": 0.2898,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0150787353515625,
      "learning_rate": 0.0001,
      "loss": 4.1049,
      "loss/crossentropy": 1.8169404864311218,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1981067657470703,
      "step": 14490
    },
    {
      "epoch": 0.28984,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.0131591796875,
      "learning_rate": 0.0001,
      "loss": 4.1456,
      "loss/crossentropy": 2.147166609764099,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1993183195590973,
      "step": 14492
    },
    {
      "epoch": 0.28988,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.011362457275390625,
      "learning_rate": 0.0001,
      "loss": 3.8507,
      "loss/crossentropy": 1.7740440964698792,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18381935358047485,
      "step": 14494
    },
    {
      "epoch": 0.28992,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.010428619384765626,
      "learning_rate": 0.0001,
      "loss": 4.0327,
      "loss/crossentropy": 2.206219792366028,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21052061766386032,
      "step": 14496
    },
    {
      "epoch": 0.28996,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008600870768229166,
      "learning_rate": 0.0001,
      "loss": 4.4421,
      "loss/crossentropy": 2.4030661582946777,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22094716131687164,
      "step": 14498
    },
    {
      "epoch": 0.29,
      "grad_norm": 2.25,
      "grad_norm_var": 0.016633097330729166,
      "learning_rate": 0.0001,
      "loss": 4.4593,
      "loss/crossentropy": 2.142518997192383,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2111787647008896,
      "step": 14500
    },
    {
      "epoch": 0.29004,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.015633138020833333,
      "learning_rate": 0.0001,
      "loss": 4.3679,
      "loss/crossentropy": 1.9228865504264832,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19389715045690536,
      "step": 14502
    },
    {
      "epoch": 0.29008,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.014141591389973958,
      "learning_rate": 0.0001,
      "loss": 4.011,
      "loss/crossentropy": 1.789183259010315,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18835698068141937,
      "step": 14504
    },
    {
      "epoch": 0.29012,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.013798014322916666,
      "learning_rate": 0.0001,
      "loss": 4.2862,
      "loss/crossentropy": 2.3876765966415405,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23156627267599106,
      "step": 14506
    },
    {
      "epoch": 0.29016,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.010151926676432292,
      "learning_rate": 0.0001,
      "loss": 4.2024,
      "loss/crossentropy": 2.055350124835968,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1995788812637329,
      "step": 14508
    },
    {
      "epoch": 0.2902,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.010007476806640625,
      "learning_rate": 0.0001,
      "loss": 4.343,
      "loss/crossentropy": 2.2045267820358276,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2197813093662262,
      "step": 14510
    },
    {
      "epoch": 0.29024,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.014503733317057291,
      "learning_rate": 0.0001,
      "loss": 3.7136,
      "loss/crossentropy": 1.9596665501594543,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17329587787389755,
      "step": 14512
    },
    {
      "epoch": 0.29028,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.017465972900390626,
      "learning_rate": 0.0001,
      "loss": 3.9546,
      "loss/crossentropy": 1.8273005485534668,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17768846452236176,
      "step": 14514
    },
    {
      "epoch": 0.29032,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.011739095052083334,
      "learning_rate": 0.0001,
      "loss": 3.9924,
      "loss/crossentropy": 2.0428807735443115,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2104019746184349,
      "step": 14516
    },
    {
      "epoch": 0.29036,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.014216868082682292,
      "learning_rate": 0.0001,
      "loss": 4.3591,
      "loss/crossentropy": 2.1242733001708984,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2153768166899681,
      "step": 14518
    },
    {
      "epoch": 0.2904,
      "grad_norm": 1.875,
      "grad_norm_var": 0.013944244384765625,
      "learning_rate": 0.0001,
      "loss": 4.0831,
      "loss/crossentropy": 1.9223615527153015,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18382133543491364,
      "step": 14520
    },
    {
      "epoch": 0.29044,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.013993326822916667,
      "learning_rate": 0.0001,
      "loss": 3.9527,
      "loss/crossentropy": 2.12492972612381,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19818739593029022,
      "step": 14522
    },
    {
      "epoch": 0.29048,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.013665517171223959,
      "learning_rate": 0.0001,
      "loss": 4.1539,
      "loss/crossentropy": 2.0251020789146423,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19633153080940247,
      "step": 14524
    },
    {
      "epoch": 0.29052,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.014949544270833334,
      "learning_rate": 0.0001,
      "loss": 4.192,
      "loss/crossentropy": 2.177084445953369,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22618898749351501,
      "step": 14526
    },
    {
      "epoch": 0.29056,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.013997141520182292,
      "learning_rate": 0.0001,
      "loss": 3.9215,
      "loss/crossentropy": 1.718012809753418,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18632279336452484,
      "step": 14528
    },
    {
      "epoch": 0.2906,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.017850494384765624,
      "learning_rate": 0.0001,
      "loss": 4.2869,
      "loss/crossentropy": 1.7026863098144531,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.211147278547287,
      "step": 14530
    },
    {
      "epoch": 0.29064,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.017682902018229165,
      "learning_rate": 0.0001,
      "loss": 3.9369,
      "loss/crossentropy": 1.6652680039405823,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16967280209064484,
      "step": 14532
    },
    {
      "epoch": 0.29068,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.013996378580729166,
      "learning_rate": 0.0001,
      "loss": 4.0815,
      "loss/crossentropy": 2.191527247428894,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20215752720832825,
      "step": 14534
    },
    {
      "epoch": 0.29072,
      "grad_norm": 2.125,
      "grad_norm_var": 0.015927886962890624,
      "learning_rate": 0.0001,
      "loss": 3.8817,
      "loss/crossentropy": 1.8385645747184753,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19906776398420334,
      "step": 14536
    },
    {
      "epoch": 0.29076,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.01665013631184896,
      "learning_rate": 0.0001,
      "loss": 3.9186,
      "loss/crossentropy": 2.0456249117851257,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2057228460907936,
      "step": 14538
    },
    {
      "epoch": 0.2908,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.018651326497395832,
      "learning_rate": 0.0001,
      "loss": 4.0636,
      "loss/crossentropy": 2.0413625836372375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20096366852521896,
      "step": 14540
    },
    {
      "epoch": 0.29084,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.016755167643229166,
      "learning_rate": 0.0001,
      "loss": 3.7819,
      "loss/crossentropy": 1.723297357559204,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.180135078728199,
      "step": 14542
    },
    {
      "epoch": 0.29088,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.01578343709309896,
      "learning_rate": 0.0001,
      "loss": 4.2115,
      "loss/crossentropy": 2.3143200874328613,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2289530113339424,
      "step": 14544
    },
    {
      "epoch": 0.29092,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.010017903645833333,
      "learning_rate": 0.0001,
      "loss": 4.408,
      "loss/crossentropy": 2.1612138748168945,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21224550902843475,
      "step": 14546
    },
    {
      "epoch": 0.29096,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.009227498372395834,
      "learning_rate": 0.0001,
      "loss": 4.1101,
      "loss/crossentropy": 2.0591673851013184,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18913161754608154,
      "step": 14548
    },
    {
      "epoch": 0.291,
      "grad_norm": 1.875,
      "grad_norm_var": 0.009505208333333333,
      "learning_rate": 0.0001,
      "loss": 4.132,
      "loss/crossentropy": 2.0901013016700745,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19315654039382935,
      "step": 14550
    },
    {
      "epoch": 0.29104,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.006833648681640625,
      "learning_rate": 0.0001,
      "loss": 3.9351,
      "loss/crossentropy": 2.0716471672058105,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1951451674103737,
      "step": 14552
    },
    {
      "epoch": 0.29108,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.008186848958333333,
      "learning_rate": 0.0001,
      "loss": 4.3656,
      "loss/crossentropy": 2.084408760070801,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21354945749044418,
      "step": 14554
    },
    {
      "epoch": 0.29112,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.006486002604166667,
      "learning_rate": 0.0001,
      "loss": 4.0342,
      "loss/crossentropy": 2.154771566390991,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20277246832847595,
      "step": 14556
    },
    {
      "epoch": 0.29116,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0069353739420572914,
      "learning_rate": 0.0001,
      "loss": 4.259,
      "loss/crossentropy": 2.24834144115448,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23204077035188675,
      "step": 14558
    },
    {
      "epoch": 0.2912,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.006951649983723958,
      "learning_rate": 0.0001,
      "loss": 3.8054,
      "loss/crossentropy": 1.8239850401878357,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18208947032690048,
      "step": 14560
    },
    {
      "epoch": 0.29124,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.006701405843098958,
      "learning_rate": 0.0001,
      "loss": 3.9465,
      "loss/crossentropy": 1.969277262687683,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2093023955821991,
      "step": 14562
    },
    {
      "epoch": 0.29128,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.007670084635416667,
      "learning_rate": 0.0001,
      "loss": 3.7176,
      "loss/crossentropy": 1.561119556427002,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1653747707605362,
      "step": 14564
    },
    {
      "epoch": 0.29132,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.008154042561848958,
      "learning_rate": 0.0001,
      "loss": 4.3307,
      "loss/crossentropy": 2.2341216802597046,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23967482149600983,
      "step": 14566
    },
    {
      "epoch": 0.29136,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.008227284749348958,
      "learning_rate": 0.0001,
      "loss": 3.9899,
      "loss/crossentropy": 1.999854326248169,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19598717242479324,
      "step": 14568
    },
    {
      "epoch": 0.2914,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.006898752848307292,
      "learning_rate": 0.0001,
      "loss": 4.1262,
      "loss/crossentropy": 1.8079020380973816,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1919228658080101,
      "step": 14570
    },
    {
      "epoch": 0.29144,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.009970855712890626,
      "learning_rate": 0.0001,
      "loss": 3.9946,
      "loss/crossentropy": 2.044901430606842,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21737617254257202,
      "step": 14572
    },
    {
      "epoch": 0.29148,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009806060791015625,
      "learning_rate": 0.0001,
      "loss": 4.0498,
      "loss/crossentropy": 1.972197949886322,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20828764885663986,
      "step": 14574
    },
    {
      "epoch": 0.29152,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010131581624348959,
      "learning_rate": 0.0001,
      "loss": 4.0011,
      "loss/crossentropy": 1.9129782915115356,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17108886688947678,
      "step": 14576
    },
    {
      "epoch": 0.29156,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.01068115234375,
      "learning_rate": 0.0001,
      "loss": 4.1458,
      "loss/crossentropy": 2.0545085072517395,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20631857216358185,
      "step": 14578
    },
    {
      "epoch": 0.2916,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.012654368082682292,
      "learning_rate": 0.0001,
      "loss": 4.355,
      "loss/crossentropy": 2.0274637937545776,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2099764049053192,
      "step": 14580
    },
    {
      "epoch": 0.29164,
      "grad_norm": 1.875,
      "grad_norm_var": 0.013765207926432292,
      "learning_rate": 0.0001,
      "loss": 3.9014,
      "loss/crossentropy": 1.8154139518737793,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19045238196849823,
      "step": 14582
    },
    {
      "epoch": 0.29168,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.014937082926432291,
      "learning_rate": 0.0001,
      "loss": 4.0327,
      "loss/crossentropy": 2.061666965484619,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21272550523281097,
      "step": 14584
    },
    {
      "epoch": 0.29172,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.01580785115559896,
      "learning_rate": 0.0001,
      "loss": 4.6019,
      "loss/crossentropy": 2.3831146955490112,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22327116876840591,
      "step": 14586
    },
    {
      "epoch": 0.29176,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.013090006510416667,
      "learning_rate": 0.0001,
      "loss": 4.2142,
      "loss/crossentropy": 1.8024229407310486,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1871618777513504,
      "step": 14588
    },
    {
      "epoch": 0.2918,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.014967600504557291,
      "learning_rate": 0.0001,
      "loss": 4.1611,
      "loss/crossentropy": 2.289118528366089,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22116435319185257,
      "step": 14590
    },
    {
      "epoch": 0.29184,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.013277180989583333,
      "learning_rate": 0.0001,
      "loss": 4.1358,
      "loss/crossentropy": 2.11151522397995,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22029711306095123,
      "step": 14592
    },
    {
      "epoch": 0.29188,
      "grad_norm": 2.125,
      "grad_norm_var": 0.013288370768229167,
      "learning_rate": 0.0001,
      "loss": 4.3103,
      "loss/crossentropy": 1.7856897711753845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1815958321094513,
      "step": 14594
    },
    {
      "epoch": 0.29192,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0101959228515625,
      "learning_rate": 0.0001,
      "loss": 4.1085,
      "loss/crossentropy": 1.9745690822601318,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2074003368616104,
      "step": 14596
    },
    {
      "epoch": 0.29196,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0091461181640625,
      "learning_rate": 0.0001,
      "loss": 4.3034,
      "loss/crossentropy": 1.9199401140213013,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19998926669359207,
      "step": 14598
    },
    {
      "epoch": 0.292,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.007743072509765625,
      "learning_rate": 0.0001,
      "loss": 4.0842,
      "loss/crossentropy": 1.9938844442367554,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20138052105903625,
      "step": 14600
    },
    {
      "epoch": 0.29204,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0059397379557291664,
      "learning_rate": 0.0001,
      "loss": 4.0025,
      "loss/crossentropy": 1.6290993094444275,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18716775625944138,
      "step": 14602
    },
    {
      "epoch": 0.29208,
      "grad_norm": 1.875,
      "grad_norm_var": 0.006115468343098959,
      "learning_rate": 0.0001,
      "loss": 3.9953,
      "loss/crossentropy": 2.116807520389557,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2113799750804901,
      "step": 14604
    },
    {
      "epoch": 0.29212,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.006392415364583333,
      "learning_rate": 0.0001,
      "loss": 4.0527,
      "loss/crossentropy": 1.6732030510902405,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19776474684476852,
      "step": 14606
    },
    {
      "epoch": 0.29216,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.006762440999348958,
      "learning_rate": 0.0001,
      "loss": 4.2572,
      "loss/crossentropy": 1.9403682351112366,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18560070544481277,
      "step": 14608
    },
    {
      "epoch": 0.2922,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.011641438802083333,
      "learning_rate": 0.0001,
      "loss": 4.2238,
      "loss/crossentropy": 1.9337440729141235,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2172246277332306,
      "step": 14610
    },
    {
      "epoch": 0.29224,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.011836751302083334,
      "learning_rate": 0.0001,
      "loss": 4.1839,
      "loss/crossentropy": 2.1603177785873413,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2076597660779953,
      "step": 14612
    },
    {
      "epoch": 0.29228,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0118804931640625,
      "learning_rate": 0.0001,
      "loss": 4.1103,
      "loss/crossentropy": 1.979960322380066,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2007075995206833,
      "step": 14614
    },
    {
      "epoch": 0.29232,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0134429931640625,
      "learning_rate": 0.0001,
      "loss": 3.9901,
      "loss/crossentropy": 2.1262378096580505,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2033269852399826,
      "step": 14616
    },
    {
      "epoch": 0.29236,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.013327789306640626,
      "learning_rate": 0.0001,
      "loss": 3.9001,
      "loss/crossentropy": 1.836608648300171,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17032578587532043,
      "step": 14618
    },
    {
      "epoch": 0.2924,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.012473297119140626,
      "learning_rate": 0.0001,
      "loss": 4.0165,
      "loss/crossentropy": 1.784467339515686,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.187791109085083,
      "step": 14620
    },
    {
      "epoch": 0.29244,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.010196940104166666,
      "learning_rate": 0.0001,
      "loss": 4.2963,
      "loss/crossentropy": 2.5023289918899536,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2469349130988121,
      "step": 14622
    },
    {
      "epoch": 0.29248,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0142974853515625,
      "learning_rate": 0.0001,
      "loss": 4.3748,
      "loss/crossentropy": 2.244086742401123,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20690800249576569,
      "step": 14624
    },
    {
      "epoch": 0.29252,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.008829752604166666,
      "learning_rate": 0.0001,
      "loss": 4.2594,
      "loss/crossentropy": 2.167649209499359,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23074676096439362,
      "step": 14626
    },
    {
      "epoch": 0.29256,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0092437744140625,
      "learning_rate": 0.0001,
      "loss": 4.1837,
      "loss/crossentropy": 1.934467613697052,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2041843980550766,
      "step": 14628
    },
    {
      "epoch": 0.2926,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0152740478515625,
      "learning_rate": 0.0001,
      "loss": 4.3277,
      "loss/crossentropy": 2.457708954811096,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24325969815254211,
      "step": 14630
    },
    {
      "epoch": 0.29264,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.014446767171223958,
      "learning_rate": 0.0001,
      "loss": 4.0523,
      "loss/crossentropy": 2.165565609931946,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20386752486228943,
      "step": 14632
    },
    {
      "epoch": 0.29268,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.013677724202473958,
      "learning_rate": 0.0001,
      "loss": 4.0767,
      "loss/crossentropy": 1.9475921988487244,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2017974555492401,
      "step": 14634
    },
    {
      "epoch": 0.29272,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.015819295247395834,
      "learning_rate": 0.0001,
      "loss": 4.1435,
      "loss/crossentropy": 1.95538991689682,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19920305162668228,
      "step": 14636
    },
    {
      "epoch": 0.29276,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.01803766886393229,
      "learning_rate": 0.0001,
      "loss": 4.4073,
      "loss/crossentropy": 2.1813108921051025,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2130545824766159,
      "step": 14638
    },
    {
      "epoch": 0.2928,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.01672337849934896,
      "learning_rate": 0.0001,
      "loss": 4.1841,
      "loss/crossentropy": 2.198926568031311,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21660766005516052,
      "step": 14640
    },
    {
      "epoch": 0.29284,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.01692682902018229,
      "learning_rate": 0.0001,
      "loss": 4.2872,
      "loss/crossentropy": 2.244659662246704,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22478342056274414,
      "step": 14642
    },
    {
      "epoch": 0.29288,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.01608861287434896,
      "learning_rate": 0.0001,
      "loss": 4.234,
      "loss/crossentropy": 2.1159621477127075,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22023546695709229,
      "step": 14644
    },
    {
      "epoch": 0.29292,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.011067708333333334,
      "learning_rate": 0.0001,
      "loss": 3.9166,
      "loss/crossentropy": 2.2336114645004272,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20683540403842926,
      "step": 14646
    },
    {
      "epoch": 0.29296,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.011128743489583334,
      "learning_rate": 0.0001,
      "loss": 3.8647,
      "loss/crossentropy": 1.9015604257583618,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2791958376765251,
      "step": 14648
    },
    {
      "epoch": 0.293,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.011180623372395834,
      "learning_rate": 0.0001,
      "loss": 3.7325,
      "loss/crossentropy": 1.6337950229644775,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17279411852359772,
      "step": 14650
    },
    {
      "epoch": 0.29304,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009618123372395834,
      "learning_rate": 0.0001,
      "loss": 4.111,
      "loss/crossentropy": 1.920238435268402,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18912465870380402,
      "step": 14652
    },
    {
      "epoch": 0.29308,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006075032552083333,
      "learning_rate": 0.0001,
      "loss": 4.4346,
      "loss/crossentropy": 2.5529074668884277,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23994534462690353,
      "step": 14654
    },
    {
      "epoch": 0.29312,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0100982666015625,
      "learning_rate": 0.0001,
      "loss": 4.3154,
      "loss/crossentropy": 2.1320537328720093,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21480896323919296,
      "step": 14656
    },
    {
      "epoch": 0.29316,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.013205718994140626,
      "learning_rate": 0.0001,
      "loss": 3.867,
      "loss/crossentropy": 1.8921156525611877,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1749916449189186,
      "step": 14658
    },
    {
      "epoch": 0.2932,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.015433502197265626,
      "learning_rate": 0.0001,
      "loss": 3.7405,
      "loss/crossentropy": 2.2475300431251526,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20675917714834213,
      "step": 14660
    },
    {
      "epoch": 0.29324,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.015384928385416666,
      "learning_rate": 0.0001,
      "loss": 4.2091,
      "loss/crossentropy": 2.291227698326111,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2244400456547737,
      "step": 14662
    },
    {
      "epoch": 0.29328,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.017976888020833335,
      "learning_rate": 0.0001,
      "loss": 4.1468,
      "loss/crossentropy": 1.8830837607383728,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21137318760156631,
      "step": 14664
    },
    {
      "epoch": 0.29332,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.019742838541666665,
      "learning_rate": 0.0001,
      "loss": 4.3424,
      "loss/crossentropy": 2.366846203804016,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22360631078481674,
      "step": 14666
    },
    {
      "epoch": 0.29336,
      "grad_norm": 1.875,
      "grad_norm_var": 0.021720123291015626,
      "learning_rate": 0.0001,
      "loss": 4.1778,
      "loss/crossentropy": 2.165378987789154,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2088456228375435,
      "step": 14668
    },
    {
      "epoch": 0.2934,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.021809895833333332,
      "learning_rate": 0.0001,
      "loss": 4.178,
      "loss/crossentropy": 1.782981276512146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17654509842395782,
      "step": 14670
    },
    {
      "epoch": 0.29344,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.021581013997395832,
      "learning_rate": 0.0001,
      "loss": 4.0796,
      "loss/crossentropy": 1.9891030192375183,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18961766362190247,
      "step": 14672
    },
    {
      "epoch": 0.29348,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.015738932291666667,
      "learning_rate": 0.0001,
      "loss": 4.0359,
      "loss/crossentropy": 1.9350959062576294,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2124071717262268,
      "step": 14674
    },
    {
      "epoch": 0.29352,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.01131591796875,
      "learning_rate": 0.0001,
      "loss": 4.2651,
      "loss/crossentropy": 2.1763634085655212,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22027570754289627,
      "step": 14676
    },
    {
      "epoch": 0.29356,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.012499745686848958,
      "learning_rate": 0.0001,
      "loss": 4.1502,
      "loss/crossentropy": 1.741381287574768,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17308862507343292,
      "step": 14678
    },
    {
      "epoch": 0.2936,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.011633046468098958,
      "learning_rate": 0.0001,
      "loss": 4.2012,
      "loss/crossentropy": 2.352774977684021,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20929308235645294,
      "step": 14680
    },
    {
      "epoch": 0.29364,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.012679036458333333,
      "learning_rate": 0.0001,
      "loss": 3.9799,
      "loss/crossentropy": 1.6342085003852844,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17740759253501892,
      "step": 14682
    },
    {
      "epoch": 0.29368,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0110260009765625,
      "learning_rate": 0.0001,
      "loss": 4.1968,
      "loss/crossentropy": 2.315647602081299,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2202264443039894,
      "step": 14684
    },
    {
      "epoch": 0.29372,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.011860911051432292,
      "learning_rate": 0.0001,
      "loss": 4.0276,
      "loss/crossentropy": 2.1667529344558716,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20557944476604462,
      "step": 14686
    },
    {
      "epoch": 0.29376,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.012970987955729167,
      "learning_rate": 0.0001,
      "loss": 3.7212,
      "loss/crossentropy": 1.9426026940345764,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17865119129419327,
      "step": 14688
    },
    {
      "epoch": 0.2938,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.013720703125,
      "learning_rate": 0.0001,
      "loss": 3.9345,
      "loss/crossentropy": 2.012951970100403,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1918969601392746,
      "step": 14690
    },
    {
      "epoch": 0.29384,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0097808837890625,
      "learning_rate": 0.0001,
      "loss": 4.2131,
      "loss/crossentropy": 2.2038668394088745,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21520675718784332,
      "step": 14692
    },
    {
      "epoch": 0.29388,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.009659576416015624,
      "learning_rate": 0.0001,
      "loss": 3.9577,
      "loss/crossentropy": 2.284528613090515,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2084675058722496,
      "step": 14694
    },
    {
      "epoch": 0.29392,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.009991200764973958,
      "learning_rate": 0.0001,
      "loss": 4.1159,
      "loss/crossentropy": 2.3619518280029297,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23107902705669403,
      "step": 14696
    },
    {
      "epoch": 0.29396,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009870402018229167,
      "learning_rate": 0.0001,
      "loss": 4.1488,
      "loss/crossentropy": 2.279863119125366,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21235084533691406,
      "step": 14698
    },
    {
      "epoch": 0.294,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.009639231363932292,
      "learning_rate": 0.0001,
      "loss": 4.2216,
      "loss/crossentropy": 2.128455936908722,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21381118893623352,
      "step": 14700
    },
    {
      "epoch": 0.29404,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.009014638264973958,
      "learning_rate": 0.0001,
      "loss": 4.0147,
      "loss/crossentropy": 2.265346884727478,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2086036428809166,
      "step": 14702
    },
    {
      "epoch": 0.29408,
      "grad_norm": 2.0,
      "grad_norm_var": 0.008421834309895833,
      "learning_rate": 0.0001,
      "loss": 4.2778,
      "loss/crossentropy": 2.4323991537094116,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21795648336410522,
      "step": 14704
    },
    {
      "epoch": 0.29412,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.30881729125976565,
      "learning_rate": 0.0001,
      "loss": 4.3864,
      "loss/crossentropy": 2.2948319911956787,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23302766680717468,
      "step": 14706
    },
    {
      "epoch": 0.29416,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.3063140869140625,
      "learning_rate": 0.0001,
      "loss": 4.121,
      "loss/crossentropy": 2.1273980140686035,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2240317091345787,
      "step": 14708
    },
    {
      "epoch": 0.2942,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.304357655843099,
      "learning_rate": 0.0001,
      "loss": 3.7468,
      "loss/crossentropy": 2.0593321323394775,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19727306067943573,
      "step": 14710
    },
    {
      "epoch": 0.29424,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.3060373942057292,
      "learning_rate": 0.0001,
      "loss": 3.9979,
      "loss/crossentropy": 2.339016914367676,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2313404530286789,
      "step": 14712
    },
    {
      "epoch": 0.29428,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.30822652180989585,
      "learning_rate": 0.0001,
      "loss": 3.9406,
      "loss/crossentropy": 2.1587395668029785,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20085373520851135,
      "step": 14714
    },
    {
      "epoch": 0.29432,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.3103912353515625,
      "learning_rate": 0.0001,
      "loss": 4.0616,
      "loss/crossentropy": 1.8317759037017822,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20150133967399597,
      "step": 14716
    },
    {
      "epoch": 0.29436,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.31324437459309895,
      "learning_rate": 0.0001,
      "loss": 3.9291,
      "loss/crossentropy": 1.849327266216278,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17959115654230118,
      "step": 14718
    },
    {
      "epoch": 0.2944,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.32099507649739584,
      "learning_rate": 0.0001,
      "loss": 3.7933,
      "loss/crossentropy": 1.652997612953186,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17243028432130814,
      "step": 14720
    },
    {
      "epoch": 0.29444,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.012995402018229166,
      "learning_rate": 0.0001,
      "loss": 4.2919,
      "loss/crossentropy": 1.968604028224945,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20564967393875122,
      "step": 14722
    },
    {
      "epoch": 0.29448,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.007466634114583333,
      "learning_rate": 0.0001,
      "loss": 4.1211,
      "loss/crossentropy": 2.349083185195923,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24004054814577103,
      "step": 14724
    },
    {
      "epoch": 0.29452,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.006200917561848958,
      "learning_rate": 0.0001,
      "loss": 4.2078,
      "loss/crossentropy": 2.1629676818847656,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2260306179523468,
      "step": 14726
    },
    {
      "epoch": 0.29456,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.012572224934895833,
      "learning_rate": 0.0001,
      "loss": 4.5995,
      "loss/crossentropy": 2.221126079559326,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21054977923631668,
      "step": 14728
    },
    {
      "epoch": 0.2946,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.3638987223307292,
      "learning_rate": 0.0001,
      "loss": 4.2588,
      "loss/crossentropy": 2.0333986282348633,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27580036222934723,
      "step": 14730
    },
    {
      "epoch": 0.29464,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.36292699178059895,
      "learning_rate": 0.0001,
      "loss": 4.2601,
      "loss/crossentropy": 2.0025470852851868,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20166245102882385,
      "step": 14732
    },
    {
      "epoch": 0.29468,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.35410868326822914,
      "learning_rate": 0.0001,
      "loss": 4.1044,
      "loss/crossentropy": 2.2882933616638184,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1972086951136589,
      "step": 14734
    },
    {
      "epoch": 0.29472,
      "grad_norm": 2.125,
      "grad_norm_var": 0.3395342508951823,
      "learning_rate": 0.0001,
      "loss": 4.1616,
      "loss/crossentropy": 2.2404085397720337,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22015579044818878,
      "step": 14736
    },
    {
      "epoch": 0.29476,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.33505223592122396,
      "learning_rate": 0.0001,
      "loss": 4.0132,
      "loss/crossentropy": 1.9309356808662415,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2082076519727707,
      "step": 14738
    },
    {
      "epoch": 0.2948,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.33227437337239585,
      "learning_rate": 0.0001,
      "loss": 4.1652,
      "loss/crossentropy": 2.2118934988975525,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20175088196992874,
      "step": 14740
    },
    {
      "epoch": 0.29484,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.36171875,
      "learning_rate": 0.0001,
      "loss": 4.1204,
      "loss/crossentropy": 2.19997900724411,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22470968216657639,
      "step": 14742
    },
    {
      "epoch": 0.29488,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.365966796875,
      "learning_rate": 0.0001,
      "loss": 4.2159,
      "loss/crossentropy": 1.8140272498130798,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20032335817813873,
      "step": 14744
    },
    {
      "epoch": 0.29492,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.07648086547851562,
      "learning_rate": 0.0001,
      "loss": 3.957,
      "loss/crossentropy": 2.0233620405197144,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20464400947093964,
      "step": 14746
    },
    {
      "epoch": 0.29496,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.07643407185872396,
      "learning_rate": 0.0001,
      "loss": 4.3946,
      "loss/crossentropy": 2.3900705575942993,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23763202875852585,
      "step": 14748
    },
    {
      "epoch": 0.295,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.07768325805664063,
      "learning_rate": 0.0001,
      "loss": 4.205,
      "loss/crossentropy": 1.9895538687705994,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20736829191446304,
      "step": 14750
    },
    {
      "epoch": 0.29504,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0857421875,
      "learning_rate": 0.0001,
      "loss": 4.1454,
      "loss/crossentropy": 1.7315555810928345,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1839754432439804,
      "step": 14752
    },
    {
      "epoch": 0.29508,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.08105367024739583,
      "learning_rate": 0.0001,
      "loss": 4.5482,
      "loss/crossentropy": 2.44241464138031,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2505839467048645,
      "step": 14754
    },
    {
      "epoch": 0.29512,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0849029541015625,
      "learning_rate": 0.0001,
      "loss": 3.782,
      "loss/crossentropy": 1.7948896884918213,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1987960934638977,
      "step": 14756
    },
    {
      "epoch": 0.29516,
      "grad_norm": 2.125,
      "grad_norm_var": 0.020334625244140626,
      "learning_rate": 0.0001,
      "loss": 4.0898,
      "loss/crossentropy": 1.710760235786438,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18624412268400192,
      "step": 14758
    },
    {
      "epoch": 0.2952,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.027205149332682293,
      "learning_rate": 0.0001,
      "loss": 4.4651,
      "loss/crossentropy": 2.0127750635147095,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20178033411502838,
      "step": 14760
    },
    {
      "epoch": 0.29524,
      "grad_norm": 2.0,
      "grad_norm_var": 0.027042388916015625,
      "learning_rate": 0.0001,
      "loss": 4.1152,
      "loss/crossentropy": 1.9764790534973145,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19000716507434845,
      "step": 14762
    },
    {
      "epoch": 0.29528,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.020930735270182292,
      "learning_rate": 0.0001,
      "loss": 3.9634,
      "loss/crossentropy": 2.0221771597862244,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1980113685131073,
      "step": 14764
    },
    {
      "epoch": 0.29532,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.019972483317057293,
      "learning_rate": 0.0001,
      "loss": 4.1721,
      "loss/crossentropy": 2.278168559074402,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.231141597032547,
      "step": 14766
    },
    {
      "epoch": 0.29536,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0168853759765625,
      "learning_rate": 0.0001,
      "loss": 4.1308,
      "loss/crossentropy": 2.1837246417999268,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2026391476392746,
      "step": 14768
    },
    {
      "epoch": 0.2954,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.019795735677083332,
      "learning_rate": 0.0001,
      "loss": 4.0694,
      "loss/crossentropy": 2.1170668601989746,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1871805638074875,
      "step": 14770
    },
    {
      "epoch": 0.29544,
      "grad_norm": 2.125,
      "grad_norm_var": 0.01784032185872396,
      "learning_rate": 0.0001,
      "loss": 4.1003,
      "loss/crossentropy": 1.9905366897583008,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1951729580760002,
      "step": 14772
    },
    {
      "epoch": 0.29548,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.017146809895833334,
      "learning_rate": 0.0001,
      "loss": 3.9811,
      "loss/crossentropy": 1.781678318977356,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19464702904224396,
      "step": 14774
    },
    {
      "epoch": 0.29552,
      "grad_norm": 1.875,
      "grad_norm_var": 0.012303670247395834,
      "learning_rate": 0.0001,
      "loss": 4.1944,
      "loss/crossentropy": 2.2671462297439575,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22269655019044876,
      "step": 14776
    },
    {
      "epoch": 0.29556,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.012511952718098959,
      "learning_rate": 0.0001,
      "loss": 4.2128,
      "loss/crossentropy": 2.0294137001037598,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21939906477928162,
      "step": 14778
    },
    {
      "epoch": 0.2956,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.011264801025390625,
      "learning_rate": 0.0001,
      "loss": 4.1295,
      "loss/crossentropy": 1.9174134731292725,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19592813402414322,
      "step": 14780
    },
    {
      "epoch": 0.29564,
      "grad_norm": 2.0,
      "grad_norm_var": 0.009528605143229167,
      "learning_rate": 0.0001,
      "loss": 3.9877,
      "loss/crossentropy": 2.0033875703811646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2121451273560524,
      "step": 14782
    },
    {
      "epoch": 0.29568,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.09626439412434896,
      "learning_rate": 0.0001,
      "loss": 4.0263,
      "loss/crossentropy": 2.209542691707611,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18373841792345047,
      "step": 14784
    },
    {
      "epoch": 0.29572,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.09244155883789062,
      "learning_rate": 0.0001,
      "loss": 4.149,
      "loss/crossentropy": 2.0651984214782715,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20486845821142197,
      "step": 14786
    },
    {
      "epoch": 0.29576,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.09123433430989583,
      "learning_rate": 0.0001,
      "loss": 4.1697,
      "loss/crossentropy": 2.311215043067932,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22553718090057373,
      "step": 14788
    },
    {
      "epoch": 0.2958,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.09251708984375,
      "learning_rate": 0.0001,
      "loss": 4.3291,
      "loss/crossentropy": 2.0282764434814453,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20748327672481537,
      "step": 14790
    },
    {
      "epoch": 0.29584,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.09575093587239583,
      "learning_rate": 0.0001,
      "loss": 3.764,
      "loss/crossentropy": 2.04198157787323,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19687633216381073,
      "step": 14792
    },
    {
      "epoch": 0.29588,
      "grad_norm": 2.125,
      "grad_norm_var": 0.09601236979166666,
      "learning_rate": 0.0001,
      "loss": 4.389,
      "loss/crossentropy": 2.049328565597534,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20265275985002518,
      "step": 14794
    },
    {
      "epoch": 0.29592,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0992876688639323,
      "learning_rate": 0.0001,
      "loss": 3.9118,
      "loss/crossentropy": 1.905173659324646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.193840891122818,
      "step": 14796
    },
    {
      "epoch": 0.29596,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.09698486328125,
      "learning_rate": 0.0001,
      "loss": 3.941,
      "loss/crossentropy": 1.8715303540229797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18632445484399796,
      "step": 14798
    },
    {
      "epoch": 0.296,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.013765207926432292,
      "learning_rate": 0.0001,
      "loss": 4.039,
      "loss/crossentropy": 2.0428889989852905,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19538581371307373,
      "step": 14800
    },
    {
      "epoch": 0.29604,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.015553538004557292,
      "learning_rate": 0.0001,
      "loss": 4.0834,
      "loss/crossentropy": 2.2959046363830566,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2372349202632904,
      "step": 14802
    },
    {
      "epoch": 0.29608,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.016190592447916666,
      "learning_rate": 0.0001,
      "loss": 4.0399,
      "loss/crossentropy": 2.083451807498932,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20915375649929047,
      "step": 14804
    },
    {
      "epoch": 0.29612,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0132232666015625,
      "learning_rate": 0.0001,
      "loss": 4.293,
      "loss/crossentropy": 2.31546950340271,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20648062229156494,
      "step": 14806
    },
    {
      "epoch": 0.29616,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.013019816080729166,
      "learning_rate": 0.0001,
      "loss": 3.7273,
      "loss/crossentropy": 1.9833638072013855,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19132772833108902,
      "step": 14808
    },
    {
      "epoch": 0.2962,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.013655344645182291,
      "learning_rate": 0.0001,
      "loss": 3.8247,
      "loss/crossentropy": 1.784917414188385,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19032159447669983,
      "step": 14810
    },
    {
      "epoch": 0.29624,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.011263020833333333,
      "learning_rate": 0.0001,
      "loss": 4.0011,
      "loss/crossentropy": 1.843966782093048,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22127597779035568,
      "step": 14812
    },
    {
      "epoch": 0.29628,
      "grad_norm": 2.921875,
      "grad_norm_var": 0.06729100545247396,
      "learning_rate": 0.0001,
      "loss": 4.0535,
      "loss/crossentropy": 1.8107360005378723,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18275444209575653,
      "step": 14814
    },
    {
      "epoch": 0.29632,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.06665827433268229,
      "learning_rate": 0.0001,
      "loss": 4.2711,
      "loss/crossentropy": 2.147618055343628,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23800157755613327,
      "step": 14816
    },
    {
      "epoch": 0.29636,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.06475804646809896,
      "learning_rate": 0.0001,
      "loss": 4.267,
      "loss/crossentropy": 2.4430278539657593,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2401380091905594,
      "step": 14818
    },
    {
      "epoch": 0.2964,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.06377665201822917,
      "learning_rate": 0.0001,
      "loss": 4.3204,
      "loss/crossentropy": 2.181501626968384,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22491848468780518,
      "step": 14820
    },
    {
      "epoch": 0.29644,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.06398111979166667,
      "learning_rate": 0.0001,
      "loss": 4.0375,
      "loss/crossentropy": 2.101313889026642,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21598663926124573,
      "step": 14822
    },
    {
      "epoch": 0.29648,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.059020741780598955,
      "learning_rate": 0.0001,
      "loss": 3.9629,
      "loss/crossentropy": 1.6834549307823181,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19430063664913177,
      "step": 14824
    },
    {
      "epoch": 0.29652,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0566162109375,
      "learning_rate": 0.0001,
      "loss": 4.0087,
      "loss/crossentropy": 1.8400229215621948,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18764454871416092,
      "step": 14826
    },
    {
      "epoch": 0.29656,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.05609944661458333,
      "learning_rate": 0.0001,
      "loss": 4.1427,
      "loss/crossentropy": 1.9417667388916016,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21304062008857727,
      "step": 14828
    },
    {
      "epoch": 0.2966,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.003639475504557292,
      "learning_rate": 0.0001,
      "loss": 4.125,
      "loss/crossentropy": 2.227192521095276,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23011230677366257,
      "step": 14830
    },
    {
      "epoch": 0.29664,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0029436747233072915,
      "learning_rate": 0.0001,
      "loss": 4.3496,
      "loss/crossentropy": 2.2136365175247192,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21694285422563553,
      "step": 14832
    },
    {
      "epoch": 0.29668,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.004137929280598958,
      "learning_rate": 0.0001,
      "loss": 3.9464,
      "loss/crossentropy": 1.9179469347000122,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1996636986732483,
      "step": 14834
    },
    {
      "epoch": 0.29672,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0043718973795572914,
      "learning_rate": 0.0001,
      "loss": 3.9001,
      "loss/crossentropy": 1.9639176726341248,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17738713324069977,
      "step": 14836
    },
    {
      "epoch": 0.29676,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.004564412434895833,
      "learning_rate": 0.0001,
      "loss": 4.2665,
      "loss/crossentropy": 1.7748695611953735,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18663641810417175,
      "step": 14838
    },
    {
      "epoch": 0.2968,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.007054646809895833,
      "learning_rate": 0.0001,
      "loss": 3.9879,
      "loss/crossentropy": 2.094521999359131,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18323734402656555,
      "step": 14840
    },
    {
      "epoch": 0.29684,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.01407470703125,
      "learning_rate": 0.0001,
      "loss": 4.5969,
      "loss/crossentropy": 2.488257050514221,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23173442482948303,
      "step": 14842
    },
    {
      "epoch": 0.29688,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.013444010416666667,
      "learning_rate": 0.0001,
      "loss": 4.2015,
      "loss/crossentropy": 2.2317086458206177,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2046324908733368,
      "step": 14844
    },
    {
      "epoch": 0.29692,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.014134724934895834,
      "learning_rate": 0.0001,
      "loss": 3.9917,
      "loss/crossentropy": 2.275767207145691,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20739831775426865,
      "step": 14846
    },
    {
      "epoch": 0.29696,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.014501698811848958,
      "learning_rate": 0.0001,
      "loss": 3.9728,
      "loss/crossentropy": 2.0288257002830505,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18860980868339539,
      "step": 14848
    },
    {
      "epoch": 0.297,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.015449778238932291,
      "learning_rate": 0.0001,
      "loss": 4.3509,
      "loss/crossentropy": 2.0609896183013916,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20585624873638153,
      "step": 14850
    },
    {
      "epoch": 0.29704,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.018214670817057292,
      "learning_rate": 0.0001,
      "loss": 3.5485,
      "loss/crossentropy": 1.9250158667564392,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1843906193971634,
      "step": 14852
    },
    {
      "epoch": 0.29708,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.018070475260416666,
      "learning_rate": 0.0001,
      "loss": 3.9761,
      "loss/crossentropy": 2.118358612060547,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2035902589559555,
      "step": 14854
    },
    {
      "epoch": 0.29712,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.015610504150390624,
      "learning_rate": 0.0001,
      "loss": 4.135,
      "loss/crossentropy": 1.8381852507591248,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1859317123889923,
      "step": 14856
    },
    {
      "epoch": 0.29716,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.009513346354166667,
      "learning_rate": 0.0001,
      "loss": 4.1235,
      "loss/crossentropy": 2.078063726425171,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21035100519657135,
      "step": 14858
    },
    {
      "epoch": 0.2972,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.010701497395833334,
      "learning_rate": 0.0001,
      "loss": 4.0549,
      "loss/crossentropy": 2.1546566486358643,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19998939335346222,
      "step": 14860
    },
    {
      "epoch": 0.29724,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.010164133707682292,
      "learning_rate": 0.0001,
      "loss": 4.2264,
      "loss/crossentropy": 2.019156754016876,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1957200989127159,
      "step": 14862
    },
    {
      "epoch": 0.29728,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.013138834635416667,
      "learning_rate": 0.0001,
      "loss": 4.0808,
      "loss/crossentropy": 1.8542814254760742,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19331540167331696,
      "step": 14864
    },
    {
      "epoch": 0.29732,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.010895792643229167,
      "learning_rate": 0.0001,
      "loss": 4.2488,
      "loss/crossentropy": 2.241651773452759,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22117780148983002,
      "step": 14866
    },
    {
      "epoch": 0.29736,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.011226145426432292,
      "learning_rate": 0.0001,
      "loss": 4.2139,
      "loss/crossentropy": 1.8856277465820312,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19262754172086716,
      "step": 14868
    },
    {
      "epoch": 0.2974,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.013388824462890626,
      "learning_rate": 0.0001,
      "loss": 4.3315,
      "loss/crossentropy": 2.0159581899642944,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21892660111188889,
      "step": 14870
    },
    {
      "epoch": 0.29744,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.014562733968098958,
      "learning_rate": 0.0001,
      "loss": 3.8965,
      "loss/crossentropy": 1.8794240355491638,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18827088177204132,
      "step": 14872
    },
    {
      "epoch": 0.29748,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.013726552327473959,
      "learning_rate": 0.0001,
      "loss": 4.2176,
      "loss/crossentropy": 2.3741602897644043,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2292410209774971,
      "step": 14874
    },
    {
      "epoch": 0.29752,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.014025624593098958,
      "learning_rate": 0.0001,
      "loss": 3.8625,
      "loss/crossentropy": 1.9994327425956726,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21266867220401764,
      "step": 14876
    },
    {
      "epoch": 0.29756,
      "grad_norm": 2.125,
      "grad_norm_var": 0.014143880208333333,
      "learning_rate": 0.0001,
      "loss": 4.1337,
      "loss/crossentropy": 2.0914021730422974,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2071390450000763,
      "step": 14878
    },
    {
      "epoch": 0.2976,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.012798817952473958,
      "learning_rate": 0.0001,
      "loss": 4.0463,
      "loss/crossentropy": 1.8055935502052307,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19375848025083542,
      "step": 14880
    },
    {
      "epoch": 0.29764,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.012896474202473958,
      "learning_rate": 0.0001,
      "loss": 4.3662,
      "loss/crossentropy": 2.614544630050659,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22156962752342224,
      "step": 14882
    },
    {
      "epoch": 0.29768,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.013205718994140626,
      "learning_rate": 0.0001,
      "loss": 3.6502,
      "loss/crossentropy": 1.7306513786315918,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17159338295459747,
      "step": 14884
    },
    {
      "epoch": 0.29772,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.010060373942057292,
      "learning_rate": 0.0001,
      "loss": 4.0683,
      "loss/crossentropy": 2.060949981212616,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20615212619304657,
      "step": 14886
    },
    {
      "epoch": 0.29776,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.008756510416666667,
      "learning_rate": 0.0001,
      "loss": 3.9903,
      "loss/crossentropy": 1.8338764309883118,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20274019241333008,
      "step": 14888
    },
    {
      "epoch": 0.2978,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.013683827718098958,
      "learning_rate": 0.0001,
      "loss": 4.0335,
      "loss/crossentropy": 2.2182517051696777,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22446341067552567,
      "step": 14890
    },
    {
      "epoch": 0.29784,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.012951405843098958,
      "learning_rate": 0.0001,
      "loss": 4.3924,
      "loss/crossentropy": 2.222801446914673,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22065874934196472,
      "step": 14892
    },
    {
      "epoch": 0.29788,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.012808990478515626,
      "learning_rate": 0.0001,
      "loss": 4.1315,
      "loss/crossentropy": 2.2181414365768433,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21953265368938446,
      "step": 14894
    },
    {
      "epoch": 0.29792,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.013166300455729167,
      "learning_rate": 0.0001,
      "loss": 3.654,
      "loss/crossentropy": 1.9304096102714539,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20096193999052048,
      "step": 14896
    },
    {
      "epoch": 0.29796,
      "grad_norm": 2.0,
      "grad_norm_var": 0.012043253580729166,
      "learning_rate": 0.0001,
      "loss": 4.2074,
      "loss/crossentropy": 2.2288341522216797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24108773469924927,
      "step": 14898
    },
    {
      "epoch": 0.298,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.012550608317057291,
      "learning_rate": 0.0001,
      "loss": 4.191,
      "loss/crossentropy": 2.1711790561676025,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2395533323287964,
      "step": 14900
    },
    {
      "epoch": 0.29804,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.019482167561848958,
      "learning_rate": 0.0001,
      "loss": 4.2686,
      "loss/crossentropy": 2.0533857345581055,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21801679581403732,
      "step": 14902
    },
    {
      "epoch": 0.29808,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.01883519490559896,
      "learning_rate": 0.0001,
      "loss": 3.8678,
      "loss/crossentropy": 1.634634256362915,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17861492931842804,
      "step": 14904
    },
    {
      "epoch": 0.29812,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.0149322509765625,
      "learning_rate": 0.0001,
      "loss": 4.2152,
      "loss/crossentropy": 2.3151358366012573,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22694342583417892,
      "step": 14906
    },
    {
      "epoch": 0.29816,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.01585871378580729,
      "learning_rate": 0.0001,
      "loss": 4.085,
      "loss/crossentropy": 1.9165552258491516,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18231045454740524,
      "step": 14908
    },
    {
      "epoch": 0.2982,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.01480712890625,
      "learning_rate": 0.0001,
      "loss": 3.9693,
      "loss/crossentropy": 2.0094637274742126,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19547566026449203,
      "step": 14910
    },
    {
      "epoch": 0.29824,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.014753214518229167,
      "learning_rate": 0.0001,
      "loss": 3.9091,
      "loss/crossentropy": 1.8764418959617615,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18009892851114273,
      "step": 14912
    },
    {
      "epoch": 0.29828,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.015868123372395834,
      "learning_rate": 0.0001,
      "loss": 4.429,
      "loss/crossentropy": 2.1255921721458435,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20359216630458832,
      "step": 14914
    },
    {
      "epoch": 0.29832,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.014891560872395833,
      "learning_rate": 0.0001,
      "loss": 3.7889,
      "loss/crossentropy": 2.1316330432891846,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1970125362277031,
      "step": 14916
    },
    {
      "epoch": 0.29836,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0089996337890625,
      "learning_rate": 0.0001,
      "loss": 4.0305,
      "loss/crossentropy": 1.7825700640678406,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1902567446231842,
      "step": 14918
    },
    {
      "epoch": 0.2984,
      "grad_norm": 2.0,
      "grad_norm_var": 0.008565012613932292,
      "learning_rate": 0.0001,
      "loss": 3.9577,
      "loss/crossentropy": 1.982073962688446,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20216460525989532,
      "step": 14920
    },
    {
      "epoch": 0.29844,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0060943603515625,
      "learning_rate": 0.0001,
      "loss": 4.04,
      "loss/crossentropy": 2.3210418224334717,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2224876508116722,
      "step": 14922
    },
    {
      "epoch": 0.29848,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.005535634358723959,
      "learning_rate": 0.0001,
      "loss": 4.0509,
      "loss/crossentropy": 1.7990338206291199,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17401983588933945,
      "step": 14924
    },
    {
      "epoch": 0.29852,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.010545857747395833,
      "learning_rate": 0.0001,
      "loss": 4.4979,
      "loss/crossentropy": 1.8772737979888916,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21376933157444,
      "step": 14926
    },
    {
      "epoch": 0.29856,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.009716542561848958,
      "learning_rate": 0.0001,
      "loss": 4.0544,
      "loss/crossentropy": 1.6914128065109253,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17860179394483566,
      "step": 14928
    },
    {
      "epoch": 0.2986,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.009691365559895833,
      "learning_rate": 0.0001,
      "loss": 3.9737,
      "loss/crossentropy": 1.9545430541038513,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2015315517783165,
      "step": 14930
    },
    {
      "epoch": 0.29864,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.009175618489583334,
      "learning_rate": 0.0001,
      "loss": 3.9181,
      "loss/crossentropy": 2.0585074424743652,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2029297649860382,
      "step": 14932
    },
    {
      "epoch": 0.29868,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.009679921468098958,
      "learning_rate": 0.0001,
      "loss": 3.8638,
      "loss/crossentropy": 1.9196518063545227,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1970217153429985,
      "step": 14934
    },
    {
      "epoch": 0.29872,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0097564697265625,
      "learning_rate": 0.0001,
      "loss": 4.1366,
      "loss/crossentropy": 2.1077409982681274,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22366851568222046,
      "step": 14936
    },
    {
      "epoch": 0.29876,
      "grad_norm": 3.921875,
      "grad_norm_var": 0.24059829711914063,
      "learning_rate": 0.0001,
      "loss": 3.8552,
      "loss/crossentropy": 1.7660444974899292,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23386957496404648,
      "step": 14938
    },
    {
      "epoch": 0.2988,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.23871841430664062,
      "learning_rate": 0.0001,
      "loss": 4.2936,
      "loss/crossentropy": 1.8436493873596191,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19589588046073914,
      "step": 14940
    },
    {
      "epoch": 0.29884,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.24443333943684895,
      "learning_rate": 0.0001,
      "loss": 4.3465,
      "loss/crossentropy": 2.2123888731002808,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26607823371887207,
      "step": 14942
    },
    {
      "epoch": 0.29888,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.2546119689941406,
      "learning_rate": 0.0001,
      "loss": 4.3234,
      "loss/crossentropy": 2.70485258102417,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2529008686542511,
      "step": 14944
    },
    {
      "epoch": 0.29892,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.25155843098958336,
      "learning_rate": 0.0001,
      "loss": 4.1379,
      "loss/crossentropy": 1.932865023612976,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22141847014427185,
      "step": 14946
    },
    {
      "epoch": 0.29896,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.25643310546875,
      "learning_rate": 0.0001,
      "loss": 4.0119,
      "loss/crossentropy": 2.213751196861267,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21382954716682434,
      "step": 14948
    },
    {
      "epoch": 0.299,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.251073964436849,
      "learning_rate": 0.0001,
      "loss": 3.9673,
      "loss/crossentropy": 1.679059088230133,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16484958678483963,
      "step": 14950
    },
    {
      "epoch": 0.29904,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.24972508748372396,
      "learning_rate": 0.0001,
      "loss": 3.9873,
      "loss/crossentropy": 1.748594582080841,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20150119066238403,
      "step": 14952
    },
    {
      "epoch": 0.29908,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.05215657552083333,
      "learning_rate": 0.0001,
      "loss": 4.8053,
      "loss/crossentropy": 2.4588215351104736,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2513396218419075,
      "step": 14954
    },
    {
      "epoch": 0.29912,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.05468114217122396,
      "learning_rate": 0.0001,
      "loss": 4.0782,
      "loss/crossentropy": 2.1104516983032227,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20738063752651215,
      "step": 14956
    },
    {
      "epoch": 0.29916,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.04905776977539063,
      "learning_rate": 0.0001,
      "loss": 4.2351,
      "loss/crossentropy": 1.7171857953071594,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27423302084207535,
      "step": 14958
    },
    {
      "epoch": 0.2992,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.03264058430989583,
      "learning_rate": 0.0001,
      "loss": 3.7507,
      "loss/crossentropy": 1.5607159733772278,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1616373285651207,
      "step": 14960
    },
    {
      "epoch": 0.29924,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.033394114176432295,
      "learning_rate": 0.0001,
      "loss": 4.0128,
      "loss/crossentropy": 1.9257365465164185,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18031761050224304,
      "step": 14962
    },
    {
      "epoch": 0.29928,
      "grad_norm": 2.0,
      "grad_norm_var": 0.031062825520833334,
      "learning_rate": 0.0001,
      "loss": 3.9406,
      "loss/crossentropy": 1.876187801361084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19531002640724182,
      "step": 14964
    },
    {
      "epoch": 0.29932,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0317047119140625,
      "learning_rate": 0.0001,
      "loss": 4.2028,
      "loss/crossentropy": 1.9325042963027954,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19496598094701767,
      "step": 14966
    },
    {
      "epoch": 0.29936,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.03322652180989583,
      "learning_rate": 0.0001,
      "loss": 4.1293,
      "loss/crossentropy": 2.148952007293701,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18833627551794052,
      "step": 14968
    },
    {
      "epoch": 0.2994,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008099110921223958,
      "learning_rate": 0.0001,
      "loss": 4.0954,
      "loss/crossentropy": 2.3639365434646606,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21851496398448944,
      "step": 14970
    },
    {
      "epoch": 0.29944,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.008371734619140625,
      "learning_rate": 0.0001,
      "loss": 3.8026,
      "loss/crossentropy": 1.941165030002594,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19548548012971878,
      "step": 14972
    },
    {
      "epoch": 0.29948,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0082916259765625,
      "learning_rate": 0.0001,
      "loss": 4.1203,
      "loss/crossentropy": 2.030466139316559,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1922580599784851,
      "step": 14974
    },
    {
      "epoch": 0.29952,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.008487955729166666,
      "learning_rate": 0.0001,
      "loss": 4.2854,
      "loss/crossentropy": 2.2279865741729736,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20831073075532913,
      "step": 14976
    },
    {
      "epoch": 0.29956,
      "grad_norm": 1.875,
      "grad_norm_var": 0.007047271728515625,
      "learning_rate": 0.0001,
      "loss": 4.1708,
      "loss/crossentropy": 2.0749244689941406,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18032152205705643,
      "step": 14978
    },
    {
      "epoch": 0.2996,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.04270731608072917,
      "learning_rate": 0.0001,
      "loss": 4.199,
      "loss/crossentropy": 2.3190104961395264,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24735668301582336,
      "step": 14980
    },
    {
      "epoch": 0.29964,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.044406890869140625,
      "learning_rate": 0.0001,
      "loss": 3.9549,
      "loss/crossentropy": 1.970013439655304,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19323202222585678,
      "step": 14982
    },
    {
      "epoch": 0.29968,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.045169830322265625,
      "learning_rate": 0.0001,
      "loss": 4.4762,
      "loss/crossentropy": 2.052343726158142,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20909954607486725,
      "step": 14984
    },
    {
      "epoch": 0.29972,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.043268839518229164,
      "learning_rate": 0.0001,
      "loss": 4.1776,
      "loss/crossentropy": 2.2586673498153687,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22640614211559296,
      "step": 14986
    },
    {
      "epoch": 0.29976,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.042557779947916666,
      "learning_rate": 0.0001,
      "loss": 3.8465,
      "loss/crossentropy": 1.8843002319335938,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19117384403944016,
      "step": 14988
    },
    {
      "epoch": 0.2998,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.042740885416666666,
      "learning_rate": 0.0001,
      "loss": 3.7343,
      "loss/crossentropy": 1.542019009590149,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18295453488826752,
      "step": 14990
    },
    {
      "epoch": 0.29984,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.043369293212890625,
      "learning_rate": 0.0001,
      "loss": 4.1335,
      "loss/crossentropy": 2.160573959350586,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22916415333747864,
      "step": 14992
    },
    {
      "epoch": 0.29988,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.062168121337890625,
      "learning_rate": 0.0001,
      "loss": 4.1898,
      "loss/crossentropy": 2.1296870708465576,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20498204976320267,
      "step": 14994
    },
    {
      "epoch": 0.29992,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.0336669921875,
      "learning_rate": 0.0001,
      "loss": 4.0554,
      "loss/crossentropy": 2.0356279015541077,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21798957884311676,
      "step": 14996
    },
    {
      "epoch": 0.29996,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.035796864827473955,
      "learning_rate": 0.0001,
      "loss": 4.6195,
      "loss/crossentropy": 2.466023027896881,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22107623517513275,
      "step": 14998
    },
    {
      "epoch": 0.3,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.03408203125,
      "learning_rate": 0.0001,
      "loss": 4.1455,
      "loss/crossentropy": 1.8457531332969666,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21306351572275162,
      "step": 15000
    },
    {
      "epoch": 0.30004,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.034211222330729166,
      "learning_rate": 0.0001,
      "loss": 4.5128,
      "loss/crossentropy": 2.202435612678528,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21542339771986008,
      "step": 15002
    },
    {
      "epoch": 0.30008,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.032613118489583336,
      "learning_rate": 0.0001,
      "loss": 4.1489,
      "loss/crossentropy": 2.142255425453186,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21016352623701096,
      "step": 15004
    },
    {
      "epoch": 0.30012,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.030631256103515626,
      "learning_rate": 0.0001,
      "loss": 4.1443,
      "loss/crossentropy": 1.949910044670105,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19256845861673355,
      "step": 15006
    },
    {
      "epoch": 0.30016,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.029750315348307292,
      "learning_rate": 0.0001,
      "loss": 4.34,
      "loss/crossentropy": 2.257867217063904,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22471557557582855,
      "step": 15008
    },
    {
      "epoch": 0.3002,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.011358388264973958,
      "learning_rate": 0.0001,
      "loss": 4.3457,
      "loss/crossentropy": 2.1541898250579834,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23054643720388412,
      "step": 15010
    },
    {
      "epoch": 0.30024,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.01165771484375,
      "learning_rate": 0.0001,
      "loss": 4.2537,
      "loss/crossentropy": 2.0029123425483704,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21815954893827438,
      "step": 15012
    },
    {
      "epoch": 0.30028,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0076416015625,
      "learning_rate": 0.0001,
      "loss": 4.0093,
      "loss/crossentropy": 2.0617172718048096,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19382908195257187,
      "step": 15014
    },
    {
      "epoch": 0.30032,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.00921630859375,
      "learning_rate": 0.0001,
      "loss": 4.3123,
      "loss/crossentropy": 2.1493303775787354,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21762489527463913,
      "step": 15016
    },
    {
      "epoch": 0.30036,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0093505859375,
      "learning_rate": 0.0001,
      "loss": 4.1282,
      "loss/crossentropy": 2.081854462623596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22617685049772263,
      "step": 15018
    },
    {
      "epoch": 0.3004,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.009842681884765624,
      "learning_rate": 0.0001,
      "loss": 4.0352,
      "loss/crossentropy": 1.8565402626991272,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18223749846220016,
      "step": 15020
    },
    {
      "epoch": 0.30044,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.010815175374348958,
      "learning_rate": 0.0001,
      "loss": 4.1779,
      "loss/crossentropy": 2.129835605621338,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21263620257377625,
      "step": 15022
    },
    {
      "epoch": 0.30048,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.011189524332682292,
      "learning_rate": 0.0001,
      "loss": 4.5131,
      "loss/crossentropy": 2.113425612449646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22930295765399933,
      "step": 15024
    },
    {
      "epoch": 0.30052,
      "grad_norm": 2.0,
      "grad_norm_var": 0.010109202067057291,
      "learning_rate": 0.0001,
      "loss": 4.1557,
      "loss/crossentropy": 2.254178762435913,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21644078195095062,
      "step": 15026
    },
    {
      "epoch": 0.30056,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.011966959635416666,
      "learning_rate": 0.0001,
      "loss": 4.0863,
      "loss/crossentropy": 2.233784854412079,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22132302820682526,
      "step": 15028
    },
    {
      "epoch": 0.3006,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0118804931640625,
      "learning_rate": 0.0001,
      "loss": 3.8657,
      "loss/crossentropy": 2.2234549522399902,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2144894078373909,
      "step": 15030
    },
    {
      "epoch": 0.30064,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0108306884765625,
      "learning_rate": 0.0001,
      "loss": 4.1315,
      "loss/crossentropy": 2.0401915907859802,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20804419368505478,
      "step": 15032
    },
    {
      "epoch": 0.30068,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.010978190104166667,
      "learning_rate": 0.0001,
      "loss": 4.0939,
      "loss/crossentropy": 2.058899462223053,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2222386747598648,
      "step": 15034
    },
    {
      "epoch": 0.30072,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.020694986979166666,
      "learning_rate": 0.0001,
      "loss": 4.5452,
      "loss/crossentropy": 2.396425485610962,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24734684079885483,
      "step": 15036
    },
    {
      "epoch": 0.30076,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.02011693318684896,
      "learning_rate": 0.0001,
      "loss": 4.1124,
      "loss/crossentropy": 1.9825797080993652,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20504453778266907,
      "step": 15038
    },
    {
      "epoch": 0.3008,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.02068049112955729,
      "learning_rate": 0.0001,
      "loss": 4.0287,
      "loss/crossentropy": 1.9550088047981262,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2051691859960556,
      "step": 15040
    },
    {
      "epoch": 0.30084,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.02102839152018229,
      "learning_rate": 0.0001,
      "loss": 4.1926,
      "loss/crossentropy": 2.166532874107361,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20486898720264435,
      "step": 15042
    },
    {
      "epoch": 0.30088,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0158355712890625,
      "learning_rate": 0.0001,
      "loss": 3.9905,
      "loss/crossentropy": 1.8752986192703247,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20286446809768677,
      "step": 15044
    },
    {
      "epoch": 0.30092,
      "grad_norm": 2.125,
      "grad_norm_var": 0.016649373372395835,
      "learning_rate": 0.0001,
      "loss": 4.3217,
      "loss/crossentropy": 1.9599428176879883,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2061413824558258,
      "step": 15046
    },
    {
      "epoch": 0.30096,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.015428670247395833,
      "learning_rate": 0.0001,
      "loss": 4.2914,
      "loss/crossentropy": 2.5601563453674316,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2123739868402481,
      "step": 15048
    },
    {
      "epoch": 0.301,
      "grad_norm": 2.0,
      "grad_norm_var": 0.015062459309895833,
      "learning_rate": 0.0001,
      "loss": 4.2784,
      "loss/crossentropy": 2.075629949569702,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20791151374578476,
      "step": 15050
    },
    {
      "epoch": 0.30104,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.008532460530598958,
      "learning_rate": 0.0001,
      "loss": 4.1578,
      "loss/crossentropy": 1.8346505165100098,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16620153188705444,
      "step": 15052
    },
    {
      "epoch": 0.30108,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.009382120768229167,
      "learning_rate": 0.0001,
      "loss": 4.003,
      "loss/crossentropy": 1.9715532660484314,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20792889595031738,
      "step": 15054
    },
    {
      "epoch": 0.30112,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0090728759765625,
      "learning_rate": 0.0001,
      "loss": 4.06,
      "loss/crossentropy": 2.135376811027527,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.218796044588089,
      "step": 15056
    },
    {
      "epoch": 0.30116,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.00941162109375,
      "learning_rate": 0.0001,
      "loss": 4.4339,
      "loss/crossentropy": 2.095982074737549,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2120692878961563,
      "step": 15058
    },
    {
      "epoch": 0.3012,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.00911865234375,
      "learning_rate": 0.0001,
      "loss": 4.3395,
      "loss/crossentropy": 2.2836159467697144,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23748520761728287,
      "step": 15060
    },
    {
      "epoch": 0.30124,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.010163370768229167,
      "learning_rate": 0.0001,
      "loss": 3.8877,
      "loss/crossentropy": 2.073283314704895,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19927848130464554,
      "step": 15062
    },
    {
      "epoch": 0.30128,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010196940104166666,
      "learning_rate": 0.0001,
      "loss": 4.1908,
      "loss/crossentropy": 2.2401102781295776,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22298122942447662,
      "step": 15064
    },
    {
      "epoch": 0.30132,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.012650299072265624,
      "learning_rate": 0.0001,
      "loss": 4.1878,
      "loss/crossentropy": 2.063979387283325,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19714537262916565,
      "step": 15066
    },
    {
      "epoch": 0.30136,
      "grad_norm": 2.0,
      "grad_norm_var": 0.012894694010416667,
      "learning_rate": 0.0001,
      "loss": 3.6407,
      "loss/crossentropy": 2.0346400141716003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18405266851186752,
      "step": 15068
    },
    {
      "epoch": 0.3014,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.013437652587890625,
      "learning_rate": 0.0001,
      "loss": 3.8641,
      "loss/crossentropy": 2.1211341619491577,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19659722596406937,
      "step": 15070
    },
    {
      "epoch": 0.30144,
      "grad_norm": 3.484375,
      "grad_norm_var": 0.1490966796875,
      "learning_rate": 0.0001,
      "loss": 3.9967,
      "loss/crossentropy": 1.937787652015686,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18829380720853806,
      "step": 15072
    },
    {
      "epoch": 0.30148,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.1507219950358073,
      "learning_rate": 0.0001,
      "loss": 3.922,
      "loss/crossentropy": 1.9047453999519348,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18311911821365356,
      "step": 15074
    },
    {
      "epoch": 0.30152,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.15110651652018228,
      "learning_rate": 0.0001,
      "loss": 4.3423,
      "loss/crossentropy": 2.1409407258033752,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21341674029827118,
      "step": 15076
    },
    {
      "epoch": 0.30156,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.15178197224934895,
      "learning_rate": 0.0001,
      "loss": 4.2815,
      "loss/crossentropy": 2.188783288002014,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2406207174062729,
      "step": 15078
    },
    {
      "epoch": 0.3016,
      "grad_norm": 2.125,
      "grad_norm_var": 0.15162938435872395,
      "learning_rate": 0.0001,
      "loss": 4.368,
      "loss/crossentropy": 2.088913679122925,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2055455967783928,
      "step": 15080
    },
    {
      "epoch": 0.30164,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.15350316365559896,
      "learning_rate": 0.0001,
      "loss": 3.9039,
      "loss/crossentropy": 1.9477434754371643,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19234874844551086,
      "step": 15082
    },
    {
      "epoch": 0.30168,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.1490069071451823,
      "learning_rate": 0.0001,
      "loss": 4.2653,
      "loss/crossentropy": 2.2611928582191467,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2250279188156128,
      "step": 15084
    },
    {
      "epoch": 0.30172,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.14735514322916668,
      "learning_rate": 0.0001,
      "loss": 4.043,
      "loss/crossentropy": 2.340996742248535,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2201436087489128,
      "step": 15086
    },
    {
      "epoch": 0.30176,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.01718724568684896,
      "learning_rate": 0.0001,
      "loss": 3.9232,
      "loss/crossentropy": 2.045587956905365,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2049652263522148,
      "step": 15088
    },
    {
      "epoch": 0.3018,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.015265909830729167,
      "learning_rate": 0.0001,
      "loss": 3.9266,
      "loss/crossentropy": 1.6079826951026917,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1584620103240013,
      "step": 15090
    },
    {
      "epoch": 0.30184,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.013494618733723958,
      "learning_rate": 0.0001,
      "loss": 4.1049,
      "loss/crossentropy": 2.122701048851013,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2095094472169876,
      "step": 15092
    },
    {
      "epoch": 0.30188,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.008243560791015625,
      "learning_rate": 0.0001,
      "loss": 3.8602,
      "loss/crossentropy": 1.8888922929763794,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19282665848731995,
      "step": 15094
    },
    {
      "epoch": 0.30192,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0063873291015625,
      "learning_rate": 0.0001,
      "loss": 4.1279,
      "loss/crossentropy": 2.121204972267151,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1894393339753151,
      "step": 15096
    },
    {
      "epoch": 0.30196,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.005991363525390625,
      "learning_rate": 0.0001,
      "loss": 4.0433,
      "loss/crossentropy": 1.836807131767273,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1851387470960617,
      "step": 15098
    },
    {
      "epoch": 0.302,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.005246734619140625,
      "learning_rate": 0.0001,
      "loss": 4.1723,
      "loss/crossentropy": 2.170082688331604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20964065939188004,
      "step": 15100
    },
    {
      "epoch": 0.30204,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0037595113118489582,
      "learning_rate": 0.0001,
      "loss": 4.0642,
      "loss/crossentropy": 2.128955125808716,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20737067610025406,
      "step": 15102
    },
    {
      "epoch": 0.30208,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0062744140625,
      "learning_rate": 0.0001,
      "loss": 4.1561,
      "loss/crossentropy": 1.970711886882782,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19101983308792114,
      "step": 15104
    },
    {
      "epoch": 0.30212,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.018040974934895832,
      "learning_rate": 0.0001,
      "loss": 4.5089,
      "loss/crossentropy": 2.105665922164917,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23741377890110016,
      "step": 15106
    },
    {
      "epoch": 0.30216,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.01962865193684896,
      "learning_rate": 0.0001,
      "loss": 4.3766,
      "loss/crossentropy": 2.274402379989624,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20633316040039062,
      "step": 15108
    },
    {
      "epoch": 0.3022,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.017996978759765626,
      "learning_rate": 0.0001,
      "loss": 4.0393,
      "loss/crossentropy": 2.134031891822815,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20932414382696152,
      "step": 15110
    },
    {
      "epoch": 0.30224,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.01930720011393229,
      "learning_rate": 0.0001,
      "loss": 3.9084,
      "loss/crossentropy": 2.015448212623596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18043135851621628,
      "step": 15112
    },
    {
      "epoch": 0.30228,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.018232981363932293,
      "learning_rate": 0.0001,
      "loss": 4.2025,
      "loss/crossentropy": 2.3585838079452515,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2323882058262825,
      "step": 15114
    },
    {
      "epoch": 0.30232,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.018607584635416667,
      "learning_rate": 0.0001,
      "loss": 3.8586,
      "loss/crossentropy": 1.790964961051941,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18150723725557327,
      "step": 15116
    },
    {
      "epoch": 0.30236,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.018802642822265625,
      "learning_rate": 0.0001,
      "loss": 4.1433,
      "loss/crossentropy": 2.0206486582756042,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2102368101477623,
      "step": 15118
    },
    {
      "epoch": 0.3024,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.016169230143229168,
      "learning_rate": 0.0001,
      "loss": 4.1773,
      "loss/crossentropy": 2.21474289894104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22742661088705063,
      "step": 15120
    },
    {
      "epoch": 0.30244,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.009781901041666667,
      "learning_rate": 0.0001,
      "loss": 3.9536,
      "loss/crossentropy": 2.0452207922935486,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19045353680849075,
      "step": 15122
    },
    {
      "epoch": 0.30248,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.0105865478515625,
      "learning_rate": 0.0001,
      "loss": 3.7914,
      "loss/crossentropy": 2.019612729549408,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2032184973359108,
      "step": 15124
    },
    {
      "epoch": 0.30252,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.010221099853515625,
      "learning_rate": 0.0001,
      "loss": 4.0303,
      "loss/crossentropy": 2.1821396350860596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22832991182804108,
      "step": 15126
    },
    {
      "epoch": 0.30256,
      "grad_norm": 1.875,
      "grad_norm_var": 0.010088857014973958,
      "learning_rate": 0.0001,
      "loss": 3.7838,
      "loss/crossentropy": 2.0395348072052,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18255837261676788,
      "step": 15128
    },
    {
      "epoch": 0.3026,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.009081013997395833,
      "learning_rate": 0.0001,
      "loss": 4.177,
      "loss/crossentropy": 2.1724050045013428,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19672775268554688,
      "step": 15130
    },
    {
      "epoch": 0.30264,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.008470662434895833,
      "learning_rate": 0.0001,
      "loss": 4.2438,
      "loss/crossentropy": 2.107967436313629,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19564508646726608,
      "step": 15132
    },
    {
      "epoch": 0.30268,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.007355753580729167,
      "learning_rate": 0.0001,
      "loss": 3.8674,
      "loss/crossentropy": 1.7179370522499084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17135357856750488,
      "step": 15134
    },
    {
      "epoch": 0.30272,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.008119455973307292,
      "learning_rate": 0.0001,
      "loss": 4.104,
      "loss/crossentropy": 1.9004579186439514,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19902431219816208,
      "step": 15136
    },
    {
      "epoch": 0.30276,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0052487691243489586,
      "learning_rate": 0.0001,
      "loss": 4.013,
      "loss/crossentropy": 1.9554911255836487,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19999338686466217,
      "step": 15138
    },
    {
      "epoch": 0.3028,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.004793294270833333,
      "learning_rate": 0.0001,
      "loss": 4.1051,
      "loss/crossentropy": 1.8687474131584167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19250088185071945,
      "step": 15140
    },
    {
      "epoch": 0.30284,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.004929351806640625,
      "learning_rate": 0.0001,
      "loss": 4.1217,
      "loss/crossentropy": 2.1360538005828857,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22316065430641174,
      "step": 15142
    },
    {
      "epoch": 0.30288,
      "grad_norm": 2.125,
      "grad_norm_var": 0.006776682535807292,
      "learning_rate": 0.0001,
      "loss": 3.7681,
      "loss/crossentropy": 1.6100040078163147,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16926150023937225,
      "step": 15144
    },
    {
      "epoch": 0.30292,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0070302327473958336,
      "learning_rate": 0.0001,
      "loss": 4.0459,
      "loss/crossentropy": 1.983083188533783,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19075944274663925,
      "step": 15146
    },
    {
      "epoch": 0.30296,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.009110514322916667,
      "learning_rate": 0.0001,
      "loss": 3.8722,
      "loss/crossentropy": 1.9027757048606873,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2069152221083641,
      "step": 15148
    },
    {
      "epoch": 0.303,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.009144846598307292,
      "learning_rate": 0.0001,
      "loss": 3.996,
      "loss/crossentropy": 1.9141475558280945,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19106873124837875,
      "step": 15150
    },
    {
      "epoch": 0.30304,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.009821573893229166,
      "learning_rate": 0.0001,
      "loss": 4.0872,
      "loss/crossentropy": 1.924220085144043,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19870451837778091,
      "step": 15152
    },
    {
      "epoch": 0.30308,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.009169260660807291,
      "learning_rate": 0.0001,
      "loss": 4.1611,
      "loss/crossentropy": 2.18874990940094,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2040283977985382,
      "step": 15154
    },
    {
      "epoch": 0.30312,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0096588134765625,
      "learning_rate": 0.0001,
      "loss": 3.6905,
      "loss/crossentropy": 2.0296109914779663,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20384181290864944,
      "step": 15156
    },
    {
      "epoch": 0.30316,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0091949462890625,
      "learning_rate": 0.0001,
      "loss": 4.2475,
      "loss/crossentropy": 2.1908326148986816,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20775151252746582,
      "step": 15158
    },
    {
      "epoch": 0.3032,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.007523345947265625,
      "learning_rate": 0.0001,
      "loss": 4.191,
      "loss/crossentropy": 2.1348751187324524,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21342730522155762,
      "step": 15160
    },
    {
      "epoch": 0.30324,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0075103759765625,
      "learning_rate": 0.0001,
      "loss": 3.9642,
      "loss/crossentropy": 2.03944593667984,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1934272050857544,
      "step": 15162
    },
    {
      "epoch": 0.30328,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.005475870768229167,
      "learning_rate": 0.0001,
      "loss": 3.9692,
      "loss/crossentropy": 1.8574647307395935,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1847086101770401,
      "step": 15164
    },
    {
      "epoch": 0.30332,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.005671183268229167,
      "learning_rate": 0.0001,
      "loss": 4.3025,
      "loss/crossentropy": 2.219611406326294,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1982950195670128,
      "step": 15166
    },
    {
      "epoch": 0.30336,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.005159250895182292,
      "learning_rate": 0.0001,
      "loss": 4.1903,
      "loss/crossentropy": 2.024726688861847,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.217759907245636,
      "step": 15168
    },
    {
      "epoch": 0.3034,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.005353800455729167,
      "learning_rate": 0.0001,
      "loss": 4.2273,
      "loss/crossentropy": 2.196849226951599,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21290963143110275,
      "step": 15170
    },
    {
      "epoch": 0.30344,
      "grad_norm": 1.7421875,
      "grad_norm_var": 0.00750732421875,
      "learning_rate": 0.0001,
      "loss": 4.0096,
      "loss/crossentropy": 1.9023171067237854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19199151545763016,
      "step": 15172
    },
    {
      "epoch": 0.30348,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008644358317057291,
      "learning_rate": 0.0001,
      "loss": 4.0798,
      "loss/crossentropy": 2.1286264657974243,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20934996753931046,
      "step": 15174
    },
    {
      "epoch": 0.30352,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.008007558186848958,
      "learning_rate": 0.0001,
      "loss": 4.2621,
      "loss/crossentropy": 2.2262455224990845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2091647908091545,
      "step": 15176
    },
    {
      "epoch": 0.30356,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.04183349609375,
      "learning_rate": 0.0001,
      "loss": 4.2439,
      "loss/crossentropy": 2.4126710891723633,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22972645610570908,
      "step": 15178
    },
    {
      "epoch": 0.3036,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.042569732666015624,
      "learning_rate": 0.0001,
      "loss": 4.4268,
      "loss/crossentropy": 2.4708153009414673,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23798953741788864,
      "step": 15180
    },
    {
      "epoch": 0.30364,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.04053929646809896,
      "learning_rate": 0.0001,
      "loss": 4.1841,
      "loss/crossentropy": 1.9924429655075073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19149084389209747,
      "step": 15182
    },
    {
      "epoch": 0.30368,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.04182103474934896,
      "learning_rate": 0.0001,
      "loss": 4.1335,
      "loss/crossentropy": 1.8018346428871155,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1883997619152069,
      "step": 15184
    },
    {
      "epoch": 0.30372,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.04639867146809896,
      "learning_rate": 0.0001,
      "loss": 3.96,
      "loss/crossentropy": 2.001536011695862,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2008758783340454,
      "step": 15186
    },
    {
      "epoch": 0.30376,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.06099828084309896,
      "learning_rate": 0.0001,
      "loss": 4.2308,
      "loss/crossentropy": 1.9239726066589355,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2068573236465454,
      "step": 15188
    },
    {
      "epoch": 0.3038,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.059845987955729166,
      "learning_rate": 0.0001,
      "loss": 4.303,
      "loss/crossentropy": 2.252953827381134,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22697407007217407,
      "step": 15190
    },
    {
      "epoch": 0.30384,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0631011962890625,
      "learning_rate": 0.0001,
      "loss": 4.174,
      "loss/crossentropy": 2.135131001472473,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2088206186890602,
      "step": 15192
    },
    {
      "epoch": 0.30388,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0391754150390625,
      "learning_rate": 0.0001,
      "loss": 4.22,
      "loss/crossentropy": 2.394876003265381,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22017298638820648,
      "step": 15194
    },
    {
      "epoch": 0.30392,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.03862711588541667,
      "learning_rate": 0.0001,
      "loss": 4.3951,
      "loss/crossentropy": 2.3517301082611084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21459681540727615,
      "step": 15196
    },
    {
      "epoch": 0.30396,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.038914998372395836,
      "learning_rate": 0.0001,
      "loss": 4.3068,
      "loss/crossentropy": 1.956869900226593,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19348695874214172,
      "step": 15198
    },
    {
      "epoch": 0.304,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.045660146077473956,
      "learning_rate": 0.0001,
      "loss": 3.5481,
      "loss/crossentropy": 1.8349076509475708,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19203926622867584,
      "step": 15200
    },
    {
      "epoch": 0.30404,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.041265614827473956,
      "learning_rate": 0.0001,
      "loss": 4.1749,
      "loss/crossentropy": 2.253283977508545,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24825416505336761,
      "step": 15202
    },
    {
      "epoch": 0.30408,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0156646728515625,
      "learning_rate": 0.0001,
      "loss": 3.9585,
      "loss/crossentropy": 1.7219101190567017,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18302495777606964,
      "step": 15204
    },
    {
      "epoch": 0.30412,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.010309855143229166,
      "learning_rate": 0.0001,
      "loss": 4.4398,
      "loss/crossentropy": 2.2931004762649536,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21909544616937637,
      "step": 15206
    },
    {
      "epoch": 0.30416,
      "grad_norm": 2.125,
      "grad_norm_var": 0.009883626302083334,
      "learning_rate": 0.0001,
      "loss": 4.0707,
      "loss/crossentropy": 1.9331820011138916,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19565977901220322,
      "step": 15208
    },
    {
      "epoch": 0.3042,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008568318684895833,
      "learning_rate": 0.0001,
      "loss": 4.0936,
      "loss/crossentropy": 1.9863171577453613,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20089460909366608,
      "step": 15210
    },
    {
      "epoch": 0.30424,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.010945383707682292,
      "learning_rate": 0.0001,
      "loss": 4.3762,
      "loss/crossentropy": 2.326872229576111,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20695248246192932,
      "step": 15212
    },
    {
      "epoch": 0.30428,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.012679036458333333,
      "learning_rate": 0.0001,
      "loss": 4.1684,
      "loss/crossentropy": 2.0216678380966187,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19882218539714813,
      "step": 15214
    },
    {
      "epoch": 0.30432,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.0074503580729166664,
      "learning_rate": 0.0001,
      "loss": 4.3682,
      "loss/crossentropy": 2.170402765274048,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22156177461147308,
      "step": 15216
    },
    {
      "epoch": 0.30436,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.007608795166015625,
      "learning_rate": 0.0001,
      "loss": 4.1655,
      "loss/crossentropy": 1.9652912616729736,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20658399909734726,
      "step": 15218
    },
    {
      "epoch": 0.3044,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0090972900390625,
      "learning_rate": 0.0001,
      "loss": 4.1234,
      "loss/crossentropy": 2.4518260955810547,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20796608924865723,
      "step": 15220
    },
    {
      "epoch": 0.30444,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.009952545166015625,
      "learning_rate": 0.0001,
      "loss": 4.0179,
      "loss/crossentropy": 1.9519163370132446,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19269376248121262,
      "step": 15222
    },
    {
      "epoch": 0.30448,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.010758209228515624,
      "learning_rate": 0.0001,
      "loss": 3.9411,
      "loss/crossentropy": 2.1493905782699585,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19713661074638367,
      "step": 15224
    },
    {
      "epoch": 0.30452,
      "grad_norm": 2.5,
      "grad_norm_var": 0.026878865559895833,
      "learning_rate": 0.0001,
      "loss": 4.2124,
      "loss/crossentropy": 2.2856662273406982,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23240232467651367,
      "step": 15226
    },
    {
      "epoch": 0.30456,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0249176025390625,
      "learning_rate": 0.0001,
      "loss": 4.1723,
      "loss/crossentropy": 2.3631181716918945,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22731658816337585,
      "step": 15228
    },
    {
      "epoch": 0.3046,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.027469635009765625,
      "learning_rate": 0.0001,
      "loss": 4.0952,
      "loss/crossentropy": 2.1270273327827454,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20736727863550186,
      "step": 15230
    },
    {
      "epoch": 0.30464,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.029320271809895833,
      "learning_rate": 0.0001,
      "loss": 4.4801,
      "loss/crossentropy": 2.1448079347610474,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20350141823291779,
      "step": 15232
    },
    {
      "epoch": 0.30468,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.059576161702473956,
      "learning_rate": 0.0001,
      "loss": 4.295,
      "loss/crossentropy": 2.0582846999168396,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19898280501365662,
      "step": 15234
    },
    {
      "epoch": 0.30472,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0597076416015625,
      "learning_rate": 0.0001,
      "loss": 4.0946,
      "loss/crossentropy": 2.2282591462135315,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22299205511808395,
      "step": 15236
    },
    {
      "epoch": 0.30476,
      "grad_norm": 4.34375,
      "grad_norm_var": 0.3757484436035156,
      "learning_rate": 0.0001,
      "loss": 4.4031,
      "loss/crossentropy": 1.902436077594757,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19752992689609528,
      "step": 15238
    },
    {
      "epoch": 0.3048,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.3626177469889323,
      "learning_rate": 0.0001,
      "loss": 4.5476,
      "loss/crossentropy": 2.5741195678710938,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25522060692310333,
      "step": 15240
    },
    {
      "epoch": 0.30484,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.3576942443847656,
      "learning_rate": 0.0001,
      "loss": 4.2319,
      "loss/crossentropy": 2.261489987373352,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2379501387476921,
      "step": 15242
    },
    {
      "epoch": 0.30488,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.3581451416015625,
      "learning_rate": 0.0001,
      "loss": 4.1728,
      "loss/crossentropy": 2.3075523376464844,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21830804646015167,
      "step": 15244
    },
    {
      "epoch": 0.30492,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.36470133463541665,
      "learning_rate": 0.0001,
      "loss": 3.9585,
      "loss/crossentropy": 2.096464157104492,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2073582485318184,
      "step": 15246
    },
    {
      "epoch": 0.30496,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.41076558430989585,
      "learning_rate": 0.0001,
      "loss": 4.0121,
      "loss/crossentropy": 1.9624161124229431,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19424450397491455,
      "step": 15248
    },
    {
      "epoch": 0.305,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.38983154296875,
      "learning_rate": 0.0001,
      "loss": 4.1351,
      "loss/crossentropy": 2.0838447213172913,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22090255469083786,
      "step": 15250
    },
    {
      "epoch": 0.30504,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.379644521077474,
      "learning_rate": 0.0001,
      "loss": 4.4244,
      "loss/crossentropy": 2.349258542060852,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20451825857162476,
      "step": 15252
    },
    {
      "epoch": 0.30508,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.06938451131184896,
      "learning_rate": 0.0001,
      "loss": 4.328,
      "loss/crossentropy": 2.3996351957321167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24550051987171173,
      "step": 15254
    },
    {
      "epoch": 0.30512,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.07430191040039062,
      "learning_rate": 0.0001,
      "loss": 4.0098,
      "loss/crossentropy": 2.325650215148926,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2189955934882164,
      "step": 15256
    },
    {
      "epoch": 0.30516,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.07815348307291667,
      "learning_rate": 0.0001,
      "loss": 4.0063,
      "loss/crossentropy": 1.8568453788757324,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18639766424894333,
      "step": 15258
    },
    {
      "epoch": 0.3052,
      "grad_norm": 1.875,
      "grad_norm_var": 0.08263346354166666,
      "learning_rate": 0.0001,
      "loss": 3.8955,
      "loss/crossentropy": 1.8268811106681824,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18669788539409637,
      "step": 15260
    },
    {
      "epoch": 0.30524,
      "grad_norm": 2.125,
      "grad_norm_var": 0.08479715983072916,
      "learning_rate": 0.0001,
      "loss": 4.1124,
      "loss/crossentropy": 2.0555724501609802,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2101765051484108,
      "step": 15262
    },
    {
      "epoch": 0.30528,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.018070475260416666,
      "learning_rate": 0.0001,
      "loss": 4.1075,
      "loss/crossentropy": 2.080612599849701,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18955913186073303,
      "step": 15264
    },
    {
      "epoch": 0.30532,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0137939453125,
      "learning_rate": 0.0001,
      "loss": 4.2678,
      "loss/crossentropy": 2.352793037891388,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21857291460037231,
      "step": 15266
    },
    {
      "epoch": 0.30536,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.010434722900390625,
      "learning_rate": 0.0001,
      "loss": 4.0871,
      "loss/crossentropy": 2.0258968472480774,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18990560621023178,
      "step": 15268
    },
    {
      "epoch": 0.3054,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.010871378580729167,
      "learning_rate": 0.0001,
      "loss": 4.3126,
      "loss/crossentropy": 2.228433310985565,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20862557739019394,
      "step": 15270
    },
    {
      "epoch": 0.30544,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.012330881754557292,
      "learning_rate": 0.0001,
      "loss": 3.7731,
      "loss/crossentropy": 1.9475300312042236,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20644760876893997,
      "step": 15272
    },
    {
      "epoch": 0.30548,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.012166341145833334,
      "learning_rate": 0.0001,
      "loss": 4.1685,
      "loss/crossentropy": 2.3502081632614136,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22096982598304749,
      "step": 15274
    },
    {
      "epoch": 0.30552,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.012064615885416666,
      "learning_rate": 0.0001,
      "loss": 4.294,
      "loss/crossentropy": 2.2581464052200317,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21158118546009064,
      "step": 15276
    },
    {
      "epoch": 0.30556,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.014054361979166667,
      "learning_rate": 0.0001,
      "loss": 4.0301,
      "loss/crossentropy": 1.9583097696304321,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19379162788391113,
      "step": 15278
    },
    {
      "epoch": 0.3056,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.016527303059895835,
      "learning_rate": 0.0001,
      "loss": 4.4464,
      "loss/crossentropy": 1.988980233669281,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19495020806789398,
      "step": 15280
    },
    {
      "epoch": 0.30564,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.017350260416666666,
      "learning_rate": 0.0001,
      "loss": 4.1733,
      "loss/crossentropy": 2.1816195249557495,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2196531891822815,
      "step": 15282
    },
    {
      "epoch": 0.30568,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.015264638264973958,
      "learning_rate": 0.0001,
      "loss": 3.7994,
      "loss/crossentropy": 2.180557370185852,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2106764316558838,
      "step": 15284
    },
    {
      "epoch": 0.30572,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.011027018229166666,
      "learning_rate": 0.0001,
      "loss": 4.0655,
      "loss/crossentropy": 1.901601493358612,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18202877044677734,
      "step": 15286
    },
    {
      "epoch": 0.30576,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.019301096598307293,
      "learning_rate": 0.0001,
      "loss": 4.1989,
      "loss/crossentropy": 2.259449601173401,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23166191577911377,
      "step": 15288
    },
    {
      "epoch": 0.3058,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.020514933268229167,
      "learning_rate": 0.0001,
      "loss": 3.9866,
      "loss/crossentropy": 1.786646544933319,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19156906753778458,
      "step": 15290
    },
    {
      "epoch": 0.30584,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.02205174763997396,
      "learning_rate": 0.0001,
      "loss": 3.9258,
      "loss/crossentropy": 1.8522255420684814,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1828666776418686,
      "step": 15292
    },
    {
      "epoch": 0.30588,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.02127685546875,
      "learning_rate": 0.0001,
      "loss": 4.2543,
      "loss/crossentropy": 2.195667266845703,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21043212711811066,
      "step": 15294
    },
    {
      "epoch": 0.30592,
      "grad_norm": 2.0,
      "grad_norm_var": 0.020918528238932293,
      "learning_rate": 0.0001,
      "loss": 3.8389,
      "loss/crossentropy": 1.8748087882995605,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20358053594827652,
      "step": 15296
    },
    {
      "epoch": 0.30596,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.017536417643229166,
      "learning_rate": 0.0001,
      "loss": 4.0789,
      "loss/crossentropy": 2.2658244371414185,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2206317037343979,
      "step": 15298
    },
    {
      "epoch": 0.306,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.018778483072916668,
      "learning_rate": 0.0001,
      "loss": 4.2882,
      "loss/crossentropy": 1.7914378643035889,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20813927054405212,
      "step": 15300
    },
    {
      "epoch": 0.30604,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.01895319620768229,
      "learning_rate": 0.0001,
      "loss": 4.1428,
      "loss/crossentropy": 2.046573519706726,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20681843161582947,
      "step": 15302
    },
    {
      "epoch": 0.30608,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.005960845947265625,
      "learning_rate": 0.0001,
      "loss": 4.2538,
      "loss/crossentropy": 2.0068886280059814,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.201521098613739,
      "step": 15304
    },
    {
      "epoch": 0.30612,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.005509440104166667,
      "learning_rate": 0.0001,
      "loss": 4.0672,
      "loss/crossentropy": 2.2020740509033203,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20413114875555038,
      "step": 15306
    },
    {
      "epoch": 0.30616,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0057769775390625,
      "learning_rate": 0.0001,
      "loss": 4.3654,
      "loss/crossentropy": 2.3031119108200073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20878810435533524,
      "step": 15308
    },
    {
      "epoch": 0.3062,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.006204986572265625,
      "learning_rate": 0.0001,
      "loss": 3.7786,
      "loss/crossentropy": 1.6445855498313904,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1618219092488289,
      "step": 15310
    },
    {
      "epoch": 0.30624,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.004572550455729167,
      "learning_rate": 0.0001,
      "loss": 4.1378,
      "loss/crossentropy": 1.995127022266388,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21076080948114395,
      "step": 15312
    },
    {
      "epoch": 0.30628,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.005411783854166667,
      "learning_rate": 0.0001,
      "loss": 3.6278,
      "loss/crossentropy": 1.8340229392051697,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19418048858642578,
      "step": 15314
    },
    {
      "epoch": 0.30632,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.007106272379557291,
      "learning_rate": 0.0001,
      "loss": 3.9496,
      "loss/crossentropy": 1.713787317276001,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17608212679624557,
      "step": 15316
    },
    {
      "epoch": 0.30636,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.006967926025390625,
      "learning_rate": 0.0001,
      "loss": 4.0223,
      "loss/crossentropy": 1.7105762362480164,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1760871261358261,
      "step": 15318
    },
    {
      "epoch": 0.3064,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0071441650390625,
      "learning_rate": 0.0001,
      "loss": 4.0326,
      "loss/crossentropy": 2.337005376815796,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2222321778535843,
      "step": 15320
    },
    {
      "epoch": 0.30644,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.008868153889973958,
      "learning_rate": 0.0001,
      "loss": 4.3269,
      "loss/crossentropy": 1.8426868915557861,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1880369558930397,
      "step": 15322
    },
    {
      "epoch": 0.30648,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.009162394205729167,
      "learning_rate": 0.0001,
      "loss": 3.7798,
      "loss/crossentropy": 2.300028443336487,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22529813647270203,
      "step": 15324
    },
    {
      "epoch": 0.30652,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.009193674723307291,
      "learning_rate": 0.0001,
      "loss": 3.9838,
      "loss/crossentropy": 2.3558120727539062,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2000514343380928,
      "step": 15326
    },
    {
      "epoch": 0.30656,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.007879384358723958,
      "learning_rate": 0.0001,
      "loss": 3.6921,
      "loss/crossentropy": 1.681145966053009,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1756032332777977,
      "step": 15328
    },
    {
      "epoch": 0.3066,
      "grad_norm": 2.0,
      "grad_norm_var": 0.010298665364583333,
      "learning_rate": 0.0001,
      "loss": 4.41,
      "loss/crossentropy": 2.1566712260246277,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21067694574594498,
      "step": 15330
    },
    {
      "epoch": 0.30664,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0075347900390625,
      "learning_rate": 0.0001,
      "loss": 3.993,
      "loss/crossentropy": 2.125000774860382,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20768606662750244,
      "step": 15332
    },
    {
      "epoch": 0.30668,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.007328033447265625,
      "learning_rate": 0.0001,
      "loss": 4.0805,
      "loss/crossentropy": 2.168649673461914,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20393794029951096,
      "step": 15334
    },
    {
      "epoch": 0.30672,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.008072662353515624,
      "learning_rate": 0.0001,
      "loss": 3.9154,
      "loss/crossentropy": 1.657529592514038,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19351129233837128,
      "step": 15336
    },
    {
      "epoch": 0.30676,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.006780751546223958,
      "learning_rate": 0.0001,
      "loss": 4.2166,
      "loss/crossentropy": 2.115979850292206,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20223377645015717,
      "step": 15338
    },
    {
      "epoch": 0.3068,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.005833943684895833,
      "learning_rate": 0.0001,
      "loss": 4.1288,
      "loss/crossentropy": 1.699661135673523,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17160461097955704,
      "step": 15340
    },
    {
      "epoch": 0.30684,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.008180491129557292,
      "learning_rate": 0.0001,
      "loss": 3.8418,
      "loss/crossentropy": 2.141602098941803,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19209860265254974,
      "step": 15342
    },
    {
      "epoch": 0.30688,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.007806142171223958,
      "learning_rate": 0.0001,
      "loss": 4.3619,
      "loss/crossentropy": 2.1502809524536133,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21510163694620132,
      "step": 15344
    },
    {
      "epoch": 0.30692,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.006078847249348958,
      "learning_rate": 0.0001,
      "loss": 4.0684,
      "loss/crossentropy": 2.1354172825813293,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19959942996501923,
      "step": 15346
    },
    {
      "epoch": 0.30696,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.006095377604166666,
      "learning_rate": 0.0001,
      "loss": 4.1196,
      "loss/crossentropy": 2.0634223222732544,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2073688805103302,
      "step": 15348
    },
    {
      "epoch": 0.307,
      "grad_norm": 2.125,
      "grad_norm_var": 0.008385976155598959,
      "learning_rate": 0.0001,
      "loss": 4.1489,
      "loss/crossentropy": 1.9851733446121216,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2060396894812584,
      "step": 15350
    },
    {
      "epoch": 0.30704,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.007490793863932292,
      "learning_rate": 0.0001,
      "loss": 4.1196,
      "loss/crossentropy": 2.3790550231933594,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23205619305372238,
      "step": 15352
    },
    {
      "epoch": 0.30708,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.008552042643229167,
      "learning_rate": 0.0001,
      "loss": 3.9085,
      "loss/crossentropy": 1.943642258644104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18238085508346558,
      "step": 15354
    },
    {
      "epoch": 0.30712,
      "grad_norm": 2.0,
      "grad_norm_var": 0.008988189697265624,
      "learning_rate": 0.0001,
      "loss": 4.112,
      "loss/crossentropy": 1.9680217504501343,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18599005788564682,
      "step": 15356
    },
    {
      "epoch": 0.30716,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0077898661295572914,
      "learning_rate": 0.0001,
      "loss": 3.7402,
      "loss/crossentropy": 1.9464871287345886,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18814007937908173,
      "step": 15358
    },
    {
      "epoch": 0.3072,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.013179524739583334,
      "learning_rate": 0.0001,
      "loss": 4.0231,
      "loss/crossentropy": 2.0038596987724304,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19203800708055496,
      "step": 15360
    },
    {
      "epoch": 0.30724,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.013071441650390625,
      "learning_rate": 0.0001,
      "loss": 4.0843,
      "loss/crossentropy": 1.93240225315094,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20595912635326385,
      "step": 15362
    },
    {
      "epoch": 0.30728,
      "grad_norm": 2.0,
      "grad_norm_var": 0.012938435872395833,
      "learning_rate": 0.0001,
      "loss": 4.2042,
      "loss/crossentropy": 2.067903220653534,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21503040939569473,
      "step": 15364
    },
    {
      "epoch": 0.30732,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.012674713134765625,
      "learning_rate": 0.0001,
      "loss": 3.9056,
      "loss/crossentropy": 2.179181218147278,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20675741881132126,
      "step": 15366
    },
    {
      "epoch": 0.30736,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.02471923828125,
      "learning_rate": 0.0001,
      "loss": 4.4657,
      "loss/crossentropy": 2.106004774570465,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23683416098356247,
      "step": 15368
    },
    {
      "epoch": 0.3074,
      "grad_norm": 2.0,
      "grad_norm_var": 0.023266347249348958,
      "learning_rate": 0.0001,
      "loss": 4.135,
      "loss/crossentropy": 2.2843246459960938,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24139763414859772,
      "step": 15370
    },
    {
      "epoch": 0.30744,
      "grad_norm": 2.0,
      "grad_norm_var": 0.028484853108723958,
      "learning_rate": 0.0001,
      "loss": 4.122,
      "loss/crossentropy": 2.2482924461364746,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2242894545197487,
      "step": 15372
    },
    {
      "epoch": 0.30748,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.025321451822916667,
      "learning_rate": 0.0001,
      "loss": 4.221,
      "loss/crossentropy": 2.1573593616485596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22176649421453476,
      "step": 15374
    },
    {
      "epoch": 0.30752,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.022705078125,
      "learning_rate": 0.0001,
      "loss": 4.219,
      "loss/crossentropy": 1.9243710041046143,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19137389957904816,
      "step": 15376
    },
    {
      "epoch": 0.30756,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.02211481730143229,
      "learning_rate": 0.0001,
      "loss": 4.0903,
      "loss/crossentropy": 2.004511833190918,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20657775551080704,
      "step": 15378
    },
    {
      "epoch": 0.3076,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.022956339518229167,
      "learning_rate": 0.0001,
      "loss": 4.0174,
      "loss/crossentropy": 1.8409560918807983,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19139550626277924,
      "step": 15380
    },
    {
      "epoch": 0.30764,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.019974772135416666,
      "learning_rate": 0.0001,
      "loss": 4.2722,
      "loss/crossentropy": 2.1442220211029053,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23278377205133438,
      "step": 15382
    },
    {
      "epoch": 0.30768,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.013651275634765625,
      "learning_rate": 0.0001,
      "loss": 4.2915,
      "loss/crossentropy": 2.086996078491211,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20602282136678696,
      "step": 15384
    },
    {
      "epoch": 0.30772,
      "grad_norm": 2.125,
      "grad_norm_var": 0.014404042561848959,
      "learning_rate": 0.0001,
      "loss": 4.4884,
      "loss/crossentropy": 2.479863405227661,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23815791308879852,
      "step": 15386
    },
    {
      "epoch": 0.30776,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.009034983317057292,
      "learning_rate": 0.0001,
      "loss": 4.0648,
      "loss/crossentropy": 1.9218324422836304,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22571419924497604,
      "step": 15388
    },
    {
      "epoch": 0.3078,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.009004720052083333,
      "learning_rate": 0.0001,
      "loss": 4.0315,
      "loss/crossentropy": 1.9488168954849243,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2004484310746193,
      "step": 15390
    },
    {
      "epoch": 0.30784,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.006585439046223958,
      "learning_rate": 0.0001,
      "loss": 4.3221,
      "loss/crossentropy": 2.307602286338806,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23881366103887558,
      "step": 15392
    },
    {
      "epoch": 0.30788,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.007283528645833333,
      "learning_rate": 0.0001,
      "loss": 4.1025,
      "loss/crossentropy": 2.1380842328071594,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2019854038953781,
      "step": 15394
    },
    {
      "epoch": 0.30792,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0072509765625,
      "learning_rate": 0.0001,
      "loss": 4.1245,
      "loss/crossentropy": 2.0551159977912903,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18669020384550095,
      "step": 15396
    },
    {
      "epoch": 0.30796,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.009250640869140625,
      "learning_rate": 0.0001,
      "loss": 3.959,
      "loss/crossentropy": 1.9213807582855225,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1843273714184761,
      "step": 15398
    },
    {
      "epoch": 0.308,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008628082275390626,
      "learning_rate": 0.0001,
      "loss": 3.9724,
      "loss/crossentropy": 2.0028855204582214,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2215728610754013,
      "step": 15400
    },
    {
      "epoch": 0.30804,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0064776102701822914,
      "learning_rate": 0.0001,
      "loss": 4.0287,
      "loss/crossentropy": 1.957477331161499,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20727651566267014,
      "step": 15402
    },
    {
      "epoch": 0.30808,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.0080718994140625,
      "learning_rate": 0.0001,
      "loss": 4.0255,
      "loss/crossentropy": 2.2241835594177246,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25234874337911606,
      "step": 15404
    },
    {
      "epoch": 0.30812,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006852213541666667,
      "learning_rate": 0.0001,
      "loss": 4.2677,
      "loss/crossentropy": 2.108901560306549,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20304366201162338,
      "step": 15406
    },
    {
      "epoch": 0.30816,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0075032552083333336,
      "learning_rate": 0.0001,
      "loss": 4.302,
      "loss/crossentropy": 2.1005085110664368,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21205321699380875,
      "step": 15408
    },
    {
      "epoch": 0.3082,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008139801025390626,
      "learning_rate": 0.0001,
      "loss": 4.2505,
      "loss/crossentropy": 2.1565412282943726,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22271832078695297,
      "step": 15410
    },
    {
      "epoch": 0.30824,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.006241607666015625,
      "learning_rate": 0.0001,
      "loss": 4.1236,
      "loss/crossentropy": 1.906798243522644,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21911519020795822,
      "step": 15412
    },
    {
      "epoch": 0.30828,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.006089019775390625,
      "learning_rate": 0.0001,
      "loss": 4.1679,
      "loss/crossentropy": 2.0200547575950623,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19892341643571854,
      "step": 15414
    },
    {
      "epoch": 0.30832,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.006037394205729167,
      "learning_rate": 0.0001,
      "loss": 4.302,
      "loss/crossentropy": 2.14735209941864,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21503761410713196,
      "step": 15416
    },
    {
      "epoch": 0.30836,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.007405344645182292,
      "learning_rate": 0.0001,
      "loss": 4.2943,
      "loss/crossentropy": 2.0479459166526794,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21268057823181152,
      "step": 15418
    },
    {
      "epoch": 0.3084,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.006831614176432291,
      "learning_rate": 0.0001,
      "loss": 4.1969,
      "loss/crossentropy": 1.4835808873176575,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16378673166036606,
      "step": 15420
    },
    {
      "epoch": 0.30844,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.007802073160807292,
      "learning_rate": 0.0001,
      "loss": 4.058,
      "loss/crossentropy": 1.7942256331443787,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18387632071971893,
      "step": 15422
    },
    {
      "epoch": 0.30848,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006534576416015625,
      "learning_rate": 0.0001,
      "loss": 4.1352,
      "loss/crossentropy": 1.8660341501235962,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20189601182937622,
      "step": 15424
    },
    {
      "epoch": 0.30852,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.006257120768229167,
      "learning_rate": 0.0001,
      "loss": 4.0359,
      "loss/crossentropy": 1.8545112013816833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21422524750232697,
      "step": 15426
    },
    {
      "epoch": 0.30856,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0069163004557291664,
      "learning_rate": 0.0001,
      "loss": 4.1411,
      "loss/crossentropy": 2.204701781272888,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21733464300632477,
      "step": 15428
    },
    {
      "epoch": 0.3086,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.006087239583333333,
      "learning_rate": 0.0001,
      "loss": 4.0865,
      "loss/crossentropy": 1.9295769929885864,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20931877195835114,
      "step": 15430
    },
    {
      "epoch": 0.30864,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.008534495035807292,
      "learning_rate": 0.0001,
      "loss": 3.9165,
      "loss/crossentropy": 2.0090243220329285,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20928232371807098,
      "step": 15432
    },
    {
      "epoch": 0.30868,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009326171875,
      "learning_rate": 0.0001,
      "loss": 4.0517,
      "loss/crossentropy": 2.1884257793426514,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20887935161590576,
      "step": 15434
    },
    {
      "epoch": 0.30872,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.009553019205729167,
      "learning_rate": 0.0001,
      "loss": 4.0101,
      "loss/crossentropy": 2.023981511592865,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1893448904156685,
      "step": 15436
    },
    {
      "epoch": 0.30876,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.00909423828125,
      "learning_rate": 0.0001,
      "loss": 4.0168,
      "loss/crossentropy": 2.427197813987732,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21649108827114105,
      "step": 15438
    },
    {
      "epoch": 0.3088,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.015478515625,
      "learning_rate": 0.0001,
      "loss": 4.1243,
      "loss/crossentropy": 2.2578816413879395,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21297947317361832,
      "step": 15440
    },
    {
      "epoch": 0.30884,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.015248362223307292,
      "learning_rate": 0.0001,
      "loss": 4.2977,
      "loss/crossentropy": 1.8909358382225037,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1924509033560753,
      "step": 15442
    },
    {
      "epoch": 0.30888,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.016228993733723957,
      "learning_rate": 0.0001,
      "loss": 4.0142,
      "loss/crossentropy": 2.344139575958252,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20414643734693527,
      "step": 15444
    },
    {
      "epoch": 0.30892,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.01624755859375,
      "learning_rate": 0.0001,
      "loss": 4.2373,
      "loss/crossentropy": 2.268660068511963,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22332928329706192,
      "step": 15446
    },
    {
      "epoch": 0.30896,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0148193359375,
      "learning_rate": 0.0001,
      "loss": 4.0944,
      "loss/crossentropy": 2.1276373267173767,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2123691812157631,
      "step": 15448
    },
    {
      "epoch": 0.309,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.012975819905598958,
      "learning_rate": 0.0001,
      "loss": 4.1025,
      "loss/crossentropy": 2.2471970319747925,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2228669673204422,
      "step": 15450
    },
    {
      "epoch": 0.30904,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0124908447265625,
      "learning_rate": 0.0001,
      "loss": 4.2153,
      "loss/crossentropy": 2.122064530849457,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22981297969818115,
      "step": 15452
    },
    {
      "epoch": 0.30908,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.014823150634765626,
      "learning_rate": 0.0001,
      "loss": 3.815,
      "loss/crossentropy": 2.016224443912506,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19105292856693268,
      "step": 15454
    },
    {
      "epoch": 0.30912,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.016193644205729166,
      "learning_rate": 0.0001,
      "loss": 4.1518,
      "loss/crossentropy": 1.9021474719047546,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23519159853458405,
      "step": 15456
    },
    {
      "epoch": 0.30916,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.016617838541666666,
      "learning_rate": 0.0001,
      "loss": 4.0971,
      "loss/crossentropy": 2.067293703556061,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20525048673152924,
      "step": 15458
    },
    {
      "epoch": 0.3092,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.021930948893229166,
      "learning_rate": 0.0001,
      "loss": 4.3147,
      "loss/crossentropy": 2.0720648169517517,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20188772678375244,
      "step": 15460
    },
    {
      "epoch": 0.30924,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.030890909830729167,
      "learning_rate": 0.0001,
      "loss": 4.212,
      "loss/crossentropy": 2.1377363204956055,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20943745225667953,
      "step": 15462
    },
    {
      "epoch": 0.30928,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.028595987955729166,
      "learning_rate": 0.0001,
      "loss": 4.0799,
      "loss/crossentropy": 1.9101733565330505,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18436391651630402,
      "step": 15464
    },
    {
      "epoch": 0.30932,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.026775868733723958,
      "learning_rate": 0.0001,
      "loss": 4.0945,
      "loss/crossentropy": 2.205660581588745,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2431517392396927,
      "step": 15466
    },
    {
      "epoch": 0.30936,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.02670262654622396,
      "learning_rate": 0.0001,
      "loss": 4.502,
      "loss/crossentropy": 2.3487337827682495,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22333069890737534,
      "step": 15468
    },
    {
      "epoch": 0.3094,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.021971638997395834,
      "learning_rate": 0.0001,
      "loss": 3.9717,
      "loss/crossentropy": 1.953830897808075,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20369011163711548,
      "step": 15470
    },
    {
      "epoch": 0.30944,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0156402587890625,
      "learning_rate": 0.0001,
      "loss": 4.2282,
      "loss/crossentropy": 2.1175807118415833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23428738862276077,
      "step": 15472
    },
    {
      "epoch": 0.30948,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.016585032145182293,
      "learning_rate": 0.0001,
      "loss": 4.0257,
      "loss/crossentropy": 2.114426612854004,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20113752037286758,
      "step": 15474
    },
    {
      "epoch": 0.30952,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.01676203409830729,
      "learning_rate": 0.0001,
      "loss": 3.8239,
      "loss/crossentropy": 2.121155083179474,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23428727686405182,
      "step": 15476
    },
    {
      "epoch": 0.30956,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.007061513264973959,
      "learning_rate": 0.0001,
      "loss": 3.9982,
      "loss/crossentropy": 2.1271342635154724,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1993299275636673,
      "step": 15478
    },
    {
      "epoch": 0.3096,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.007502237955729167,
      "learning_rate": 0.0001,
      "loss": 4.0211,
      "loss/crossentropy": 2.132619023323059,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20428313314914703,
      "step": 15480
    },
    {
      "epoch": 0.30964,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.013720703125,
      "learning_rate": 0.0001,
      "loss": 4.2644,
      "loss/crossentropy": 1.8355163931846619,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1969365030527115,
      "step": 15482
    },
    {
      "epoch": 0.30968,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.016695149739583335,
      "learning_rate": 0.0001,
      "loss": 4.0001,
      "loss/crossentropy": 1.6765839457511902,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19648748636245728,
      "step": 15484
    },
    {
      "epoch": 0.30972,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0168609619140625,
      "learning_rate": 0.0001,
      "loss": 4.0806,
      "loss/crossentropy": 1.7628436088562012,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1850176900625229,
      "step": 15486
    },
    {
      "epoch": 0.30976,
      "grad_norm": 2.125,
      "grad_norm_var": 0.017014312744140624,
      "learning_rate": 0.0001,
      "loss": 4.5902,
      "loss/crossentropy": 2.407894253730774,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2243446335196495,
      "step": 15488
    },
    {
      "epoch": 0.3098,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.017409006754557293,
      "learning_rate": 0.0001,
      "loss": 4.0274,
      "loss/crossentropy": 2.0359743237495422,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2140478640794754,
      "step": 15490
    },
    {
      "epoch": 0.30984,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.012111155192057292,
      "learning_rate": 0.0001,
      "loss": 4.086,
      "loss/crossentropy": 2.1705892086029053,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2108132392168045,
      "step": 15492
    },
    {
      "epoch": 0.30988,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.011126454671223958,
      "learning_rate": 0.0001,
      "loss": 3.9782,
      "loss/crossentropy": 1.836803376674652,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19436977803707123,
      "step": 15494
    },
    {
      "epoch": 0.30992,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010432688395182292,
      "learning_rate": 0.0001,
      "loss": 4.1018,
      "loss/crossentropy": 2.1445683240890503,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21950577944517136,
      "step": 15496
    },
    {
      "epoch": 0.30996,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.007299550374348958,
      "learning_rate": 0.0001,
      "loss": 4.1382,
      "loss/crossentropy": 2.258531093597412,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.212137870490551,
      "step": 15498
    },
    {
      "epoch": 0.31,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.004052480061848958,
      "learning_rate": 0.0001,
      "loss": 4.2473,
      "loss/crossentropy": 1.9310460686683655,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18316251039505005,
      "step": 15500
    },
    {
      "epoch": 0.31004,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.004788970947265625,
      "learning_rate": 0.0001,
      "loss": 3.8656,
      "loss/crossentropy": 1.867807149887085,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18478462100028992,
      "step": 15502
    },
    {
      "epoch": 0.31008,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.008042144775390624,
      "learning_rate": 0.0001,
      "loss": 4.4691,
      "loss/crossentropy": 2.0724986791610718,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20986932516098022,
      "step": 15504
    },
    {
      "epoch": 0.31012,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.010138956705729167,
      "learning_rate": 0.0001,
      "loss": 4.5188,
      "loss/crossentropy": 2.1599953174591064,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2250317633152008,
      "step": 15506
    },
    {
      "epoch": 0.31016,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.015608469645182291,
      "learning_rate": 0.0001,
      "loss": 3.7842,
      "loss/crossentropy": 1.975223183631897,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2104134038090706,
      "step": 15508
    },
    {
      "epoch": 0.3102,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.019539388020833333,
      "learning_rate": 0.0001,
      "loss": 4.0028,
      "loss/crossentropy": 2.133773148059845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20153696089982986,
      "step": 15510
    },
    {
      "epoch": 0.31024,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.02014948527018229,
      "learning_rate": 0.0001,
      "loss": 3.9948,
      "loss/crossentropy": 1.994078278541565,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20573429763317108,
      "step": 15512
    },
    {
      "epoch": 0.31028,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.019280751546223957,
      "learning_rate": 0.0001,
      "loss": 3.8209,
      "loss/crossentropy": 1.7885381579399109,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1889629364013672,
      "step": 15514
    },
    {
      "epoch": 0.31032,
      "grad_norm": 2.0,
      "grad_norm_var": 0.019191233317057292,
      "learning_rate": 0.0001,
      "loss": 4.2765,
      "loss/crossentropy": 2.495113968849182,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22110097110271454,
      "step": 15516
    },
    {
      "epoch": 0.31036,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.019160715738932292,
      "learning_rate": 0.0001,
      "loss": 4.0465,
      "loss/crossentropy": 2.0757648944854736,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20948036015033722,
      "step": 15518
    },
    {
      "epoch": 0.3104,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.015793609619140624,
      "learning_rate": 0.0001,
      "loss": 4.1369,
      "loss/crossentropy": 2.151356875896454,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22598887234926224,
      "step": 15520
    },
    {
      "epoch": 0.31044,
      "grad_norm": 2.0,
      "grad_norm_var": 0.010933176676432291,
      "learning_rate": 0.0001,
      "loss": 4.2193,
      "loss/crossentropy": 2.1210484504699707,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23365220427513123,
      "step": 15522
    },
    {
      "epoch": 0.31048,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.008414459228515626,
      "learning_rate": 0.0001,
      "loss": 4.0584,
      "loss/crossentropy": 2.3766754865646362,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22161870449781418,
      "step": 15524
    },
    {
      "epoch": 0.31052,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0043853759765625,
      "learning_rate": 0.0001,
      "loss": 4.0968,
      "loss/crossentropy": 2.33975088596344,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21598009765148163,
      "step": 15526
    },
    {
      "epoch": 0.31056,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.006468709309895833,
      "learning_rate": 0.0001,
      "loss": 4.4643,
      "loss/crossentropy": 2.3024542331695557,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20656803250312805,
      "step": 15528
    },
    {
      "epoch": 0.3106,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.007666015625,
      "learning_rate": 0.0001,
      "loss": 4.1137,
      "loss/crossentropy": 2.009307861328125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19054396450519562,
      "step": 15530
    },
    {
      "epoch": 0.31064,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.008902994791666667,
      "learning_rate": 0.0001,
      "loss": 3.9413,
      "loss/crossentropy": 1.8119140267372131,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19199557602405548,
      "step": 15532
    },
    {
      "epoch": 0.31068,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.009614817301432292,
      "learning_rate": 0.0001,
      "loss": 3.8753,
      "loss/crossentropy": 1.9586234092712402,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18981865793466568,
      "step": 15534
    },
    {
      "epoch": 0.31072,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.010798899332682292,
      "learning_rate": 0.0001,
      "loss": 4.4319,
      "loss/crossentropy": 2.1208658814430237,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20590776205062866,
      "step": 15536
    },
    {
      "epoch": 0.31076,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.012336222330729167,
      "learning_rate": 0.0001,
      "loss": 3.7806,
      "loss/crossentropy": 1.539306402206421,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1615876704454422,
      "step": 15538
    },
    {
      "epoch": 0.3108,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.013768513997395834,
      "learning_rate": 0.0001,
      "loss": 4.0686,
      "loss/crossentropy": 2.1376689672470093,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21377786993980408,
      "step": 15540
    },
    {
      "epoch": 0.31084,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.016249338785807293,
      "learning_rate": 0.0001,
      "loss": 4.368,
      "loss/crossentropy": 1.954556941986084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19270780682563782,
      "step": 15542
    },
    {
      "epoch": 0.31088,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.013575998942057292,
      "learning_rate": 0.0001,
      "loss": 4.1313,
      "loss/crossentropy": 2.0225048661231995,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1937834918498993,
      "step": 15544
    },
    {
      "epoch": 0.31092,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.011310831705729166,
      "learning_rate": 0.0001,
      "loss": 4.0919,
      "loss/crossentropy": 2.149065852165222,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21962979435920715,
      "step": 15546
    },
    {
      "epoch": 0.31096,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.013073476155598958,
      "learning_rate": 0.0001,
      "loss": 4.0192,
      "loss/crossentropy": 1.9546958804130554,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20022059231996536,
      "step": 15548
    },
    {
      "epoch": 0.311,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.013728841145833334,
      "learning_rate": 0.0001,
      "loss": 4.2815,
      "loss/crossentropy": 2.3048110008239746,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2232206016778946,
      "step": 15550
    },
    {
      "epoch": 0.31104,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.014525349934895833,
      "learning_rate": 0.0001,
      "loss": 3.9263,
      "loss/crossentropy": 2.0329134464263916,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20285522937774658,
      "step": 15552
    },
    {
      "epoch": 0.31108,
      "grad_norm": 2.0,
      "grad_norm_var": 0.013138580322265624,
      "learning_rate": 0.0001,
      "loss": 4.1654,
      "loss/crossentropy": 2.0910086631774902,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20308854430913925,
      "step": 15554
    },
    {
      "epoch": 0.31112,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.012361399332682292,
      "learning_rate": 0.0001,
      "loss": 4.1922,
      "loss/crossentropy": 2.3473485708236694,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2138630449771881,
      "step": 15556
    },
    {
      "epoch": 0.31116,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0098297119140625,
      "learning_rate": 0.0001,
      "loss": 4.3456,
      "loss/crossentropy": 2.4176318645477295,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23624707758426666,
      "step": 15558
    },
    {
      "epoch": 0.3112,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.009943644205729166,
      "learning_rate": 0.0001,
      "loss": 4.0898,
      "loss/crossentropy": 2.190505266189575,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21235676854848862,
      "step": 15560
    },
    {
      "epoch": 0.31124,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.1935198465983073,
      "learning_rate": 0.0001,
      "loss": 4.3688,
      "loss/crossentropy": 2.1774531602859497,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2568993717432022,
      "step": 15562
    },
    {
      "epoch": 0.31128,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.1934832255045573,
      "learning_rate": 0.0001,
      "loss": 4.004,
      "loss/crossentropy": 2.0320950150489807,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21587081998586655,
      "step": 15564
    },
    {
      "epoch": 0.31132,
      "grad_norm": 2.125,
      "grad_norm_var": 0.19068781534830728,
      "learning_rate": 0.0001,
      "loss": 4.3037,
      "loss/crossentropy": 2.0477761030197144,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21909283846616745,
      "step": 15566
    },
    {
      "epoch": 0.31136,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.18454996744791666,
      "learning_rate": 0.0001,
      "loss": 4.2288,
      "loss/crossentropy": 2.3907222747802734,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21334309875965118,
      "step": 15568
    },
    {
      "epoch": 0.3114,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.1816973368326823,
      "learning_rate": 0.0001,
      "loss": 4.28,
      "loss/crossentropy": 2.2539732456207275,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21502045542001724,
      "step": 15570
    },
    {
      "epoch": 0.31144,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.17766520182291667,
      "learning_rate": 0.0001,
      "loss": 3.9675,
      "loss/crossentropy": 2.036223292350769,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1959664672613144,
      "step": 15572
    },
    {
      "epoch": 0.31148,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.17722880045572917,
      "learning_rate": 0.0001,
      "loss": 4.0387,
      "loss/crossentropy": 1.7659193873405457,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19535844773054123,
      "step": 15574
    },
    {
      "epoch": 0.31152,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.17946751912434897,
      "learning_rate": 0.0001,
      "loss": 3.8892,
      "loss/crossentropy": 2.0053369402885437,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24516403675079346,
      "step": 15576
    },
    {
      "epoch": 0.31156,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.0060384114583333336,
      "learning_rate": 0.0001,
      "loss": 3.9851,
      "loss/crossentropy": 2.1804131269454956,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2349890023469925,
      "step": 15578
    },
    {
      "epoch": 0.3116,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.008552042643229167,
      "learning_rate": 0.0001,
      "loss": 3.9369,
      "loss/crossentropy": 2.314085602760315,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21073052287101746,
      "step": 15580
    },
    {
      "epoch": 0.31164,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.008973948160807292,
      "learning_rate": 0.0001,
      "loss": 3.7951,
      "loss/crossentropy": 2.1868577003479004,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21520060300827026,
      "step": 15582
    },
    {
      "epoch": 0.31168,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.007957967122395833,
      "learning_rate": 0.0001,
      "loss": 4.1618,
      "loss/crossentropy": 1.9785407185554504,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18188606202602386,
      "step": 15584
    },
    {
      "epoch": 0.31172,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.009598795572916667,
      "learning_rate": 0.0001,
      "loss": 3.9173,
      "loss/crossentropy": 1.9320110082626343,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2022753208875656,
      "step": 15586
    },
    {
      "epoch": 0.31176,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.015653483072916665,
      "learning_rate": 0.0001,
      "loss": 4.0995,
      "loss/crossentropy": 1.7925443649291992,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17203447967767715,
      "step": 15588
    },
    {
      "epoch": 0.3118,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.016556803385416666,
      "learning_rate": 0.0001,
      "loss": 3.8876,
      "loss/crossentropy": 2.048543393611908,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18696243315935135,
      "step": 15590
    },
    {
      "epoch": 0.31184,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.01658935546875,
      "learning_rate": 0.0001,
      "loss": 4.125,
      "loss/crossentropy": 2.376060724258423,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20351988822221756,
      "step": 15592
    },
    {
      "epoch": 0.31188,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.0164947509765625,
      "learning_rate": 0.0001,
      "loss": 4.1778,
      "loss/crossentropy": 1.5962707996368408,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1538907140493393,
      "step": 15594
    },
    {
      "epoch": 0.31192,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.016281890869140624,
      "learning_rate": 0.0001,
      "loss": 3.7529,
      "loss/crossentropy": 1.7845428586006165,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18107154965400696,
      "step": 15596
    },
    {
      "epoch": 0.31196,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.01536865234375,
      "learning_rate": 0.0001,
      "loss": 4.0385,
      "loss/crossentropy": 1.8557100296020508,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2010527327656746,
      "step": 15598
    },
    {
      "epoch": 0.312,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.015372721354166667,
      "learning_rate": 0.0001,
      "loss": 3.9735,
      "loss/crossentropy": 1.8659257888793945,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19198190420866013,
      "step": 15600
    },
    {
      "epoch": 0.31204,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.014078521728515625,
      "learning_rate": 0.0001,
      "loss": 3.9842,
      "loss/crossentropy": 2.0340747833251953,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19963373243808746,
      "step": 15602
    },
    {
      "epoch": 0.31208,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.00931396484375,
      "learning_rate": 0.0001,
      "loss": 4.1192,
      "loss/crossentropy": 2.0676557421684265,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20130064338445663,
      "step": 15604
    },
    {
      "epoch": 0.31212,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.008397420247395834,
      "learning_rate": 0.0001,
      "loss": 3.8069,
      "loss/crossentropy": 2.038145124912262,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2034953236579895,
      "step": 15606
    },
    {
      "epoch": 0.31216,
      "grad_norm": 2.65625,
      "grad_norm_var": 0.03449605305989583,
      "learning_rate": 0.0001,
      "loss": 4.3615,
      "loss/crossentropy": 2.3371706008911133,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21727421879768372,
      "step": 15608
    },
    {
      "epoch": 0.3122,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.03327611287434896,
      "learning_rate": 0.0001,
      "loss": 4.1988,
      "loss/crossentropy": 2.3314318656921387,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2064831778407097,
      "step": 15610
    },
    {
      "epoch": 0.31224,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.03206558227539062,
      "learning_rate": 0.0001,
      "loss": 3.9634,
      "loss/crossentropy": 1.8074566721916199,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1755329668521881,
      "step": 15612
    },
    {
      "epoch": 0.31228,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.032138824462890625,
      "learning_rate": 0.0001,
      "loss": 4.1849,
      "loss/crossentropy": 2.1219228506088257,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23183195292949677,
      "step": 15614
    },
    {
      "epoch": 0.31232,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.031434885660807294,
      "learning_rate": 0.0001,
      "loss": 4.1418,
      "loss/crossentropy": 2.3573983907699585,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20713336020708084,
      "step": 15616
    },
    {
      "epoch": 0.31236,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.033933258056640624,
      "learning_rate": 0.0001,
      "loss": 4.048,
      "loss/crossentropy": 2.0748316049575806,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19384868443012238,
      "step": 15618
    },
    {
      "epoch": 0.3124,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.033882395426432295,
      "learning_rate": 0.0001,
      "loss": 4.1834,
      "loss/crossentropy": 1.7933568358421326,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19711802154779434,
      "step": 15620
    },
    {
      "epoch": 0.31244,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.03115819295247396,
      "learning_rate": 0.0001,
      "loss": 4.2023,
      "loss/crossentropy": 1.8922778367996216,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19847699999809265,
      "step": 15622
    },
    {
      "epoch": 0.31248,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0048004150390625,
      "learning_rate": 0.0001,
      "loss": 4.2326,
      "loss/crossentropy": 2.3432679176330566,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22804179042577744,
      "step": 15624
    },
    {
      "epoch": 0.31252,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.007469685872395834,
      "learning_rate": 0.0001,
      "loss": 4.1215,
      "loss/crossentropy": 2.340356707572937,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22454006224870682,
      "step": 15626
    },
    {
      "epoch": 0.31256,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.010737864176432292,
      "learning_rate": 0.0001,
      "loss": 4.0874,
      "loss/crossentropy": 2.037777841091156,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20374249666929245,
      "step": 15628
    },
    {
      "epoch": 0.3126,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.010562896728515625,
      "learning_rate": 0.0001,
      "loss": 4.3193,
      "loss/crossentropy": 2.2576274275779724,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19910124689340591,
      "step": 15630
    },
    {
      "epoch": 0.31264,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.011302693684895834,
      "learning_rate": 0.0001,
      "loss": 4.1561,
      "loss/crossentropy": 2.1354891061782837,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20646820217370987,
      "step": 15632
    },
    {
      "epoch": 0.31268,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.011136627197265625,
      "learning_rate": 0.0001,
      "loss": 3.906,
      "loss/crossentropy": 1.8888733386993408,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20730097591876984,
      "step": 15634
    },
    {
      "epoch": 0.31272,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.012474568684895833,
      "learning_rate": 0.0001,
      "loss": 3.9522,
      "loss/crossentropy": 2.1037773489952087,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20308485627174377,
      "step": 15636
    },
    {
      "epoch": 0.31276,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.017085774739583334,
      "learning_rate": 0.0001,
      "loss": 4.244,
      "loss/crossentropy": 1.996969223022461,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2080031782388687,
      "step": 15638
    },
    {
      "epoch": 0.3128,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.0469146728515625,
      "learning_rate": 0.0001,
      "loss": 3.915,
      "loss/crossentropy": 2.0528018474578857,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2248053327202797,
      "step": 15640
    },
    {
      "epoch": 0.31284,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.044864654541015625,
      "learning_rate": 0.0001,
      "loss": 4.1718,
      "loss/crossentropy": 2.270516335964203,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19578035175800323,
      "step": 15642
    },
    {
      "epoch": 0.31288,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0400299072265625,
      "learning_rate": 0.0001,
      "loss": 4.1463,
      "loss/crossentropy": 2.146915912628174,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21282948553562164,
      "step": 15644
    },
    {
      "epoch": 0.31292,
      "grad_norm": 2.125,
      "grad_norm_var": 0.041562652587890624,
      "learning_rate": 0.0001,
      "loss": 4.3275,
      "loss/crossentropy": 2.058235287666321,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20604285597801208,
      "step": 15646
    },
    {
      "epoch": 0.31296,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0414215087890625,
      "learning_rate": 0.0001,
      "loss": 4.4304,
      "loss/crossentropy": 2.3217705488204956,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24202678352594376,
      "step": 15648
    },
    {
      "epoch": 0.313,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.041257476806640624,
      "learning_rate": 0.0001,
      "loss": 4.1372,
      "loss/crossentropy": 2.108501672744751,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20358111709356308,
      "step": 15650
    },
    {
      "epoch": 0.31304,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.041025543212890626,
      "learning_rate": 0.0001,
      "loss": 4.1224,
      "loss/crossentropy": 2.0875802636146545,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20186328142881393,
      "step": 15652
    },
    {
      "epoch": 0.31308,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.036321767171223956,
      "learning_rate": 0.0001,
      "loss": 4.1994,
      "loss/crossentropy": 2.0159415006637573,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2053827941417694,
      "step": 15654
    },
    {
      "epoch": 0.31312,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008402252197265625,
      "learning_rate": 0.0001,
      "loss": 3.9711,
      "loss/crossentropy": 1.8664127588272095,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18666206300258636,
      "step": 15656
    },
    {
      "epoch": 0.31316,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.008548736572265625,
      "learning_rate": 0.0001,
      "loss": 3.9399,
      "loss/crossentropy": 1.9351357221603394,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.189142145216465,
      "step": 15658
    },
    {
      "epoch": 0.3132,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.010872141520182291,
      "learning_rate": 0.0001,
      "loss": 4.3288,
      "loss/crossentropy": 1.9937176704406738,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18212904781103134,
      "step": 15660
    },
    {
      "epoch": 0.31324,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009801991780598958,
      "learning_rate": 0.0001,
      "loss": 4.1238,
      "loss/crossentropy": 2.0626447796821594,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20908795297145844,
      "step": 15662
    },
    {
      "epoch": 0.31328,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.008506011962890626,
      "learning_rate": 0.0001,
      "loss": 4.0436,
      "loss/crossentropy": 1.8491488695144653,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18241358548402786,
      "step": 15664
    },
    {
      "epoch": 0.31332,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.0084625244140625,
      "learning_rate": 0.0001,
      "loss": 3.9885,
      "loss/crossentropy": 2.2744827270507812,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2295297086238861,
      "step": 15666
    },
    {
      "epoch": 0.31336,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0078521728515625,
      "learning_rate": 0.0001,
      "loss": 3.9213,
      "loss/crossentropy": 2.30733585357666,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23149366676807404,
      "step": 15668
    },
    {
      "epoch": 0.3134,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.00892333984375,
      "learning_rate": 0.0001,
      "loss": 4.1357,
      "loss/crossentropy": 2.038204550743103,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19608831405639648,
      "step": 15670
    },
    {
      "epoch": 0.31344,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.008676910400390625,
      "learning_rate": 0.0001,
      "loss": 4.1235,
      "loss/crossentropy": 2.076035261154175,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21801196038722992,
      "step": 15672
    },
    {
      "epoch": 0.31348,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.011310831705729166,
      "learning_rate": 0.0001,
      "loss": 4.0923,
      "loss/crossentropy": 1.9234278202056885,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18380460888147354,
      "step": 15674
    },
    {
      "epoch": 0.31352,
      "grad_norm": 2.125,
      "grad_norm_var": 0.00931396484375,
      "learning_rate": 0.0001,
      "loss": 4.1002,
      "loss/crossentropy": 1.9312421679496765,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18474044650793076,
      "step": 15676
    },
    {
      "epoch": 0.31356,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009716542561848958,
      "learning_rate": 0.0001,
      "loss": 4.079,
      "loss/crossentropy": 2.014132261276245,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2045312151312828,
      "step": 15678
    },
    {
      "epoch": 0.3136,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0099365234375,
      "learning_rate": 0.0001,
      "loss": 4.2325,
      "loss/crossentropy": 1.925381362438202,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18306680023670197,
      "step": 15680
    },
    {
      "epoch": 0.31364,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.009346516927083333,
      "learning_rate": 0.0001,
      "loss": 4.0766,
      "loss/crossentropy": 1.932526171207428,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20741897821426392,
      "step": 15682
    },
    {
      "epoch": 0.31368,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008963775634765626,
      "learning_rate": 0.0001,
      "loss": 4.2234,
      "loss/crossentropy": 2.1144750714302063,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19398797303438187,
      "step": 15684
    },
    {
      "epoch": 0.31372,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0076983133951822914,
      "learning_rate": 0.0001,
      "loss": 3.8411,
      "loss/crossentropy": 1.8115296363830566,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1869400143623352,
      "step": 15686
    },
    {
      "epoch": 0.31376,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.014686838785807291,
      "learning_rate": 0.0001,
      "loss": 4.0935,
      "loss/crossentropy": 2.0550562739372253,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2093418911099434,
      "step": 15688
    },
    {
      "epoch": 0.3138,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.013541412353515626,
      "learning_rate": 0.0001,
      "loss": 3.8549,
      "loss/crossentropy": 2.144737482070923,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20025025308132172,
      "step": 15690
    },
    {
      "epoch": 0.31384,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.012629191080729166,
      "learning_rate": 0.0001,
      "loss": 3.9911,
      "loss/crossentropy": 2.2495052814483643,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21178698539733887,
      "step": 15692
    },
    {
      "epoch": 0.31388,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.012740071614583333,
      "learning_rate": 0.0001,
      "loss": 4.2208,
      "loss/crossentropy": 2.1858155727386475,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20891498774290085,
      "step": 15694
    },
    {
      "epoch": 0.31392,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0131744384765625,
      "learning_rate": 0.0001,
      "loss": 4.1841,
      "loss/crossentropy": 2.1480127573013306,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2149479240179062,
      "step": 15696
    },
    {
      "epoch": 0.31396,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.013329060872395833,
      "learning_rate": 0.0001,
      "loss": 3.9218,
      "loss/crossentropy": 2.1585733294487,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20317788422107697,
      "step": 15698
    },
    {
      "epoch": 0.314,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.2103179931640625,
      "learning_rate": 0.0001,
      "loss": 4.06,
      "loss/crossentropy": 2.1281010508537292,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2032032608985901,
      "step": 15700
    },
    {
      "epoch": 0.31404,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.2096588134765625,
      "learning_rate": 0.0001,
      "loss": 4.0189,
      "loss/crossentropy": 2.160776972770691,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20999448001384735,
      "step": 15702
    },
    {
      "epoch": 0.31408,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.20640869140625,
      "learning_rate": 0.0001,
      "loss": 3.961,
      "loss/crossentropy": 2.1508986949920654,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21675271540880203,
      "step": 15704
    },
    {
      "epoch": 0.31412,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.2011871337890625,
      "learning_rate": 0.0001,
      "loss": 4.1573,
      "loss/crossentropy": 2.083084225654602,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21110595017671585,
      "step": 15706
    },
    {
      "epoch": 0.31416,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.20275472005208334,
      "learning_rate": 0.0001,
      "loss": 3.9785,
      "loss/crossentropy": 2.1268805265426636,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20728368312120438,
      "step": 15708
    },
    {
      "epoch": 0.3142,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.20265706380208334,
      "learning_rate": 0.0001,
      "loss": 4.4291,
      "loss/crossentropy": 2.156775116920471,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22175253927707672,
      "step": 15710
    },
    {
      "epoch": 0.31424,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.20224583943684896,
      "learning_rate": 0.0001,
      "loss": 4.1537,
      "loss/crossentropy": 2.2541415691375732,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22773633152246475,
      "step": 15712
    },
    {
      "epoch": 0.31428,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.20350723266601561,
      "learning_rate": 0.0001,
      "loss": 3.9166,
      "loss/crossentropy": 1.6463716626167297,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18172992020845413,
      "step": 15714
    },
    {
      "epoch": 0.31432,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.011669921875,
      "learning_rate": 0.0001,
      "loss": 4.0144,
      "loss/crossentropy": 1.9325580596923828,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19147639721632004,
      "step": 15716
    },
    {
      "epoch": 0.31436,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.014115142822265624,
      "learning_rate": 0.0001,
      "loss": 3.9952,
      "loss/crossentropy": 2.4016847610473633,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2058410421013832,
      "step": 15718
    },
    {
      "epoch": 0.3144,
      "grad_norm": 2.125,
      "grad_norm_var": 0.014338175455729166,
      "learning_rate": 0.0001,
      "loss": 4.1847,
      "loss/crossentropy": 2.3456791639328003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22646376490592957,
      "step": 15720
    },
    {
      "epoch": 0.31444,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.013785807291666667,
      "learning_rate": 0.0001,
      "loss": 4.0551,
      "loss/crossentropy": 1.6703909635543823,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1818019300699234,
      "step": 15722
    },
    {
      "epoch": 0.31448,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.011065419514973958,
      "learning_rate": 0.0001,
      "loss": 4.0561,
      "loss/crossentropy": 2.0004162192344666,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2033773437142372,
      "step": 15724
    },
    {
      "epoch": 0.31452,
      "grad_norm": 2.0,
      "grad_norm_var": 0.009905751546223958,
      "learning_rate": 0.0001,
      "loss": 4.1799,
      "loss/crossentropy": 2.018653154373169,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20320426672697067,
      "step": 15726
    },
    {
      "epoch": 0.31456,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.009592437744140625,
      "learning_rate": 0.0001,
      "loss": 4.0991,
      "loss/crossentropy": 2.0881033539772034,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2126169577240944,
      "step": 15728
    },
    {
      "epoch": 0.3146,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0067942301432291664,
      "learning_rate": 0.0001,
      "loss": 4.2159,
      "loss/crossentropy": 2.1103954911231995,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2163369059562683,
      "step": 15730
    },
    {
      "epoch": 0.31464,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.007868448893229166,
      "learning_rate": 0.0001,
      "loss": 4.142,
      "loss/crossentropy": 1.9757474660873413,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21398526430130005,
      "step": 15732
    },
    {
      "epoch": 0.31468,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.006304677327473958,
      "learning_rate": 0.0001,
      "loss": 3.8965,
      "loss/crossentropy": 2.114749312400818,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21620750427246094,
      "step": 15734
    },
    {
      "epoch": 0.31472,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.007163238525390625,
      "learning_rate": 0.0001,
      "loss": 3.9688,
      "loss/crossentropy": 1.9022989869117737,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1895657405257225,
      "step": 15736
    },
    {
      "epoch": 0.31476,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.013059234619140625,
      "learning_rate": 0.0001,
      "loss": 4.2254,
      "loss/crossentropy": 2.0457635521888733,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21281428635120392,
      "step": 15738
    },
    {
      "epoch": 0.3148,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011954752604166667,
      "learning_rate": 0.0001,
      "loss": 3.8985,
      "loss/crossentropy": 1.9585834741592407,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2003040909767151,
      "step": 15740
    },
    {
      "epoch": 0.31484,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.014745076497395834,
      "learning_rate": 0.0001,
      "loss": 4.1682,
      "loss/crossentropy": 2.06454074382782,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22020584344863892,
      "step": 15742
    },
    {
      "epoch": 0.31488,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.015665690104166668,
      "learning_rate": 0.0001,
      "loss": 4.2384,
      "loss/crossentropy": 2.2651820182800293,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22998705506324768,
      "step": 15744
    },
    {
      "epoch": 0.31492,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0172607421875,
      "learning_rate": 0.0001,
      "loss": 4.258,
      "loss/crossentropy": 2.269286036491394,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21012098342180252,
      "step": 15746
    },
    {
      "epoch": 0.31496,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.017891438802083333,
      "learning_rate": 0.0001,
      "loss": 4.0604,
      "loss/crossentropy": 1.6778026223182678,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17330823838710785,
      "step": 15748
    },
    {
      "epoch": 0.315,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.020775349934895833,
      "learning_rate": 0.0001,
      "loss": 3.9785,
      "loss/crossentropy": 1.8929405808448792,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18170948326587677,
      "step": 15750
    },
    {
      "epoch": 0.31504,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.01980768839518229,
      "learning_rate": 0.0001,
      "loss": 4.0874,
      "loss/crossentropy": 1.8814340233802795,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2097821682691574,
      "step": 15752
    },
    {
      "epoch": 0.31508,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.014465077718098959,
      "learning_rate": 0.0001,
      "loss": 3.8622,
      "loss/crossentropy": 1.7101264595985413,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1834770292043686,
      "step": 15754
    },
    {
      "epoch": 0.31512,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.01607233683268229,
      "learning_rate": 0.0001,
      "loss": 3.8498,
      "loss/crossentropy": 2.0287395119667053,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19603880494832993,
      "step": 15756
    },
    {
      "epoch": 0.31516,
      "grad_norm": 1.875,
      "grad_norm_var": 0.010467274983723959,
      "learning_rate": 0.0001,
      "loss": 3.7485,
      "loss/crossentropy": 2.0119568705558777,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.212313212454319,
      "step": 15758
    },
    {
      "epoch": 0.3152,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011441802978515625,
      "learning_rate": 0.0001,
      "loss": 4.5141,
      "loss/crossentropy": 1.8937278985977173,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1774902269244194,
      "step": 15760
    },
    {
      "epoch": 0.31524,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.009104156494140625,
      "learning_rate": 0.0001,
      "loss": 4.3562,
      "loss/crossentropy": 2.557780146598816,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22111114859580994,
      "step": 15762
    },
    {
      "epoch": 0.31528,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.011262003580729167,
      "learning_rate": 0.0001,
      "loss": 4.1646,
      "loss/crossentropy": 2.1467400789260864,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2134222611784935,
      "step": 15764
    },
    {
      "epoch": 0.31532,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.00814208984375,
      "learning_rate": 0.0001,
      "loss": 3.9631,
      "loss/crossentropy": 1.9758468270301819,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1853625252842903,
      "step": 15766
    },
    {
      "epoch": 0.31536,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.009699503580729166,
      "learning_rate": 0.0001,
      "loss": 4.2144,
      "loss/crossentropy": 1.8620794415473938,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18791437149047852,
      "step": 15768
    },
    {
      "epoch": 0.3154,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008893839518229167,
      "learning_rate": 0.0001,
      "loss": 4.2066,
      "loss/crossentropy": 2.187020480632782,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2102060467004776,
      "step": 15770
    },
    {
      "epoch": 0.31544,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.006522369384765625,
      "learning_rate": 0.0001,
      "loss": 4.1828,
      "loss/crossentropy": 2.0049954652786255,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19090547412633896,
      "step": 15772
    },
    {
      "epoch": 0.31548,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.010357411702473958,
      "learning_rate": 0.0001,
      "loss": 4.195,
      "loss/crossentropy": 2.0030736327171326,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20134840160608292,
      "step": 15774
    },
    {
      "epoch": 0.31552,
      "grad_norm": 2.0,
      "grad_norm_var": 0.010687001546223958,
      "learning_rate": 0.0001,
      "loss": 4.3092,
      "loss/crossentropy": 1.8941256999969482,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20826192945241928,
      "step": 15776
    },
    {
      "epoch": 0.31556,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.015596516927083333,
      "learning_rate": 0.0001,
      "loss": 4.0767,
      "loss/crossentropy": 2.0645129680633545,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19137012213468552,
      "step": 15778
    },
    {
      "epoch": 0.3156,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.014240519205729166,
      "learning_rate": 0.0001,
      "loss": 4.0349,
      "loss/crossentropy": 2.198352813720703,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21606199443340302,
      "step": 15780
    },
    {
      "epoch": 0.31564,
      "grad_norm": 2.125,
      "grad_norm_var": 0.014020792643229167,
      "learning_rate": 0.0001,
      "loss": 4.1544,
      "loss/crossentropy": 1.9041627049446106,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19438787549734116,
      "step": 15782
    },
    {
      "epoch": 0.31568,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.027741495768229166,
      "learning_rate": 0.0001,
      "loss": 4.2615,
      "loss/crossentropy": 2.110167443752289,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2026640698313713,
      "step": 15784
    },
    {
      "epoch": 0.31572,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.02655029296875,
      "learning_rate": 0.0001,
      "loss": 4.1932,
      "loss/crossentropy": 2.5422832369804382,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20274986326694489,
      "step": 15786
    },
    {
      "epoch": 0.31576,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.02840576171875,
      "learning_rate": 0.0001,
      "loss": 4.2038,
      "loss/crossentropy": 2.106445074081421,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20275159180164337,
      "step": 15788
    },
    {
      "epoch": 0.3158,
      "grad_norm": 2.375,
      "grad_norm_var": 0.029857381184895834,
      "learning_rate": 0.0001,
      "loss": 4.2718,
      "loss/crossentropy": 2.341952681541443,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22970512509346008,
      "step": 15790
    },
    {
      "epoch": 0.31584,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.030228678385416666,
      "learning_rate": 0.0001,
      "loss": 4.238,
      "loss/crossentropy": 2.120057761669159,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21171308308839798,
      "step": 15792
    },
    {
      "epoch": 0.31588,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.027644856770833334,
      "learning_rate": 0.0001,
      "loss": 3.8819,
      "loss/crossentropy": 1.5601414442062378,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20278310775756836,
      "step": 15794
    },
    {
      "epoch": 0.31592,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.027596028645833333,
      "learning_rate": 0.0001,
      "loss": 4.0743,
      "loss/crossentropy": 2.0634138584136963,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1815958246588707,
      "step": 15796
    },
    {
      "epoch": 0.31596,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0312652587890625,
      "learning_rate": 0.0001,
      "loss": 3.8183,
      "loss/crossentropy": 1.8437300324440002,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18099966645240784,
      "step": 15798
    },
    {
      "epoch": 0.316,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.015897369384765624,
      "learning_rate": 0.0001,
      "loss": 4.1102,
      "loss/crossentropy": 1.9323074221611023,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1884680539369583,
      "step": 15800
    },
    {
      "epoch": 0.31604,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.018027496337890626,
      "learning_rate": 0.0001,
      "loss": 4.3145,
      "loss/crossentropy": 2.158499598503113,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23394601047039032,
      "step": 15802
    },
    {
      "epoch": 0.31608,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.018062337239583334,
      "learning_rate": 0.0001,
      "loss": 4.0894,
      "loss/crossentropy": 1.9635908007621765,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20692527294158936,
      "step": 15804
    },
    {
      "epoch": 0.31612,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.010503896077473958,
      "learning_rate": 0.0001,
      "loss": 4.1912,
      "loss/crossentropy": 1.96099454164505,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1971232146024704,
      "step": 15806
    },
    {
      "epoch": 0.31616,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.010643513997395833,
      "learning_rate": 0.0001,
      "loss": 4.2205,
      "loss/crossentropy": 1.9690999388694763,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19965435564517975,
      "step": 15808
    },
    {
      "epoch": 0.3162,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011354319254557292,
      "learning_rate": 0.0001,
      "loss": 4.113,
      "loss/crossentropy": 2.3924723863601685,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2192596197128296,
      "step": 15810
    },
    {
      "epoch": 0.31624,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.013999176025390626,
      "learning_rate": 0.0001,
      "loss": 3.8794,
      "loss/crossentropy": 2.137382209300995,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20949169248342514,
      "step": 15812
    },
    {
      "epoch": 0.31628,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.013103993733723958,
      "learning_rate": 0.0001,
      "loss": 4.0187,
      "loss/crossentropy": 1.986245334148407,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19678276032209396,
      "step": 15814
    },
    {
      "epoch": 0.31632,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.013492584228515625,
      "learning_rate": 0.0001,
      "loss": 3.9623,
      "loss/crossentropy": 1.9210307598114014,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18748773634433746,
      "step": 15816
    },
    {
      "epoch": 0.31636,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0087066650390625,
      "learning_rate": 0.0001,
      "loss": 4.1328,
      "loss/crossentropy": 1.9330076575279236,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21187862008810043,
      "step": 15818
    },
    {
      "epoch": 0.3164,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.007306925455729167,
      "learning_rate": 0.0001,
      "loss": 4.2475,
      "loss/crossentropy": 2.275822639465332,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2303369864821434,
      "step": 15820
    },
    {
      "epoch": 0.31644,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0082275390625,
      "learning_rate": 0.0001,
      "loss": 4.3641,
      "loss/crossentropy": 1.9317327737808228,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19543734192848206,
      "step": 15822
    },
    {
      "epoch": 0.31648,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.008715565999348958,
      "learning_rate": 0.0001,
      "loss": 4.3209,
      "loss/crossentropy": 2.2146100997924805,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22172697633504868,
      "step": 15824
    },
    {
      "epoch": 0.31652,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.007173411051432292,
      "learning_rate": 0.0001,
      "loss": 4.0226,
      "loss/crossentropy": 2.296873092651367,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22240595519542694,
      "step": 15826
    },
    {
      "epoch": 0.31656,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.005900065104166667,
      "learning_rate": 0.0001,
      "loss": 4.2914,
      "loss/crossentropy": 2.130843997001648,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21288827806711197,
      "step": 15828
    },
    {
      "epoch": 0.3166,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.017856597900390625,
      "learning_rate": 0.0001,
      "loss": 4.1067,
      "loss/crossentropy": 2.042823553085327,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1972874104976654,
      "step": 15830
    },
    {
      "epoch": 0.31664,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.01701838175455729,
      "learning_rate": 0.0001,
      "loss": 3.8878,
      "loss/crossentropy": 1.8040228486061096,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19943677634000778,
      "step": 15832
    },
    {
      "epoch": 0.31668,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.018400065104166665,
      "learning_rate": 0.0001,
      "loss": 4.2647,
      "loss/crossentropy": 2.085720181465149,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2592329904437065,
      "step": 15834
    },
    {
      "epoch": 0.31672,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.022334798177083334,
      "learning_rate": 0.0001,
      "loss": 3.7694,
      "loss/crossentropy": 1.9271164536476135,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1766722872853279,
      "step": 15836
    },
    {
      "epoch": 0.31676,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.022106679280598958,
      "learning_rate": 0.0001,
      "loss": 4.0742,
      "loss/crossentropy": 2.188960611820221,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19409415870904922,
      "step": 15838
    },
    {
      "epoch": 0.3168,
      "grad_norm": 2.0,
      "grad_norm_var": 0.020949045817057293,
      "learning_rate": 0.0001,
      "loss": 3.7821,
      "loss/crossentropy": 1.5765752792358398,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16517025232315063,
      "step": 15840
    },
    {
      "epoch": 0.31684,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.020804595947265626,
      "learning_rate": 0.0001,
      "loss": 4.3356,
      "loss/crossentropy": 1.9913761019706726,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20838972181081772,
      "step": 15842
    },
    {
      "epoch": 0.31688,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.034366607666015625,
      "learning_rate": 0.0001,
      "loss": 4.3525,
      "loss/crossentropy": 1.8618363738059998,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19304241985082626,
      "step": 15844
    },
    {
      "epoch": 0.31692,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.026911417643229168,
      "learning_rate": 0.0001,
      "loss": 3.8087,
      "loss/crossentropy": 1.9099775552749634,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2251715511083603,
      "step": 15846
    },
    {
      "epoch": 0.31696,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.026486968994140624,
      "learning_rate": 0.0001,
      "loss": 4.0781,
      "loss/crossentropy": 2.3534491062164307,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19920790195465088,
      "step": 15848
    },
    {
      "epoch": 0.317,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.02804743448893229,
      "learning_rate": 0.0001,
      "loss": 4.1676,
      "loss/crossentropy": 2.1890534162521362,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2140251249074936,
      "step": 15850
    },
    {
      "epoch": 0.31704,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.0298980712890625,
      "learning_rate": 0.0001,
      "loss": 3.6303,
      "loss/crossentropy": 1.880380094051361,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18094944953918457,
      "step": 15852
    },
    {
      "epoch": 0.31708,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.03880106608072917,
      "learning_rate": 0.0001,
      "loss": 4.3276,
      "loss/crossentropy": 1.8723257184028625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2101321741938591,
      "step": 15854
    },
    {
      "epoch": 0.31712,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.039582316080729166,
      "learning_rate": 0.0001,
      "loss": 4.077,
      "loss/crossentropy": 2.0442943572998047,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1947987675666809,
      "step": 15856
    },
    {
      "epoch": 0.31716,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.03819071451822917,
      "learning_rate": 0.0001,
      "loss": 4.2969,
      "loss/crossentropy": 2.0415892601013184,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2022911012172699,
      "step": 15858
    },
    {
      "epoch": 0.3172,
      "grad_norm": 1.875,
      "grad_norm_var": 0.02176513671875,
      "learning_rate": 0.0001,
      "loss": 3.8801,
      "loss/crossentropy": 1.8290876150131226,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1914537101984024,
      "step": 15860
    },
    {
      "epoch": 0.31724,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.022904459635416666,
      "learning_rate": 0.0001,
      "loss": 3.8257,
      "loss/crossentropy": 1.6169148087501526,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21117067337036133,
      "step": 15862
    },
    {
      "epoch": 0.31728,
      "grad_norm": 2.0,
      "grad_norm_var": 0.022648111979166666,
      "learning_rate": 0.0001,
      "loss": 4.2553,
      "loss/crossentropy": 2.3701289892196655,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2557762861251831,
      "step": 15864
    },
    {
      "epoch": 0.31732,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.019657135009765625,
      "learning_rate": 0.0001,
      "loss": 4.2299,
      "loss/crossentropy": 2.2519643306732178,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21390919387340546,
      "step": 15866
    },
    {
      "epoch": 0.31736,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.015409088134765625,
      "learning_rate": 0.0001,
      "loss": 4.266,
      "loss/crossentropy": 2.3287068009376526,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22211260348558426,
      "step": 15868
    },
    {
      "epoch": 0.3174,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.004401652018229166,
      "learning_rate": 0.0001,
      "loss": 3.9622,
      "loss/crossentropy": 2.144728899002075,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2148151397705078,
      "step": 15870
    },
    {
      "epoch": 0.31744,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.004847971598307291,
      "learning_rate": 0.0001,
      "loss": 3.9042,
      "loss/crossentropy": 2.120876669883728,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21692734956741333,
      "step": 15872
    },
    {
      "epoch": 0.31748,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.004703521728515625,
      "learning_rate": 0.0001,
      "loss": 3.9254,
      "loss/crossentropy": 2.2827231884002686,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20714152604341507,
      "step": 15874
    },
    {
      "epoch": 0.31752,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.004788970947265625,
      "learning_rate": 0.0001,
      "loss": 3.8742,
      "loss/crossentropy": 1.8011438846588135,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19294942915439606,
      "step": 15876
    },
    {
      "epoch": 0.31756,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.0048258463541666664,
      "learning_rate": 0.0001,
      "loss": 3.9082,
      "loss/crossentropy": 1.9359464049339294,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19418440759181976,
      "step": 15878
    },
    {
      "epoch": 0.3176,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.006371053059895834,
      "learning_rate": 0.0001,
      "loss": 4.4035,
      "loss/crossentropy": 2.220693826675415,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2583991438150406,
      "step": 15880
    },
    {
      "epoch": 0.31764,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.006254069010416667,
      "learning_rate": 0.0001,
      "loss": 3.9762,
      "loss/crossentropy": 1.7859990000724792,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1756511777639389,
      "step": 15882
    },
    {
      "epoch": 0.31768,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.00587158203125,
      "learning_rate": 0.0001,
      "loss": 3.7008,
      "loss/crossentropy": 1.9461398720741272,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18634959310293198,
      "step": 15884
    },
    {
      "epoch": 0.31772,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.010833485921223959,
      "learning_rate": 0.0001,
      "loss": 4.3732,
      "loss/crossentropy": 2.1066328287124634,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2553827613592148,
      "step": 15886
    },
    {
      "epoch": 0.31776,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.011087799072265625,
      "learning_rate": 0.0001,
      "loss": 3.8751,
      "loss/crossentropy": 1.7809287309646606,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19362767785787582,
      "step": 15888
    },
    {
      "epoch": 0.3178,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.01077880859375,
      "learning_rate": 0.0001,
      "loss": 4.0424,
      "loss/crossentropy": 1.9554992318153381,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18544971197843552,
      "step": 15890
    },
    {
      "epoch": 0.31784,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.0327392578125,
      "learning_rate": 0.0001,
      "loss": 3.9083,
      "loss/crossentropy": 1.5843109488487244,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15569136291742325,
      "step": 15892
    },
    {
      "epoch": 0.31788,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0307525634765625,
      "learning_rate": 0.0001,
      "loss": 4.1213,
      "loss/crossentropy": 2.2000880241394043,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2167157679796219,
      "step": 15894
    },
    {
      "epoch": 0.31792,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.031870269775390626,
      "learning_rate": 0.0001,
      "loss": 4.265,
      "loss/crossentropy": 2.2597590684890747,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21171507239341736,
      "step": 15896
    },
    {
      "epoch": 0.31796,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.03181330362955729,
      "learning_rate": 0.0001,
      "loss": 3.9516,
      "loss/crossentropy": 2.0513627529144287,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21095634251832962,
      "step": 15898
    },
    {
      "epoch": 0.318,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.029271443684895832,
      "learning_rate": 0.0001,
      "loss": 4.0667,
      "loss/crossentropy": 2.1822222471237183,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21117551624774933,
      "step": 15900
    },
    {
      "epoch": 0.31804,
      "grad_norm": 2.125,
      "grad_norm_var": 0.027784983317057293,
      "learning_rate": 0.0001,
      "loss": 4.3526,
      "loss/crossentropy": 1.9500460624694824,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20769241452217102,
      "step": 15902
    },
    {
      "epoch": 0.31808,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.02684326171875,
      "learning_rate": 0.0001,
      "loss": 4.2862,
      "loss/crossentropy": 2.179791212081909,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20147693157196045,
      "step": 15904
    },
    {
      "epoch": 0.31812,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.025217437744140626,
      "learning_rate": 0.0001,
      "loss": 4.0485,
      "loss/crossentropy": 1.939364731311798,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18659047782421112,
      "step": 15906
    },
    {
      "epoch": 0.31816,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0059478759765625,
      "learning_rate": 0.0001,
      "loss": 3.9782,
      "loss/crossentropy": 2.219409167766571,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19967788457870483,
      "step": 15908
    },
    {
      "epoch": 0.3182,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.007741038004557292,
      "learning_rate": 0.0001,
      "loss": 3.829,
      "loss/crossentropy": 1.8363550901412964,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18773943930864334,
      "step": 15910
    },
    {
      "epoch": 0.31824,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.006151326497395833,
      "learning_rate": 0.0001,
      "loss": 4.2196,
      "loss/crossentropy": 1.779226541519165,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20676184445619583,
      "step": 15912
    },
    {
      "epoch": 0.31828,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.006172688802083334,
      "learning_rate": 0.0001,
      "loss": 3.9532,
      "loss/crossentropy": 2.1175760626792908,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20498017221689224,
      "step": 15914
    },
    {
      "epoch": 0.31832,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.005893707275390625,
      "learning_rate": 0.0001,
      "loss": 4.1601,
      "loss/crossentropy": 2.0964329838752747,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1898122876882553,
      "step": 15916
    },
    {
      "epoch": 0.31836,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.005680084228515625,
      "learning_rate": 0.0001,
      "loss": 3.9093,
      "loss/crossentropy": 1.9687228798866272,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1931748390197754,
      "step": 15918
    },
    {
      "epoch": 0.3184,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.005197906494140625,
      "learning_rate": 0.0001,
      "loss": 4.0003,
      "loss/crossentropy": 1.8344767689704895,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19926829636096954,
      "step": 15920
    },
    {
      "epoch": 0.31844,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.004443359375,
      "learning_rate": 0.0001,
      "loss": 4.0426,
      "loss/crossentropy": 2.154146194458008,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21284651011228561,
      "step": 15922
    },
    {
      "epoch": 0.31848,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.004223378499348959,
      "learning_rate": 0.0001,
      "loss": 4.2112,
      "loss/crossentropy": 1.877986490726471,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18935512751340866,
      "step": 15924
    },
    {
      "epoch": 0.31852,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.009297434488932292,
      "learning_rate": 0.0001,
      "loss": 4.4296,
      "loss/crossentropy": 2.361938714981079,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23210977017879486,
      "step": 15926
    },
    {
      "epoch": 0.31856,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.01043701171875,
      "learning_rate": 0.0001,
      "loss": 4.1209,
      "loss/crossentropy": 2.3886083364486694,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22222661972045898,
      "step": 15928
    },
    {
      "epoch": 0.3186,
      "grad_norm": 2.25,
      "grad_norm_var": 0.014806874593098958,
      "learning_rate": 0.0001,
      "loss": 4.1164,
      "loss/crossentropy": 2.073238492012024,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20130325853824615,
      "step": 15930
    },
    {
      "epoch": 0.31864,
      "grad_norm": 2.125,
      "grad_norm_var": 0.015569814046223958,
      "learning_rate": 0.0001,
      "loss": 4.1692,
      "loss/crossentropy": 1.9937713742256165,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.199419766664505,
      "step": 15932
    },
    {
      "epoch": 0.31868,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0135894775390625,
      "learning_rate": 0.0001,
      "loss": 4.0594,
      "loss/crossentropy": 2.003056764602661,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18682510405778885,
      "step": 15934
    },
    {
      "epoch": 0.31872,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0137359619140625,
      "learning_rate": 0.0001,
      "loss": 3.9555,
      "loss/crossentropy": 1.7363762259483337,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17377281934022903,
      "step": 15936
    },
    {
      "epoch": 0.31876,
      "grad_norm": 2.0,
      "grad_norm_var": 0.014412180582682291,
      "learning_rate": 0.0001,
      "loss": 4.1321,
      "loss/crossentropy": 1.9791433811187744,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24057063460350037,
      "step": 15938
    },
    {
      "epoch": 0.3188,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.011378733317057292,
      "learning_rate": 0.0001,
      "loss": 3.9662,
      "loss/crossentropy": 1.9962583780288696,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19619113206863403,
      "step": 15940
    },
    {
      "epoch": 0.31884,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.01053466796875,
      "learning_rate": 0.0001,
      "loss": 4.2093,
      "loss/crossentropy": 1.8568945527076721,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1979837343096733,
      "step": 15942
    },
    {
      "epoch": 0.31888,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.008991495768229166,
      "learning_rate": 0.0001,
      "loss": 4.0148,
      "loss/crossentropy": 2.031070291996002,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1951117366552353,
      "step": 15944
    },
    {
      "epoch": 0.31892,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.007916005452473958,
      "learning_rate": 0.0001,
      "loss": 4.0139,
      "loss/crossentropy": 2.2226544618606567,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21066214889287949,
      "step": 15946
    },
    {
      "epoch": 0.31896,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.007783762613932292,
      "learning_rate": 0.0001,
      "loss": 4.0275,
      "loss/crossentropy": 2.090229034423828,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.225972518324852,
      "step": 15948
    },
    {
      "epoch": 0.319,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0080078125,
      "learning_rate": 0.0001,
      "loss": 3.9792,
      "loss/crossentropy": 1.8562734723091125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17633049190044403,
      "step": 15950
    },
    {
      "epoch": 0.31904,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.00992431640625,
      "learning_rate": 0.0001,
      "loss": 4.0227,
      "loss/crossentropy": 1.6677230596542358,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16521989554166794,
      "step": 15952
    },
    {
      "epoch": 0.31908,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.008939361572265625,
      "learning_rate": 0.0001,
      "loss": 4.2391,
      "loss/crossentropy": 2.1918782591819763,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21977706253528595,
      "step": 15954
    },
    {
      "epoch": 0.31912,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.008701324462890625,
      "learning_rate": 0.0001,
      "loss": 3.8932,
      "loss/crossentropy": 1.8623137474060059,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1937999278306961,
      "step": 15956
    },
    {
      "epoch": 0.31916,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.014839426676432291,
      "learning_rate": 0.0001,
      "loss": 4.2378,
      "loss/crossentropy": 2.262889266014099,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19410346448421478,
      "step": 15958
    },
    {
      "epoch": 0.3192,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.014654286702473958,
      "learning_rate": 0.0001,
      "loss": 4.0487,
      "loss/crossentropy": 1.9620846509933472,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1978064849972725,
      "step": 15960
    },
    {
      "epoch": 0.31924,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.014207967122395833,
      "learning_rate": 0.0001,
      "loss": 4.2746,
      "loss/crossentropy": 2.453945517539978,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21578459441661835,
      "step": 15962
    },
    {
      "epoch": 0.31928,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.013529459635416666,
      "learning_rate": 0.0001,
      "loss": 3.9692,
      "loss/crossentropy": 1.9565781354904175,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17789364606142044,
      "step": 15964
    },
    {
      "epoch": 0.31932,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.012589518229166667,
      "learning_rate": 0.0001,
      "loss": 4.3504,
      "loss/crossentropy": 2.1175976991653442,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2239348292350769,
      "step": 15966
    },
    {
      "epoch": 0.31936,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.010554758707682292,
      "learning_rate": 0.0001,
      "loss": 4.0815,
      "loss/crossentropy": 1.8919953107833862,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19832338392734528,
      "step": 15968
    },
    {
      "epoch": 0.3194,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0126220703125,
      "learning_rate": 0.0001,
      "loss": 4.1282,
      "loss/crossentropy": 1.9757861495018005,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19513867795467377,
      "step": 15970
    },
    {
      "epoch": 0.31944,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.012630208333333334,
      "learning_rate": 0.0001,
      "loss": 4.1045,
      "loss/crossentropy": 1.8716764450073242,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18571900576353073,
      "step": 15972
    },
    {
      "epoch": 0.31948,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.006575520833333333,
      "learning_rate": 0.0001,
      "loss": 4.2572,
      "loss/crossentropy": 2.251496374607086,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2199217826128006,
      "step": 15974
    },
    {
      "epoch": 0.31952,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.008424631754557292,
      "learning_rate": 0.0001,
      "loss": 4.1033,
      "loss/crossentropy": 1.8639826774597168,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18889226019382477,
      "step": 15976
    },
    {
      "epoch": 0.31956,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.008499908447265624,
      "learning_rate": 0.0001,
      "loss": 3.8386,
      "loss/crossentropy": 1.824375331401825,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1809096783399582,
      "step": 15978
    },
    {
      "epoch": 0.3196,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.010361480712890624,
      "learning_rate": 0.0001,
      "loss": 4.4576,
      "loss/crossentropy": 2.357889413833618,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21762560307979584,
      "step": 15980
    },
    {
      "epoch": 0.31964,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.011527252197265626,
      "learning_rate": 0.0001,
      "loss": 4.0706,
      "loss/crossentropy": 2.0944892168045044,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18413915485143661,
      "step": 15982
    },
    {
      "epoch": 0.31968,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.011736806233723958,
      "learning_rate": 0.0001,
      "loss": 3.9458,
      "loss/crossentropy": 1.8600184321403503,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18580162525177002,
      "step": 15984
    },
    {
      "epoch": 0.31972,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008697255452473959,
      "learning_rate": 0.0001,
      "loss": 4.1152,
      "loss/crossentropy": 1.9213852882385254,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20599720627069473,
      "step": 15986
    },
    {
      "epoch": 0.31976,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0119537353515625,
      "learning_rate": 0.0001,
      "loss": 4.2187,
      "loss/crossentropy": 2.0295584201812744,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2125483900308609,
      "step": 15988
    },
    {
      "epoch": 0.3198,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.012198638916015626,
      "learning_rate": 0.0001,
      "loss": 4.2448,
      "loss/crossentropy": 2.148836612701416,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22027160972356796,
      "step": 15990
    },
    {
      "epoch": 0.31984,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.010201009114583333,
      "learning_rate": 0.0001,
      "loss": 4.122,
      "loss/crossentropy": 1.9927499294281006,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20413245260715485,
      "step": 15992
    },
    {
      "epoch": 0.31988,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.0113433837890625,
      "learning_rate": 0.0001,
      "loss": 4.2675,
      "loss/crossentropy": 2.090071678161621,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21670149266719818,
      "step": 15994
    },
    {
      "epoch": 0.31992,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.010074869791666666,
      "learning_rate": 0.0001,
      "loss": 4.1333,
      "loss/crossentropy": 1.8375160098075867,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1938413679599762,
      "step": 15996
    },
    {
      "epoch": 0.31996,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.010498046875,
      "learning_rate": 0.0001,
      "loss": 3.8915,
      "loss/crossentropy": 1.5196507573127747,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16407200694084167,
      "step": 15998
    },
    {
      "epoch": 0.32,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.011344146728515626,
      "learning_rate": 0.0001,
      "loss": 4.0965,
      "loss/crossentropy": 2.0738461017608643,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21532364934682846,
      "step": 16000
    },
    {
      "epoch": 0.32004,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.011948394775390624,
      "learning_rate": 0.0001,
      "loss": 4.0687,
      "loss/crossentropy": 2.3243744373321533,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23544494807720184,
      "step": 16002
    },
    {
      "epoch": 0.32008,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.009862263997395834,
      "learning_rate": 0.0001,
      "loss": 3.9637,
      "loss/crossentropy": 1.7145931124687195,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1816524863243103,
      "step": 16004
    },
    {
      "epoch": 0.32012,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.010506184895833333,
      "learning_rate": 0.0001,
      "loss": 4.2317,
      "loss/crossentropy": 2.369017481803894,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21978877484798431,
      "step": 16006
    },
    {
      "epoch": 0.32016,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.010276031494140626,
      "learning_rate": 0.0001,
      "loss": 4.2283,
      "loss/crossentropy": 2.2246848344802856,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21983492374420166,
      "step": 16008
    },
    {
      "epoch": 0.3202,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.03774185180664062,
      "learning_rate": 0.0001,
      "loss": 4.4077,
      "loss/crossentropy": 2.0235647559165955,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22780708968639374,
      "step": 16010
    },
    {
      "epoch": 0.32024,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.03724543253580729,
      "learning_rate": 0.0001,
      "loss": 4.1692,
      "loss/crossentropy": 2.3185853958129883,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22308620065450668,
      "step": 16012
    },
    {
      "epoch": 0.32028,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0337890625,
      "learning_rate": 0.0001,
      "loss": 4.1314,
      "loss/crossentropy": 1.8504603505134583,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1909484937787056,
      "step": 16014
    },
    {
      "epoch": 0.32032,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.03502604166666667,
      "learning_rate": 0.0001,
      "loss": 3.8247,
      "loss/crossentropy": 1.7095224857330322,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18588167428970337,
      "step": 16016
    },
    {
      "epoch": 0.32036,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.03749974568684896,
      "learning_rate": 0.0001,
      "loss": 4.1015,
      "loss/crossentropy": 1.8398523330688477,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18085161596536636,
      "step": 16018
    },
    {
      "epoch": 0.3204,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.03569717407226562,
      "learning_rate": 0.0001,
      "loss": 4.1463,
      "loss/crossentropy": 2.297620415687561,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20937249064445496,
      "step": 16020
    },
    {
      "epoch": 0.32044,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.03618545532226562,
      "learning_rate": 0.0001,
      "loss": 4.3518,
      "loss/crossentropy": 1.9124106764793396,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2072141021490097,
      "step": 16022
    },
    {
      "epoch": 0.32048,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.03752415974934896,
      "learning_rate": 0.0001,
      "loss": 4.2695,
      "loss/crossentropy": 2.119450092315674,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20875975489616394,
      "step": 16024
    },
    {
      "epoch": 0.32052,
      "grad_norm": 2.125,
      "grad_norm_var": 0.008125559488932291,
      "learning_rate": 0.0001,
      "loss": 4.35,
      "loss/crossentropy": 1.722611904144287,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1751066967844963,
      "step": 16026
    },
    {
      "epoch": 0.32056,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.008760579427083333,
      "learning_rate": 0.0001,
      "loss": 4.0689,
      "loss/crossentropy": 2.246767997741699,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1999046877026558,
      "step": 16028
    },
    {
      "epoch": 0.3206,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0083404541015625,
      "learning_rate": 0.0001,
      "loss": 3.9949,
      "loss/crossentropy": 2.0449349880218506,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1969577819108963,
      "step": 16030
    },
    {
      "epoch": 0.32064,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.010050201416015625,
      "learning_rate": 0.0001,
      "loss": 4.1257,
      "loss/crossentropy": 2.125577926635742,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22252507507801056,
      "step": 16032
    },
    {
      "epoch": 0.32068,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.009079742431640624,
      "learning_rate": 0.0001,
      "loss": 3.639,
      "loss/crossentropy": 1.8161216974258423,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17801867425441742,
      "step": 16034
    },
    {
      "epoch": 0.32072,
      "grad_norm": 2.0,
      "grad_norm_var": 0.009422810872395833,
      "learning_rate": 0.0001,
      "loss": 3.9475,
      "loss/crossentropy": 2.0116894841194153,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18890849500894547,
      "step": 16036
    },
    {
      "epoch": 0.32076,
      "grad_norm": 1.875,
      "grad_norm_var": 0.0100982666015625,
      "learning_rate": 0.0001,
      "loss": 3.8274,
      "loss/crossentropy": 1.505588710308075,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17151623219251633,
      "step": 16038
    },
    {
      "epoch": 0.3208,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.008585357666015625,
      "learning_rate": 0.0001,
      "loss": 4.2162,
      "loss/crossentropy": 2.1405014991760254,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2036641389131546,
      "step": 16040
    },
    {
      "epoch": 0.32084,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.006628163655598958,
      "learning_rate": 0.0001,
      "loss": 4.0743,
      "loss/crossentropy": 2.1562809348106384,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21301954984664917,
      "step": 16042
    },
    {
      "epoch": 0.32088,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.006628163655598958,
      "learning_rate": 0.0001,
      "loss": 3.9773,
      "loss/crossentropy": 1.9317167401313782,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19167031347751617,
      "step": 16044
    },
    {
      "epoch": 0.32092,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.007330067952473958,
      "learning_rate": 0.0001,
      "loss": 4.0523,
      "loss/crossentropy": 1.6257455348968506,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1676744669675827,
      "step": 16046
    },
    {
      "epoch": 0.32096,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.006640625,
      "learning_rate": 0.0001,
      "loss": 4.2454,
      "loss/crossentropy": 2.0711347460746765,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2268730029463768,
      "step": 16048
    },
    {
      "epoch": 0.321,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0071489969889322914,
      "learning_rate": 0.0001,
      "loss": 4.3333,
      "loss/crossentropy": 2.173452377319336,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1981828361749649,
      "step": 16050
    },
    {
      "epoch": 0.32104,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.009593709309895834,
      "learning_rate": 0.0001,
      "loss": 4.4601,
      "loss/crossentropy": 2.348356604576111,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2538425847887993,
      "step": 16052
    },
    {
      "epoch": 0.32108,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0089263916015625,
      "learning_rate": 0.0001,
      "loss": 4.3408,
      "loss/crossentropy": 2.5441235303878784,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22638535499572754,
      "step": 16054
    },
    {
      "epoch": 0.32112,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.0098876953125,
      "learning_rate": 0.0001,
      "loss": 4.08,
      "loss/crossentropy": 2.129339337348938,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20430771261453629,
      "step": 16056
    },
    {
      "epoch": 0.32116,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.01142578125,
      "learning_rate": 0.0001,
      "loss": 3.9048,
      "loss/crossentropy": 2.1599501371383667,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20718347281217575,
      "step": 16058
    },
    {
      "epoch": 0.3212,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.010296376546223958,
      "learning_rate": 0.0001,
      "loss": 4.2746,
      "loss/crossentropy": 2.2704890966415405,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22025802731513977,
      "step": 16060
    },
    {
      "epoch": 0.32124,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.011702473958333333,
      "learning_rate": 0.0001,
      "loss": 3.7282,
      "loss/crossentropy": 1.7704021334648132,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17811128497123718,
      "step": 16062
    },
    {
      "epoch": 0.32128,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.015705362955729166,
      "learning_rate": 0.0001,
      "loss": 4.2366,
      "loss/crossentropy": 1.8030991554260254,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20078134536743164,
      "step": 16064
    },
    {
      "epoch": 0.32132,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0161041259765625,
      "learning_rate": 0.0001,
      "loss": 3.9832,
      "loss/crossentropy": 2.14484703540802,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21000836789608002,
      "step": 16066
    },
    {
      "epoch": 0.32136,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.016521962483723958,
      "learning_rate": 0.0001,
      "loss": 4.0812,
      "loss/crossentropy": 2.1680833101272583,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22287416458129883,
      "step": 16068
    },
    {
      "epoch": 0.3214,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.01708958943684896,
      "learning_rate": 0.0001,
      "loss": 4.1355,
      "loss/crossentropy": 1.9971441626548767,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20709602534770966,
      "step": 16070
    },
    {
      "epoch": 0.32144,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.016462198893229165,
      "learning_rate": 0.0001,
      "loss": 4.2735,
      "loss/crossentropy": 2.180538833141327,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2097538262605667,
      "step": 16072
    },
    {
      "epoch": 0.32148,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.013529459635416666,
      "learning_rate": 0.0001,
      "loss": 4.0725,
      "loss/crossentropy": 2.074296534061432,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20961299538612366,
      "step": 16074
    },
    {
      "epoch": 0.32152,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.014998372395833333,
      "learning_rate": 0.0001,
      "loss": 4.3,
      "loss/crossentropy": 2.0936360359191895,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18253444135189056,
      "step": 16076
    },
    {
      "epoch": 0.32156,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0121337890625,
      "learning_rate": 0.0001,
      "loss": 3.9992,
      "loss/crossentropy": 2.1265164613723755,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2127028927206993,
      "step": 16078
    },
    {
      "epoch": 0.3216,
      "grad_norm": 2.0,
      "grad_norm_var": 0.008410390218098958,
      "learning_rate": 0.0001,
      "loss": 4.1907,
      "loss/crossentropy": 2.0804547667503357,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23104286193847656,
      "step": 16080
    },
    {
      "epoch": 0.32164,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.007759602864583334,
      "learning_rate": 0.0001,
      "loss": 3.9653,
      "loss/crossentropy": 1.9458459615707397,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18746595829725266,
      "step": 16082
    },
    {
      "epoch": 0.32168,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0056793212890625,
      "learning_rate": 0.0001,
      "loss": 4.0581,
      "loss/crossentropy": 2.186868667602539,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20350559055805206,
      "step": 16084
    },
    {
      "epoch": 0.32172,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.007340494791666667,
      "learning_rate": 0.0001,
      "loss": 4.2364,
      "loss/crossentropy": 1.5468108654022217,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17710693180561066,
      "step": 16086
    },
    {
      "epoch": 0.32176,
      "grad_norm": 2.0,
      "grad_norm_var": 0.006180572509765625,
      "learning_rate": 0.0001,
      "loss": 4.2635,
      "loss/crossentropy": 2.222296118736267,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22016968578100204,
      "step": 16088
    },
    {
      "epoch": 0.3218,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.005619049072265625,
      "learning_rate": 0.0001,
      "loss": 4.0659,
      "loss/crossentropy": 1.7950996160507202,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1892552226781845,
      "step": 16090
    },
    {
      "epoch": 0.32184,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.007120768229166667,
      "learning_rate": 0.0001,
      "loss": 4.2899,
      "loss/crossentropy": 2.293117642402649,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2351122871041298,
      "step": 16092
    },
    {
      "epoch": 0.32188,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.006208292643229167,
      "learning_rate": 0.0001,
      "loss": 3.8235,
      "loss/crossentropy": 1.6602438688278198,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18052196502685547,
      "step": 16094
    },
    {
      "epoch": 0.32192,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.008501942952473958,
      "learning_rate": 0.0001,
      "loss": 4.114,
      "loss/crossentropy": 1.9335945844650269,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20585428178310394,
      "step": 16096
    },
    {
      "epoch": 0.32196,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.011378733317057292,
      "learning_rate": 0.0001,
      "loss": 4.3911,
      "loss/crossentropy": 2.0107831358909607,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19297368824481964,
      "step": 16098
    },
    {
      "epoch": 0.322,
      "grad_norm": 2.0,
      "grad_norm_var": 0.011801910400390626,
      "learning_rate": 0.0001,
      "loss": 4.0153,
      "loss/crossentropy": 1.9804013967514038,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17699319869279861,
      "step": 16100
    },
    {
      "epoch": 0.32204,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.01048583984375,
      "learning_rate": 0.0001,
      "loss": 4.1069,
      "loss/crossentropy": 1.9009913206100464,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1931428462266922,
      "step": 16102
    },
    {
      "epoch": 0.32208,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.011358388264973958,
      "learning_rate": 0.0001,
      "loss": 4.5134,
      "loss/crossentropy": 1.9558063745498657,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2139815017580986,
      "step": 16104
    },
    {
      "epoch": 0.32212,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.013323720296223958,
      "learning_rate": 0.0001,
      "loss": 4.0989,
      "loss/crossentropy": 2.0626373887062073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2022782266139984,
      "step": 16106
    },
    {
      "epoch": 0.32216,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.013814036051432292,
      "learning_rate": 0.0001,
      "loss": 3.9408,
      "loss/crossentropy": 2.166835308074951,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22070696204900742,
      "step": 16108
    },
    {
      "epoch": 0.3222,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.012963612874348959,
      "learning_rate": 0.0001,
      "loss": 4.1382,
      "loss/crossentropy": 2.2970025539398193,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23291311413049698,
      "step": 16110
    },
    {
      "epoch": 0.32224,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.012174224853515625,
      "learning_rate": 0.0001,
      "loss": 3.9577,
      "loss/crossentropy": 1.9400970935821533,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.199058435857296,
      "step": 16112
    },
    {
      "epoch": 0.32228,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.009110260009765624,
      "learning_rate": 0.0001,
      "loss": 3.9659,
      "loss/crossentropy": 1.6467864513397217,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19500216841697693,
      "step": 16114
    },
    {
      "epoch": 0.32232,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.010546875,
      "learning_rate": 0.0001,
      "loss": 4.1791,
      "loss/crossentropy": 2.477790355682373,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22540342807769775,
      "step": 16116
    },
    {
      "epoch": 0.32236,
      "grad_norm": 1.875,
      "grad_norm_var": 0.011066691080729166,
      "learning_rate": 0.0001,
      "loss": 4.0661,
      "loss/crossentropy": 1.946933627128601,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17866672575473785,
      "step": 16118
    },
    {
      "epoch": 0.3224,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.01153564453125,
      "learning_rate": 0.0001,
      "loss": 4.1416,
      "loss/crossentropy": 2.4484145641326904,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21314629167318344,
      "step": 16120
    },
    {
      "epoch": 0.32244,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.009155019124348959,
      "learning_rate": 0.0001,
      "loss": 3.9754,
      "loss/crossentropy": 1.9663755893707275,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19814299046993256,
      "step": 16122
    },
    {
      "epoch": 0.32248,
      "grad_norm": 2.125,
      "grad_norm_var": 0.5887278238932292,
      "learning_rate": 0.0001,
      "loss": 3.7417,
      "loss/crossentropy": 1.9786240458488464,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19666218757629395,
      "step": 16124
    },
    {
      "epoch": 0.32252,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.5901079813639323,
      "learning_rate": 0.0001,
      "loss": 3.9504,
      "loss/crossentropy": 1.90617835521698,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1803947538137436,
      "step": 16126
    },
    {
      "epoch": 0.32256,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.5877764383951823,
      "learning_rate": 0.0001,
      "loss": 4.3267,
      "loss/crossentropy": 2.1199004650115967,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21409574151039124,
      "step": 16128
    },
    {
      "epoch": 0.3226,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.5875445048014323,
      "learning_rate": 0.0001,
      "loss": 4.184,
      "loss/crossentropy": 2.081290364265442,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20103489607572556,
      "step": 16130
    },
    {
      "epoch": 0.32264,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.5902565002441407,
      "learning_rate": 0.0001,
      "loss": 4.216,
      "loss/crossentropy": 1.9839922785758972,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20679790526628494,
      "step": 16132
    },
    {
      "epoch": 0.32268,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.5830800374348958,
      "learning_rate": 0.0001,
      "loss": 4.3441,
      "loss/crossentropy": 2.2765486240386963,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21012597531080246,
      "step": 16134
    },
    {
      "epoch": 0.32272,
      "grad_norm": 2.125,
      "grad_norm_var": 0.57388916015625,
      "learning_rate": 0.0001,
      "loss": 4.0003,
      "loss/crossentropy": 1.8353520035743713,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19367212802171707,
      "step": 16136
    },
    {
      "epoch": 0.32276,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.5781572977701823,
      "learning_rate": 0.0001,
      "loss": 3.8104,
      "loss/crossentropy": 1.8009640574455261,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18030469119548798,
      "step": 16138
    },
    {
      "epoch": 0.3228,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.020975494384765626,
      "learning_rate": 0.0001,
      "loss": 4.1914,
      "loss/crossentropy": 1.865959644317627,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20606407523155212,
      "step": 16140
    },
    {
      "epoch": 0.32284,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.021445465087890626,
      "learning_rate": 0.0001,
      "loss": 4.0253,
      "loss/crossentropy": 2.238165020942688,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22822897136211395,
      "step": 16142
    },
    {
      "epoch": 0.32288,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.02393366495768229,
      "learning_rate": 0.0001,
      "loss": 4.0425,
      "loss/crossentropy": 2.175424814224243,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2260037586092949,
      "step": 16144
    },
    {
      "epoch": 0.32292,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.025050608317057292,
      "learning_rate": 0.0001,
      "loss": 4.1493,
      "loss/crossentropy": 2.2071722745895386,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23026321083307266,
      "step": 16146
    },
    {
      "epoch": 0.32296,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.024468739827473957,
      "learning_rate": 0.0001,
      "loss": 4.2384,
      "loss/crossentropy": 2.211812973022461,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20882735401391983,
      "step": 16148
    },
    {
      "epoch": 0.323,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.02444636027018229,
      "learning_rate": 0.0001,
      "loss": 4.0356,
      "loss/crossentropy": 1.6503818035125732,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1640765517950058,
      "step": 16150
    },
    {
      "epoch": 0.32304,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.02520726521809896,
      "learning_rate": 0.0001,
      "loss": 4.0692,
      "loss/crossentropy": 1.9449399709701538,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1876515969634056,
      "step": 16152
    },
    {
      "epoch": 0.32308,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.022581990559895834,
      "learning_rate": 0.0001,
      "loss": 4.1665,
      "loss/crossentropy": 2.001387894153595,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21191075444221497,
      "step": 16154
    },
    {
      "epoch": 0.32312,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.0143218994140625,
      "learning_rate": 0.0001,
      "loss": 4.2115,
      "loss/crossentropy": 2.1146541833877563,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2692951038479805,
      "step": 16156
    },
    {
      "epoch": 0.32316,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.015602366129557291,
      "learning_rate": 0.0001,
      "loss": 4.1534,
      "loss/crossentropy": 2.0765844583511353,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19518104195594788,
      "step": 16158
    },
    {
      "epoch": 0.3232,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0128814697265625,
      "learning_rate": 0.0001,
      "loss": 4.1432,
      "loss/crossentropy": 2.5394046306610107,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22651594877243042,
      "step": 16160
    },
    {
      "epoch": 0.32324,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.011702473958333333,
      "learning_rate": 0.0001,
      "loss": 4.0412,
      "loss/crossentropy": 2.0140721797943115,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18245946615934372,
      "step": 16162
    },
    {
      "epoch": 0.32328,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.012303670247395834,
      "learning_rate": 0.0001,
      "loss": 4.3029,
      "loss/crossentropy": 2.472638249397278,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23127944022417068,
      "step": 16164
    },
    {
      "epoch": 0.32332,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.014924112955729167,
      "learning_rate": 0.0001,
      "loss": 4.1974,
      "loss/crossentropy": 2.1551883220672607,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2093031033873558,
      "step": 16166
    },
    {
      "epoch": 0.32336,
      "grad_norm": 2.25,
      "grad_norm_var": 0.01641845703125,
      "learning_rate": 0.0001,
      "loss": 4.1221,
      "loss/crossentropy": 2.042950928211212,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2049141526222229,
      "step": 16168
    },
    {
      "epoch": 0.3234,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.014849599202473958,
      "learning_rate": 0.0001,
      "loss": 4.4795,
      "loss/crossentropy": 2.0262559056282043,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18649922311306,
      "step": 16170
    },
    {
      "epoch": 0.32344,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.011675771077473958,
      "learning_rate": 0.0001,
      "loss": 4.0477,
      "loss/crossentropy": 2.3014276027679443,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21770215779542923,
      "step": 16172
    },
    {
      "epoch": 0.32348,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.009349568684895834,
      "learning_rate": 0.0001,
      "loss": 4.2802,
      "loss/crossentropy": 2.213370680809021,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18508044630289078,
      "step": 16174
    },
    {
      "epoch": 0.32352,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009075673421223958,
      "learning_rate": 0.0001,
      "loss": 4.2529,
      "loss/crossentropy": 1.973215639591217,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1937488242983818,
      "step": 16176
    },
    {
      "epoch": 0.32356,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.018553670247395834,
      "learning_rate": 0.0001,
      "loss": 4.3534,
      "loss/crossentropy": 2.1585731506347656,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2029733881354332,
      "step": 16178
    },
    {
      "epoch": 0.3236,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.021647135416666668,
      "learning_rate": 0.0001,
      "loss": 4.3324,
      "loss/crossentropy": 2.3878796100616455,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19985762238502502,
      "step": 16180
    },
    {
      "epoch": 0.32364,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.027860514322916665,
      "learning_rate": 0.0001,
      "loss": 3.8043,
      "loss/crossentropy": 2.07839834690094,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1994347795844078,
      "step": 16182
    },
    {
      "epoch": 0.32368,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.02607421875,
      "learning_rate": 0.0001,
      "loss": 4.3147,
      "loss/crossentropy": 2.1800806522369385,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21444538235664368,
      "step": 16184
    },
    {
      "epoch": 0.32372,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.029352823893229168,
      "learning_rate": 0.0001,
      "loss": 3.9927,
      "loss/crossentropy": 2.0900917053222656,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20476030558347702,
      "step": 16186
    },
    {
      "epoch": 0.32376,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.02899169921875,
      "learning_rate": 0.0001,
      "loss": 4.209,
      "loss/crossentropy": 2.1523303985595703,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20984850078821182,
      "step": 16188
    },
    {
      "epoch": 0.3238,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.03593648274739583,
      "learning_rate": 0.0001,
      "loss": 3.799,
      "loss/crossentropy": 1.8808646202087402,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18587414175271988,
      "step": 16190
    },
    {
      "epoch": 0.32384,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.03593648274739583,
      "learning_rate": 0.0001,
      "loss": 4.1002,
      "loss/crossentropy": 1.981968104839325,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20611849427223206,
      "step": 16192
    },
    {
      "epoch": 0.32388,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.020951334635416666,
      "learning_rate": 0.0001,
      "loss": 4.2811,
      "loss/crossentropy": 2.1454352140426636,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21696189790964127,
      "step": 16194
    },
    {
      "epoch": 0.32392,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.011905924479166666,
      "learning_rate": 0.0001,
      "loss": 4.159,
      "loss/crossentropy": 2.026822328567505,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2112257331609726,
      "step": 16196
    },
    {
      "epoch": 0.32396,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.009706370035807292,
      "learning_rate": 0.0001,
      "loss": 4.0838,
      "loss/crossentropy": 2.332329034805298,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20129209011793137,
      "step": 16198
    },
    {
      "epoch": 0.324,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.007260894775390625,
      "learning_rate": 0.0001,
      "loss": 4.11,
      "loss/crossentropy": 2.271657705307007,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2460309863090515,
      "step": 16200
    },
    {
      "epoch": 0.32404,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0064084370930989586,
      "learning_rate": 0.0001,
      "loss": 4.0835,
      "loss/crossentropy": 2.0890920162200928,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2132192626595497,
      "step": 16202
    },
    {
      "epoch": 0.32408,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.007226308186848958,
      "learning_rate": 0.0001,
      "loss": 4.0271,
      "loss/crossentropy": 1.8368538618087769,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1891050636768341,
      "step": 16204
    },
    {
      "epoch": 0.32412,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.004375966389973959,
      "learning_rate": 0.0001,
      "loss": 4.2859,
      "loss/crossentropy": 2.3123772144317627,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22660605609416962,
      "step": 16206
    },
    {
      "epoch": 0.32416,
      "grad_norm": 2.125,
      "grad_norm_var": 0.005411529541015625,
      "learning_rate": 0.0001,
      "loss": 4.0989,
      "loss/crossentropy": 2.21794331073761,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21767260879278183,
      "step": 16208
    },
    {
      "epoch": 0.3242,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.010591379801432292,
      "learning_rate": 0.0001,
      "loss": 4.5407,
      "loss/crossentropy": 2.169134736061096,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2172672376036644,
      "step": 16210
    },
    {
      "epoch": 0.32424,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.011331939697265625,
      "learning_rate": 0.0001,
      "loss": 4.0327,
      "loss/crossentropy": 2.2830699682235718,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22920189797878265,
      "step": 16212
    },
    {
      "epoch": 0.32428,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.012770334879557291,
      "learning_rate": 0.0001,
      "loss": 3.9278,
      "loss/crossentropy": 2.024571657180786,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21096136420965195,
      "step": 16214
    },
    {
      "epoch": 0.32432,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.013057200113932292,
      "learning_rate": 0.0001,
      "loss": 4.0244,
      "loss/crossentropy": 2.060720980167389,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21043655276298523,
      "step": 16216
    },
    {
      "epoch": 0.32436,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.014134724934895834,
      "learning_rate": 0.0001,
      "loss": 4.074,
      "loss/crossentropy": 2.0332056283950806,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19506454467773438,
      "step": 16218
    },
    {
      "epoch": 0.3244,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.014021555582682291,
      "learning_rate": 0.0001,
      "loss": 3.8173,
      "loss/crossentropy": 1.811396062374115,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19359800219535828,
      "step": 16220
    },
    {
      "epoch": 0.32444,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.013117472330729166,
      "learning_rate": 0.0001,
      "loss": 4.1846,
      "loss/crossentropy": 2.074379801750183,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21763048321008682,
      "step": 16222
    },
    {
      "epoch": 0.32448,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.014682769775390625,
      "learning_rate": 0.0001,
      "loss": 4.1984,
      "loss/crossentropy": 2.1836538314819336,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20814575254917145,
      "step": 16224
    },
    {
      "epoch": 0.32452,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.010794830322265626,
      "learning_rate": 0.0001,
      "loss": 4.2866,
      "loss/crossentropy": 2.365849494934082,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24219170212745667,
      "step": 16226
    },
    {
      "epoch": 0.32456,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.010343424479166667,
      "learning_rate": 0.0001,
      "loss": 4.1976,
      "loss/crossentropy": 1.8738153576850891,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22152718901634216,
      "step": 16228
    },
    {
      "epoch": 0.3246,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.010209147135416667,
      "learning_rate": 0.0001,
      "loss": 4.1458,
      "loss/crossentropy": 2.2568124532699585,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21668314933776855,
      "step": 16230
    },
    {
      "epoch": 0.32464,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.010900624593098958,
      "learning_rate": 0.0001,
      "loss": 4.2889,
      "loss/crossentropy": 2.4037723541259766,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2225075364112854,
      "step": 16232
    },
    {
      "epoch": 0.32468,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.01375732421875,
      "learning_rate": 0.0001,
      "loss": 3.7288,
      "loss/crossentropy": 2.2305864095687866,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19381006807088852,
      "step": 16234
    },
    {
      "epoch": 0.32472,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.013862864176432291,
      "learning_rate": 0.0001,
      "loss": 4.186,
      "loss/crossentropy": 2.1115931272506714,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20033665746450424,
      "step": 16236
    },
    {
      "epoch": 0.32476,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.014054361979166667,
      "learning_rate": 0.0001,
      "loss": 4.1602,
      "loss/crossentropy": 2.0450429916381836,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2080780565738678,
      "step": 16238
    },
    {
      "epoch": 0.3248,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.011888631184895833,
      "learning_rate": 0.0001,
      "loss": 4.056,
      "loss/crossentropy": 2.003769636154175,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20560404658317566,
      "step": 16240
    },
    {
      "epoch": 0.32484,
      "grad_norm": 2.125,
      "grad_norm_var": 0.010383860270182291,
      "learning_rate": 0.0001,
      "loss": 4.3504,
      "loss/crossentropy": 1.9452934265136719,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2063707485795021,
      "step": 16242
    },
    {
      "epoch": 0.32488,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.012497711181640624,
      "learning_rate": 0.0001,
      "loss": 3.9172,
      "loss/crossentropy": 2.1412742137908936,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20292682200670242,
      "step": 16244
    },
    {
      "epoch": 0.32492,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.011140950520833333,
      "learning_rate": 0.0001,
      "loss": 4.3025,
      "loss/crossentropy": 2.2285404205322266,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22571790218353271,
      "step": 16246
    },
    {
      "epoch": 0.32496,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.010713704427083333,
      "learning_rate": 0.0001,
      "loss": 3.6582,
      "loss/crossentropy": 1.82416570186615,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1771545708179474,
      "step": 16248
    },
    {
      "epoch": 0.325,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.010422515869140624,
      "learning_rate": 0.0001,
      "loss": 4.2533,
      "loss/crossentropy": 2.1432559490203857,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20037438720464706,
      "step": 16250
    },
    {
      "epoch": 0.32504,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.015337880452473958,
      "learning_rate": 0.0001,
      "loss": 4.162,
      "loss/crossentropy": 2.1962517499923706,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22743911296129227,
      "step": 16252
    },
    {
      "epoch": 0.32508,
      "grad_norm": 2.125,
      "grad_norm_var": 0.01636530558268229,
      "learning_rate": 0.0001,
      "loss": 4.2223,
      "loss/crossentropy": 1.932455599308014,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2043299823999405,
      "step": 16254
    },
    {
      "epoch": 0.32512,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.015824127197265624,
      "learning_rate": 0.0001,
      "loss": 3.9024,
      "loss/crossentropy": 1.9702956676483154,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19528628885746002,
      "step": 16256
    },
    {
      "epoch": 0.32516,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.020475260416666665,
      "learning_rate": 0.0001,
      "loss": 4.0628,
      "loss/crossentropy": 2.0042436718940735,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19861597567796707,
      "step": 16258
    },
    {
      "epoch": 0.3252,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.019573720296223958,
      "learning_rate": 0.0001,
      "loss": 4.1876,
      "loss/crossentropy": 2.3544296622276306,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22385136783123016,
      "step": 16260
    },
    {
      "epoch": 0.32524,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.01942723592122396,
      "learning_rate": 0.0001,
      "loss": 4.1435,
      "loss/crossentropy": 2.1482596397399902,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21662700176239014,
      "step": 16262
    },
    {
      "epoch": 0.32528,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.01395263671875,
      "learning_rate": 0.0001,
      "loss": 4.3441,
      "loss/crossentropy": 2.243022322654724,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22729884833097458,
      "step": 16264
    },
    {
      "epoch": 0.32532,
      "grad_norm": 1.875,
      "grad_norm_var": 0.018656158447265626,
      "learning_rate": 0.0001,
      "loss": 3.9013,
      "loss/crossentropy": 2.2499197721481323,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20558901876211166,
      "step": 16266
    },
    {
      "epoch": 0.32536,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.022907511393229166,
      "learning_rate": 0.0001,
      "loss": 4.2313,
      "loss/crossentropy": 1.6232356429100037,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18864237517118454,
      "step": 16268
    },
    {
      "epoch": 0.3254,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.02402521769205729,
      "learning_rate": 0.0001,
      "loss": 4.0435,
      "loss/crossentropy": 1.559517502784729,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1704516038298607,
      "step": 16270
    },
    {
      "epoch": 0.32544,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.02484308878580729,
      "learning_rate": 0.0001,
      "loss": 3.9018,
      "loss/crossentropy": 1.7346046566963196,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17380736768245697,
      "step": 16272
    },
    {
      "epoch": 0.32548,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.02237523396809896,
      "learning_rate": 0.0001,
      "loss": 4.2894,
      "loss/crossentropy": 2.1451315879821777,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21778792887926102,
      "step": 16274
    },
    {
      "epoch": 0.32552,
      "grad_norm": 2.25,
      "grad_norm_var": 0.022684733072916668,
      "learning_rate": 0.0001,
      "loss": 3.8737,
      "loss/crossentropy": 1.8013367056846619,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18631768971681595,
      "step": 16276
    },
    {
      "epoch": 0.32556,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.025655110677083332,
      "learning_rate": 0.0001,
      "loss": 3.8573,
      "loss/crossentropy": 1.8858280181884766,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1940329447388649,
      "step": 16278
    },
    {
      "epoch": 0.3256,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0259429931640625,
      "learning_rate": 0.0001,
      "loss": 4.1268,
      "loss/crossentropy": 1.8475046157836914,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18596403300762177,
      "step": 16280
    },
    {
      "epoch": 0.32564,
      "grad_norm": 2.0,
      "grad_norm_var": 0.023636881510416666,
      "learning_rate": 0.0001,
      "loss": 4.2602,
      "loss/crossentropy": 2.2585566639900208,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22430463135242462,
      "step": 16282
    },
    {
      "epoch": 0.32568,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.015900675455729166,
      "learning_rate": 0.0001,
      "loss": 3.7224,
      "loss/crossentropy": 1.7233783602714539,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17318468540906906,
      "step": 16284
    },
    {
      "epoch": 0.32572,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.022172037760416666,
      "learning_rate": 0.0001,
      "loss": 4.3042,
      "loss/crossentropy": 2.174088716506958,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21329404413700104,
      "step": 16286
    },
    {
      "epoch": 0.32576,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.022345987955729167,
      "learning_rate": 0.0001,
      "loss": 3.8932,
      "loss/crossentropy": 1.8016705513000488,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20395102351903915,
      "step": 16288
    },
    {
      "epoch": 0.3258,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.020182037353515626,
      "learning_rate": 0.0001,
      "loss": 3.8634,
      "loss/crossentropy": 2.166901111602783,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21237251162528992,
      "step": 16290
    },
    {
      "epoch": 0.32584,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.014766184488932292,
      "learning_rate": 0.0001,
      "loss": 4.07,
      "loss/crossentropy": 2.024773359298706,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1973796784877777,
      "step": 16292
    },
    {
      "epoch": 0.32588,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.015466054280598959,
      "learning_rate": 0.0001,
      "loss": 4.173,
      "loss/crossentropy": 2.457033157348633,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21896487474441528,
      "step": 16294
    },
    {
      "epoch": 0.32592,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.014180501302083334,
      "learning_rate": 0.0001,
      "loss": 4.11,
      "loss/crossentropy": 2.4872595071792603,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2348102256655693,
      "step": 16296
    },
    {
      "epoch": 0.32596,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.014249420166015625,
      "learning_rate": 0.0001,
      "loss": 4.0756,
      "loss/crossentropy": 2.110987067222595,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21079359203577042,
      "step": 16298
    },
    {
      "epoch": 0.326,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0133941650390625,
      "learning_rate": 0.0001,
      "loss": 4.2334,
      "loss/crossentropy": 2.2166510820388794,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19656993448734283,
      "step": 16300
    },
    {
      "epoch": 0.32604,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.008125813802083333,
      "learning_rate": 0.0001,
      "loss": 4.0212,
      "loss/crossentropy": 2.014153838157654,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1956234946846962,
      "step": 16302
    },
    {
      "epoch": 0.32608,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.03588231404622396,
      "learning_rate": 0.0001,
      "loss": 4.3045,
      "loss/crossentropy": 2.298485517501831,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21710477769374847,
      "step": 16304
    },
    {
      "epoch": 0.32612,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.033599599202473955,
      "learning_rate": 0.0001,
      "loss": 4.3134,
      "loss/crossentropy": 2.233761191368103,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21609390527009964,
      "step": 16306
    },
    {
      "epoch": 0.32616,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.03675918579101563,
      "learning_rate": 0.0001,
      "loss": 3.9141,
      "loss/crossentropy": 1.9988956451416016,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1924634352326393,
      "step": 16308
    },
    {
      "epoch": 0.3262,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.035194651285807295,
      "learning_rate": 0.0001,
      "loss": 3.9726,
      "loss/crossentropy": 1.9185590147972107,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20295168459415436,
      "step": 16310
    },
    {
      "epoch": 0.32624,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.03476155598958333,
      "learning_rate": 0.0001,
      "loss": 4.0805,
      "loss/crossentropy": 2.219870448112488,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21425887942314148,
      "step": 16312
    },
    {
      "epoch": 0.32628,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.03612442016601562,
      "learning_rate": 0.0001,
      "loss": 4.5119,
      "loss/crossentropy": 2.219307541847229,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22380203753709793,
      "step": 16314
    },
    {
      "epoch": 0.32632,
      "grad_norm": 1.875,
      "grad_norm_var": 0.03746515909830729,
      "learning_rate": 0.0001,
      "loss": 3.9652,
      "loss/crossentropy": 1.7907224893569946,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18037232011556625,
      "step": 16316
    },
    {
      "epoch": 0.32636,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.03876927693684896,
      "learning_rate": 0.0001,
      "loss": 4.1337,
      "loss/crossentropy": 1.9702708721160889,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18266448378562927,
      "step": 16318
    },
    {
      "epoch": 0.3264,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.012481435139973959,
      "learning_rate": 0.0001,
      "loss": 4.0551,
      "loss/crossentropy": 2.1748844385147095,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20725325495004654,
      "step": 16320
    },
    {
      "epoch": 0.32644,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0119781494140625,
      "learning_rate": 0.0001,
      "loss": 4.2882,
      "loss/crossentropy": 2.2458006143569946,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21462788432836533,
      "step": 16322
    },
    {
      "epoch": 0.32648,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.009276326497395833,
      "learning_rate": 0.0001,
      "loss": 4.1231,
      "loss/crossentropy": 1.923665463924408,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.223393976688385,
      "step": 16324
    },
    {
      "epoch": 0.32652,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.010129547119140625,
      "learning_rate": 0.0001,
      "loss": 4.1643,
      "loss/crossentropy": 2.36092209815979,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22316965460777283,
      "step": 16326
    },
    {
      "epoch": 0.32656,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.010456339518229166,
      "learning_rate": 0.0001,
      "loss": 4.0778,
      "loss/crossentropy": 1.7765440344810486,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1736016422510147,
      "step": 16328
    },
    {
      "epoch": 0.3266,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.006638336181640625,
      "learning_rate": 0.0001,
      "loss": 4.4885,
      "loss/crossentropy": 2.2588138580322266,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21762139350175858,
      "step": 16330
    },
    {
      "epoch": 0.32664,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0064084370930989586,
      "learning_rate": 0.0001,
      "loss": 4.2476,
      "loss/crossentropy": 2.177221417427063,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19594799727201462,
      "step": 16332
    },
    {
      "epoch": 0.32668,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0051422119140625,
      "learning_rate": 0.0001,
      "loss": 3.941,
      "loss/crossentropy": 1.8424673676490784,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19562739878892899,
      "step": 16334
    },
    {
      "epoch": 0.32672,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.006473541259765625,
      "learning_rate": 0.0001,
      "loss": 3.861,
      "loss/crossentropy": 1.8677524328231812,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19708558171987534,
      "step": 16336
    },
    {
      "epoch": 0.32676,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.007081858317057292,
      "learning_rate": 0.0001,
      "loss": 3.9222,
      "loss/crossentropy": 1.9559081196784973,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19585052132606506,
      "step": 16338
    },
    {
      "epoch": 0.3268,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.006892649332682291,
      "learning_rate": 0.0001,
      "loss": 4.144,
      "loss/crossentropy": 1.979565978050232,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2045053392648697,
      "step": 16340
    },
    {
      "epoch": 0.32684,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.00714111328125,
      "learning_rate": 0.0001,
      "loss": 4.2517,
      "loss/crossentropy": 2.296669840812683,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.207671657204628,
      "step": 16342
    },
    {
      "epoch": 0.32688,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0065305074055989586,
      "learning_rate": 0.0001,
      "loss": 4.0279,
      "loss/crossentropy": 1.9962339401245117,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19833384454250336,
      "step": 16344
    },
    {
      "epoch": 0.32692,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.006078084309895833,
      "learning_rate": 0.0001,
      "loss": 4.1779,
      "loss/crossentropy": 2.020018517971039,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19721734523773193,
      "step": 16346
    },
    {
      "epoch": 0.32696,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.006815592447916667,
      "learning_rate": 0.0001,
      "loss": 3.7614,
      "loss/crossentropy": 2.112026810646057,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20460353046655655,
      "step": 16348
    },
    {
      "epoch": 0.327,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.006540679931640625,
      "learning_rate": 0.0001,
      "loss": 4.1509,
      "loss/crossentropy": 2.2069387435913086,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2233232483267784,
      "step": 16350
    },
    {
      "epoch": 0.32704,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.004792277018229167,
      "learning_rate": 0.0001,
      "loss": 3.8675,
      "loss/crossentropy": 1.9365113377571106,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19625376164913177,
      "step": 16352
    },
    {
      "epoch": 0.32708,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0052154541015625,
      "learning_rate": 0.0001,
      "loss": 4.1831,
      "loss/crossentropy": 2.2316179871559143,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22712747752666473,
      "step": 16354
    },
    {
      "epoch": 0.32712,
      "grad_norm": 2.0,
      "grad_norm_var": 0.035672760009765624,
      "learning_rate": 0.0001,
      "loss": 3.9148,
      "loss/crossentropy": 1.801742434501648,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1831187978386879,
      "step": 16356
    },
    {
      "epoch": 0.32716,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.03590087890625,
      "learning_rate": 0.0001,
      "loss": 3.9819,
      "loss/crossentropy": 2.157936453819275,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21520215272903442,
      "step": 16358
    },
    {
      "epoch": 0.3272,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.035835520426432295,
      "learning_rate": 0.0001,
      "loss": 4.0743,
      "loss/crossentropy": 1.6054654717445374,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1769917830824852,
      "step": 16360
    },
    {
      "epoch": 0.32724,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.039613596598307294,
      "learning_rate": 0.0001,
      "loss": 4.0751,
      "loss/crossentropy": 2.0573782324790955,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22550390660762787,
      "step": 16362
    },
    {
      "epoch": 0.32728,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.035131581624348956,
      "learning_rate": 0.0001,
      "loss": 4.281,
      "loss/crossentropy": 2.0643117427825928,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21060281991958618,
      "step": 16364
    },
    {
      "epoch": 0.32732,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.042525227864583334,
      "learning_rate": 0.0001,
      "loss": 4.0884,
      "loss/crossentropy": 2.1980225443840027,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19246292114257812,
      "step": 16366
    },
    {
      "epoch": 0.32736,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.042577107747395836,
      "learning_rate": 0.0001,
      "loss": 4.05,
      "loss/crossentropy": 1.8917307257652283,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17911095917224884,
      "step": 16368
    },
    {
      "epoch": 0.3274,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.0429595947265625,
      "learning_rate": 0.0001,
      "loss": 4.1415,
      "loss/crossentropy": 2.0520957708358765,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19265015423297882,
      "step": 16370
    },
    {
      "epoch": 0.32744,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.019510904947916668,
      "learning_rate": 0.0001,
      "loss": 4.3026,
      "loss/crossentropy": 2.4500341415405273,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21713324636220932,
      "step": 16372
    },
    {
      "epoch": 0.32748,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.02032648722330729,
      "learning_rate": 0.0001,
      "loss": 4.0754,
      "loss/crossentropy": 1.8100037574768066,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16967355459928513,
      "step": 16374
    },
    {
      "epoch": 0.32752,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.021174875895182292,
      "learning_rate": 0.0001,
      "loss": 4.0244,
      "loss/crossentropy": 2.092597723007202,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1998964473605156,
      "step": 16376
    },
    {
      "epoch": 0.32756,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.016076405843098957,
      "learning_rate": 0.0001,
      "loss": 4.3627,
      "loss/crossentropy": 2.151292622089386,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19120946526527405,
      "step": 16378
    },
    {
      "epoch": 0.3276,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.020906575520833335,
      "learning_rate": 0.0001,
      "loss": 3.6949,
      "loss/crossentropy": 1.7384315729141235,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18261852860450745,
      "step": 16380
    },
    {
      "epoch": 0.32764,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.019147745768229165,
      "learning_rate": 0.0001,
      "loss": 3.9341,
      "loss/crossentropy": 1.662436068058014,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16703163087368011,
      "step": 16382
    },
    {
      "epoch": 0.32768,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.018871053059895834,
      "learning_rate": 0.0001,
      "loss": 4.3202,
      "loss/crossentropy": 2.2057151794433594,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20989537239074707,
      "step": 16384
    },
    {
      "epoch": 0.32772,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.068359375,
      "learning_rate": 0.0001,
      "loss": 4.3033,
      "loss/crossentropy": 2.1689382791519165,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20804189145565033,
      "step": 16386
    },
    {
      "epoch": 0.32776,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.06450169881184896,
      "learning_rate": 0.0001,
      "loss": 4.2241,
      "loss/crossentropy": 2.3121412992477417,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20552029460668564,
      "step": 16388
    },
    {
      "epoch": 0.3278,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.06318333943684896,
      "learning_rate": 0.0001,
      "loss": 4.1396,
      "loss/crossentropy": 2.0784353017807007,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20427466928958893,
      "step": 16390
    },
    {
      "epoch": 0.32784,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.06282145182291667,
      "learning_rate": 0.0001,
      "loss": 4.1132,
      "loss/crossentropy": 2.147130608558655,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2181989550590515,
      "step": 16392
    },
    {
      "epoch": 0.32788,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.06468505859375,
      "learning_rate": 0.0001,
      "loss": 3.812,
      "loss/crossentropy": 1.7986091375350952,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18294771760702133,
      "step": 16394
    },
    {
      "epoch": 0.32792,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.059004720052083334,
      "learning_rate": 0.0001,
      "loss": 4.037,
      "loss/crossentropy": 2.0563793778419495,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19180169701576233,
      "step": 16396
    },
    {
      "epoch": 0.32796,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.05506591796875,
      "learning_rate": 0.0001,
      "loss": 4.3111,
      "loss/crossentropy": 2.462133765220642,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21563740074634552,
      "step": 16398
    },
    {
      "epoch": 0.328,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.059427897135416664,
      "learning_rate": 0.0001,
      "loss": 3.9743,
      "loss/crossentropy": 1.8900938630104065,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1845797374844551,
      "step": 16400
    },
    {
      "epoch": 0.32804,
      "grad_norm": 2.375,
      "grad_norm_var": 0.0195068359375,
      "learning_rate": 0.0001,
      "loss": 4.1653,
      "loss/crossentropy": 2.0627527832984924,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20268447697162628,
      "step": 16402
    },
    {
      "epoch": 0.32808,
      "grad_norm": 2.125,
      "grad_norm_var": 0.030147043863932292,
      "learning_rate": 0.0001,
      "loss": 4.3042,
      "loss/crossentropy": 2.058929443359375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19984640181064606,
      "step": 16404
    },
    {
      "epoch": 0.32812,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.030564117431640624,
      "learning_rate": 0.0001,
      "loss": 4.1162,
      "loss/crossentropy": 1.816510558128357,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1754368245601654,
      "step": 16406
    },
    {
      "epoch": 0.32816,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0302886962890625,
      "learning_rate": 0.0001,
      "loss": 4.263,
      "loss/crossentropy": 1.9079806208610535,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18917685747146606,
      "step": 16408
    },
    {
      "epoch": 0.3282,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.028612263997395835,
      "learning_rate": 0.0001,
      "loss": 3.9264,
      "loss/crossentropy": 2.0913302898406982,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2088220864534378,
      "step": 16410
    },
    {
      "epoch": 0.32824,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.028547159830729165,
      "learning_rate": 0.0001,
      "loss": 3.9029,
      "loss/crossentropy": 2.1609301567077637,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2010297030210495,
      "step": 16412
    },
    {
      "epoch": 0.32828,
      "grad_norm": 3.078125,
      "grad_norm_var": 0.0982666015625,
      "learning_rate": 0.0001,
      "loss": 4.2563,
      "loss/crossentropy": 1.8449677228927612,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23620254546403885,
      "step": 16414
    },
    {
      "epoch": 0.32832,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.09134699503580729,
      "learning_rate": 0.0001,
      "loss": 4.1053,
      "loss/crossentropy": 2.059410274028778,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20259736478328705,
      "step": 16416
    },
    {
      "epoch": 0.32836,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.08761571248372396,
      "learning_rate": 0.0001,
      "loss": 3.8718,
      "loss/crossentropy": 1.6095005869865417,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17026344686746597,
      "step": 16418
    },
    {
      "epoch": 0.3284,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.07932510375976562,
      "learning_rate": 0.0001,
      "loss": 4.2623,
      "loss/crossentropy": 2.200170636177063,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2259170413017273,
      "step": 16420
    },
    {
      "epoch": 0.32844,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.08116226196289063,
      "learning_rate": 0.0001,
      "loss": 4.0724,
      "loss/crossentropy": 1.970030963420868,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19053932279348373,
      "step": 16422
    },
    {
      "epoch": 0.32848,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.08263753255208334,
      "learning_rate": 0.0001,
      "loss": 4.106,
      "loss/crossentropy": 2.2675901651382446,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22277145832777023,
      "step": 16424
    },
    {
      "epoch": 0.32852,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.08910725911458334,
      "learning_rate": 0.0001,
      "loss": 3.723,
      "loss/crossentropy": 1.6446372866630554,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15234287828207016,
      "step": 16426
    },
    {
      "epoch": 0.32856,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0875689188639323,
      "learning_rate": 0.0001,
      "loss": 4.3421,
      "loss/crossentropy": 2.350424647331238,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2190476581454277,
      "step": 16428
    },
    {
      "epoch": 0.3286,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.009279123942057292,
      "learning_rate": 0.0001,
      "loss": 3.9663,
      "loss/crossentropy": 1.8809763193130493,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19404269009828568,
      "step": 16430
    },
    {
      "epoch": 0.32864,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.008257802327473958,
      "learning_rate": 0.0001,
      "loss": 4.1731,
      "loss/crossentropy": 2.1159849166870117,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21232837438583374,
      "step": 16432
    },
    {
      "epoch": 0.32868,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.02133763631184896,
      "learning_rate": 0.0001,
      "loss": 4.2875,
      "loss/crossentropy": 2.3252066373825073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24867044389247894,
      "step": 16434
    },
    {
      "epoch": 0.32872,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.020186360677083334,
      "learning_rate": 0.0001,
      "loss": 4.2462,
      "loss/crossentropy": 2.0642696619033813,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18409788608551025,
      "step": 16436
    },
    {
      "epoch": 0.32876,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.022725423177083332,
      "learning_rate": 0.0001,
      "loss": 4.3832,
      "loss/crossentropy": 2.359580874443054,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.218842551112175,
      "step": 16438
    },
    {
      "epoch": 0.3288,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.02264378865559896,
      "learning_rate": 0.0001,
      "loss": 3.7996,
      "loss/crossentropy": 2.0959436893463135,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1993689462542534,
      "step": 16440
    },
    {
      "epoch": 0.32884,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.017679595947265626,
      "learning_rate": 0.0001,
      "loss": 4.0838,
      "loss/crossentropy": 2.1507667303085327,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20800111442804337,
      "step": 16442
    },
    {
      "epoch": 0.32888,
      "grad_norm": 1.875,
      "grad_norm_var": 0.020005035400390624,
      "learning_rate": 0.0001,
      "loss": 3.9428,
      "loss/crossentropy": 1.8771533370018005,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20476838946342468,
      "step": 16444
    },
    {
      "epoch": 0.32892,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.020005035400390624,
      "learning_rate": 0.0001,
      "loss": 4.2689,
      "loss/crossentropy": 2.365285038948059,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22512932121753693,
      "step": 16446
    },
    {
      "epoch": 0.32896,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.019701131184895835,
      "learning_rate": 0.0001,
      "loss": 4.2928,
      "loss/crossentropy": 2.2584418058395386,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21187593042850494,
      "step": 16448
    },
    {
      "epoch": 0.329,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.010270182291666667,
      "learning_rate": 0.0001,
      "loss": 3.8186,
      "loss/crossentropy": 1.9227403402328491,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1945207566022873,
      "step": 16450
    },
    {
      "epoch": 0.32904,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.011897786458333334,
      "learning_rate": 0.0001,
      "loss": 3.8512,
      "loss/crossentropy": 2.062865734100342,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2064652517437935,
      "step": 16452
    },
    {
      "epoch": 0.32908,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.007505035400390625,
      "learning_rate": 0.0001,
      "loss": 4.0493,
      "loss/crossentropy": 1.98399817943573,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22567245364189148,
      "step": 16454
    },
    {
      "epoch": 0.32912,
      "grad_norm": 2.0,
      "grad_norm_var": 0.007428995768229167,
      "learning_rate": 0.0001,
      "loss": 4.2888,
      "loss/crossentropy": 2.0655510425567627,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20936425030231476,
      "step": 16456
    },
    {
      "epoch": 0.32916,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.01739501953125,
      "learning_rate": 0.0001,
      "loss": 4.1486,
      "loss/crossentropy": 1.8360095024108887,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18788952380418777,
      "step": 16458
    },
    {
      "epoch": 0.3292,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.01672948201497396,
      "learning_rate": 0.0001,
      "loss": 4.2265,
      "loss/crossentropy": 2.1088568568229675,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20265556871891022,
      "step": 16460
    },
    {
      "epoch": 0.32924,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.017601521809895833,
      "learning_rate": 0.0001,
      "loss": 4.1004,
      "loss/crossentropy": 2.3746429681777954,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21173583716154099,
      "step": 16462
    },
    {
      "epoch": 0.32928,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.01946996053059896,
      "learning_rate": 0.0001,
      "loss": 3.9577,
      "loss/crossentropy": 2.0129969716072083,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18141476809978485,
      "step": 16464
    },
    {
      "epoch": 0.32932,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.019160970052083334,
      "learning_rate": 0.0001,
      "loss": 4.2122,
      "loss/crossentropy": 2.385707139968872,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22456685453653336,
      "step": 16466
    },
    {
      "epoch": 0.32936,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.015958658854166665,
      "learning_rate": 0.0001,
      "loss": 4.183,
      "loss/crossentropy": 2.466804623603821,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23166514188051224,
      "step": 16468
    },
    {
      "epoch": 0.3294,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.018700154622395833,
      "learning_rate": 0.0001,
      "loss": 3.8552,
      "loss/crossentropy": 1.9182489514350891,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1918247565627098,
      "step": 16470
    },
    {
      "epoch": 0.32944,
      "grad_norm": 2.125,
      "grad_norm_var": 0.021993001302083332,
      "learning_rate": 0.0001,
      "loss": 4.4364,
      "loss/crossentropy": 2.27492892742157,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2237030565738678,
      "step": 16472
    },
    {
      "epoch": 0.32948,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0144927978515625,
      "learning_rate": 0.0001,
      "loss": 4.0022,
      "loss/crossentropy": 1.856327474117279,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17718566209077835,
      "step": 16474
    },
    {
      "epoch": 0.32952,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.0145263671875,
      "learning_rate": 0.0001,
      "loss": 3.9994,
      "loss/crossentropy": 2.014274477958679,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20536774396896362,
      "step": 16476
    },
    {
      "epoch": 0.32956,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.016383616129557292,
      "learning_rate": 0.0001,
      "loss": 4.4606,
      "loss/crossentropy": 2.098844289779663,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18847080320119858,
      "step": 16478
    },
    {
      "epoch": 0.3296,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.014647420247395833,
      "learning_rate": 0.0001,
      "loss": 4.1978,
      "loss/crossentropy": 2.0402814149856567,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1929381936788559,
      "step": 16480
    },
    {
      "epoch": 0.32964,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.013315582275390625,
      "learning_rate": 0.0001,
      "loss": 4.2737,
      "loss/crossentropy": 2.413077712059021,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2426547110080719,
      "step": 16482
    },
    {
      "epoch": 0.32968,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.013401031494140625,
      "learning_rate": 0.0001,
      "loss": 3.9544,
      "loss/crossentropy": 2.081916332244873,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2190306931734085,
      "step": 16484
    },
    {
      "epoch": 0.32972,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.010765584309895833,
      "learning_rate": 0.0001,
      "loss": 4.1093,
      "loss/crossentropy": 2.1017117500305176,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19809769093990326,
      "step": 16486
    },
    {
      "epoch": 0.32976,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.008139801025390626,
      "learning_rate": 0.0001,
      "loss": 3.9585,
      "loss/crossentropy": 2.289687156677246,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21503399312496185,
      "step": 16488
    },
    {
      "epoch": 0.3298,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.008642323811848958,
      "learning_rate": 0.0001,
      "loss": 4.1952,
      "loss/crossentropy": 2.0613157749176025,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20428171753883362,
      "step": 16490
    },
    {
      "epoch": 0.32984,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0073964436848958336,
      "learning_rate": 0.0001,
      "loss": 4.3191,
      "loss/crossentropy": 2.1484400033950806,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20850211381912231,
      "step": 16492
    },
    {
      "epoch": 0.32988,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.0072629292805989586,
      "learning_rate": 0.0001,
      "loss": 3.875,
      "loss/crossentropy": 1.6853107213974,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1736336275935173,
      "step": 16494
    },
    {
      "epoch": 0.32992,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.007389068603515625,
      "learning_rate": 0.0001,
      "loss": 4.2047,
      "loss/crossentropy": 2.323481321334839,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22742202132940292,
      "step": 16496
    },
    {
      "epoch": 0.32996,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.00738525390625,
      "learning_rate": 0.0001,
      "loss": 4.124,
      "loss/crossentropy": 2.1196956038475037,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19792834669351578,
      "step": 16498
    },
    {
      "epoch": 0.33,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.007417805989583333,
      "learning_rate": 0.0001,
      "loss": 4.0013,
      "loss/crossentropy": 1.947241187095642,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21067001670598984,
      "step": 16500
    },
    {
      "epoch": 0.33004,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.00758056640625,
      "learning_rate": 0.0001,
      "loss": 4.0631,
      "loss/crossentropy": 2.256502628326416,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22300700098276138,
      "step": 16502
    },
    {
      "epoch": 0.33008,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.007574208577473958,
      "learning_rate": 0.0001,
      "loss": 3.7874,
      "loss/crossentropy": 1.7947281002998352,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1860191375017166,
      "step": 16504
    },
    {
      "epoch": 0.33012,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.007995351155598959,
      "learning_rate": 0.0001,
      "loss": 4.0189,
      "loss/crossentropy": 1.9331985712051392,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20170047879219055,
      "step": 16506
    },
    {
      "epoch": 0.33016,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.011030832926432291,
      "learning_rate": 0.0001,
      "loss": 4.2148,
      "loss/crossentropy": 2.2274327278137207,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1977931559085846,
      "step": 16508
    },
    {
      "epoch": 0.3302,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.007884724934895834,
      "learning_rate": 0.0001,
      "loss": 4.15,
      "loss/crossentropy": 1.8204082250595093,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18195096403360367,
      "step": 16510
    },
    {
      "epoch": 0.33024,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.009633127848307292,
      "learning_rate": 0.0001,
      "loss": 4.2363,
      "loss/crossentropy": 2.308950901031494,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22067535668611526,
      "step": 16512
    },
    {
      "epoch": 0.33028,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.014794921875,
      "learning_rate": 0.0001,
      "loss": 4.3895,
      "loss/crossentropy": 2.1682082414627075,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2021712213754654,
      "step": 16514
    },
    {
      "epoch": 0.33032,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.016047159830729168,
      "learning_rate": 0.0001,
      "loss": 3.9213,
      "loss/crossentropy": 1.9669402837753296,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19773626327514648,
      "step": 16516
    },
    {
      "epoch": 0.33036,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.016047159830729168,
      "learning_rate": 0.0001,
      "loss": 4.1723,
      "loss/crossentropy": 2.0478034019470215,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19190094619989395,
      "step": 16518
    },
    {
      "epoch": 0.3304,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.014127349853515625,
      "learning_rate": 0.0001,
      "loss": 4.1399,
      "loss/crossentropy": 1.8262990713119507,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19281092286109924,
      "step": 16520
    },
    {
      "epoch": 0.33044,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0139801025390625,
      "learning_rate": 0.0001,
      "loss": 4.385,
      "loss/crossentropy": 2.3943055868148804,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2286393940448761,
      "step": 16522
    },
    {
      "epoch": 0.33048,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0124664306640625,
      "learning_rate": 0.0001,
      "loss": 4.1117,
      "loss/crossentropy": 1.7113690972328186,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20282406359910965,
      "step": 16524
    },
    {
      "epoch": 0.33052,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.012924957275390624,
      "learning_rate": 0.0001,
      "loss": 3.9182,
      "loss/crossentropy": 1.9077526926994324,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18814775347709656,
      "step": 16526
    },
    {
      "epoch": 0.33056,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0106597900390625,
      "learning_rate": 0.0001,
      "loss": 4.3404,
      "loss/crossentropy": 1.9100408554077148,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1897171437740326,
      "step": 16528
    },
    {
      "epoch": 0.3306,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.0066314697265625,
      "learning_rate": 0.0001,
      "loss": 4.1403,
      "loss/crossentropy": 1.8145674467086792,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18221604079008102,
      "step": 16530
    },
    {
      "epoch": 0.33064,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0085113525390625,
      "learning_rate": 0.0001,
      "loss": 4.0647,
      "loss/crossentropy": 2.017480492591858,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19173581898212433,
      "step": 16532
    },
    {
      "epoch": 0.33068,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.008739980061848958,
      "learning_rate": 0.0001,
      "loss": 4.1918,
      "loss/crossentropy": 1.8937708139419556,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1974407583475113,
      "step": 16534
    },
    {
      "epoch": 0.33072,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.012664540608723959,
      "learning_rate": 0.0001,
      "loss": 4.1354,
      "loss/crossentropy": 1.824431598186493,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18114721775054932,
      "step": 16536
    },
    {
      "epoch": 0.33076,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.018464152018229166,
      "learning_rate": 0.0001,
      "loss": 4.034,
      "loss/crossentropy": 1.6330693364143372,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17686167359352112,
      "step": 16538
    },
    {
      "epoch": 0.3308,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.028595987955729166,
      "learning_rate": 0.0001,
      "loss": 4.2359,
      "loss/crossentropy": 2.1948810815811157,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21977168321609497,
      "step": 16540
    },
    {
      "epoch": 0.33084,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.03220926920572917,
      "learning_rate": 0.0001,
      "loss": 4.1942,
      "loss/crossentropy": 2.1222333908081055,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19481626898050308,
      "step": 16542
    },
    {
      "epoch": 0.33088,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.03241780598958333,
      "learning_rate": 0.0001,
      "loss": 3.9908,
      "loss/crossentropy": 1.9703376293182373,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20912524312734604,
      "step": 16544
    },
    {
      "epoch": 0.33092,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.03206965128580729,
      "learning_rate": 0.0001,
      "loss": 4.0774,
      "loss/crossentropy": 2.2693361043930054,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2137979120016098,
      "step": 16546
    },
    {
      "epoch": 0.33096,
      "grad_norm": 3.703125,
      "grad_norm_var": 0.19548746744791667,
      "learning_rate": 0.0001,
      "loss": 4.2372,
      "loss/crossentropy": 2.029613673686981,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21167294681072235,
      "step": 16548
    },
    {
      "epoch": 0.331,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.19552586873372396,
      "learning_rate": 0.0001,
      "loss": 3.7223,
      "loss/crossentropy": 2.0102853775024414,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2100009247660637,
      "step": 16550
    },
    {
      "epoch": 0.33104,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.19806722005208333,
      "learning_rate": 0.0001,
      "loss": 4.6058,
      "loss/crossentropy": 2.1570287942886353,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21619540452957153,
      "step": 16552
    },
    {
      "epoch": 0.33108,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.20053609212239584,
      "learning_rate": 0.0001,
      "loss": 4.0271,
      "loss/crossentropy": 1.9157934188842773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.193254753947258,
      "step": 16554
    },
    {
      "epoch": 0.33112,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.2052642822265625,
      "learning_rate": 0.0001,
      "loss": 3.6593,
      "loss/crossentropy": 1.7680367827415466,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19283122569322586,
      "step": 16556
    },
    {
      "epoch": 0.33116,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.2058013916015625,
      "learning_rate": 0.0001,
      "loss": 3.8828,
      "loss/crossentropy": 1.8805989623069763,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20299813151359558,
      "step": 16558
    },
    {
      "epoch": 0.3312,
      "grad_norm": 1.71875,
      "grad_norm_var": 0.21938069661458334,
      "learning_rate": 0.0001,
      "loss": 3.7657,
      "loss/crossentropy": 1.9145240187644958,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18402665108442307,
      "step": 16560
    },
    {
      "epoch": 0.33124,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.2167144775390625,
      "learning_rate": 0.0001,
      "loss": 4.1984,
      "loss/crossentropy": 1.7958417534828186,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17038261890411377,
      "step": 16562
    },
    {
      "epoch": 0.33128,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.038919830322265626,
      "learning_rate": 0.0001,
      "loss": 3.8916,
      "loss/crossentropy": 2.0782148838043213,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21738532930612564,
      "step": 16564
    },
    {
      "epoch": 0.33132,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.03910903930664063,
      "learning_rate": 0.0001,
      "loss": 4.3098,
      "loss/crossentropy": 1.9767170548439026,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2127690687775612,
      "step": 16566
    },
    {
      "epoch": 0.33136,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.01185302734375,
      "learning_rate": 0.0001,
      "loss": 3.9133,
      "loss/crossentropy": 2.086383044719696,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19811449944972992,
      "step": 16568
    },
    {
      "epoch": 0.3314,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.011991119384765625,
      "learning_rate": 0.0001,
      "loss": 4.0358,
      "loss/crossentropy": 2.0471088886260986,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20480120182037354,
      "step": 16570
    },
    {
      "epoch": 0.33144,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.012485504150390625,
      "learning_rate": 0.0001,
      "loss": 4.1097,
      "loss/crossentropy": 2.2637689113616943,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21112071722745895,
      "step": 16572
    },
    {
      "epoch": 0.33148,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.012018839518229166,
      "learning_rate": 0.0001,
      "loss": 4.023,
      "loss/crossentropy": 1.8035425543785095,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18357180804014206,
      "step": 16574
    },
    {
      "epoch": 0.33152,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.007189687093098958,
      "learning_rate": 0.0001,
      "loss": 3.7453,
      "loss/crossentropy": 1.647614598274231,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17512448877096176,
      "step": 16576
    },
    {
      "epoch": 0.33156,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0063873291015625,
      "learning_rate": 0.0001,
      "loss": 4.0452,
      "loss/crossentropy": 2.0647078156471252,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21568534523248672,
      "step": 16578
    },
    {
      "epoch": 0.3316,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.00615234375,
      "learning_rate": 0.0001,
      "loss": 3.8656,
      "loss/crossentropy": 1.7758954763412476,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1891157627105713,
      "step": 16580
    },
    {
      "epoch": 0.33164,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0060618082682291664,
      "learning_rate": 0.0001,
      "loss": 4.1022,
      "loss/crossentropy": 2.0523850321769714,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21198877692222595,
      "step": 16582
    },
    {
      "epoch": 0.33168,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.006725819905598959,
      "learning_rate": 0.0001,
      "loss": 4.0584,
      "loss/crossentropy": 1.99459570646286,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20555292069911957,
      "step": 16584
    },
    {
      "epoch": 0.33172,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.007771809895833333,
      "learning_rate": 0.0001,
      "loss": 4.4979,
      "loss/crossentropy": 2.353522777557373,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23193839937448502,
      "step": 16586
    },
    {
      "epoch": 0.33176,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.020970662434895832,
      "learning_rate": 0.0001,
      "loss": 4.1303,
      "loss/crossentropy": 2.0257768630981445,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20071294903755188,
      "step": 16588
    },
    {
      "epoch": 0.3318,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.020467122395833332,
      "learning_rate": 0.0001,
      "loss": 4.2135,
      "loss/crossentropy": 2.2084985971450806,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20876871049404144,
      "step": 16590
    },
    {
      "epoch": 0.33184,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.019791412353515624,
      "learning_rate": 0.0001,
      "loss": 4.3275,
      "loss/crossentropy": 1.9118791818618774,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24547216296195984,
      "step": 16592
    },
    {
      "epoch": 0.33188,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.022188313802083335,
      "learning_rate": 0.0001,
      "loss": 4.0069,
      "loss/crossentropy": 2.184453248977661,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1988314613699913,
      "step": 16594
    },
    {
      "epoch": 0.33192,
      "grad_norm": 1.71875,
      "grad_norm_var": 0.02801488240559896,
      "learning_rate": 0.0001,
      "loss": 3.8711,
      "loss/crossentropy": 2.2105953097343445,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20877376198768616,
      "step": 16596
    },
    {
      "epoch": 0.33196,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.028586578369140626,
      "learning_rate": 0.0001,
      "loss": 4.2969,
      "loss/crossentropy": 2.028349459171295,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20705370604991913,
      "step": 16598
    },
    {
      "epoch": 0.332,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.02982177734375,
      "learning_rate": 0.0001,
      "loss": 3.8968,
      "loss/crossentropy": 2.0778703689575195,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19761346280574799,
      "step": 16600
    },
    {
      "epoch": 0.33204,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.029670206705729167,
      "learning_rate": 0.0001,
      "loss": 3.9786,
      "loss/crossentropy": 2.046931743621826,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21325716376304626,
      "step": 16602
    },
    {
      "epoch": 0.33208,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.011486562093098958,
      "learning_rate": 0.0001,
      "loss": 4.1525,
      "loss/crossentropy": 1.7907955050468445,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1822371482849121,
      "step": 16604
    },
    {
      "epoch": 0.33212,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.012108357747395833,
      "learning_rate": 0.0001,
      "loss": 4.1586,
      "loss/crossentropy": 1.7620025277137756,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19196610152721405,
      "step": 16606
    },
    {
      "epoch": 0.33216,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010282135009765625,
      "learning_rate": 0.0001,
      "loss": 3.9175,
      "loss/crossentropy": 1.6695470213890076,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16451346129179,
      "step": 16608
    },
    {
      "epoch": 0.3322,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.012717437744140626,
      "learning_rate": 0.0001,
      "loss": 4.0304,
      "loss/crossentropy": 2.2249897718429565,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21898073703050613,
      "step": 16610
    },
    {
      "epoch": 0.33224,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.009474436442057291,
      "learning_rate": 0.0001,
      "loss": 4.1589,
      "loss/crossentropy": 2.0861976146698,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1993977203965187,
      "step": 16612
    },
    {
      "epoch": 0.33228,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.009527333577473958,
      "learning_rate": 0.0001,
      "loss": 4.1516,
      "loss/crossentropy": 2.195378541946411,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2103184312582016,
      "step": 16614
    },
    {
      "epoch": 0.33232,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.009478505452473958,
      "learning_rate": 0.0001,
      "loss": 3.6947,
      "loss/crossentropy": 1.9064915180206299,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17310689389705658,
      "step": 16616
    },
    {
      "epoch": 0.33236,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.009452311197916667,
      "learning_rate": 0.0001,
      "loss": 3.9698,
      "loss/crossentropy": 1.7967488169670105,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18099892884492874,
      "step": 16618
    },
    {
      "epoch": 0.3324,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.012565104166666667,
      "learning_rate": 0.0001,
      "loss": 4.1319,
      "loss/crossentropy": 2.0776742696762085,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2005431056022644,
      "step": 16620
    },
    {
      "epoch": 0.33244,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0109375,
      "learning_rate": 0.0001,
      "loss": 4.1881,
      "loss/crossentropy": 2.0590518712997437,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20676207542419434,
      "step": 16622
    },
    {
      "epoch": 0.33248,
      "grad_norm": 2.0,
      "grad_norm_var": 0.011628977457682292,
      "learning_rate": 0.0001,
      "loss": 4.2067,
      "loss/crossentropy": 1.8861089944839478,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1860756129026413,
      "step": 16624
    },
    {
      "epoch": 0.33252,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.009308878580729167,
      "learning_rate": 0.0001,
      "loss": 3.9757,
      "loss/crossentropy": 2.4207329750061035,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22214040160179138,
      "step": 16626
    },
    {
      "epoch": 0.33256,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.009295399983723958,
      "learning_rate": 0.0001,
      "loss": 3.934,
      "loss/crossentropy": 2.089089274406433,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21007104218006134,
      "step": 16628
    },
    {
      "epoch": 0.3326,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.011380767822265625,
      "learning_rate": 0.0001,
      "loss": 4.3451,
      "loss/crossentropy": 2.357658624649048,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22044895589351654,
      "step": 16630
    },
    {
      "epoch": 0.33264,
      "grad_norm": 1.875,
      "grad_norm_var": 0.010223134358723959,
      "learning_rate": 0.0001,
      "loss": 3.9142,
      "loss/crossentropy": 2.1036278009414673,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2003040835261345,
      "step": 16632
    },
    {
      "epoch": 0.33268,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.010749308268229167,
      "learning_rate": 0.0001,
      "loss": 4.0802,
      "loss/crossentropy": 2.202688694000244,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21549009531736374,
      "step": 16634
    },
    {
      "epoch": 0.33272,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.008642323811848958,
      "learning_rate": 0.0001,
      "loss": 4.1642,
      "loss/crossentropy": 2.261624753475189,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2209767997264862,
      "step": 16636
    },
    {
      "epoch": 0.33276,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.013570149739583334,
      "learning_rate": 0.0001,
      "loss": 4.0556,
      "loss/crossentropy": 1.7452040910720825,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18558355420827866,
      "step": 16638
    },
    {
      "epoch": 0.3328,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.012797037760416666,
      "learning_rate": 0.0001,
      "loss": 3.9364,
      "loss/crossentropy": 1.9910151362419128,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2115819901227951,
      "step": 16640
    },
    {
      "epoch": 0.33284,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.014583079020182292,
      "learning_rate": 0.0001,
      "loss": 3.9412,
      "loss/crossentropy": 2.0230116844177246,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2001063972711563,
      "step": 16642
    },
    {
      "epoch": 0.33288,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.015818023681640626,
      "learning_rate": 0.0001,
      "loss": 4.0476,
      "loss/crossentropy": 1.9575697183609009,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20243427157402039,
      "step": 16644
    },
    {
      "epoch": 0.33292,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.012890370686848958,
      "learning_rate": 0.0001,
      "loss": 4.052,
      "loss/crossentropy": 1.9650630354881287,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.203716441988945,
      "step": 16646
    },
    {
      "epoch": 0.33296,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.0132476806640625,
      "learning_rate": 0.0001,
      "loss": 4.1345,
      "loss/crossentropy": 1.497282326221466,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15893913060426712,
      "step": 16648
    },
    {
      "epoch": 0.333,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.014491526285807292,
      "learning_rate": 0.0001,
      "loss": 4.1273,
      "loss/crossentropy": 1.8017843961715698,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18981081247329712,
      "step": 16650
    },
    {
      "epoch": 0.33304,
      "grad_norm": 2.125,
      "grad_norm_var": 0.014296213785807291,
      "learning_rate": 0.0001,
      "loss": 4.07,
      "loss/crossentropy": 2.089366614818573,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2076069712638855,
      "step": 16652
    },
    {
      "epoch": 0.33308,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.008695475260416667,
      "learning_rate": 0.0001,
      "loss": 3.9662,
      "loss/crossentropy": 1.9445012211799622,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18241490423679352,
      "step": 16654
    },
    {
      "epoch": 0.33312,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.008740234375,
      "learning_rate": 0.0001,
      "loss": 3.9922,
      "loss/crossentropy": 1.844423532485962,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18664997816085815,
      "step": 16656
    },
    {
      "epoch": 0.33316,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.007085927327473958,
      "learning_rate": 0.0001,
      "loss": 4.2221,
      "loss/crossentropy": 2.1095730662345886,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21053151786327362,
      "step": 16658
    },
    {
      "epoch": 0.3332,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.007477823893229167,
      "learning_rate": 0.0001,
      "loss": 4.255,
      "loss/crossentropy": 2.139304041862488,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19131766259670258,
      "step": 16660
    },
    {
      "epoch": 0.33324,
      "grad_norm": 2.0,
      "grad_norm_var": 0.007450103759765625,
      "learning_rate": 0.0001,
      "loss": 3.9358,
      "loss/crossentropy": 1.8872849345207214,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19200573861598969,
      "step": 16662
    },
    {
      "epoch": 0.33328,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.006695302327473959,
      "learning_rate": 0.0001,
      "loss": 4.0939,
      "loss/crossentropy": 2.488931655883789,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24411997199058533,
      "step": 16664
    },
    {
      "epoch": 0.33332,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.006151326497395833,
      "learning_rate": 0.0001,
      "loss": 3.7294,
      "loss/crossentropy": 1.8066997528076172,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19391655176877975,
      "step": 16666
    },
    {
      "epoch": 0.33336,
      "grad_norm": 2.0,
      "grad_norm_var": 0.005370076497395833,
      "learning_rate": 0.0001,
      "loss": 4.1831,
      "loss/crossentropy": 2.085490345954895,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21987342834472656,
      "step": 16668
    },
    {
      "epoch": 0.3334,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.006459299723307292,
      "learning_rate": 0.0001,
      "loss": 3.7791,
      "loss/crossentropy": 2.052145302295685,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21673783659934998,
      "step": 16670
    },
    {
      "epoch": 0.33344,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.00640869140625,
      "learning_rate": 0.0001,
      "loss": 4.1648,
      "loss/crossentropy": 2.0924419164657593,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21523578464984894,
      "step": 16672
    },
    {
      "epoch": 0.33348,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.006525675455729167,
      "learning_rate": 0.0001,
      "loss": 4.125,
      "loss/crossentropy": 2.0340868830680847,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1785074919462204,
      "step": 16674
    },
    {
      "epoch": 0.33352,
      "grad_norm": 2.25,
      "grad_norm_var": 0.010990397135416666,
      "learning_rate": 0.0001,
      "loss": 4.2404,
      "loss/crossentropy": 2.4421908855438232,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24263421446084976,
      "step": 16676
    },
    {
      "epoch": 0.33356,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0108551025390625,
      "learning_rate": 0.0001,
      "loss": 4.1879,
      "loss/crossentropy": 1.9141955971717834,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18666712939739227,
      "step": 16678
    },
    {
      "epoch": 0.3336,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011336008707682291,
      "learning_rate": 0.0001,
      "loss": 4.0378,
      "loss/crossentropy": 2.0019638538360596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17722105979919434,
      "step": 16680
    },
    {
      "epoch": 0.33364,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.009511057535807292,
      "learning_rate": 0.0001,
      "loss": 4.2047,
      "loss/crossentropy": 2.0835896134376526,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.251879021525383,
      "step": 16682
    },
    {
      "epoch": 0.33368,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.015192667643229166,
      "learning_rate": 0.0001,
      "loss": 4.2406,
      "loss/crossentropy": 1.8940032720565796,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19034714996814728,
      "step": 16684
    },
    {
      "epoch": 0.33372,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.014817047119140624,
      "learning_rate": 0.0001,
      "loss": 4.0845,
      "loss/crossentropy": 1.863099992275238,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19675587117671967,
      "step": 16686
    },
    {
      "epoch": 0.33376,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.024836222330729168,
      "learning_rate": 0.0001,
      "loss": 4.2785,
      "loss/crossentropy": 2.027459740638733,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19650273770093918,
      "step": 16688
    },
    {
      "epoch": 0.3338,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.026008097330729167,
      "learning_rate": 0.0001,
      "loss": 4.2892,
      "loss/crossentropy": 2.2324944734573364,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23016268759965897,
      "step": 16690
    },
    {
      "epoch": 0.33384,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.02474950154622396,
      "learning_rate": 0.0001,
      "loss": 4.155,
      "loss/crossentropy": 2.213741898536682,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21159538626670837,
      "step": 16692
    },
    {
      "epoch": 0.33388,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.025340779622395834,
      "learning_rate": 0.0001,
      "loss": 4.4426,
      "loss/crossentropy": 2.084823966026306,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22060541808605194,
      "step": 16694
    },
    {
      "epoch": 0.33392,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.02417780558268229,
      "learning_rate": 0.0001,
      "loss": 4.1847,
      "loss/crossentropy": 2.2417017221450806,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2050381377339363,
      "step": 16696
    },
    {
      "epoch": 0.33396,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.032134755452473955,
      "learning_rate": 0.0001,
      "loss": 4.3037,
      "loss/crossentropy": 2.3696242570877075,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22559890151023865,
      "step": 16698
    },
    {
      "epoch": 0.334,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.030295562744140626,
      "learning_rate": 0.0001,
      "loss": 4.158,
      "loss/crossentropy": 2.0899609327316284,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20470503717660904,
      "step": 16700
    },
    {
      "epoch": 0.33404,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.029386138916015624,
      "learning_rate": 0.0001,
      "loss": 3.924,
      "loss/crossentropy": 1.8663234114646912,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1815531924366951,
      "step": 16702
    },
    {
      "epoch": 0.33408,
      "grad_norm": 3.0,
      "grad_norm_var": 0.08069432576497396,
      "learning_rate": 0.0001,
      "loss": 4.1651,
      "loss/crossentropy": 2.030432403087616,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2063748836517334,
      "step": 16704
    },
    {
      "epoch": 0.33412,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.08033218383789062,
      "learning_rate": 0.0001,
      "loss": 4.152,
      "loss/crossentropy": 2.12885981798172,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19536730647087097,
      "step": 16706
    },
    {
      "epoch": 0.33416,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.07674153645833333,
      "learning_rate": 0.0001,
      "loss": 3.909,
      "loss/crossentropy": 2.0870607495307922,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20478558540344238,
      "step": 16708
    },
    {
      "epoch": 0.3342,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.07646382649739583,
      "learning_rate": 0.0001,
      "loss": 4.2058,
      "loss/crossentropy": 2.215463638305664,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2118721902370453,
      "step": 16710
    },
    {
      "epoch": 0.33424,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.08145243326822917,
      "learning_rate": 0.0001,
      "loss": 3.8417,
      "loss/crossentropy": 2.0292049646377563,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18440847843885422,
      "step": 16712
    },
    {
      "epoch": 0.33428,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.08017756144205729,
      "learning_rate": 0.0001,
      "loss": 3.7722,
      "loss/crossentropy": 1.9214341640472412,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18510671705007553,
      "step": 16714
    },
    {
      "epoch": 0.33432,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.07769266764322917,
      "learning_rate": 0.0001,
      "loss": 4.3905,
      "loss/crossentropy": 2.1761534214019775,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20527766644954681,
      "step": 16716
    },
    {
      "epoch": 0.33436,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.07639567057291667,
      "learning_rate": 0.0001,
      "loss": 3.9234,
      "loss/crossentropy": 2.020824670791626,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20723462104797363,
      "step": 16718
    },
    {
      "epoch": 0.3344,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0129547119140625,
      "learning_rate": 0.0001,
      "loss": 4.167,
      "loss/crossentropy": 2.243465781211853,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21866197139024734,
      "step": 16720
    },
    {
      "epoch": 0.33444,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.013244374593098959,
      "learning_rate": 0.0001,
      "loss": 3.9963,
      "loss/crossentropy": 1.9071390628814697,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20368033647537231,
      "step": 16722
    },
    {
      "epoch": 0.33448,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.011966959635416666,
      "learning_rate": 0.0001,
      "loss": 3.8974,
      "loss/crossentropy": 1.72525554895401,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18200145661830902,
      "step": 16724
    },
    {
      "epoch": 0.33452,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0154052734375,
      "learning_rate": 0.0001,
      "loss": 4.101,
      "loss/crossentropy": 2.174055576324463,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21181844174861908,
      "step": 16726
    },
    {
      "epoch": 0.33456,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.014251454671223959,
      "learning_rate": 0.0001,
      "loss": 3.8004,
      "loss/crossentropy": 1.6175724864006042,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15338046848773956,
      "step": 16728
    },
    {
      "epoch": 0.3346,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.012353515625,
      "learning_rate": 0.0001,
      "loss": 4.1101,
      "loss/crossentropy": 2.0030194520950317,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20353461802005768,
      "step": 16730
    },
    {
      "epoch": 0.33464,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.012092081705729167,
      "learning_rate": 0.0001,
      "loss": 4.3876,
      "loss/crossentropy": 2.272015690803528,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22368235886096954,
      "step": 16732
    },
    {
      "epoch": 0.33468,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.012189737955729167,
      "learning_rate": 0.0001,
      "loss": 4.3365,
      "loss/crossentropy": 2.149975538253784,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20634270459413528,
      "step": 16734
    },
    {
      "epoch": 0.33472,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.009869130452473958,
      "learning_rate": 0.0001,
      "loss": 3.9201,
      "loss/crossentropy": 2.095462441444397,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20027077198028564,
      "step": 16736
    },
    {
      "epoch": 0.33476,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.008819325764973959,
      "learning_rate": 0.0001,
      "loss": 4.0859,
      "loss/crossentropy": 1.988598644733429,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20503661036491394,
      "step": 16738
    },
    {
      "epoch": 0.3348,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0066569010416666664,
      "learning_rate": 0.0001,
      "loss": 4.2618,
      "loss/crossentropy": 2.0615572333335876,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20363956689834595,
      "step": 16740
    },
    {
      "epoch": 0.33484,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0027903238932291668,
      "learning_rate": 0.0001,
      "loss": 3.9476,
      "loss/crossentropy": 1.8172455430030823,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20091129839420319,
      "step": 16742
    },
    {
      "epoch": 0.33488,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0043853759765625,
      "learning_rate": 0.0001,
      "loss": 3.9541,
      "loss/crossentropy": 2.212782144546509,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19710610061883926,
      "step": 16744
    },
    {
      "epoch": 0.33492,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.008674875895182291,
      "learning_rate": 0.0001,
      "loss": 4.2629,
      "loss/crossentropy": 2.374183773994446,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23687051236629486,
      "step": 16746
    },
    {
      "epoch": 0.33496,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.008902740478515626,
      "learning_rate": 0.0001,
      "loss": 4.3642,
      "loss/crossentropy": 1.9374622702598572,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19821622967720032,
      "step": 16748
    },
    {
      "epoch": 0.335,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.008740234375,
      "learning_rate": 0.0001,
      "loss": 4.1289,
      "loss/crossentropy": 1.6063715815544128,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16296840459108353,
      "step": 16750
    },
    {
      "epoch": 0.33504,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.0096832275390625,
      "learning_rate": 0.0001,
      "loss": 4.0315,
      "loss/crossentropy": 2.1251469254493713,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17997874319553375,
      "step": 16752
    },
    {
      "epoch": 0.33508,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0097564697265625,
      "learning_rate": 0.0001,
      "loss": 4.03,
      "loss/crossentropy": 2.126043915748596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2283206582069397,
      "step": 16754
    },
    {
      "epoch": 0.33512,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.009901682535807291,
      "learning_rate": 0.0001,
      "loss": 4.2969,
      "loss/crossentropy": 2.3409340381622314,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21817704290151596,
      "step": 16756
    },
    {
      "epoch": 0.33516,
      "grad_norm": 2.0,
      "grad_norm_var": 0.01024169921875,
      "learning_rate": 0.0001,
      "loss": 4.0083,
      "loss/crossentropy": 1.9816790223121643,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19743956625461578,
      "step": 16758
    },
    {
      "epoch": 0.3352,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.012967681884765625,
      "learning_rate": 0.0001,
      "loss": 4.3056,
      "loss/crossentropy": 2.246092438697815,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22662658989429474,
      "step": 16760
    },
    {
      "epoch": 0.33524,
      "grad_norm": 2.125,
      "grad_norm_var": 0.010794830322265626,
      "learning_rate": 0.0001,
      "loss": 4.2842,
      "loss/crossentropy": 2.1571802496910095,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20506682991981506,
      "step": 16762
    },
    {
      "epoch": 0.33528,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010636138916015624,
      "learning_rate": 0.0001,
      "loss": 4.0372,
      "loss/crossentropy": 1.966954231262207,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18581678718328476,
      "step": 16764
    },
    {
      "epoch": 0.33532,
      "grad_norm": 2.125,
      "grad_norm_var": 0.012961578369140626,
      "learning_rate": 0.0001,
      "loss": 3.9978,
      "loss/crossentropy": 1.9360153079032898,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20846693962812424,
      "step": 16766
    },
    {
      "epoch": 0.33536,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.012644195556640625,
      "learning_rate": 0.0001,
      "loss": 4.0953,
      "loss/crossentropy": 2.3978073596954346,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20406542718410492,
      "step": 16768
    },
    {
      "epoch": 0.3354,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.013639068603515625,
      "learning_rate": 0.0001,
      "loss": 4.0933,
      "loss/crossentropy": 2.239229917526245,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21536701172590256,
      "step": 16770
    },
    {
      "epoch": 0.33544,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.013600413004557292,
      "learning_rate": 0.0001,
      "loss": 4.2153,
      "loss/crossentropy": 2.1333428621292114,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2062046378850937,
      "step": 16772
    },
    {
      "epoch": 0.33548,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.014362589518229166,
      "learning_rate": 0.0001,
      "loss": 4.0094,
      "loss/crossentropy": 2.3503127098083496,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20422626286745071,
      "step": 16774
    },
    {
      "epoch": 0.33552,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.009772745768229167,
      "learning_rate": 0.0001,
      "loss": 4.1466,
      "loss/crossentropy": 1.9181209802627563,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16844520717859268,
      "step": 16776
    },
    {
      "epoch": 0.33556,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.009269205729166667,
      "learning_rate": 0.0001,
      "loss": 4.1975,
      "loss/crossentropy": 2.022172212600708,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19443488866090775,
      "step": 16778
    },
    {
      "epoch": 0.3356,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.010951487223307292,
      "learning_rate": 0.0001,
      "loss": 3.9544,
      "loss/crossentropy": 2.0416210293769836,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17731749266386032,
      "step": 16780
    },
    {
      "epoch": 0.33564,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009748331705729167,
      "learning_rate": 0.0001,
      "loss": 4.0757,
      "loss/crossentropy": 2.1722596883773804,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19995421171188354,
      "step": 16782
    },
    {
      "epoch": 0.33568,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.009146881103515626,
      "learning_rate": 0.0001,
      "loss": 4.0942,
      "loss/crossentropy": 2.4041545391082764,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22784388065338135,
      "step": 16784
    },
    {
      "epoch": 0.33572,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.007757314046223958,
      "learning_rate": 0.0001,
      "loss": 4.0692,
      "loss/crossentropy": 2.189204216003418,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20079431682825089,
      "step": 16786
    },
    {
      "epoch": 0.33576,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.007130686442057292,
      "learning_rate": 0.0001,
      "loss": 3.9771,
      "loss/crossentropy": 2.2308130860328674,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22352158278226852,
      "step": 16788
    },
    {
      "epoch": 0.3358,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.005248006184895833,
      "learning_rate": 0.0001,
      "loss": 4.0901,
      "loss/crossentropy": 2.238118886947632,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21165399253368378,
      "step": 16790
    },
    {
      "epoch": 0.33584,
      "grad_norm": 11.9375,
      "grad_norm_var": 6.248060862223308,
      "learning_rate": 0.0001,
      "loss": 4.5633,
      "loss/crossentropy": 2.452346444129944,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2192159816622734,
      "step": 16792
    },
    {
      "epoch": 0.33588,
      "grad_norm": 1.9765625,
      "grad_norm_var": 6.234175364176433,
      "learning_rate": 0.0001,
      "loss": 4.2032,
      "loss/crossentropy": 2.115506410598755,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19228096306324005,
      "step": 16794
    },
    {
      "epoch": 0.33592,
      "grad_norm": 2.046875,
      "grad_norm_var": 6.207706705729167,
      "learning_rate": 0.0001,
      "loss": 4.313,
      "loss/crossentropy": 2.1052395701408386,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24314512312412262,
      "step": 16796
    },
    {
      "epoch": 0.33596,
      "grad_norm": 1.9765625,
      "grad_norm_var": 6.191576131184896,
      "learning_rate": 0.0001,
      "loss": 4.201,
      "loss/crossentropy": 1.7643597722053528,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17998731136322021,
      "step": 16798
    },
    {
      "epoch": 0.336,
      "grad_norm": 1.890625,
      "grad_norm_var": 6.213846588134766,
      "learning_rate": 0.0001,
      "loss": 3.886,
      "loss/crossentropy": 2.0926302671432495,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.199835367500782,
      "step": 16800
    },
    {
      "epoch": 0.33604,
      "grad_norm": 2.109375,
      "grad_norm_var": 6.19762954711914,
      "learning_rate": 0.0001,
      "loss": 4.0106,
      "loss/crossentropy": 1.9939849972724915,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19728610664606094,
      "step": 16802
    },
    {
      "epoch": 0.33608,
      "grad_norm": 2.15625,
      "grad_norm_var": 6.1746826171875,
      "learning_rate": 0.0001,
      "loss": 4.3381,
      "loss/crossentropy": 2.154136300086975,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2071739211678505,
      "step": 16804
    },
    {
      "epoch": 0.33612,
      "grad_norm": 1.9296875,
      "grad_norm_var": 6.173281860351563,
      "learning_rate": 0.0001,
      "loss": 4.06,
      "loss/crossentropy": 1.6244451403617859,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1825866624712944,
      "step": 16806
    },
    {
      "epoch": 0.33616,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.010945383707682292,
      "learning_rate": 0.0001,
      "loss": 3.6796,
      "loss/crossentropy": 1.936405599117279,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20031629502773285,
      "step": 16808
    },
    {
      "epoch": 0.3362,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.011568196614583333,
      "learning_rate": 0.0001,
      "loss": 4.276,
      "loss/crossentropy": 2.117881119251251,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21941083669662476,
      "step": 16810
    },
    {
      "epoch": 0.33624,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0125152587890625,
      "learning_rate": 0.0001,
      "loss": 4.1104,
      "loss/crossentropy": 2.0384327173233032,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2019578069448471,
      "step": 16812
    },
    {
      "epoch": 0.33628,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.011893463134765626,
      "learning_rate": 0.0001,
      "loss": 4.1383,
      "loss/crossentropy": 2.058988571166992,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20959977805614471,
      "step": 16814
    },
    {
      "epoch": 0.33632,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.010680898030598959,
      "learning_rate": 0.0001,
      "loss": 4.0734,
      "loss/crossentropy": 2.117241382598877,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2039024829864502,
      "step": 16816
    },
    {
      "epoch": 0.33636,
      "grad_norm": 2.0,
      "grad_norm_var": 0.009968058268229166,
      "learning_rate": 0.0001,
      "loss": 4.157,
      "loss/crossentropy": 1.8142318725585938,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1877349317073822,
      "step": 16818
    },
    {
      "epoch": 0.3364,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.014440663655598958,
      "learning_rate": 0.0001,
      "loss": 4.2402,
      "loss/crossentropy": 2.1766676902770996,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21788865327835083,
      "step": 16820
    },
    {
      "epoch": 0.33644,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0163726806640625,
      "learning_rate": 0.0001,
      "loss": 4.2953,
      "loss/crossentropy": 2.2596739530563354,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23641519993543625,
      "step": 16822
    },
    {
      "epoch": 0.33648,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.012572987874348959,
      "learning_rate": 0.0001,
      "loss": 4.1689,
      "loss/crossentropy": 2.1782987117767334,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20562626421451569,
      "step": 16824
    },
    {
      "epoch": 0.33652,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.01633275349934896,
      "learning_rate": 0.0001,
      "loss": 4.0693,
      "loss/crossentropy": 1.9579410552978516,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20332375168800354,
      "step": 16826
    },
    {
      "epoch": 0.33656,
      "grad_norm": 1.75,
      "grad_norm_var": 0.020977528889973958,
      "learning_rate": 0.0001,
      "loss": 4.0193,
      "loss/crossentropy": 1.917382538318634,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.188996322453022,
      "step": 16828
    },
    {
      "epoch": 0.3366,
      "grad_norm": 2.125,
      "grad_norm_var": 0.02045466105143229,
      "learning_rate": 0.0001,
      "loss": 3.9885,
      "loss/crossentropy": 1.8045424222946167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18587132543325424,
      "step": 16830
    },
    {
      "epoch": 0.33664,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.07043355305989583,
      "learning_rate": 0.0001,
      "loss": 4.0171,
      "loss/crossentropy": 2.113844871520996,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20383527874946594,
      "step": 16832
    },
    {
      "epoch": 0.33668,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.06795654296875,
      "learning_rate": 0.0001,
      "loss": 4.1255,
      "loss/crossentropy": 1.9635592699050903,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20447230339050293,
      "step": 16834
    },
    {
      "epoch": 0.33672,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.06646219889322917,
      "learning_rate": 0.0001,
      "loss": 4.1336,
      "loss/crossentropy": 1.8918602466583252,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21786046028137207,
      "step": 16836
    },
    {
      "epoch": 0.33676,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.06734390258789062,
      "learning_rate": 0.0001,
      "loss": 4.3372,
      "loss/crossentropy": 2.347909450531006,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23047995567321777,
      "step": 16838
    },
    {
      "epoch": 0.3368,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.06854248046875,
      "learning_rate": 0.0001,
      "loss": 4.0242,
      "loss/crossentropy": 1.834633469581604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1915532797574997,
      "step": 16840
    },
    {
      "epoch": 0.33684,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.06948954264322917,
      "learning_rate": 0.0001,
      "loss": 3.6871,
      "loss/crossentropy": 1.9172112345695496,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18901804834604263,
      "step": 16842
    },
    {
      "epoch": 0.33688,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.06441141764322916,
      "learning_rate": 0.0001,
      "loss": 4.3911,
      "loss/crossentropy": 2.37927508354187,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2328021451830864,
      "step": 16844
    },
    {
      "epoch": 0.33692,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.06558024088541667,
      "learning_rate": 0.0001,
      "loss": 3.8573,
      "loss/crossentropy": 1.7887099385261536,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17989587783813477,
      "step": 16846
    },
    {
      "epoch": 0.33696,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0051513671875,
      "learning_rate": 0.0001,
      "loss": 4.0826,
      "loss/crossentropy": 2.008473217487335,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20677915215492249,
      "step": 16848
    },
    {
      "epoch": 0.337,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.005362955729166666,
      "learning_rate": 0.0001,
      "loss": 4.1751,
      "loss/crossentropy": 2.101797103881836,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19833290576934814,
      "step": 16850
    },
    {
      "epoch": 0.33704,
      "grad_norm": 2.0,
      "grad_norm_var": 0.005779774983723959,
      "learning_rate": 0.0001,
      "loss": 4.0087,
      "loss/crossentropy": 1.9306662678718567,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1893247440457344,
      "step": 16852
    },
    {
      "epoch": 0.33708,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.005293782552083333,
      "learning_rate": 0.0001,
      "loss": 3.9986,
      "loss/crossentropy": 2.1301704049110413,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1975892335176468,
      "step": 16854
    },
    {
      "epoch": 0.33712,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.007173411051432292,
      "learning_rate": 0.0001,
      "loss": 3.9583,
      "loss/crossentropy": 2.020972192287445,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19376973807811737,
      "step": 16856
    },
    {
      "epoch": 0.33716,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.006159464518229167,
      "learning_rate": 0.0001,
      "loss": 3.9632,
      "loss/crossentropy": 1.8190429210662842,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1629476398229599,
      "step": 16858
    },
    {
      "epoch": 0.3372,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.007183583577473959,
      "learning_rate": 0.0001,
      "loss": 4.2147,
      "loss/crossentropy": 2.360092878341675,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22874057292938232,
      "step": 16860
    },
    {
      "epoch": 0.33724,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.008853912353515625,
      "learning_rate": 0.0001,
      "loss": 4.232,
      "loss/crossentropy": 2.3477792739868164,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2215813398361206,
      "step": 16862
    },
    {
      "epoch": 0.33728,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.009919230143229167,
      "learning_rate": 0.0001,
      "loss": 4.2551,
      "loss/crossentropy": 2.474452257156372,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2185184359550476,
      "step": 16864
    },
    {
      "epoch": 0.33732,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010396067301432292,
      "learning_rate": 0.0001,
      "loss": 3.7782,
      "loss/crossentropy": 1.8387269973754883,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18137617409229279,
      "step": 16866
    },
    {
      "epoch": 0.33736,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.009924062093098958,
      "learning_rate": 0.0001,
      "loss": 3.9414,
      "loss/crossentropy": 1.8537965416908264,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19480426609516144,
      "step": 16868
    },
    {
      "epoch": 0.3374,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.010057576497395833,
      "learning_rate": 0.0001,
      "loss": 3.9808,
      "loss/crossentropy": 1.9274648427963257,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19545693695545197,
      "step": 16870
    },
    {
      "epoch": 0.33744,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.010273996988932292,
      "learning_rate": 0.0001,
      "loss": 4.1097,
      "loss/crossentropy": 1.9180024862289429,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2180539146065712,
      "step": 16872
    },
    {
      "epoch": 0.33748,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.011016591389973959,
      "learning_rate": 0.0001,
      "loss": 4.4008,
      "loss/crossentropy": 2.185365915298462,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2225860357284546,
      "step": 16874
    },
    {
      "epoch": 0.33752,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.011523183186848958,
      "learning_rate": 0.0001,
      "loss": 3.9882,
      "loss/crossentropy": 1.9749634861946106,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19897626340389252,
      "step": 16876
    },
    {
      "epoch": 0.33756,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.010284169514973959,
      "learning_rate": 0.0001,
      "loss": 4.237,
      "loss/crossentropy": 2.1505807638168335,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20921844244003296,
      "step": 16878
    },
    {
      "epoch": 0.3376,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.01002197265625,
      "learning_rate": 0.0001,
      "loss": 4.0692,
      "loss/crossentropy": 1.7159577012062073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19649318605661392,
      "step": 16880
    },
    {
      "epoch": 0.33764,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.010251617431640625,
      "learning_rate": 0.0001,
      "loss": 4.1615,
      "loss/crossentropy": 2.187040388584137,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20251552015542984,
      "step": 16882
    },
    {
      "epoch": 0.33768,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.014609527587890626,
      "learning_rate": 0.0001,
      "loss": 4.0587,
      "loss/crossentropy": 1.9247627258300781,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17710037529468536,
      "step": 16884
    },
    {
      "epoch": 0.33772,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0141845703125,
      "learning_rate": 0.0001,
      "loss": 4.001,
      "loss/crossentropy": 1.8499276041984558,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18950388580560684,
      "step": 16886
    },
    {
      "epoch": 0.33776,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.0138824462890625,
      "learning_rate": 0.0001,
      "loss": 4.0998,
      "loss/crossentropy": 2.0127341747283936,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18379998207092285,
      "step": 16888
    },
    {
      "epoch": 0.3378,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.013602447509765626,
      "learning_rate": 0.0001,
      "loss": 3.9096,
      "loss/crossentropy": 2.2561115026474,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20785757154226303,
      "step": 16890
    },
    {
      "epoch": 0.33784,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0131103515625,
      "learning_rate": 0.0001,
      "loss": 3.9638,
      "loss/crossentropy": 1.9391398429870605,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19481037557125092,
      "step": 16892
    },
    {
      "epoch": 0.33788,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.014827219645182292,
      "learning_rate": 0.0001,
      "loss": 4.2368,
      "loss/crossentropy": 2.3129481077194214,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2065483033657074,
      "step": 16894
    },
    {
      "epoch": 0.33792,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.01984430948893229,
      "learning_rate": 0.0001,
      "loss": 4.0774,
      "loss/crossentropy": 1.8482372760772705,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19291523844003677,
      "step": 16896
    },
    {
      "epoch": 0.33796,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.020031483968098958,
      "learning_rate": 0.0001,
      "loss": 4.3876,
      "loss/crossentropy": 1.973912537097931,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22137422114610672,
      "step": 16898
    },
    {
      "epoch": 0.338,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.01654841105143229,
      "learning_rate": 0.0001,
      "loss": 3.9599,
      "loss/crossentropy": 1.7987132668495178,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18385548144578934,
      "step": 16900
    },
    {
      "epoch": 0.33804,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.01727879842122396,
      "learning_rate": 0.0001,
      "loss": 4.3673,
      "loss/crossentropy": 2.199060797691345,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2016109600663185,
      "step": 16902
    },
    {
      "epoch": 0.33808,
      "grad_norm": 3.046875,
      "grad_norm_var": 0.0830718994140625,
      "learning_rate": 0.0001,
      "loss": 4.27,
      "loss/crossentropy": 2.179081439971924,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21342043578624725,
      "step": 16904
    },
    {
      "epoch": 0.33812,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0810455322265625,
      "learning_rate": 0.0001,
      "loss": 3.9861,
      "loss/crossentropy": 1.953968107700348,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20462830364704132,
      "step": 16906
    },
    {
      "epoch": 0.33816,
      "grad_norm": 2.0,
      "grad_norm_var": 0.08123270670572917,
      "learning_rate": 0.0001,
      "loss": 4.3432,
      "loss/crossentropy": 2.3202494382858276,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21390480548143387,
      "step": 16908
    },
    {
      "epoch": 0.3382,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.07872899373372395,
      "learning_rate": 0.0001,
      "loss": 4.3085,
      "loss/crossentropy": 2.085767686367035,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20946675539016724,
      "step": 16910
    },
    {
      "epoch": 0.33824,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.07372945149739583,
      "learning_rate": 0.0001,
      "loss": 4.217,
      "loss/crossentropy": 1.789705514907837,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2247294932603836,
      "step": 16912
    },
    {
      "epoch": 0.33828,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.07440770467122396,
      "learning_rate": 0.0001,
      "loss": 4.0409,
      "loss/crossentropy": 1.9354140758514404,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17958863079547882,
      "step": 16914
    },
    {
      "epoch": 0.33832,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.07529296875,
      "learning_rate": 0.0001,
      "loss": 3.6949,
      "loss/crossentropy": 1.878045916557312,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1812344640493393,
      "step": 16916
    },
    {
      "epoch": 0.33836,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.0769488016764323,
      "learning_rate": 0.0001,
      "loss": 3.9392,
      "loss/crossentropy": 1.7474132776260376,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19444099068641663,
      "step": 16918
    },
    {
      "epoch": 0.3384,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.007185618082682292,
      "learning_rate": 0.0001,
      "loss": 4.1524,
      "loss/crossentropy": 2.0462751984596252,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20077970623970032,
      "step": 16920
    },
    {
      "epoch": 0.33844,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.006912994384765625,
      "learning_rate": 0.0001,
      "loss": 4.205,
      "loss/crossentropy": 2.1104917526245117,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21799270808696747,
      "step": 16922
    },
    {
      "epoch": 0.33848,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.007083892822265625,
      "learning_rate": 0.0001,
      "loss": 4.1915,
      "loss/crossentropy": 2.11525696516037,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20585542917251587,
      "step": 16924
    },
    {
      "epoch": 0.33852,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.005222320556640625,
      "learning_rate": 0.0001,
      "loss": 4.3629,
      "loss/crossentropy": 2.3451485633850098,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22672247886657715,
      "step": 16926
    },
    {
      "epoch": 0.33856,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.004906209309895834,
      "learning_rate": 0.0001,
      "loss": 4.0493,
      "loss/crossentropy": 1.8872862458229065,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1973365843296051,
      "step": 16928
    },
    {
      "epoch": 0.3386,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.00777587890625,
      "learning_rate": 0.0001,
      "loss": 4.1082,
      "loss/crossentropy": 1.9465213418006897,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2297648787498474,
      "step": 16930
    },
    {
      "epoch": 0.33864,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0084625244140625,
      "learning_rate": 0.0001,
      "loss": 3.9397,
      "loss/crossentropy": 1.7463516592979431,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.193784698843956,
      "step": 16932
    },
    {
      "epoch": 0.33868,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.006738026936848958,
      "learning_rate": 0.0001,
      "loss": 4.1561,
      "loss/crossentropy": 2.2214877605438232,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2064037024974823,
      "step": 16934
    },
    {
      "epoch": 0.33872,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.007600657145182292,
      "learning_rate": 0.0001,
      "loss": 4.0355,
      "loss/crossentropy": 1.896401822566986,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21860874444246292,
      "step": 16936
    },
    {
      "epoch": 0.33876,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.007209269205729166,
      "learning_rate": 0.0001,
      "loss": 4.0701,
      "loss/crossentropy": 2.508669376373291,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23076221346855164,
      "step": 16938
    },
    {
      "epoch": 0.3388,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.009266916910807292,
      "learning_rate": 0.0001,
      "loss": 3.8421,
      "loss/crossentropy": 1.9412622451782227,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1833793818950653,
      "step": 16940
    },
    {
      "epoch": 0.33884,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.025406646728515624,
      "learning_rate": 0.0001,
      "loss": 4.362,
      "loss/crossentropy": 2.4757901430130005,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23629899322986603,
      "step": 16942
    },
    {
      "epoch": 0.33888,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.02580744425455729,
      "learning_rate": 0.0001,
      "loss": 4.1204,
      "loss/crossentropy": 1.788071870803833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1894787922501564,
      "step": 16944
    },
    {
      "epoch": 0.33892,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.023436482747395834,
      "learning_rate": 0.0001,
      "loss": 4.0626,
      "loss/crossentropy": 2.1902449131011963,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21310994774103165,
      "step": 16946
    },
    {
      "epoch": 0.33896,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.021394856770833335,
      "learning_rate": 0.0001,
      "loss": 3.9828,
      "loss/crossentropy": 2.0444132685661316,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1755325198173523,
      "step": 16948
    },
    {
      "epoch": 0.339,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.02200902303059896,
      "learning_rate": 0.0001,
      "loss": 4.1609,
      "loss/crossentropy": 1.9242961406707764,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18970132619142532,
      "step": 16950
    },
    {
      "epoch": 0.33904,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0216552734375,
      "learning_rate": 0.0001,
      "loss": 3.9819,
      "loss/crossentropy": 1.7046860456466675,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19807563722133636,
      "step": 16952
    },
    {
      "epoch": 0.33908,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.02184015909830729,
      "learning_rate": 0.0001,
      "loss": 3.8975,
      "loss/crossentropy": 1.8960286974906921,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17571083456277847,
      "step": 16954
    },
    {
      "epoch": 0.33912,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.019481404622395834,
      "learning_rate": 0.0001,
      "loss": 3.922,
      "loss/crossentropy": 1.9351304769515991,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1918834000825882,
      "step": 16956
    },
    {
      "epoch": 0.33916,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.017728424072265624,
      "learning_rate": 0.0001,
      "loss": 4.167,
      "loss/crossentropy": 1.974304735660553,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20828718692064285,
      "step": 16958
    },
    {
      "epoch": 0.3392,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.017867024739583334,
      "learning_rate": 0.0001,
      "loss": 3.8506,
      "loss/crossentropy": 2.3836190700531006,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2074722871184349,
      "step": 16960
    },
    {
      "epoch": 0.33924,
      "grad_norm": 1.875,
      "grad_norm_var": 0.018123372395833334,
      "learning_rate": 0.0001,
      "loss": 4.0827,
      "loss/crossentropy": 2.1663198471069336,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20097027719020844,
      "step": 16962
    },
    {
      "epoch": 0.33928,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.01810277303059896,
      "learning_rate": 0.0001,
      "loss": 4.2466,
      "loss/crossentropy": 2.275226354598999,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22942040860652924,
      "step": 16964
    },
    {
      "epoch": 0.33932,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.017796834309895832,
      "learning_rate": 0.0001,
      "loss": 3.9802,
      "loss/crossentropy": 1.7741501331329346,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1822734773159027,
      "step": 16966
    },
    {
      "epoch": 0.33936,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.017215983072916666,
      "learning_rate": 0.0001,
      "loss": 3.9016,
      "loss/crossentropy": 1.791014850139618,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2008860930800438,
      "step": 16968
    },
    {
      "epoch": 0.3394,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.016888173421223958,
      "learning_rate": 0.0001,
      "loss": 4.1941,
      "loss/crossentropy": 1.9239555597305298,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.192842036485672,
      "step": 16970
    },
    {
      "epoch": 0.33944,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.016788482666015625,
      "learning_rate": 0.0001,
      "loss": 4.108,
      "loss/crossentropy": 2.1543636322021484,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20888526737689972,
      "step": 16972
    },
    {
      "epoch": 0.33948,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.005012003580729166,
      "learning_rate": 0.0001,
      "loss": 3.975,
      "loss/crossentropy": 2.2322418093681335,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19024743884801865,
      "step": 16974
    },
    {
      "epoch": 0.33952,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.012422434488932292,
      "learning_rate": 0.0001,
      "loss": 4.3208,
      "loss/crossentropy": 1.9934163093566895,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26193471997976303,
      "step": 16976
    },
    {
      "epoch": 0.33956,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.013244374593098959,
      "learning_rate": 0.0001,
      "loss": 4.2753,
      "loss/crossentropy": 2.340088725090027,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2110593616962433,
      "step": 16978
    },
    {
      "epoch": 0.3396,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.03021214803059896,
      "learning_rate": 0.0001,
      "loss": 4.2047,
      "loss/crossentropy": 2.057206869125366,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2071269005537033,
      "step": 16980
    },
    {
      "epoch": 0.33964,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.030210113525390624,
      "learning_rate": 0.0001,
      "loss": 4.0262,
      "loss/crossentropy": 2.002032458782196,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19903026521205902,
      "step": 16982
    },
    {
      "epoch": 0.33968,
      "grad_norm": 2.0,
      "grad_norm_var": 0.030804189046223958,
      "learning_rate": 0.0001,
      "loss": 4.246,
      "loss/crossentropy": 2.01567679643631,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20198361575603485,
      "step": 16984
    },
    {
      "epoch": 0.33972,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.032814280192057295,
      "learning_rate": 0.0001,
      "loss": 4.0606,
      "loss/crossentropy": 1.661778211593628,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16601823270320892,
      "step": 16986
    },
    {
      "epoch": 0.33976,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.03467203776041667,
      "learning_rate": 0.0001,
      "loss": 3.9411,
      "loss/crossentropy": 2.284485101699829,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20108629018068314,
      "step": 16988
    },
    {
      "epoch": 0.3398,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.03200861612955729,
      "learning_rate": 0.0001,
      "loss": 4.1066,
      "loss/crossentropy": 2.1860578656196594,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22195414453744888,
      "step": 16990
    },
    {
      "epoch": 0.33984,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.02719904581705729,
      "learning_rate": 0.0001,
      "loss": 4.0212,
      "loss/crossentropy": 2.032013416290283,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19930274784564972,
      "step": 16992
    },
    {
      "epoch": 0.33988,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.1898577372233073,
      "learning_rate": 0.0001,
      "loss": 4.2847,
      "loss/crossentropy": 2.227185010910034,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19806954264640808,
      "step": 16994
    },
    {
      "epoch": 0.33992,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.17535807291666666,
      "learning_rate": 0.0001,
      "loss": 4.1526,
      "loss/crossentropy": 2.222020983695984,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2354518324136734,
      "step": 16996
    },
    {
      "epoch": 0.33996,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.1755767822265625,
      "learning_rate": 0.0001,
      "loss": 4.1766,
      "loss/crossentropy": 2.336432456970215,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19555255770683289,
      "step": 16998
    },
    {
      "epoch": 0.34,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.1748443603515625,
      "learning_rate": 0.0001,
      "loss": 4.3312,
      "loss/crossentropy": 2.17374986410141,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21119412034749985,
      "step": 17000
    },
    {
      "epoch": 0.34004,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.17195536295572916,
      "learning_rate": 0.0001,
      "loss": 4.1445,
      "loss/crossentropy": 2.135382056236267,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2263253927230835,
      "step": 17002
    },
    {
      "epoch": 0.34008,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.20149917602539064,
      "learning_rate": 0.0001,
      "loss": 3.811,
      "loss/crossentropy": 1.9646037220954895,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18890808522701263,
      "step": 17004
    },
    {
      "epoch": 0.34012,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.20098368326822916,
      "learning_rate": 0.0001,
      "loss": 4.0889,
      "loss/crossentropy": 1.8009583353996277,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1882813572883606,
      "step": 17006
    },
    {
      "epoch": 0.34016,
      "grad_norm": 3.109375,
      "grad_norm_var": 0.24433492024739584,
      "learning_rate": 0.0001,
      "loss": 4.4921,
      "loss/crossentropy": 2.0362982153892517,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.208044171333313,
      "step": 17008
    },
    {
      "epoch": 0.3402,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.11477432250976563,
      "learning_rate": 0.0001,
      "loss": 4.1781,
      "loss/crossentropy": 2.2055057287216187,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21141493320465088,
      "step": 17010
    },
    {
      "epoch": 0.34024,
      "grad_norm": 2.0,
      "grad_norm_var": 0.11963475545247396,
      "learning_rate": 0.0001,
      "loss": 3.772,
      "loss/crossentropy": 1.386088252067566,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16025834530591965,
      "step": 17012
    },
    {
      "epoch": 0.34028,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.11599299112955729,
      "learning_rate": 0.0001,
      "loss": 4.2389,
      "loss/crossentropy": 2.131182312965393,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20460083335638046,
      "step": 17014
    },
    {
      "epoch": 0.34032,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.11592992146809895,
      "learning_rate": 0.0001,
      "loss": 4.2681,
      "loss/crossentropy": 2.2682281732559204,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22915159910917282,
      "step": 17016
    },
    {
      "epoch": 0.34036,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.12011693318684896,
      "learning_rate": 0.0001,
      "loss": 4.0521,
      "loss/crossentropy": 2.0855059027671814,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19398894160985947,
      "step": 17018
    },
    {
      "epoch": 0.3404,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0842437744140625,
      "learning_rate": 0.0001,
      "loss": 4.0421,
      "loss/crossentropy": 1.895095944404602,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2172701731324196,
      "step": 17020
    },
    {
      "epoch": 0.34044,
      "grad_norm": 2.0,
      "grad_norm_var": 0.08455403645833333,
      "learning_rate": 0.0001,
      "loss": 4.0814,
      "loss/crossentropy": 2.10223788022995,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2144509255886078,
      "step": 17022
    },
    {
      "epoch": 0.34048,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.014241282145182292,
      "learning_rate": 0.0001,
      "loss": 4.1211,
      "loss/crossentropy": 2.1131081581115723,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21176592260599136,
      "step": 17024
    },
    {
      "epoch": 0.34052,
      "grad_norm": 2.0,
      "grad_norm_var": 0.009991200764973958,
      "learning_rate": 0.0001,
      "loss": 4.1581,
      "loss/crossentropy": 2.1125452518463135,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1931193768978119,
      "step": 17026
    },
    {
      "epoch": 0.34056,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.017775217692057293,
      "learning_rate": 0.0001,
      "loss": 4.2624,
      "loss/crossentropy": 2.213385283946991,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2576863020658493,
      "step": 17028
    },
    {
      "epoch": 0.3406,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.017773183186848958,
      "learning_rate": 0.0001,
      "loss": 4.408,
      "loss/crossentropy": 2.0793206095695496,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20338336378335953,
      "step": 17030
    },
    {
      "epoch": 0.34064,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.01807225545247396,
      "learning_rate": 0.0001,
      "loss": 4.3102,
      "loss/crossentropy": 1.8870239853858948,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18544750660657883,
      "step": 17032
    },
    {
      "epoch": 0.34068,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.017254384358723958,
      "learning_rate": 0.0001,
      "loss": 4.1538,
      "loss/crossentropy": 1.9893989562988281,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19679997861385345,
      "step": 17034
    },
    {
      "epoch": 0.34072,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.01845270792643229,
      "learning_rate": 0.0001,
      "loss": 3.9447,
      "loss/crossentropy": 2.0401668548583984,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19242482632398605,
      "step": 17036
    },
    {
      "epoch": 0.34076,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.015472157796223959,
      "learning_rate": 0.0001,
      "loss": 4.0509,
      "loss/crossentropy": 2.0786415934562683,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21781006455421448,
      "step": 17038
    },
    {
      "epoch": 0.3408,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.015533192952473959,
      "learning_rate": 0.0001,
      "loss": 3.9472,
      "loss/crossentropy": 2.143572211265564,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20608216524124146,
      "step": 17040
    },
    {
      "epoch": 0.34084,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0154693603515625,
      "learning_rate": 0.0001,
      "loss": 4.3305,
      "loss/crossentropy": 2.0578572750091553,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22006108611822128,
      "step": 17042
    },
    {
      "epoch": 0.34088,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0061920166015625,
      "learning_rate": 0.0001,
      "loss": 4.1295,
      "loss/crossentropy": 2.1016032099723816,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2059948667883873,
      "step": 17044
    },
    {
      "epoch": 0.34092,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.00438232421875,
      "learning_rate": 0.0001,
      "loss": 3.9293,
      "loss/crossentropy": 1.814449965953827,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1946963220834732,
      "step": 17046
    },
    {
      "epoch": 0.34096,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.004233551025390625,
      "learning_rate": 0.0001,
      "loss": 4.0696,
      "loss/crossentropy": 2.2421000599861145,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2114911824464798,
      "step": 17048
    },
    {
      "epoch": 0.341,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0070231119791666664,
      "learning_rate": 0.0001,
      "loss": 3.984,
      "loss/crossentropy": 2.1200218200683594,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.212583489716053,
      "step": 17050
    },
    {
      "epoch": 0.34104,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.007441965738932291,
      "learning_rate": 0.0001,
      "loss": 3.8656,
      "loss/crossentropy": 1.9486305713653564,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18631484359502792,
      "step": 17052
    },
    {
      "epoch": 0.34108,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.007721964518229167,
      "learning_rate": 0.0001,
      "loss": 4.3888,
      "loss/crossentropy": 2.4838292598724365,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23271672427654266,
      "step": 17054
    },
    {
      "epoch": 0.34112,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.008137766520182292,
      "learning_rate": 0.0001,
      "loss": 3.9099,
      "loss/crossentropy": 1.8606489896774292,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1955883800983429,
      "step": 17056
    },
    {
      "epoch": 0.34116,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.013199615478515624,
      "learning_rate": 0.0001,
      "loss": 4.2847,
      "loss/crossentropy": 2.1702204942703247,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21619878709316254,
      "step": 17058
    },
    {
      "epoch": 0.3412,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.013158162434895834,
      "learning_rate": 0.0001,
      "loss": 4.1987,
      "loss/crossentropy": 2.1011393070220947,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2120063453912735,
      "step": 17060
    },
    {
      "epoch": 0.34124,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.013142903645833334,
      "learning_rate": 0.0001,
      "loss": 4.1704,
      "loss/crossentropy": 2.267427682876587,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20903929322957993,
      "step": 17062
    },
    {
      "epoch": 0.34128,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.01620457967122396,
      "learning_rate": 0.0001,
      "loss": 4.3312,
      "loss/crossentropy": 2.125304937362671,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18540818989276886,
      "step": 17064
    },
    {
      "epoch": 0.34132,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.013925933837890625,
      "learning_rate": 0.0001,
      "loss": 3.9552,
      "loss/crossentropy": 1.8741024136543274,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19243815541267395,
      "step": 17066
    },
    {
      "epoch": 0.34136,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.023451487223307293,
      "learning_rate": 0.0001,
      "loss": 4.0252,
      "loss/crossentropy": 2.12781822681427,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23974300920963287,
      "step": 17068
    },
    {
      "epoch": 0.3414,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.02448298136393229,
      "learning_rate": 0.0001,
      "loss": 4.2728,
      "loss/crossentropy": 2.0786141753196716,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22311320155858994,
      "step": 17070
    },
    {
      "epoch": 0.34144,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.02408421834309896,
      "learning_rate": 0.0001,
      "loss": 4.1686,
      "loss/crossentropy": 2.1569892168045044,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20391946285963058,
      "step": 17072
    },
    {
      "epoch": 0.34148,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.024607086181640626,
      "learning_rate": 0.0001,
      "loss": 3.8035,
      "loss/crossentropy": 1.8860353231430054,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1937229335308075,
      "step": 17074
    },
    {
      "epoch": 0.34152,
      "grad_norm": 2.0,
      "grad_norm_var": 0.02448094685872396,
      "learning_rate": 0.0001,
      "loss": 4.0993,
      "loss/crossentropy": 2.25645911693573,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2088249772787094,
      "step": 17076
    },
    {
      "epoch": 0.34156,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.02972997029622396,
      "learning_rate": 0.0001,
      "loss": 4.3486,
      "loss/crossentropy": 2.379481792449951,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2341819554567337,
      "step": 17078
    },
    {
      "epoch": 0.3416,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0273345947265625,
      "learning_rate": 0.0001,
      "loss": 4.0841,
      "loss/crossentropy": 2.1564733386039734,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2162168025970459,
      "step": 17080
    },
    {
      "epoch": 0.34164,
      "grad_norm": 3.015625,
      "grad_norm_var": 0.0860260009765625,
      "learning_rate": 0.0001,
      "loss": 4.2526,
      "loss/crossentropy": 1.8906886577606201,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18614360690116882,
      "step": 17082
    },
    {
      "epoch": 0.34168,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.07732747395833334,
      "learning_rate": 0.0001,
      "loss": 4.0135,
      "loss/crossentropy": 2.093637704849243,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19867172092199326,
      "step": 17084
    },
    {
      "epoch": 0.34172,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0760210673014323,
      "learning_rate": 0.0001,
      "loss": 4.2364,
      "loss/crossentropy": 2.2579731941223145,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2153376340866089,
      "step": 17086
    },
    {
      "epoch": 0.34176,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.07312393188476562,
      "learning_rate": 0.0001,
      "loss": 4.3275,
      "loss/crossentropy": 2.250504732131958,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2759099751710892,
      "step": 17088
    },
    {
      "epoch": 0.3418,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.07190653483072916,
      "learning_rate": 0.0001,
      "loss": 3.8977,
      "loss/crossentropy": 1.6062138676643372,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15309542417526245,
      "step": 17090
    },
    {
      "epoch": 0.34184,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.07587865193684896,
      "learning_rate": 0.0001,
      "loss": 4.2199,
      "loss/crossentropy": 2.227494239807129,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20563329756259918,
      "step": 17092
    },
    {
      "epoch": 0.34188,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.07314631144205729,
      "learning_rate": 0.0001,
      "loss": 4.1095,
      "loss/crossentropy": 2.1047890186309814,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19775691628456116,
      "step": 17094
    },
    {
      "epoch": 0.34192,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.07301813761393229,
      "learning_rate": 0.0001,
      "loss": 4.0508,
      "loss/crossentropy": 2.044450581073761,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19867002964019775,
      "step": 17096
    },
    {
      "epoch": 0.34196,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.009511057535807292,
      "learning_rate": 0.0001,
      "loss": 4.2319,
      "loss/crossentropy": 1.9139947891235352,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20143548399209976,
      "step": 17098
    },
    {
      "epoch": 0.342,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.008162180582682291,
      "learning_rate": 0.0001,
      "loss": 3.7159,
      "loss/crossentropy": 1.6752634048461914,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17691873013973236,
      "step": 17100
    },
    {
      "epoch": 0.34204,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0156402587890625,
      "learning_rate": 0.0001,
      "loss": 4.2159,
      "loss/crossentropy": 2.277552843093872,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2049017697572708,
      "step": 17102
    },
    {
      "epoch": 0.34208,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.015608469645182291,
      "learning_rate": 0.0001,
      "loss": 4.0663,
      "loss/crossentropy": 2.458711266517639,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21726765483617783,
      "step": 17104
    },
    {
      "epoch": 0.34212,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.01627197265625,
      "learning_rate": 0.0001,
      "loss": 4.0384,
      "loss/crossentropy": 1.9302194714546204,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19446631520986557,
      "step": 17106
    },
    {
      "epoch": 0.34216,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.01622314453125,
      "learning_rate": 0.0001,
      "loss": 3.8935,
      "loss/crossentropy": 2.021213114261627,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19394385814666748,
      "step": 17108
    },
    {
      "epoch": 0.3422,
      "grad_norm": 2.0,
      "grad_norm_var": 0.01689453125,
      "learning_rate": 0.0001,
      "loss": 4.4094,
      "loss/crossentropy": 2.4856090545654297,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2156004160642624,
      "step": 17110
    },
    {
      "epoch": 0.34224,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.016869862874348957,
      "learning_rate": 0.0001,
      "loss": 4.4135,
      "loss/crossentropy": 2.410070061683655,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21203476190567017,
      "step": 17112
    },
    {
      "epoch": 0.34228,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.019136555989583335,
      "learning_rate": 0.0001,
      "loss": 4.0484,
      "loss/crossentropy": 2.026827871799469,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20053605735301971,
      "step": 17114
    },
    {
      "epoch": 0.34232,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.016283162434895835,
      "learning_rate": 0.0001,
      "loss": 4.2319,
      "loss/crossentropy": 2.1833966970443726,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2193789780139923,
      "step": 17116
    },
    {
      "epoch": 0.34236,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.010798899332682292,
      "learning_rate": 0.0001,
      "loss": 4.0921,
      "loss/crossentropy": 2.137472152709961,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21218669414520264,
      "step": 17118
    },
    {
      "epoch": 0.3424,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.013741048177083333,
      "learning_rate": 0.0001,
      "loss": 4.2064,
      "loss/crossentropy": 2.0607420206069946,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2138473093509674,
      "step": 17120
    },
    {
      "epoch": 0.34244,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0123687744140625,
      "learning_rate": 0.0001,
      "loss": 4.1787,
      "loss/crossentropy": 1.9109066724777222,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20130306482315063,
      "step": 17122
    },
    {
      "epoch": 0.34248,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.010944620768229166,
      "learning_rate": 0.0001,
      "loss": 3.7531,
      "loss/crossentropy": 2.1049917936325073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20878130197525024,
      "step": 17124
    },
    {
      "epoch": 0.34252,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.010008748372395833,
      "learning_rate": 0.0001,
      "loss": 4.0951,
      "loss/crossentropy": 1.9087567925453186,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18959704041481018,
      "step": 17126
    },
    {
      "epoch": 0.34256,
      "grad_norm": 2.125,
      "grad_norm_var": 0.010033162434895833,
      "learning_rate": 0.0001,
      "loss": 4.3854,
      "loss/crossentropy": 2.491095781326294,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21739569306373596,
      "step": 17128
    },
    {
      "epoch": 0.3426,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0081207275390625,
      "learning_rate": 0.0001,
      "loss": 4.0227,
      "loss/crossentropy": 2.045976400375366,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18868304044008255,
      "step": 17130
    },
    {
      "epoch": 0.34264,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.009318033854166666,
      "learning_rate": 0.0001,
      "loss": 4.1221,
      "loss/crossentropy": 2.021378517150879,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1955355852842331,
      "step": 17132
    },
    {
      "epoch": 0.34268,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.010247548421223959,
      "learning_rate": 0.0001,
      "loss": 4.0244,
      "loss/crossentropy": 2.052547872066498,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18654850870370865,
      "step": 17134
    },
    {
      "epoch": 0.34272,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.008886464436848958,
      "learning_rate": 0.0001,
      "loss": 4.2536,
      "loss/crossentropy": 2.1709023118019104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21335267275571823,
      "step": 17136
    },
    {
      "epoch": 0.34276,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.010477701822916666,
      "learning_rate": 0.0001,
      "loss": 4.1682,
      "loss/crossentropy": 2.310550093650818,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18719495832920074,
      "step": 17138
    },
    {
      "epoch": 0.3428,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.011262003580729167,
      "learning_rate": 0.0001,
      "loss": 4.0145,
      "loss/crossentropy": 1.9357584714889526,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1914527416229248,
      "step": 17140
    },
    {
      "epoch": 0.34284,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.012684885660807292,
      "learning_rate": 0.0001,
      "loss": 3.9673,
      "loss/crossentropy": 1.7561541199684143,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1711764633655548,
      "step": 17142
    },
    {
      "epoch": 0.34288,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.7546994527180989,
      "learning_rate": 0.0001,
      "loss": 4.562,
      "loss/crossentropy": 2.4081480503082275,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2405518889427185,
      "step": 17144
    },
    {
      "epoch": 0.34292,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.7456451416015625,
      "learning_rate": 0.0001,
      "loss": 4.4023,
      "loss/crossentropy": 2.44531512260437,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23487353324890137,
      "step": 17146
    },
    {
      "epoch": 0.34296,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.7445696512858073,
      "learning_rate": 0.0001,
      "loss": 4.0845,
      "loss/crossentropy": 1.9712305068969727,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20045534521341324,
      "step": 17148
    },
    {
      "epoch": 0.343,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.7468706766764323,
      "learning_rate": 0.0001,
      "loss": 3.8479,
      "loss/crossentropy": 1.9718485474586487,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18550659716129303,
      "step": 17150
    },
    {
      "epoch": 0.34304,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.7527414957682291,
      "learning_rate": 0.0001,
      "loss": 4.2708,
      "loss/crossentropy": 2.1901930570602417,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2082839384675026,
      "step": 17152
    },
    {
      "epoch": 0.34308,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.7463905334472656,
      "learning_rate": 0.0001,
      "loss": 4.2104,
      "loss/crossentropy": 2.098864793777466,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19527551531791687,
      "step": 17154
    },
    {
      "epoch": 0.34312,
      "grad_norm": 2.0,
      "grad_norm_var": 0.7467437744140625,
      "learning_rate": 0.0001,
      "loss": 3.9873,
      "loss/crossentropy": 1.7448402643203735,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1862412989139557,
      "step": 17156
    },
    {
      "epoch": 0.34316,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.7429603576660156,
      "learning_rate": 0.0001,
      "loss": 4.1016,
      "loss/crossentropy": 1.9973859190940857,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19645331799983978,
      "step": 17158
    },
    {
      "epoch": 0.3432,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.016874186197916665,
      "learning_rate": 0.0001,
      "loss": 4.2022,
      "loss/crossentropy": 2.2570079565048218,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20290197432041168,
      "step": 17160
    },
    {
      "epoch": 0.34324,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.008642323811848958,
      "learning_rate": 0.0001,
      "loss": 4.2304,
      "loss/crossentropy": 1.7943695187568665,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1924503892660141,
      "step": 17162
    },
    {
      "epoch": 0.34328,
      "grad_norm": 2.0,
      "grad_norm_var": 0.008225250244140624,
      "learning_rate": 0.0001,
      "loss": 4.2142,
      "loss/crossentropy": 2.1875799894332886,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1968172788619995,
      "step": 17164
    },
    {
      "epoch": 0.34332,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.007477823893229167,
      "learning_rate": 0.0001,
      "loss": 4.1844,
      "loss/crossentropy": 2.242374360561371,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21981361508369446,
      "step": 17166
    },
    {
      "epoch": 0.34336,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006754302978515625,
      "learning_rate": 0.0001,
      "loss": 4.0406,
      "loss/crossentropy": 1.8542492985725403,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19412916153669357,
      "step": 17168
    },
    {
      "epoch": 0.3434,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0067860921223958336,
      "learning_rate": 0.0001,
      "loss": 3.8804,
      "loss/crossentropy": 1.9377904534339905,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20015903562307358,
      "step": 17170
    },
    {
      "epoch": 0.34344,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.006319173177083333,
      "learning_rate": 0.0001,
      "loss": 4.1691,
      "loss/crossentropy": 2.2250781655311584,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1943892240524292,
      "step": 17172
    },
    {
      "epoch": 0.34348,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0054094950358072914,
      "learning_rate": 0.0001,
      "loss": 4.0274,
      "loss/crossentropy": 2.0463303923606873,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1931215077638626,
      "step": 17174
    },
    {
      "epoch": 0.34352,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.004984537760416667,
      "learning_rate": 0.0001,
      "loss": 3.826,
      "loss/crossentropy": 2.0576277375221252,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24007725715637207,
      "step": 17176
    },
    {
      "epoch": 0.34356,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.003999582926432292,
      "learning_rate": 0.0001,
      "loss": 3.9865,
      "loss/crossentropy": 1.8423307538032532,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17171106487512589,
      "step": 17178
    },
    {
      "epoch": 0.3436,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0053375244140625,
      "learning_rate": 0.0001,
      "loss": 4.3089,
      "loss/crossentropy": 2.05659818649292,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20058965682983398,
      "step": 17180
    },
    {
      "epoch": 0.34364,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0055620829264322914,
      "learning_rate": 0.0001,
      "loss": 3.8965,
      "loss/crossentropy": 1.926946997642517,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1770440638065338,
      "step": 17182
    },
    {
      "epoch": 0.34368,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.005098215738932292,
      "learning_rate": 0.0001,
      "loss": 3.9781,
      "loss/crossentropy": 1.8181686401367188,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19718249142169952,
      "step": 17184
    },
    {
      "epoch": 0.34372,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.006624094645182292,
      "learning_rate": 0.0001,
      "loss": 4.0732,
      "loss/crossentropy": 2.021128237247467,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1968582272529602,
      "step": 17186
    },
    {
      "epoch": 0.34376,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.008365631103515625,
      "learning_rate": 0.0001,
      "loss": 4.2861,
      "loss/crossentropy": 2.04776668548584,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22718969732522964,
      "step": 17188
    },
    {
      "epoch": 0.3438,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.008957672119140624,
      "learning_rate": 0.0001,
      "loss": 4.2196,
      "loss/crossentropy": 2.2225213050842285,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20369096100330353,
      "step": 17190
    },
    {
      "epoch": 0.34384,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.008089192708333333,
      "learning_rate": 0.0001,
      "loss": 4.1137,
      "loss/crossentropy": 1.8544179201126099,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20437808334827423,
      "step": 17192
    },
    {
      "epoch": 0.34388,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.006730143229166667,
      "learning_rate": 0.0001,
      "loss": 4.387,
      "loss/crossentropy": 2.1270517110824585,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1997273936867714,
      "step": 17194
    },
    {
      "epoch": 0.34392,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0060791015625,
      "learning_rate": 0.0001,
      "loss": 3.8062,
      "loss/crossentropy": 1.8699182271957397,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19382557272911072,
      "step": 17196
    },
    {
      "epoch": 0.34396,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006941731770833333,
      "learning_rate": 0.0001,
      "loss": 3.9864,
      "loss/crossentropy": 1.8734498023986816,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18053626269102097,
      "step": 17198
    },
    {
      "epoch": 0.344,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.008097330729166666,
      "learning_rate": 0.0001,
      "loss": 3.9234,
      "loss/crossentropy": 1.9386133551597595,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1851777583360672,
      "step": 17200
    },
    {
      "epoch": 0.34404,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0069163004557291664,
      "learning_rate": 0.0001,
      "loss": 3.9619,
      "loss/crossentropy": 2.1074278354644775,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22176695615053177,
      "step": 17202
    },
    {
      "epoch": 0.34408,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.005895741780598958,
      "learning_rate": 0.0001,
      "loss": 4.1231,
      "loss/crossentropy": 1.826387107372284,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19368141889572144,
      "step": 17204
    },
    {
      "epoch": 0.34412,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.005680084228515625,
      "learning_rate": 0.0001,
      "loss": 4.1367,
      "loss/crossentropy": 1.9109328389167786,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2023227959871292,
      "step": 17206
    },
    {
      "epoch": 0.34416,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.006517537434895833,
      "learning_rate": 0.0001,
      "loss": 4.1495,
      "loss/crossentropy": 1.8752732872962952,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17867545038461685,
      "step": 17208
    },
    {
      "epoch": 0.3442,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.00972900390625,
      "learning_rate": 0.0001,
      "loss": 3.9998,
      "loss/crossentropy": 1.95436429977417,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19947312772274017,
      "step": 17210
    },
    {
      "epoch": 0.34424,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.00947265625,
      "learning_rate": 0.0001,
      "loss": 4.0074,
      "loss/crossentropy": 1.9237809777259827,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2055458500981331,
      "step": 17212
    },
    {
      "epoch": 0.34428,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.007995351155598959,
      "learning_rate": 0.0001,
      "loss": 4.1351,
      "loss/crossentropy": 2.4497138261795044,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23067011684179306,
      "step": 17214
    },
    {
      "epoch": 0.34432,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.007791900634765625,
      "learning_rate": 0.0001,
      "loss": 3.9144,
      "loss/crossentropy": 2.33384370803833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22732949256896973,
      "step": 17216
    },
    {
      "epoch": 0.34436,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0075439453125,
      "learning_rate": 0.0001,
      "loss": 3.9414,
      "loss/crossentropy": 2.022156059741974,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2063901051878929,
      "step": 17218
    },
    {
      "epoch": 0.3444,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.007614898681640625,
      "learning_rate": 0.0001,
      "loss": 4.0642,
      "loss/crossentropy": 2.183963179588318,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21098726987838745,
      "step": 17220
    },
    {
      "epoch": 0.34444,
      "grad_norm": 6.875,
      "grad_norm_var": 1.501227823893229,
      "learning_rate": 0.0001,
      "loss": 4.3139,
      "loss/crossentropy": 2.256633758544922,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21449647843837738,
      "step": 17222
    },
    {
      "epoch": 0.34448,
      "grad_norm": 14.3125,
      "grad_norm_var": 10.478612263997396,
      "learning_rate": 0.0001,
      "loss": 3.838,
      "loss/crossentropy": 1.8303287625312805,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18412896245718002,
      "step": 17224
    },
    {
      "epoch": 0.34452,
      "grad_norm": 2.0625,
      "grad_norm_var": 10.405972290039063,
      "learning_rate": 0.0001,
      "loss": 4.1852,
      "loss/crossentropy": 2.3248833417892456,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23459070920944214,
      "step": 17226
    },
    {
      "epoch": 0.34456,
      "grad_norm": 2.109375,
      "grad_norm_var": 10.418485514322917,
      "learning_rate": 0.0001,
      "loss": 4.062,
      "loss/crossentropy": 2.018009066581726,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21266046166419983,
      "step": 17228
    },
    {
      "epoch": 0.3446,
      "grad_norm": 1.984375,
      "grad_norm_var": 10.434609985351562,
      "learning_rate": 0.0001,
      "loss": 4.186,
      "loss/crossentropy": 2.337665557861328,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.225304514169693,
      "step": 17230
    },
    {
      "epoch": 0.34464,
      "grad_norm": 1.8046875,
      "grad_norm_var": 10.453043365478516,
      "learning_rate": 0.0001,
      "loss": 3.86,
      "loss/crossentropy": 1.8581790924072266,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18770471215248108,
      "step": 17232
    },
    {
      "epoch": 0.34468,
      "grad_norm": 1.8515625,
      "grad_norm_var": 10.462247721354167,
      "learning_rate": 0.0001,
      "loss": 3.9213,
      "loss/crossentropy": 1.9208187460899353,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19106490910053253,
      "step": 17234
    },
    {
      "epoch": 0.34472,
      "grad_norm": 1.8515625,
      "grad_norm_var": 10.454630279541016,
      "learning_rate": 0.0001,
      "loss": 3.8284,
      "loss/crossentropy": 1.8844050765037537,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17407642304897308,
      "step": 17236
    },
    {
      "epoch": 0.34476,
      "grad_norm": 1.9296875,
      "grad_norm_var": 9.497085571289062,
      "learning_rate": 0.0001,
      "loss": 3.7423,
      "loss/crossentropy": 2.0443355441093445,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.198170468211174,
      "step": 17238
    },
    {
      "epoch": 0.3448,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.027342732747395834,
      "learning_rate": 0.0001,
      "loss": 4.0965,
      "loss/crossentropy": 1.8563520908355713,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1958291083574295,
      "step": 17240
    },
    {
      "epoch": 0.34484,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0127349853515625,
      "learning_rate": 0.0001,
      "loss": 4.3028,
      "loss/crossentropy": 2.1252214908599854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22386425733566284,
      "step": 17242
    },
    {
      "epoch": 0.34488,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.011848958333333333,
      "learning_rate": 0.0001,
      "loss": 3.926,
      "loss/crossentropy": 2.1379681825637817,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19781935214996338,
      "step": 17244
    },
    {
      "epoch": 0.34492,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.013529205322265625,
      "learning_rate": 0.0001,
      "loss": 3.7403,
      "loss/crossentropy": 1.9774981141090393,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18523120880126953,
      "step": 17246
    },
    {
      "epoch": 0.34496,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0120025634765625,
      "learning_rate": 0.0001,
      "loss": 4.2668,
      "loss/crossentropy": 2.3234479427337646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21939057111740112,
      "step": 17248
    },
    {
      "epoch": 0.345,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.011706288655598958,
      "learning_rate": 0.0001,
      "loss": 4.11,
      "loss/crossentropy": 2.0521583557128906,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20204473286867142,
      "step": 17250
    },
    {
      "epoch": 0.34504,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.03452860514322917,
      "learning_rate": 0.0001,
      "loss": 4.2184,
      "loss/crossentropy": 2.486477255821228,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21673013269901276,
      "step": 17252
    },
    {
      "epoch": 0.34508,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.035456339518229164,
      "learning_rate": 0.0001,
      "loss": 4.0215,
      "loss/crossentropy": 2.285652995109558,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21322200447320938,
      "step": 17254
    },
    {
      "epoch": 0.34512,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.03436686197916667,
      "learning_rate": 0.0001,
      "loss": 4.1855,
      "loss/crossentropy": 2.0946747064590454,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19791530817747116,
      "step": 17256
    },
    {
      "epoch": 0.34516,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.03250732421875,
      "learning_rate": 0.0001,
      "loss": 4.1129,
      "loss/crossentropy": 1.7509565353393555,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16754039376974106,
      "step": 17258
    },
    {
      "epoch": 0.3452,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0325347900390625,
      "learning_rate": 0.0001,
      "loss": 4.3411,
      "loss/crossentropy": 2.3828574419021606,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22555553913116455,
      "step": 17260
    },
    {
      "epoch": 0.34524,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.029060872395833333,
      "learning_rate": 0.0001,
      "loss": 4.215,
      "loss/crossentropy": 1.997941255569458,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1815095767378807,
      "step": 17262
    },
    {
      "epoch": 0.34528,
      "grad_norm": 2.25,
      "grad_norm_var": 0.0296630859375,
      "learning_rate": 0.0001,
      "loss": 4.3814,
      "loss/crossentropy": 2.3036177158355713,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2305641770362854,
      "step": 17264
    },
    {
      "epoch": 0.34532,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.03178075154622396,
      "learning_rate": 0.0001,
      "loss": 4.1668,
      "loss/crossentropy": 2.202543616294861,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19594155251979828,
      "step": 17266
    },
    {
      "epoch": 0.34536,
      "grad_norm": 1.875,
      "grad_norm_var": 0.011199696858723959,
      "learning_rate": 0.0001,
      "loss": 4.1792,
      "loss/crossentropy": 2.1162038445472717,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1945410966873169,
      "step": 17268
    },
    {
      "epoch": 0.3454,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.009883626302083334,
      "learning_rate": 0.0001,
      "loss": 4.1109,
      "loss/crossentropy": 2.273344039916992,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24327096343040466,
      "step": 17270
    },
    {
      "epoch": 0.34544,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.011351521809895833,
      "learning_rate": 0.0001,
      "loss": 3.8254,
      "loss/crossentropy": 1.851362407207489,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18494703620672226,
      "step": 17272
    },
    {
      "epoch": 0.34548,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.01114501953125,
      "learning_rate": 0.0001,
      "loss": 4.0846,
      "loss/crossentropy": 1.9861729145050049,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20077189058065414,
      "step": 17274
    },
    {
      "epoch": 0.34552,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.011716461181640625,
      "learning_rate": 0.0001,
      "loss": 4.0692,
      "loss/crossentropy": 2.0234099626541138,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20506415516138077,
      "step": 17276
    },
    {
      "epoch": 0.34556,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.012326812744140625,
      "learning_rate": 0.0001,
      "loss": 3.9843,
      "loss/crossentropy": 2.1666045784950256,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18108145147562027,
      "step": 17278
    },
    {
      "epoch": 0.3456,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0060536702473958336,
      "learning_rate": 0.0001,
      "loss": 4.0877,
      "loss/crossentropy": 2.066905975341797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2112976461648941,
      "step": 17280
    },
    {
      "epoch": 0.34564,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0059722900390625,
      "learning_rate": 0.0001,
      "loss": 4.224,
      "loss/crossentropy": 2.1789294481277466,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20467212051153183,
      "step": 17282
    },
    {
      "epoch": 0.34568,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.005736287434895833,
      "learning_rate": 0.0001,
      "loss": 3.9716,
      "loss/crossentropy": 1.8310245275497437,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1974945366382599,
      "step": 17284
    },
    {
      "epoch": 0.34572,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.006563059488932292,
      "learning_rate": 0.0001,
      "loss": 4.1832,
      "loss/crossentropy": 2.033496856689453,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1989584118127823,
      "step": 17286
    },
    {
      "epoch": 0.34576,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0091705322265625,
      "learning_rate": 0.0001,
      "loss": 3.8748,
      "loss/crossentropy": 1.8883211016654968,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1889789029955864,
      "step": 17288
    },
    {
      "epoch": 0.3458,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008495076497395834,
      "learning_rate": 0.0001,
      "loss": 4.1637,
      "loss/crossentropy": 2.4520593881607056,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2183564007282257,
      "step": 17290
    },
    {
      "epoch": 0.34584,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.008272298177083333,
      "learning_rate": 0.0001,
      "loss": 3.9375,
      "loss/crossentropy": 2.030746340751648,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19382008165121078,
      "step": 17292
    },
    {
      "epoch": 0.34588,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.007295735677083333,
      "learning_rate": 0.0001,
      "loss": 4.2182,
      "loss/crossentropy": 1.9897491931915283,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21130450069904327,
      "step": 17294
    },
    {
      "epoch": 0.34592,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.006883748372395833,
      "learning_rate": 0.0001,
      "loss": 3.982,
      "loss/crossentropy": 1.8000388145446777,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20359987765550613,
      "step": 17296
    },
    {
      "epoch": 0.34596,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0065610249837239586,
      "learning_rate": 0.0001,
      "loss": 4.0341,
      "loss/crossentropy": 2.0408164262771606,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18042601644992828,
      "step": 17298
    },
    {
      "epoch": 0.346,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.006742350260416667,
      "learning_rate": 0.0001,
      "loss": 4.0849,
      "loss/crossentropy": 2.28415846824646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20799801498651505,
      "step": 17300
    },
    {
      "epoch": 0.34604,
      "grad_norm": 2.25,
      "grad_norm_var": 0.010925038655598959,
      "learning_rate": 0.0001,
      "loss": 4.3258,
      "loss/crossentropy": 1.9420422315597534,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20019559562206268,
      "step": 17302
    },
    {
      "epoch": 0.34608,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.0102691650390625,
      "learning_rate": 0.0001,
      "loss": 4.1892,
      "loss/crossentropy": 2.100687026977539,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21129299700260162,
      "step": 17304
    },
    {
      "epoch": 0.34612,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.010970052083333333,
      "learning_rate": 0.0001,
      "loss": 4.0209,
      "loss/crossentropy": 1.8657938241958618,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2081414759159088,
      "step": 17306
    },
    {
      "epoch": 0.34616,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.012853749593098958,
      "learning_rate": 0.0001,
      "loss": 4.025,
      "loss/crossentropy": 2.0629169940948486,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21117018163204193,
      "step": 17308
    },
    {
      "epoch": 0.3462,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0126129150390625,
      "learning_rate": 0.0001,
      "loss": 4.159,
      "loss/crossentropy": 2.2877084016799927,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1959577053785324,
      "step": 17310
    },
    {
      "epoch": 0.34624,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.012280019124348958,
      "learning_rate": 0.0001,
      "loss": 4.1135,
      "loss/crossentropy": 2.2501026391983032,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2108026072382927,
      "step": 17312
    },
    {
      "epoch": 0.34628,
      "grad_norm": 2.0,
      "grad_norm_var": 0.012011464436848958,
      "learning_rate": 0.0001,
      "loss": 4.1498,
      "loss/crossentropy": 1.9485750794410706,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2126113697886467,
      "step": 17314
    },
    {
      "epoch": 0.34632,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.011500803629557292,
      "learning_rate": 0.0001,
      "loss": 4.13,
      "loss/crossentropy": 2.193112373352051,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19680321961641312,
      "step": 17316
    },
    {
      "epoch": 0.34636,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.010084788004557291,
      "learning_rate": 0.0001,
      "loss": 4.1181,
      "loss/crossentropy": 2.30054771900177,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2211720421910286,
      "step": 17318
    },
    {
      "epoch": 0.3464,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.026869455973307293,
      "learning_rate": 0.0001,
      "loss": 4.2737,
      "loss/crossentropy": 2.0091487169265747,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19040243327617645,
      "step": 17320
    },
    {
      "epoch": 0.34644,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.036628977457682295,
      "learning_rate": 0.0001,
      "loss": 4.2257,
      "loss/crossentropy": 2.1841423511505127,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2112671434879303,
      "step": 17322
    },
    {
      "epoch": 0.34648,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.03555501302083333,
      "learning_rate": 0.0001,
      "loss": 3.8837,
      "loss/crossentropy": 2.092822253704071,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2037336230278015,
      "step": 17324
    },
    {
      "epoch": 0.34652,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0335601806640625,
      "learning_rate": 0.0001,
      "loss": 4.0322,
      "loss/crossentropy": 2.280580163002014,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20753052085638046,
      "step": 17326
    },
    {
      "epoch": 0.34656,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.03884048461914062,
      "learning_rate": 0.0001,
      "loss": 4.1206,
      "loss/crossentropy": 1.60361909866333,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15820877254009247,
      "step": 17328
    },
    {
      "epoch": 0.3466,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.04915949503580729,
      "learning_rate": 0.0001,
      "loss": 3.6214,
      "loss/crossentropy": 1.608814001083374,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16808264702558517,
      "step": 17330
    },
    {
      "epoch": 0.34664,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.06932373046875,
      "learning_rate": 0.0001,
      "loss": 4.4983,
      "loss/crossentropy": 2.3868669271469116,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23717772960662842,
      "step": 17332
    },
    {
      "epoch": 0.34668,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.06720759073893229,
      "learning_rate": 0.0001,
      "loss": 4.0862,
      "loss/crossentropy": 2.1908692717552185,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21705647557973862,
      "step": 17334
    },
    {
      "epoch": 0.34672,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.05272191365559896,
      "learning_rate": 0.0001,
      "loss": 4.2436,
      "loss/crossentropy": 2.459980010986328,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24167515337467194,
      "step": 17336
    },
    {
      "epoch": 0.34676,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.04353612263997396,
      "learning_rate": 0.0001,
      "loss": 4.219,
      "loss/crossentropy": 2.221674919128418,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2097555324435234,
      "step": 17338
    },
    {
      "epoch": 0.3468,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.04521458943684896,
      "learning_rate": 0.0001,
      "loss": 3.7565,
      "loss/crossentropy": 2.2840858697891235,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2250964418053627,
      "step": 17340
    },
    {
      "epoch": 0.34684,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.04426854451497396,
      "learning_rate": 0.0001,
      "loss": 3.6803,
      "loss/crossentropy": 2.0715824365615845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2149435505270958,
      "step": 17342
    },
    {
      "epoch": 0.34688,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0420806884765625,
      "learning_rate": 0.0001,
      "loss": 4.1017,
      "loss/crossentropy": 2.2389899492263794,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22377658635377884,
      "step": 17344
    },
    {
      "epoch": 0.34692,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.035278065999348955,
      "learning_rate": 0.0001,
      "loss": 3.9827,
      "loss/crossentropy": 2.0308732986450195,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2074751853942871,
      "step": 17346
    },
    {
      "epoch": 0.34696,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.010864003499348959,
      "learning_rate": 0.0001,
      "loss": 3.7901,
      "loss/crossentropy": 1.8123770356178284,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19080865383148193,
      "step": 17348
    },
    {
      "epoch": 0.347,
      "grad_norm": 1.734375,
      "grad_norm_var": 0.010367838541666667,
      "learning_rate": 0.0001,
      "loss": 3.8772,
      "loss/crossentropy": 1.884379267692566,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1799243837594986,
      "step": 17350
    },
    {
      "epoch": 0.34704,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.012442779541015626,
      "learning_rate": 0.0001,
      "loss": 4.4317,
      "loss/crossentropy": 2.237663149833679,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20197343081235886,
      "step": 17352
    },
    {
      "epoch": 0.34708,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.013337961832682292,
      "learning_rate": 0.0001,
      "loss": 3.7444,
      "loss/crossentropy": 1.9563414454460144,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18722663074731827,
      "step": 17354
    },
    {
      "epoch": 0.34712,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.020817057291666666,
      "learning_rate": 0.0001,
      "loss": 4.2799,
      "loss/crossentropy": 2.1188591718673706,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21183937788009644,
      "step": 17356
    },
    {
      "epoch": 0.34716,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0219146728515625,
      "learning_rate": 0.0001,
      "loss": 4.4551,
      "loss/crossentropy": 2.276149272918701,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2232571393251419,
      "step": 17358
    },
    {
      "epoch": 0.3472,
      "grad_norm": 2.0,
      "grad_norm_var": 0.02123998006184896,
      "learning_rate": 0.0001,
      "loss": 4.4329,
      "loss/crossentropy": 2.360138177871704,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22683367878198624,
      "step": 17360
    },
    {
      "epoch": 0.34724,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.020576985677083333,
      "learning_rate": 0.0001,
      "loss": 4.2762,
      "loss/crossentropy": 1.99272620677948,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18479204177856445,
      "step": 17362
    },
    {
      "epoch": 0.34728,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.019760894775390624,
      "learning_rate": 0.0001,
      "loss": 4.2303,
      "loss/crossentropy": 2.053771197795868,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20591863989830017,
      "step": 17364
    },
    {
      "epoch": 0.34732,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.016428375244140626,
      "learning_rate": 0.0001,
      "loss": 3.9703,
      "loss/crossentropy": 2.1311055421829224,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2049081027507782,
      "step": 17366
    },
    {
      "epoch": 0.34736,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.01246337890625,
      "learning_rate": 0.0001,
      "loss": 4.1102,
      "loss/crossentropy": 2.0712148547172546,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19809392094612122,
      "step": 17368
    },
    {
      "epoch": 0.3474,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.009468587239583333,
      "learning_rate": 0.0001,
      "loss": 4.0732,
      "loss/crossentropy": 2.144728183746338,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21964795887470245,
      "step": 17370
    },
    {
      "epoch": 0.34744,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.003856404622395833,
      "learning_rate": 0.0001,
      "loss": 3.9145,
      "loss/crossentropy": 2.024275004863739,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19414371997117996,
      "step": 17372
    },
    {
      "epoch": 0.34748,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.00400390625,
      "learning_rate": 0.0001,
      "loss": 4.0925,
      "loss/crossentropy": 2.460733413696289,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2185375839471817,
      "step": 17374
    },
    {
      "epoch": 0.34752,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.010423787434895833,
      "learning_rate": 0.0001,
      "loss": 4.308,
      "loss/crossentropy": 2.3502203226089478,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2140796035528183,
      "step": 17376
    },
    {
      "epoch": 0.34756,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.012547810872395834,
      "learning_rate": 0.0001,
      "loss": 4.0238,
      "loss/crossentropy": 2.2896264791488647,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2149006575345993,
      "step": 17378
    },
    {
      "epoch": 0.3476,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.013118489583333334,
      "learning_rate": 0.0001,
      "loss": 3.9757,
      "loss/crossentropy": 2.101209044456482,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20842310786247253,
      "step": 17380
    },
    {
      "epoch": 0.34764,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0189361572265625,
      "learning_rate": 0.0001,
      "loss": 4.3287,
      "loss/crossentropy": 1.8995551466941833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1851196512579918,
      "step": 17382
    },
    {
      "epoch": 0.34768,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.01887995402018229,
      "learning_rate": 0.0001,
      "loss": 3.9219,
      "loss/crossentropy": 1.850643277168274,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16962846368551254,
      "step": 17384
    },
    {
      "epoch": 0.34772,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.019528961181640624,
      "learning_rate": 0.0001,
      "loss": 4.2478,
      "loss/crossentropy": 2.1131063103675842,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19152897596359253,
      "step": 17386
    },
    {
      "epoch": 0.34776,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.020140584309895834,
      "learning_rate": 0.0001,
      "loss": 4.1932,
      "loss/crossentropy": 2.172673463821411,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20645174384117126,
      "step": 17388
    },
    {
      "epoch": 0.3478,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.019147745768229165,
      "learning_rate": 0.0001,
      "loss": 4.269,
      "loss/crossentropy": 2.431947708129883,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20513125509023666,
      "step": 17390
    },
    {
      "epoch": 0.34784,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.014461008707682292,
      "learning_rate": 0.0001,
      "loss": 3.9314,
      "loss/crossentropy": 2.062328279018402,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1929447129368782,
      "step": 17392
    },
    {
      "epoch": 0.34788,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010609690348307292,
      "learning_rate": 0.0001,
      "loss": 4.2277,
      "loss/crossentropy": 2.189133882522583,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20945163816213608,
      "step": 17394
    },
    {
      "epoch": 0.34792,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.010846964518229167,
      "learning_rate": 0.0001,
      "loss": 3.9426,
      "loss/crossentropy": 2.070523977279663,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20401575416326523,
      "step": 17396
    },
    {
      "epoch": 0.34796,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0075266520182291664,
      "learning_rate": 0.0001,
      "loss": 4.0668,
      "loss/crossentropy": 2.3387876749038696,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19960296899080276,
      "step": 17398
    },
    {
      "epoch": 0.348,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.00955810546875,
      "learning_rate": 0.0001,
      "loss": 4.0225,
      "loss/crossentropy": 1.9857355952262878,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19195467978715897,
      "step": 17400
    },
    {
      "epoch": 0.34804,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.009041086832682291,
      "learning_rate": 0.0001,
      "loss": 4.0715,
      "loss/crossentropy": 2.2523770332336426,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22015457600355148,
      "step": 17402
    },
    {
      "epoch": 0.34808,
      "grad_norm": 1.875,
      "grad_norm_var": 0.0074045817057291664,
      "learning_rate": 0.0001,
      "loss": 3.9181,
      "loss/crossentropy": 1.7088012099266052,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1644933521747589,
      "step": 17404
    },
    {
      "epoch": 0.34812,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.006772613525390625,
      "learning_rate": 0.0001,
      "loss": 3.8759,
      "loss/crossentropy": 2.022001802921295,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19237526506185532,
      "step": 17406
    },
    {
      "epoch": 0.34816,
      "grad_norm": 2.125,
      "grad_norm_var": 0.015669504801432293,
      "learning_rate": 0.0001,
      "loss": 4.2196,
      "loss/crossentropy": 2.240332841873169,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2214636355638504,
      "step": 17408
    },
    {
      "epoch": 0.3482,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.015099843343098959,
      "learning_rate": 0.0001,
      "loss": 4.0211,
      "loss/crossentropy": 2.1392452716827393,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21769221127033234,
      "step": 17410
    },
    {
      "epoch": 0.34824,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.016078440348307292,
      "learning_rate": 0.0001,
      "loss": 4.0243,
      "loss/crossentropy": 1.9182460308074951,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.202327162027359,
      "step": 17412
    },
    {
      "epoch": 0.34828,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0153961181640625,
      "learning_rate": 0.0001,
      "loss": 4.0654,
      "loss/crossentropy": 2.036426305770874,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19997400790452957,
      "step": 17414
    },
    {
      "epoch": 0.34832,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.014353179931640625,
      "learning_rate": 0.0001,
      "loss": 3.9086,
      "loss/crossentropy": 2.0429354906082153,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20519915968179703,
      "step": 17416
    },
    {
      "epoch": 0.34836,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.013923136393229167,
      "learning_rate": 0.0001,
      "loss": 4.2742,
      "loss/crossentropy": 2.3943710327148438,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2106815129518509,
      "step": 17418
    },
    {
      "epoch": 0.3484,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0130615234375,
      "learning_rate": 0.0001,
      "loss": 4.1596,
      "loss/crossentropy": 2.060115098953247,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20672088861465454,
      "step": 17420
    },
    {
      "epoch": 0.34844,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.012471516927083334,
      "learning_rate": 0.0001,
      "loss": 4.2223,
      "loss/crossentropy": 2.2820589542388916,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23306988179683685,
      "step": 17422
    },
    {
      "epoch": 0.34848,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.008788045247395833,
      "learning_rate": 0.0001,
      "loss": 4.1665,
      "loss/crossentropy": 2.13969624042511,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20909114927053452,
      "step": 17424
    },
    {
      "epoch": 0.34852,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.011214192708333333,
      "learning_rate": 0.0001,
      "loss": 3.9572,
      "loss/crossentropy": 2.0822505950927734,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20429539680480957,
      "step": 17426
    },
    {
      "epoch": 0.34856,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.009759267171223959,
      "learning_rate": 0.0001,
      "loss": 3.9435,
      "loss/crossentropy": 2.512156844139099,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21776874363422394,
      "step": 17428
    },
    {
      "epoch": 0.3486,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.018700917561848957,
      "learning_rate": 0.0001,
      "loss": 4.1145,
      "loss/crossentropy": 2.0379759669303894,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20926732569932938,
      "step": 17430
    },
    {
      "epoch": 0.34864,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.017295074462890626,
      "learning_rate": 0.0001,
      "loss": 4.1998,
      "loss/crossentropy": 2.443945050239563,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22597475349903107,
      "step": 17432
    },
    {
      "epoch": 0.34868,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.019576009114583334,
      "learning_rate": 0.0001,
      "loss": 3.8442,
      "loss/crossentropy": 1.8295652866363525,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18034228682518005,
      "step": 17434
    },
    {
      "epoch": 0.34872,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.020428212483723958,
      "learning_rate": 0.0001,
      "loss": 4.0614,
      "loss/crossentropy": 1.9487475156784058,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18921684473752975,
      "step": 17436
    },
    {
      "epoch": 0.34876,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.018721262613932293,
      "learning_rate": 0.0001,
      "loss": 4.1153,
      "loss/crossentropy": 2.099911689758301,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2079886943101883,
      "step": 17438
    },
    {
      "epoch": 0.3488,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.016739654541015624,
      "learning_rate": 0.0001,
      "loss": 4.1762,
      "loss/crossentropy": 2.52177894115448,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22080882638692856,
      "step": 17440
    },
    {
      "epoch": 0.34884,
      "grad_norm": 2.125,
      "grad_norm_var": 0.014900461832682291,
      "learning_rate": 0.0001,
      "loss": 4.1076,
      "loss/crossentropy": 2.222637891769409,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22301562875509262,
      "step": 17442
    },
    {
      "epoch": 0.34888,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.014725494384765624,
      "learning_rate": 0.0001,
      "loss": 4.3965,
      "loss/crossentropy": 2.2042930126190186,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2051277905702591,
      "step": 17444
    },
    {
      "epoch": 0.34892,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0065093994140625,
      "learning_rate": 0.0001,
      "loss": 4.2452,
      "loss/crossentropy": 2.0599029064178467,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20651433616876602,
      "step": 17446
    },
    {
      "epoch": 0.34896,
      "grad_norm": 2.0,
      "grad_norm_var": 0.007027180989583334,
      "learning_rate": 0.0001,
      "loss": 3.9174,
      "loss/crossentropy": 1.9319151639938354,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18495241552591324,
      "step": 17448
    },
    {
      "epoch": 0.349,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.006197102864583333,
      "learning_rate": 0.0001,
      "loss": 4.0844,
      "loss/crossentropy": 2.341706871986389,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20252344757318497,
      "step": 17450
    },
    {
      "epoch": 0.34904,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.004833984375,
      "learning_rate": 0.0001,
      "loss": 4.0301,
      "loss/crossentropy": 2.0161609053611755,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2176668643951416,
      "step": 17452
    },
    {
      "epoch": 0.34908,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.0062334696451822914,
      "learning_rate": 0.0001,
      "loss": 3.6046,
      "loss/crossentropy": 1.7961083054542542,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16384318470954895,
      "step": 17454
    },
    {
      "epoch": 0.34912,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.01791966756184896,
      "learning_rate": 0.0001,
      "loss": 4.6054,
      "loss/crossentropy": 2.108873188495636,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21137713640928268,
      "step": 17456
    },
    {
      "epoch": 0.34916,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.017830403645833333,
      "learning_rate": 0.0001,
      "loss": 3.9543,
      "loss/crossentropy": 2.158664584159851,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19957562536001205,
      "step": 17458
    },
    {
      "epoch": 0.3492,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.018660227457682293,
      "learning_rate": 0.0001,
      "loss": 3.9454,
      "loss/crossentropy": 1.7799381017684937,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.193112812936306,
      "step": 17460
    },
    {
      "epoch": 0.34924,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.019128163655598957,
      "learning_rate": 0.0001,
      "loss": 4.4072,
      "loss/crossentropy": 2.372501015663147,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2180323675274849,
      "step": 17462
    },
    {
      "epoch": 0.34928,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0185699462890625,
      "learning_rate": 0.0001,
      "loss": 4.0343,
      "loss/crossentropy": 1.9338968396186829,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19117196649312973,
      "step": 17464
    },
    {
      "epoch": 0.34932,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.020774078369140626,
      "learning_rate": 0.0001,
      "loss": 3.6429,
      "loss/crossentropy": 1.4602742195129395,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1677519753575325,
      "step": 17466
    },
    {
      "epoch": 0.34936,
      "grad_norm": 2.0,
      "grad_norm_var": 0.02080078125,
      "learning_rate": 0.0001,
      "loss": 4.2498,
      "loss/crossentropy": 2.1464006304740906,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21201838552951813,
      "step": 17468
    },
    {
      "epoch": 0.3494,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.018529256184895832,
      "learning_rate": 0.0001,
      "loss": 4.1736,
      "loss/crossentropy": 2.071315884590149,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19322700798511505,
      "step": 17470
    },
    {
      "epoch": 0.34944,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.010309855143229166,
      "learning_rate": 0.0001,
      "loss": 4.2653,
      "loss/crossentropy": 1.9166500568389893,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23960981518030167,
      "step": 17472
    },
    {
      "epoch": 0.34948,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.011457316080729167,
      "learning_rate": 0.0001,
      "loss": 3.9433,
      "loss/crossentropy": 1.8666390180587769,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1875123456120491,
      "step": 17474
    },
    {
      "epoch": 0.34952,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.009736887613932292,
      "learning_rate": 0.0001,
      "loss": 4.0984,
      "loss/crossentropy": 1.9197289943695068,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21075090020895004,
      "step": 17476
    },
    {
      "epoch": 0.34956,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.011018625895182292,
      "learning_rate": 0.0001,
      "loss": 4.0971,
      "loss/crossentropy": 1.904780387878418,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20412997901439667,
      "step": 17478
    },
    {
      "epoch": 0.3496,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.012325032552083334,
      "learning_rate": 0.0001,
      "loss": 3.735,
      "loss/crossentropy": 1.6775096654891968,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1669272631406784,
      "step": 17480
    },
    {
      "epoch": 0.34964,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0089111328125,
      "learning_rate": 0.0001,
      "loss": 4.1039,
      "loss/crossentropy": 2.050463318824768,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19382070004940033,
      "step": 17482
    },
    {
      "epoch": 0.34968,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.009224192301432291,
      "learning_rate": 0.0001,
      "loss": 3.8739,
      "loss/crossentropy": 1.8124673962593079,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20674846321344376,
      "step": 17484
    },
    {
      "epoch": 0.34972,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.008969879150390625,
      "learning_rate": 0.0001,
      "loss": 3.6797,
      "loss/crossentropy": 1.6487451791763306,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17874807119369507,
      "step": 17486
    },
    {
      "epoch": 0.34976,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.009430948893229167,
      "learning_rate": 0.0001,
      "loss": 4.4025,
      "loss/crossentropy": 2.101687431335449,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24856076389551163,
      "step": 17488
    },
    {
      "epoch": 0.3498,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.009205881754557292,
      "learning_rate": 0.0001,
      "loss": 4.0236,
      "loss/crossentropy": 1.9233632683753967,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21046236902475357,
      "step": 17490
    },
    {
      "epoch": 0.34984,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.010589345296223959,
      "learning_rate": 0.0001,
      "loss": 4.0376,
      "loss/crossentropy": 2.2882679104804993,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23568424582481384,
      "step": 17492
    },
    {
      "epoch": 0.34988,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.022342681884765625,
      "learning_rate": 0.0001,
      "loss": 4.4205,
      "loss/crossentropy": 2.205874502658844,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21892941743135452,
      "step": 17494
    },
    {
      "epoch": 0.34992,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.0232177734375,
      "learning_rate": 0.0001,
      "loss": 4.3628,
      "loss/crossentropy": 2.278464913368225,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22409500926733017,
      "step": 17496
    },
    {
      "epoch": 0.34996,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.026082356770833332,
      "learning_rate": 0.0001,
      "loss": 4.0736,
      "loss/crossentropy": 2.5003273487091064,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22478827834129333,
      "step": 17498
    },
    {
      "epoch": 0.35,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.04237848917643229,
      "learning_rate": 0.0001,
      "loss": 4.3658,
      "loss/crossentropy": 2.034249722957611,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19176610559225082,
      "step": 17500
    },
    {
      "epoch": 0.35004,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0401031494140625,
      "learning_rate": 0.0001,
      "loss": 4.1586,
      "loss/crossentropy": 2.271065592765808,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21523310244083405,
      "step": 17502
    },
    {
      "epoch": 0.35008,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.03870824178059896,
      "learning_rate": 0.0001,
      "loss": 4.1198,
      "loss/crossentropy": 2.182866334915161,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21311558783054352,
      "step": 17504
    },
    {
      "epoch": 0.35012,
      "grad_norm": 2.125,
      "grad_norm_var": 0.03878173828125,
      "learning_rate": 0.0001,
      "loss": 4.1471,
      "loss/crossentropy": 1.7134016752243042,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16884544491767883,
      "step": 17506
    },
    {
      "epoch": 0.35016,
      "grad_norm": 2.125,
      "grad_norm_var": 0.03687718709309896,
      "learning_rate": 0.0001,
      "loss": 4.2464,
      "loss/crossentropy": 2.198198080062866,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1916792169213295,
      "step": 17508
    },
    {
      "epoch": 0.3502,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0272125244140625,
      "learning_rate": 0.0001,
      "loss": 4.2651,
      "loss/crossentropy": 1.9370547533035278,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23538394272327423,
      "step": 17510
    },
    {
      "epoch": 0.35024,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.02469456990559896,
      "learning_rate": 0.0001,
      "loss": 4.0664,
      "loss/crossentropy": 2.1743874549865723,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22514399886131287,
      "step": 17512
    },
    {
      "epoch": 0.35028,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.02174657185872396,
      "learning_rate": 0.0001,
      "loss": 4.2767,
      "loss/crossentropy": 1.8546866178512573,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1772822067141533,
      "step": 17514
    },
    {
      "epoch": 0.35032,
      "grad_norm": 2.0,
      "grad_norm_var": 0.004154205322265625,
      "learning_rate": 0.0001,
      "loss": 4.2603,
      "loss/crossentropy": 1.8439211249351501,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19603434205055237,
      "step": 17516
    },
    {
      "epoch": 0.35036,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.006843058268229166,
      "learning_rate": 0.0001,
      "loss": 3.9506,
      "loss/crossentropy": 2.2717851400375366,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21598497033119202,
      "step": 17518
    },
    {
      "epoch": 0.3504,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.006843058268229166,
      "learning_rate": 0.0001,
      "loss": 4.1807,
      "loss/crossentropy": 2.2645692825317383,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2229163572192192,
      "step": 17520
    },
    {
      "epoch": 0.35044,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.0071604410807291664,
      "learning_rate": 0.0001,
      "loss": 4.09,
      "loss/crossentropy": 2.0507450103759766,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19268949329853058,
      "step": 17522
    },
    {
      "epoch": 0.35048,
      "grad_norm": 2.0,
      "grad_norm_var": 0.005521647135416667,
      "learning_rate": 0.0001,
      "loss": 4.1468,
      "loss/crossentropy": 2.3831188678741455,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21931231766939163,
      "step": 17524
    },
    {
      "epoch": 0.35052,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.006281534830729167,
      "learning_rate": 0.0001,
      "loss": 4.116,
      "loss/crossentropy": 1.8574120998382568,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17972451448440552,
      "step": 17526
    },
    {
      "epoch": 0.35056,
      "grad_norm": 1.875,
      "grad_norm_var": 0.006013997395833333,
      "learning_rate": 0.0001,
      "loss": 3.9063,
      "loss/crossentropy": 2.141213893890381,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21798025816679,
      "step": 17528
    },
    {
      "epoch": 0.3506,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.006520334879557292,
      "learning_rate": 0.0001,
      "loss": 3.9454,
      "loss/crossentropy": 2.2224762439727783,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2031020149588585,
      "step": 17530
    },
    {
      "epoch": 0.35064,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.006476847330729166,
      "learning_rate": 0.0001,
      "loss": 3.765,
      "loss/crossentropy": 2.163232743740082,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22002054750919342,
      "step": 17532
    },
    {
      "epoch": 0.35068,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0055653889973958336,
      "learning_rate": 0.0001,
      "loss": 4.1397,
      "loss/crossentropy": 2.1463611721992493,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20084905624389648,
      "step": 17534
    },
    {
      "epoch": 0.35072,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.006566365559895833,
      "learning_rate": 0.0001,
      "loss": 3.9751,
      "loss/crossentropy": 2.1863406896591187,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2150622010231018,
      "step": 17536
    },
    {
      "epoch": 0.35076,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.005744425455729166,
      "learning_rate": 0.0001,
      "loss": 4.0958,
      "loss/crossentropy": 2.2162610292434692,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20196689665317535,
      "step": 17538
    },
    {
      "epoch": 0.3508,
      "grad_norm": 10.6875,
      "grad_norm_var": 4.776464589436849,
      "learning_rate": 0.0001,
      "loss": 4.1004,
      "loss/crossentropy": 1.8908233642578125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23238417506217957,
      "step": 17540
    },
    {
      "epoch": 0.35084,
      "grad_norm": 2.125,
      "grad_norm_var": 4.763142903645833,
      "learning_rate": 0.0001,
      "loss": 4.416,
      "loss/crossentropy": 2.3578076362609863,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21626722812652588,
      "step": 17542
    },
    {
      "epoch": 0.35088,
      "grad_norm": 2.234375,
      "grad_norm_var": 4.736722819010416,
      "learning_rate": 0.0001,
      "loss": 4.3243,
      "loss/crossentropy": 2.4079915285110474,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2292388305068016,
      "step": 17544
    },
    {
      "epoch": 0.35092,
      "grad_norm": 2.015625,
      "grad_norm_var": 4.717746734619141,
      "learning_rate": 0.0001,
      "loss": 4.2647,
      "loss/crossentropy": 2.1021666526794434,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20725515484809875,
      "step": 17546
    },
    {
      "epoch": 0.35096,
      "grad_norm": 1.96875,
      "grad_norm_var": 4.707061513264974,
      "learning_rate": 0.0001,
      "loss": 4.0905,
      "loss/crossentropy": 2.039289176464081,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20614883303642273,
      "step": 17548
    },
    {
      "epoch": 0.351,
      "grad_norm": 2.203125,
      "grad_norm_var": 4.68468017578125,
      "learning_rate": 0.0001,
      "loss": 4.269,
      "loss/crossentropy": 1.9731826782226562,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2164870798587799,
      "step": 17550
    },
    {
      "epoch": 0.35104,
      "grad_norm": 1.875,
      "grad_norm_var": 4.69991455078125,
      "learning_rate": 0.0001,
      "loss": 4.0785,
      "loss/crossentropy": 2.0166266560554504,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1977868676185608,
      "step": 17552
    },
    {
      "epoch": 0.35108,
      "grad_norm": 2.0,
      "grad_norm_var": 4.692333730061849,
      "learning_rate": 0.0001,
      "loss": 4.2823,
      "loss/crossentropy": 2.284690737724304,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2204381301999092,
      "step": 17554
    },
    {
      "epoch": 0.35112,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.012977854410807291,
      "learning_rate": 0.0001,
      "loss": 4.2629,
      "loss/crossentropy": 2.293992757797241,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21718238294124603,
      "step": 17556
    },
    {
      "epoch": 0.35116,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.013133748372395834,
      "learning_rate": 0.0001,
      "loss": 4.0687,
      "loss/crossentropy": 2.252563714981079,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23132775723934174,
      "step": 17558
    },
    {
      "epoch": 0.3512,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0097564697265625,
      "learning_rate": 0.0001,
      "loss": 3.977,
      "loss/crossentropy": 1.8454533219337463,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20084110647439957,
      "step": 17560
    },
    {
      "epoch": 0.35124,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009616851806640625,
      "learning_rate": 0.0001,
      "loss": 4.1526,
      "loss/crossentropy": 2.3160746097564697,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21168464422225952,
      "step": 17562
    },
    {
      "epoch": 0.35128,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.00953369140625,
      "learning_rate": 0.0001,
      "loss": 4.031,
      "loss/crossentropy": 2.0470253229141235,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20728407055139542,
      "step": 17564
    },
    {
      "epoch": 0.35132,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0064389546712239586,
      "learning_rate": 0.0001,
      "loss": 4.404,
      "loss/crossentropy": 2.098679304122925,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2133883535861969,
      "step": 17566
    },
    {
      "epoch": 0.35136,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.005615234375,
      "learning_rate": 0.0001,
      "loss": 4.0509,
      "loss/crossentropy": 1.975411057472229,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19393378496170044,
      "step": 17568
    },
    {
      "epoch": 0.3514,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.005606842041015625,
      "learning_rate": 0.0001,
      "loss": 4.2861,
      "loss/crossentropy": 2.594779372215271,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2499995082616806,
      "step": 17570
    },
    {
      "epoch": 0.35144,
      "grad_norm": 1.875,
      "grad_norm_var": 0.005915323893229167,
      "learning_rate": 0.0001,
      "loss": 4.1045,
      "loss/crossentropy": 2.075734496116638,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.207140251994133,
      "step": 17572
    },
    {
      "epoch": 0.35148,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.0062978108723958336,
      "learning_rate": 0.0001,
      "loss": 3.869,
      "loss/crossentropy": 2.124872624874115,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20058108121156693,
      "step": 17574
    },
    {
      "epoch": 0.35152,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.012056477864583333,
      "learning_rate": 0.0001,
      "loss": 4.1552,
      "loss/crossentropy": 2.0275574922561646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2105395644903183,
      "step": 17576
    },
    {
      "epoch": 0.35156,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.011913045247395834,
      "learning_rate": 0.0001,
      "loss": 4.248,
      "loss/crossentropy": 2.0694713592529297,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21342340111732483,
      "step": 17578
    },
    {
      "epoch": 0.3516,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.012239583333333333,
      "learning_rate": 0.0001,
      "loss": 3.9729,
      "loss/crossentropy": 1.8013625741004944,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1766640990972519,
      "step": 17580
    },
    {
      "epoch": 0.35164,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0123443603515625,
      "learning_rate": 0.0001,
      "loss": 4.0313,
      "loss/crossentropy": 2.1990463733673096,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19608185440301895,
      "step": 17582
    },
    {
      "epoch": 0.35168,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.013361612955729166,
      "learning_rate": 0.0001,
      "loss": 3.9041,
      "loss/crossentropy": 2.0714540481567383,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20064443349838257,
      "step": 17584
    },
    {
      "epoch": 0.35172,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.015242258707682291,
      "learning_rate": 0.0001,
      "loss": 3.7431,
      "loss/crossentropy": 1.7437097430229187,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17001167684793472,
      "step": 17586
    },
    {
      "epoch": 0.35176,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.020531972249348957,
      "learning_rate": 0.0001,
      "loss": 4.1987,
      "loss/crossentropy": 2.072261691093445,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21019883453845978,
      "step": 17588
    },
    {
      "epoch": 0.3518,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.021922810872395834,
      "learning_rate": 0.0001,
      "loss": 4.0826,
      "loss/crossentropy": 2.1804715394973755,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21589763462543488,
      "step": 17590
    },
    {
      "epoch": 0.35184,
      "grad_norm": 2.0,
      "grad_norm_var": 0.01734619140625,
      "learning_rate": 0.0001,
      "loss": 4.0964,
      "loss/crossentropy": 2.0044930577278137,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20204917341470718,
      "step": 17592
    },
    {
      "epoch": 0.35188,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.017362467447916665,
      "learning_rate": 0.0001,
      "loss": 4.1394,
      "loss/crossentropy": 1.8961025476455688,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1865018978714943,
      "step": 17594
    },
    {
      "epoch": 0.35192,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.016971588134765625,
      "learning_rate": 0.0001,
      "loss": 3.8503,
      "loss/crossentropy": 1.7005563378334045,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18951866030693054,
      "step": 17596
    },
    {
      "epoch": 0.35196,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.017002105712890625,
      "learning_rate": 0.0001,
      "loss": 4.0044,
      "loss/crossentropy": 2.002982437610626,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2076285257935524,
      "step": 17598
    },
    {
      "epoch": 0.352,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.016950480143229165,
      "learning_rate": 0.0001,
      "loss": 4.0595,
      "loss/crossentropy": 2.0283620357513428,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20175684988498688,
      "step": 17600
    },
    {
      "epoch": 0.35204,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.014289347330729167,
      "learning_rate": 0.0001,
      "loss": 3.9495,
      "loss/crossentropy": 1.855184018611908,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19368328154087067,
      "step": 17602
    },
    {
      "epoch": 0.35208,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.009089152018229166,
      "learning_rate": 0.0001,
      "loss": 4.2364,
      "loss/crossentropy": 2.2288308143615723,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2121831700205803,
      "step": 17604
    },
    {
      "epoch": 0.35212,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.002512359619140625,
      "learning_rate": 0.0001,
      "loss": 4.1398,
      "loss/crossentropy": 1.9114368557929993,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17937570065259933,
      "step": 17606
    },
    {
      "epoch": 0.35216,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0026751200358072916,
      "learning_rate": 0.0001,
      "loss": 3.9921,
      "loss/crossentropy": 2.238133430480957,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23264692723751068,
      "step": 17608
    },
    {
      "epoch": 0.3522,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0024169921875,
      "learning_rate": 0.0001,
      "loss": 3.85,
      "loss/crossentropy": 1.7219743728637695,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17108283936977386,
      "step": 17610
    },
    {
      "epoch": 0.35224,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.005037434895833333,
      "learning_rate": 0.0001,
      "loss": 3.8406,
      "loss/crossentropy": 1.9613978862762451,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2042389214038849,
      "step": 17612
    },
    {
      "epoch": 0.35228,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.005132802327473958,
      "learning_rate": 0.0001,
      "loss": 3.8567,
      "loss/crossentropy": 1.946107029914856,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19604258984327316,
      "step": 17614
    },
    {
      "epoch": 0.35232,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.005077870686848959,
      "learning_rate": 0.0001,
      "loss": 4.1404,
      "loss/crossentropy": 2.255744218826294,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2184707298874855,
      "step": 17616
    },
    {
      "epoch": 0.35236,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0058062235514322914,
      "learning_rate": 0.0001,
      "loss": 4.1852,
      "loss/crossentropy": 2.339258551597595,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21353042870759964,
      "step": 17618
    },
    {
      "epoch": 0.3524,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0054595947265625,
      "learning_rate": 0.0001,
      "loss": 3.9494,
      "loss/crossentropy": 1.8912869691848755,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19134657084941864,
      "step": 17620
    },
    {
      "epoch": 0.35244,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0058258056640625,
      "learning_rate": 0.0001,
      "loss": 4.2607,
      "loss/crossentropy": 2.186914384365082,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20765449851751328,
      "step": 17622
    },
    {
      "epoch": 0.35248,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.005773671468098958,
      "learning_rate": 0.0001,
      "loss": 3.8567,
      "loss/crossentropy": 1.6339558959007263,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18255474418401718,
      "step": 17624
    },
    {
      "epoch": 0.35252,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.005796051025390625,
      "learning_rate": 0.0001,
      "loss": 3.9593,
      "loss/crossentropy": 2.030660629272461,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1943136677145958,
      "step": 17626
    },
    {
      "epoch": 0.35256,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.004173787434895834,
      "learning_rate": 0.0001,
      "loss": 3.9738,
      "loss/crossentropy": 1.8040945529937744,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18390020728111267,
      "step": 17628
    },
    {
      "epoch": 0.3526,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.0055539449055989586,
      "learning_rate": 0.0001,
      "loss": 4.4613,
      "loss/crossentropy": 1.9993014335632324,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20794235169887543,
      "step": 17630
    },
    {
      "epoch": 0.35264,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.004881795247395833,
      "learning_rate": 0.0001,
      "loss": 3.8539,
      "loss/crossentropy": 2.0751482248306274,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20385687798261642,
      "step": 17632
    },
    {
      "epoch": 0.35268,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0052073160807291664,
      "learning_rate": 0.0001,
      "loss": 4.0059,
      "loss/crossentropy": 1.8923512697219849,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19879557192325592,
      "step": 17634
    },
    {
      "epoch": 0.35272,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.004935709635416666,
      "learning_rate": 0.0001,
      "loss": 4.2176,
      "loss/crossentropy": 2.068901300430298,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21545641124248505,
      "step": 17636
    },
    {
      "epoch": 0.35276,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.005968983968098958,
      "learning_rate": 0.0001,
      "loss": 3.8627,
      "loss/crossentropy": 1.8329379558563232,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1875568851828575,
      "step": 17638
    },
    {
      "epoch": 0.3528,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.007811482747395833,
      "learning_rate": 0.0001,
      "loss": 4.0878,
      "loss/crossentropy": 2.2368232011795044,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22392207384109497,
      "step": 17640
    },
    {
      "epoch": 0.35284,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.014972941080729166,
      "learning_rate": 0.0001,
      "loss": 4.0397,
      "loss/crossentropy": 2.2503843307495117,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21132144331932068,
      "step": 17642
    },
    {
      "epoch": 0.35288,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.01582819620768229,
      "learning_rate": 0.0001,
      "loss": 4.0459,
      "loss/crossentropy": 2.1919764280319214,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.197752483189106,
      "step": 17644
    },
    {
      "epoch": 0.35292,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0148345947265625,
      "learning_rate": 0.0001,
      "loss": 3.9122,
      "loss/crossentropy": 1.9867302775382996,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2115078568458557,
      "step": 17646
    },
    {
      "epoch": 0.35296,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.017899576822916666,
      "learning_rate": 0.0001,
      "loss": 3.8369,
      "loss/crossentropy": 1.9015426635742188,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17188573628664017,
      "step": 17648
    },
    {
      "epoch": 0.353,
      "grad_norm": 1.875,
      "grad_norm_var": 0.019006093343098957,
      "learning_rate": 0.0001,
      "loss": 3.6023,
      "loss/crossentropy": 1.8092533946037292,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1764475554227829,
      "step": 17650
    },
    {
      "epoch": 0.35304,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.01936620076497396,
      "learning_rate": 0.0001,
      "loss": 4.0542,
      "loss/crossentropy": 2.0736488103866577,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20046666264533997,
      "step": 17652
    },
    {
      "epoch": 0.35308,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.019364166259765624,
      "learning_rate": 0.0001,
      "loss": 3.9716,
      "loss/crossentropy": 2.1787428855895996,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23303276300430298,
      "step": 17654
    },
    {
      "epoch": 0.35312,
      "grad_norm": 2.0,
      "grad_norm_var": 0.017020416259765626,
      "learning_rate": 0.0001,
      "loss": 4.4889,
      "loss/crossentropy": 2.419381618499756,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23756036162376404,
      "step": 17656
    },
    {
      "epoch": 0.35316,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.012111155192057292,
      "learning_rate": 0.0001,
      "loss": 4.4202,
      "loss/crossentropy": 2.4017220735549927,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2373708263039589,
      "step": 17658
    },
    {
      "epoch": 0.3532,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.011919911702473958,
      "learning_rate": 0.0001,
      "loss": 3.8805,
      "loss/crossentropy": 1.7796767354011536,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19661501049995422,
      "step": 17660
    },
    {
      "epoch": 0.35324,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.013158162434895834,
      "learning_rate": 0.0001,
      "loss": 4.2215,
      "loss/crossentropy": 2.1815608739852905,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20645780861377716,
      "step": 17662
    },
    {
      "epoch": 0.35328,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.010178375244140624,
      "learning_rate": 0.0001,
      "loss": 4.2148,
      "loss/crossentropy": 2.056411921977997,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21581003069877625,
      "step": 17664
    },
    {
      "epoch": 0.35332,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.007824452718098958,
      "learning_rate": 0.0001,
      "loss": 3.8676,
      "loss/crossentropy": 1.8901747465133667,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19498124718666077,
      "step": 17666
    },
    {
      "epoch": 0.35336,
      "grad_norm": 2.0,
      "grad_norm_var": 0.006493123372395834,
      "learning_rate": 0.0001,
      "loss": 4.0884,
      "loss/crossentropy": 2.0241716504096985,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19405542314052582,
      "step": 17668
    },
    {
      "epoch": 0.3534,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.008241526285807292,
      "learning_rate": 0.0001,
      "loss": 4.1452,
      "loss/crossentropy": 2.016151189804077,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1856973022222519,
      "step": 17670
    },
    {
      "epoch": 0.35344,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.009096018473307292,
      "learning_rate": 0.0001,
      "loss": 4.3895,
      "loss/crossentropy": 2.5066399574279785,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23949292302131653,
      "step": 17672
    },
    {
      "epoch": 0.35348,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.007236480712890625,
      "learning_rate": 0.0001,
      "loss": 4.1432,
      "loss/crossentropy": 2.1384177207946777,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21078043431043625,
      "step": 17674
    },
    {
      "epoch": 0.35352,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0070709228515625,
      "learning_rate": 0.0001,
      "loss": 4.1732,
      "loss/crossentropy": 2.1211976408958435,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2064737230539322,
      "step": 17676
    },
    {
      "epoch": 0.35356,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0072418212890625,
      "learning_rate": 0.0001,
      "loss": 4.3103,
      "loss/crossentropy": 2.018574059009552,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20446718484163284,
      "step": 17678
    },
    {
      "epoch": 0.3536,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008186848958333333,
      "learning_rate": 0.0001,
      "loss": 3.9862,
      "loss/crossentropy": 2.10149747133255,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1974436193704605,
      "step": 17680
    },
    {
      "epoch": 0.35364,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009698232014973959,
      "learning_rate": 0.0001,
      "loss": 4.3133,
      "loss/crossentropy": 2.288322687149048,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22535154223442078,
      "step": 17682
    },
    {
      "epoch": 0.35368,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.010400136311848959,
      "learning_rate": 0.0001,
      "loss": 4.3396,
      "loss/crossentropy": 2.002479314804077,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20964795351028442,
      "step": 17684
    },
    {
      "epoch": 0.35372,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0099029541015625,
      "learning_rate": 0.0001,
      "loss": 4.0116,
      "loss/crossentropy": 2.2548930644989014,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19946102052927017,
      "step": 17686
    },
    {
      "epoch": 0.35376,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.015006510416666667,
      "learning_rate": 0.0001,
      "loss": 4.1777,
      "loss/crossentropy": 2.3636194467544556,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21626830101013184,
      "step": 17688
    },
    {
      "epoch": 0.3538,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.016471099853515626,
      "learning_rate": 0.0001,
      "loss": 4.1137,
      "loss/crossentropy": 2.2401771545410156,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.202116459608078,
      "step": 17690
    },
    {
      "epoch": 0.35384,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.01602783203125,
      "learning_rate": 0.0001,
      "loss": 4.2248,
      "loss/crossentropy": 2.192560911178589,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20161370187997818,
      "step": 17692
    },
    {
      "epoch": 0.35388,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.01495361328125,
      "learning_rate": 0.0001,
      "loss": 4.175,
      "loss/crossentropy": 2.31100332736969,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23697054386138916,
      "step": 17694
    },
    {
      "epoch": 0.35392,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.014139811197916666,
      "learning_rate": 0.0001,
      "loss": 4.0894,
      "loss/crossentropy": 2.018012821674347,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1954345926642418,
      "step": 17696
    },
    {
      "epoch": 0.35396,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.0168853759765625,
      "learning_rate": 0.0001,
      "loss": 4.3128,
      "loss/crossentropy": 2.0922133326530457,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2219238504767418,
      "step": 17698
    },
    {
      "epoch": 0.354,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.01591796875,
      "learning_rate": 0.0001,
      "loss": 4.3803,
      "loss/crossentropy": 2.727385640144348,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21042727679014206,
      "step": 17700
    },
    {
      "epoch": 0.35404,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.014631144205729167,
      "learning_rate": 0.0001,
      "loss": 3.9535,
      "loss/crossentropy": 1.8388070464134216,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18624315410852432,
      "step": 17702
    },
    {
      "epoch": 0.35408,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008719889322916667,
      "learning_rate": 0.0001,
      "loss": 3.9327,
      "loss/crossentropy": 1.879045307636261,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1827290803194046,
      "step": 17704
    },
    {
      "epoch": 0.35412,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0075266520182291664,
      "learning_rate": 0.0001,
      "loss": 4.2715,
      "loss/crossentropy": 2.2604658603668213,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22230257838964462,
      "step": 17706
    },
    {
      "epoch": 0.35416,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.01329345703125,
      "learning_rate": 0.0001,
      "loss": 3.7828,
      "loss/crossentropy": 2.1200287342071533,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20524511486291885,
      "step": 17708
    },
    {
      "epoch": 0.3542,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.014135487874348958,
      "learning_rate": 0.0001,
      "loss": 3.9624,
      "loss/crossentropy": 1.933307945728302,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19991742819547653,
      "step": 17710
    },
    {
      "epoch": 0.35424,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.013231404622395833,
      "learning_rate": 0.0001,
      "loss": 4.0673,
      "loss/crossentropy": 1.9442221522331238,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1861218512058258,
      "step": 17712
    },
    {
      "epoch": 0.35428,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.008885701497395834,
      "learning_rate": 0.0001,
      "loss": 3.6592,
      "loss/crossentropy": 1.7169482111930847,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16892607510089874,
      "step": 17714
    },
    {
      "epoch": 0.35432,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.006891886393229167,
      "learning_rate": 0.0001,
      "loss": 4.3472,
      "loss/crossentropy": 2.105340003967285,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19026386737823486,
      "step": 17716
    },
    {
      "epoch": 0.35436,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0069976806640625,
      "learning_rate": 0.0001,
      "loss": 4.1666,
      "loss/crossentropy": 2.0366504192352295,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19709386676549911,
      "step": 17718
    },
    {
      "epoch": 0.3544,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.007393391927083334,
      "learning_rate": 0.0001,
      "loss": 3.9829,
      "loss/crossentropy": 1.9578897356987,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19842654466629028,
      "step": 17720
    },
    {
      "epoch": 0.35444,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.006888834635416666,
      "learning_rate": 0.0001,
      "loss": 3.8608,
      "loss/crossentropy": 2.139336943626404,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19530436396598816,
      "step": 17722
    },
    {
      "epoch": 0.35448,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.009669748942057292,
      "learning_rate": 0.0001,
      "loss": 4.3803,
      "loss/crossentropy": 2.4077011346817017,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2209417074918747,
      "step": 17724
    },
    {
      "epoch": 0.35452,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.009373982747395834,
      "learning_rate": 0.0001,
      "loss": 4.2412,
      "loss/crossentropy": 2.1217586994171143,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2052367851138115,
      "step": 17726
    },
    {
      "epoch": 0.35456,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.009751129150390624,
      "learning_rate": 0.0001,
      "loss": 4.1872,
      "loss/crossentropy": 2.078941583633423,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2033359408378601,
      "step": 17728
    },
    {
      "epoch": 0.3546,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0072265625,
      "learning_rate": 0.0001,
      "loss": 4.3514,
      "loss/crossentropy": 2.1763141751289368,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22765995562076569,
      "step": 17730
    },
    {
      "epoch": 0.35464,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.00784912109375,
      "learning_rate": 0.0001,
      "loss": 4.1094,
      "loss/crossentropy": 1.83991938829422,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18603162467479706,
      "step": 17732
    },
    {
      "epoch": 0.35468,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.009132639567057291,
      "learning_rate": 0.0001,
      "loss": 3.93,
      "loss/crossentropy": 2.021396040916443,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19647164642810822,
      "step": 17734
    },
    {
      "epoch": 0.35472,
      "grad_norm": 2.0,
      "grad_norm_var": 0.009297434488932292,
      "learning_rate": 0.0001,
      "loss": 4.2137,
      "loss/crossentropy": 2.2339202165603638,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20721730589866638,
      "step": 17736
    },
    {
      "epoch": 0.35476,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.008017730712890626,
      "learning_rate": 0.0001,
      "loss": 4.1859,
      "loss/crossentropy": 2.2280211448669434,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21610353142023087,
      "step": 17738
    },
    {
      "epoch": 0.3548,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.0082672119140625,
      "learning_rate": 0.0001,
      "loss": 3.8836,
      "loss/crossentropy": 1.8112387657165527,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18565281480550766,
      "step": 17740
    },
    {
      "epoch": 0.35484,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.009098307291666666,
      "learning_rate": 0.0001,
      "loss": 4.5429,
      "loss/crossentropy": 2.1125651597976685,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1949760466814041,
      "step": 17742
    },
    {
      "epoch": 0.35488,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.009163411458333333,
      "learning_rate": 0.0001,
      "loss": 3.9613,
      "loss/crossentropy": 2.1020091772079468,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2126646414399147,
      "step": 17744
    },
    {
      "epoch": 0.35492,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.010553995768229166,
      "learning_rate": 0.0001,
      "loss": 4.0522,
      "loss/crossentropy": 1.8875654339790344,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1913822665810585,
      "step": 17746
    },
    {
      "epoch": 0.35496,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.011921946207682292,
      "learning_rate": 0.0001,
      "loss": 4.2408,
      "loss/crossentropy": 2.1043163537979126,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18798817694187164,
      "step": 17748
    },
    {
      "epoch": 0.355,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.012373606363932291,
      "learning_rate": 0.0001,
      "loss": 4.0744,
      "loss/crossentropy": 2.0751022696495056,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19110247492790222,
      "step": 17750
    },
    {
      "epoch": 0.35504,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.011815388997395834,
      "learning_rate": 0.0001,
      "loss": 4.2077,
      "loss/crossentropy": 2.0335286259651184,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18653328716754913,
      "step": 17752
    },
    {
      "epoch": 0.35508,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.012230428059895833,
      "learning_rate": 0.0001,
      "loss": 4.2833,
      "loss/crossentropy": 2.091266691684723,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19435560703277588,
      "step": 17754
    },
    {
      "epoch": 0.35512,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.015702056884765624,
      "learning_rate": 0.0001,
      "loss": 4.4468,
      "loss/crossentropy": 2.2889565229415894,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21760451793670654,
      "step": 17756
    },
    {
      "epoch": 0.35516,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0162017822265625,
      "learning_rate": 0.0001,
      "loss": 4.1216,
      "loss/crossentropy": 1.818089485168457,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18759652972221375,
      "step": 17758
    },
    {
      "epoch": 0.3552,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.015531158447265625,
      "learning_rate": 0.0001,
      "loss": 4.0322,
      "loss/crossentropy": 1.8579466938972473,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20318903028964996,
      "step": 17760
    },
    {
      "epoch": 0.35524,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0152008056640625,
      "learning_rate": 0.0001,
      "loss": 4.058,
      "loss/crossentropy": 2.0463815927505493,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20810134708881378,
      "step": 17762
    },
    {
      "epoch": 0.35528,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.0146392822265625,
      "learning_rate": 0.0001,
      "loss": 3.9702,
      "loss/crossentropy": 1.8263658285140991,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1781037524342537,
      "step": 17764
    },
    {
      "epoch": 0.35532,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0123199462890625,
      "learning_rate": 0.0001,
      "loss": 4.0973,
      "loss/crossentropy": 2.0662325620651245,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2161395400762558,
      "step": 17766
    },
    {
      "epoch": 0.35536,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.013598378499348958,
      "learning_rate": 0.0001,
      "loss": 4.1028,
      "loss/crossentropy": 1.9901453256607056,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20523327589035034,
      "step": 17768
    },
    {
      "epoch": 0.3554,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.011766560872395833,
      "learning_rate": 0.0001,
      "loss": 4.1567,
      "loss/crossentropy": 1.9877265095710754,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20279201865196228,
      "step": 17770
    },
    {
      "epoch": 0.35544,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.013695271809895833,
      "learning_rate": 0.0001,
      "loss": 4.5072,
      "loss/crossentropy": 2.1376627683639526,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20117300748825073,
      "step": 17772
    },
    {
      "epoch": 0.35548,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.015327962239583333,
      "learning_rate": 0.0001,
      "loss": 3.8024,
      "loss/crossentropy": 2.0174089074134827,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2042396143078804,
      "step": 17774
    },
    {
      "epoch": 0.35552,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.019090779622395835,
      "learning_rate": 0.0001,
      "loss": 3.7872,
      "loss/crossentropy": 2.1650888919830322,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2062518149614334,
      "step": 17776
    },
    {
      "epoch": 0.35556,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.017836252848307293,
      "learning_rate": 0.0001,
      "loss": 4.275,
      "loss/crossentropy": 2.078373670578003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.205182746052742,
      "step": 17778
    },
    {
      "epoch": 0.3556,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.017693837483723957,
      "learning_rate": 0.0001,
      "loss": 3.7109,
      "loss/crossentropy": 1.7496679425239563,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17602672427892685,
      "step": 17780
    },
    {
      "epoch": 0.35564,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.01779963175455729,
      "learning_rate": 0.0001,
      "loss": 3.8767,
      "loss/crossentropy": 2.3132810592651367,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21808438748121262,
      "step": 17782
    },
    {
      "epoch": 0.35568,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.017210896809895834,
      "learning_rate": 0.0001,
      "loss": 3.8465,
      "loss/crossentropy": 1.9900661706924438,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20264087617397308,
      "step": 17784
    },
    {
      "epoch": 0.35572,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.01665013631184896,
      "learning_rate": 0.0001,
      "loss": 4.2735,
      "loss/crossentropy": 2.095883369445801,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19925827533006668,
      "step": 17786
    },
    {
      "epoch": 0.35576,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.008771769205729167,
      "learning_rate": 0.0001,
      "loss": 3.9558,
      "loss/crossentropy": 1.9932443499565125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1864292174577713,
      "step": 17788
    },
    {
      "epoch": 0.3558,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.008455149332682292,
      "learning_rate": 0.0001,
      "loss": 4.0811,
      "loss/crossentropy": 1.8468709588050842,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18770359456539154,
      "step": 17790
    },
    {
      "epoch": 0.35584,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.006502278645833333,
      "learning_rate": 0.0001,
      "loss": 4.1749,
      "loss/crossentropy": 2.293928623199463,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20280499011278152,
      "step": 17792
    },
    {
      "epoch": 0.35588,
      "grad_norm": 2.0,
      "grad_norm_var": 0.00618896484375,
      "learning_rate": 0.0001,
      "loss": 4.0155,
      "loss/crossentropy": 1.8333890438079834,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20070593804121017,
      "step": 17794
    },
    {
      "epoch": 0.35592,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0061767578125,
      "learning_rate": 0.0001,
      "loss": 3.9903,
      "loss/crossentropy": 1.7173805236816406,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17676317691802979,
      "step": 17796
    },
    {
      "epoch": 0.35596,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0058977762858072914,
      "learning_rate": 0.0001,
      "loss": 3.8196,
      "loss/crossentropy": 1.8198468685150146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20926716923713684,
      "step": 17798
    },
    {
      "epoch": 0.356,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0054840087890625,
      "learning_rate": 0.0001,
      "loss": 4.2473,
      "loss/crossentropy": 2.354674279689789,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23675478994846344,
      "step": 17800
    },
    {
      "epoch": 0.35604,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.005610911051432291,
      "learning_rate": 0.0001,
      "loss": 4.1092,
      "loss/crossentropy": 2.055357277393341,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18840713798999786,
      "step": 17802
    },
    {
      "epoch": 0.35608,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.007089996337890625,
      "learning_rate": 0.0001,
      "loss": 4.1351,
      "loss/crossentropy": 2.3461071252822876,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22149299830198288,
      "step": 17804
    },
    {
      "epoch": 0.35612,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0083892822265625,
      "learning_rate": 0.0001,
      "loss": 4.165,
      "loss/crossentropy": 2.2675124406814575,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21928898990154266,
      "step": 17806
    },
    {
      "epoch": 0.35616,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.007346343994140625,
      "learning_rate": 0.0001,
      "loss": 4.2397,
      "loss/crossentropy": 2.239398717880249,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20473385602235794,
      "step": 17808
    },
    {
      "epoch": 0.3562,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.009262847900390624,
      "learning_rate": 0.0001,
      "loss": 3.8152,
      "loss/crossentropy": 2.022661864757538,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1910046860575676,
      "step": 17810
    },
    {
      "epoch": 0.35624,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.009236653645833334,
      "learning_rate": 0.0001,
      "loss": 4.053,
      "loss/crossentropy": 1.874055802822113,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19950387626886368,
      "step": 17812
    },
    {
      "epoch": 0.35628,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.009178670247395833,
      "learning_rate": 0.0001,
      "loss": 4.058,
      "loss/crossentropy": 1.7601851224899292,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18487969785928726,
      "step": 17814
    },
    {
      "epoch": 0.35632,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.008348592122395833,
      "learning_rate": 0.0001,
      "loss": 4.0428,
      "loss/crossentropy": 1.8841391801834106,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17906954884529114,
      "step": 17816
    },
    {
      "epoch": 0.35636,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.011092122395833333,
      "learning_rate": 0.0001,
      "loss": 3.6292,
      "loss/crossentropy": 1.7313326597213745,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16394374519586563,
      "step": 17818
    },
    {
      "epoch": 0.3564,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0076901753743489586,
      "learning_rate": 0.0001,
      "loss": 3.7718,
      "loss/crossentropy": 2.094825506210327,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20041480660438538,
      "step": 17820
    },
    {
      "epoch": 0.35644,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.004854329427083333,
      "learning_rate": 0.0001,
      "loss": 4.075,
      "loss/crossentropy": 2.222295045852661,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19912637770175934,
      "step": 17822
    },
    {
      "epoch": 0.35648,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.005980428059895833,
      "learning_rate": 0.0001,
      "loss": 3.971,
      "loss/crossentropy": 2.1671608090400696,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20070582628250122,
      "step": 17824
    },
    {
      "epoch": 0.35652,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.006884511311848958,
      "learning_rate": 0.0001,
      "loss": 4.0805,
      "loss/crossentropy": 2.164097785949707,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2535782679915428,
      "step": 17826
    },
    {
      "epoch": 0.35656,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.008405558268229167,
      "learning_rate": 0.0001,
      "loss": 4.128,
      "loss/crossentropy": 1.738038182258606,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17469938844442368,
      "step": 17828
    },
    {
      "epoch": 0.3566,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.010636393229166667,
      "learning_rate": 0.0001,
      "loss": 4.3157,
      "loss/crossentropy": 2.3383474349975586,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2087705433368683,
      "step": 17830
    },
    {
      "epoch": 0.35664,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.012870279947916667,
      "learning_rate": 0.0001,
      "loss": 3.854,
      "loss/crossentropy": 2.0826202034950256,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19418098032474518,
      "step": 17832
    },
    {
      "epoch": 0.35668,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.011287434895833334,
      "learning_rate": 0.0001,
      "loss": 4.1733,
      "loss/crossentropy": 1.943938970565796,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20667240023612976,
      "step": 17834
    },
    {
      "epoch": 0.35672,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.011226145426432292,
      "learning_rate": 0.0001,
      "loss": 4.1312,
      "loss/crossentropy": 1.9756113290786743,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1973763257265091,
      "step": 17836
    },
    {
      "epoch": 0.35676,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.012898763020833334,
      "learning_rate": 0.0001,
      "loss": 3.6895,
      "loss/crossentropy": 1.8583598732948303,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17464379966259003,
      "step": 17838
    },
    {
      "epoch": 0.3568,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.011791737874348958,
      "learning_rate": 0.0001,
      "loss": 3.9421,
      "loss/crossentropy": 1.9786911606788635,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20051777362823486,
      "step": 17840
    },
    {
      "epoch": 0.35684,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.010414377848307291,
      "learning_rate": 0.0001,
      "loss": 4.0804,
      "loss/crossentropy": 2.1271785497665405,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21754157543182373,
      "step": 17842
    },
    {
      "epoch": 0.35688,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.009639485677083334,
      "learning_rate": 0.0001,
      "loss": 3.9739,
      "loss/crossentropy": 2.264941096305847,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20562931895256042,
      "step": 17844
    },
    {
      "epoch": 0.35692,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0083160400390625,
      "learning_rate": 0.0001,
      "loss": 4.3865,
      "loss/crossentropy": 2.141028881072998,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23258531093597412,
      "step": 17846
    },
    {
      "epoch": 0.35696,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.009781901041666667,
      "learning_rate": 0.0001,
      "loss": 4.1484,
      "loss/crossentropy": 1.9527946710586548,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24948758631944656,
      "step": 17848
    },
    {
      "epoch": 0.357,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008854166666666666,
      "learning_rate": 0.0001,
      "loss": 4.0745,
      "loss/crossentropy": 1.9595746397972107,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19834068417549133,
      "step": 17850
    },
    {
      "epoch": 0.35704,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.010396321614583334,
      "learning_rate": 0.0001,
      "loss": 4.1407,
      "loss/crossentropy": 1.8414466977119446,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19133952260017395,
      "step": 17852
    },
    {
      "epoch": 0.35708,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.010326894124348958,
      "learning_rate": 0.0001,
      "loss": 3.6912,
      "loss/crossentropy": 1.6589071154594421,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17579880356788635,
      "step": 17854
    },
    {
      "epoch": 0.35712,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.011213175455729167,
      "learning_rate": 0.0001,
      "loss": 3.8164,
      "loss/crossentropy": 1.8462252020835876,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19051598757505417,
      "step": 17856
    },
    {
      "epoch": 0.35716,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.015083567301432291,
      "learning_rate": 0.0001,
      "loss": 3.9453,
      "loss/crossentropy": 2.0473897457122803,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21250004321336746,
      "step": 17858
    },
    {
      "epoch": 0.3572,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.014664459228515624,
      "learning_rate": 0.0001,
      "loss": 4.1836,
      "loss/crossentropy": 2.220176875591278,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22177650034427643,
      "step": 17860
    },
    {
      "epoch": 0.35724,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.13489176432291666,
      "learning_rate": 0.0001,
      "loss": 3.5816,
      "loss/crossentropy": 1.6774207949638367,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16900332272052765,
      "step": 17862
    },
    {
      "epoch": 0.35728,
      "grad_norm": 2.0,
      "grad_norm_var": 0.13516616821289062,
      "learning_rate": 0.0001,
      "loss": 3.8075,
      "loss/crossentropy": 1.9064326286315918,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1925061270594597,
      "step": 17864
    },
    {
      "epoch": 0.35732,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.1366607666015625,
      "learning_rate": 0.0001,
      "loss": 4.1698,
      "loss/crossentropy": 2.188231110572815,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20198386162519455,
      "step": 17866
    },
    {
      "epoch": 0.35736,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.13647359212239582,
      "learning_rate": 0.0001,
      "loss": 4.0141,
      "loss/crossentropy": 1.9380639791488647,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19545384496450424,
      "step": 17868
    },
    {
      "epoch": 0.3574,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.13407389322916666,
      "learning_rate": 0.0001,
      "loss": 4.1965,
      "loss/crossentropy": 1.995704710483551,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19666064530611038,
      "step": 17870
    },
    {
      "epoch": 0.35744,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.13017349243164061,
      "learning_rate": 0.0001,
      "loss": 4.0191,
      "loss/crossentropy": 2.2148354053497314,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23265502601861954,
      "step": 17872
    },
    {
      "epoch": 0.35748,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.1243072509765625,
      "learning_rate": 0.0001,
      "loss": 4.0021,
      "loss/crossentropy": 1.7091269493103027,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21242853999137878,
      "step": 17874
    },
    {
      "epoch": 0.35752,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.12219950358072916,
      "learning_rate": 0.0001,
      "loss": 4.3081,
      "loss/crossentropy": 2.2395507097244263,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21505478769540787,
      "step": 17876
    },
    {
      "epoch": 0.35756,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.008512369791666667,
      "learning_rate": 0.0001,
      "loss": 3.8636,
      "loss/crossentropy": 2.014355480670929,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20956294238567352,
      "step": 17878
    },
    {
      "epoch": 0.3576,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.012383778889973959,
      "learning_rate": 0.0001,
      "loss": 4.3642,
      "loss/crossentropy": 2.3606066703796387,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22014011442661285,
      "step": 17880
    },
    {
      "epoch": 0.35764,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.011368560791015624,
      "learning_rate": 0.0001,
      "loss": 3.9368,
      "loss/crossentropy": 1.6152977347373962,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1630660966038704,
      "step": 17882
    },
    {
      "epoch": 0.35768,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.011138661702473959,
      "learning_rate": 0.0001,
      "loss": 4.156,
      "loss/crossentropy": 2.1977567076683044,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20369882881641388,
      "step": 17884
    },
    {
      "epoch": 0.35772,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.012359364827473959,
      "learning_rate": 0.0001,
      "loss": 3.9927,
      "loss/crossentropy": 1.9582098126411438,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20284517109394073,
      "step": 17886
    },
    {
      "epoch": 0.35776,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.013818105061848959,
      "learning_rate": 0.0001,
      "loss": 4.108,
      "loss/crossentropy": 2.1300426721572876,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19376185536384583,
      "step": 17888
    },
    {
      "epoch": 0.3578,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.01883519490559896,
      "learning_rate": 0.0001,
      "loss": 4.3648,
      "loss/crossentropy": 2.057590961456299,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21069013327360153,
      "step": 17890
    },
    {
      "epoch": 0.35784,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.01968994140625,
      "learning_rate": 0.0001,
      "loss": 4.0006,
      "loss/crossentropy": 2.21635901927948,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19591110199689865,
      "step": 17892
    },
    {
      "epoch": 0.35788,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.019809722900390625,
      "learning_rate": 0.0001,
      "loss": 4.4868,
      "loss/crossentropy": 2.6530216932296753,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24127788096666336,
      "step": 17894
    },
    {
      "epoch": 0.35792,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.013256581624348958,
      "learning_rate": 0.0001,
      "loss": 4.0727,
      "loss/crossentropy": 1.9931264519691467,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20592481642961502,
      "step": 17896
    },
    {
      "epoch": 0.35796,
      "grad_norm": 2.0,
      "grad_norm_var": 0.013305409749348959,
      "learning_rate": 0.0001,
      "loss": 4.1574,
      "loss/crossentropy": 2.178193688392639,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2204412743449211,
      "step": 17898
    },
    {
      "epoch": 0.358,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.012878163655598959,
      "learning_rate": 0.0001,
      "loss": 4.1748,
      "loss/crossentropy": 2.192083954811096,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22313552349805832,
      "step": 17900
    },
    {
      "epoch": 0.35804,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.011281077067057292,
      "learning_rate": 0.0001,
      "loss": 4.1695,
      "loss/crossentropy": 2.228816568851471,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2120378389954567,
      "step": 17902
    },
    {
      "epoch": 0.35808,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.010871378580729167,
      "learning_rate": 0.0001,
      "loss": 4.1369,
      "loss/crossentropy": 2.0921813249588013,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19552303105592728,
      "step": 17904
    },
    {
      "epoch": 0.35812,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.004687245686848958,
      "learning_rate": 0.0001,
      "loss": 4.2206,
      "loss/crossentropy": 2.0568217635154724,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19668184220790863,
      "step": 17906
    },
    {
      "epoch": 0.35816,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0038937886555989584,
      "learning_rate": 0.0001,
      "loss": 3.9946,
      "loss/crossentropy": 1.8241485357284546,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15895915031433105,
      "step": 17908
    },
    {
      "epoch": 0.3582,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.003082021077473958,
      "learning_rate": 0.0001,
      "loss": 3.9904,
      "loss/crossentropy": 2.2571341395378113,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2065516859292984,
      "step": 17910
    },
    {
      "epoch": 0.35824,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0032297770182291665,
      "learning_rate": 0.0001,
      "loss": 3.9289,
      "loss/crossentropy": 2.0472288727760315,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21586833149194717,
      "step": 17912
    },
    {
      "epoch": 0.35828,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0035947163899739585,
      "learning_rate": 0.0001,
      "loss": 4.0652,
      "loss/crossentropy": 2.0345569252967834,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19726714491844177,
      "step": 17914
    },
    {
      "epoch": 0.35832,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0037127176920572916,
      "learning_rate": 0.0001,
      "loss": 4.0294,
      "loss/crossentropy": 2.15024471282959,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20401738584041595,
      "step": 17916
    },
    {
      "epoch": 0.35836,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.004874420166015625,
      "learning_rate": 0.0001,
      "loss": 4.3122,
      "loss/crossentropy": 2.110986351966858,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2299470156431198,
      "step": 17918
    },
    {
      "epoch": 0.3584,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.003897857666015625,
      "learning_rate": 0.0001,
      "loss": 4.2147,
      "loss/crossentropy": 1.7297720909118652,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18035240471363068,
      "step": 17920
    },
    {
      "epoch": 0.35844,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.01236572265625,
      "learning_rate": 0.0001,
      "loss": 4.3997,
      "loss/crossentropy": 1.9483368396759033,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2060721442103386,
      "step": 17922
    },
    {
      "epoch": 0.35848,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0123443603515625,
      "learning_rate": 0.0001,
      "loss": 4.1527,
      "loss/crossentropy": 1.9840999841690063,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20700833201408386,
      "step": 17924
    },
    {
      "epoch": 0.35852,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.011525217692057292,
      "learning_rate": 0.0001,
      "loss": 4.1754,
      "loss/crossentropy": 2.4118131399154663,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20695140957832336,
      "step": 17926
    },
    {
      "epoch": 0.35856,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.011195627848307292,
      "learning_rate": 0.0001,
      "loss": 3.9363,
      "loss/crossentropy": 1.9709432721138,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18202026933431625,
      "step": 17928
    },
    {
      "epoch": 0.3586,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.010206858317057291,
      "learning_rate": 0.0001,
      "loss": 4.2972,
      "loss/crossentropy": 1.9450251460075378,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21454186737537384,
      "step": 17930
    },
    {
      "epoch": 0.35864,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0094390869140625,
      "learning_rate": 0.0001,
      "loss": 4.3317,
      "loss/crossentropy": 2.0738271474838257,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21376954019069672,
      "step": 17932
    },
    {
      "epoch": 0.35868,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.010109202067057291,
      "learning_rate": 0.0001,
      "loss": 4.1613,
      "loss/crossentropy": 2.1857110261917114,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2238282486796379,
      "step": 17934
    },
    {
      "epoch": 0.35872,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.010643513997395833,
      "learning_rate": 0.0001,
      "loss": 4.2208,
      "loss/crossentropy": 2.2790093421936035,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1987098902463913,
      "step": 17936
    },
    {
      "epoch": 0.35876,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.0054280598958333336,
      "learning_rate": 0.0001,
      "loss": 3.9261,
      "loss/crossentropy": 2.1101399064064026,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2034989446401596,
      "step": 17938
    },
    {
      "epoch": 0.3588,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.005793253580729167,
      "learning_rate": 0.0001,
      "loss": 3.7024,
      "loss/crossentropy": 1.6923771500587463,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18680892139673233,
      "step": 17940
    },
    {
      "epoch": 0.35884,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.005342356363932292,
      "learning_rate": 0.0001,
      "loss": 3.9282,
      "loss/crossentropy": 1.6772454977035522,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17377988994121552,
      "step": 17942
    },
    {
      "epoch": 0.35888,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.007722981770833333,
      "learning_rate": 0.0001,
      "loss": 4.141,
      "loss/crossentropy": 2.2448233366012573,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21804184466600418,
      "step": 17944
    },
    {
      "epoch": 0.35892,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.006688435872395833,
      "learning_rate": 0.0001,
      "loss": 3.9984,
      "loss/crossentropy": 1.934161365032196,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1834414005279541,
      "step": 17946
    },
    {
      "epoch": 0.35896,
      "grad_norm": 2.0,
      "grad_norm_var": 0.007344563802083333,
      "learning_rate": 0.0001,
      "loss": 3.9943,
      "loss/crossentropy": 2.06991970539093,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21084094047546387,
      "step": 17948
    },
    {
      "epoch": 0.359,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.007916005452473958,
      "learning_rate": 0.0001,
      "loss": 4.0493,
      "loss/crossentropy": 2.0942054986953735,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21652275323867798,
      "step": 17950
    },
    {
      "epoch": 0.35904,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.009134928385416666,
      "learning_rate": 0.0001,
      "loss": 4.1125,
      "loss/crossentropy": 2.0005252361297607,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.198953777551651,
      "step": 17952
    },
    {
      "epoch": 0.35908,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.0327301025390625,
      "learning_rate": 0.0001,
      "loss": 4.1639,
      "loss/crossentropy": 2.1563133597373962,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2072513848543167,
      "step": 17954
    },
    {
      "epoch": 0.35912,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0312896728515625,
      "learning_rate": 0.0001,
      "loss": 4.1789,
      "loss/crossentropy": 2.3533977270126343,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21483703702688217,
      "step": 17956
    },
    {
      "epoch": 0.35916,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.029581705729166668,
      "learning_rate": 0.0001,
      "loss": 4.0921,
      "loss/crossentropy": 2.1915602684020996,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2241670861840248,
      "step": 17958
    },
    {
      "epoch": 0.3592,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0314117431640625,
      "learning_rate": 0.0001,
      "loss": 3.9225,
      "loss/crossentropy": 1.9529941082000732,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1920214593410492,
      "step": 17960
    },
    {
      "epoch": 0.35924,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.030350748697916666,
      "learning_rate": 0.0001,
      "loss": 4.4467,
      "loss/crossentropy": 2.308731436729431,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2238362655043602,
      "step": 17962
    },
    {
      "epoch": 0.35928,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.04334208170572917,
      "learning_rate": 0.0001,
      "loss": 4.1682,
      "loss/crossentropy": 2.0917986631393433,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2061680108308792,
      "step": 17964
    },
    {
      "epoch": 0.35932,
      "grad_norm": 2.265625,
      "grad_norm_var": 10.265104166666667,
      "learning_rate": 0.0001,
      "loss": 4.9501,
      "loss/crossentropy": 2.16925585269928,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20239810645580292,
      "step": 17966
    },
    {
      "epoch": 0.35936,
      "grad_norm": 2.0,
      "grad_norm_var": 10.219252268473307,
      "learning_rate": 0.0001,
      "loss": 4.287,
      "loss/crossentropy": 2.1585946083068848,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21381714940071106,
      "step": 17968
    },
    {
      "epoch": 0.3594,
      "grad_norm": 1.8359375,
      "grad_norm_var": 10.281192016601562,
      "learning_rate": 0.0001,
      "loss": 4.0133,
      "loss/crossentropy": 1.8752552270889282,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19667387753725052,
      "step": 17970
    },
    {
      "epoch": 0.35944,
      "grad_norm": 1.9140625,
      "grad_norm_var": 10.291275024414062,
      "learning_rate": 0.0001,
      "loss": 3.9797,
      "loss/crossentropy": 1.8533543944358826,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1844579428434372,
      "step": 17972
    },
    {
      "epoch": 0.35948,
      "grad_norm": 2.234375,
      "grad_norm_var": 10.282754516601562,
      "learning_rate": 0.0001,
      "loss": 4.5316,
      "loss/crossentropy": 2.5859906673431396,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23142188042402267,
      "step": 17974
    },
    {
      "epoch": 0.35952,
      "grad_norm": 2.15625,
      "grad_norm_var": 10.236201985677083,
      "learning_rate": 0.0001,
      "loss": 4.2607,
      "loss/crossentropy": 2.2086023092269897,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21764708310365677,
      "step": 17976
    },
    {
      "epoch": 0.35956,
      "grad_norm": 1.953125,
      "grad_norm_var": 10.258858235677083,
      "learning_rate": 0.0001,
      "loss": 4.2578,
      "loss/crossentropy": 2.3860682249069214,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22804389894008636,
      "step": 17978
    },
    {
      "epoch": 0.3596,
      "grad_norm": 2.0,
      "grad_norm_var": 10.314815266927083,
      "learning_rate": 0.0001,
      "loss": 4.2055,
      "loss/crossentropy": 2.1412216424942017,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.200544573366642,
      "step": 17980
    },
    {
      "epoch": 0.35964,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.022395833333333334,
      "learning_rate": 0.0001,
      "loss": 4.0412,
      "loss/crossentropy": 2.198704957962036,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2043847218155861,
      "step": 17982
    },
    {
      "epoch": 0.35968,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.010249837239583334,
      "learning_rate": 0.0001,
      "loss": 4.037,
      "loss/crossentropy": 2.0053776502609253,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2072744369506836,
      "step": 17984
    },
    {
      "epoch": 0.35972,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.04221979777018229,
      "learning_rate": 0.0001,
      "loss": 4.127,
      "loss/crossentropy": 2.5015710592269897,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23083895444869995,
      "step": 17986
    },
    {
      "epoch": 0.35976,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.04145889282226563,
      "learning_rate": 0.0001,
      "loss": 4.0531,
      "loss/crossentropy": 2.037220776081085,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19487662613391876,
      "step": 17988
    },
    {
      "epoch": 0.3598,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.03911107381184896,
      "learning_rate": 0.0001,
      "loss": 3.9801,
      "loss/crossentropy": 1.9316660165786743,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20183449983596802,
      "step": 17990
    },
    {
      "epoch": 0.35984,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.04090067545572917,
      "learning_rate": 0.0001,
      "loss": 3.804,
      "loss/crossentropy": 1.892772138118744,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17082024365663528,
      "step": 17992
    },
    {
      "epoch": 0.35988,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.04088109334309896,
      "learning_rate": 0.0001,
      "loss": 4.0732,
      "loss/crossentropy": 1.9325169324874878,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1900368332862854,
      "step": 17994
    },
    {
      "epoch": 0.35992,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.050388336181640625,
      "learning_rate": 0.0001,
      "loss": 4.1392,
      "loss/crossentropy": 1.9746126532554626,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21207977831363678,
      "step": 17996
    },
    {
      "epoch": 0.35996,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.048620351155598956,
      "learning_rate": 0.0001,
      "loss": 4.0308,
      "loss/crossentropy": 2.1249493956565857,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18674498051404953,
      "step": 17998
    },
    {
      "epoch": 0.36,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.04625422159830729,
      "learning_rate": 0.0001,
      "loss": 4.0854,
      "loss/crossentropy": 2.1279499530792236,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.214716374874115,
      "step": 18000
    },
    {
      "epoch": 0.36004,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0177398681640625,
      "learning_rate": 0.0001,
      "loss": 3.9238,
      "loss/crossentropy": 2.301763415336609,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20897136628627777,
      "step": 18002
    },
    {
      "epoch": 0.36008,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.01788330078125,
      "learning_rate": 0.0001,
      "loss": 4.0944,
      "loss/crossentropy": 1.8079062104225159,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19506293535232544,
      "step": 18004
    },
    {
      "epoch": 0.36012,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0205078125,
      "learning_rate": 0.0001,
      "loss": 4.0503,
      "loss/crossentropy": 1.9429230093955994,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19762147217988968,
      "step": 18006
    },
    {
      "epoch": 0.36016,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.019608561197916666,
      "learning_rate": 0.0001,
      "loss": 3.9801,
      "loss/crossentropy": 2.307368576526642,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21261631697416306,
      "step": 18008
    },
    {
      "epoch": 0.3602,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.021952311197916668,
      "learning_rate": 0.0001,
      "loss": 3.8723,
      "loss/crossentropy": 2.2834020853042603,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20086131989955902,
      "step": 18010
    },
    {
      "epoch": 0.36024,
      "grad_norm": 1.875,
      "grad_norm_var": 0.0117340087890625,
      "learning_rate": 0.0001,
      "loss": 4.1708,
      "loss/crossentropy": 2.1730109453201294,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20470792055130005,
      "step": 18012
    },
    {
      "epoch": 0.36028,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.012809244791666667,
      "learning_rate": 0.0001,
      "loss": 4.4047,
      "loss/crossentropy": 2.491591691970825,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2361597716808319,
      "step": 18014
    },
    {
      "epoch": 0.36032,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.013240305582682292,
      "learning_rate": 0.0001,
      "loss": 4.167,
      "loss/crossentropy": 2.024519979953766,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2042919620871544,
      "step": 18016
    },
    {
      "epoch": 0.36036,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.016123199462890626,
      "learning_rate": 0.0001,
      "loss": 3.9521,
      "loss/crossentropy": 2.055600941181183,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17253537476062775,
      "step": 18018
    },
    {
      "epoch": 0.3604,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0188629150390625,
      "learning_rate": 0.0001,
      "loss": 3.9665,
      "loss/crossentropy": 1.8796368837356567,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21168731898069382,
      "step": 18020
    },
    {
      "epoch": 0.36044,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.014825185139973959,
      "learning_rate": 0.0001,
      "loss": 4.0291,
      "loss/crossentropy": 1.9885223507881165,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20676057040691376,
      "step": 18022
    },
    {
      "epoch": 0.36048,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.010839589436848958,
      "learning_rate": 0.0001,
      "loss": 3.8401,
      "loss/crossentropy": 2.0412577986717224,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18211868405342102,
      "step": 18024
    },
    {
      "epoch": 0.36052,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.010445149739583333,
      "learning_rate": 0.0001,
      "loss": 3.8696,
      "loss/crossentropy": 2.1391053199768066,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20121745765209198,
      "step": 18026
    },
    {
      "epoch": 0.36056,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.010087076822916667,
      "learning_rate": 0.0001,
      "loss": 3.9683,
      "loss/crossentropy": 1.8888981938362122,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19372491538524628,
      "step": 18028
    },
    {
      "epoch": 0.3606,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.007087198893229166,
      "learning_rate": 0.0001,
      "loss": 4.2467,
      "loss/crossentropy": 2.002028524875641,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18978165835142136,
      "step": 18030
    },
    {
      "epoch": 0.36064,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.007828776041666667,
      "learning_rate": 0.0001,
      "loss": 4.0703,
      "loss/crossentropy": 2.3448036909103394,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21162079274654388,
      "step": 18032
    },
    {
      "epoch": 0.36068,
      "grad_norm": 2.5625,
      "grad_norm_var": 0.03154474894205729,
      "learning_rate": 0.0001,
      "loss": 4.283,
      "loss/crossentropy": 2.086554765701294,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25456031411886215,
      "step": 18034
    },
    {
      "epoch": 0.36072,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.02948582967122396,
      "learning_rate": 0.0001,
      "loss": 3.8122,
      "loss/crossentropy": 1.9166680574417114,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18382181227207184,
      "step": 18036
    },
    {
      "epoch": 0.36076,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.029642740885416668,
      "learning_rate": 0.0001,
      "loss": 3.9937,
      "loss/crossentropy": 2.050579786300659,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20126450806856155,
      "step": 18038
    },
    {
      "epoch": 0.3608,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.030049641927083332,
      "learning_rate": 0.0001,
      "loss": 3.9273,
      "loss/crossentropy": 2.2339882850646973,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20789660513401031,
      "step": 18040
    },
    {
      "epoch": 0.36084,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.029243977864583333,
      "learning_rate": 0.0001,
      "loss": 4.0086,
      "loss/crossentropy": 2.248544931411743,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20925041288137436,
      "step": 18042
    },
    {
      "epoch": 0.36088,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.028955078125,
      "learning_rate": 0.0001,
      "loss": 4.0448,
      "loss/crossentropy": 1.8894963264465332,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18232882767915726,
      "step": 18044
    },
    {
      "epoch": 0.36092,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0289703369140625,
      "learning_rate": 0.0001,
      "loss": 4.1838,
      "loss/crossentropy": 2.0220844745635986,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18812591582536697,
      "step": 18046
    },
    {
      "epoch": 0.36096,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.027913411458333332,
      "learning_rate": 0.0001,
      "loss": 4.2088,
      "loss/crossentropy": 2.1896166801452637,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.205679252743721,
      "step": 18048
    },
    {
      "epoch": 0.361,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.004686482747395833,
      "learning_rate": 0.0001,
      "loss": 4.0511,
      "loss/crossentropy": 2.0670089721679688,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18881987780332565,
      "step": 18050
    },
    {
      "epoch": 0.36104,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.005500284830729166,
      "learning_rate": 0.0001,
      "loss": 4.2426,
      "loss/crossentropy": 2.155470609664917,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2147108018398285,
      "step": 18052
    },
    {
      "epoch": 0.36108,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.0076812744140625,
      "learning_rate": 0.0001,
      "loss": 4.2007,
      "loss/crossentropy": 1.8182223439216614,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18691811710596085,
      "step": 18054
    },
    {
      "epoch": 0.36112,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0066314697265625,
      "learning_rate": 0.0001,
      "loss": 3.8601,
      "loss/crossentropy": 1.7366089820861816,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18361609429121017,
      "step": 18056
    },
    {
      "epoch": 0.36116,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.005704498291015625,
      "learning_rate": 0.0001,
      "loss": 4.027,
      "loss/crossentropy": 1.9429153203964233,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1974082887172699,
      "step": 18058
    },
    {
      "epoch": 0.3612,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.00562744140625,
      "learning_rate": 0.0001,
      "loss": 4.2751,
      "loss/crossentropy": 2.230627417564392,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22216492146253586,
      "step": 18060
    },
    {
      "epoch": 0.36124,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.006371053059895834,
      "learning_rate": 0.0001,
      "loss": 4.1759,
      "loss/crossentropy": 2.0506786704063416,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20492341369390488,
      "step": 18062
    },
    {
      "epoch": 0.36128,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.007116444905598958,
      "learning_rate": 0.0001,
      "loss": 3.778,
      "loss/crossentropy": 2.025477647781372,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20034795254468918,
      "step": 18064
    },
    {
      "epoch": 0.36132,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006617991129557291,
      "learning_rate": 0.0001,
      "loss": 4.2998,
      "loss/crossentropy": 1.9828922748565674,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19445167481899261,
      "step": 18066
    },
    {
      "epoch": 0.36136,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006485748291015625,
      "learning_rate": 0.0001,
      "loss": 4.2566,
      "loss/crossentropy": 2.2869513630867004,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21809721738100052,
      "step": 18068
    },
    {
      "epoch": 0.3614,
      "grad_norm": 1.875,
      "grad_norm_var": 0.004129791259765625,
      "learning_rate": 0.0001,
      "loss": 3.9636,
      "loss/crossentropy": 2.1584482192993164,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21123671531677246,
      "step": 18070
    },
    {
      "epoch": 0.36144,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0046078999837239586,
      "learning_rate": 0.0001,
      "loss": 3.9628,
      "loss/crossentropy": 2.112669587135315,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20795845240354538,
      "step": 18072
    },
    {
      "epoch": 0.36148,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0059234619140625,
      "learning_rate": 0.0001,
      "loss": 4.2516,
      "loss/crossentropy": 2.3283581733703613,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22724319994449615,
      "step": 18074
    },
    {
      "epoch": 0.36152,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.006078084309895833,
      "learning_rate": 0.0001,
      "loss": 3.8966,
      "loss/crossentropy": 2.286331057548523,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20937074720859528,
      "step": 18076
    },
    {
      "epoch": 0.36156,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.009110260009765624,
      "learning_rate": 0.0001,
      "loss": 4.1124,
      "loss/crossentropy": 2.1055954694747925,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1924988478422165,
      "step": 18078
    },
    {
      "epoch": 0.3616,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.008186594645182291,
      "learning_rate": 0.0001,
      "loss": 4.0349,
      "loss/crossentropy": 2.180675983428955,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21886169910430908,
      "step": 18080
    },
    {
      "epoch": 0.36164,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.008621978759765624,
      "learning_rate": 0.0001,
      "loss": 3.9958,
      "loss/crossentropy": 1.7595775127410889,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17430071532726288,
      "step": 18082
    },
    {
      "epoch": 0.36168,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.011472320556640625,
      "learning_rate": 0.0001,
      "loss": 4.0306,
      "loss/crossentropy": 2.1191208958625793,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19574576616287231,
      "step": 18084
    },
    {
      "epoch": 0.36172,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0110595703125,
      "learning_rate": 0.0001,
      "loss": 4.1563,
      "loss/crossentropy": 2.0964609384536743,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19742169976234436,
      "step": 18086
    },
    {
      "epoch": 0.36176,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0108154296875,
      "learning_rate": 0.0001,
      "loss": 4.3584,
      "loss/crossentropy": 2.3081077337265015,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2177809551358223,
      "step": 18088
    },
    {
      "epoch": 0.3618,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.009928385416666666,
      "learning_rate": 0.0001,
      "loss": 4.0493,
      "loss/crossentropy": 2.340154528617859,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23811470717191696,
      "step": 18090
    },
    {
      "epoch": 0.36184,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.009777577718098958,
      "learning_rate": 0.0001,
      "loss": 3.828,
      "loss/crossentropy": 1.9872968196868896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19391798973083496,
      "step": 18092
    },
    {
      "epoch": 0.36188,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0067779541015625,
      "learning_rate": 0.0001,
      "loss": 3.9404,
      "loss/crossentropy": 1.9162002205848694,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.203052818775177,
      "step": 18094
    },
    {
      "epoch": 0.36192,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.009718577067057291,
      "learning_rate": 0.0001,
      "loss": 3.7458,
      "loss/crossentropy": 2.240887403488159,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19722212105989456,
      "step": 18096
    },
    {
      "epoch": 0.36196,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.009496053059895834,
      "learning_rate": 0.0001,
      "loss": 4.1473,
      "loss/crossentropy": 1.8601738214492798,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1938677802681923,
      "step": 18098
    },
    {
      "epoch": 0.362,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.005729166666666666,
      "learning_rate": 0.0001,
      "loss": 4.0131,
      "loss/crossentropy": 1.9588143229484558,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19559185206890106,
      "step": 18100
    },
    {
      "epoch": 0.36204,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.0060618082682291664,
      "learning_rate": 0.0001,
      "loss": 3.9199,
      "loss/crossentropy": 1.909518837928772,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1974424198269844,
      "step": 18102
    },
    {
      "epoch": 0.36208,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.005641428629557291,
      "learning_rate": 0.0001,
      "loss": 4.15,
      "loss/crossentropy": 2.0762908458709717,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1981927454471588,
      "step": 18104
    },
    {
      "epoch": 0.36212,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.006339263916015625,
      "learning_rate": 0.0001,
      "loss": 4.3226,
      "loss/crossentropy": 2.4555106163024902,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22001618146896362,
      "step": 18106
    },
    {
      "epoch": 0.36216,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.013533528645833333,
      "learning_rate": 0.0001,
      "loss": 4.0595,
      "loss/crossentropy": 1.9983150959014893,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19945750385522842,
      "step": 18108
    },
    {
      "epoch": 0.3622,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.013181304931640625,
      "learning_rate": 0.0001,
      "loss": 4.0863,
      "loss/crossentropy": 2.152313530445099,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21388111263513565,
      "step": 18110
    },
    {
      "epoch": 0.36224,
      "grad_norm": 2.125,
      "grad_norm_var": 0.016886393229166668,
      "learning_rate": 0.0001,
      "loss": 4.1128,
      "loss/crossentropy": 1.690669596195221,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1883523240685463,
      "step": 18112
    },
    {
      "epoch": 0.36228,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.017179107666015624,
      "learning_rate": 0.0001,
      "loss": 3.9978,
      "loss/crossentropy": 1.9404500126838684,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18945956230163574,
      "step": 18114
    },
    {
      "epoch": 0.36232,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.016007486979166666,
      "learning_rate": 0.0001,
      "loss": 4.4396,
      "loss/crossentropy": 2.213471293449402,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22550443559885025,
      "step": 18116
    },
    {
      "epoch": 0.36236,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.014233144124348958,
      "learning_rate": 0.0001,
      "loss": 4.3084,
      "loss/crossentropy": 2.1008135080337524,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2002386674284935,
      "step": 18118
    },
    {
      "epoch": 0.3624,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.013866933186848958,
      "learning_rate": 0.0001,
      "loss": 3.8664,
      "loss/crossentropy": 1.9704426527023315,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1806606948375702,
      "step": 18120
    },
    {
      "epoch": 0.36244,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.01473388671875,
      "learning_rate": 0.0001,
      "loss": 3.7456,
      "loss/crossentropy": 1.8658949732780457,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19861871004104614,
      "step": 18122
    },
    {
      "epoch": 0.36248,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.011034901936848958,
      "learning_rate": 0.0001,
      "loss": 4.0341,
      "loss/crossentropy": 1.9327979683876038,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19395866990089417,
      "step": 18124
    },
    {
      "epoch": 0.36252,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.009641265869140625,
      "learning_rate": 0.0001,
      "loss": 4.071,
      "loss/crossentropy": 2.377102255821228,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21928569674491882,
      "step": 18126
    },
    {
      "epoch": 0.36256,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.006994374593098958,
      "learning_rate": 0.0001,
      "loss": 3.9623,
      "loss/crossentropy": 2.1650161743164062,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2292800396680832,
      "step": 18128
    },
    {
      "epoch": 0.3626,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.007458241780598959,
      "learning_rate": 0.0001,
      "loss": 4.3135,
      "loss/crossentropy": 2.281827986240387,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2190621793270111,
      "step": 18130
    },
    {
      "epoch": 0.36264,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.00848388671875,
      "learning_rate": 0.0001,
      "loss": 3.8335,
      "loss/crossentropy": 2.0073219537734985,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18984179198741913,
      "step": 18132
    },
    {
      "epoch": 0.36268,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.01141357421875,
      "learning_rate": 0.0001,
      "loss": 3.7551,
      "loss/crossentropy": 1.7675580978393555,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18771683424711227,
      "step": 18134
    },
    {
      "epoch": 0.36272,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.011154937744140624,
      "learning_rate": 0.0001,
      "loss": 3.8816,
      "loss/crossentropy": 2.064103126525879,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20835159718990326,
      "step": 18136
    },
    {
      "epoch": 0.36276,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0126861572265625,
      "learning_rate": 0.0001,
      "loss": 4.2122,
      "loss/crossentropy": 2.1821314096450806,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2391536980867386,
      "step": 18138
    },
    {
      "epoch": 0.3628,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.010117340087890624,
      "learning_rate": 0.0001,
      "loss": 4.2307,
      "loss/crossentropy": 2.295500636100769,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.215906023979187,
      "step": 18140
    },
    {
      "epoch": 0.36284,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.010245768229166667,
      "learning_rate": 0.0001,
      "loss": 4.1144,
      "loss/crossentropy": 2.0736570954322815,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18491360545158386,
      "step": 18142
    },
    {
      "epoch": 0.36288,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.009496053059895834,
      "learning_rate": 0.0001,
      "loss": 4.1326,
      "loss/crossentropy": 2.0513535737991333,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20204297453165054,
      "step": 18144
    },
    {
      "epoch": 0.36292,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.010601552327473958,
      "learning_rate": 0.0001,
      "loss": 4.1648,
      "loss/crossentropy": 2.1471269130706787,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.204594686627388,
      "step": 18146
    },
    {
      "epoch": 0.36296,
      "grad_norm": 2.125,
      "grad_norm_var": 0.011372884114583334,
      "learning_rate": 0.0001,
      "loss": 4.2728,
      "loss/crossentropy": 2.093555986881256,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20173487067222595,
      "step": 18148
    },
    {
      "epoch": 0.363,
      "grad_norm": 1.6953125,
      "grad_norm_var": 0.013627115885416667,
      "learning_rate": 0.0001,
      "loss": 3.7183,
      "loss/crossentropy": 1.6938685178756714,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1703190803527832,
      "step": 18150
    },
    {
      "epoch": 0.36304,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.013337961832682292,
      "learning_rate": 0.0001,
      "loss": 4.116,
      "loss/crossentropy": 2.180016875267029,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23546195775270462,
      "step": 18152
    },
    {
      "epoch": 0.36308,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011277008056640624,
      "learning_rate": 0.0001,
      "loss": 4.1826,
      "loss/crossentropy": 2.145975947380066,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20554804801940918,
      "step": 18154
    },
    {
      "epoch": 0.36312,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.011131795247395833,
      "learning_rate": 0.0001,
      "loss": 4.0113,
      "loss/crossentropy": 2.068985939025879,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20410801470279694,
      "step": 18156
    },
    {
      "epoch": 0.36316,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.012572224934895833,
      "learning_rate": 0.0001,
      "loss": 3.8273,
      "loss/crossentropy": 1.8428975343704224,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19165785610675812,
      "step": 18158
    },
    {
      "epoch": 0.3632,
      "grad_norm": 1.875,
      "grad_norm_var": 0.0131744384765625,
      "learning_rate": 0.0001,
      "loss": 3.7808,
      "loss/crossentropy": 1.859747588634491,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18524104356765747,
      "step": 18160
    },
    {
      "epoch": 0.36324,
      "grad_norm": 2.0,
      "grad_norm_var": 0.011521148681640624,
      "learning_rate": 0.0001,
      "loss": 4.1458,
      "loss/crossentropy": 2.1383343935012817,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21649780869483948,
      "step": 18162
    },
    {
      "epoch": 0.36328,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.010758209228515624,
      "learning_rate": 0.0001,
      "loss": 3.9889,
      "loss/crossentropy": 2.0153123140335083,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2131347879767418,
      "step": 18164
    },
    {
      "epoch": 0.36332,
      "grad_norm": 1.875,
      "grad_norm_var": 0.007004547119140625,
      "learning_rate": 0.0001,
      "loss": 4.0827,
      "loss/crossentropy": 1.8382813930511475,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18116765469312668,
      "step": 18166
    },
    {
      "epoch": 0.36336,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.007287343343098958,
      "learning_rate": 0.0001,
      "loss": 3.8883,
      "loss/crossentropy": 1.9639039039611816,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2014933079481125,
      "step": 18168
    },
    {
      "epoch": 0.3634,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.012446848551432292,
      "learning_rate": 0.0001,
      "loss": 4.2886,
      "loss/crossentropy": 2.057366132736206,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2134721800684929,
      "step": 18170
    },
    {
      "epoch": 0.36344,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.012690989176432292,
      "learning_rate": 0.0001,
      "loss": 4.0801,
      "loss/crossentropy": 2.244979500770569,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.219241164624691,
      "step": 18172
    },
    {
      "epoch": 0.36348,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.010625966389973958,
      "learning_rate": 0.0001,
      "loss": 4.1271,
      "loss/crossentropy": 2.1159361600875854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19503474980592728,
      "step": 18174
    },
    {
      "epoch": 0.36352,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0099761962890625,
      "learning_rate": 0.0001,
      "loss": 4.2031,
      "loss/crossentropy": 2.240646004676819,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22175125032663345,
      "step": 18176
    },
    {
      "epoch": 0.36356,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.011327107747395834,
      "learning_rate": 0.0001,
      "loss": 3.9508,
      "loss/crossentropy": 1.7946885228157043,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17505639791488647,
      "step": 18178
    },
    {
      "epoch": 0.3636,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.010420735677083333,
      "learning_rate": 0.0001,
      "loss": 4.1138,
      "loss/crossentropy": 2.2058286666870117,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21548640727996826,
      "step": 18180
    },
    {
      "epoch": 0.36364,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0101226806640625,
      "learning_rate": 0.0001,
      "loss": 4.0831,
      "loss/crossentropy": 2.168944835662842,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22227579355239868,
      "step": 18182
    },
    {
      "epoch": 0.36368,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.008392079671223959,
      "learning_rate": 0.0001,
      "loss": 4.1482,
      "loss/crossentropy": 2.019882082939148,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20161078870296478,
      "step": 18184
    },
    {
      "epoch": 0.36372,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0034993489583333335,
      "learning_rate": 0.0001,
      "loss": 4.1305,
      "loss/crossentropy": 2.000952959060669,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19938969612121582,
      "step": 18186
    },
    {
      "epoch": 0.36376,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0024920145670572916,
      "learning_rate": 0.0001,
      "loss": 3.9158,
      "loss/crossentropy": 2.0612798929214478,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2022939696907997,
      "step": 18188
    },
    {
      "epoch": 0.3638,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.0042307535807291664,
      "learning_rate": 0.0001,
      "loss": 3.8822,
      "loss/crossentropy": 1.664880096912384,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16950316727161407,
      "step": 18190
    },
    {
      "epoch": 0.36384,
      "grad_norm": 1.875,
      "grad_norm_var": 0.003360748291015625,
      "learning_rate": 0.0001,
      "loss": 3.9881,
      "loss/crossentropy": 1.8047854900360107,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1823924407362938,
      "step": 18192
    },
    {
      "epoch": 0.36388,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.0060808817545572914,
      "learning_rate": 0.0001,
      "loss": 3.7857,
      "loss/crossentropy": 2.167177438735962,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1963154897093773,
      "step": 18194
    },
    {
      "epoch": 0.36392,
      "grad_norm": 1.875,
      "grad_norm_var": 0.006151326497395833,
      "learning_rate": 0.0001,
      "loss": 4.0372,
      "loss/crossentropy": 2.396019458770752,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20359576493501663,
      "step": 18196
    },
    {
      "epoch": 0.36396,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.005866495768229166,
      "learning_rate": 0.0001,
      "loss": 3.9349,
      "loss/crossentropy": 2.0866541862487793,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2082398235797882,
      "step": 18198
    },
    {
      "epoch": 0.364,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0060791015625,
      "learning_rate": 0.0001,
      "loss": 4.1046,
      "loss/crossentropy": 2.170537829399109,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19030719250440598,
      "step": 18200
    },
    {
      "epoch": 0.36404,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.024234771728515625,
      "learning_rate": 0.0001,
      "loss": 4.3921,
      "loss/crossentropy": 1.938852846622467,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19586428999900818,
      "step": 18202
    },
    {
      "epoch": 0.36408,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.023996734619140626,
      "learning_rate": 0.0001,
      "loss": 4.0629,
      "loss/crossentropy": 1.8909979462623596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17625273764133453,
      "step": 18204
    },
    {
      "epoch": 0.36412,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.022078450520833334,
      "learning_rate": 0.0001,
      "loss": 4.1729,
      "loss/crossentropy": 1.9127016067504883,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1888885200023651,
      "step": 18206
    },
    {
      "epoch": 0.36416,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.03103612263997396,
      "learning_rate": 0.0001,
      "loss": 4.1826,
      "loss/crossentropy": 2.182482957839966,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21610775589942932,
      "step": 18208
    },
    {
      "epoch": 0.3642,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.027497355143229166,
      "learning_rate": 0.0001,
      "loss": 4.1941,
      "loss/crossentropy": 2.0900736451148987,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20886409282684326,
      "step": 18210
    },
    {
      "epoch": 0.36424,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0248199462890625,
      "learning_rate": 0.0001,
      "loss": 3.9184,
      "loss/crossentropy": 2.2132604122161865,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21213480830192566,
      "step": 18212
    },
    {
      "epoch": 0.36428,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.024825032552083334,
      "learning_rate": 0.0001,
      "loss": 4.0002,
      "loss/crossentropy": 1.7810762524604797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.188262477517128,
      "step": 18214
    },
    {
      "epoch": 0.36432,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.02490208943684896,
      "learning_rate": 0.0001,
      "loss": 3.8339,
      "loss/crossentropy": 2.072141647338867,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20580865442752838,
      "step": 18216
    },
    {
      "epoch": 0.36436,
      "grad_norm": 2.125,
      "grad_norm_var": 0.016778310139973957,
      "learning_rate": 0.0001,
      "loss": 4.0777,
      "loss/crossentropy": 1.9672082662582397,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19521142542362213,
      "step": 18218
    },
    {
      "epoch": 0.3644,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.017574055989583334,
      "learning_rate": 0.0001,
      "loss": 4.0763,
      "loss/crossentropy": 2.0122682452201843,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21602813154459,
      "step": 18220
    },
    {
      "epoch": 0.36444,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0169342041015625,
      "learning_rate": 0.0001,
      "loss": 4.0574,
      "loss/crossentropy": 2.1709738969802856,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2267712950706482,
      "step": 18222
    },
    {
      "epoch": 0.36448,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009091949462890625,
      "learning_rate": 0.0001,
      "loss": 4.2092,
      "loss/crossentropy": 2.246786117553711,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2128008008003235,
      "step": 18224
    },
    {
      "epoch": 0.36452,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0063250223795572914,
      "learning_rate": 0.0001,
      "loss": 4.1051,
      "loss/crossentropy": 1.686498999595642,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18233107775449753,
      "step": 18226
    },
    {
      "epoch": 0.36456,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.006786855061848959,
      "learning_rate": 0.0001,
      "loss": 4.3823,
      "loss/crossentropy": 2.5849136114120483,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23062889277935028,
      "step": 18228
    },
    {
      "epoch": 0.3646,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.005804189046223958,
      "learning_rate": 0.0001,
      "loss": 4.04,
      "loss/crossentropy": 2.068212151527405,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20198806375265121,
      "step": 18230
    },
    {
      "epoch": 0.36464,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.00618896484375,
      "learning_rate": 0.0001,
      "loss": 3.8729,
      "loss/crossentropy": 1.727245271205902,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17305339127779007,
      "step": 18232
    },
    {
      "epoch": 0.36468,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.005574289957682292,
      "learning_rate": 0.0001,
      "loss": 4.0496,
      "loss/crossentropy": 2.0269583463668823,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2060471773147583,
      "step": 18234
    },
    {
      "epoch": 0.36472,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.004801177978515625,
      "learning_rate": 0.0001,
      "loss": 4.1707,
      "loss/crossentropy": 1.9194093346595764,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19781427085399628,
      "step": 18236
    },
    {
      "epoch": 0.36476,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0055328369140625,
      "learning_rate": 0.0001,
      "loss": 4.1438,
      "loss/crossentropy": 1.7539438605308533,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19985094666481018,
      "step": 18238
    },
    {
      "epoch": 0.3648,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.006068674723307291,
      "learning_rate": 0.0001,
      "loss": 4.1287,
      "loss/crossentropy": 2.1092851161956787,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23365512490272522,
      "step": 18240
    },
    {
      "epoch": 0.36484,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.00771484375,
      "learning_rate": 0.0001,
      "loss": 3.6435,
      "loss/crossentropy": 1.772037386894226,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18215186148881912,
      "step": 18242
    },
    {
      "epoch": 0.36488,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.007845052083333333,
      "learning_rate": 0.0001,
      "loss": 4.1017,
      "loss/crossentropy": 2.0108843445777893,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20684240013360977,
      "step": 18244
    },
    {
      "epoch": 0.36492,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.008348592122395833,
      "learning_rate": 0.0001,
      "loss": 4.0077,
      "loss/crossentropy": 2.0076091289520264,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1842496171593666,
      "step": 18246
    },
    {
      "epoch": 0.36496,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.008085123697916667,
      "learning_rate": 0.0001,
      "loss": 4.1357,
      "loss/crossentropy": 2.1824593544006348,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19478252530097961,
      "step": 18248
    },
    {
      "epoch": 0.365,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0068318684895833336,
      "learning_rate": 0.0001,
      "loss": 4.2231,
      "loss/crossentropy": 2.1033846139907837,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21361663192510605,
      "step": 18250
    },
    {
      "epoch": 0.36504,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.009299468994140626,
      "learning_rate": 0.0001,
      "loss": 3.7738,
      "loss/crossentropy": 2.142418146133423,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20618261396884918,
      "step": 18252
    },
    {
      "epoch": 0.36508,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.008131663004557291,
      "learning_rate": 0.0001,
      "loss": 4.1037,
      "loss/crossentropy": 2.061814546585083,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20519836992025375,
      "step": 18254
    },
    {
      "epoch": 0.36512,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008512115478515625,
      "learning_rate": 0.0001,
      "loss": 4.0512,
      "loss/crossentropy": 2.0216987133026123,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1956964135169983,
      "step": 18256
    },
    {
      "epoch": 0.36516,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.006681315104166667,
      "learning_rate": 0.0001,
      "loss": 4.301,
      "loss/crossentropy": 2.2619231939315796,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21638523787260056,
      "step": 18258
    },
    {
      "epoch": 0.3652,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.006281534830729167,
      "learning_rate": 0.0001,
      "loss": 4.165,
      "loss/crossentropy": 1.8058243989944458,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19046328961849213,
      "step": 18260
    },
    {
      "epoch": 0.36524,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.005639394124348958,
      "learning_rate": 0.0001,
      "loss": 4.1876,
      "loss/crossentropy": 2.0969032049179077,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20291081070899963,
      "step": 18262
    },
    {
      "epoch": 0.36528,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.006794993082682292,
      "learning_rate": 0.0001,
      "loss": 3.8754,
      "loss/crossentropy": 1.9567083716392517,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19624605774879456,
      "step": 18264
    },
    {
      "epoch": 0.36532,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.006493123372395834,
      "learning_rate": 0.0001,
      "loss": 3.9861,
      "loss/crossentropy": 1.9015939235687256,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18771862238645554,
      "step": 18266
    },
    {
      "epoch": 0.36536,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.017575836181640624,
      "learning_rate": 0.0001,
      "loss": 4.0288,
      "loss/crossentropy": 1.9432410597801208,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1785590723156929,
      "step": 18268
    },
    {
      "epoch": 0.3654,
      "grad_norm": 2.0,
      "grad_norm_var": 0.017438761393229165,
      "learning_rate": 0.0001,
      "loss": 4.3256,
      "loss/crossentropy": 2.216074585914612,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21463461220264435,
      "step": 18270
    },
    {
      "epoch": 0.36544,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.020401763916015624,
      "learning_rate": 0.0001,
      "loss": 3.7192,
      "loss/crossentropy": 1.5440006256103516,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1621650904417038,
      "step": 18272
    },
    {
      "epoch": 0.36548,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.02197850545247396,
      "learning_rate": 0.0001,
      "loss": 3.9072,
      "loss/crossentropy": 1.9478511214256287,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20275350660085678,
      "step": 18274
    },
    {
      "epoch": 0.36552,
      "grad_norm": 2.0,
      "grad_norm_var": 0.02217381795247396,
      "learning_rate": 0.0001,
      "loss": 4.1986,
      "loss/crossentropy": 2.16433984041214,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21180611103773117,
      "step": 18276
    },
    {
      "epoch": 0.36556,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.02182184855143229,
      "learning_rate": 0.0001,
      "loss": 4.0345,
      "loss/crossentropy": 1.9628196954727173,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.198894202709198,
      "step": 18278
    },
    {
      "epoch": 0.3656,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.020409901936848957,
      "learning_rate": 0.0001,
      "loss": 4.0616,
      "loss/crossentropy": 1.7818017601966858,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19511079788208008,
      "step": 18280
    },
    {
      "epoch": 0.36564,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.02315241495768229,
      "learning_rate": 0.0001,
      "loss": 4.1769,
      "loss/crossentropy": 2.0149444341659546,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2128366306424141,
      "step": 18282
    },
    {
      "epoch": 0.36568,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.010351308186848958,
      "learning_rate": 0.0001,
      "loss": 4.0475,
      "loss/crossentropy": 2.1580519676208496,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21166586130857468,
      "step": 18284
    },
    {
      "epoch": 0.36572,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.03432591756184896,
      "learning_rate": 0.0001,
      "loss": 3.9196,
      "loss/crossentropy": 2.308629631996155,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20283856242895126,
      "step": 18286
    },
    {
      "epoch": 0.36576,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.03199055989583333,
      "learning_rate": 0.0001,
      "loss": 4.0587,
      "loss/crossentropy": 2.1761614084243774,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.212914377450943,
      "step": 18288
    },
    {
      "epoch": 0.3658,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.029842122395833334,
      "learning_rate": 0.0001,
      "loss": 4.191,
      "loss/crossentropy": 1.9851300120353699,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1898990124464035,
      "step": 18290
    },
    {
      "epoch": 0.36584,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.031172688802083334,
      "learning_rate": 0.0001,
      "loss": 4.028,
      "loss/crossentropy": 2.1123871207237244,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20982655137777328,
      "step": 18292
    },
    {
      "epoch": 0.36588,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.03831761678059896,
      "learning_rate": 0.0001,
      "loss": 4.3007,
      "loss/crossentropy": 2.2666051387786865,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23328793793916702,
      "step": 18294
    },
    {
      "epoch": 0.36592,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.03904393513997396,
      "learning_rate": 0.0001,
      "loss": 3.8741,
      "loss/crossentropy": 1.9444871544837952,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20335105061531067,
      "step": 18296
    },
    {
      "epoch": 0.36596,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.036717732747395836,
      "learning_rate": 0.0001,
      "loss": 4.0799,
      "loss/crossentropy": 2.136604368686676,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20185644924640656,
      "step": 18298
    },
    {
      "epoch": 0.366,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.059024810791015625,
      "learning_rate": 0.0001,
      "loss": 4.0605,
      "loss/crossentropy": 1.9029142260551453,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24649157375097275,
      "step": 18300
    },
    {
      "epoch": 0.36604,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0400054931640625,
      "learning_rate": 0.0001,
      "loss": 4.1685,
      "loss/crossentropy": 2.0363988876342773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20920708030462265,
      "step": 18302
    },
    {
      "epoch": 0.36608,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.04248860677083333,
      "learning_rate": 0.0001,
      "loss": 3.5556,
      "loss/crossentropy": 1.8357294797897339,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18201576173305511,
      "step": 18304
    },
    {
      "epoch": 0.36612,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.04253743489583333,
      "learning_rate": 0.0001,
      "loss": 4.3381,
      "loss/crossentropy": 2.194098114967346,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20245323330163956,
      "step": 18306
    },
    {
      "epoch": 0.36616,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0409332275390625,
      "learning_rate": 0.0001,
      "loss": 4.2501,
      "loss/crossentropy": 2.3840869665145874,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2243075668811798,
      "step": 18308
    },
    {
      "epoch": 0.3662,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.034398396809895836,
      "learning_rate": 0.0001,
      "loss": 4.1895,
      "loss/crossentropy": 2.3093236684799194,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22531048208475113,
      "step": 18310
    },
    {
      "epoch": 0.36624,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.03472493489583333,
      "learning_rate": 0.0001,
      "loss": 3.7629,
      "loss/crossentropy": 2.155013680458069,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19654065370559692,
      "step": 18312
    },
    {
      "epoch": 0.36628,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.03486226399739583,
      "learning_rate": 0.0001,
      "loss": 4.0241,
      "loss/crossentropy": 1.8823603391647339,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18333254009485245,
      "step": 18314
    },
    {
      "epoch": 0.36632,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0090972900390625,
      "learning_rate": 0.0001,
      "loss": 3.8092,
      "loss/crossentropy": 1.9217600226402283,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18095016479492188,
      "step": 18316
    },
    {
      "epoch": 0.36636,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.009004720052083333,
      "learning_rate": 0.0001,
      "loss": 3.8178,
      "loss/crossentropy": 1.7448294758796692,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18972519785165787,
      "step": 18318
    },
    {
      "epoch": 0.3664,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.008442942301432292,
      "learning_rate": 0.0001,
      "loss": 3.9394,
      "loss/crossentropy": 1.8730336427688599,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18396812677383423,
      "step": 18320
    },
    {
      "epoch": 0.36644,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.006955718994140625,
      "learning_rate": 0.0001,
      "loss": 3.8763,
      "loss/crossentropy": 2.075575351715088,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20251936465501785,
      "step": 18322
    },
    {
      "epoch": 0.36648,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.004131825764973959,
      "learning_rate": 0.0001,
      "loss": 4.0955,
      "loss/crossentropy": 2.083173990249634,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2003224566578865,
      "step": 18324
    },
    {
      "epoch": 0.36652,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.00396728515625,
      "learning_rate": 0.0001,
      "loss": 4.0245,
      "loss/crossentropy": 2.4401766061782837,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21872484683990479,
      "step": 18326
    },
    {
      "epoch": 0.36656,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0038533528645833333,
      "learning_rate": 0.0001,
      "loss": 4.0249,
      "loss/crossentropy": 2.2402881383895874,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1977379471063614,
      "step": 18328
    },
    {
      "epoch": 0.3666,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.004650624593098959,
      "learning_rate": 0.0001,
      "loss": 4.1047,
      "loss/crossentropy": 2.2246369123458862,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21132111549377441,
      "step": 18330
    },
    {
      "epoch": 0.36664,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0043365478515625,
      "learning_rate": 0.0001,
      "loss": 4.1607,
      "loss/crossentropy": 1.8048500418663025,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18298982083797455,
      "step": 18332
    },
    {
      "epoch": 0.36668,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.006514231363932292,
      "learning_rate": 0.0001,
      "loss": 4.0987,
      "loss/crossentropy": 2.3368008136749268,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21743982285261154,
      "step": 18334
    },
    {
      "epoch": 0.36672,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.010188547770182292,
      "learning_rate": 0.0001,
      "loss": 4.3158,
      "loss/crossentropy": 2.3520134687423706,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22603602707386017,
      "step": 18336
    },
    {
      "epoch": 0.36676,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.011248524983723958,
      "learning_rate": 0.0001,
      "loss": 4.0037,
      "loss/crossentropy": 2.0071199536323547,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18584266304969788,
      "step": 18338
    },
    {
      "epoch": 0.3668,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.019260406494140625,
      "learning_rate": 0.0001,
      "loss": 4.2787,
      "loss/crossentropy": 1.9329636693000793,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21350353956222534,
      "step": 18340
    },
    {
      "epoch": 0.36684,
      "grad_norm": 2.0,
      "grad_norm_var": 0.015990193684895834,
      "learning_rate": 0.0001,
      "loss": 4.0387,
      "loss/crossentropy": 1.8231948018074036,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2064230963587761,
      "step": 18342
    },
    {
      "epoch": 0.36688,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.01605809529622396,
      "learning_rate": 0.0001,
      "loss": 4.0888,
      "loss/crossentropy": 1.9928399324417114,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1996312066912651,
      "step": 18344
    },
    {
      "epoch": 0.36692,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.015916951497395835,
      "learning_rate": 0.0001,
      "loss": 4.1736,
      "loss/crossentropy": 2.139783501625061,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20407474786043167,
      "step": 18346
    },
    {
      "epoch": 0.36696,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.017166900634765624,
      "learning_rate": 0.0001,
      "loss": 4.0357,
      "loss/crossentropy": 1.8519954681396484,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18404167890548706,
      "step": 18348
    },
    {
      "epoch": 0.367,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.016361236572265625,
      "learning_rate": 0.0001,
      "loss": 4.1325,
      "loss/crossentropy": 2.037365674972534,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2384558469057083,
      "step": 18350
    },
    {
      "epoch": 0.36704,
      "grad_norm": 1.875,
      "grad_norm_var": 0.015600331624348958,
      "learning_rate": 0.0001,
      "loss": 3.887,
      "loss/crossentropy": 2.0023937821388245,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19700156897306442,
      "step": 18352
    },
    {
      "epoch": 0.36708,
      "grad_norm": 2.125,
      "grad_norm_var": 0.015282185872395833,
      "learning_rate": 0.0001,
      "loss": 4.0475,
      "loss/crossentropy": 1.829107940196991,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19559869915246964,
      "step": 18354
    },
    {
      "epoch": 0.36712,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.006012980143229167,
      "learning_rate": 0.0001,
      "loss": 4.0962,
      "loss/crossentropy": 2.206624150276184,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2166636437177658,
      "step": 18356
    },
    {
      "epoch": 0.36716,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.005907185872395833,
      "learning_rate": 0.0001,
      "loss": 4.1191,
      "loss/crossentropy": 2.1971875429153442,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20924220979213715,
      "step": 18358
    },
    {
      "epoch": 0.3672,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0069539388020833336,
      "learning_rate": 0.0001,
      "loss": 4.4876,
      "loss/crossentropy": 2.554604172706604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2638286352157593,
      "step": 18360
    },
    {
      "epoch": 0.36724,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.007981109619140624,
      "learning_rate": 0.0001,
      "loss": 3.9076,
      "loss/crossentropy": 2.088346302509308,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1948780044913292,
      "step": 18362
    },
    {
      "epoch": 0.36728,
      "grad_norm": 1.875,
      "grad_norm_var": 0.0078033447265625,
      "learning_rate": 0.0001,
      "loss": 4.0101,
      "loss/crossentropy": 2.24162495136261,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21616502106189728,
      "step": 18364
    },
    {
      "epoch": 0.36732,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.008567047119140626,
      "learning_rate": 0.0001,
      "loss": 3.8693,
      "loss/crossentropy": 2.249878764152527,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2312733307480812,
      "step": 18366
    },
    {
      "epoch": 0.36736,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0083404541015625,
      "learning_rate": 0.0001,
      "loss": 4.0711,
      "loss/crossentropy": 2.0440531969070435,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18561603128910065,
      "step": 18368
    },
    {
      "epoch": 0.3674,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.006696573893229167,
      "learning_rate": 0.0001,
      "loss": 4.2582,
      "loss/crossentropy": 2.2324228286743164,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20741964876651764,
      "step": 18370
    },
    {
      "epoch": 0.36744,
      "grad_norm": 2.125,
      "grad_norm_var": 0.008634440104166667,
      "learning_rate": 0.0001,
      "loss": 4.2807,
      "loss/crossentropy": 2.168972373008728,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.215408056974411,
      "step": 18372
    },
    {
      "epoch": 0.36748,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.008955637613932291,
      "learning_rate": 0.0001,
      "loss": 4.0341,
      "loss/crossentropy": 2.076392412185669,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2023996263742447,
      "step": 18374
    },
    {
      "epoch": 0.36752,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.005968983968098958,
      "learning_rate": 0.0001,
      "loss": 4.2958,
      "loss/crossentropy": 2.0497928857803345,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21541842073202133,
      "step": 18376
    },
    {
      "epoch": 0.36756,
      "grad_norm": 1.6875,
      "grad_norm_var": 0.0102294921875,
      "learning_rate": 0.0001,
      "loss": 3.6889,
      "loss/crossentropy": 1.8894451260566711,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18294879794120789,
      "step": 18378
    },
    {
      "epoch": 0.3676,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.010545857747395833,
      "learning_rate": 0.0001,
      "loss": 4.0237,
      "loss/crossentropy": 2.1654014587402344,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18372679501771927,
      "step": 18380
    },
    {
      "epoch": 0.36764,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.013516998291015625,
      "learning_rate": 0.0001,
      "loss": 4.0657,
      "loss/crossentropy": 2.203396499156952,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22593200951814651,
      "step": 18382
    },
    {
      "epoch": 0.36768,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.014090728759765626,
      "learning_rate": 0.0001,
      "loss": 3.946,
      "loss/crossentropy": 1.7496543526649475,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1825375333428383,
      "step": 18384
    },
    {
      "epoch": 0.36772,
      "grad_norm": 1.875,
      "grad_norm_var": 0.014249420166015625,
      "learning_rate": 0.0001,
      "loss": 3.7667,
      "loss/crossentropy": 1.9488004446029663,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18244057893753052,
      "step": 18386
    },
    {
      "epoch": 0.36776,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.01231689453125,
      "learning_rate": 0.0001,
      "loss": 3.9406,
      "loss/crossentropy": 1.9792875051498413,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20204880088567734,
      "step": 18388
    },
    {
      "epoch": 0.3678,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.012412261962890626,
      "learning_rate": 0.0001,
      "loss": 4.0077,
      "loss/crossentropy": 1.890614092350006,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19333010911941528,
      "step": 18390
    },
    {
      "epoch": 0.36784,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.011832427978515626,
      "learning_rate": 0.0001,
      "loss": 4.0615,
      "loss/crossentropy": 2.0901471972465515,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20534122735261917,
      "step": 18392
    },
    {
      "epoch": 0.36788,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.00716552734375,
      "learning_rate": 0.0001,
      "loss": 4.2502,
      "loss/crossentropy": 2.2931089401245117,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2130991816520691,
      "step": 18394
    },
    {
      "epoch": 0.36792,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0077392578125,
      "learning_rate": 0.0001,
      "loss": 3.9862,
      "loss/crossentropy": 2.2144237756729126,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22273198515176773,
      "step": 18396
    },
    {
      "epoch": 0.36796,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.006624094645182292,
      "learning_rate": 0.0001,
      "loss": 3.7365,
      "loss/crossentropy": 1.9135422110557556,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1983061358332634,
      "step": 18398
    },
    {
      "epoch": 0.368,
      "grad_norm": 2.0,
      "grad_norm_var": 0.007456207275390625,
      "learning_rate": 0.0001,
      "loss": 3.7556,
      "loss/crossentropy": 1.692852795124054,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1670849658548832,
      "step": 18400
    },
    {
      "epoch": 0.36804,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0083160400390625,
      "learning_rate": 0.0001,
      "loss": 4.0043,
      "loss/crossentropy": 1.908652126789093,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21787738054990768,
      "step": 18402
    },
    {
      "epoch": 0.36808,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008432769775390625,
      "learning_rate": 0.0001,
      "loss": 4.2343,
      "loss/crossentropy": 2.0613549947738647,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20495950430631638,
      "step": 18404
    },
    {
      "epoch": 0.36812,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.026759592692057292,
      "learning_rate": 0.0001,
      "loss": 4.0549,
      "loss/crossentropy": 1.9765326976776123,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18859146535396576,
      "step": 18406
    },
    {
      "epoch": 0.36816,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.027147420247395835,
      "learning_rate": 0.0001,
      "loss": 4.0196,
      "loss/crossentropy": 1.8403696417808533,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18926746398210526,
      "step": 18408
    },
    {
      "epoch": 0.3682,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.029080963134765624,
      "learning_rate": 0.0001,
      "loss": 4.1859,
      "loss/crossentropy": 2.1792644262313843,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22263357043266296,
      "step": 18410
    },
    {
      "epoch": 0.36824,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.02910334269205729,
      "learning_rate": 0.0001,
      "loss": 4.2387,
      "loss/crossentropy": 2.079226016998291,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20990607887506485,
      "step": 18412
    },
    {
      "epoch": 0.36828,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.027457427978515626,
      "learning_rate": 0.0001,
      "loss": 4.2093,
      "loss/crossentropy": 2.1428889632225037,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22140806913375854,
      "step": 18414
    },
    {
      "epoch": 0.36832,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.024589029947916667,
      "learning_rate": 0.0001,
      "loss": 4.3761,
      "loss/crossentropy": 2.1461609601974487,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20594316720962524,
      "step": 18416
    },
    {
      "epoch": 0.36836,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.02399470011393229,
      "learning_rate": 0.0001,
      "loss": 4.1597,
      "loss/crossentropy": 1.980036735534668,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.201970636844635,
      "step": 18418
    },
    {
      "epoch": 0.3684,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.024773915608723957,
      "learning_rate": 0.0001,
      "loss": 3.9847,
      "loss/crossentropy": 1.6273554563522339,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1938486322760582,
      "step": 18420
    },
    {
      "epoch": 0.36844,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.006251780192057291,
      "learning_rate": 0.0001,
      "loss": 4.1784,
      "loss/crossentropy": 1.9100900292396545,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18063092976808548,
      "step": 18422
    },
    {
      "epoch": 0.36848,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0063555399576822914,
      "learning_rate": 0.0001,
      "loss": 4.1282,
      "loss/crossentropy": 2.1882822513580322,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21261122822761536,
      "step": 18424
    },
    {
      "epoch": 0.36852,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.0046384175618489586,
      "learning_rate": 0.0001,
      "loss": 3.9556,
      "loss/crossentropy": 2.210257649421692,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1929008737206459,
      "step": 18426
    },
    {
      "epoch": 0.36856,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.005041249593098958,
      "learning_rate": 0.0001,
      "loss": 4.1691,
      "loss/crossentropy": 2.0247724056243896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2004971206188202,
      "step": 18428
    },
    {
      "epoch": 0.3686,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.004133097330729167,
      "learning_rate": 0.0001,
      "loss": 4.2092,
      "loss/crossentropy": 1.9844316244125366,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.210312619805336,
      "step": 18430
    },
    {
      "epoch": 0.36864,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.005000559488932291,
      "learning_rate": 0.0001,
      "loss": 4.0368,
      "loss/crossentropy": 2.2416625022888184,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21065251529216766,
      "step": 18432
    },
    {
      "epoch": 0.36868,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.020104726155598957,
      "learning_rate": 0.0001,
      "loss": 4.4109,
      "loss/crossentropy": 2.0283551812171936,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2150302305817604,
      "step": 18434
    },
    {
      "epoch": 0.36872,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.01968994140625,
      "learning_rate": 0.0001,
      "loss": 4.1379,
      "loss/crossentropy": 2.210233688354492,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21831902861595154,
      "step": 18436
    },
    {
      "epoch": 0.36876,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0205474853515625,
      "learning_rate": 0.0001,
      "loss": 3.9993,
      "loss/crossentropy": 2.0454147458076477,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2432640790939331,
      "step": 18438
    },
    {
      "epoch": 0.3688,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.020444488525390624,
      "learning_rate": 0.0001,
      "loss": 4.1243,
      "loss/crossentropy": 2.024592399597168,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20496471971273422,
      "step": 18440
    },
    {
      "epoch": 0.36884,
      "grad_norm": 2.125,
      "grad_norm_var": 0.018822987874348957,
      "learning_rate": 0.0001,
      "loss": 4.3058,
      "loss/crossentropy": 2.4205459356307983,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22281523793935776,
      "step": 18442
    },
    {
      "epoch": 0.36888,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0170806884765625,
      "learning_rate": 0.0001,
      "loss": 4.0341,
      "loss/crossentropy": 1.9198943376541138,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1916896402835846,
      "step": 18444
    },
    {
      "epoch": 0.36892,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0171051025390625,
      "learning_rate": 0.0001,
      "loss": 4.0475,
      "loss/crossentropy": 1.928059160709381,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1899876967072487,
      "step": 18446
    },
    {
      "epoch": 0.36896,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.020792388916015626,
      "learning_rate": 0.0001,
      "loss": 4.3327,
      "loss/crossentropy": 2.346290349960327,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22388508170843124,
      "step": 18448
    },
    {
      "epoch": 0.369,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.009946441650390625,
      "learning_rate": 0.0001,
      "loss": 3.9064,
      "loss/crossentropy": 2.0108938217163086,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19462434202432632,
      "step": 18450
    },
    {
      "epoch": 0.36904,
      "grad_norm": 2.0,
      "grad_norm_var": 0.00986328125,
      "learning_rate": 0.0001,
      "loss": 4.2468,
      "loss/crossentropy": 1.7108886241912842,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18806710839271545,
      "step": 18452
    },
    {
      "epoch": 0.36908,
      "grad_norm": 5.1875,
      "grad_norm_var": 0.6345499674479167,
      "learning_rate": 0.0001,
      "loss": 4.6972,
      "loss/crossentropy": 2.4162802696228027,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3217930570244789,
      "step": 18454
    },
    {
      "epoch": 0.36912,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.639013671875,
      "learning_rate": 0.0001,
      "loss": 3.4762,
      "loss/crossentropy": 1.6459838151931763,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17543485760688782,
      "step": 18456
    },
    {
      "epoch": 0.36916,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.640679677327474,
      "learning_rate": 0.0001,
      "loss": 4.1194,
      "loss/crossentropy": 2.2111966013908386,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22191359847784042,
      "step": 18458
    },
    {
      "epoch": 0.3692,
      "grad_norm": 2.125,
      "grad_norm_var": 0.6325887044270834,
      "learning_rate": 0.0001,
      "loss": 4.301,
      "loss/crossentropy": 2.2514692544937134,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21384654194116592,
      "step": 18460
    },
    {
      "epoch": 0.36924,
      "grad_norm": 2.125,
      "grad_norm_var": 0.63665771484375,
      "learning_rate": 0.0001,
      "loss": 4.307,
      "loss/crossentropy": 2.097872793674469,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21747223287820816,
      "step": 18462
    },
    {
      "epoch": 0.36928,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.6418690999348958,
      "learning_rate": 0.0001,
      "loss": 4.1033,
      "loss/crossentropy": 2.1611807346343994,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2042219042778015,
      "step": 18464
    },
    {
      "epoch": 0.36932,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.62939453125,
      "learning_rate": 0.0001,
      "loss": 3.7871,
      "loss/crossentropy": 1.8941562175750732,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.192266546189785,
      "step": 18466
    },
    {
      "epoch": 0.36936,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.72021484375,
      "learning_rate": 0.0001,
      "loss": 4.0358,
      "loss/crossentropy": 1.8930317163467407,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19436348974704742,
      "step": 18468
    },
    {
      "epoch": 0.3694,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.15979588826497396,
      "learning_rate": 0.0001,
      "loss": 4.1046,
      "loss/crossentropy": 2.1041005849838257,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21314333379268646,
      "step": 18470
    },
    {
      "epoch": 0.36944,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.15750732421875,
      "learning_rate": 0.0001,
      "loss": 4.2096,
      "loss/crossentropy": 1.8747637867927551,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19172174483537674,
      "step": 18472
    },
    {
      "epoch": 0.36948,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.15751113891601562,
      "learning_rate": 0.0001,
      "loss": 3.9809,
      "loss/crossentropy": 2.1883610486984253,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19927512109279633,
      "step": 18474
    },
    {
      "epoch": 0.36952,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.1606353759765625,
      "learning_rate": 0.0001,
      "loss": 3.9303,
      "loss/crossentropy": 1.972772240638733,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17948149144649506,
      "step": 18476
    },
    {
      "epoch": 0.36956,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.14570490519205728,
      "learning_rate": 0.0001,
      "loss": 4.1246,
      "loss/crossentropy": 1.8912597298622131,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18716244399547577,
      "step": 18478
    },
    {
      "epoch": 0.3696,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.1458740234375,
      "learning_rate": 0.0001,
      "loss": 4.0816,
      "loss/crossentropy": 2.044828712940216,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2076854333281517,
      "step": 18480
    },
    {
      "epoch": 0.36964,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.14629618326822916,
      "learning_rate": 0.0001,
      "loss": 4.2007,
      "loss/crossentropy": 2.0936968326568604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22010911256074905,
      "step": 18482
    },
    {
      "epoch": 0.36968,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0027414957682291665,
      "learning_rate": 0.0001,
      "loss": 4.08,
      "loss/crossentropy": 2.2678394317626953,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2073991820216179,
      "step": 18484
    },
    {
      "epoch": 0.36972,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0031064351399739585,
      "learning_rate": 0.0001,
      "loss": 3.9361,
      "loss/crossentropy": 2.281362295150757,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2015322744846344,
      "step": 18486
    },
    {
      "epoch": 0.36976,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0034739176432291665,
      "learning_rate": 0.0001,
      "loss": 4.2318,
      "loss/crossentropy": 2.0029674768447876,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2155751883983612,
      "step": 18488
    },
    {
      "epoch": 0.3698,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.003525543212890625,
      "learning_rate": 0.0001,
      "loss": 4.3761,
      "loss/crossentropy": 2.370519280433655,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2308923304080963,
      "step": 18490
    },
    {
      "epoch": 0.36984,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.004109446207682292,
      "learning_rate": 0.0001,
      "loss": 4.3137,
      "loss/crossentropy": 1.9416582584381104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18809420615434647,
      "step": 18492
    },
    {
      "epoch": 0.36988,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.003885650634765625,
      "learning_rate": 0.0001,
      "loss": 4.1381,
      "loss/crossentropy": 2.065304160118103,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19029072672128677,
      "step": 18494
    },
    {
      "epoch": 0.36992,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0036516825358072916,
      "learning_rate": 0.0001,
      "loss": 4.1366,
      "loss/crossentropy": 1.9798340201377869,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1716388538479805,
      "step": 18496
    },
    {
      "epoch": 0.36996,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0035540262858072915,
      "learning_rate": 0.0001,
      "loss": 4.2321,
      "loss/crossentropy": 2.336732864379883,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19805839657783508,
      "step": 18498
    },
    {
      "epoch": 0.37,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0033444722493489584,
      "learning_rate": 0.0001,
      "loss": 4.08,
      "loss/crossentropy": 1.9556902050971985,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19576922804117203,
      "step": 18500
    },
    {
      "epoch": 0.37004,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.006581370035807292,
      "learning_rate": 0.0001,
      "loss": 3.7281,
      "loss/crossentropy": 2.0062427520751953,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19548972696065903,
      "step": 18502
    },
    {
      "epoch": 0.37008,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0061419169108072914,
      "learning_rate": 0.0001,
      "loss": 4.0954,
      "loss/crossentropy": 2.1030293703079224,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22147215902805328,
      "step": 18504
    },
    {
      "epoch": 0.37012,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.005163319905598958,
      "learning_rate": 0.0001,
      "loss": 3.9545,
      "loss/crossentropy": 1.7618860006332397,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17934025079011917,
      "step": 18506
    },
    {
      "epoch": 0.37016,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.005454254150390625,
      "learning_rate": 0.0001,
      "loss": 4.239,
      "loss/crossentropy": 2.4424854516983032,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23131447285413742,
      "step": 18508
    },
    {
      "epoch": 0.3702,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0076904296875,
      "learning_rate": 0.0001,
      "loss": 4.0365,
      "loss/crossentropy": 2.125267446041107,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19127625226974487,
      "step": 18510
    },
    {
      "epoch": 0.37024,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.008506011962890626,
      "learning_rate": 0.0001,
      "loss": 3.9498,
      "loss/crossentropy": 2.2222214937210083,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.204838365316391,
      "step": 18512
    },
    {
      "epoch": 0.37028,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.009056599934895833,
      "learning_rate": 0.0001,
      "loss": 4.1012,
      "loss/crossentropy": 2.1930031776428223,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21769095957279205,
      "step": 18514
    },
    {
      "epoch": 0.37032,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.010814412434895834,
      "learning_rate": 0.0001,
      "loss": 4.0068,
      "loss/crossentropy": 2.078152060508728,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19528701901435852,
      "step": 18516
    },
    {
      "epoch": 0.37036,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.009016927083333333,
      "learning_rate": 0.0001,
      "loss": 4.0135,
      "loss/crossentropy": 1.9786349534988403,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19193317741155624,
      "step": 18518
    },
    {
      "epoch": 0.3704,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0119293212890625,
      "learning_rate": 0.0001,
      "loss": 4.1458,
      "loss/crossentropy": 2.279360294342041,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20865458250045776,
      "step": 18520
    },
    {
      "epoch": 0.37044,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.012548828125,
      "learning_rate": 0.0001,
      "loss": 4.195,
      "loss/crossentropy": 2.0668978691101074,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19060996919870377,
      "step": 18522
    },
    {
      "epoch": 0.37048,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.011116536458333333,
      "learning_rate": 0.0001,
      "loss": 4.2661,
      "loss/crossentropy": 2.4433913230895996,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2263498529791832,
      "step": 18524
    },
    {
      "epoch": 0.37052,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.009049224853515624,
      "learning_rate": 0.0001,
      "loss": 3.7953,
      "loss/crossentropy": 1.7785582542419434,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18417692929506302,
      "step": 18526
    },
    {
      "epoch": 0.37056,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008499908447265624,
      "learning_rate": 0.0001,
      "loss": 4.2958,
      "loss/crossentropy": 1.9753797054290771,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1898263543844223,
      "step": 18528
    },
    {
      "epoch": 0.3706,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.007696278889973958,
      "learning_rate": 0.0001,
      "loss": 4.1945,
      "loss/crossentropy": 2.302052319049835,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21289903670549393,
      "step": 18530
    },
    {
      "epoch": 0.37064,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.007523600260416667,
      "learning_rate": 0.0001,
      "loss": 4.1132,
      "loss/crossentropy": 2.3275071382522583,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2265043556690216,
      "step": 18532
    },
    {
      "epoch": 0.37068,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.00826416015625,
      "learning_rate": 0.0001,
      "loss": 4.2004,
      "loss/crossentropy": 2.3155715465545654,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20101473480463028,
      "step": 18534
    },
    {
      "epoch": 0.37072,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.0075681050618489586,
      "learning_rate": 0.0001,
      "loss": 3.8546,
      "loss/crossentropy": 2.0893908739089966,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18266596645116806,
      "step": 18536
    },
    {
      "epoch": 0.37076,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.006546783447265625,
      "learning_rate": 0.0001,
      "loss": 4.297,
      "loss/crossentropy": 2.075433909893036,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20270193368196487,
      "step": 18538
    },
    {
      "epoch": 0.3708,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.006605784098307292,
      "learning_rate": 0.0001,
      "loss": 4.186,
      "loss/crossentropy": 2.2795485258102417,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21547292172908783,
      "step": 18540
    },
    {
      "epoch": 0.37084,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.006956990559895833,
      "learning_rate": 0.0001,
      "loss": 4.0351,
      "loss/crossentropy": 2.0038467049598694,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19927001744508743,
      "step": 18542
    },
    {
      "epoch": 0.37088,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0056111653645833336,
      "learning_rate": 0.0001,
      "loss": 3.979,
      "loss/crossentropy": 1.8877951502799988,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19283732771873474,
      "step": 18544
    },
    {
      "epoch": 0.37092,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0065305074055989586,
      "learning_rate": 0.0001,
      "loss": 4.0169,
      "loss/crossentropy": 1.8111292719841003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20587582886219025,
      "step": 18546
    },
    {
      "epoch": 0.37096,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.004548136393229167,
      "learning_rate": 0.0001,
      "loss": 4.0766,
      "loss/crossentropy": 2.12824147939682,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1996314972639084,
      "step": 18548
    },
    {
      "epoch": 0.371,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0035845438639322915,
      "learning_rate": 0.0001,
      "loss": 3.9549,
      "loss/crossentropy": 2.22495698928833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20922152698040009,
      "step": 18550
    },
    {
      "epoch": 0.37104,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.003110504150390625,
      "learning_rate": 0.0001,
      "loss": 4.2245,
      "loss/crossentropy": 2.1434414386749268,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2070355862379074,
      "step": 18552
    },
    {
      "epoch": 0.37108,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.003979237874348959,
      "learning_rate": 0.0001,
      "loss": 4.0359,
      "loss/crossentropy": 2.248233437538147,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21400754898786545,
      "step": 18554
    },
    {
      "epoch": 0.37112,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0051513671875,
      "learning_rate": 0.0001,
      "loss": 4.2072,
      "loss/crossentropy": 2.246641755104065,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22847777605056763,
      "step": 18556
    },
    {
      "epoch": 0.37116,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.004654693603515625,
      "learning_rate": 0.0001,
      "loss": 4.0805,
      "loss/crossentropy": 2.057798206806183,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23668289184570312,
      "step": 18558
    },
    {
      "epoch": 0.3712,
      "grad_norm": 2.0,
      "grad_norm_var": 0.004587554931640625,
      "learning_rate": 0.0001,
      "loss": 3.8454,
      "loss/crossentropy": 1.7570822834968567,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1820850819349289,
      "step": 18560
    },
    {
      "epoch": 0.37124,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.015372467041015626,
      "learning_rate": 0.0001,
      "loss": 4.0077,
      "loss/crossentropy": 1.7747303247451782,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1704205796122551,
      "step": 18562
    },
    {
      "epoch": 0.37128,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.015632120768229167,
      "learning_rate": 0.0001,
      "loss": 4.0611,
      "loss/crossentropy": 2.3060861825942993,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20002726465463638,
      "step": 18564
    },
    {
      "epoch": 0.37132,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.015242258707682291,
      "learning_rate": 0.0001,
      "loss": 4.258,
      "loss/crossentropy": 2.2742475271224976,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22524481266736984,
      "step": 18566
    },
    {
      "epoch": 0.37136,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.01647923787434896,
      "learning_rate": 0.0001,
      "loss": 3.9816,
      "loss/crossentropy": 1.7929689288139343,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1819063350558281,
      "step": 18568
    },
    {
      "epoch": 0.3714,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.015718587239583335,
      "learning_rate": 0.0001,
      "loss": 4.2487,
      "loss/crossentropy": 2.4505950212478638,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22857815772294998,
      "step": 18570
    },
    {
      "epoch": 0.37144,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.017658487955729166,
      "learning_rate": 0.0001,
      "loss": 3.5856,
      "loss/crossentropy": 1.640372097492218,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16958069056272507,
      "step": 18572
    },
    {
      "epoch": 0.37148,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.01916681925455729,
      "learning_rate": 0.0001,
      "loss": 4.1601,
      "loss/crossentropy": 2.1556472778320312,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20749886333942413,
      "step": 18574
    },
    {
      "epoch": 0.37152,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.01846491495768229,
      "learning_rate": 0.0001,
      "loss": 4.1579,
      "loss/crossentropy": 2.1131449937820435,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20596522092819214,
      "step": 18576
    },
    {
      "epoch": 0.37156,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.008998362223307292,
      "learning_rate": 0.0001,
      "loss": 3.8036,
      "loss/crossentropy": 1.8966050148010254,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19289513677358627,
      "step": 18578
    },
    {
      "epoch": 0.3716,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.010306803385416667,
      "learning_rate": 0.0001,
      "loss": 4.1883,
      "loss/crossentropy": 2.1016936898231506,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1917622685432434,
      "step": 18580
    },
    {
      "epoch": 0.37164,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.01512451171875,
      "learning_rate": 0.0001,
      "loss": 4.2596,
      "loss/crossentropy": 2.289618492126465,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2033960521221161,
      "step": 18582
    },
    {
      "epoch": 0.37168,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.020287068684895833,
      "learning_rate": 0.0001,
      "loss": 4.0441,
      "loss/crossentropy": 2.205715775489807,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20015332102775574,
      "step": 18584
    },
    {
      "epoch": 0.37172,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.02072118123372396,
      "learning_rate": 0.0001,
      "loss": 4.0834,
      "loss/crossentropy": 2.3037471771240234,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2424507886171341,
      "step": 18586
    },
    {
      "epoch": 0.37176,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.016686757405598957,
      "learning_rate": 0.0001,
      "loss": 4.0718,
      "loss/crossentropy": 2.1757054328918457,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20777830481529236,
      "step": 18588
    },
    {
      "epoch": 0.3718,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.015860748291015626,
      "learning_rate": 0.0001,
      "loss": 4.2905,
      "loss/crossentropy": 2.3257133960723877,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22848188877105713,
      "step": 18590
    },
    {
      "epoch": 0.37184,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.017179107666015624,
      "learning_rate": 0.0001,
      "loss": 3.9132,
      "loss/crossentropy": 2.3375465869903564,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21214719116687775,
      "step": 18592
    },
    {
      "epoch": 0.37188,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0162750244140625,
      "learning_rate": 0.0001,
      "loss": 4.507,
      "loss/crossentropy": 2.1896092891693115,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2303280457854271,
      "step": 18594
    },
    {
      "epoch": 0.37192,
      "grad_norm": 2.0,
      "grad_norm_var": 0.016287994384765626,
      "learning_rate": 0.0001,
      "loss": 3.9078,
      "loss/crossentropy": 1.8913645148277283,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17311514914035797,
      "step": 18596
    },
    {
      "epoch": 0.37196,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.012442779541015626,
      "learning_rate": 0.0001,
      "loss": 4.2212,
      "loss/crossentropy": 2.5278197526931763,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23568174242973328,
      "step": 18598
    },
    {
      "epoch": 0.372,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0101959228515625,
      "learning_rate": 0.0001,
      "loss": 4.1568,
      "loss/crossentropy": 2.073936700820923,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21885155141353607,
      "step": 18600
    },
    {
      "epoch": 0.37204,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.012684885660807292,
      "learning_rate": 0.0001,
      "loss": 4.388,
      "loss/crossentropy": 2.162986159324646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2012084275484085,
      "step": 18602
    },
    {
      "epoch": 0.37208,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.013108062744140624,
      "learning_rate": 0.0001,
      "loss": 3.9659,
      "loss/crossentropy": 1.9332863092422485,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18284077942371368,
      "step": 18604
    },
    {
      "epoch": 0.37212,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.013809967041015624,
      "learning_rate": 0.0001,
      "loss": 4.0279,
      "loss/crossentropy": 2.083172380924225,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.186289481818676,
      "step": 18606
    },
    {
      "epoch": 0.37216,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.013588205973307291,
      "learning_rate": 0.0001,
      "loss": 3.9932,
      "loss/crossentropy": 2.100313901901245,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1977698802947998,
      "step": 18608
    },
    {
      "epoch": 0.3722,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.016792805989583333,
      "learning_rate": 0.0001,
      "loss": 4.1086,
      "loss/crossentropy": 2.2336236238479614,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21973469108343124,
      "step": 18610
    },
    {
      "epoch": 0.37224,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.017048136393229166,
      "learning_rate": 0.0001,
      "loss": 3.9422,
      "loss/crossentropy": 2.0263352394104004,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19232206791639328,
      "step": 18612
    },
    {
      "epoch": 0.37228,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.01727472941080729,
      "learning_rate": 0.0001,
      "loss": 4.2295,
      "loss/crossentropy": 2.0719032287597656,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22896382957696915,
      "step": 18614
    },
    {
      "epoch": 0.37232,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.015400950113932292,
      "learning_rate": 0.0001,
      "loss": 3.881,
      "loss/crossentropy": 1.9514707326889038,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1899290755391121,
      "step": 18616
    },
    {
      "epoch": 0.37236,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.010172526041666666,
      "learning_rate": 0.0001,
      "loss": 4.0163,
      "loss/crossentropy": 2.0810786485671997,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1983262002468109,
      "step": 18618
    },
    {
      "epoch": 0.3724,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.010302734375,
      "learning_rate": 0.0001,
      "loss": 4.1311,
      "loss/crossentropy": 1.823366403579712,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18820683658123016,
      "step": 18620
    },
    {
      "epoch": 0.37244,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.023933919270833333,
      "learning_rate": 0.0001,
      "loss": 4.2361,
      "loss/crossentropy": 2.2814120054244995,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19500700384378433,
      "step": 18622
    },
    {
      "epoch": 0.37248,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.023339589436848957,
      "learning_rate": 0.0001,
      "loss": 3.9548,
      "loss/crossentropy": 1.5761349201202393,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17777466028928757,
      "step": 18624
    },
    {
      "epoch": 0.37252,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.018853505452473957,
      "learning_rate": 0.0001,
      "loss": 3.9134,
      "loss/crossentropy": 1.8996286988258362,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21907050907611847,
      "step": 18626
    },
    {
      "epoch": 0.37256,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.020869700113932292,
      "learning_rate": 0.0001,
      "loss": 4.2361,
      "loss/crossentropy": 1.6650620698928833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18808475136756897,
      "step": 18628
    },
    {
      "epoch": 0.3726,
      "grad_norm": 2.125,
      "grad_norm_var": 0.02156550089518229,
      "learning_rate": 0.0001,
      "loss": 4.4004,
      "loss/crossentropy": 2.2887942790985107,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20416373759508133,
      "step": 18630
    },
    {
      "epoch": 0.37264,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.018629709879557293,
      "learning_rate": 0.0001,
      "loss": 3.9606,
      "loss/crossentropy": 1.9479430317878723,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18489708751440048,
      "step": 18632
    },
    {
      "epoch": 0.37268,
      "grad_norm": 2.0,
      "grad_norm_var": 0.016676584879557293,
      "learning_rate": 0.0001,
      "loss": 4.3041,
      "loss/crossentropy": 2.2156901359558105,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20997263491153717,
      "step": 18634
    },
    {
      "epoch": 0.37272,
      "grad_norm": 2.0,
      "grad_norm_var": 0.015950520833333332,
      "learning_rate": 0.0001,
      "loss": 4.1619,
      "loss/crossentropy": 2.50583279132843,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23347270488739014,
      "step": 18636
    },
    {
      "epoch": 0.37276,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.008650716145833333,
      "learning_rate": 0.0001,
      "loss": 3.8241,
      "loss/crossentropy": 1.7688243985176086,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18601343035697937,
      "step": 18638
    },
    {
      "epoch": 0.3728,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.009439849853515625,
      "learning_rate": 0.0001,
      "loss": 4.2241,
      "loss/crossentropy": 2.3786444664001465,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2287142351269722,
      "step": 18640
    },
    {
      "epoch": 0.37284,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.009479777018229166,
      "learning_rate": 0.0001,
      "loss": 4.243,
      "loss/crossentropy": 2.2686339616775513,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21086719632148743,
      "step": 18642
    },
    {
      "epoch": 0.37288,
      "grad_norm": 2.0,
      "grad_norm_var": 0.007490793863932292,
      "learning_rate": 0.0001,
      "loss": 3.8341,
      "loss/crossentropy": 1.9386130571365356,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19289222359657288,
      "step": 18644
    },
    {
      "epoch": 0.37292,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.006819407145182292,
      "learning_rate": 0.0001,
      "loss": 4.3357,
      "loss/crossentropy": 2.3655601739883423,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2207036018371582,
      "step": 18646
    },
    {
      "epoch": 0.37296,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.007100168863932292,
      "learning_rate": 0.0001,
      "loss": 3.9039,
      "loss/crossentropy": 1.9600831270217896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18018172681331635,
      "step": 18648
    },
    {
      "epoch": 0.373,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.024873860677083335,
      "learning_rate": 0.0001,
      "loss": 3.9492,
      "loss/crossentropy": 1.86528480052948,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19763591885566711,
      "step": 18650
    },
    {
      "epoch": 0.37304,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.024925740559895833,
      "learning_rate": 0.0001,
      "loss": 4.07,
      "loss/crossentropy": 2.0938327312469482,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2051714062690735,
      "step": 18652
    },
    {
      "epoch": 0.37308,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.022454579671223957,
      "learning_rate": 0.0001,
      "loss": 3.878,
      "loss/crossentropy": 1.964399516582489,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20198997855186462,
      "step": 18654
    },
    {
      "epoch": 0.37312,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0245849609375,
      "learning_rate": 0.0001,
      "loss": 3.9929,
      "loss/crossentropy": 2.0186336040496826,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19310183823108673,
      "step": 18656
    },
    {
      "epoch": 0.37316,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.025052642822265624,
      "learning_rate": 0.0001,
      "loss": 3.8569,
      "loss/crossentropy": 1.8805240392684937,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20392487198114395,
      "step": 18658
    },
    {
      "epoch": 0.3732,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.025480143229166665,
      "learning_rate": 0.0001,
      "loss": 4.0306,
      "loss/crossentropy": 2.45102322101593,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2161283940076828,
      "step": 18660
    },
    {
      "epoch": 0.37324,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.025679270426432293,
      "learning_rate": 0.0001,
      "loss": 4.1015,
      "loss/crossentropy": 2.2971357107162476,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23155176639556885,
      "step": 18662
    },
    {
      "epoch": 0.37328,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.025614166259765626,
      "learning_rate": 0.0001,
      "loss": 4.285,
      "loss/crossentropy": 2.3736867904663086,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22897879034280777,
      "step": 18664
    },
    {
      "epoch": 0.37332,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.005500284830729166,
      "learning_rate": 0.0001,
      "loss": 3.9884,
      "loss/crossentropy": 1.5138108134269714,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1526506468653679,
      "step": 18666
    },
    {
      "epoch": 0.37336,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.0061948140462239586,
      "learning_rate": 0.0001,
      "loss": 3.9507,
      "loss/crossentropy": 2.415435791015625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21689960360527039,
      "step": 18668
    },
    {
      "epoch": 0.3734,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.006306966145833333,
      "learning_rate": 0.0001,
      "loss": 4.014,
      "loss/crossentropy": 2.1235941648483276,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20921900868415833,
      "step": 18670
    },
    {
      "epoch": 0.37344,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.004428863525390625,
      "learning_rate": 0.0001,
      "loss": 4.2549,
      "loss/crossentropy": 2.3612579703330994,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21255160123109818,
      "step": 18672
    },
    {
      "epoch": 0.37348,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.054351552327473955,
      "learning_rate": 0.0001,
      "loss": 4.1722,
      "loss/crossentropy": 2.23550283908844,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18890902400016785,
      "step": 18674
    },
    {
      "epoch": 0.37352,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.05468317667643229,
      "learning_rate": 0.0001,
      "loss": 4.0985,
      "loss/crossentropy": 1.7998243570327759,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19070486724376678,
      "step": 18676
    },
    {
      "epoch": 0.37356,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.055214182535807295,
      "learning_rate": 0.0001,
      "loss": 4.151,
      "loss/crossentropy": 1.892760992050171,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19817107915878296,
      "step": 18678
    },
    {
      "epoch": 0.3736,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.055757395426432294,
      "learning_rate": 0.0001,
      "loss": 3.9862,
      "loss/crossentropy": 1.9619091153144836,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21299303323030472,
      "step": 18680
    },
    {
      "epoch": 0.37364,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.057889556884765624,
      "learning_rate": 0.0001,
      "loss": 3.8419,
      "loss/crossentropy": 2.0857014656066895,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20634424686431885,
      "step": 18682
    },
    {
      "epoch": 0.37368,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.05814793904622396,
      "learning_rate": 0.0001,
      "loss": 3.9589,
      "loss/crossentropy": 2.1093358397483826,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1933949589729309,
      "step": 18684
    },
    {
      "epoch": 0.37372,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.05987548828125,
      "learning_rate": 0.0001,
      "loss": 3.9537,
      "loss/crossentropy": 2.1040873527526855,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2061748430132866,
      "step": 18686
    },
    {
      "epoch": 0.37376,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.059081776936848955,
      "learning_rate": 0.0001,
      "loss": 4.1676,
      "loss/crossentropy": 2.4795751571655273,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2075038179755211,
      "step": 18688
    },
    {
      "epoch": 0.3738,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.006599934895833334,
      "learning_rate": 0.0001,
      "loss": 3.9992,
      "loss/crossentropy": 1.864591360092163,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17119919508695602,
      "step": 18690
    },
    {
      "epoch": 0.37384,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.007063547770182292,
      "learning_rate": 0.0001,
      "loss": 3.7565,
      "loss/crossentropy": 1.6705753207206726,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19292957335710526,
      "step": 18692
    },
    {
      "epoch": 0.37388,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.004209136962890625,
      "learning_rate": 0.0001,
      "loss": 4.0448,
      "loss/crossentropy": 2.16109561920166,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20515639334917068,
      "step": 18694
    },
    {
      "epoch": 0.37392,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.014212799072265626,
      "learning_rate": 0.0001,
      "loss": 4.1683,
      "loss/crossentropy": 1.771731436252594,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20304742455482483,
      "step": 18696
    },
    {
      "epoch": 0.37396,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.014330037434895833,
      "learning_rate": 0.0001,
      "loss": 3.9636,
      "loss/crossentropy": 2.046182096004486,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1976921409368515,
      "step": 18698
    },
    {
      "epoch": 0.374,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.0148345947265625,
      "learning_rate": 0.0001,
      "loss": 4.0257,
      "loss/crossentropy": 2.0343292355537415,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19037891179323196,
      "step": 18700
    },
    {
      "epoch": 0.37404,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.014782460530598958,
      "learning_rate": 0.0001,
      "loss": 4.1727,
      "loss/crossentropy": 2.034367859363556,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20182596892118454,
      "step": 18702
    },
    {
      "epoch": 0.37408,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.015616861979166667,
      "learning_rate": 0.0001,
      "loss": 3.966,
      "loss/crossentropy": 2.048890709877014,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19922567903995514,
      "step": 18704
    },
    {
      "epoch": 0.37412,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0163970947265625,
      "learning_rate": 0.0001,
      "loss": 4.0864,
      "loss/crossentropy": 1.8759222626686096,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19452380388975143,
      "step": 18706
    },
    {
      "epoch": 0.37416,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.015681966145833334,
      "learning_rate": 0.0001,
      "loss": 3.9964,
      "loss/crossentropy": 1.693844199180603,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16944261640310287,
      "step": 18708
    },
    {
      "epoch": 0.3742,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.01727879842122396,
      "learning_rate": 0.0001,
      "loss": 4.293,
      "loss/crossentropy": 2.266264319419861,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2103852555155754,
      "step": 18710
    },
    {
      "epoch": 0.37424,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.008250935872395834,
      "learning_rate": 0.0001,
      "loss": 3.8547,
      "loss/crossentropy": 2.029367506504059,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18859465420246124,
      "step": 18712
    },
    {
      "epoch": 0.37428,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.039184315999348955,
      "learning_rate": 0.0001,
      "loss": 4.1141,
      "loss/crossentropy": 1.9427701234817505,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1960265040397644,
      "step": 18714
    },
    {
      "epoch": 0.37432,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.03737360636393229,
      "learning_rate": 0.0001,
      "loss": 4.3087,
      "loss/crossentropy": 2.2656137943267822,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22287357598543167,
      "step": 18716
    },
    {
      "epoch": 0.37436,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0365386962890625,
      "learning_rate": 0.0001,
      "loss": 4.163,
      "loss/crossentropy": 2.0447250604629517,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20505183935165405,
      "step": 18718
    },
    {
      "epoch": 0.3744,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.04125137329101562,
      "learning_rate": 0.0001,
      "loss": 3.6507,
      "loss/crossentropy": 1.8176262378692627,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17877614498138428,
      "step": 18720
    },
    {
      "epoch": 0.37444,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.04017333984375,
      "learning_rate": 0.0001,
      "loss": 3.9368,
      "loss/crossentropy": 1.8362378478050232,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18403278291225433,
      "step": 18722
    },
    {
      "epoch": 0.37448,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.04104410807291667,
      "learning_rate": 0.0001,
      "loss": 4.4322,
      "loss/crossentropy": 2.43450927734375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21623602509498596,
      "step": 18724
    },
    {
      "epoch": 0.37452,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.04133275349934896,
      "learning_rate": 0.0001,
      "loss": 4.3905,
      "loss/crossentropy": 2.227471947669983,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20000503957271576,
      "step": 18726
    },
    {
      "epoch": 0.37456,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.04317626953125,
      "learning_rate": 0.0001,
      "loss": 3.9138,
      "loss/crossentropy": 1.9392182230949402,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19422952830791473,
      "step": 18728
    },
    {
      "epoch": 0.3746,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.014469401041666666,
      "learning_rate": 0.0001,
      "loss": 4.1284,
      "loss/crossentropy": 2.0168241262435913,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1774912029504776,
      "step": 18730
    },
    {
      "epoch": 0.37464,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.012859853108723958,
      "learning_rate": 0.0001,
      "loss": 3.8982,
      "loss/crossentropy": 1.6861125230789185,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16301175951957703,
      "step": 18732
    },
    {
      "epoch": 0.37468,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.014111328125,
      "learning_rate": 0.0001,
      "loss": 4.2366,
      "loss/crossentropy": 2.2991716861724854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21094633638858795,
      "step": 18734
    },
    {
      "epoch": 0.37472,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.008115386962890625,
      "learning_rate": 0.0001,
      "loss": 4.0184,
      "loss/crossentropy": 1.8746486902236938,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19248205423355103,
      "step": 18736
    },
    {
      "epoch": 0.37476,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.009642537434895833,
      "learning_rate": 0.0001,
      "loss": 4.1886,
      "loss/crossentropy": 2.137068212032318,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1996043175458908,
      "step": 18738
    },
    {
      "epoch": 0.3748,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.015636952718098958,
      "learning_rate": 0.0001,
      "loss": 4.1503,
      "loss/crossentropy": 1.724601149559021,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23018527776002884,
      "step": 18740
    },
    {
      "epoch": 0.37484,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.015584309895833334,
      "learning_rate": 0.0001,
      "loss": 4.1593,
      "loss/crossentropy": 2.1407381296157837,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21157334744930267,
      "step": 18742
    },
    {
      "epoch": 0.37488,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.013090006510416667,
      "learning_rate": 0.0001,
      "loss": 4.0371,
      "loss/crossentropy": 2.2977999448776245,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19883693754673004,
      "step": 18744
    },
    {
      "epoch": 0.37492,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.018155924479166665,
      "learning_rate": 0.0001,
      "loss": 3.8912,
      "loss/crossentropy": 2.096716046333313,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21084215492010117,
      "step": 18746
    },
    {
      "epoch": 0.37496,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.019254302978515624,
      "learning_rate": 0.0001,
      "loss": 3.8508,
      "loss/crossentropy": 1.7292688488960266,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18280881643295288,
      "step": 18748
    },
    {
      "epoch": 0.375,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0191314697265625,
      "learning_rate": 0.0001,
      "loss": 3.9798,
      "loss/crossentropy": 1.7562988996505737,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1889612227678299,
      "step": 18750
    },
    {
      "epoch": 0.37504,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.02008641560872396,
      "learning_rate": 0.0001,
      "loss": 4.1498,
      "loss/crossentropy": 2.1730951070785522,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20177915692329407,
      "step": 18752
    },
    {
      "epoch": 0.37508,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.018161773681640625,
      "learning_rate": 0.0001,
      "loss": 4.0316,
      "loss/crossentropy": 1.7671055793762207,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17018838971853256,
      "step": 18754
    },
    {
      "epoch": 0.37512,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010228474934895834,
      "learning_rate": 0.0001,
      "loss": 4.1922,
      "loss/crossentropy": 2.2650365829467773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22090423107147217,
      "step": 18756
    },
    {
      "epoch": 0.37516,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.01002197265625,
      "learning_rate": 0.0001,
      "loss": 4.1985,
      "loss/crossentropy": 2.252619981765747,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20853855460882187,
      "step": 18758
    },
    {
      "epoch": 0.3752,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.011766560872395833,
      "learning_rate": 0.0001,
      "loss": 3.8947,
      "loss/crossentropy": 2.022092342376709,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19293325394392014,
      "step": 18760
    },
    {
      "epoch": 0.37524,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.010081990559895834,
      "learning_rate": 0.0001,
      "loss": 4.0289,
      "loss/crossentropy": 1.7500890493392944,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18832595646381378,
      "step": 18762
    },
    {
      "epoch": 0.37528,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.009220123291015625,
      "learning_rate": 0.0001,
      "loss": 3.7769,
      "loss/crossentropy": 1.8201736211776733,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1910964399576187,
      "step": 18764
    },
    {
      "epoch": 0.37532,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.009989420572916666,
      "learning_rate": 0.0001,
      "loss": 4.0508,
      "loss/crossentropy": 2.1644541025161743,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20023848116397858,
      "step": 18766
    },
    {
      "epoch": 0.37536,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.010436757405598959,
      "learning_rate": 0.0001,
      "loss": 4.163,
      "loss/crossentropy": 2.275505781173706,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23095671832561493,
      "step": 18768
    },
    {
      "epoch": 0.3754,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.010773722330729167,
      "learning_rate": 0.0001,
      "loss": 4.0557,
      "loss/crossentropy": 1.976850986480713,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2125934213399887,
      "step": 18770
    },
    {
      "epoch": 0.37544,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.012035115559895834,
      "learning_rate": 0.0001,
      "loss": 4.146,
      "loss/crossentropy": 2.150836706161499,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19192685186862946,
      "step": 18772
    },
    {
      "epoch": 0.37548,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.012400054931640625,
      "learning_rate": 0.0001,
      "loss": 4.2302,
      "loss/crossentropy": 2.0790088176727295,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2147749587893486,
      "step": 18774
    },
    {
      "epoch": 0.37552,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.010697428385416667,
      "learning_rate": 0.0001,
      "loss": 3.7918,
      "loss/crossentropy": 1.8366054892539978,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19662949442863464,
      "step": 18776
    },
    {
      "epoch": 0.37556,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008278147379557291,
      "learning_rate": 0.0001,
      "loss": 3.9056,
      "loss/crossentropy": 1.6928801536560059,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18292289972305298,
      "step": 18778
    },
    {
      "epoch": 0.3756,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.007591756184895834,
      "learning_rate": 0.0001,
      "loss": 4.2872,
      "loss/crossentropy": 2.015128195285797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20940368622541428,
      "step": 18780
    },
    {
      "epoch": 0.37564,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0061757405598958336,
      "learning_rate": 0.0001,
      "loss": 4.3304,
      "loss/crossentropy": 2.026209592819214,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2227947860956192,
      "step": 18782
    },
    {
      "epoch": 0.37568,
      "grad_norm": 2.0,
      "grad_norm_var": 0.004349772135416667,
      "learning_rate": 0.0001,
      "loss": 4.0591,
      "loss/crossentropy": 2.29964280128479,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23132510483264923,
      "step": 18784
    },
    {
      "epoch": 0.37572,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.005366770426432291,
      "learning_rate": 0.0001,
      "loss": 3.9675,
      "loss/crossentropy": 1.881381332874298,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17967566847801208,
      "step": 18786
    },
    {
      "epoch": 0.37576,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.004198201497395833,
      "learning_rate": 0.0001,
      "loss": 4.1834,
      "loss/crossentropy": 2.2231001257896423,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21163878589868546,
      "step": 18788
    },
    {
      "epoch": 0.3758,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.004526519775390625,
      "learning_rate": 0.0001,
      "loss": 4.0475,
      "loss/crossentropy": 2.3531078100204468,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21745989471673965,
      "step": 18790
    },
    {
      "epoch": 0.37584,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.004801432291666667,
      "learning_rate": 0.0001,
      "loss": 4.2549,
      "loss/crossentropy": 2.1178460121154785,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23571017384529114,
      "step": 18792
    },
    {
      "epoch": 0.37588,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.005890909830729167,
      "learning_rate": 0.0001,
      "loss": 3.8373,
      "loss/crossentropy": 2.0659135580062866,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20817245543003082,
      "step": 18794
    },
    {
      "epoch": 0.37592,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.006318918863932292,
      "learning_rate": 0.0001,
      "loss": 4.0103,
      "loss/crossentropy": 1.579395353794098,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16713083535432816,
      "step": 18796
    },
    {
      "epoch": 0.37596,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.005940500895182292,
      "learning_rate": 0.0001,
      "loss": 4.1272,
      "loss/crossentropy": 2.102541923522949,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2174636349081993,
      "step": 18798
    },
    {
      "epoch": 0.376,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0067942301432291664,
      "learning_rate": 0.0001,
      "loss": 4.2622,
      "loss/crossentropy": 1.9342190027236938,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19780533015727997,
      "step": 18800
    },
    {
      "epoch": 0.37604,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.006121571858723958,
      "learning_rate": 0.0001,
      "loss": 4.0659,
      "loss/crossentropy": 2.140671730041504,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21276966482400894,
      "step": 18802
    },
    {
      "epoch": 0.37608,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.009590657552083333,
      "learning_rate": 0.0001,
      "loss": 4.1929,
      "loss/crossentropy": 1.9004405736923218,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17683426290750504,
      "step": 18804
    },
    {
      "epoch": 0.37612,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.010155232747395833,
      "learning_rate": 0.0001,
      "loss": 3.7163,
      "loss/crossentropy": 1.9487649202346802,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19463464617729187,
      "step": 18806
    },
    {
      "epoch": 0.37616,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.011262766520182292,
      "learning_rate": 0.0001,
      "loss": 3.9517,
      "loss/crossentropy": 2.15252423286438,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20531394332647324,
      "step": 18808
    },
    {
      "epoch": 0.3762,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.011592356363932292,
      "learning_rate": 0.0001,
      "loss": 4.0494,
      "loss/crossentropy": 2.114433467388153,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20183662325143814,
      "step": 18810
    },
    {
      "epoch": 0.37624,
      "grad_norm": 2.0,
      "grad_norm_var": 0.011205037434895834,
      "learning_rate": 0.0001,
      "loss": 3.9035,
      "loss/crossentropy": 1.9229055047035217,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18662934005260468,
      "step": 18812
    },
    {
      "epoch": 0.37628,
      "grad_norm": 2.046875,
      "grad_norm_var": 1.8609944661458333,
      "learning_rate": 0.0001,
      "loss": 4.2516,
      "loss/crossentropy": 1.8242689371109009,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17858239263296127,
      "step": 18814
    },
    {
      "epoch": 0.37632,
      "grad_norm": 1.9765625,
      "grad_norm_var": 1.864818318684896,
      "learning_rate": 0.0001,
      "loss": 3.8917,
      "loss/crossentropy": 2.3987890481948853,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2246004194021225,
      "step": 18816
    },
    {
      "epoch": 0.37636,
      "grad_norm": 2.234375,
      "grad_norm_var": 1.8565500895182292,
      "learning_rate": 0.0001,
      "loss": 4.3493,
      "loss/crossentropy": 2.4376614093780518,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24541915208101273,
      "step": 18818
    },
    {
      "epoch": 0.3764,
      "grad_norm": 1.984375,
      "grad_norm_var": 1.865612538655599,
      "learning_rate": 0.0001,
      "loss": 4.1344,
      "loss/crossentropy": 2.2477601766586304,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2178046852350235,
      "step": 18820
    },
    {
      "epoch": 0.37644,
      "grad_norm": 1.96875,
      "grad_norm_var": 1.8636464436848958,
      "learning_rate": 0.0001,
      "loss": 3.9685,
      "loss/crossentropy": 1.9782747626304626,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19459142535924911,
      "step": 18822
    },
    {
      "epoch": 0.37648,
      "grad_norm": 2.0625,
      "grad_norm_var": 1.8451738993326823,
      "learning_rate": 0.0001,
      "loss": 4.043,
      "loss/crossentropy": 2.1214100122451782,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2082664743065834,
      "step": 18824
    },
    {
      "epoch": 0.37652,
      "grad_norm": 1.90625,
      "grad_norm_var": 1.8489461263020834,
      "learning_rate": 0.0001,
      "loss": 3.8256,
      "loss/crossentropy": 1.68446546792984,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.167055182158947,
      "step": 18826
    },
    {
      "epoch": 0.37656,
      "grad_norm": 1.9921875,
      "grad_norm_var": 1.8380022684733073,
      "learning_rate": 0.0001,
      "loss": 4.0588,
      "loss/crossentropy": 1.7720499634742737,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18632127344608307,
      "step": 18828
    },
    {
      "epoch": 0.3766,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.013492838541666666,
      "learning_rate": 0.0001,
      "loss": 4.2288,
      "loss/crossentropy": 2.283332347869873,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23949767649173737,
      "step": 18830
    },
    {
      "epoch": 0.37664,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.014644114176432292,
      "learning_rate": 0.0001,
      "loss": 3.9134,
      "loss/crossentropy": 1.9924857020378113,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20181651413440704,
      "step": 18832
    },
    {
      "epoch": 0.37668,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.011091105143229167,
      "learning_rate": 0.0001,
      "loss": 3.8424,
      "loss/crossentropy": 2.4015761613845825,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2279737964272499,
      "step": 18834
    },
    {
      "epoch": 0.37672,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.014152018229166667,
      "learning_rate": 0.0001,
      "loss": 3.8636,
      "loss/crossentropy": 2.0717111229896545,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19089852273464203,
      "step": 18836
    },
    {
      "epoch": 0.37676,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0149566650390625,
      "learning_rate": 0.0001,
      "loss": 4.2694,
      "loss/crossentropy": 1.9993728995323181,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19489262998104095,
      "step": 18838
    },
    {
      "epoch": 0.3768,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.015949503580729166,
      "learning_rate": 0.0001,
      "loss": 3.7009,
      "loss/crossentropy": 1.915448248386383,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1923104077577591,
      "step": 18840
    },
    {
      "epoch": 0.37684,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.016511027018229166,
      "learning_rate": 0.0001,
      "loss": 4.0424,
      "loss/crossentropy": 2.292284607887268,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22159253805875778,
      "step": 18842
    },
    {
      "epoch": 0.37688,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.012784830729166667,
      "learning_rate": 0.0001,
      "loss": 3.9904,
      "loss/crossentropy": 1.9938938617706299,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19937817752361298,
      "step": 18844
    },
    {
      "epoch": 0.37692,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.016454060872395832,
      "learning_rate": 0.0001,
      "loss": 4.1914,
      "loss/crossentropy": 2.045258641242981,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2652823179960251,
      "step": 18846
    },
    {
      "epoch": 0.37696,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.016707356770833334,
      "learning_rate": 0.0001,
      "loss": 4.0391,
      "loss/crossentropy": 2.1330259442329407,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19155749678611755,
      "step": 18848
    },
    {
      "epoch": 0.377,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.017101796468098958,
      "learning_rate": 0.0001,
      "loss": 4.1422,
      "loss/crossentropy": 2.2135089635849,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20062117278575897,
      "step": 18850
    },
    {
      "epoch": 0.37704,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.013826243082682292,
      "learning_rate": 0.0001,
      "loss": 4.1188,
      "loss/crossentropy": 2.073060691356659,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19443543255329132,
      "step": 18852
    },
    {
      "epoch": 0.37708,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.014170074462890625,
      "learning_rate": 0.0001,
      "loss": 4.089,
      "loss/crossentropy": 1.930641233921051,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19604724645614624,
      "step": 18854
    },
    {
      "epoch": 0.37712,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011252593994140626,
      "learning_rate": 0.0001,
      "loss": 4.0238,
      "loss/crossentropy": 2.0813130140304565,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21019181609153748,
      "step": 18856
    },
    {
      "epoch": 0.37716,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.010080718994140625,
      "learning_rate": 0.0001,
      "loss": 3.9855,
      "loss/crossentropy": 1.8528355956077576,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19945576786994934,
      "step": 18858
    },
    {
      "epoch": 0.3772,
      "grad_norm": 2.0,
      "grad_norm_var": 0.011896769205729166,
      "learning_rate": 0.0001,
      "loss": 3.9655,
      "loss/crossentropy": 2.32794725894928,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22688604146242142,
      "step": 18860
    },
    {
      "epoch": 0.37724,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.007264963785807292,
      "learning_rate": 0.0001,
      "loss": 4.25,
      "loss/crossentropy": 2.061101734638214,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1961553767323494,
      "step": 18862
    },
    {
      "epoch": 0.37728,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.00604248046875,
      "learning_rate": 0.0001,
      "loss": 3.7839,
      "loss/crossentropy": 1.9127929210662842,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20082338899374008,
      "step": 18864
    },
    {
      "epoch": 0.37732,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.14662272135416668,
      "learning_rate": 0.0001,
      "loss": 4.289,
      "loss/crossentropy": 2.001932919025421,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1881283074617386,
      "step": 18866
    },
    {
      "epoch": 0.37736,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.14527180989583333,
      "learning_rate": 0.0001,
      "loss": 4.0851,
      "loss/crossentropy": 1.693075716495514,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18698371946811676,
      "step": 18868
    },
    {
      "epoch": 0.3774,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.1481402079264323,
      "learning_rate": 0.0001,
      "loss": 4.0484,
      "loss/crossentropy": 1.879252314567566,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18412816524505615,
      "step": 18870
    },
    {
      "epoch": 0.37744,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.1550066630045573,
      "learning_rate": 0.0001,
      "loss": 4.3488,
      "loss/crossentropy": 2.1819299459457397,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20243355631828308,
      "step": 18872
    },
    {
      "epoch": 0.37748,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.1576568603515625,
      "learning_rate": 0.0001,
      "loss": 4.0889,
      "loss/crossentropy": 2.2487794160842896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20594948530197144,
      "step": 18874
    },
    {
      "epoch": 0.37752,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.15466079711914063,
      "learning_rate": 0.0001,
      "loss": 4.303,
      "loss/crossentropy": 1.990889549255371,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19230055809020996,
      "step": 18876
    },
    {
      "epoch": 0.37756,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.1582763671875,
      "learning_rate": 0.0001,
      "loss": 4.0435,
      "loss/crossentropy": 2.091560959815979,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19504332542419434,
      "step": 18878
    },
    {
      "epoch": 0.3776,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.1599273681640625,
      "learning_rate": 0.0001,
      "loss": 4.0776,
      "loss/crossentropy": 1.9213955998420715,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1994621828198433,
      "step": 18880
    },
    {
      "epoch": 0.37764,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.02360814412434896,
      "learning_rate": 0.0001,
      "loss": 4.0103,
      "loss/crossentropy": 2.1728278398513794,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21039214730262756,
      "step": 18882
    },
    {
      "epoch": 0.37768,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.026341756184895832,
      "learning_rate": 0.0001,
      "loss": 4.052,
      "loss/crossentropy": 1.889222264289856,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17232363671064377,
      "step": 18884
    },
    {
      "epoch": 0.37772,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.026228841145833334,
      "learning_rate": 0.0001,
      "loss": 4.118,
      "loss/crossentropy": 2.264981746673584,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.213841512799263,
      "step": 18886
    },
    {
      "epoch": 0.37776,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.013598378499348958,
      "learning_rate": 0.0001,
      "loss": 3.8843,
      "loss/crossentropy": 1.8615361452102661,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1745932251214981,
      "step": 18888
    },
    {
      "epoch": 0.3778,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.009266916910807292,
      "learning_rate": 0.0001,
      "loss": 3.5687,
      "loss/crossentropy": 2.1459723711013794,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21431762725114822,
      "step": 18890
    },
    {
      "epoch": 0.37784,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.010479482014973958,
      "learning_rate": 0.0001,
      "loss": 4.2125,
      "loss/crossentropy": 1.8859283328056335,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20627497136592865,
      "step": 18892
    },
    {
      "epoch": 0.37788,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.012211100260416666,
      "learning_rate": 0.0001,
      "loss": 3.9978,
      "loss/crossentropy": 1.824287474155426,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17877592146396637,
      "step": 18894
    },
    {
      "epoch": 0.37792,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.012414296468098959,
      "learning_rate": 0.0001,
      "loss": 3.9857,
      "loss/crossentropy": 2.486180305480957,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2033834606409073,
      "step": 18896
    },
    {
      "epoch": 0.37796,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.012981923421223958,
      "learning_rate": 0.0001,
      "loss": 4.2339,
      "loss/crossentropy": 2.0677965879440308,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21111362427473068,
      "step": 18898
    },
    {
      "epoch": 0.378,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.011107381184895833,
      "learning_rate": 0.0001,
      "loss": 4.2559,
      "loss/crossentropy": 1.941792368888855,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18762809038162231,
      "step": 18900
    },
    {
      "epoch": 0.37804,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.012504069010416667,
      "learning_rate": 0.0001,
      "loss": 3.8028,
      "loss/crossentropy": 1.907107174396515,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2050766795873642,
      "step": 18902
    },
    {
      "epoch": 0.37808,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.009242502848307292,
      "learning_rate": 0.0001,
      "loss": 3.9742,
      "loss/crossentropy": 1.5920222997665405,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17366845160722733,
      "step": 18904
    },
    {
      "epoch": 0.37812,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0066650390625,
      "learning_rate": 0.0001,
      "loss": 3.9581,
      "loss/crossentropy": 2.054854154586792,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20102889090776443,
      "step": 18906
    },
    {
      "epoch": 0.37816,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0067789713541666664,
      "learning_rate": 0.0001,
      "loss": 4.2582,
      "loss/crossentropy": 2.42835795879364,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2154095396399498,
      "step": 18908
    },
    {
      "epoch": 0.3782,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.007169596354166667,
      "learning_rate": 0.0001,
      "loss": 4.1466,
      "loss/crossentropy": 2.1060370206832886,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20842333883047104,
      "step": 18910
    },
    {
      "epoch": 0.37824,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.008194986979166667,
      "learning_rate": 0.0001,
      "loss": 4.1587,
      "loss/crossentropy": 2.0559674501419067,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2195875570178032,
      "step": 18912
    },
    {
      "epoch": 0.37828,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008888498942057291,
      "learning_rate": 0.0001,
      "loss": 4.2786,
      "loss/crossentropy": 2.240954041481018,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20908969640731812,
      "step": 18914
    },
    {
      "epoch": 0.37832,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.008747355143229166,
      "learning_rate": 0.0001,
      "loss": 4.0729,
      "loss/crossentropy": 2.0393139123916626,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19933026283979416,
      "step": 18916
    },
    {
      "epoch": 0.37836,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.007682291666666666,
      "learning_rate": 0.0001,
      "loss": 4.0969,
      "loss/crossentropy": 2.0294516682624817,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20086780190467834,
      "step": 18918
    },
    {
      "epoch": 0.3784,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.007425944010416667,
      "learning_rate": 0.0001,
      "loss": 3.9138,
      "loss/crossentropy": 1.8509765267372131,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18834587186574936,
      "step": 18920
    },
    {
      "epoch": 0.37844,
      "grad_norm": 1.875,
      "grad_norm_var": 0.0068267822265625,
      "learning_rate": 0.0001,
      "loss": 3.8669,
      "loss/crossentropy": 1.8817378878593445,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17679119110107422,
      "step": 18922
    },
    {
      "epoch": 0.37848,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.005975087483723958,
      "learning_rate": 0.0001,
      "loss": 3.9276,
      "loss/crossentropy": 1.766166627407074,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16948368400335312,
      "step": 18924
    },
    {
      "epoch": 0.37852,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.006786855061848959,
      "learning_rate": 0.0001,
      "loss": 3.8948,
      "loss/crossentropy": 2.251446485519409,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22272750735282898,
      "step": 18926
    },
    {
      "epoch": 0.37856,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0045125325520833336,
      "learning_rate": 0.0001,
      "loss": 4.1119,
      "loss/crossentropy": 2.0006097555160522,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21003134548664093,
      "step": 18928
    },
    {
      "epoch": 0.3786,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0036936442057291666,
      "learning_rate": 0.0001,
      "loss": 3.7731,
      "loss/crossentropy": 2.2906641960144043,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23650038242340088,
      "step": 18930
    },
    {
      "epoch": 0.37864,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0039866129557291664,
      "learning_rate": 0.0001,
      "loss": 4.1848,
      "loss/crossentropy": 2.185898005962372,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21387950330972672,
      "step": 18932
    },
    {
      "epoch": 0.37868,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0028928120930989585,
      "learning_rate": 0.0001,
      "loss": 4.1808,
      "loss/crossentropy": 2.2457324266433716,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21054718643426895,
      "step": 18934
    },
    {
      "epoch": 0.37872,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0032793680826822915,
      "learning_rate": 0.0001,
      "loss": 4.2711,
      "loss/crossentropy": 2.1835416555404663,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23258651047945023,
      "step": 18936
    },
    {
      "epoch": 0.37876,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0029436747233072915,
      "learning_rate": 0.0001,
      "loss": 4.1555,
      "loss/crossentropy": 2.1908507347106934,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20261384546756744,
      "step": 18938
    },
    {
      "epoch": 0.3788,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0035336812337239585,
      "learning_rate": 0.0001,
      "loss": 4.1359,
      "loss/crossentropy": 2.276697278022766,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20843130350112915,
      "step": 18940
    },
    {
      "epoch": 0.37884,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0023251851399739582,
      "learning_rate": 0.0001,
      "loss": 4.1309,
      "loss/crossentropy": 2.16153222322464,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19693513214588165,
      "step": 18942
    },
    {
      "epoch": 0.37888,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0023590087890625,
      "learning_rate": 0.0001,
      "loss": 3.9822,
      "loss/crossentropy": 2.293270707130432,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21085364371538162,
      "step": 18944
    },
    {
      "epoch": 0.37892,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0025042215983072918,
      "learning_rate": 0.0001,
      "loss": 4.1291,
      "loss/crossentropy": 2.0382518768310547,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21567383408546448,
      "step": 18946
    },
    {
      "epoch": 0.37896,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.00894775390625,
      "learning_rate": 0.0001,
      "loss": 3.9993,
      "loss/crossentropy": 2.038296341896057,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19963379949331284,
      "step": 18948
    },
    {
      "epoch": 0.379,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.00994873046875,
      "learning_rate": 0.0001,
      "loss": 4.2363,
      "loss/crossentropy": 2.1765074729919434,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19551265239715576,
      "step": 18950
    },
    {
      "epoch": 0.37904,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010591379801432292,
      "learning_rate": 0.0001,
      "loss": 3.8641,
      "loss/crossentropy": 1.870033621788025,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17858022451400757,
      "step": 18952
    },
    {
      "epoch": 0.37908,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.011083730061848958,
      "learning_rate": 0.0001,
      "loss": 4.1877,
      "loss/crossentropy": 1.8716632723808289,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20534101128578186,
      "step": 18954
    },
    {
      "epoch": 0.37912,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0111236572265625,
      "learning_rate": 0.0001,
      "loss": 4.1662,
      "loss/crossentropy": 1.925970435142517,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19596435129642487,
      "step": 18956
    },
    {
      "epoch": 0.37916,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.011800130208333334,
      "learning_rate": 0.0001,
      "loss": 4.126,
      "loss/crossentropy": 2.2280589938163757,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21037188172340393,
      "step": 18958
    },
    {
      "epoch": 0.3792,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0150299072265625,
      "learning_rate": 0.0001,
      "loss": 3.7683,
      "loss/crossentropy": 1.8350458145141602,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1821446716785431,
      "step": 18960
    },
    {
      "epoch": 0.37924,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.015018463134765625,
      "learning_rate": 0.0001,
      "loss": 4.0376,
      "loss/crossentropy": 2.0393940210342407,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18378175050020218,
      "step": 18962
    },
    {
      "epoch": 0.37928,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.00897216796875,
      "learning_rate": 0.0001,
      "loss": 4.0018,
      "loss/crossentropy": 1.7470228672027588,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19222351908683777,
      "step": 18964
    },
    {
      "epoch": 0.37932,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.011742146809895833,
      "learning_rate": 0.0001,
      "loss": 4.1791,
      "loss/crossentropy": 2.3943980932235718,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23214909434318542,
      "step": 18966
    },
    {
      "epoch": 0.37936,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.012239329020182292,
      "learning_rate": 0.0001,
      "loss": 3.7877,
      "loss/crossentropy": 1.8153263330459595,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19808300584554672,
      "step": 18968
    },
    {
      "epoch": 0.3794,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.011171213785807292,
      "learning_rate": 0.0001,
      "loss": 4.0122,
      "loss/crossentropy": 2.0037755370140076,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20173701643943787,
      "step": 18970
    },
    {
      "epoch": 0.37944,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.012168121337890626,
      "learning_rate": 0.0001,
      "loss": 4.1069,
      "loss/crossentropy": 2.144823908805847,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21421342343091965,
      "step": 18972
    },
    {
      "epoch": 0.37948,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011565907796223959,
      "learning_rate": 0.0001,
      "loss": 4.2788,
      "loss/crossentropy": 2.2151081562042236,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20057211816310883,
      "step": 18974
    },
    {
      "epoch": 0.37952,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008976236979166666,
      "learning_rate": 0.0001,
      "loss": 4.5061,
      "loss/crossentropy": 2.341770827770233,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21535535156726837,
      "step": 18976
    },
    {
      "epoch": 0.37956,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.009520467122395833,
      "learning_rate": 0.0001,
      "loss": 4.1399,
      "loss/crossentropy": 1.8952747583389282,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17530933022499084,
      "step": 18978
    },
    {
      "epoch": 0.3796,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008885701497395834,
      "learning_rate": 0.0001,
      "loss": 4.141,
      "loss/crossentropy": 2.2774226665496826,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20682721585035324,
      "step": 18980
    },
    {
      "epoch": 0.37964,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.008508046468098959,
      "learning_rate": 0.0001,
      "loss": 3.982,
      "loss/crossentropy": 1.9596800208091736,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18774595111608505,
      "step": 18982
    },
    {
      "epoch": 0.37968,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.009309895833333333,
      "learning_rate": 0.0001,
      "loss": 4.008,
      "loss/crossentropy": 2.1485220193862915,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19221040606498718,
      "step": 18984
    },
    {
      "epoch": 0.37972,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.009039052327473958,
      "learning_rate": 0.0001,
      "loss": 4.142,
      "loss/crossentropy": 2.2140207290649414,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2012096345424652,
      "step": 18986
    },
    {
      "epoch": 0.37976,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.008063761393229167,
      "learning_rate": 0.0001,
      "loss": 4.1088,
      "loss/crossentropy": 1.9045360684394836,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19290773570537567,
      "step": 18988
    },
    {
      "epoch": 0.3798,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008072662353515624,
      "learning_rate": 0.0001,
      "loss": 4.0349,
      "loss/crossentropy": 1.8351407051086426,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18892163783311844,
      "step": 18990
    },
    {
      "epoch": 0.37984,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.007865397135416667,
      "learning_rate": 0.0001,
      "loss": 3.9136,
      "loss/crossentropy": 2.021119713783264,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19214972108602524,
      "step": 18992
    },
    {
      "epoch": 0.37988,
      "grad_norm": 2.0,
      "grad_norm_var": 0.006394195556640625,
      "learning_rate": 0.0001,
      "loss": 4.2961,
      "loss/crossentropy": 2.0591527223587036,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20295391231775284,
      "step": 18994
    },
    {
      "epoch": 0.37992,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.006566365559895833,
      "learning_rate": 0.0001,
      "loss": 4.1135,
      "loss/crossentropy": 1.9794987440109253,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20964892208576202,
      "step": 18996
    },
    {
      "epoch": 0.37996,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0042111714680989586,
      "learning_rate": 0.0001,
      "loss": 4.0699,
      "loss/crossentropy": 2.116178512573242,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1962205320596695,
      "step": 18998
    },
    {
      "epoch": 0.38,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.003714752197265625,
      "learning_rate": 0.0001,
      "loss": 4.0683,
      "loss/crossentropy": 2.2458595037460327,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1981743946671486,
      "step": 19000
    },
    {
      "epoch": 0.38004,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0038266499837239583,
      "learning_rate": 0.0001,
      "loss": 4.2085,
      "loss/crossentropy": 2.0612659454345703,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20526322722434998,
      "step": 19002
    },
    {
      "epoch": 0.38008,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.00347900390625,
      "learning_rate": 0.0001,
      "loss": 4.191,
      "loss/crossentropy": 2.0810243487358093,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.222077377140522,
      "step": 19004
    },
    {
      "epoch": 0.38012,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.003940582275390625,
      "learning_rate": 0.0001,
      "loss": 4.0957,
      "loss/crossentropy": 2.2651617527008057,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2106148600578308,
      "step": 19006
    },
    {
      "epoch": 0.38016,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.0056793212890625,
      "learning_rate": 0.0001,
      "loss": 3.9671,
      "loss/crossentropy": 1.9701108932495117,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18778277933597565,
      "step": 19008
    },
    {
      "epoch": 0.3802,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.007024892171223958,
      "learning_rate": 0.0001,
      "loss": 4.1265,
      "loss/crossentropy": 1.9822896122932434,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25280050933361053,
      "step": 19010
    },
    {
      "epoch": 0.38024,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0063435872395833336,
      "learning_rate": 0.0001,
      "loss": 3.9577,
      "loss/crossentropy": 2.020545542240143,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19008295983076096,
      "step": 19012
    },
    {
      "epoch": 0.38028,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.007336171468098959,
      "learning_rate": 0.0001,
      "loss": 4.2044,
      "loss/crossentropy": 2.0922394394874573,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1972956582903862,
      "step": 19014
    },
    {
      "epoch": 0.38032,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.007067616780598958,
      "learning_rate": 0.0001,
      "loss": 3.9726,
      "loss/crossentropy": 2.028052031993866,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18549586832523346,
      "step": 19016
    },
    {
      "epoch": 0.38036,
      "grad_norm": 1.6796875,
      "grad_norm_var": 0.011395009358723958,
      "learning_rate": 0.0001,
      "loss": 4.0118,
      "loss/crossentropy": 1.9518752098083496,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17747169733047485,
      "step": 19018
    },
    {
      "epoch": 0.3804,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.013492838541666666,
      "learning_rate": 0.0001,
      "loss": 4.1404,
      "loss/crossentropy": 1.8896766901016235,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20509368181228638,
      "step": 19020
    },
    {
      "epoch": 0.38044,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.01407470703125,
      "learning_rate": 0.0001,
      "loss": 4.1159,
      "loss/crossentropy": 2.10469388961792,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2058955579996109,
      "step": 19022
    },
    {
      "epoch": 0.38048,
      "grad_norm": 1.875,
      "grad_norm_var": 0.012717437744140626,
      "learning_rate": 0.0001,
      "loss": 3.8601,
      "loss/crossentropy": 1.7341394424438477,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19358114898204803,
      "step": 19024
    },
    {
      "epoch": 0.38052,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.013634999593098959,
      "learning_rate": 0.0001,
      "loss": 3.8528,
      "loss/crossentropy": 2.186724543571472,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1981809437274933,
      "step": 19026
    },
    {
      "epoch": 0.38056,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.014070638020833333,
      "learning_rate": 0.0001,
      "loss": 4.2421,
      "loss/crossentropy": 2.228869318962097,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20712755620479584,
      "step": 19028
    },
    {
      "epoch": 0.3806,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0140869140625,
      "learning_rate": 0.0001,
      "loss": 4.126,
      "loss/crossentropy": 2.2783373594284058,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20916558057069778,
      "step": 19030
    },
    {
      "epoch": 0.38064,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.013270823160807292,
      "learning_rate": 0.0001,
      "loss": 4.2744,
      "loss/crossentropy": 2.2671823501586914,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19389723241329193,
      "step": 19032
    },
    {
      "epoch": 0.38068,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.008540852864583334,
      "learning_rate": 0.0001,
      "loss": 4.0267,
      "loss/crossentropy": 1.948801040649414,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18766260892152786,
      "step": 19034
    },
    {
      "epoch": 0.38072,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.007818349202473958,
      "learning_rate": 0.0001,
      "loss": 4.0856,
      "loss/crossentropy": 1.895260751247406,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1848379746079445,
      "step": 19036
    },
    {
      "epoch": 0.38076,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.008504231770833334,
      "learning_rate": 0.0001,
      "loss": 3.8189,
      "loss/crossentropy": 1.8295226097106934,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1816689372062683,
      "step": 19038
    },
    {
      "epoch": 0.3808,
      "grad_norm": 2.0,
      "grad_norm_var": 0.007830556233723958,
      "learning_rate": 0.0001,
      "loss": 4.0363,
      "loss/crossentropy": 1.6487661004066467,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.178475059568882,
      "step": 19040
    },
    {
      "epoch": 0.38084,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0058095296223958336,
      "learning_rate": 0.0001,
      "loss": 3.828,
      "loss/crossentropy": 1.9264054894447327,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2009180784225464,
      "step": 19042
    },
    {
      "epoch": 0.38088,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.007989247639973959,
      "learning_rate": 0.0001,
      "loss": 4.2636,
      "loss/crossentropy": 2.3018531799316406,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2039026990532875,
      "step": 19044
    },
    {
      "epoch": 0.38092,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.009093983968098959,
      "learning_rate": 0.0001,
      "loss": 4.0201,
      "loss/crossentropy": 2.135006010532379,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.172908216714859,
      "step": 19046
    },
    {
      "epoch": 0.38096,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.009476725260416667,
      "learning_rate": 0.0001,
      "loss": 4.2064,
      "loss/crossentropy": 2.1111900806427,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19583145529031754,
      "step": 19048
    },
    {
      "epoch": 0.381,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0096099853515625,
      "learning_rate": 0.0001,
      "loss": 4.1165,
      "loss/crossentropy": 2.208159327507019,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21107421815395355,
      "step": 19050
    },
    {
      "epoch": 0.38104,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.012894694010416667,
      "learning_rate": 0.0001,
      "loss": 4.1722,
      "loss/crossentropy": 2.144998788833618,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2108323574066162,
      "step": 19052
    },
    {
      "epoch": 0.38108,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0116455078125,
      "learning_rate": 0.0001,
      "loss": 4.0938,
      "loss/crossentropy": 2.0046940445899963,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20454590767621994,
      "step": 19054
    },
    {
      "epoch": 0.38112,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.010992177327473958,
      "learning_rate": 0.0001,
      "loss": 4.2219,
      "loss/crossentropy": 2.042281448841095,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21530431509017944,
      "step": 19056
    },
    {
      "epoch": 0.38116,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0102691650390625,
      "learning_rate": 0.0001,
      "loss": 4.2701,
      "loss/crossentropy": 2.0579177141189575,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20769091695547104,
      "step": 19058
    },
    {
      "epoch": 0.3812,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.012719472249348959,
      "learning_rate": 0.0001,
      "loss": 4.0561,
      "loss/crossentropy": 1.8340198993682861,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1927536353468895,
      "step": 19060
    },
    {
      "epoch": 0.38124,
      "grad_norm": 2.375,
      "grad_norm_var": 0.016584269205729165,
      "learning_rate": 0.0001,
      "loss": 4.0412,
      "loss/crossentropy": 2.146699070930481,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18840950727462769,
      "step": 19062
    },
    {
      "epoch": 0.38128,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.017600250244140626,
      "learning_rate": 0.0001,
      "loss": 3.7448,
      "loss/crossentropy": 2.0206886529922485,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20354034006595612,
      "step": 19064
    },
    {
      "epoch": 0.38132,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.023372141520182292,
      "learning_rate": 0.0001,
      "loss": 4.0047,
      "loss/crossentropy": 1.9568690061569214,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.183781698346138,
      "step": 19066
    },
    {
      "epoch": 0.38136,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.022855631510416665,
      "learning_rate": 0.0001,
      "loss": 4.0898,
      "loss/crossentropy": 2.092573404312134,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20443417131900787,
      "step": 19068
    },
    {
      "epoch": 0.3814,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.025770823160807293,
      "learning_rate": 0.0001,
      "loss": 4.2253,
      "loss/crossentropy": 2.17366099357605,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2064819559454918,
      "step": 19070
    },
    {
      "epoch": 0.38144,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0265777587890625,
      "learning_rate": 0.0001,
      "loss": 4.0398,
      "loss/crossentropy": 1.7935467958450317,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19256697595119476,
      "step": 19072
    },
    {
      "epoch": 0.38148,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.026554107666015625,
      "learning_rate": 0.0001,
      "loss": 3.9846,
      "loss/crossentropy": 1.995141625404358,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19474036246538162,
      "step": 19074
    },
    {
      "epoch": 0.38152,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.023221588134765624,
      "learning_rate": 0.0001,
      "loss": 3.9058,
      "loss/crossentropy": 2.140002131462097,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19610393047332764,
      "step": 19076
    },
    {
      "epoch": 0.38156,
      "grad_norm": 2.0,
      "grad_norm_var": 0.013288370768229167,
      "learning_rate": 0.0001,
      "loss": 4.1176,
      "loss/crossentropy": 2.1740564107894897,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20218566060066223,
      "step": 19078
    },
    {
      "epoch": 0.3816,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.015925089518229168,
      "learning_rate": 0.0001,
      "loss": 4.3043,
      "loss/crossentropy": 2.065530776977539,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21683599054813385,
      "step": 19080
    },
    {
      "epoch": 0.38164,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0127349853515625,
      "learning_rate": 0.0001,
      "loss": 4.2859,
      "loss/crossentropy": 1.844248354434967,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19479060918092728,
      "step": 19082
    },
    {
      "epoch": 0.38168,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.011136627197265625,
      "learning_rate": 0.0001,
      "loss": 3.886,
      "loss/crossentropy": 1.8596556186676025,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2059667930006981,
      "step": 19084
    },
    {
      "epoch": 0.38172,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.014731597900390626,
      "learning_rate": 0.0001,
      "loss": 4.0982,
      "loss/crossentropy": 2.08061683177948,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20623548328876495,
      "step": 19086
    },
    {
      "epoch": 0.38176,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.014615631103515625,
      "learning_rate": 0.0001,
      "loss": 4.0013,
      "loss/crossentropy": 1.888766884803772,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18000122904777527,
      "step": 19088
    },
    {
      "epoch": 0.3818,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.017032877604166666,
      "learning_rate": 0.0001,
      "loss": 3.8784,
      "loss/crossentropy": 2.021374225616455,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21468255668878555,
      "step": 19090
    },
    {
      "epoch": 0.38184,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.016805013020833332,
      "learning_rate": 0.0001,
      "loss": 3.8327,
      "loss/crossentropy": 2.083697557449341,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21188172698020935,
      "step": 19092
    },
    {
      "epoch": 0.38188,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.02019017537434896,
      "learning_rate": 0.0001,
      "loss": 3.9617,
      "loss/crossentropy": 2.1484656929969788,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21978207677602768,
      "step": 19094
    },
    {
      "epoch": 0.38192,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.0218414306640625,
      "learning_rate": 0.0001,
      "loss": 4.1728,
      "loss/crossentropy": 2.321221709251404,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22688252478837967,
      "step": 19096
    },
    {
      "epoch": 0.38196,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.0518218994140625,
      "learning_rate": 0.0001,
      "loss": 4.157,
      "loss/crossentropy": 2.0623167753219604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2148035168647766,
      "step": 19098
    },
    {
      "epoch": 0.382,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.05501708984375,
      "learning_rate": 0.0001,
      "loss": 4.0255,
      "loss/crossentropy": 1.8845015168190002,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1880025863647461,
      "step": 19100
    },
    {
      "epoch": 0.38204,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.05045750935872396,
      "learning_rate": 0.0001,
      "loss": 3.8361,
      "loss/crossentropy": 1.927463173866272,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17446774244308472,
      "step": 19102
    },
    {
      "epoch": 0.38208,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0536529541015625,
      "learning_rate": 0.0001,
      "loss": 4.2909,
      "loss/crossentropy": 2.6780699491500854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22951193153858185,
      "step": 19104
    },
    {
      "epoch": 0.38212,
      "grad_norm": 1.875,
      "grad_norm_var": 0.05358250935872396,
      "learning_rate": 0.0001,
      "loss": 3.9188,
      "loss/crossentropy": 2.2530760765075684,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2141779437661171,
      "step": 19106
    },
    {
      "epoch": 0.38216,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.07016499837239583,
      "learning_rate": 0.0001,
      "loss": 4.1647,
      "loss/crossentropy": 2.0292217135429382,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20607471466064453,
      "step": 19108
    },
    {
      "epoch": 0.3822,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.07394790649414062,
      "learning_rate": 0.0001,
      "loss": 3.8144,
      "loss/crossentropy": 1.8341345191001892,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18437029421329498,
      "step": 19110
    },
    {
      "epoch": 0.38224,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.06982421875,
      "learning_rate": 0.0001,
      "loss": 4.1782,
      "loss/crossentropy": 1.9262341260910034,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1894589066505432,
      "step": 19112
    },
    {
      "epoch": 0.38228,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.04212824503580729,
      "learning_rate": 0.0001,
      "loss": 4.2473,
      "loss/crossentropy": 1.9369717240333557,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17737293988466263,
      "step": 19114
    },
    {
      "epoch": 0.38232,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.03845621744791667,
      "learning_rate": 0.0001,
      "loss": 4.0904,
      "loss/crossentropy": 1.7547513842582703,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1895698457956314,
      "step": 19116
    },
    {
      "epoch": 0.38236,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.042789459228515625,
      "learning_rate": 0.0001,
      "loss": 4.2814,
      "loss/crossentropy": 2.2367511987686157,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22208665311336517,
      "step": 19118
    },
    {
      "epoch": 0.3824,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.040169016520182295,
      "learning_rate": 0.0001,
      "loss": 4.1671,
      "loss/crossentropy": 2.248104691505432,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2132478505373001,
      "step": 19120
    },
    {
      "epoch": 0.38244,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.040415191650390626,
      "learning_rate": 0.0001,
      "loss": 4.0371,
      "loss/crossentropy": 2.237170696258545,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21920116990804672,
      "step": 19122
    },
    {
      "epoch": 0.38248,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.018027496337890626,
      "learning_rate": 0.0001,
      "loss": 4.1011,
      "loss/crossentropy": 1.82357919216156,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1723759025335312,
      "step": 19124
    },
    {
      "epoch": 0.38252,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.018668619791666667,
      "learning_rate": 0.0001,
      "loss": 3.8541,
      "loss/crossentropy": 1.9371765851974487,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19782865047454834,
      "step": 19126
    },
    {
      "epoch": 0.38256,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0208404541015625,
      "learning_rate": 0.0001,
      "loss": 4.1305,
      "loss/crossentropy": 1.8295999765396118,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1756351813673973,
      "step": 19128
    },
    {
      "epoch": 0.3826,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.044077301025390626,
      "learning_rate": 0.0001,
      "loss": 4.0657,
      "loss/crossentropy": 2.0991050601005554,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17175115644931793,
      "step": 19130
    },
    {
      "epoch": 0.38264,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.04390640258789062,
      "learning_rate": 0.0001,
      "loss": 4.1359,
      "loss/crossentropy": 2.2707191705703735,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22085107117891312,
      "step": 19132
    },
    {
      "epoch": 0.38268,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.03630345662434896,
      "learning_rate": 0.0001,
      "loss": 4.352,
      "loss/crossentropy": 2.237929582595825,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2524951994419098,
      "step": 19134
    },
    {
      "epoch": 0.38272,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.03629735310872396,
      "learning_rate": 0.0001,
      "loss": 4.0293,
      "loss/crossentropy": 1.9790211915969849,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18156784772872925,
      "step": 19136
    },
    {
      "epoch": 0.38276,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.034795888264973956,
      "learning_rate": 0.0001,
      "loss": 3.9467,
      "loss/crossentropy": 1.9819305539131165,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18941760063171387,
      "step": 19138
    },
    {
      "epoch": 0.3828,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.032022857666015626,
      "learning_rate": 0.0001,
      "loss": 3.9699,
      "loss/crossentropy": 1.9541404843330383,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.195252887904644,
      "step": 19140
    },
    {
      "epoch": 0.38284,
      "grad_norm": 2.0,
      "grad_norm_var": 0.028824869791666666,
      "learning_rate": 0.0001,
      "loss": 3.9563,
      "loss/crossentropy": 2.1265366673469543,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20113253593444824,
      "step": 19142
    },
    {
      "epoch": 0.38288,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.0292724609375,
      "learning_rate": 0.0001,
      "loss": 4.0248,
      "loss/crossentropy": 1.767483413219452,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18520043045282364,
      "step": 19144
    },
    {
      "epoch": 0.38292,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.005352528889973959,
      "learning_rate": 0.0001,
      "loss": 4.2516,
      "loss/crossentropy": 2.1124974489212036,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20909518748521805,
      "step": 19146
    },
    {
      "epoch": 0.38296,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.005181630452473958,
      "learning_rate": 0.0001,
      "loss": 4.3104,
      "loss/crossentropy": 2.2959529161453247,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2183123677968979,
      "step": 19148
    },
    {
      "epoch": 0.383,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.005078125,
      "learning_rate": 0.0001,
      "loss": 4.3217,
      "loss/crossentropy": 2.3522391319274902,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22062842547893524,
      "step": 19150
    },
    {
      "epoch": 0.38304,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0048095703125,
      "learning_rate": 0.0001,
      "loss": 3.7375,
      "loss/crossentropy": 1.6673399806022644,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16702746599912643,
      "step": 19152
    },
    {
      "epoch": 0.38308,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.004073079427083333,
      "learning_rate": 0.0001,
      "loss": 4.2058,
      "loss/crossentropy": 2.021436333656311,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19833282381296158,
      "step": 19154
    },
    {
      "epoch": 0.38312,
      "grad_norm": 2.0,
      "grad_norm_var": 0.007439931233723958,
      "learning_rate": 0.0001,
      "loss": 3.7624,
      "loss/crossentropy": 1.4123128056526184,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1485099121928215,
      "step": 19156
    },
    {
      "epoch": 0.38316,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.009959920247395834,
      "learning_rate": 0.0001,
      "loss": 4.2364,
      "loss/crossentropy": 1.9472095966339111,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16975026577711105,
      "step": 19158
    },
    {
      "epoch": 0.3832,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0094390869140625,
      "learning_rate": 0.0001,
      "loss": 4.2025,
      "loss/crossentropy": 2.112083077430725,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2198200300335884,
      "step": 19160
    },
    {
      "epoch": 0.38324,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0121246337890625,
      "learning_rate": 0.0001,
      "loss": 3.9661,
      "loss/crossentropy": 2.1301331520080566,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2039676010608673,
      "step": 19162
    },
    {
      "epoch": 0.38328,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.019456990559895835,
      "learning_rate": 0.0001,
      "loss": 4.0655,
      "loss/crossentropy": 1.8461318016052246,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17811527848243713,
      "step": 19164
    },
    {
      "epoch": 0.38332,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.018973795572916667,
      "learning_rate": 0.0001,
      "loss": 4.0408,
      "loss/crossentropy": 2.102015793323517,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21166063100099564,
      "step": 19166
    },
    {
      "epoch": 0.38336,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.020167795817057292,
      "learning_rate": 0.0001,
      "loss": 3.9572,
      "loss/crossentropy": 1.9496545791625977,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17148377001285553,
      "step": 19168
    },
    {
      "epoch": 0.3834,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.020643870035807293,
      "learning_rate": 0.0001,
      "loss": 4.0675,
      "loss/crossentropy": 2.059878885746002,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1984856054186821,
      "step": 19170
    },
    {
      "epoch": 0.38344,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.017923990885416668,
      "learning_rate": 0.0001,
      "loss": 3.6476,
      "loss/crossentropy": 1.7966619729995728,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17823103070259094,
      "step": 19172
    },
    {
      "epoch": 0.38348,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.01645075480143229,
      "learning_rate": 0.0001,
      "loss": 3.6671,
      "loss/crossentropy": 2.0173062086105347,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19681749492883682,
      "step": 19174
    },
    {
      "epoch": 0.38352,
      "grad_norm": 1.7265625,
      "grad_norm_var": 0.019108072916666666,
      "learning_rate": 0.0001,
      "loss": 3.6439,
      "loss/crossentropy": 1.6330417394638062,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17260564863681793,
      "step": 19176
    },
    {
      "epoch": 0.38356,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.01646728515625,
      "learning_rate": 0.0001,
      "loss": 3.8957,
      "loss/crossentropy": 1.6575458645820618,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17257437109947205,
      "step": 19178
    },
    {
      "epoch": 0.3836,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.006266021728515625,
      "learning_rate": 0.0001,
      "loss": 3.8633,
      "loss/crossentropy": 2.094203770160675,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2039681375026703,
      "step": 19180
    },
    {
      "epoch": 0.38364,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.006306966145833333,
      "learning_rate": 0.0001,
      "loss": 3.9579,
      "loss/crossentropy": 1.9561032056808472,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19183862209320068,
      "step": 19182
    },
    {
      "epoch": 0.38368,
      "grad_norm": 2.609375,
      "grad_norm_var": 0.03765055338541667,
      "learning_rate": 0.0001,
      "loss": 4.4288,
      "loss/crossentropy": 1.950503408908844,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.198556549847126,
      "step": 19184
    },
    {
      "epoch": 0.38372,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0398193359375,
      "learning_rate": 0.0001,
      "loss": 4.2354,
      "loss/crossentropy": 2.421600103378296,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.208856999874115,
      "step": 19186
    },
    {
      "epoch": 0.38376,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.04036356608072917,
      "learning_rate": 0.0001,
      "loss": 3.9318,
      "loss/crossentropy": 1.7086477279663086,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19327964633703232,
      "step": 19188
    },
    {
      "epoch": 0.3838,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.041071573893229164,
      "learning_rate": 0.0001,
      "loss": 4.1488,
      "loss/crossentropy": 2.1670665740966797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19333725422620773,
      "step": 19190
    },
    {
      "epoch": 0.38384,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.03430582682291667,
      "learning_rate": 0.0001,
      "loss": 4.3145,
      "loss/crossentropy": 2.218670129776001,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20835422724485397,
      "step": 19192
    },
    {
      "epoch": 0.38388,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.037699127197265626,
      "learning_rate": 0.0001,
      "loss": 3.7753,
      "loss/crossentropy": 1.909380555152893,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18018050491809845,
      "step": 19194
    },
    {
      "epoch": 0.38392,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.038211822509765625,
      "learning_rate": 0.0001,
      "loss": 4.1049,
      "loss/crossentropy": 1.9840248227119446,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19120849668979645,
      "step": 19196
    },
    {
      "epoch": 0.38396,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.042012532552083336,
      "learning_rate": 0.0001,
      "loss": 4.4124,
      "loss/crossentropy": 2.3527809381484985,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20009320974349976,
      "step": 19198
    },
    {
      "epoch": 0.384,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0181793212890625,
      "learning_rate": 0.0001,
      "loss": 4.1011,
      "loss/crossentropy": 2.2111377716064453,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19603972136974335,
      "step": 19200
    },
    {
      "epoch": 0.38404,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.01685358683268229,
      "learning_rate": 0.0001,
      "loss": 3.9995,
      "loss/crossentropy": 1.6832327842712402,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18738068640232086,
      "step": 19202
    },
    {
      "epoch": 0.38408,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.015805816650390624,
      "learning_rate": 0.0001,
      "loss": 3.9789,
      "loss/crossentropy": 1.9412715435028076,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1835481896996498,
      "step": 19204
    },
    {
      "epoch": 0.38412,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.013451131184895833,
      "learning_rate": 0.0001,
      "loss": 4.203,
      "loss/crossentropy": 2.0641706585884094,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20165929198265076,
      "step": 19206
    },
    {
      "epoch": 0.38416,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.011400349934895833,
      "learning_rate": 0.0001,
      "loss": 4.0054,
      "loss/crossentropy": 2.2053582668304443,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2108134999871254,
      "step": 19208
    },
    {
      "epoch": 0.3842,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.010015614827473958,
      "learning_rate": 0.0001,
      "loss": 3.8549,
      "loss/crossentropy": 2.0534805059432983,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19088375568389893,
      "step": 19210
    },
    {
      "epoch": 0.38424,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.010969034830729167,
      "learning_rate": 0.0001,
      "loss": 4.06,
      "loss/crossentropy": 1.8782867789268494,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17865915596485138,
      "step": 19212
    },
    {
      "epoch": 0.38428,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0059201558430989586,
      "learning_rate": 0.0001,
      "loss": 4.0641,
      "loss/crossentropy": 1.9773831963539124,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20622816681861877,
      "step": 19214
    },
    {
      "epoch": 0.38432,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0067942301432291664,
      "learning_rate": 0.0001,
      "loss": 4.1479,
      "loss/crossentropy": 2.0890401005744934,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2173013687133789,
      "step": 19216
    },
    {
      "epoch": 0.38436,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.008676910400390625,
      "learning_rate": 0.0001,
      "loss": 3.7221,
      "loss/crossentropy": 1.858555793762207,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15820113569498062,
      "step": 19218
    },
    {
      "epoch": 0.3844,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.00911865234375,
      "learning_rate": 0.0001,
      "loss": 3.9924,
      "loss/crossentropy": 2.2149851322174072,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2221473827958107,
      "step": 19220
    },
    {
      "epoch": 0.38444,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.009490712483723959,
      "learning_rate": 0.0001,
      "loss": 3.7057,
      "loss/crossentropy": 2.086311161518097,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1791597157716751,
      "step": 19222
    },
    {
      "epoch": 0.38448,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.009112294514973958,
      "learning_rate": 0.0001,
      "loss": 3.9117,
      "loss/crossentropy": 1.810127079486847,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2018662691116333,
      "step": 19224
    },
    {
      "epoch": 0.38452,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.009064737955729167,
      "learning_rate": 0.0001,
      "loss": 4.3156,
      "loss/crossentropy": 2.132863163948059,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22217299789190292,
      "step": 19226
    },
    {
      "epoch": 0.38456,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.008299763997395833,
      "learning_rate": 0.0001,
      "loss": 3.8697,
      "loss/crossentropy": 2.0617652535438538,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1968286782503128,
      "step": 19228
    },
    {
      "epoch": 0.3846,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.007250722249348958,
      "learning_rate": 0.0001,
      "loss": 3.8566,
      "loss/crossentropy": 1.9286837577819824,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20616496354341507,
      "step": 19230
    },
    {
      "epoch": 0.38464,
      "grad_norm": 2.125,
      "grad_norm_var": 0.008150227864583333,
      "learning_rate": 0.0001,
      "loss": 4.2854,
      "loss/crossentropy": 2.009112238883972,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19038759917020798,
      "step": 19232
    },
    {
      "epoch": 0.38468,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.006818644205729167,
      "learning_rate": 0.0001,
      "loss": 4.0879,
      "loss/crossentropy": 2.1937737464904785,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20164573192596436,
      "step": 19234
    },
    {
      "epoch": 0.38472,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.00587158203125,
      "learning_rate": 0.0001,
      "loss": 4.2356,
      "loss/crossentropy": 1.981977641582489,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20445218682289124,
      "step": 19236
    },
    {
      "epoch": 0.38476,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.004648590087890625,
      "learning_rate": 0.0001,
      "loss": 3.9515,
      "loss/crossentropy": 1.6332372426986694,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1629931926727295,
      "step": 19238
    },
    {
      "epoch": 0.3848,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.003952789306640625,
      "learning_rate": 0.0001,
      "loss": 3.9061,
      "loss/crossentropy": 1.8656854629516602,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18546659499406815,
      "step": 19240
    },
    {
      "epoch": 0.38484,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.005594635009765625,
      "learning_rate": 0.0001,
      "loss": 4.1251,
      "loss/crossentropy": 2.219391703605652,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21327267587184906,
      "step": 19242
    },
    {
      "epoch": 0.38488,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.007883453369140625,
      "learning_rate": 0.0001,
      "loss": 4.448,
      "loss/crossentropy": 1.9518468976020813,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23419301211833954,
      "step": 19244
    },
    {
      "epoch": 0.38492,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.007411448160807291,
      "learning_rate": 0.0001,
      "loss": 4.2947,
      "loss/crossentropy": 2.190311014652252,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2218230664730072,
      "step": 19246
    },
    {
      "epoch": 0.38496,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.009633127848307292,
      "learning_rate": 0.0001,
      "loss": 3.8475,
      "loss/crossentropy": 1.8347881436347961,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20036222785711288,
      "step": 19248
    },
    {
      "epoch": 0.385,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.012157185872395834,
      "learning_rate": 0.0001,
      "loss": 3.9732,
      "loss/crossentropy": 1.949516236782074,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20307819545269012,
      "step": 19250
    },
    {
      "epoch": 0.38504,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.012892405192057291,
      "learning_rate": 0.0001,
      "loss": 4.007,
      "loss/crossentropy": 1.883664846420288,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1825740560889244,
      "step": 19252
    },
    {
      "epoch": 0.38508,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0146881103515625,
      "learning_rate": 0.0001,
      "loss": 4.1794,
      "loss/crossentropy": 2.1245557069778442,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20935190469026566,
      "step": 19254
    },
    {
      "epoch": 0.38512,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.015166982014973959,
      "learning_rate": 0.0001,
      "loss": 3.9125,
      "loss/crossentropy": 1.546354353427887,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15509265661239624,
      "step": 19256
    },
    {
      "epoch": 0.38516,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.014290110270182291,
      "learning_rate": 0.0001,
      "loss": 4.1744,
      "loss/crossentropy": 1.979533076286316,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20994720607995987,
      "step": 19258
    },
    {
      "epoch": 0.3852,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.014074452718098958,
      "learning_rate": 0.0001,
      "loss": 3.7774,
      "loss/crossentropy": 2.125569224357605,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21175500005483627,
      "step": 19260
    },
    {
      "epoch": 0.38524,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.0124267578125,
      "learning_rate": 0.0001,
      "loss": 3.911,
      "loss/crossentropy": 2.027758777141571,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20255477726459503,
      "step": 19262
    },
    {
      "epoch": 0.38528,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.011058553059895834,
      "learning_rate": 0.0001,
      "loss": 3.4728,
      "loss/crossentropy": 1.9937690496444702,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1836088001728058,
      "step": 19264
    },
    {
      "epoch": 0.38532,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.00863037109375,
      "learning_rate": 0.0001,
      "loss": 4.1797,
      "loss/crossentropy": 2.0846773386001587,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18216369301080704,
      "step": 19266
    },
    {
      "epoch": 0.38536,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.008429972330729167,
      "learning_rate": 0.0001,
      "loss": 3.8911,
      "loss/crossentropy": 1.445238471031189,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17925872653722763,
      "step": 19268
    },
    {
      "epoch": 0.3854,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0061757405598958336,
      "learning_rate": 0.0001,
      "loss": 4.0615,
      "loss/crossentropy": 2.0442845821380615,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2024901956319809,
      "step": 19270
    },
    {
      "epoch": 0.38544,
      "grad_norm": 1.7265625,
      "grad_norm_var": 0.009144846598307292,
      "learning_rate": 0.0001,
      "loss": 4.0898,
      "loss/crossentropy": 1.9336092472076416,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17869817465543747,
      "step": 19272
    },
    {
      "epoch": 0.38548,
      "grad_norm": 2.125,
      "grad_norm_var": 0.010884348551432292,
      "learning_rate": 0.0001,
      "loss": 4.1939,
      "loss/crossentropy": 1.815299928188324,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.182998888194561,
      "step": 19274
    },
    {
      "epoch": 0.38552,
      "grad_norm": 2.0,
      "grad_norm_var": 0.008886464436848958,
      "learning_rate": 0.0001,
      "loss": 4.2121,
      "loss/crossentropy": 2.091490864753723,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19843900948762894,
      "step": 19276
    },
    {
      "epoch": 0.38556,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009608713785807292,
      "learning_rate": 0.0001,
      "loss": 4.098,
      "loss/crossentropy": 1.9923794865608215,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.186911903321743,
      "step": 19278
    },
    {
      "epoch": 0.3856,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.009867350260416666,
      "learning_rate": 0.0001,
      "loss": 4.0658,
      "loss/crossentropy": 1.9323501586914062,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1931384950876236,
      "step": 19280
    },
    {
      "epoch": 0.38564,
      "grad_norm": 1.875,
      "grad_norm_var": 0.010262044270833333,
      "learning_rate": 0.0001,
      "loss": 3.8819,
      "loss/crossentropy": 1.8469224572181702,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1835423856973648,
      "step": 19282
    },
    {
      "epoch": 0.38568,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.010453287760416667,
      "learning_rate": 0.0001,
      "loss": 3.9027,
      "loss/crossentropy": 1.9476045370101929,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17014781385660172,
      "step": 19284
    },
    {
      "epoch": 0.38572,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.010285441080729167,
      "learning_rate": 0.0001,
      "loss": 3.9943,
      "loss/crossentropy": 1.7913609743118286,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18661632388830185,
      "step": 19286
    },
    {
      "epoch": 0.38576,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.00760498046875,
      "learning_rate": 0.0001,
      "loss": 4.0283,
      "loss/crossentropy": 1.9682837128639221,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1799641102552414,
      "step": 19288
    },
    {
      "epoch": 0.3858,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.006331125895182292,
      "learning_rate": 0.0001,
      "loss": 4.3075,
      "loss/crossentropy": 2.117633819580078,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.208334282040596,
      "step": 19290
    },
    {
      "epoch": 0.38584,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.005761464436848958,
      "learning_rate": 0.0001,
      "loss": 4.1215,
      "loss/crossentropy": 2.1208746433258057,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20928698778152466,
      "step": 19292
    },
    {
      "epoch": 0.38588,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.005549112955729167,
      "learning_rate": 0.0001,
      "loss": 4.0572,
      "loss/crossentropy": 1.9424505829811096,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18668201565742493,
      "step": 19294
    },
    {
      "epoch": 0.38592,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.005277252197265625,
      "learning_rate": 0.0001,
      "loss": 4.1575,
      "loss/crossentropy": 2.251755177974701,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21153685450553894,
      "step": 19296
    },
    {
      "epoch": 0.38596,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.007114410400390625,
      "learning_rate": 0.0001,
      "loss": 4.1885,
      "loss/crossentropy": 2.1629436016082764,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.206063412129879,
      "step": 19298
    },
    {
      "epoch": 0.386,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0069732666015625,
      "learning_rate": 0.0001,
      "loss": 4.1267,
      "loss/crossentropy": 2.0333832502365112,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1872929260134697,
      "step": 19300
    },
    {
      "epoch": 0.38604,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.007641347249348959,
      "learning_rate": 0.0001,
      "loss": 3.8433,
      "loss/crossentropy": 1.8364217281341553,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17408161610364914,
      "step": 19302
    },
    {
      "epoch": 0.38608,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.005973307291666666,
      "learning_rate": 0.0001,
      "loss": 3.9175,
      "loss/crossentropy": 1.975037932395935,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18705998361110687,
      "step": 19304
    },
    {
      "epoch": 0.38612,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.006477864583333334,
      "learning_rate": 0.0001,
      "loss": 3.7868,
      "loss/crossentropy": 1.6889175176620483,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16312924772500992,
      "step": 19306
    },
    {
      "epoch": 0.38616,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.006392161051432292,
      "learning_rate": 0.0001,
      "loss": 4.1299,
      "loss/crossentropy": 2.2644211053848267,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18390889465808868,
      "step": 19308
    },
    {
      "epoch": 0.3862,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.009723917643229166,
      "learning_rate": 0.0001,
      "loss": 4.325,
      "loss/crossentropy": 2.180909037590027,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20263531804084778,
      "step": 19310
    },
    {
      "epoch": 0.38624,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010138956705729167,
      "learning_rate": 0.0001,
      "loss": 4.2001,
      "loss/crossentropy": 2.4007444381713867,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23251044005155563,
      "step": 19312
    },
    {
      "epoch": 0.38628,
      "grad_norm": 2.0,
      "grad_norm_var": 0.010534413655598958,
      "learning_rate": 0.0001,
      "loss": 3.9862,
      "loss/crossentropy": 1.8122236728668213,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18613354116678238,
      "step": 19314
    },
    {
      "epoch": 0.38632,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.010404459635416667,
      "learning_rate": 0.0001,
      "loss": 4.1379,
      "loss/crossentropy": 2.159746825695038,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21711497008800507,
      "step": 19316
    },
    {
      "epoch": 0.38636,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.009627024332682291,
      "learning_rate": 0.0001,
      "loss": 3.9929,
      "loss/crossentropy": 2.129119336605072,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19623322039842606,
      "step": 19318
    },
    {
      "epoch": 0.3864,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.009934234619140624,
      "learning_rate": 0.0001,
      "loss": 4.1578,
      "loss/crossentropy": 2.322758913040161,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20978064835071564,
      "step": 19320
    },
    {
      "epoch": 0.38644,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010322825113932291,
      "learning_rate": 0.0001,
      "loss": 4.0266,
      "loss/crossentropy": 2.2643767595291138,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21338346600532532,
      "step": 19322
    },
    {
      "epoch": 0.38648,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.010731760660807292,
      "learning_rate": 0.0001,
      "loss": 4.0614,
      "loss/crossentropy": 1.8038535118103027,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19042538851499557,
      "step": 19324
    },
    {
      "epoch": 0.38652,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0069620768229166664,
      "learning_rate": 0.0001,
      "loss": 3.9641,
      "loss/crossentropy": 2.0847853422164917,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1965423971414566,
      "step": 19326
    },
    {
      "epoch": 0.38656,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.007692209879557292,
      "learning_rate": 0.0001,
      "loss": 4.3347,
      "loss/crossentropy": 2.2121591567993164,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21294504404067993,
      "step": 19328
    },
    {
      "epoch": 0.3866,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.005985260009765625,
      "learning_rate": 0.0001,
      "loss": 3.9869,
      "loss/crossentropy": 2.100327968597412,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20181559026241302,
      "step": 19330
    },
    {
      "epoch": 0.38664,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.00750732421875,
      "learning_rate": 0.0001,
      "loss": 3.642,
      "loss/crossentropy": 1.7100898623466492,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18045856803655624,
      "step": 19332
    },
    {
      "epoch": 0.38668,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.008337148030598958,
      "learning_rate": 0.0001,
      "loss": 3.712,
      "loss/crossentropy": 1.6435258388519287,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1821775734424591,
      "step": 19334
    },
    {
      "epoch": 0.38672,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.008674875895182291,
      "learning_rate": 0.0001,
      "loss": 3.8334,
      "loss/crossentropy": 2.338008165359497,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2040715217590332,
      "step": 19336
    },
    {
      "epoch": 0.38676,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.008454386393229167,
      "learning_rate": 0.0001,
      "loss": 3.8911,
      "loss/crossentropy": 1.8373408913612366,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17516936361789703,
      "step": 19338
    },
    {
      "epoch": 0.3868,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.008422597249348959,
      "learning_rate": 0.0001,
      "loss": 3.9902,
      "loss/crossentropy": 2.0346588492393494,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20979416370391846,
      "step": 19340
    },
    {
      "epoch": 0.38684,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.009399159749348959,
      "learning_rate": 0.0001,
      "loss": 4.1759,
      "loss/crossentropy": 2.083326816558838,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20702984929084778,
      "step": 19342
    },
    {
      "epoch": 0.38688,
      "grad_norm": 2.0,
      "grad_norm_var": 0.012473297119140626,
      "learning_rate": 0.0001,
      "loss": 4.611,
      "loss/crossentropy": 2.4100319147109985,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2617071568965912,
      "step": 19344
    },
    {
      "epoch": 0.38692,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.012550608317057291,
      "learning_rate": 0.0001,
      "loss": 3.8864,
      "loss/crossentropy": 1.7147992849349976,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.174342080950737,
      "step": 19346
    },
    {
      "epoch": 0.38696,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.013602701822916667,
      "learning_rate": 0.0001,
      "loss": 4.1508,
      "loss/crossentropy": 2.041933536529541,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2024158239364624,
      "step": 19348
    },
    {
      "epoch": 0.387,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.012434895833333333,
      "learning_rate": 0.0001,
      "loss": 4.0929,
      "loss/crossentropy": 1.835128128528595,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1861354559659958,
      "step": 19350
    },
    {
      "epoch": 0.38704,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.013370768229166666,
      "learning_rate": 0.0001,
      "loss": 4.164,
      "loss/crossentropy": 1.9472790360450745,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19485513865947723,
      "step": 19352
    },
    {
      "epoch": 0.38708,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.012428538004557291,
      "learning_rate": 0.0001,
      "loss": 4.0217,
      "loss/crossentropy": 2.0546197295188904,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1883516013622284,
      "step": 19354
    },
    {
      "epoch": 0.38712,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.013169097900390624,
      "learning_rate": 0.0001,
      "loss": 3.6439,
      "loss/crossentropy": 1.8285245299339294,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1730484738945961,
      "step": 19356
    },
    {
      "epoch": 0.38716,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.013890584309895834,
      "learning_rate": 0.0001,
      "loss": 4.2169,
      "loss/crossentropy": 2.320949673652649,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21517502516508102,
      "step": 19358
    },
    {
      "epoch": 0.3872,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.011289215087890625,
      "learning_rate": 0.0001,
      "loss": 4.0196,
      "loss/crossentropy": 2.2944518327713013,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20343206077814102,
      "step": 19360
    },
    {
      "epoch": 0.38724,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0112945556640625,
      "learning_rate": 0.0001,
      "loss": 4.2113,
      "loss/crossentropy": 2.134889602661133,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19354654103517532,
      "step": 19362
    },
    {
      "epoch": 0.38728,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.009639231363932292,
      "learning_rate": 0.0001,
      "loss": 4.1244,
      "loss/crossentropy": 2.206334412097931,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20670340210199356,
      "step": 19364
    },
    {
      "epoch": 0.38732,
      "grad_norm": 2.0,
      "grad_norm_var": 0.009769694010416666,
      "learning_rate": 0.0001,
      "loss": 3.8194,
      "loss/crossentropy": 2.1286932229995728,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19814791530370712,
      "step": 19366
    },
    {
      "epoch": 0.38736,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0066314697265625,
      "learning_rate": 0.0001,
      "loss": 4.1533,
      "loss/crossentropy": 2.119162678718567,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18044983595609665,
      "step": 19368
    },
    {
      "epoch": 0.3874,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.005060831705729167,
      "learning_rate": 0.0001,
      "loss": 4.0061,
      "loss/crossentropy": 2.4316102266311646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22509240359067917,
      "step": 19370
    },
    {
      "epoch": 0.38744,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.0060117085774739586,
      "learning_rate": 0.0001,
      "loss": 3.9984,
      "loss/crossentropy": 2.540325403213501,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22844604402780533,
      "step": 19372
    },
    {
      "epoch": 0.38748,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.005782063802083333,
      "learning_rate": 0.0001,
      "loss": 4.1795,
      "loss/crossentropy": 2.3002058267593384,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21805168688297272,
      "step": 19374
    },
    {
      "epoch": 0.38752,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0058258056640625,
      "learning_rate": 0.0001,
      "loss": 4.274,
      "loss/crossentropy": 2.2381847500801086,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19565512239933014,
      "step": 19376
    },
    {
      "epoch": 0.38756,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0060302734375,
      "learning_rate": 0.0001,
      "loss": 4.2683,
      "loss/crossentropy": 2.101936161518097,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.198813758790493,
      "step": 19378
    },
    {
      "epoch": 0.3876,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0060943603515625,
      "learning_rate": 0.0001,
      "loss": 3.8679,
      "loss/crossentropy": 2.1088311672210693,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21237140893936157,
      "step": 19380
    },
    {
      "epoch": 0.38764,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0060943603515625,
      "learning_rate": 0.0001,
      "loss": 3.9588,
      "loss/crossentropy": 2.0985374450683594,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18979168683290482,
      "step": 19382
    },
    {
      "epoch": 0.38768,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.005690256754557292,
      "learning_rate": 0.0001,
      "loss": 4.0798,
      "loss/crossentropy": 2.3004449605941772,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2053247168660164,
      "step": 19384
    },
    {
      "epoch": 0.38772,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.005277252197265625,
      "learning_rate": 0.0001,
      "loss": 4.0285,
      "loss/crossentropy": 2.1257707476615906,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22222986072301865,
      "step": 19386
    },
    {
      "epoch": 0.38776,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.003733062744140625,
      "learning_rate": 0.0001,
      "loss": 4.1336,
      "loss/crossentropy": 2.3034613132476807,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2174137905240059,
      "step": 19388
    },
    {
      "epoch": 0.3878,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.006245930989583333,
      "learning_rate": 0.0001,
      "loss": 3.9854,
      "loss/crossentropy": 2.1303011178970337,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22152414172887802,
      "step": 19390
    },
    {
      "epoch": 0.38784,
      "grad_norm": 2.0,
      "grad_norm_var": 0.010412343343098958,
      "learning_rate": 0.0001,
      "loss": 4.1868,
      "loss/crossentropy": 1.8575092554092407,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18628983944654465,
      "step": 19392
    },
    {
      "epoch": 0.38788,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.011694081624348958,
      "learning_rate": 0.0001,
      "loss": 4.1219,
      "loss/crossentropy": 2.0332913994789124,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2499927133321762,
      "step": 19394
    },
    {
      "epoch": 0.38792,
      "grad_norm": 2.125,
      "grad_norm_var": 0.013004302978515625,
      "learning_rate": 0.0001,
      "loss": 4.2275,
      "loss/crossentropy": 2.0926660895347595,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19264871627092361,
      "step": 19396
    },
    {
      "epoch": 0.38796,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.0156005859375,
      "learning_rate": 0.0001,
      "loss": 4.2337,
      "loss/crossentropy": 2.08814400434494,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21137948334217072,
      "step": 19398
    },
    {
      "epoch": 0.388,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.019162750244140624,
      "learning_rate": 0.0001,
      "loss": 4.0263,
      "loss/crossentropy": 2.074121594429016,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20524942874908447,
      "step": 19400
    },
    {
      "epoch": 0.38804,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.020589192708333332,
      "learning_rate": 0.0001,
      "loss": 3.9437,
      "loss/crossentropy": 1.8818755745887756,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1794123500585556,
      "step": 19402
    },
    {
      "epoch": 0.38808,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.020992024739583334,
      "learning_rate": 0.0001,
      "loss": 3.8638,
      "loss/crossentropy": 2.5050524473190308,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22337081283330917,
      "step": 19404
    },
    {
      "epoch": 0.38812,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0180572509765625,
      "learning_rate": 0.0001,
      "loss": 4.2158,
      "loss/crossentropy": 2.2035861015319824,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20753300189971924,
      "step": 19406
    },
    {
      "epoch": 0.38816,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.014012654622395834,
      "learning_rate": 0.0001,
      "loss": 3.8975,
      "loss/crossentropy": 2.2270091772079468,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20527157932519913,
      "step": 19408
    },
    {
      "epoch": 0.3882,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.013504791259765624,
      "learning_rate": 0.0001,
      "loss": 3.951,
      "loss/crossentropy": 1.6546313762664795,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17243140190839767,
      "step": 19410
    },
    {
      "epoch": 0.38824,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.012756093343098959,
      "learning_rate": 0.0001,
      "loss": 4.0653,
      "loss/crossentropy": 1.8858160376548767,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18621040880680084,
      "step": 19412
    },
    {
      "epoch": 0.38828,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.011875152587890625,
      "learning_rate": 0.0001,
      "loss": 4.4744,
      "loss/crossentropy": 1.9876565337181091,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.30138373374938965,
      "step": 19414
    },
    {
      "epoch": 0.38832,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.013053131103515626,
      "learning_rate": 0.0001,
      "loss": 3.7802,
      "loss/crossentropy": 2.1887502670288086,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1910400092601776,
      "step": 19416
    },
    {
      "epoch": 0.38836,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.012511952718098959,
      "learning_rate": 0.0001,
      "loss": 3.9398,
      "loss/crossentropy": 2.246508002281189,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2018478363752365,
      "step": 19418
    },
    {
      "epoch": 0.3884,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.011510976155598958,
      "learning_rate": 0.0001,
      "loss": 4.1703,
      "loss/crossentropy": 2.041202425956726,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1890542358160019,
      "step": 19420
    },
    {
      "epoch": 0.38844,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.016904449462890624,
      "learning_rate": 0.0001,
      "loss": 4.4154,
      "loss/crossentropy": 2.0470627546310425,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19648576527833939,
      "step": 19422
    },
    {
      "epoch": 0.38848,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0155181884765625,
      "learning_rate": 0.0001,
      "loss": 4.1794,
      "loss/crossentropy": 2.172736167907715,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22087720036506653,
      "step": 19424
    },
    {
      "epoch": 0.38852,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.015811920166015625,
      "learning_rate": 0.0001,
      "loss": 4.0106,
      "loss/crossentropy": 2.0988662242889404,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18553303182125092,
      "step": 19426
    },
    {
      "epoch": 0.38856,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.017134348551432293,
      "learning_rate": 0.0001,
      "loss": 4.1368,
      "loss/crossentropy": 1.821892261505127,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19670143723487854,
      "step": 19428
    },
    {
      "epoch": 0.3886,
      "grad_norm": 1.875,
      "grad_norm_var": 0.016747792561848957,
      "learning_rate": 0.0001,
      "loss": 3.7878,
      "loss/crossentropy": 1.8893300294876099,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18850861489772797,
      "step": 19430
    },
    {
      "epoch": 0.38864,
      "grad_norm": 2.984375,
      "grad_norm_var": 0.07888997395833333,
      "learning_rate": 0.0001,
      "loss": 4.243,
      "loss/crossentropy": 2.210429847240448,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17908993363380432,
      "step": 19432
    },
    {
      "epoch": 0.38868,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.07731526692708333,
      "learning_rate": 0.0001,
      "loss": 3.9939,
      "loss/crossentropy": 2.164771556854248,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1919529214501381,
      "step": 19434
    },
    {
      "epoch": 0.38872,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.07656631469726563,
      "learning_rate": 0.0001,
      "loss": 4.1721,
      "loss/crossentropy": 2.1335190534591675,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2263372465968132,
      "step": 19436
    },
    {
      "epoch": 0.38876,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.073779296875,
      "learning_rate": 0.0001,
      "loss": 4.06,
      "loss/crossentropy": 2.199760317802429,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20560061931610107,
      "step": 19438
    },
    {
      "epoch": 0.3888,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.07665608723958334,
      "learning_rate": 0.0001,
      "loss": 3.9367,
      "loss/crossentropy": 2.2948896884918213,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2016046792268753,
      "step": 19440
    },
    {
      "epoch": 0.38884,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.07588882446289062,
      "learning_rate": 0.0001,
      "loss": 4.1444,
      "loss/crossentropy": 2.120785415172577,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18029560148715973,
      "step": 19442
    },
    {
      "epoch": 0.38888,
      "grad_norm": 1.875,
      "grad_norm_var": 0.07493260701497396,
      "learning_rate": 0.0001,
      "loss": 3.8872,
      "loss/crossentropy": 2.049036145210266,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19056373089551926,
      "step": 19444
    },
    {
      "epoch": 0.38892,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.07226130167643229,
      "learning_rate": 0.0001,
      "loss": 4.2652,
      "loss/crossentropy": 2.0076504945755005,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20951516926288605,
      "step": 19446
    },
    {
      "epoch": 0.38896,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.006740061442057291,
      "learning_rate": 0.0001,
      "loss": 3.8178,
      "loss/crossentropy": 1.9398083090782166,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21217594295740128,
      "step": 19448
    },
    {
      "epoch": 0.389,
      "grad_norm": 1.875,
      "grad_norm_var": 0.00546875,
      "learning_rate": 0.0001,
      "loss": 4.0245,
      "loss/crossentropy": 2.0976104736328125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19852425903081894,
      "step": 19450
    },
    {
      "epoch": 0.38904,
      "grad_norm": 7.625,
      "grad_norm_var": 2.023509724934896,
      "learning_rate": 0.0001,
      "loss": 3.9741,
      "loss/crossentropy": 1.8561761379241943,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17980807274580002,
      "step": 19452
    },
    {
      "epoch": 0.38908,
      "grad_norm": 1.859375,
      "grad_norm_var": 2.0183570861816404,
      "learning_rate": 0.0001,
      "loss": 3.913,
      "loss/crossentropy": 2.1107255816459656,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20205769687891006,
      "step": 19454
    },
    {
      "epoch": 0.38912,
      "grad_norm": 1.8203125,
      "grad_norm_var": 2.009368642171224,
      "learning_rate": 0.0001,
      "loss": 3.9216,
      "loss/crossentropy": 2.018397331237793,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18721628934144974,
      "step": 19456
    },
    {
      "epoch": 0.38916,
      "grad_norm": 1.84375,
      "grad_norm_var": 2.010087076822917,
      "learning_rate": 0.0001,
      "loss": 3.8738,
      "loss/crossentropy": 2.016912341117859,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20780544728040695,
      "step": 19458
    },
    {
      "epoch": 0.3892,
      "grad_norm": 1.8671875,
      "grad_norm_var": 2.0083984375,
      "learning_rate": 0.0001,
      "loss": 3.9523,
      "loss/crossentropy": 1.945086121559143,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19544267654418945,
      "step": 19460
    },
    {
      "epoch": 0.38924,
      "grad_norm": 1.9375,
      "grad_norm_var": 2.01300048828125,
      "learning_rate": 0.0001,
      "loss": 4.1854,
      "loss/crossentropy": 2.059622883796692,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21501117944717407,
      "step": 19462
    },
    {
      "epoch": 0.38928,
      "grad_norm": 1.96875,
      "grad_norm_var": 2.024466705322266,
      "learning_rate": 0.0001,
      "loss": 3.8336,
      "loss/crossentropy": 1.8915096521377563,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1852220892906189,
      "step": 19464
    },
    {
      "epoch": 0.38932,
      "grad_norm": 2.015625,
      "grad_norm_var": 2.0162534077962238,
      "learning_rate": 0.0001,
      "loss": 4.0612,
      "loss/crossentropy": 1.5394552946090698,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16873380541801453,
      "step": 19466
    },
    {
      "epoch": 0.38936,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.03472468058268229,
      "learning_rate": 0.0001,
      "loss": 4.2426,
      "loss/crossentropy": 2.0290380716323853,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19913609325885773,
      "step": 19468
    },
    {
      "epoch": 0.3894,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.033841705322265624,
      "learning_rate": 0.0001,
      "loss": 4.1607,
      "loss/crossentropy": 2.208653211593628,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.190569207072258,
      "step": 19470
    },
    {
      "epoch": 0.38944,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0092041015625,
      "learning_rate": 0.0001,
      "loss": 4.2646,
      "loss/crossentropy": 2.2873799800872803,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20589639246463776,
      "step": 19472
    },
    {
      "epoch": 0.38948,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.008014933268229166,
      "learning_rate": 0.0001,
      "loss": 4.1662,
      "loss/crossentropy": 2.152024030685425,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20272599160671234,
      "step": 19474
    },
    {
      "epoch": 0.38952,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.009374745686848958,
      "learning_rate": 0.0001,
      "loss": 3.9932,
      "loss/crossentropy": 2.0079659819602966,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19110675901174545,
      "step": 19476
    },
    {
      "epoch": 0.38956,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.010081990559895834,
      "learning_rate": 0.0001,
      "loss": 4.1061,
      "loss/crossentropy": 2.3006476163864136,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2253967523574829,
      "step": 19478
    },
    {
      "epoch": 0.3896,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.010389963785807291,
      "learning_rate": 0.0001,
      "loss": 4.0678,
      "loss/crossentropy": 2.1429306864738464,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20268720388412476,
      "step": 19480
    },
    {
      "epoch": 0.38964,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.011417643229166666,
      "learning_rate": 0.0001,
      "loss": 4.0581,
      "loss/crossentropy": 2.260764956474304,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2121199518442154,
      "step": 19482
    },
    {
      "epoch": 0.38968,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010636393229166667,
      "learning_rate": 0.0001,
      "loss": 4.1607,
      "loss/crossentropy": 2.3874053955078125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2229148969054222,
      "step": 19484
    },
    {
      "epoch": 0.38972,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.009227498372395834,
      "learning_rate": 0.0001,
      "loss": 4.1456,
      "loss/crossentropy": 2.2740933895111084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2509172707796097,
      "step": 19486
    },
    {
      "epoch": 0.38976,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.009679921468098958,
      "learning_rate": 0.0001,
      "loss": 4.1831,
      "loss/crossentropy": 2.1949650049209595,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19400373846292496,
      "step": 19488
    },
    {
      "epoch": 0.3898,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.009549713134765625,
      "learning_rate": 0.0001,
      "loss": 3.9984,
      "loss/crossentropy": 1.4600969552993774,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15833207219839096,
      "step": 19490
    },
    {
      "epoch": 0.38984,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.019913482666015624,
      "learning_rate": 0.0001,
      "loss": 4.1694,
      "loss/crossentropy": 2.0005985498428345,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20003372430801392,
      "step": 19492
    },
    {
      "epoch": 0.38988,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.020304107666015626,
      "learning_rate": 0.0001,
      "loss": 3.8336,
      "loss/crossentropy": 1.9609830379486084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19764738529920578,
      "step": 19494
    },
    {
      "epoch": 0.38992,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.01943359375,
      "learning_rate": 0.0001,
      "loss": 3.8708,
      "loss/crossentropy": 1.7853738069534302,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18979863077402115,
      "step": 19496
    },
    {
      "epoch": 0.38996,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.018961588541666668,
      "learning_rate": 0.0001,
      "loss": 4.0335,
      "loss/crossentropy": 2.004499912261963,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20765355974435806,
      "step": 19498
    },
    {
      "epoch": 0.39,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.019870758056640625,
      "learning_rate": 0.0001,
      "loss": 4.1277,
      "loss/crossentropy": 2.0148558020591736,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18088079243898392,
      "step": 19500
    },
    {
      "epoch": 0.39004,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.022965240478515624,
      "learning_rate": 0.0001,
      "loss": 4.1213,
      "loss/crossentropy": 2.2602895498275757,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22686412185430527,
      "step": 19502
    },
    {
      "epoch": 0.39008,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.024326324462890625,
      "learning_rate": 0.0001,
      "loss": 4.0541,
      "loss/crossentropy": 2.1789051294326782,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1989632546901703,
      "step": 19504
    },
    {
      "epoch": 0.39012,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.024568684895833335,
      "learning_rate": 0.0001,
      "loss": 3.9234,
      "loss/crossentropy": 1.93650484085083,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1976509690284729,
      "step": 19506
    },
    {
      "epoch": 0.39016,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.008351389567057292,
      "learning_rate": 0.0001,
      "loss": 3.9519,
      "loss/crossentropy": 1.9904287457466125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17271699011325836,
      "step": 19508
    },
    {
      "epoch": 0.3902,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008194986979166667,
      "learning_rate": 0.0001,
      "loss": 3.7009,
      "loss/crossentropy": 1.4440776705741882,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16108233481645584,
      "step": 19510
    },
    {
      "epoch": 0.39024,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.008111317952473959,
      "learning_rate": 0.0001,
      "loss": 4.0218,
      "loss/crossentropy": 2.143462061882019,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2024468332529068,
      "step": 19512
    },
    {
      "epoch": 0.39028,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008436838785807291,
      "learning_rate": 0.0001,
      "loss": 4.1267,
      "loss/crossentropy": 2.085771322250366,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22135765105485916,
      "step": 19514
    },
    {
      "epoch": 0.39032,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.008337148030598958,
      "learning_rate": 0.0001,
      "loss": 4.1423,
      "loss/crossentropy": 2.251617908477783,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2026226669549942,
      "step": 19516
    },
    {
      "epoch": 0.39036,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.003885650634765625,
      "learning_rate": 0.0001,
      "loss": 4.3156,
      "loss/crossentropy": 2.3808701038360596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2263621687889099,
      "step": 19518
    },
    {
      "epoch": 0.3904,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.004019927978515625,
      "learning_rate": 0.0001,
      "loss": 4.0323,
      "loss/crossentropy": 2.000342011451721,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20008830726146698,
      "step": 19520
    },
    {
      "epoch": 0.39044,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0044830322265625,
      "learning_rate": 0.0001,
      "loss": 4.3285,
      "loss/crossentropy": 2.242555856704712,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2443876415491104,
      "step": 19522
    },
    {
      "epoch": 0.39048,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.005060831705729167,
      "learning_rate": 0.0001,
      "loss": 4.344,
      "loss/crossentropy": 2.2867462635040283,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21522662043571472,
      "step": 19524
    },
    {
      "epoch": 0.39052,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0053955078125,
      "learning_rate": 0.0001,
      "loss": 4.0117,
      "loss/crossentropy": 2.1892699003219604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.205935537815094,
      "step": 19526
    },
    {
      "epoch": 0.39056,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.007972971598307291,
      "learning_rate": 0.0001,
      "loss": 3.8673,
      "loss/crossentropy": 2.182734966278076,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1980958804488182,
      "step": 19528
    },
    {
      "epoch": 0.3906,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0075927734375,
      "learning_rate": 0.0001,
      "loss": 4.0193,
      "loss/crossentropy": 2.3624355792999268,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22833774983882904,
      "step": 19530
    },
    {
      "epoch": 0.39064,
      "grad_norm": 1.875,
      "grad_norm_var": 0.007671864827473959,
      "learning_rate": 0.0001,
      "loss": 4.0497,
      "loss/crossentropy": 2.037912607192993,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2032359093427658,
      "step": 19532
    },
    {
      "epoch": 0.39068,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.007513173421223958,
      "learning_rate": 0.0001,
      "loss": 3.9812,
      "loss/crossentropy": 2.076040804386139,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21829531341791153,
      "step": 19534
    },
    {
      "epoch": 0.39072,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0058502197265625,
      "learning_rate": 0.0001,
      "loss": 4.0407,
      "loss/crossentropy": 2.2695836424827576,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2125643789768219,
      "step": 19536
    },
    {
      "epoch": 0.39076,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0061279296875,
      "learning_rate": 0.0001,
      "loss": 4.2664,
      "loss/crossentropy": 1.9343088269233704,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2045460194349289,
      "step": 19538
    },
    {
      "epoch": 0.3908,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.005272420247395834,
      "learning_rate": 0.0001,
      "loss": 4.0757,
      "loss/crossentropy": 1.9968918561935425,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17788998782634735,
      "step": 19540
    },
    {
      "epoch": 0.39084,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.005467732747395833,
      "learning_rate": 0.0001,
      "loss": 4.094,
      "loss/crossentropy": 2.157910704612732,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19516880810260773,
      "step": 19542
    },
    {
      "epoch": 0.39088,
      "grad_norm": 2.0,
      "grad_norm_var": 0.004042307535807292,
      "learning_rate": 0.0001,
      "loss": 4.1056,
      "loss/crossentropy": 2.1784998178482056,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20280858874320984,
      "step": 19544
    },
    {
      "epoch": 0.39092,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.00438232421875,
      "learning_rate": 0.0001,
      "loss": 4.0796,
      "loss/crossentropy": 2.3444844484329224,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2418832629919052,
      "step": 19546
    },
    {
      "epoch": 0.39096,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0038083394368489585,
      "learning_rate": 0.0001,
      "loss": 3.9802,
      "loss/crossentropy": 1.90557062625885,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2010379433631897,
      "step": 19548
    },
    {
      "epoch": 0.391,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.004107411702473958,
      "learning_rate": 0.0001,
      "loss": 3.7756,
      "loss/crossentropy": 1.4939787983894348,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1362278200685978,
      "step": 19550
    },
    {
      "epoch": 0.39104,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0063517252604166664,
      "learning_rate": 0.0001,
      "loss": 4.1738,
      "loss/crossentropy": 2.037553310394287,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2159510999917984,
      "step": 19552
    },
    {
      "epoch": 0.39108,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.004816691080729167,
      "learning_rate": 0.0001,
      "loss": 3.887,
      "loss/crossentropy": 1.9987242221832275,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19919036328792572,
      "step": 19554
    },
    {
      "epoch": 0.39112,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0048906962076822914,
      "learning_rate": 0.0001,
      "loss": 4.0298,
      "loss/crossentropy": 2.1593196392059326,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21370293200016022,
      "step": 19556
    },
    {
      "epoch": 0.39116,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.005631510416666667,
      "learning_rate": 0.0001,
      "loss": 3.8348,
      "loss/crossentropy": 1.8323914408683777,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20382094383239746,
      "step": 19558
    },
    {
      "epoch": 0.3912,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0053059895833333336,
      "learning_rate": 0.0001,
      "loss": 4.1138,
      "loss/crossentropy": 2.0240999460220337,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19374582171440125,
      "step": 19560
    },
    {
      "epoch": 0.39124,
      "grad_norm": 2.0,
      "grad_norm_var": 0.005304972330729167,
      "learning_rate": 0.0001,
      "loss": 4.4349,
      "loss/crossentropy": 2.377955436706543,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2213267982006073,
      "step": 19562
    },
    {
      "epoch": 0.39128,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.006154123942057292,
      "learning_rate": 0.0001,
      "loss": 4.0878,
      "loss/crossentropy": 1.724283754825592,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20119256526231766,
      "step": 19564
    },
    {
      "epoch": 0.39132,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.005655670166015625,
      "learning_rate": 0.0001,
      "loss": 3.9066,
      "loss/crossentropy": 2.0352718234062195,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19105417281389236,
      "step": 19566
    },
    {
      "epoch": 0.39136,
      "grad_norm": 1.875,
      "grad_norm_var": 0.0037913004557291667,
      "learning_rate": 0.0001,
      "loss": 3.7868,
      "loss/crossentropy": 2.078265905380249,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19992397725582123,
      "step": 19568
    },
    {
      "epoch": 0.3914,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.004329172770182291,
      "learning_rate": 0.0001,
      "loss": 3.8276,
      "loss/crossentropy": 1.9722577929496765,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17697854340076447,
      "step": 19570
    },
    {
      "epoch": 0.39144,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.004915364583333333,
      "learning_rate": 0.0001,
      "loss": 4.0587,
      "loss/crossentropy": 2.2238826751708984,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20041973888874054,
      "step": 19572
    },
    {
      "epoch": 0.39148,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0037923177083333333,
      "learning_rate": 0.0001,
      "loss": 4.0226,
      "loss/crossentropy": 1.8374757170677185,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18086174875497818,
      "step": 19574
    },
    {
      "epoch": 0.39152,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.00635986328125,
      "learning_rate": 0.0001,
      "loss": 4.2903,
      "loss/crossentropy": 2.0642590522766113,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19544780254364014,
      "step": 19576
    },
    {
      "epoch": 0.39156,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.020493316650390624,
      "learning_rate": 0.0001,
      "loss": 4.1231,
      "loss/crossentropy": 2.090702533721924,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19859656691551208,
      "step": 19578
    },
    {
      "epoch": 0.3916,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.023538970947265626,
      "learning_rate": 0.0001,
      "loss": 3.9075,
      "loss/crossentropy": 1.8087154030799866,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1918642893433571,
      "step": 19580
    },
    {
      "epoch": 0.39164,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.023527018229166665,
      "learning_rate": 0.0001,
      "loss": 3.8869,
      "loss/crossentropy": 2.0180618166923523,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19190336763858795,
      "step": 19582
    },
    {
      "epoch": 0.39168,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.025886027018229167,
      "learning_rate": 0.0001,
      "loss": 4.2298,
      "loss/crossentropy": 1.980249285697937,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20591172575950623,
      "step": 19584
    },
    {
      "epoch": 0.39172,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.026374308268229167,
      "learning_rate": 0.0001,
      "loss": 3.9883,
      "loss/crossentropy": 2.1208410263061523,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20714347064495087,
      "step": 19586
    },
    {
      "epoch": 0.39176,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.024179840087890626,
      "learning_rate": 0.0001,
      "loss": 4.3101,
      "loss/crossentropy": 2.351140856742859,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26494763791561127,
      "step": 19588
    },
    {
      "epoch": 0.3918,
      "grad_norm": 2.125,
      "grad_norm_var": 0.024448394775390625,
      "learning_rate": 0.0001,
      "loss": 4.1259,
      "loss/crossentropy": 1.9821222424507141,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2062402218580246,
      "step": 19590
    },
    {
      "epoch": 0.39184,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.02462158203125,
      "learning_rate": 0.0001,
      "loss": 4.1373,
      "loss/crossentropy": 2.363589644432068,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22435829043388367,
      "step": 19592
    },
    {
      "epoch": 0.39188,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.013963826497395833,
      "learning_rate": 0.0001,
      "loss": 4.1845,
      "loss/crossentropy": 1.925455391407013,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17338209599256516,
      "step": 19594
    },
    {
      "epoch": 0.39192,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.012254842122395833,
      "learning_rate": 0.0001,
      "loss": 3.9777,
      "loss/crossentropy": 2.176175117492676,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21482165157794952,
      "step": 19596
    },
    {
      "epoch": 0.39196,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.014330037434895833,
      "learning_rate": 0.0001,
      "loss": 3.9266,
      "loss/crossentropy": 2.0013960003852844,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20629072189331055,
      "step": 19598
    },
    {
      "epoch": 0.392,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.011797841389973958,
      "learning_rate": 0.0001,
      "loss": 3.9402,
      "loss/crossentropy": 2.066355049610138,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22155777364969254,
      "step": 19600
    },
    {
      "epoch": 0.39204,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.010809071858723958,
      "learning_rate": 0.0001,
      "loss": 4.0319,
      "loss/crossentropy": 2.0513144731521606,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19442218542099,
      "step": 19602
    },
    {
      "epoch": 0.39208,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.010896809895833333,
      "learning_rate": 0.0001,
      "loss": 4.2601,
      "loss/crossentropy": 1.9635959267616272,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20168906450271606,
      "step": 19604
    },
    {
      "epoch": 0.39212,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.009025065104166667,
      "learning_rate": 0.0001,
      "loss": 3.9406,
      "loss/crossentropy": 1.9395795464515686,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17990338802337646,
      "step": 19606
    },
    {
      "epoch": 0.39216,
      "grad_norm": 2.125,
      "grad_norm_var": 0.011004384358723958,
      "learning_rate": 0.0001,
      "loss": 4.1936,
      "loss/crossentropy": 2.111960232257843,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20662778615951538,
      "step": 19608
    },
    {
      "epoch": 0.3922,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.011466471354166667,
      "learning_rate": 0.0001,
      "loss": 3.9238,
      "loss/crossentropy": 1.855182707309723,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19665290415287018,
      "step": 19610
    },
    {
      "epoch": 0.39224,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0156494140625,
      "learning_rate": 0.0001,
      "loss": 4.262,
      "loss/crossentropy": 2.1587076783180237,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20080996304750443,
      "step": 19612
    },
    {
      "epoch": 0.39228,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.01597874959309896,
      "learning_rate": 0.0001,
      "loss": 4.1227,
      "loss/crossentropy": 2.130104422569275,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19484283030033112,
      "step": 19614
    },
    {
      "epoch": 0.39232,
      "grad_norm": 1.75,
      "grad_norm_var": 0.017992146809895835,
      "learning_rate": 0.0001,
      "loss": 3.9046,
      "loss/crossentropy": 1.6983461380004883,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18040503561496735,
      "step": 19616
    },
    {
      "epoch": 0.39236,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.017756144205729168,
      "learning_rate": 0.0001,
      "loss": 4.0141,
      "loss/crossentropy": 2.0672999024391174,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20455461740493774,
      "step": 19618
    },
    {
      "epoch": 0.3924,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.018202463785807293,
      "learning_rate": 0.0001,
      "loss": 3.83,
      "loss/crossentropy": 2.0494508743286133,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18307264149188995,
      "step": 19620
    },
    {
      "epoch": 0.39244,
      "grad_norm": 1.875,
      "grad_norm_var": 0.018277740478515624,
      "learning_rate": 0.0001,
      "loss": 3.9901,
      "loss/crossentropy": 1.683717966079712,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16951359808444977,
      "step": 19622
    },
    {
      "epoch": 0.39248,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.015197499593098959,
      "learning_rate": 0.0001,
      "loss": 3.8383,
      "loss/crossentropy": 2.62760066986084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22649705410003662,
      "step": 19624
    },
    {
      "epoch": 0.39252,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.024072011311848957,
      "learning_rate": 0.0001,
      "loss": 4.19,
      "loss/crossentropy": 1.857836663722992,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1827384978532791,
      "step": 19626
    },
    {
      "epoch": 0.39256,
      "grad_norm": 2.0,
      "grad_norm_var": 0.016108957926432292,
      "learning_rate": 0.0001,
      "loss": 3.7635,
      "loss/crossentropy": 2.1203905940055847,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20033711194992065,
      "step": 19628
    },
    {
      "epoch": 0.3926,
      "grad_norm": 2.125,
      "grad_norm_var": 0.01721165974934896,
      "learning_rate": 0.0001,
      "loss": 4.4474,
      "loss/crossentropy": 2.378189444541931,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20892338454723358,
      "step": 19630
    },
    {
      "epoch": 0.39264,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.015819295247395834,
      "learning_rate": 0.0001,
      "loss": 4.0897,
      "loss/crossentropy": 1.896558940410614,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18873175233602524,
      "step": 19632
    },
    {
      "epoch": 0.39268,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.020798492431640624,
      "learning_rate": 0.0001,
      "loss": 4.3611,
      "loss/crossentropy": 2.3503568172454834,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23128806054592133,
      "step": 19634
    },
    {
      "epoch": 0.39272,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.021345011393229165,
      "learning_rate": 0.0001,
      "loss": 4.0515,
      "loss/crossentropy": 2.081954002380371,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20059917122125626,
      "step": 19636
    },
    {
      "epoch": 0.39276,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.021512858072916665,
      "learning_rate": 0.0001,
      "loss": 4.0986,
      "loss/crossentropy": 1.9573910236358643,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1903771460056305,
      "step": 19638
    },
    {
      "epoch": 0.3928,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.022025299072265626,
      "learning_rate": 0.0001,
      "loss": 3.9333,
      "loss/crossentropy": 2.098384141921997,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19905343651771545,
      "step": 19640
    },
    {
      "epoch": 0.39284,
      "grad_norm": 2.0,
      "grad_norm_var": 0.015892537434895833,
      "learning_rate": 0.0001,
      "loss": 4.0854,
      "loss/crossentropy": 1.8510947227478027,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18955568969249725,
      "step": 19642
    },
    {
      "epoch": 0.39288,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.016440582275390626,
      "learning_rate": 0.0001,
      "loss": 4.2948,
      "loss/crossentropy": 2.2839618921279907,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23230211436748505,
      "step": 19644
    },
    {
      "epoch": 0.39292,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.014778391520182291,
      "learning_rate": 0.0001,
      "loss": 4.0112,
      "loss/crossentropy": 1.8055492639541626,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20515839755535126,
      "step": 19646
    },
    {
      "epoch": 0.39296,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.013348134358723958,
      "learning_rate": 0.0001,
      "loss": 4.0556,
      "loss/crossentropy": 1.6756115555763245,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17216521501541138,
      "step": 19648
    },
    {
      "epoch": 0.393,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.007287343343098958,
      "learning_rate": 0.0001,
      "loss": 4.4001,
      "loss/crossentropy": 2.3247755765914917,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21122244000434875,
      "step": 19650
    },
    {
      "epoch": 0.39304,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.004044596354166667,
      "learning_rate": 0.0001,
      "loss": 3.895,
      "loss/crossentropy": 1.9654970169067383,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20018797367811203,
      "step": 19652
    },
    {
      "epoch": 0.39308,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0029042561848958335,
      "learning_rate": 0.0001,
      "loss": 3.9018,
      "loss/crossentropy": 1.9368168115615845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19478464126586914,
      "step": 19654
    },
    {
      "epoch": 0.39312,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0023671468098958332,
      "learning_rate": 0.0001,
      "loss": 3.7869,
      "loss/crossentropy": 2.099206328392029,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20060646533966064,
      "step": 19656
    },
    {
      "epoch": 0.39316,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.002561187744140625,
      "learning_rate": 0.0001,
      "loss": 4.3438,
      "loss/crossentropy": 2.1585338711738586,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20852376520633698,
      "step": 19658
    },
    {
      "epoch": 0.3932,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.0044830322265625,
      "learning_rate": 0.0001,
      "loss": 4.0533,
      "loss/crossentropy": 2.285220742225647,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1983182057738304,
      "step": 19660
    },
    {
      "epoch": 0.39324,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0047190348307291664,
      "learning_rate": 0.0001,
      "loss": 4.0046,
      "loss/crossentropy": 1.955579936504364,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21165720373392105,
      "step": 19662
    },
    {
      "epoch": 0.39328,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.00645751953125,
      "learning_rate": 0.0001,
      "loss": 3.823,
      "loss/crossentropy": 1.9925233721733093,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18789846450090408,
      "step": 19664
    },
    {
      "epoch": 0.39332,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.006490071614583333,
      "learning_rate": 0.0001,
      "loss": 4.0123,
      "loss/crossentropy": 1.9267281293869019,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19545376300811768,
      "step": 19666
    },
    {
      "epoch": 0.39336,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.006493123372395834,
      "learning_rate": 0.0001,
      "loss": 4.0401,
      "loss/crossentropy": 2.179477632045746,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19430068880319595,
      "step": 19668
    },
    {
      "epoch": 0.3934,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.007228342692057291,
      "learning_rate": 0.0001,
      "loss": 4.1508,
      "loss/crossentropy": 2.1411179900169373,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2276124805212021,
      "step": 19670
    },
    {
      "epoch": 0.39344,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.014438629150390625,
      "learning_rate": 0.0001,
      "loss": 4.2689,
      "loss/crossentropy": 2.085427463054657,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21856296062469482,
      "step": 19672
    },
    {
      "epoch": 0.39348,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.014495595296223959,
      "learning_rate": 0.0001,
      "loss": 4.0302,
      "loss/crossentropy": 1.7399682402610779,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16918571293354034,
      "step": 19674
    },
    {
      "epoch": 0.39352,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.011356608072916666,
      "learning_rate": 0.0001,
      "loss": 4.0404,
      "loss/crossentropy": 1.8636209964752197,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17654836922883987,
      "step": 19676
    },
    {
      "epoch": 0.39356,
      "grad_norm": 1.7421875,
      "grad_norm_var": 0.015290323893229167,
      "learning_rate": 0.0001,
      "loss": 4.0157,
      "loss/crossentropy": 2.143825590610504,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2125091552734375,
      "step": 19678
    },
    {
      "epoch": 0.3936,
      "grad_norm": 1.6953125,
      "grad_norm_var": 0.01873753865559896,
      "learning_rate": 0.0001,
      "loss": 3.7692,
      "loss/crossentropy": 1.8222747445106506,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17713025212287903,
      "step": 19680
    },
    {
      "epoch": 0.39364,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.01969172159830729,
      "learning_rate": 0.0001,
      "loss": 3.8254,
      "loss/crossentropy": 1.7044820189476013,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17669418454170227,
      "step": 19682
    },
    {
      "epoch": 0.39368,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.020411936442057292,
      "learning_rate": 0.0001,
      "loss": 4.0987,
      "loss/crossentropy": 2.144485831260681,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20580272376537323,
      "step": 19684
    },
    {
      "epoch": 0.39372,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.019774373372395834,
      "learning_rate": 0.0001,
      "loss": 4.0994,
      "loss/crossentropy": 2.17536997795105,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21704821288585663,
      "step": 19686
    },
    {
      "epoch": 0.39376,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.012963612874348959,
      "learning_rate": 0.0001,
      "loss": 4.2592,
      "loss/crossentropy": 2.1875303983688354,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1996757537126541,
      "step": 19688
    },
    {
      "epoch": 0.3938,
      "grad_norm": 2.125,
      "grad_norm_var": 0.014121246337890626,
      "learning_rate": 0.0001,
      "loss": 4.1943,
      "loss/crossentropy": 2.270558476448059,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23406681418418884,
      "step": 19690
    },
    {
      "epoch": 0.39384,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.015860748291015626,
      "learning_rate": 0.0001,
      "loss": 4.2097,
      "loss/crossentropy": 2.0614060163497925,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21256985515356064,
      "step": 19692
    },
    {
      "epoch": 0.39388,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.015091705322265624,
      "learning_rate": 0.0001,
      "loss": 4.042,
      "loss/crossentropy": 2.1036725640296936,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19555094093084335,
      "step": 19694
    },
    {
      "epoch": 0.39392,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.010396321614583334,
      "learning_rate": 0.0001,
      "loss": 3.7702,
      "loss/crossentropy": 1.8713775277137756,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1973343789577484,
      "step": 19696
    },
    {
      "epoch": 0.39396,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.009411366780598958,
      "learning_rate": 0.0001,
      "loss": 3.9581,
      "loss/crossentropy": 1.9986143708229065,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1925104334950447,
      "step": 19698
    },
    {
      "epoch": 0.394,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0090484619140625,
      "learning_rate": 0.0001,
      "loss": 3.8134,
      "loss/crossentropy": 1.8336694836616516,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18453969806432724,
      "step": 19700
    },
    {
      "epoch": 0.39404,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.01065673828125,
      "learning_rate": 0.0001,
      "loss": 3.9279,
      "loss/crossentropy": 1.8948233723640442,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17115242034196854,
      "step": 19702
    },
    {
      "epoch": 0.39408,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.008958943684895833,
      "learning_rate": 0.0001,
      "loss": 4.244,
      "loss/crossentropy": 2.5506935119628906,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22056522965431213,
      "step": 19704
    },
    {
      "epoch": 0.39412,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.02890625,
      "learning_rate": 0.0001,
      "loss": 4.1832,
      "loss/crossentropy": 2.127421021461487,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20175430178642273,
      "step": 19706
    },
    {
      "epoch": 0.39416,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.027197265625,
      "learning_rate": 0.0001,
      "loss": 4.0251,
      "loss/crossentropy": 2.246693968772888,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22178302705287933,
      "step": 19708
    },
    {
      "epoch": 0.3942,
      "grad_norm": 1.6953125,
      "grad_norm_var": 0.03328221638997396,
      "learning_rate": 0.0001,
      "loss": 3.9336,
      "loss/crossentropy": 1.970005750656128,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18967190384864807,
      "step": 19710
    },
    {
      "epoch": 0.39424,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.032956695556640624,
      "learning_rate": 0.0001,
      "loss": 4.1035,
      "loss/crossentropy": 1.944337785243988,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1952769234776497,
      "step": 19712
    },
    {
      "epoch": 0.39428,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0328277587890625,
      "learning_rate": 0.0001,
      "loss": 4.2289,
      "loss/crossentropy": 2.291213870048523,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2077646553516388,
      "step": 19714
    },
    {
      "epoch": 0.39432,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.03361790974934896,
      "learning_rate": 0.0001,
      "loss": 4.0433,
      "loss/crossentropy": 2.404030203819275,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2200116366147995,
      "step": 19716
    },
    {
      "epoch": 0.39436,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.032364908854166666,
      "learning_rate": 0.0001,
      "loss": 3.8609,
      "loss/crossentropy": 1.9242625832557678,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19318503141403198,
      "step": 19718
    },
    {
      "epoch": 0.3944,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.03240966796875,
      "learning_rate": 0.0001,
      "loss": 3.9789,
      "loss/crossentropy": 2.319575071334839,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21668671071529388,
      "step": 19720
    },
    {
      "epoch": 0.39444,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.012287394205729166,
      "learning_rate": 0.0001,
      "loss": 4.2219,
      "loss/crossentropy": 2.0032835006713867,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19786667823791504,
      "step": 19722
    },
    {
      "epoch": 0.39448,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.0146636962890625,
      "learning_rate": 0.0001,
      "loss": 3.9509,
      "loss/crossentropy": 1.9524562358856201,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18965402245521545,
      "step": 19724
    },
    {
      "epoch": 0.39452,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.007521311442057292,
      "learning_rate": 0.0001,
      "loss": 3.5677,
      "loss/crossentropy": 1.5887231826782227,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1512351706624031,
      "step": 19726
    },
    {
      "epoch": 0.39456,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.008576456705729167,
      "learning_rate": 0.0001,
      "loss": 4.1379,
      "loss/crossentropy": 2.107685923576355,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18645642697811127,
      "step": 19728
    },
    {
      "epoch": 0.3946,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.009028879801432292,
      "learning_rate": 0.0001,
      "loss": 3.9594,
      "loss/crossentropy": 2.3551766872406006,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2207053080201149,
      "step": 19730
    },
    {
      "epoch": 0.39464,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.010847727457682291,
      "learning_rate": 0.0001,
      "loss": 4.3542,
      "loss/crossentropy": 2.1983554363250732,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23001667857170105,
      "step": 19732
    },
    {
      "epoch": 0.39468,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009639485677083334,
      "learning_rate": 0.0001,
      "loss": 4.3158,
      "loss/crossentropy": 2.0236815214157104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19230765849351883,
      "step": 19734
    },
    {
      "epoch": 0.39472,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.010529581705729167,
      "learning_rate": 0.0001,
      "loss": 4.0511,
      "loss/crossentropy": 2.2027645111083984,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1864710971713066,
      "step": 19736
    },
    {
      "epoch": 0.39476,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.013152821858723959,
      "learning_rate": 0.0001,
      "loss": 4.008,
      "loss/crossentropy": 1.921549379825592,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21788031607866287,
      "step": 19738
    },
    {
      "epoch": 0.3948,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.010416666666666666,
      "learning_rate": 0.0001,
      "loss": 3.8353,
      "loss/crossentropy": 2.121790587902069,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21380367130041122,
      "step": 19740
    },
    {
      "epoch": 0.39484,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.008217112223307291,
      "learning_rate": 0.0001,
      "loss": 3.9975,
      "loss/crossentropy": 1.9595564007759094,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17919503152370453,
      "step": 19742
    },
    {
      "epoch": 0.39488,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.010993448893229167,
      "learning_rate": 0.0001,
      "loss": 3.9935,
      "loss/crossentropy": 2.155984342098236,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2002984657883644,
      "step": 19744
    },
    {
      "epoch": 0.39492,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.012589518229166667,
      "learning_rate": 0.0001,
      "loss": 3.875,
      "loss/crossentropy": 1.766005277633667,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1717890352010727,
      "step": 19746
    },
    {
      "epoch": 0.39496,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010424550374348958,
      "learning_rate": 0.0001,
      "loss": 3.9178,
      "loss/crossentropy": 1.8891428112983704,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1965286061167717,
      "step": 19748
    },
    {
      "epoch": 0.395,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.010343170166015625,
      "learning_rate": 0.0001,
      "loss": 4.3124,
      "loss/crossentropy": 1.9521282315254211,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21058151125907898,
      "step": 19750
    },
    {
      "epoch": 0.39504,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.014711252848307292,
      "learning_rate": 0.0001,
      "loss": 4.2242,
      "loss/crossentropy": 2.3142699003219604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24145027250051498,
      "step": 19752
    },
    {
      "epoch": 0.39508,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.011777496337890625,
      "learning_rate": 0.0001,
      "loss": 4.1059,
      "loss/crossentropy": 2.0375224351882935,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19370558112859726,
      "step": 19754
    },
    {
      "epoch": 0.39512,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.011962890625,
      "learning_rate": 0.0001,
      "loss": 4.2367,
      "loss/crossentropy": 2.193490743637085,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2012534961104393,
      "step": 19756
    },
    {
      "epoch": 0.39516,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011993153889973959,
      "learning_rate": 0.0001,
      "loss": 4.0781,
      "loss/crossentropy": 2.2397992610931396,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19738413393497467,
      "step": 19758
    },
    {
      "epoch": 0.3952,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.010223134358723959,
      "learning_rate": 0.0001,
      "loss": 3.9153,
      "loss/crossentropy": 1.7602161169052124,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18599014729261398,
      "step": 19760
    },
    {
      "epoch": 0.39524,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.008640289306640625,
      "learning_rate": 0.0001,
      "loss": 4.182,
      "loss/crossentropy": 2.536779046058655,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2301289290189743,
      "step": 19762
    },
    {
      "epoch": 0.39528,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.008965810139973959,
      "learning_rate": 0.0001,
      "loss": 3.9999,
      "loss/crossentropy": 2.3237764835357666,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21476060152053833,
      "step": 19764
    },
    {
      "epoch": 0.39532,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.011107381184895833,
      "learning_rate": 0.0001,
      "loss": 4.2102,
      "loss/crossentropy": 2.370723605155945,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24277979880571365,
      "step": 19766
    },
    {
      "epoch": 0.39536,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.006917063395182292,
      "learning_rate": 0.0001,
      "loss": 4.0838,
      "loss/crossentropy": 1.6399320363998413,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1740519106388092,
      "step": 19768
    },
    {
      "epoch": 0.3954,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.02394383748372396,
      "learning_rate": 0.0001,
      "loss": 3.6633,
      "loss/crossentropy": 1.8633801341056824,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1810051053762436,
      "step": 19770
    },
    {
      "epoch": 0.39544,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.023583984375,
      "learning_rate": 0.0001,
      "loss": 4.1102,
      "loss/crossentropy": 2.318315625190735,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21214767545461655,
      "step": 19772
    },
    {
      "epoch": 0.39548,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.026387532552083332,
      "learning_rate": 0.0001,
      "loss": 4.1232,
      "loss/crossentropy": 2.2803520560264587,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21136894822120667,
      "step": 19774
    },
    {
      "epoch": 0.39552,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.026387532552083332,
      "learning_rate": 0.0001,
      "loss": 3.9034,
      "loss/crossentropy": 2.1529780626296997,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18172463029623032,
      "step": 19776
    },
    {
      "epoch": 0.39556,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.025585683186848958,
      "learning_rate": 0.0001,
      "loss": 4.2851,
      "loss/crossentropy": 2.1246083974838257,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2181987464427948,
      "step": 19778
    },
    {
      "epoch": 0.3956,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.029515584309895832,
      "learning_rate": 0.0001,
      "loss": 4.0097,
      "loss/crossentropy": 2.027298629283905,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1877325028181076,
      "step": 19780
    },
    {
      "epoch": 0.39564,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0288482666015625,
      "learning_rate": 0.0001,
      "loss": 4.2326,
      "loss/crossentropy": 1.9338072538375854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1746418997645378,
      "step": 19782
    },
    {
      "epoch": 0.39568,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.03104222615559896,
      "learning_rate": 0.0001,
      "loss": 4.2551,
      "loss/crossentropy": 2.050130307674408,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2011866271495819,
      "step": 19784
    },
    {
      "epoch": 0.39572,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.014720662434895834,
      "learning_rate": 0.0001,
      "loss": 4.3227,
      "loss/crossentropy": 2.48315691947937,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22591856867074966,
      "step": 19786
    },
    {
      "epoch": 0.39576,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.021019490559895833,
      "learning_rate": 0.0001,
      "loss": 4.3671,
      "loss/crossentropy": 2.578279137611389,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2216324657201767,
      "step": 19788
    },
    {
      "epoch": 0.3958,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.018192291259765625,
      "learning_rate": 0.0001,
      "loss": 4.259,
      "loss/crossentropy": 1.7862395644187927,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17009516060352325,
      "step": 19790
    },
    {
      "epoch": 0.39584,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.017032877604166666,
      "learning_rate": 0.0001,
      "loss": 3.8506,
      "loss/crossentropy": 2.042613208293915,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1844482272863388,
      "step": 19792
    },
    {
      "epoch": 0.39588,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.020926666259765626,
      "learning_rate": 0.0001,
      "loss": 4.1919,
      "loss/crossentropy": 2.317251443862915,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2135745733976364,
      "step": 19794
    },
    {
      "epoch": 0.39592,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.01964111328125,
      "learning_rate": 0.0001,
      "loss": 4.0061,
      "loss/crossentropy": 1.849199891090393,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20549577474594116,
      "step": 19796
    },
    {
      "epoch": 0.39596,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0199859619140625,
      "learning_rate": 0.0001,
      "loss": 4.0551,
      "loss/crossentropy": 2.1876507997512817,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20098386704921722,
      "step": 19798
    },
    {
      "epoch": 0.396,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.01718317667643229,
      "learning_rate": 0.0001,
      "loss": 4.154,
      "loss/crossentropy": 1.9740530848503113,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19494586437940598,
      "step": 19800
    },
    {
      "epoch": 0.39604,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.017577107747395834,
      "learning_rate": 0.0001,
      "loss": 4.1405,
      "loss/crossentropy": 2.208943724632263,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22833283245563507,
      "step": 19802
    },
    {
      "epoch": 0.39608,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.009488677978515625,
      "learning_rate": 0.0001,
      "loss": 3.9082,
      "loss/crossentropy": 2.024399518966675,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19786083698272705,
      "step": 19804
    },
    {
      "epoch": 0.39612,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009356435139973958,
      "learning_rate": 0.0001,
      "loss": 4.1475,
      "loss/crossentropy": 2.1283441185951233,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21942409127950668,
      "step": 19806
    },
    {
      "epoch": 0.39616,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.01080322265625,
      "learning_rate": 0.0001,
      "loss": 4.0101,
      "loss/crossentropy": 2.3675626516342163,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22466859221458435,
      "step": 19808
    },
    {
      "epoch": 0.3962,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0082672119140625,
      "learning_rate": 0.0001,
      "loss": 3.8682,
      "loss/crossentropy": 1.886322796344757,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1904991939663887,
      "step": 19810
    },
    {
      "epoch": 0.39624,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.005012003580729166,
      "learning_rate": 0.0001,
      "loss": 3.9071,
      "loss/crossentropy": 2.098154127597809,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2065964937210083,
      "step": 19812
    },
    {
      "epoch": 0.39628,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.006788889567057292,
      "learning_rate": 0.0001,
      "loss": 3.8447,
      "loss/crossentropy": 2.0527199506759644,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19717100262641907,
      "step": 19814
    },
    {
      "epoch": 0.39632,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.011244455973307291,
      "learning_rate": 0.0001,
      "loss": 4.2692,
      "loss/crossentropy": 2.0650912523269653,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2021411582827568,
      "step": 19816
    },
    {
      "epoch": 0.39636,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.018700154622395833,
      "learning_rate": 0.0001,
      "loss": 4.3611,
      "loss/crossentropy": 2.2239702939987183,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19744005054235458,
      "step": 19818
    },
    {
      "epoch": 0.3964,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.020369211832682293,
      "learning_rate": 0.0001,
      "loss": 3.9029,
      "loss/crossentropy": 2.123336434364319,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18976306170225143,
      "step": 19820
    },
    {
      "epoch": 0.39644,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.02154515584309896,
      "learning_rate": 0.0001,
      "loss": 3.9966,
      "loss/crossentropy": 2.121657133102417,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2159118503332138,
      "step": 19822
    },
    {
      "epoch": 0.39648,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0185455322265625,
      "learning_rate": 0.0001,
      "loss": 3.7621,
      "loss/crossentropy": 1.7323416471481323,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18072299659252167,
      "step": 19824
    },
    {
      "epoch": 0.39652,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.020243072509765626,
      "learning_rate": 0.0001,
      "loss": 4.0183,
      "loss/crossentropy": 2.181841015815735,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21452204138040543,
      "step": 19826
    },
    {
      "epoch": 0.39656,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.02104670206705729,
      "learning_rate": 0.0001,
      "loss": 4.1181,
      "loss/crossentropy": 1.9022215008735657,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18924792110919952,
      "step": 19828
    },
    {
      "epoch": 0.3966,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.018317667643229167,
      "learning_rate": 0.0001,
      "loss": 4.2505,
      "loss/crossentropy": 2.1240022778511047,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18618234246969223,
      "step": 19830
    },
    {
      "epoch": 0.39664,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.015710194905598957,
      "learning_rate": 0.0001,
      "loss": 3.8611,
      "loss/crossentropy": 1.5959683060646057,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17317308485507965,
      "step": 19832
    },
    {
      "epoch": 0.39668,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.007682291666666666,
      "learning_rate": 0.0001,
      "loss": 4.1281,
      "loss/crossentropy": 2.1812866926193237,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.192174032330513,
      "step": 19834
    },
    {
      "epoch": 0.39672,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.010651652018229167,
      "learning_rate": 0.0001,
      "loss": 4.188,
      "loss/crossentropy": 1.8964568972587585,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20922592282295227,
      "step": 19836
    },
    {
      "epoch": 0.39676,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.012013498942057292,
      "learning_rate": 0.0001,
      "loss": 4.1369,
      "loss/crossentropy": 2.0822665691375732,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18019618093967438,
      "step": 19838
    },
    {
      "epoch": 0.3968,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.010985310872395833,
      "learning_rate": 0.0001,
      "loss": 4.0271,
      "loss/crossentropy": 2.116270899772644,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21604043990373611,
      "step": 19840
    },
    {
      "epoch": 0.39684,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.007759602864583334,
      "learning_rate": 0.0001,
      "loss": 3.9132,
      "loss/crossentropy": 1.7761988639831543,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20087965577840805,
      "step": 19842
    },
    {
      "epoch": 0.39688,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.008649698893229167,
      "learning_rate": 0.0001,
      "loss": 4.0187,
      "loss/crossentropy": 1.6091360449790955,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18282026052474976,
      "step": 19844
    },
    {
      "epoch": 0.39692,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.009547678629557292,
      "learning_rate": 0.0001,
      "loss": 4.4441,
      "loss/crossentropy": 2.6394678354263306,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20579150319099426,
      "step": 19846
    },
    {
      "epoch": 0.39696,
      "grad_norm": 2.0,
      "grad_norm_var": 0.010011545817057292,
      "learning_rate": 0.0001,
      "loss": 4.1778,
      "loss/crossentropy": 2.0911704897880554,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20075388252735138,
      "step": 19848
    },
    {
      "epoch": 0.397,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.010139719645182291,
      "learning_rate": 0.0001,
      "loss": 4.3458,
      "loss/crossentropy": 2.1299456357955933,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2067631408572197,
      "step": 19850
    },
    {
      "epoch": 0.39704,
      "grad_norm": 1.875,
      "grad_norm_var": 0.004937489827473958,
      "learning_rate": 0.0001,
      "loss": 3.9037,
      "loss/crossentropy": 1.9630563855171204,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.187811940908432,
      "step": 19852
    },
    {
      "epoch": 0.39708,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.004808553059895833,
      "learning_rate": 0.0001,
      "loss": 4.1681,
      "loss/crossentropy": 2.1914453506469727,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2064526304602623,
      "step": 19854
    },
    {
      "epoch": 0.39712,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.012412261962890626,
      "learning_rate": 0.0001,
      "loss": 4.1404,
      "loss/crossentropy": 2.3003474473953247,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20841488242149353,
      "step": 19856
    },
    {
      "epoch": 0.39716,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.01682306925455729,
      "learning_rate": 0.0001,
      "loss": 3.96,
      "loss/crossentropy": 2.1625255346298218,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20437289774417877,
      "step": 19858
    },
    {
      "epoch": 0.3972,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.015868123372395834,
      "learning_rate": 0.0001,
      "loss": 3.8594,
      "loss/crossentropy": 1.6742416620254517,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16787201166152954,
      "step": 19860
    },
    {
      "epoch": 0.39724,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.01546630859375,
      "learning_rate": 0.0001,
      "loss": 3.8889,
      "loss/crossentropy": 2.0682146549224854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1715894490480423,
      "step": 19862
    },
    {
      "epoch": 0.39728,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.017071278889973958,
      "learning_rate": 0.0001,
      "loss": 4.0319,
      "loss/crossentropy": 1.99592924118042,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19570383429527283,
      "step": 19864
    },
    {
      "epoch": 0.39732,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.01967137654622396,
      "learning_rate": 0.0001,
      "loss": 4.0123,
      "loss/crossentropy": 2.086844265460968,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2123936414718628,
      "step": 19866
    },
    {
      "epoch": 0.39736,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.017256673177083334,
      "learning_rate": 0.0001,
      "loss": 4.2245,
      "loss/crossentropy": 1.7985658645629883,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18117651343345642,
      "step": 19868
    },
    {
      "epoch": 0.3974,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.018949381510416665,
      "learning_rate": 0.0001,
      "loss": 3.8449,
      "loss/crossentropy": 1.9804646372795105,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18682067096233368,
      "step": 19870
    },
    {
      "epoch": 0.39744,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.01546630859375,
      "learning_rate": 0.0001,
      "loss": 4.0692,
      "loss/crossentropy": 1.9027678966522217,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18195898830890656,
      "step": 19872
    },
    {
      "epoch": 0.39748,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.01660944620768229,
      "learning_rate": 0.0001,
      "loss": 4.1191,
      "loss/crossentropy": 2.3370686769485474,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2376203015446663,
      "step": 19874
    },
    {
      "epoch": 0.39752,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.018363189697265626,
      "learning_rate": 0.0001,
      "loss": 3.8472,
      "loss/crossentropy": 1.7463279366493225,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16485755145549774,
      "step": 19876
    },
    {
      "epoch": 0.39756,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.017967732747395833,
      "learning_rate": 0.0001,
      "loss": 4.4039,
      "loss/crossentropy": 2.1171544194221497,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20631500333547592,
      "step": 19878
    },
    {
      "epoch": 0.3976,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.016733551025390626,
      "learning_rate": 0.0001,
      "loss": 3.7242,
      "loss/crossentropy": 2.1109176874160767,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18512246757745743,
      "step": 19880
    },
    {
      "epoch": 0.39764,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.013618977864583333,
      "learning_rate": 0.0001,
      "loss": 3.6843,
      "loss/crossentropy": 1.927691638469696,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19118069112300873,
      "step": 19882
    },
    {
      "epoch": 0.39768,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.01337890625,
      "learning_rate": 0.0001,
      "loss": 3.9539,
      "loss/crossentropy": 2.041896104812622,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21281379461288452,
      "step": 19884
    },
    {
      "epoch": 0.39772,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.013272857666015625,
      "learning_rate": 0.0001,
      "loss": 3.9146,
      "loss/crossentropy": 1.8779407739639282,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.180915005505085,
      "step": 19886
    },
    {
      "epoch": 0.39776,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.019001261393229166,
      "learning_rate": 0.0001,
      "loss": 4.2158,
      "loss/crossentropy": 1.9899646639823914,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19102467596530914,
      "step": 19888
    },
    {
      "epoch": 0.3978,
      "grad_norm": 2.0,
      "grad_norm_var": 0.012562815348307292,
      "learning_rate": 0.0001,
      "loss": 4.1822,
      "loss/crossentropy": 2.3294299840927124,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20739557594060898,
      "step": 19890
    },
    {
      "epoch": 0.39784,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.011864980061848959,
      "learning_rate": 0.0001,
      "loss": 4.2829,
      "loss/crossentropy": 2.2126539945602417,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23066286742687225,
      "step": 19892
    },
    {
      "epoch": 0.39788,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.016454060872395832,
      "learning_rate": 0.0001,
      "loss": 3.8665,
      "loss/crossentropy": 1.9949330687522888,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19172564148902893,
      "step": 19894
    },
    {
      "epoch": 0.39792,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.02158177693684896,
      "learning_rate": 0.0001,
      "loss": 3.9567,
      "loss/crossentropy": 2.0896310210227966,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19692128896713257,
      "step": 19896
    },
    {
      "epoch": 0.39796,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.020310211181640624,
      "learning_rate": 0.0001,
      "loss": 4.1856,
      "loss/crossentropy": 2.191789746284485,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2073511779308319,
      "step": 19898
    },
    {
      "epoch": 0.398,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.022415924072265624,
      "learning_rate": 0.0001,
      "loss": 3.9965,
      "loss/crossentropy": 1.9925153255462646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20123805850744247,
      "step": 19900
    },
    {
      "epoch": 0.39804,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.021476236979166667,
      "learning_rate": 0.0001,
      "loss": 3.7917,
      "loss/crossentropy": 1.8903232216835022,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18577788770198822,
      "step": 19902
    },
    {
      "epoch": 0.39808,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.01817804972330729,
      "learning_rate": 0.0001,
      "loss": 4.2818,
      "loss/crossentropy": 2.055288314819336,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19745147973299026,
      "step": 19904
    },
    {
      "epoch": 0.39812,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.018570709228515624,
      "learning_rate": 0.0001,
      "loss": 4.196,
      "loss/crossentropy": 1.7709746360778809,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17147859930992126,
      "step": 19906
    },
    {
      "epoch": 0.39816,
      "grad_norm": 2.0,
      "grad_norm_var": 0.01843846638997396,
      "learning_rate": 0.0001,
      "loss": 4.182,
      "loss/crossentropy": 2.3295364379882812,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20680927485227585,
      "step": 19908
    },
    {
      "epoch": 0.3982,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.014414215087890625,
      "learning_rate": 0.0001,
      "loss": 4.1633,
      "loss/crossentropy": 1.9828099608421326,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.202943354845047,
      "step": 19910
    },
    {
      "epoch": 0.39824,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.009352366129557291,
      "learning_rate": 0.0001,
      "loss": 4.2294,
      "loss/crossentropy": 2.140046715736389,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1967788189649582,
      "step": 19912
    },
    {
      "epoch": 0.39828,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.008736165364583333,
      "learning_rate": 0.0001,
      "loss": 4.1673,
      "loss/crossentropy": 2.3365899324417114,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2261614426970482,
      "step": 19914
    },
    {
      "epoch": 0.39832,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0075642903645833336,
      "learning_rate": 0.0001,
      "loss": 3.8716,
      "loss/crossentropy": 2.101326584815979,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19668899476528168,
      "step": 19916
    },
    {
      "epoch": 0.39836,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.005037180582682292,
      "learning_rate": 0.0001,
      "loss": 4.1572,
      "loss/crossentropy": 1.9804238080978394,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2117360234260559,
      "step": 19918
    },
    {
      "epoch": 0.3984,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0037676493326822915,
      "learning_rate": 0.0001,
      "loss": 3.945,
      "loss/crossentropy": 1.858399510383606,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20013604313135147,
      "step": 19920
    },
    {
      "epoch": 0.39844,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.005832672119140625,
      "learning_rate": 0.0001,
      "loss": 4.0302,
      "loss/crossentropy": 2.087529957294464,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19001878052949905,
      "step": 19922
    },
    {
      "epoch": 0.39848,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.0068267822265625,
      "learning_rate": 0.0001,
      "loss": 3.9425,
      "loss/crossentropy": 2.2521530389785767,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20182852447032928,
      "step": 19924
    },
    {
      "epoch": 0.39852,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.00784912109375,
      "learning_rate": 0.0001,
      "loss": 4.3362,
      "loss/crossentropy": 2.370557188987732,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27234383672475815,
      "step": 19926
    },
    {
      "epoch": 0.39856,
      "grad_norm": 13.0625,
      "grad_norm_var": 7.6871192932128904,
      "learning_rate": 0.0001,
      "loss": 4.0383,
      "loss/crossentropy": 2.18235445022583,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20694056898355484,
      "step": 19928
    },
    {
      "epoch": 0.3986,
      "grad_norm": 2.125,
      "grad_norm_var": 7.659780883789063,
      "learning_rate": 0.0001,
      "loss": 3.8946,
      "loss/crossentropy": 1.9220272898674011,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21540776640176773,
      "step": 19930
    },
    {
      "epoch": 0.39864,
      "grad_norm": 2.015625,
      "grad_norm_var": 7.6551513671875,
      "learning_rate": 0.0001,
      "loss": 4.015,
      "loss/crossentropy": 2.064575970172882,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20172829926013947,
      "step": 19932
    },
    {
      "epoch": 0.39868,
      "grad_norm": 2.296875,
      "grad_norm_var": 7.620402018229167,
      "learning_rate": 0.0001,
      "loss": 4.5782,
      "loss/crossentropy": 2.1824090480804443,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23097260296344757,
      "step": 19934
    },
    {
      "epoch": 0.39872,
      "grad_norm": 2.0,
      "grad_norm_var": 7.594489542643229,
      "learning_rate": 0.0001,
      "loss": 3.9227,
      "loss/crossentropy": 2.211961567401886,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2150636538863182,
      "step": 19936
    },
    {
      "epoch": 0.39876,
      "grad_norm": 2.4375,
      "grad_norm_var": 7.586128743489583,
      "learning_rate": 0.0001,
      "loss": 4.1724,
      "loss/crossentropy": 1.8789254426956177,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18855369836091995,
      "step": 19938
    },
    {
      "epoch": 0.3988,
      "grad_norm": 2.03125,
      "grad_norm_var": 7.576968383789063,
      "learning_rate": 0.0001,
      "loss": 4.212,
      "loss/crossentropy": 2.322842240333557,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19432100653648376,
      "step": 19940
    },
    {
      "epoch": 0.39884,
      "grad_norm": 1.9375,
      "grad_norm_var": 7.590311686197917,
      "learning_rate": 0.0001,
      "loss": 4.2201,
      "loss/crossentropy": 2.1524535417556763,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19931253790855408,
      "step": 19942
    },
    {
      "epoch": 0.39888,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.0280670166015625,
      "learning_rate": 0.0001,
      "loss": 4.0166,
      "loss/crossentropy": 2.0835599303245544,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18674355000257492,
      "step": 19944
    },
    {
      "epoch": 0.39892,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.026718902587890624,
      "learning_rate": 0.0001,
      "loss": 3.9398,
      "loss/crossentropy": 1.8204763531684875,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1970055252313614,
      "step": 19946
    },
    {
      "epoch": 0.39896,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.02769953409830729,
      "learning_rate": 0.0001,
      "loss": 3.8403,
      "loss/crossentropy": 2.1432100534439087,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1960032731294632,
      "step": 19948
    },
    {
      "epoch": 0.399,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0227294921875,
      "learning_rate": 0.0001,
      "loss": 4.1183,
      "loss/crossentropy": 1.9959533214569092,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18608924746513367,
      "step": 19950
    },
    {
      "epoch": 0.39904,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.02399266560872396,
      "learning_rate": 0.0001,
      "loss": 4.0235,
      "loss/crossentropy": 1.9874342679977417,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18600600212812424,
      "step": 19952
    },
    {
      "epoch": 0.39908,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.00849609375,
      "learning_rate": 0.0001,
      "loss": 4.2266,
      "loss/crossentropy": 2.5082361698150635,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22386416047811508,
      "step": 19954
    },
    {
      "epoch": 0.39912,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.008153279622395834,
      "learning_rate": 0.0001,
      "loss": 3.9735,
      "loss/crossentropy": 2.1550523042678833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2169199213385582,
      "step": 19956
    },
    {
      "epoch": 0.39916,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.008172353108723959,
      "learning_rate": 0.0001,
      "loss": 3.8834,
      "loss/crossentropy": 2.079386830329895,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19080224633216858,
      "step": 19958
    },
    {
      "epoch": 0.3992,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.006030019124348958,
      "learning_rate": 0.0001,
      "loss": 3.973,
      "loss/crossentropy": 1.8762348890304565,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1813136711716652,
      "step": 19960
    },
    {
      "epoch": 0.39924,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.006455230712890625,
      "learning_rate": 0.0001,
      "loss": 4.0965,
      "loss/crossentropy": 1.789110004901886,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17166541516780853,
      "step": 19962
    },
    {
      "epoch": 0.39928,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.009056599934895833,
      "learning_rate": 0.0001,
      "loss": 4.0912,
      "loss/crossentropy": 2.0196239948272705,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20005135238170624,
      "step": 19964
    },
    {
      "epoch": 0.39932,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0080718994140625,
      "learning_rate": 0.0001,
      "loss": 4.2787,
      "loss/crossentropy": 2.192206025123596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1979481726884842,
      "step": 19966
    },
    {
      "epoch": 0.39936,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.007157135009765625,
      "learning_rate": 0.0001,
      "loss": 4.3155,
      "loss/crossentropy": 2.263342499732971,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2187722995877266,
      "step": 19968
    },
    {
      "epoch": 0.3994,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.008161417643229167,
      "learning_rate": 0.0001,
      "loss": 4.2727,
      "loss/crossentropy": 2.3387625217437744,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21375280618667603,
      "step": 19970
    },
    {
      "epoch": 0.39944,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.008733876546223958,
      "learning_rate": 0.0001,
      "loss": 4.0442,
      "loss/crossentropy": 1.5837730765342712,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17303457856178284,
      "step": 19972
    },
    {
      "epoch": 0.39948,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.008678944905598958,
      "learning_rate": 0.0001,
      "loss": 3.8398,
      "loss/crossentropy": 1.9090858697891235,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1838761642575264,
      "step": 19974
    },
    {
      "epoch": 0.39952,
      "grad_norm": 1.875,
      "grad_norm_var": 0.008780924479166667,
      "learning_rate": 0.0001,
      "loss": 3.9345,
      "loss/crossentropy": 2.107620596885681,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19132380187511444,
      "step": 19976
    },
    {
      "epoch": 0.39956,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.009200032552083333,
      "learning_rate": 0.0001,
      "loss": 4.0709,
      "loss/crossentropy": 2.1706109046936035,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19275012612342834,
      "step": 19978
    },
    {
      "epoch": 0.3996,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.007356516520182292,
      "learning_rate": 0.0001,
      "loss": 3.8486,
      "loss/crossentropy": 2.0092907547950745,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18047627061605453,
      "step": 19980
    },
    {
      "epoch": 0.39964,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.025349934895833332,
      "learning_rate": 0.0001,
      "loss": 4.2226,
      "loss/crossentropy": 2.1216511726379395,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20591023564338684,
      "step": 19982
    },
    {
      "epoch": 0.39968,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.02671076456705729,
      "learning_rate": 0.0001,
      "loss": 3.9895,
      "loss/crossentropy": 2.3002817630767822,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20734255760908127,
      "step": 19984
    },
    {
      "epoch": 0.39972,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.02664972941080729,
      "learning_rate": 0.0001,
      "loss": 4.1494,
      "loss/crossentropy": 2.3097801208496094,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20194757729768753,
      "step": 19986
    },
    {
      "epoch": 0.39976,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.029130045572916666,
      "learning_rate": 0.0001,
      "loss": 3.9706,
      "loss/crossentropy": 2.2645580768585205,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21866093575954437,
      "step": 19988
    },
    {
      "epoch": 0.3998,
      "grad_norm": 1.875,
      "grad_norm_var": 0.03004150390625,
      "learning_rate": 0.0001,
      "loss": 4.0634,
      "loss/crossentropy": 2.224185347557068,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19011163711547852,
      "step": 19990
    },
    {
      "epoch": 0.39984,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.030248006184895832,
      "learning_rate": 0.0001,
      "loss": 3.9552,
      "loss/crossentropy": 1.958588182926178,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18359722197055817,
      "step": 19992
    },
    {
      "epoch": 0.39988,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.029361724853515625,
      "learning_rate": 0.0001,
      "loss": 4.0449,
      "loss/crossentropy": 2.5081194639205933,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22576630860567093,
      "step": 19994
    },
    {
      "epoch": 0.39992,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.0306549072265625,
      "learning_rate": 0.0001,
      "loss": 3.8833,
      "loss/crossentropy": 1.9870144724845886,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19262682646512985,
      "step": 19996
    },
    {
      "epoch": 0.39996,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.0118560791015625,
      "learning_rate": 0.0001,
      "loss": 4.1321,
      "loss/crossentropy": 2.227096438407898,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2056111991405487,
      "step": 19998
    },
    {
      "epoch": 0.4,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.012837473551432292,
      "learning_rate": 0.0001,
      "loss": 4.1659,
      "loss/crossentropy": 1.8458907008171082,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18857256323099136,
      "step": 20000
    },
    {
      "epoch": 0.40004,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.010536448160807291,
      "learning_rate": 0.0001,
      "loss": 4.1497,
      "loss/crossentropy": 2.184568405151367,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20918434113264084,
      "step": 20002
    },
    {
      "epoch": 0.40008,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.007136027018229167,
      "learning_rate": 0.0001,
      "loss": 3.8604,
      "loss/crossentropy": 2.046003818511963,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17669742554426193,
      "step": 20004
    },
    {
      "epoch": 0.40012,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.006151326497395833,
      "learning_rate": 0.0001,
      "loss": 4.1618,
      "loss/crossentropy": 2.330680012702942,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2159157246351242,
      "step": 20006
    },
    {
      "epoch": 0.40016,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.005973052978515625,
      "learning_rate": 0.0001,
      "loss": 3.643,
      "loss/crossentropy": 1.9141735434532166,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17788738757371902,
      "step": 20008
    },
    {
      "epoch": 0.4002,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.006589508056640625,
      "learning_rate": 0.0001,
      "loss": 4.1782,
      "loss/crossentropy": 2.3000227212905884,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21896196901798248,
      "step": 20010
    },
    {
      "epoch": 0.40024,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0067535400390625,
      "learning_rate": 0.0001,
      "loss": 4.2057,
      "loss/crossentropy": 2.151344060897827,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20495032519102097,
      "step": 20012
    },
    {
      "epoch": 0.40028,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.007911936442057291,
      "learning_rate": 0.0001,
      "loss": 3.9086,
      "loss/crossentropy": 2.1465260982513428,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1846243441104889,
      "step": 20014
    },
    {
      "epoch": 0.40032,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.006453450520833333,
      "learning_rate": 0.0001,
      "loss": 4.0024,
      "loss/crossentropy": 1.662541925907135,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18223944306373596,
      "step": 20016
    },
    {
      "epoch": 0.40036,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.006418609619140625,
      "learning_rate": 0.0001,
      "loss": 4.0052,
      "loss/crossentropy": 1.9482674598693848,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18713432550430298,
      "step": 20018
    },
    {
      "epoch": 0.4004,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0058837890625,
      "learning_rate": 0.0001,
      "loss": 3.7834,
      "loss/crossentropy": 1.7436261773109436,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18203437328338623,
      "step": 20020
    },
    {
      "epoch": 0.40044,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.005716705322265625,
      "learning_rate": 0.0001,
      "loss": 3.8905,
      "loss/crossentropy": 2.0031047463417053,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20406989008188248,
      "step": 20022
    },
    {
      "epoch": 0.40048,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.005415852864583333,
      "learning_rate": 0.0001,
      "loss": 3.8527,
      "loss/crossentropy": 1.9428189992904663,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17989446222782135,
      "step": 20024
    },
    {
      "epoch": 0.40052,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.005208333333333333,
      "learning_rate": 0.0001,
      "loss": 3.9674,
      "loss/crossentropy": 2.021029829978943,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1908131018280983,
      "step": 20026
    },
    {
      "epoch": 0.40056,
      "grad_norm": 2.0,
      "grad_norm_var": 0.02068049112955729,
      "learning_rate": 0.0001,
      "loss": 4.3471,
      "loss/crossentropy": 1.8795804381370544,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17839516699314117,
      "step": 20028
    },
    {
      "epoch": 0.4006,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.022823079427083334,
      "learning_rate": 0.0001,
      "loss": 4.4826,
      "loss/crossentropy": 1.9421144723892212,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18655938655138016,
      "step": 20030
    },
    {
      "epoch": 0.40064,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.022809855143229165,
      "learning_rate": 0.0001,
      "loss": 3.9615,
      "loss/crossentropy": 2.1828919649124146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21237139403820038,
      "step": 20032
    },
    {
      "epoch": 0.40068,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.02103271484375,
      "learning_rate": 0.0001,
      "loss": 4.0375,
      "loss/crossentropy": 1.9509565830230713,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19763849675655365,
      "step": 20034
    },
    {
      "epoch": 0.40072,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.022507476806640624,
      "learning_rate": 0.0001,
      "loss": 4.0011,
      "loss/crossentropy": 2.1327446699142456,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2048783376812935,
      "step": 20036
    },
    {
      "epoch": 0.40076,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.022904205322265624,
      "learning_rate": 0.0001,
      "loss": 4.2094,
      "loss/crossentropy": 2.3295645713806152,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22039268165826797,
      "step": 20038
    },
    {
      "epoch": 0.4008,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.03260879516601563,
      "learning_rate": 0.0001,
      "loss": 3.7464,
      "loss/crossentropy": 1.786357820034027,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1681244671344757,
      "step": 20040
    },
    {
      "epoch": 0.40084,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.033113352457682294,
      "learning_rate": 0.0001,
      "loss": 4.244,
      "loss/crossentropy": 2.0342337489128113,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19485876709222794,
      "step": 20042
    },
    {
      "epoch": 0.40088,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.027795155843098957,
      "learning_rate": 0.0001,
      "loss": 3.8838,
      "loss/crossentropy": 2.23150634765625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20408733934164047,
      "step": 20044
    },
    {
      "epoch": 0.40092,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.026805623372395834,
      "learning_rate": 0.0001,
      "loss": 3.9708,
      "loss/crossentropy": 2.086575150489807,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20418858528137207,
      "step": 20046
    },
    {
      "epoch": 0.40096,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.026741536458333333,
      "learning_rate": 0.0001,
      "loss": 4.1508,
      "loss/crossentropy": 2.2025340795516968,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21201904118061066,
      "step": 20048
    },
    {
      "epoch": 0.401,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.02859471638997396,
      "learning_rate": 0.0001,
      "loss": 4.2056,
      "loss/crossentropy": 2.175204277038574,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20766476541757584,
      "step": 20050
    },
    {
      "epoch": 0.40104,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.027581532796223957,
      "learning_rate": 0.0001,
      "loss": 4.0223,
      "loss/crossentropy": 2.159320116043091,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2013096585869789,
      "step": 20052
    },
    {
      "epoch": 0.40108,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.028452301025390626,
      "learning_rate": 0.0001,
      "loss": 3.8412,
      "loss/crossentropy": 1.5438128113746643,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1641307920217514,
      "step": 20054
    },
    {
      "epoch": 0.40112,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0180084228515625,
      "learning_rate": 0.0001,
      "loss": 3.8949,
      "loss/crossentropy": 2.0549490451812744,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2176165133714676,
      "step": 20056
    },
    {
      "epoch": 0.40116,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.01363525390625,
      "learning_rate": 0.0001,
      "loss": 4.1048,
      "loss/crossentropy": 1.8520516753196716,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19935637712478638,
      "step": 20058
    },
    {
      "epoch": 0.4012,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.007112375895182292,
      "learning_rate": 0.0001,
      "loss": 3.8383,
      "loss/crossentropy": 2.061866581439972,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1928197741508484,
      "step": 20060
    },
    {
      "epoch": 0.40124,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.006780751546223958,
      "learning_rate": 0.0001,
      "loss": 3.9205,
      "loss/crossentropy": 2.0396493673324585,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19679231941699982,
      "step": 20062
    },
    {
      "epoch": 0.40128,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.008231353759765626,
      "learning_rate": 0.0001,
      "loss": 4.2929,
      "loss/crossentropy": 1.8019860982894897,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17440176755189896,
      "step": 20064
    },
    {
      "epoch": 0.40132,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.011027018229166666,
      "learning_rate": 0.0001,
      "loss": 4.1233,
      "loss/crossentropy": 2.087724268436432,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18905764818191528,
      "step": 20066
    },
    {
      "epoch": 0.40136,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0110504150390625,
      "learning_rate": 0.0001,
      "loss": 3.9207,
      "loss/crossentropy": 1.899372398853302,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20874439179897308,
      "step": 20068
    },
    {
      "epoch": 0.4014,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.009073893229166666,
      "learning_rate": 0.0001,
      "loss": 3.9521,
      "loss/crossentropy": 1.740720272064209,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21033668518066406,
      "step": 20070
    },
    {
      "epoch": 0.40144,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.009748331705729167,
      "learning_rate": 0.0001,
      "loss": 3.9399,
      "loss/crossentropy": 2.043474793434143,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20285866409540176,
      "step": 20072
    },
    {
      "epoch": 0.40148,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.010114542643229167,
      "learning_rate": 0.0001,
      "loss": 3.8251,
      "loss/crossentropy": 1.907653033733368,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19814437627792358,
      "step": 20074
    },
    {
      "epoch": 0.40152,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0099029541015625,
      "learning_rate": 0.0001,
      "loss": 4.004,
      "loss/crossentropy": 1.803492784500122,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19028827548027039,
      "step": 20076
    },
    {
      "epoch": 0.40156,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.017463175455729167,
      "learning_rate": 0.0001,
      "loss": 4.0304,
      "loss/crossentropy": 2.2277488708496094,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2069825902581215,
      "step": 20078
    },
    {
      "epoch": 0.4016,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.01710205078125,
      "learning_rate": 0.0001,
      "loss": 4.0578,
      "loss/crossentropy": 1.9414427280426025,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.182416632771492,
      "step": 20080
    },
    {
      "epoch": 0.40164,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.014086659749348958,
      "learning_rate": 0.0001,
      "loss": 3.9105,
      "loss/crossentropy": 1.9231160879135132,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17756588757038116,
      "step": 20082
    },
    {
      "epoch": 0.40168,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.01566136678059896,
      "learning_rate": 0.0001,
      "loss": 4.2187,
      "loss/crossentropy": 2.1724199056625366,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21560527384281158,
      "step": 20084
    },
    {
      "epoch": 0.40172,
      "grad_norm": 2.125,
      "grad_norm_var": 0.017365519205729166,
      "learning_rate": 0.0001,
      "loss": 4.1919,
      "loss/crossentropy": 1.7942206859588623,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1705407276749611,
      "step": 20086
    },
    {
      "epoch": 0.40176,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.017183430989583335,
      "learning_rate": 0.0001,
      "loss": 4.0294,
      "loss/crossentropy": 2.1846182346343994,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2066875621676445,
      "step": 20088
    },
    {
      "epoch": 0.4018,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.0183746337890625,
      "learning_rate": 0.0001,
      "loss": 3.6378,
      "loss/crossentropy": 1.9378133416175842,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18584155291318893,
      "step": 20090
    },
    {
      "epoch": 0.40184,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.018379720052083333,
      "learning_rate": 0.0001,
      "loss": 4.2124,
      "loss/crossentropy": 1.9822070598602295,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22574876248836517,
      "step": 20092
    },
    {
      "epoch": 0.40188,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.009626261393229167,
      "learning_rate": 0.0001,
      "loss": 3.8348,
      "loss/crossentropy": 2.133267104625702,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20023657381534576,
      "step": 20094
    },
    {
      "epoch": 0.40192,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.012465159098307291,
      "learning_rate": 0.0001,
      "loss": 3.6025,
      "loss/crossentropy": 1.9899320602416992,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17226950079202652,
      "step": 20096
    },
    {
      "epoch": 0.40196,
      "grad_norm": 2.625,
      "grad_norm_var": 0.042699940999348956,
      "learning_rate": 0.0001,
      "loss": 4.6337,
      "loss/crossentropy": 2.3320037126541138,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.33343885838985443,
      "step": 20098
    },
    {
      "epoch": 0.402,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.04210586547851562,
      "learning_rate": 0.0001,
      "loss": 4.1064,
      "loss/crossentropy": 2.2026573419570923,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20648924261331558,
      "step": 20100
    },
    {
      "epoch": 0.40204,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.04067789713541667,
      "learning_rate": 0.0001,
      "loss": 4.0363,
      "loss/crossentropy": 2.1894484758377075,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1947491317987442,
      "step": 20102
    },
    {
      "epoch": 0.40208,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.040421549479166666,
      "learning_rate": 0.0001,
      "loss": 4.1506,
      "loss/crossentropy": 2.102599799633026,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19991370290517807,
      "step": 20104
    },
    {
      "epoch": 0.40212,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.038590494791666666,
      "learning_rate": 0.0001,
      "loss": 4.0476,
      "loss/crossentropy": 2.0289117097854614,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18713568150997162,
      "step": 20106
    },
    {
      "epoch": 0.40216,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.039098866780598956,
      "learning_rate": 0.0001,
      "loss": 4.0656,
      "loss/crossentropy": 2.0425156950950623,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19697192311286926,
      "step": 20108
    },
    {
      "epoch": 0.4022,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.0391265869140625,
      "learning_rate": 0.0001,
      "loss": 3.7796,
      "loss/crossentropy": 1.9846921563148499,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20306336879730225,
      "step": 20110
    },
    {
      "epoch": 0.40224,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.035166168212890626,
      "learning_rate": 0.0001,
      "loss": 3.8168,
      "loss/crossentropy": 1.846974492073059,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19833409041166306,
      "step": 20112
    },
    {
      "epoch": 0.40228,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.005052693684895833,
      "learning_rate": 0.0001,
      "loss": 4.0727,
      "loss/crossentropy": 2.3157109022140503,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2169334515929222,
      "step": 20114
    },
    {
      "epoch": 0.40232,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.004349772135416667,
      "learning_rate": 0.0001,
      "loss": 4.0974,
      "loss/crossentropy": 2.28099262714386,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20791316777467728,
      "step": 20116
    },
    {
      "epoch": 0.40236,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0042073567708333336,
      "learning_rate": 0.0001,
      "loss": 4.0327,
      "loss/crossentropy": 2.3558366298675537,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2205863893032074,
      "step": 20118
    },
    {
      "epoch": 0.4024,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0045074462890625,
      "learning_rate": 0.0001,
      "loss": 4.2792,
      "loss/crossentropy": 2.0106826424598694,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1982533410191536,
      "step": 20120
    },
    {
      "epoch": 0.40244,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.005773671468098958,
      "learning_rate": 0.0001,
      "loss": 4.2738,
      "loss/crossentropy": 2.321291446685791,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20979785174131393,
      "step": 20122
    },
    {
      "epoch": 0.40248,
      "grad_norm": 2.375,
      "grad_norm_var": 0.016658528645833334,
      "learning_rate": 0.0001,
      "loss": 4.3524,
      "loss/crossentropy": 2.3632254600524902,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2184157818555832,
      "step": 20124
    },
    {
      "epoch": 0.40252,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.015144856770833333,
      "learning_rate": 0.0001,
      "loss": 3.959,
      "loss/crossentropy": 2.1488123536109924,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2031005695462227,
      "step": 20126
    },
    {
      "epoch": 0.40256,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.015533192952473959,
      "learning_rate": 0.0001,
      "loss": 4.3341,
      "loss/crossentropy": 1.8036405444145203,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19172652810811996,
      "step": 20128
    },
    {
      "epoch": 0.4026,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.014925130208333333,
      "learning_rate": 0.0001,
      "loss": 4.1131,
      "loss/crossentropy": 2.0299129486083984,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20876885205507278,
      "step": 20130
    },
    {
      "epoch": 0.40264,
      "grad_norm": 2.0,
      "grad_norm_var": 0.014399973551432292,
      "learning_rate": 0.0001,
      "loss": 3.9618,
      "loss/crossentropy": 1.9177062511444092,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19626032561063766,
      "step": 20132
    },
    {
      "epoch": 0.40268,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.015897623697916665,
      "learning_rate": 0.0001,
      "loss": 4.1608,
      "loss/crossentropy": 2.2073251008987427,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2061092108488083,
      "step": 20134
    },
    {
      "epoch": 0.40272,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.01739476521809896,
      "learning_rate": 0.0001,
      "loss": 3.7592,
      "loss/crossentropy": 2.0243424773216248,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2109317108988762,
      "step": 20136
    },
    {
      "epoch": 0.40276,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.02050959269205729,
      "learning_rate": 0.0001,
      "loss": 3.845,
      "loss/crossentropy": 2.0052929520606995,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17854592204093933,
      "step": 20138
    },
    {
      "epoch": 0.4028,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.010790761311848958,
      "learning_rate": 0.0001,
      "loss": 4.0656,
      "loss/crossentropy": 2.067071557044983,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18235646188259125,
      "step": 20140
    },
    {
      "epoch": 0.40284,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.009354400634765624,
      "learning_rate": 0.0001,
      "loss": 4.0343,
      "loss/crossentropy": 2.050750732421875,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19028235971927643,
      "step": 20142
    },
    {
      "epoch": 0.40288,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.005680084228515625,
      "learning_rate": 0.0001,
      "loss": 4.2734,
      "loss/crossentropy": 2.1326886415481567,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2158709168434143,
      "step": 20144
    },
    {
      "epoch": 0.40292,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.006571451822916667,
      "learning_rate": 0.0001,
      "loss": 4.1029,
      "loss/crossentropy": 1.9970600605010986,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20081285387277603,
      "step": 20146
    },
    {
      "epoch": 0.40296,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.006178538004557292,
      "learning_rate": 0.0001,
      "loss": 3.8243,
      "loss/crossentropy": 2.042023479938507,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2022870033979416,
      "step": 20148
    },
    {
      "epoch": 0.403,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.006349436442057292,
      "learning_rate": 0.0001,
      "loss": 4.3217,
      "loss/crossentropy": 2.3379902839660645,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2027904912829399,
      "step": 20150
    },
    {
      "epoch": 0.40304,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.007080078125,
      "learning_rate": 0.0001,
      "loss": 4.0321,
      "loss/crossentropy": 1.804058849811554,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18429075926542282,
      "step": 20152
    },
    {
      "epoch": 0.40308,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.005716705322265625,
      "learning_rate": 0.0001,
      "loss": 4.1405,
      "loss/crossentropy": 1.93051016330719,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1912556290626526,
      "step": 20154
    },
    {
      "epoch": 0.40312,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.004634348551432291,
      "learning_rate": 0.0001,
      "loss": 4.0571,
      "loss/crossentropy": 2.3044906854629517,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22676818072795868,
      "step": 20156
    },
    {
      "epoch": 0.40316,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0047686258951822914,
      "learning_rate": 0.0001,
      "loss": 4.1262,
      "loss/crossentropy": 2.1352078914642334,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19985031336545944,
      "step": 20158
    },
    {
      "epoch": 0.4032,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0031565348307291668,
      "learning_rate": 0.0001,
      "loss": 4.1973,
      "loss/crossentropy": 2.0967469811439514,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1917915642261505,
      "step": 20160
    },
    {
      "epoch": 0.40324,
      "grad_norm": 1.875,
      "grad_norm_var": 0.0033078511555989583,
      "learning_rate": 0.0001,
      "loss": 4.2332,
      "loss/crossentropy": 2.1989063024520874,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2067013531923294,
      "step": 20162
    },
    {
      "epoch": 0.40328,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0032623291015625,
      "learning_rate": 0.0001,
      "loss": 4.144,
      "loss/crossentropy": 2.047918140888214,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19593901932239532,
      "step": 20164
    },
    {
      "epoch": 0.40332,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.003824615478515625,
      "learning_rate": 0.0001,
      "loss": 4.1674,
      "loss/crossentropy": 2.258637487888336,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20108382403850555,
      "step": 20166
    },
    {
      "epoch": 0.40336,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.003574371337890625,
      "learning_rate": 0.0001,
      "loss": 4.3006,
      "loss/crossentropy": 2.4297776222229004,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21948332339525223,
      "step": 20168
    },
    {
      "epoch": 0.4034,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.007413482666015625,
      "learning_rate": 0.0001,
      "loss": 4.1809,
      "loss/crossentropy": 2.029311180114746,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19954024255275726,
      "step": 20170
    },
    {
      "epoch": 0.40344,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.008123524983723958,
      "learning_rate": 0.0001,
      "loss": 4.0891,
      "loss/crossentropy": 2.099509835243225,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19811799377202988,
      "step": 20172
    },
    {
      "epoch": 0.40348,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.009732818603515625,
      "learning_rate": 0.0001,
      "loss": 3.996,
      "loss/crossentropy": 1.9699227809906006,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16898275911808014,
      "step": 20174
    },
    {
      "epoch": 0.40352,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.010668690999348958,
      "learning_rate": 0.0001,
      "loss": 4.0507,
      "loss/crossentropy": 2.0661654472351074,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20182892680168152,
      "step": 20176
    },
    {
      "epoch": 0.40356,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.0108306884765625,
      "learning_rate": 0.0001,
      "loss": 4.0142,
      "loss/crossentropy": 2.0732903480529785,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1912563294172287,
      "step": 20178
    },
    {
      "epoch": 0.4036,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.010578409830729166,
      "learning_rate": 0.0001,
      "loss": 4.0337,
      "loss/crossentropy": 1.9951130747795105,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17538709938526154,
      "step": 20180
    },
    {
      "epoch": 0.40364,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.010117340087890624,
      "learning_rate": 0.0001,
      "loss": 3.994,
      "loss/crossentropy": 1.8298532366752625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18598610162734985,
      "step": 20182
    },
    {
      "epoch": 0.40368,
      "grad_norm": 2.5,
      "grad_norm_var": 0.028742472330729168,
      "learning_rate": 0.0001,
      "loss": 4.3272,
      "loss/crossentropy": 2.19102144241333,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25994810461997986,
      "step": 20184
    },
    {
      "epoch": 0.40372,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.02552490234375,
      "learning_rate": 0.0001,
      "loss": 4.1003,
      "loss/crossentropy": 2.021101176738739,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19094721227884293,
      "step": 20186
    },
    {
      "epoch": 0.40376,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.02437108357747396,
      "learning_rate": 0.0001,
      "loss": 4.0057,
      "loss/crossentropy": 1.9469704627990723,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19880877435207367,
      "step": 20188
    },
    {
      "epoch": 0.4038,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.023374176025390624,
      "learning_rate": 0.0001,
      "loss": 4.1993,
      "loss/crossentropy": 2.0968725085258484,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23307666182518005,
      "step": 20190
    },
    {
      "epoch": 0.40384,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.0236572265625,
      "learning_rate": 0.0001,
      "loss": 4.1409,
      "loss/crossentropy": 1.7614120244979858,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1854785978794098,
      "step": 20192
    },
    {
      "epoch": 0.40388,
      "grad_norm": 1.875,
      "grad_norm_var": 0.021714019775390624,
      "learning_rate": 0.0001,
      "loss": 3.9778,
      "loss/crossentropy": 2.1977179050445557,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19997768104076385,
      "step": 20194
    },
    {
      "epoch": 0.40392,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.022359212239583332,
      "learning_rate": 0.0001,
      "loss": 3.6169,
      "loss/crossentropy": 1.6404736638069153,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18520022183656693,
      "step": 20196
    },
    {
      "epoch": 0.40396,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.022345987955729167,
      "learning_rate": 0.0001,
      "loss": 4.3767,
      "loss/crossentropy": 2.4365806579589844,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23998292535543442,
      "step": 20198
    },
    {
      "epoch": 0.404,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.005098215738932292,
      "learning_rate": 0.0001,
      "loss": 4.026,
      "loss/crossentropy": 2.191064238548279,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20327729731798172,
      "step": 20200
    },
    {
      "epoch": 0.40404,
      "grad_norm": 1.875,
      "grad_norm_var": 0.004992421468098958,
      "learning_rate": 0.0001,
      "loss": 3.9392,
      "loss/crossentropy": 2.1551318764686584,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.226949080824852,
      "step": 20202
    },
    {
      "epoch": 0.40408,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.017463175455729167,
      "learning_rate": 0.0001,
      "loss": 4.4831,
      "loss/crossentropy": 2.4735565185546875,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20860131084918976,
      "step": 20204
    },
    {
      "epoch": 0.40412,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.017288970947265624,
      "learning_rate": 0.0001,
      "loss": 4.1105,
      "loss/crossentropy": 2.1407764554023743,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2044227048754692,
      "step": 20206
    },
    {
      "epoch": 0.40416,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.016471099853515626,
      "learning_rate": 0.0001,
      "loss": 4.3293,
      "loss/crossentropy": 2.1775436401367188,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21017960458993912,
      "step": 20208
    },
    {
      "epoch": 0.4042,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.018195597330729167,
      "learning_rate": 0.0001,
      "loss": 3.9514,
      "loss/crossentropy": 1.8976858854293823,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17403876036405563,
      "step": 20210
    },
    {
      "epoch": 0.40424,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.018143463134765624,
      "learning_rate": 0.0001,
      "loss": 3.7933,
      "loss/crossentropy": 1.7951022386550903,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1805388182401657,
      "step": 20212
    },
    {
      "epoch": 0.40428,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.01911188761393229,
      "learning_rate": 0.0001,
      "loss": 4.0215,
      "loss/crossentropy": 2.0954891443252563,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20514150708913803,
      "step": 20214
    },
    {
      "epoch": 0.40432,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.018212636311848957,
      "learning_rate": 0.0001,
      "loss": 4.1753,
      "loss/crossentropy": 2.125544309616089,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19341818243265152,
      "step": 20216
    },
    {
      "epoch": 0.40436,
      "grad_norm": 2.125,
      "grad_norm_var": 0.01802546183268229,
      "learning_rate": 0.0001,
      "loss": 4.1361,
      "loss/crossentropy": 1.7649177312850952,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18554037809371948,
      "step": 20218
    },
    {
      "epoch": 0.4044,
      "grad_norm": 2.0,
      "grad_norm_var": 0.007529449462890625,
      "learning_rate": 0.0001,
      "loss": 4.1548,
      "loss/crossentropy": 2.1141446232795715,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.219147227704525,
      "step": 20220
    },
    {
      "epoch": 0.40444,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.007328033447265625,
      "learning_rate": 0.0001,
      "loss": 3.803,
      "loss/crossentropy": 1.9603995084762573,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19251137226819992,
      "step": 20222
    },
    {
      "epoch": 0.40448,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.006941731770833333,
      "learning_rate": 0.0001,
      "loss": 4.1023,
      "loss/crossentropy": 2.1824519634246826,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2134433463215828,
      "step": 20224
    },
    {
      "epoch": 0.40452,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.005796051025390625,
      "learning_rate": 0.0001,
      "loss": 4.0501,
      "loss/crossentropy": 2.0491157174110413,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19854432344436646,
      "step": 20226
    },
    {
      "epoch": 0.40456,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.007108561197916667,
      "learning_rate": 0.0001,
      "loss": 3.9262,
      "loss/crossentropy": 2.3731456995010376,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2264491394162178,
      "step": 20228
    },
    {
      "epoch": 0.4046,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.006150054931640625,
      "learning_rate": 0.0001,
      "loss": 3.9746,
      "loss/crossentropy": 1.9901636242866516,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17958952486515045,
      "step": 20230
    },
    {
      "epoch": 0.40464,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0072265625,
      "learning_rate": 0.0001,
      "loss": 4.2473,
      "loss/crossentropy": 1.9541369080543518,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19508858770132065,
      "step": 20232
    },
    {
      "epoch": 0.40468,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0056149800618489586,
      "learning_rate": 0.0001,
      "loss": 3.8702,
      "loss/crossentropy": 1.9723637700080872,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17709529399871826,
      "step": 20234
    },
    {
      "epoch": 0.40472,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006669108072916667,
      "learning_rate": 0.0001,
      "loss": 4.1318,
      "loss/crossentropy": 2.2354401350021362,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21117457002401352,
      "step": 20236
    },
    {
      "epoch": 0.40476,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.017787424723307292,
      "learning_rate": 0.0001,
      "loss": 4.1353,
      "loss/crossentropy": 2.1230952739715576,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19268101453781128,
      "step": 20238
    },
    {
      "epoch": 0.4048,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.024057769775390626,
      "learning_rate": 0.0001,
      "loss": 4.1973,
      "loss/crossentropy": 2.250411033630371,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24458947777748108,
      "step": 20240
    },
    {
      "epoch": 0.40484,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.021882120768229166,
      "learning_rate": 0.0001,
      "loss": 4.0774,
      "loss/crossentropy": 2.0546218752861023,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20313245803117752,
      "step": 20242
    },
    {
      "epoch": 0.40488,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.020808664957682292,
      "learning_rate": 0.0001,
      "loss": 3.827,
      "loss/crossentropy": 1.8805654644966125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19259311258792877,
      "step": 20244
    },
    {
      "epoch": 0.40492,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.023209381103515624,
      "learning_rate": 0.0001,
      "loss": 3.7371,
      "loss/crossentropy": 1.9676281809806824,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1743411347270012,
      "step": 20246
    },
    {
      "epoch": 0.40496,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.024894205729166667,
      "learning_rate": 0.0001,
      "loss": 3.7542,
      "loss/crossentropy": 2.2166742086410522,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19619230180978775,
      "step": 20248
    },
    {
      "epoch": 0.405,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.027103424072265625,
      "learning_rate": 0.0001,
      "loss": 4.2001,
      "loss/crossentropy": 2.2599674463272095,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19451096653938293,
      "step": 20250
    },
    {
      "epoch": 0.40504,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.027296702067057293,
      "learning_rate": 0.0001,
      "loss": 3.7535,
      "loss/crossentropy": 2.206246018409729,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20522233098745346,
      "step": 20252
    },
    {
      "epoch": 0.40508,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.01790135701497396,
      "learning_rate": 0.0001,
      "loss": 4.0881,
      "loss/crossentropy": 2.0229114294052124,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21543244272470474,
      "step": 20254
    },
    {
      "epoch": 0.40512,
      "grad_norm": 2.0,
      "grad_norm_var": 0.00960693359375,
      "learning_rate": 0.0001,
      "loss": 4.0279,
      "loss/crossentropy": 2.245228886604309,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20587065815925598,
      "step": 20256
    },
    {
      "epoch": 0.40516,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.009368642171223959,
      "learning_rate": 0.0001,
      "loss": 3.934,
      "loss/crossentropy": 2.030737340450287,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20048178732395172,
      "step": 20258
    },
    {
      "epoch": 0.4052,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.010027821858723958,
      "learning_rate": 0.0001,
      "loss": 4.0323,
      "loss/crossentropy": 2.084582805633545,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19962488114833832,
      "step": 20260
    },
    {
      "epoch": 0.40524,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.009075673421223958,
      "learning_rate": 0.0001,
      "loss": 3.8496,
      "loss/crossentropy": 2.2461657524108887,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21189837902784348,
      "step": 20262
    },
    {
      "epoch": 0.40528,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0095123291015625,
      "learning_rate": 0.0001,
      "loss": 4.0405,
      "loss/crossentropy": 1.9522897005081177,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18683286011219025,
      "step": 20264
    },
    {
      "epoch": 0.40532,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.008104451497395833,
      "learning_rate": 0.0001,
      "loss": 3.9797,
      "loss/crossentropy": 1.9156713485717773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18374722450971603,
      "step": 20266
    },
    {
      "epoch": 0.40536,
      "grad_norm": 2.96875,
      "grad_norm_var": 0.07459208170572916,
      "learning_rate": 0.0001,
      "loss": 4.2083,
      "loss/crossentropy": 2.2467408180236816,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2305024415254593,
      "step": 20268
    },
    {
      "epoch": 0.4054,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.07486063639322917,
      "learning_rate": 0.0001,
      "loss": 3.9249,
      "loss/crossentropy": 2.0827468633651733,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20297909528017044,
      "step": 20270
    },
    {
      "epoch": 0.40544,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.07484130859375,
      "learning_rate": 0.0001,
      "loss": 4.1911,
      "loss/crossentropy": 2.1979604959487915,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19396250694990158,
      "step": 20272
    },
    {
      "epoch": 0.40548,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0729156494140625,
      "learning_rate": 0.0001,
      "loss": 4.0384,
      "loss/crossentropy": 2.0157305002212524,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2022785097360611,
      "step": 20274
    },
    {
      "epoch": 0.40552,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.06889012654622396,
      "learning_rate": 0.0001,
      "loss": 4.1509,
      "loss/crossentropy": 1.838208019733429,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19091250747442245,
      "step": 20276
    },
    {
      "epoch": 0.40556,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.06746800740559895,
      "learning_rate": 0.0001,
      "loss": 4.1206,
      "loss/crossentropy": 2.1762728691101074,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19559895247220993,
      "step": 20278
    },
    {
      "epoch": 0.4056,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.06892903645833333,
      "learning_rate": 0.0001,
      "loss": 4.2381,
      "loss/crossentropy": 2.2119935154914856,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1954266056418419,
      "step": 20280
    },
    {
      "epoch": 0.40564,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0665728251139323,
      "learning_rate": 0.0001,
      "loss": 3.9296,
      "loss/crossentropy": 1.9655035138130188,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19419729709625244,
      "step": 20282
    },
    {
      "epoch": 0.40568,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.009501139322916666,
      "learning_rate": 0.0001,
      "loss": 3.8007,
      "loss/crossentropy": 1.7471181750297546,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15437456965446472,
      "step": 20284
    },
    {
      "epoch": 0.40572,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.006799062093098958,
      "learning_rate": 0.0001,
      "loss": 3.9379,
      "loss/crossentropy": 1.8075406551361084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18546781688928604,
      "step": 20286
    },
    {
      "epoch": 0.40576,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.006925455729166667,
      "learning_rate": 0.0001,
      "loss": 4.2841,
      "loss/crossentropy": 2.220236897468567,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20516446232795715,
      "step": 20288
    },
    {
      "epoch": 0.4058,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.006965128580729166,
      "learning_rate": 0.0001,
      "loss": 4.0263,
      "loss/crossentropy": 1.887015163898468,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1859249323606491,
      "step": 20290
    },
    {
      "epoch": 0.40584,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.004776763916015625,
      "learning_rate": 0.0001,
      "loss": 4.1448,
      "loss/crossentropy": 2.2080272436141968,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20920713245868683,
      "step": 20292
    },
    {
      "epoch": 0.40588,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.005722808837890625,
      "learning_rate": 0.0001,
      "loss": 4.2826,
      "loss/crossentropy": 1.9298865795135498,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1954272910952568,
      "step": 20294
    },
    {
      "epoch": 0.40592,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0064208984375,
      "learning_rate": 0.0001,
      "loss": 3.9118,
      "loss/crossentropy": 1.914223849773407,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2010154277086258,
      "step": 20296
    },
    {
      "epoch": 0.40596,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.005782063802083333,
      "learning_rate": 0.0001,
      "loss": 3.7257,
      "loss/crossentropy": 1.7863758206367493,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19244936853647232,
      "step": 20298
    },
    {
      "epoch": 0.406,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.005782063802083333,
      "learning_rate": 0.0001,
      "loss": 3.8874,
      "loss/crossentropy": 2.0487464666366577,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20337236672639847,
      "step": 20300
    },
    {
      "epoch": 0.40604,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.007281239827473958,
      "learning_rate": 0.0001,
      "loss": 3.8666,
      "loss/crossentropy": 1.8796595335006714,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18166716396808624,
      "step": 20302
    },
    {
      "epoch": 0.40608,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0071441650390625,
      "learning_rate": 0.0001,
      "loss": 4.0397,
      "loss/crossentropy": 2.0738844871520996,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20000334084033966,
      "step": 20304
    },
    {
      "epoch": 0.40612,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0070709228515625,
      "learning_rate": 0.0001,
      "loss": 3.8526,
      "loss/crossentropy": 2.392453908920288,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19829116761684418,
      "step": 20306
    },
    {
      "epoch": 0.40616,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.010117340087890624,
      "learning_rate": 0.0001,
      "loss": 4.0658,
      "loss/crossentropy": 2.137113928794861,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19673822075128555,
      "step": 20308
    },
    {
      "epoch": 0.4062,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.008593495686848958,
      "learning_rate": 0.0001,
      "loss": 4.1409,
      "loss/crossentropy": 2.0520911812782288,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19579464942216873,
      "step": 20310
    },
    {
      "epoch": 0.40624,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.007303873697916667,
      "learning_rate": 0.0001,
      "loss": 4.2045,
      "loss/crossentropy": 1.9204559922218323,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19526654481887817,
      "step": 20312
    },
    {
      "epoch": 0.40628,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.009822336832682292,
      "learning_rate": 0.0001,
      "loss": 3.9996,
      "loss/crossentropy": 2.0489402413368225,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19304056465625763,
      "step": 20314
    },
    {
      "epoch": 0.40632,
      "grad_norm": 2.4375,
      "grad_norm_var": 10.828910319010417,
      "learning_rate": 0.0001,
      "loss": 5.057,
      "loss/crossentropy": 2.127828896045685,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21591445803642273,
      "step": 20316
    },
    {
      "epoch": 0.40636,
      "grad_norm": 1.9453125,
      "grad_norm_var": 10.844437408447266,
      "learning_rate": 0.0001,
      "loss": 4.0046,
      "loss/crossentropy": 1.9867295026779175,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19691205769777298,
      "step": 20318
    },
    {
      "epoch": 0.4064,
      "grad_norm": 2.0,
      "grad_norm_var": 10.848514811197917,
      "learning_rate": 0.0001,
      "loss": 4.1028,
      "loss/crossentropy": 2.324278950691223,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2344008907675743,
      "step": 20320
    },
    {
      "epoch": 0.40644,
      "grad_norm": 2.15625,
      "grad_norm_var": 10.786188761393229,
      "learning_rate": 0.0001,
      "loss": 4.3306,
      "loss/crossentropy": 2.2904698848724365,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21537532657384872,
      "step": 20322
    },
    {
      "epoch": 0.40648,
      "grad_norm": 1.96875,
      "grad_norm_var": 10.77518081665039,
      "learning_rate": 0.0001,
      "loss": 3.9667,
      "loss/crossentropy": 2.0894381999969482,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20012424886226654,
      "step": 20324
    },
    {
      "epoch": 0.40652,
      "grad_norm": 1.984375,
      "grad_norm_var": 10.77937723795573,
      "learning_rate": 0.0001,
      "loss": 4.0753,
      "loss/crossentropy": 2.109618663787842,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21001552045345306,
      "step": 20326
    },
    {
      "epoch": 0.40656,
      "grad_norm": 2.078125,
      "grad_norm_var": 10.745448557535807,
      "learning_rate": 0.0001,
      "loss": 4.1139,
      "loss/crossentropy": 2.048314094543457,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19623054563999176,
      "step": 20328
    },
    {
      "epoch": 0.4066,
      "grad_norm": 2.09375,
      "grad_norm_var": 10.683837636311848,
      "learning_rate": 0.0001,
      "loss": 4.3341,
      "loss/crossentropy": 2.2704076766967773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24112576246261597,
      "step": 20330
    },
    {
      "epoch": 0.40664,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010057576497395833,
      "learning_rate": 0.0001,
      "loss": 3.9603,
      "loss/crossentropy": 2.1316241025924683,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20026037096977234,
      "step": 20332
    },
    {
      "epoch": 0.40668,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.015740712483723957,
      "learning_rate": 0.0001,
      "loss": 4.2839,
      "loss/crossentropy": 2.1923957467079163,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25954584032297134,
      "step": 20334
    },
    {
      "epoch": 0.40672,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.01502685546875,
      "learning_rate": 0.0001,
      "loss": 4.0627,
      "loss/crossentropy": 2.228965699672699,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21487966179847717,
      "step": 20336
    },
    {
      "epoch": 0.40676,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.01531982421875,
      "learning_rate": 0.0001,
      "loss": 3.8906,
      "loss/crossentropy": 1.9097226858139038,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1784234344959259,
      "step": 20338
    },
    {
      "epoch": 0.4068,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.020646158854166666,
      "learning_rate": 0.0001,
      "loss": 4.1256,
      "loss/crossentropy": 2.125716209411621,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19827620685100555,
      "step": 20340
    },
    {
      "epoch": 0.40684,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.02122802734375,
      "learning_rate": 0.0001,
      "loss": 4.1436,
      "loss/crossentropy": 2.025933086872101,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20766756683588028,
      "step": 20342
    },
    {
      "epoch": 0.40688,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.022638956705729168,
      "learning_rate": 0.0001,
      "loss": 3.7289,
      "loss/crossentropy": 2.0030741095542908,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19484290480613708,
      "step": 20344
    },
    {
      "epoch": 0.40692,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.02196019490559896,
      "learning_rate": 0.0001,
      "loss": 4.2287,
      "loss/crossentropy": 2.402723550796509,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21893122792243958,
      "step": 20346
    },
    {
      "epoch": 0.40696,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0221343994140625,
      "learning_rate": 0.0001,
      "loss": 3.8234,
      "loss/crossentropy": 2.077975869178772,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20634907484054565,
      "step": 20348
    },
    {
      "epoch": 0.407,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.015306599934895833,
      "learning_rate": 0.0001,
      "loss": 3.7596,
      "loss/crossentropy": 1.86312997341156,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18826880306005478,
      "step": 20350
    },
    {
      "epoch": 0.40704,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.015225982666015625,
      "learning_rate": 0.0001,
      "loss": 3.982,
      "loss/crossentropy": 2.090232253074646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20021233707666397,
      "step": 20352
    },
    {
      "epoch": 0.40708,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.015851847330729165,
      "learning_rate": 0.0001,
      "loss": 3.9013,
      "loss/crossentropy": 2.056272864341736,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1975172832608223,
      "step": 20354
    },
    {
      "epoch": 0.40712,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008689117431640626,
      "learning_rate": 0.0001,
      "loss": 4.2957,
      "loss/crossentropy": 2.0955827832221985,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1951812580227852,
      "step": 20356
    },
    {
      "epoch": 0.40716,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.007950846354166667,
      "learning_rate": 0.0001,
      "loss": 4.0598,
      "loss/crossentropy": 1.6103880405426025,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17600858211517334,
      "step": 20358
    },
    {
      "epoch": 0.4072,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.00782470703125,
      "learning_rate": 0.0001,
      "loss": 4.1568,
      "loss/crossentropy": 2.2216384410858154,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22146118432283401,
      "step": 20360
    },
    {
      "epoch": 0.40724,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.007033030192057292,
      "learning_rate": 0.0001,
      "loss": 4.3917,
      "loss/crossentropy": 2.1781840324401855,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19006534665822983,
      "step": 20362
    },
    {
      "epoch": 0.40728,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0074371337890625,
      "learning_rate": 0.0001,
      "loss": 3.9049,
      "loss/crossentropy": 1.8517277240753174,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17870192229747772,
      "step": 20364
    },
    {
      "epoch": 0.40732,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.006105295817057292,
      "learning_rate": 0.0001,
      "loss": 4.1469,
      "loss/crossentropy": 2.0296677947044373,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20260153710842133,
      "step": 20366
    },
    {
      "epoch": 0.40736,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.006046295166015625,
      "learning_rate": 0.0001,
      "loss": 4.0516,
      "loss/crossentropy": 2.088002324104309,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1896708756685257,
      "step": 20368
    },
    {
      "epoch": 0.4074,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.005122629801432291,
      "learning_rate": 0.0001,
      "loss": 4.0888,
      "loss/crossentropy": 2.0305171608924866,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1964031085371971,
      "step": 20370
    },
    {
      "epoch": 0.40744,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0046770731608072914,
      "learning_rate": 0.0001,
      "loss": 4.0663,
      "loss/crossentropy": 1.9005139470100403,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1981399655342102,
      "step": 20372
    },
    {
      "epoch": 0.40748,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0036333719889322918,
      "learning_rate": 0.0001,
      "loss": 4.0646,
      "loss/crossentropy": 2.1551105976104736,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21034018695354462,
      "step": 20374
    },
    {
      "epoch": 0.40752,
      "grad_norm": 1.875,
      "grad_norm_var": 0.00372314453125,
      "learning_rate": 0.0001,
      "loss": 3.9726,
      "loss/crossentropy": 2.0319311022758484,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1935891956090927,
      "step": 20376
    },
    {
      "epoch": 0.40756,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0052530924479166664,
      "learning_rate": 0.0001,
      "loss": 4.028,
      "loss/crossentropy": 2.06658136844635,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19683608412742615,
      "step": 20378
    },
    {
      "epoch": 0.4076,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.01574885050455729,
      "learning_rate": 0.0001,
      "loss": 4.5496,
      "loss/crossentropy": 2.667450428009033,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24288231134414673,
      "step": 20380
    },
    {
      "epoch": 0.40764,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.021061197916666666,
      "learning_rate": 0.0001,
      "loss": 3.6242,
      "loss/crossentropy": 1.9372497200965881,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18491743505001068,
      "step": 20382
    },
    {
      "epoch": 0.40768,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.021512858072916665,
      "learning_rate": 0.0001,
      "loss": 3.9358,
      "loss/crossentropy": 1.96478271484375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18305277079343796,
      "step": 20384
    },
    {
      "epoch": 0.40772,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.020096842447916666,
      "learning_rate": 0.0001,
      "loss": 4.0033,
      "loss/crossentropy": 1.813466727733612,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1827232465147972,
      "step": 20386
    },
    {
      "epoch": 0.40776,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.020420074462890625,
      "learning_rate": 0.0001,
      "loss": 3.9655,
      "loss/crossentropy": 1.6925603151321411,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17452678084373474,
      "step": 20388
    },
    {
      "epoch": 0.4078,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.02113622029622396,
      "learning_rate": 0.0001,
      "loss": 4.0845,
      "loss/crossentropy": 1.8859028220176697,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.187887504696846,
      "step": 20390
    },
    {
      "epoch": 0.40784,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.02066218058268229,
      "learning_rate": 0.0001,
      "loss": 3.9992,
      "loss/crossentropy": 2.0689834356307983,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20971451699733734,
      "step": 20392
    },
    {
      "epoch": 0.40788,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0203033447265625,
      "learning_rate": 0.0001,
      "loss": 4.2332,
      "loss/crossentropy": 2.237601161003113,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20401395112276077,
      "step": 20394
    },
    {
      "epoch": 0.40792,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.008560943603515624,
      "learning_rate": 0.0001,
      "loss": 3.9913,
      "loss/crossentropy": 1.8701192736625671,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.192036435008049,
      "step": 20396
    },
    {
      "epoch": 0.40796,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.005680084228515625,
      "learning_rate": 0.0001,
      "loss": 4.2942,
      "loss/crossentropy": 2.3522990942001343,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2178252711892128,
      "step": 20398
    },
    {
      "epoch": 0.408,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.005069732666015625,
      "learning_rate": 0.0001,
      "loss": 4.245,
      "loss/crossentropy": 2.267996072769165,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22347165644168854,
      "step": 20400
    },
    {
      "epoch": 0.40804,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.005208333333333333,
      "learning_rate": 0.0001,
      "loss": 4.079,
      "loss/crossentropy": 2.274789035320282,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2143096998333931,
      "step": 20402
    },
    {
      "epoch": 0.40808,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.004797108968098958,
      "learning_rate": 0.0001,
      "loss": 4.0509,
      "loss/crossentropy": 2.0331249237060547,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19468723237514496,
      "step": 20404
    },
    {
      "epoch": 0.40812,
      "grad_norm": 1.875,
      "grad_norm_var": 0.004980214436848958,
      "learning_rate": 0.0001,
      "loss": 3.9698,
      "loss/crossentropy": 1.9442223906517029,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19499845802783966,
      "step": 20406
    },
    {
      "epoch": 0.40816,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.00697021484375,
      "learning_rate": 0.0001,
      "loss": 3.6943,
      "loss/crossentropy": 1.9122044444084167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19391249865293503,
      "step": 20408
    },
    {
      "epoch": 0.4082,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.005651601155598958,
      "learning_rate": 0.0001,
      "loss": 3.9492,
      "loss/crossentropy": 1.9939789175987244,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18136083334684372,
      "step": 20410
    },
    {
      "epoch": 0.40824,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.006036122639973958,
      "learning_rate": 0.0001,
      "loss": 3.7378,
      "loss/crossentropy": 2.212642192840576,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18931927531957626,
      "step": 20412
    },
    {
      "epoch": 0.40828,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.006990559895833333,
      "learning_rate": 0.0001,
      "loss": 4.1713,
      "loss/crossentropy": 1.8282644152641296,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.190102256834507,
      "step": 20414
    },
    {
      "epoch": 0.40832,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.00745849609375,
      "learning_rate": 0.0001,
      "loss": 3.8334,
      "loss/crossentropy": 2.176365852355957,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19906923174858093,
      "step": 20416
    },
    {
      "epoch": 0.40836,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0073626200358072914,
      "learning_rate": 0.0001,
      "loss": 4.1297,
      "loss/crossentropy": 2.1126151084899902,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.180545412003994,
      "step": 20418
    },
    {
      "epoch": 0.4084,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.00784912109375,
      "learning_rate": 0.0001,
      "loss": 4.1878,
      "loss/crossentropy": 1.834926426410675,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1794508993625641,
      "step": 20420
    },
    {
      "epoch": 0.40844,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.007746378580729167,
      "learning_rate": 0.0001,
      "loss": 4.0114,
      "loss/crossentropy": 2.2909014225006104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21252857148647308,
      "step": 20422
    },
    {
      "epoch": 0.40848,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.005866495768229166,
      "learning_rate": 0.0001,
      "loss": 4.3771,
      "loss/crossentropy": 2.0966813564300537,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.197954960167408,
      "step": 20424
    },
    {
      "epoch": 0.40852,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.005228424072265625,
      "learning_rate": 0.0001,
      "loss": 4.0475,
      "loss/crossentropy": 2.0520461201667786,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19458714127540588,
      "step": 20426
    },
    {
      "epoch": 0.40856,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006400553385416666,
      "learning_rate": 0.0001,
      "loss": 4.4592,
      "loss/crossentropy": 2.2313653230667114,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.227764293551445,
      "step": 20428
    },
    {
      "epoch": 0.4086,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.006956990559895833,
      "learning_rate": 0.0001,
      "loss": 4.0398,
      "loss/crossentropy": 2.2874799966812134,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21484653651714325,
      "step": 20430
    },
    {
      "epoch": 0.40864,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.006400553385416666,
      "learning_rate": 0.0001,
      "loss": 4.1991,
      "loss/crossentropy": 2.2390183806419373,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21978556364774704,
      "step": 20432
    },
    {
      "epoch": 0.40868,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.008742014567057291,
      "learning_rate": 0.0001,
      "loss": 4.0057,
      "loss/crossentropy": 2.2040212154388428,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21448630839586258,
      "step": 20434
    },
    {
      "epoch": 0.40872,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.008356730143229166,
      "learning_rate": 0.0001,
      "loss": 4.1285,
      "loss/crossentropy": 2.2119942903518677,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19829177856445312,
      "step": 20436
    },
    {
      "epoch": 0.40876,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.007897694905598959,
      "learning_rate": 0.0001,
      "loss": 3.819,
      "loss/crossentropy": 1.769053339958191,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17577596008777618,
      "step": 20438
    },
    {
      "epoch": 0.4088,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.009226226806640625,
      "learning_rate": 0.0001,
      "loss": 4.0063,
      "loss/crossentropy": 2.1607295274734497,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20044966787099838,
      "step": 20440
    },
    {
      "epoch": 0.40884,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.009228515625,
      "learning_rate": 0.0001,
      "loss": 4.0617,
      "loss/crossentropy": 1.4168038368225098,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18440894782543182,
      "step": 20442
    },
    {
      "epoch": 0.40888,
      "grad_norm": 2.0,
      "grad_norm_var": 0.005490875244140625,
      "learning_rate": 0.0001,
      "loss": 4.2437,
      "loss/crossentropy": 2.1797818541526794,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22209269553422928,
      "step": 20444
    },
    {
      "epoch": 0.40892,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.005086008707682292,
      "learning_rate": 0.0001,
      "loss": 4.1425,
      "loss/crossentropy": 2.255184292793274,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20536670833826065,
      "step": 20446
    },
    {
      "epoch": 0.40896,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.005191802978515625,
      "learning_rate": 0.0001,
      "loss": 4.1206,
      "loss/crossentropy": 1.9633015990257263,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20300059020519257,
      "step": 20448
    },
    {
      "epoch": 0.409,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.002685292561848958,
      "learning_rate": 0.0001,
      "loss": 3.9409,
      "loss/crossentropy": 2.0943238735198975,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1932680904865265,
      "step": 20450
    },
    {
      "epoch": 0.40904,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0073321024576822914,
      "learning_rate": 0.0001,
      "loss": 4.3528,
      "loss/crossentropy": 2.3319740295410156,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24838463962078094,
      "step": 20452
    },
    {
      "epoch": 0.40908,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.0081787109375,
      "learning_rate": 0.0001,
      "loss": 3.9134,
      "loss/crossentropy": 2.0098442435264587,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19818133860826492,
      "step": 20454
    },
    {
      "epoch": 0.40912,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.020344034830729166,
      "learning_rate": 0.0001,
      "loss": 4.048,
      "loss/crossentropy": 1.9752068519592285,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20243104547262192,
      "step": 20456
    },
    {
      "epoch": 0.40916,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.024006144205729166,
      "learning_rate": 0.0001,
      "loss": 4.2057,
      "loss/crossentropy": 2.2999367713928223,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22019093483686447,
      "step": 20458
    },
    {
      "epoch": 0.4092,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.026395416259765624,
      "learning_rate": 0.0001,
      "loss": 3.9169,
      "loss/crossentropy": 1.770751178264618,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1787494421005249,
      "step": 20460
    },
    {
      "epoch": 0.40924,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.026667277018229168,
      "learning_rate": 0.0001,
      "loss": 3.9317,
      "loss/crossentropy": 2.0412501096725464,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2063281610608101,
      "step": 20462
    },
    {
      "epoch": 0.40928,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.027733357747395833,
      "learning_rate": 0.0001,
      "loss": 3.9852,
      "loss/crossentropy": 2.3771307468414307,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21928168088197708,
      "step": 20464
    },
    {
      "epoch": 0.40932,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.027551015218098957,
      "learning_rate": 0.0001,
      "loss": 4.0884,
      "loss/crossentropy": 1.91280996799469,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17027543485164642,
      "step": 20466
    },
    {
      "epoch": 0.40936,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.028831990559895833,
      "learning_rate": 0.0001,
      "loss": 3.9003,
      "loss/crossentropy": 2.024249494075775,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20218047499656677,
      "step": 20468
    },
    {
      "epoch": 0.4094,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.03514989217122396,
      "learning_rate": 0.0001,
      "loss": 4.0608,
      "loss/crossentropy": 1.787394940853119,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18133901804685593,
      "step": 20470
    },
    {
      "epoch": 0.40944,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.02319920857747396,
      "learning_rate": 0.0001,
      "loss": 3.9609,
      "loss/crossentropy": 2.085790276527405,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18643641471862793,
      "step": 20472
    },
    {
      "epoch": 0.40948,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.017929840087890624,
      "learning_rate": 0.0001,
      "loss": 4.2251,
      "loss/crossentropy": 2.1670111417770386,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19420959055423737,
      "step": 20474
    },
    {
      "epoch": 0.40952,
      "grad_norm": 1.875,
      "grad_norm_var": 0.01706720987955729,
      "learning_rate": 0.0001,
      "loss": 4.0588,
      "loss/crossentropy": 2.0304930210113525,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18586499989032745,
      "step": 20476
    },
    {
      "epoch": 0.40956,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.017832183837890626,
      "learning_rate": 0.0001,
      "loss": 4.2371,
      "loss/crossentropy": 2.0918440222740173,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21831007301807404,
      "step": 20478
    },
    {
      "epoch": 0.4096,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.017765045166015625,
      "learning_rate": 0.0001,
      "loss": 3.8786,
      "loss/crossentropy": 2.027481257915497,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19717323035001755,
      "step": 20480
    },
    {
      "epoch": 0.40964,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.017333984375,
      "learning_rate": 0.0001,
      "loss": 4.3524,
      "loss/crossentropy": 2.227196455001831,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21321603655815125,
      "step": 20482
    },
    {
      "epoch": 0.40968,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.015329742431640625,
      "learning_rate": 0.0001,
      "loss": 3.9722,
      "loss/crossentropy": 1.8985567688941956,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19271692633628845,
      "step": 20484
    },
    {
      "epoch": 0.40972,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.011498006184895833,
      "learning_rate": 0.0001,
      "loss": 4.1364,
      "loss/crossentropy": 1.973772943019867,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18885072320699692,
      "step": 20486
    },
    {
      "epoch": 0.40976,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.01171875,
      "learning_rate": 0.0001,
      "loss": 3.8451,
      "loss/crossentropy": 2.002311408519745,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2053375542163849,
      "step": 20488
    },
    {
      "epoch": 0.4098,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0702301025390625,
      "learning_rate": 0.0001,
      "loss": 3.8449,
      "loss/crossentropy": 1.7301526069641113,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16717776656150818,
      "step": 20490
    },
    {
      "epoch": 0.40984,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0685546875,
      "learning_rate": 0.0001,
      "loss": 4.1871,
      "loss/crossentropy": 1.86936616897583,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1944454312324524,
      "step": 20492
    },
    {
      "epoch": 0.40988,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.06875381469726563,
      "learning_rate": 0.0001,
      "loss": 4.1013,
      "loss/crossentropy": 1.5966055393218994,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19996708631515503,
      "step": 20494
    },
    {
      "epoch": 0.40992,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0671783447265625,
      "learning_rate": 0.0001,
      "loss": 3.9716,
      "loss/crossentropy": 1.8741289377212524,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18652669340372086,
      "step": 20496
    },
    {
      "epoch": 0.40996,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.06702372233072916,
      "learning_rate": 0.0001,
      "loss": 4.0857,
      "loss/crossentropy": 1.958758294582367,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19158867746591568,
      "step": 20498
    },
    {
      "epoch": 0.41,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.06591771443684896,
      "learning_rate": 0.0001,
      "loss": 3.7938,
      "loss/crossentropy": 1.8465643525123596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1774417608976364,
      "step": 20500
    },
    {
      "epoch": 0.41004,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.06303609212239583,
      "learning_rate": 0.0001,
      "loss": 4.0265,
      "loss/crossentropy": 1.9578893780708313,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19635465741157532,
      "step": 20502
    },
    {
      "epoch": 0.41008,
      "grad_norm": 1.875,
      "grad_norm_var": 0.06118977864583333,
      "learning_rate": 0.0001,
      "loss": 3.7317,
      "loss/crossentropy": 1.8893834352493286,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1778636872768402,
      "step": 20504
    },
    {
      "epoch": 0.41012,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.011189778645833334,
      "learning_rate": 0.0001,
      "loss": 4.1129,
      "loss/crossentropy": 2.2161590456962585,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19613313674926758,
      "step": 20506
    },
    {
      "epoch": 0.41016,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.011368560791015624,
      "learning_rate": 0.0001,
      "loss": 4.2363,
      "loss/crossentropy": 2.1272462606430054,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20562685281038284,
      "step": 20508
    },
    {
      "epoch": 0.4102,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.011319732666015625,
      "learning_rate": 0.0001,
      "loss": 3.9087,
      "loss/crossentropy": 1.8443754315376282,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20733944326639175,
      "step": 20510
    },
    {
      "epoch": 0.41024,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.012007649739583333,
      "learning_rate": 0.0001,
      "loss": 3.8766,
      "loss/crossentropy": 1.8213927149772644,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17855563014745712,
      "step": 20512
    },
    {
      "epoch": 0.41028,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009787750244140626,
      "learning_rate": 0.0001,
      "loss": 4.0914,
      "loss/crossentropy": 1.969380497932434,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20201627910137177,
      "step": 20514
    },
    {
      "epoch": 0.41032,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.009479777018229166,
      "learning_rate": 0.0001,
      "loss": 4.0491,
      "loss/crossentropy": 1.8974932432174683,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20217570662498474,
      "step": 20516
    },
    {
      "epoch": 0.41036,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.010312652587890625,
      "learning_rate": 0.0001,
      "loss": 4.0019,
      "loss/crossentropy": 1.9362964630126953,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1832883059978485,
      "step": 20518
    },
    {
      "epoch": 0.4104,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.010754140218098958,
      "learning_rate": 0.0001,
      "loss": 4.2071,
      "loss/crossentropy": 2.239969849586487,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22399117052555084,
      "step": 20520
    },
    {
      "epoch": 0.41044,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.004987589518229167,
      "learning_rate": 0.0001,
      "loss": 3.9606,
      "loss/crossentropy": 1.927116334438324,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.194569431245327,
      "step": 20522
    },
    {
      "epoch": 0.41048,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.00474853515625,
      "learning_rate": 0.0001,
      "loss": 4.0993,
      "loss/crossentropy": 1.921772539615631,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19974206387996674,
      "step": 20524
    },
    {
      "epoch": 0.41052,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.00509033203125,
      "learning_rate": 0.0001,
      "loss": 3.9591,
      "loss/crossentropy": 2.2263039350509644,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2088855803012848,
      "step": 20526
    },
    {
      "epoch": 0.41056,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.00343017578125,
      "learning_rate": 0.0001,
      "loss": 3.9004,
      "loss/crossentropy": 1.9309766292572021,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1984400451183319,
      "step": 20528
    },
    {
      "epoch": 0.4106,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0071441650390625,
      "learning_rate": 0.0001,
      "loss": 4.3564,
      "loss/crossentropy": 2.2890199422836304,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.210438072681427,
      "step": 20530
    },
    {
      "epoch": 0.41064,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.007199859619140625,
      "learning_rate": 0.0001,
      "loss": 4.2302,
      "loss/crossentropy": 1.961566150188446,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20430031418800354,
      "step": 20532
    },
    {
      "epoch": 0.41068,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0051513671875,
      "learning_rate": 0.0001,
      "loss": 4.0714,
      "loss/crossentropy": 1.9776958227157593,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20376411080360413,
      "step": 20534
    },
    {
      "epoch": 0.41072,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.005417633056640625,
      "learning_rate": 0.0001,
      "loss": 3.9352,
      "loss/crossentropy": 2.386608600616455,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20900936424732208,
      "step": 20536
    },
    {
      "epoch": 0.41076,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.007233683268229167,
      "learning_rate": 0.0001,
      "loss": 3.8075,
      "loss/crossentropy": 2.0592793226242065,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19810590893030167,
      "step": 20538
    },
    {
      "epoch": 0.4108,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.007420857747395833,
      "learning_rate": 0.0001,
      "loss": 4.2639,
      "loss/crossentropy": 2.176212787628174,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20612400770187378,
      "step": 20540
    },
    {
      "epoch": 0.41084,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.007134755452473958,
      "learning_rate": 0.0001,
      "loss": 4.0386,
      "loss/crossentropy": 1.920596957206726,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19120831042528152,
      "step": 20542
    },
    {
      "epoch": 0.41088,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.0086578369140625,
      "learning_rate": 0.0001,
      "loss": 4.0088,
      "loss/crossentropy": 1.8587758541107178,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1729830950498581,
      "step": 20544
    },
    {
      "epoch": 0.41092,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.004538726806640625,
      "learning_rate": 0.0001,
      "loss": 4.1263,
      "loss/crossentropy": 2.115676999092102,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2130855768918991,
      "step": 20546
    },
    {
      "epoch": 0.41096,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0043332417805989586,
      "learning_rate": 0.0001,
      "loss": 4.1933,
      "loss/crossentropy": 2.3977283239364624,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2074846476316452,
      "step": 20548
    },
    {
      "epoch": 0.411,
      "grad_norm": 1.7421875,
      "grad_norm_var": 0.005700429280598958,
      "learning_rate": 0.0001,
      "loss": 3.9028,
      "loss/crossentropy": 2.091398298740387,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1975005343556404,
      "step": 20550
    },
    {
      "epoch": 0.41104,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.005861155192057292,
      "learning_rate": 0.0001,
      "loss": 4.1509,
      "loss/crossentropy": 2.1864093542099,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2201247364282608,
      "step": 20552
    },
    {
      "epoch": 0.41108,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.005206044514973958,
      "learning_rate": 0.0001,
      "loss": 3.8462,
      "loss/crossentropy": 1.9154713153839111,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18139449506998062,
      "step": 20554
    },
    {
      "epoch": 0.41112,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.005682118733723958,
      "learning_rate": 0.0001,
      "loss": 4.0527,
      "loss/crossentropy": 2.392449378967285,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2069767862558365,
      "step": 20556
    },
    {
      "epoch": 0.41116,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0071604410807291664,
      "learning_rate": 0.0001,
      "loss": 4.0525,
      "loss/crossentropy": 1.9300146102905273,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20329315215349197,
      "step": 20558
    },
    {
      "epoch": 0.4112,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.007313791910807292,
      "learning_rate": 0.0001,
      "loss": 3.6521,
      "loss/crossentropy": 1.9400085806846619,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1855112686753273,
      "step": 20560
    },
    {
      "epoch": 0.41124,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.008321126302083334,
      "learning_rate": 0.0001,
      "loss": 4.1355,
      "loss/crossentropy": 2.1216121912002563,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1966063678264618,
      "step": 20562
    },
    {
      "epoch": 0.41128,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.008154042561848958,
      "learning_rate": 0.0001,
      "loss": 3.9513,
      "loss/crossentropy": 2.136322498321533,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19735725224018097,
      "step": 20564
    },
    {
      "epoch": 0.41132,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0060943603515625,
      "learning_rate": 0.0001,
      "loss": 3.9084,
      "loss/crossentropy": 2.1143118143081665,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21534033119678497,
      "step": 20566
    },
    {
      "epoch": 0.41136,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006577301025390625,
      "learning_rate": 0.0001,
      "loss": 3.9451,
      "loss/crossentropy": 1.8363113403320312,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17482534050941467,
      "step": 20568
    },
    {
      "epoch": 0.4114,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.00645751953125,
      "learning_rate": 0.0001,
      "loss": 4.0373,
      "loss/crossentropy": 1.8840075135231018,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19201497733592987,
      "step": 20570
    },
    {
      "epoch": 0.41144,
      "grad_norm": 2.0,
      "grad_norm_var": 0.006270090738932292,
      "learning_rate": 0.0001,
      "loss": 4.0909,
      "loss/crossentropy": 2.0200312733650208,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2130940854549408,
      "step": 20572
    },
    {
      "epoch": 0.41148,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.005078125,
      "learning_rate": 0.0001,
      "loss": 3.8886,
      "loss/crossentropy": 2.136338949203491,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22277235239744186,
      "step": 20574
    },
    {
      "epoch": 0.41152,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.004168446858723958,
      "learning_rate": 0.0001,
      "loss": 4.2314,
      "loss/crossentropy": 2.2381343841552734,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20784687995910645,
      "step": 20576
    },
    {
      "epoch": 0.41156,
      "grad_norm": 2.0,
      "grad_norm_var": 0.003780110677083333,
      "learning_rate": 0.0001,
      "loss": 4.0749,
      "loss/crossentropy": 2.224187970161438,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20546535402536392,
      "step": 20578
    },
    {
      "epoch": 0.4116,
      "grad_norm": 1.875,
      "grad_norm_var": 0.004117838541666667,
      "learning_rate": 0.0001,
      "loss": 3.5968,
      "loss/crossentropy": 1.966045081615448,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19925507158041,
      "step": 20580
    },
    {
      "epoch": 0.41164,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0036041259765625,
      "learning_rate": 0.0001,
      "loss": 3.6756,
      "loss/crossentropy": 1.7995998859405518,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1794610470533371,
      "step": 20582
    },
    {
      "epoch": 0.41168,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.011474609375,
      "learning_rate": 0.0001,
      "loss": 4.2112,
      "loss/crossentropy": 1.874330759048462,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23802601546049118,
      "step": 20584
    },
    {
      "epoch": 0.41172,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.011582183837890624,
      "learning_rate": 0.0001,
      "loss": 3.9275,
      "loss/crossentropy": 2.1704328060150146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21792975068092346,
      "step": 20586
    },
    {
      "epoch": 0.41176,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.012648264567057291,
      "learning_rate": 0.0001,
      "loss": 4.0073,
      "loss/crossentropy": 1.9361745119094849,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19227440655231476,
      "step": 20588
    },
    {
      "epoch": 0.4118,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0129058837890625,
      "learning_rate": 0.0001,
      "loss": 4.297,
      "loss/crossentropy": 2.19529128074646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23097511380910873,
      "step": 20590
    },
    {
      "epoch": 0.41184,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.012621815999348958,
      "learning_rate": 0.0001,
      "loss": 4.1085,
      "loss/crossentropy": 2.0319225788116455,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19555176049470901,
      "step": 20592
    },
    {
      "epoch": 0.41188,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.012393951416015625,
      "learning_rate": 0.0001,
      "loss": 4.0767,
      "loss/crossentropy": 2.395743250846863,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20953691750764847,
      "step": 20594
    },
    {
      "epoch": 0.41192,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.013901519775390624,
      "learning_rate": 0.0001,
      "loss": 3.8942,
      "loss/crossentropy": 1.7772584557533264,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19968461245298386,
      "step": 20596
    },
    {
      "epoch": 0.41196,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0158843994140625,
      "learning_rate": 0.0001,
      "loss": 3.7776,
      "loss/crossentropy": 1.89765065908432,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18085069954395294,
      "step": 20598
    },
    {
      "epoch": 0.412,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.007846832275390625,
      "learning_rate": 0.0001,
      "loss": 3.9754,
      "loss/crossentropy": 2.1154285073280334,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20126362890005112,
      "step": 20600
    },
    {
      "epoch": 0.41204,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.008259073893229166,
      "learning_rate": 0.0001,
      "loss": 4.2659,
      "loss/crossentropy": 2.15169358253479,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20982082933187485,
      "step": 20602
    },
    {
      "epoch": 0.41208,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.007824452718098958,
      "learning_rate": 0.0001,
      "loss": 3.9542,
      "loss/crossentropy": 1.8561811447143555,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18480198085308075,
      "step": 20604
    },
    {
      "epoch": 0.41212,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.008766428629557291,
      "learning_rate": 0.0001,
      "loss": 3.8989,
      "loss/crossentropy": 1.8246020078659058,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18010297417640686,
      "step": 20606
    },
    {
      "epoch": 0.41216,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0090484619140625,
      "learning_rate": 0.0001,
      "loss": 4.1666,
      "loss/crossentropy": 2.3279634714126587,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.224753275513649,
      "step": 20608
    },
    {
      "epoch": 0.4122,
      "grad_norm": 2.0,
      "grad_norm_var": 0.009171295166015624,
      "learning_rate": 0.0001,
      "loss": 4.0695,
      "loss/crossentropy": 2.122196078300476,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2060379832983017,
      "step": 20610
    },
    {
      "epoch": 0.41224,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.009666951497395833,
      "learning_rate": 0.0001,
      "loss": 4.1282,
      "loss/crossentropy": 2.0729124546051025,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20537251234054565,
      "step": 20612
    },
    {
      "epoch": 0.41228,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.008443196614583334,
      "learning_rate": 0.0001,
      "loss": 4.2078,
      "loss/crossentropy": 2.1832374930381775,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19204121828079224,
      "step": 20614
    },
    {
      "epoch": 0.41232,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.007982381184895833,
      "learning_rate": 0.0001,
      "loss": 3.8745,
      "loss/crossentropy": 1.6739779114723206,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15309840440750122,
      "step": 20616
    },
    {
      "epoch": 0.41236,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.008754221598307292,
      "learning_rate": 0.0001,
      "loss": 4.1438,
      "loss/crossentropy": 2.1585883498191833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20271800458431244,
      "step": 20618
    },
    {
      "epoch": 0.4124,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.008235422770182292,
      "learning_rate": 0.0001,
      "loss": 4.4289,
      "loss/crossentropy": 2.391141653060913,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2225441113114357,
      "step": 20620
    },
    {
      "epoch": 0.41244,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0053619384765625,
      "learning_rate": 0.0001,
      "loss": 4.1355,
      "loss/crossentropy": 2.055308997631073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1839761957526207,
      "step": 20622
    },
    {
      "epoch": 0.41248,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.005125935872395833,
      "learning_rate": 0.0001,
      "loss": 4.1143,
      "loss/crossentropy": 2.1222954988479614,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18773587048053741,
      "step": 20624
    },
    {
      "epoch": 0.41252,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.005647532145182292,
      "learning_rate": 0.0001,
      "loss": 4.1506,
      "loss/crossentropy": 2.011850595474243,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19865088164806366,
      "step": 20626
    },
    {
      "epoch": 0.41256,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0039784749348958336,
      "learning_rate": 0.0001,
      "loss": 4.0122,
      "loss/crossentropy": 1.841840922832489,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18899217247962952,
      "step": 20628
    },
    {
      "epoch": 0.4126,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.004313151041666667,
      "learning_rate": 0.0001,
      "loss": 4.2477,
      "loss/crossentropy": 2.335289478302002,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22496683150529861,
      "step": 20630
    },
    {
      "epoch": 0.41264,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.004487864176432292,
      "learning_rate": 0.0001,
      "loss": 3.9252,
      "loss/crossentropy": 2.0089204907417297,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18921320140361786,
      "step": 20632
    },
    {
      "epoch": 0.41268,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.004146067301432291,
      "learning_rate": 0.0001,
      "loss": 4.052,
      "loss/crossentropy": 2.115709662437439,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2033844143152237,
      "step": 20634
    },
    {
      "epoch": 0.41272,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.004282379150390625,
      "learning_rate": 0.0001,
      "loss": 4.0311,
      "loss/crossentropy": 2.175198018550873,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20709756761789322,
      "step": 20636
    },
    {
      "epoch": 0.41276,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.005248006184895833,
      "learning_rate": 0.0001,
      "loss": 4.0452,
      "loss/crossentropy": 2.128443717956543,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18793444335460663,
      "step": 20638
    },
    {
      "epoch": 0.4128,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.005830891927083333,
      "learning_rate": 0.0001,
      "loss": 3.9456,
      "loss/crossentropy": 1.890213668346405,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17856723070144653,
      "step": 20640
    },
    {
      "epoch": 0.41284,
      "grad_norm": 2.125,
      "grad_norm_var": 0.008283487955729167,
      "learning_rate": 0.0001,
      "loss": 4.1055,
      "loss/crossentropy": 1.925938606262207,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19046395272016525,
      "step": 20642
    },
    {
      "epoch": 0.41288,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.008442942301432292,
      "learning_rate": 0.0001,
      "loss": 4.2369,
      "loss/crossentropy": 2.0149936079978943,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20876548439264297,
      "step": 20644
    },
    {
      "epoch": 0.41292,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.008174641927083334,
      "learning_rate": 0.0001,
      "loss": 4.1904,
      "loss/crossentropy": 2.237368941307068,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20562848448753357,
      "step": 20646
    },
    {
      "epoch": 0.41296,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.008451080322265625,
      "learning_rate": 0.0001,
      "loss": 3.8332,
      "loss/crossentropy": 2.0298978090286255,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20426444709300995,
      "step": 20648
    },
    {
      "epoch": 0.413,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0090240478515625,
      "learning_rate": 0.0001,
      "loss": 3.872,
      "loss/crossentropy": 1.8883816599845886,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18203142285346985,
      "step": 20650
    },
    {
      "epoch": 0.41304,
      "grad_norm": 2.25,
      "grad_norm_var": 0.014288075764973958,
      "learning_rate": 0.0001,
      "loss": 4.0831,
      "loss/crossentropy": 2.051727771759033,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19955651462078094,
      "step": 20652
    },
    {
      "epoch": 0.41308,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.013335927327473959,
      "learning_rate": 0.0001,
      "loss": 4.1298,
      "loss/crossentropy": 2.3192938566207886,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23360159993171692,
      "step": 20654
    },
    {
      "epoch": 0.41312,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.014647420247395833,
      "learning_rate": 0.0001,
      "loss": 3.8486,
      "loss/crossentropy": 2.2148059606552124,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20121797919273376,
      "step": 20656
    },
    {
      "epoch": 0.41316,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011787923177083333,
      "learning_rate": 0.0001,
      "loss": 4.2523,
      "loss/crossentropy": 2.0665117502212524,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18279288709163666,
      "step": 20658
    },
    {
      "epoch": 0.4132,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.013060506184895833,
      "learning_rate": 0.0001,
      "loss": 3.8782,
      "loss/crossentropy": 2.0858163833618164,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19623829424381256,
      "step": 20660
    },
    {
      "epoch": 0.41324,
      "grad_norm": 1.875,
      "grad_norm_var": 0.013630167643229166,
      "learning_rate": 0.0001,
      "loss": 3.8134,
      "loss/crossentropy": 2.238997220993042,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20264007151126862,
      "step": 20662
    },
    {
      "epoch": 0.41328,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.016605377197265625,
      "learning_rate": 0.0001,
      "loss": 4.2317,
      "loss/crossentropy": 2.2540037631988525,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21315360069274902,
      "step": 20664
    },
    {
      "epoch": 0.41332,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.014975738525390626,
      "learning_rate": 0.0001,
      "loss": 4.1657,
      "loss/crossentropy": 2.3957645893096924,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2319154515862465,
      "step": 20666
    },
    {
      "epoch": 0.41336,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.009527333577473958,
      "learning_rate": 0.0001,
      "loss": 4.1946,
      "loss/crossentropy": 2.046703338623047,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1992144137620926,
      "step": 20668
    },
    {
      "epoch": 0.4134,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.010677083333333334,
      "learning_rate": 0.0001,
      "loss": 3.7664,
      "loss/crossentropy": 1.7997825145721436,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1673276051878929,
      "step": 20670
    },
    {
      "epoch": 0.41344,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009069569905598958,
      "learning_rate": 0.0001,
      "loss": 3.9432,
      "loss/crossentropy": 2.102105975151062,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19461503624916077,
      "step": 20672
    },
    {
      "epoch": 0.41348,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.009557851155598958,
      "learning_rate": 0.0001,
      "loss": 3.8547,
      "loss/crossentropy": 1.7070594429969788,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17999255657196045,
      "step": 20674
    },
    {
      "epoch": 0.41352,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.00869140625,
      "learning_rate": 0.0001,
      "loss": 4.0361,
      "loss/crossentropy": 2.1908987760543823,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2010250836610794,
      "step": 20676
    },
    {
      "epoch": 0.41356,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.007582346598307292,
      "learning_rate": 0.0001,
      "loss": 4.2711,
      "loss/crossentropy": 2.1943222284317017,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21789705008268356,
      "step": 20678
    },
    {
      "epoch": 0.4136,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.003842926025390625,
      "learning_rate": 0.0001,
      "loss": 4.2444,
      "loss/crossentropy": 2.1216484904289246,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20987947285175323,
      "step": 20680
    },
    {
      "epoch": 0.41364,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0038889567057291668,
      "learning_rate": 0.0001,
      "loss": 4.0256,
      "loss/crossentropy": 2.3570865392684937,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23686359077692032,
      "step": 20682
    },
    {
      "epoch": 0.41368,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0040771484375,
      "learning_rate": 0.0001,
      "loss": 4.1448,
      "loss/crossentropy": 1.834035336971283,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17401690781116486,
      "step": 20684
    },
    {
      "epoch": 0.41372,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0040891011555989586,
      "learning_rate": 0.0001,
      "loss": 4.2049,
      "loss/crossentropy": 2.251484513282776,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22543959319591522,
      "step": 20686
    },
    {
      "epoch": 0.41376,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.004571278889973958,
      "learning_rate": 0.0001,
      "loss": 4.3775,
      "loss/crossentropy": 2.233001708984375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18968196213245392,
      "step": 20688
    },
    {
      "epoch": 0.4138,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.004198964436848958,
      "learning_rate": 0.0001,
      "loss": 3.9482,
      "loss/crossentropy": 1.6730775833129883,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1740739718079567,
      "step": 20690
    },
    {
      "epoch": 0.41384,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.0100341796875,
      "learning_rate": 0.0001,
      "loss": 3.7717,
      "loss/crossentropy": 2.3700443506240845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20850200206041336,
      "step": 20692
    },
    {
      "epoch": 0.41388,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0098541259765625,
      "learning_rate": 0.0001,
      "loss": 4.2457,
      "loss/crossentropy": 2.253411650657654,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19234959036111832,
      "step": 20694
    },
    {
      "epoch": 0.41392,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.011836751302083334,
      "learning_rate": 0.0001,
      "loss": 3.9595,
      "loss/crossentropy": 2.414412260055542,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20320549607276917,
      "step": 20696
    },
    {
      "epoch": 0.41396,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.012562815348307292,
      "learning_rate": 0.0001,
      "loss": 3.8084,
      "loss/crossentropy": 1.8252301216125488,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19211787730455399,
      "step": 20698
    },
    {
      "epoch": 0.414,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.012387847900390625,
      "learning_rate": 0.0001,
      "loss": 4.123,
      "loss/crossentropy": 2.210301160812378,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21196797490119934,
      "step": 20700
    },
    {
      "epoch": 0.41404,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.013480377197265626,
      "learning_rate": 0.0001,
      "loss": 4.0629,
      "loss/crossentropy": 1.6429123878479004,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15947452187538147,
      "step": 20702
    },
    {
      "epoch": 0.41408,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.014218902587890625,
      "learning_rate": 0.0001,
      "loss": 3.9954,
      "loss/crossentropy": 1.8459800481796265,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18668671697378159,
      "step": 20704
    },
    {
      "epoch": 0.41412,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.014147694905598958,
      "learning_rate": 0.0001,
      "loss": 3.9468,
      "loss/crossentropy": 2.20779287815094,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2034989595413208,
      "step": 20706
    },
    {
      "epoch": 0.41416,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.03316218058268229,
      "learning_rate": 0.0001,
      "loss": 4.0503,
      "loss/crossentropy": 1.9622939229011536,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19208793342113495,
      "step": 20708
    },
    {
      "epoch": 0.4142,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.032990519205729166,
      "learning_rate": 0.0001,
      "loss": 4.2467,
      "loss/crossentropy": 1.9145439267158508,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17929863929748535,
      "step": 20710
    },
    {
      "epoch": 0.41424,
      "grad_norm": 1.6953125,
      "grad_norm_var": 0.03706029256184896,
      "learning_rate": 0.0001,
      "loss": 3.9202,
      "loss/crossentropy": 1.9035375714302063,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21188101917505264,
      "step": 20712
    },
    {
      "epoch": 0.41428,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.0380767822265625,
      "learning_rate": 0.0001,
      "loss": 4.0475,
      "loss/crossentropy": 2.0500373244285583,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2162824273109436,
      "step": 20714
    },
    {
      "epoch": 0.41432,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.03862075805664063,
      "learning_rate": 0.0001,
      "loss": 4.1101,
      "loss/crossentropy": 1.9774840474128723,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20070379972457886,
      "step": 20716
    },
    {
      "epoch": 0.41436,
      "grad_norm": 2.125,
      "grad_norm_var": 0.03785985310872396,
      "learning_rate": 0.0001,
      "loss": 4.3021,
      "loss/crossentropy": 1.9861173629760742,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19038750231266022,
      "step": 20718
    },
    {
      "epoch": 0.4144,
      "grad_norm": 1.875,
      "grad_norm_var": 0.053098297119140624,
      "learning_rate": 0.0001,
      "loss": 3.669,
      "loss/crossentropy": 1.9744009375572205,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19013763219118118,
      "step": 20720
    },
    {
      "epoch": 0.41444,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.054441070556640624,
      "learning_rate": 0.0001,
      "loss": 3.9343,
      "loss/crossentropy": 1.7110218405723572,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1715855970978737,
      "step": 20722
    },
    {
      "epoch": 0.41448,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.031556955973307294,
      "learning_rate": 0.0001,
      "loss": 3.9484,
      "loss/crossentropy": 1.8336329460144043,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19409292936325073,
      "step": 20724
    },
    {
      "epoch": 0.41452,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.03246027628580729,
      "learning_rate": 0.0001,
      "loss": 3.9218,
      "loss/crossentropy": 1.8363453149795532,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16922436654567719,
      "step": 20726
    },
    {
      "epoch": 0.41456,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.025699869791666666,
      "learning_rate": 0.0001,
      "loss": 4.1121,
      "loss/crossentropy": 1.8911715745925903,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19849538803100586,
      "step": 20728
    },
    {
      "epoch": 0.4146,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.02716064453125,
      "learning_rate": 0.0001,
      "loss": 3.8365,
      "loss/crossentropy": 1.7989731431007385,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18458950519561768,
      "step": 20730
    },
    {
      "epoch": 0.41464,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.02692235310872396,
      "learning_rate": 0.0001,
      "loss": 3.9896,
      "loss/crossentropy": 1.9423270225524902,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18612459301948547,
      "step": 20732
    },
    {
      "epoch": 0.41468,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.06059137980143229,
      "learning_rate": 0.0001,
      "loss": 3.7741,
      "loss/crossentropy": 1.7587624788284302,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1545337438583374,
      "step": 20734
    },
    {
      "epoch": 0.41472,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.04381510416666667,
      "learning_rate": 0.0001,
      "loss": 3.9651,
      "loss/crossentropy": 2.314169406890869,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22612107545137405,
      "step": 20736
    },
    {
      "epoch": 0.41476,
      "grad_norm": 2.421875,
      "grad_norm_var": 0.0535308837890625,
      "learning_rate": 0.0001,
      "loss": 4.272,
      "loss/crossentropy": 2.088030219078064,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19648675620555878,
      "step": 20738
    },
    {
      "epoch": 0.4148,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.05324071248372396,
      "learning_rate": 0.0001,
      "loss": 3.8286,
      "loss/crossentropy": 1.6665831208229065,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17022767663002014,
      "step": 20740
    },
    {
      "epoch": 0.41484,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.05115458170572917,
      "learning_rate": 0.0001,
      "loss": 4.1142,
      "loss/crossentropy": 2.425256609916687,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20156945288181305,
      "step": 20742
    },
    {
      "epoch": 0.41488,
      "grad_norm": 2.0,
      "grad_norm_var": 0.05110244750976563,
      "learning_rate": 0.0001,
      "loss": 3.8343,
      "loss/crossentropy": 1.6360740661621094,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16550948470830917,
      "step": 20744
    },
    {
      "epoch": 0.41492,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.044960276285807295,
      "learning_rate": 0.0001,
      "loss": 4.0782,
      "loss/crossentropy": 2.0676532983779907,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21137626469135284,
      "step": 20746
    },
    {
      "epoch": 0.41496,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.04678141276041667,
      "learning_rate": 0.0001,
      "loss": 4.122,
      "loss/crossentropy": 2.059161067008972,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20899715274572372,
      "step": 20748
    },
    {
      "epoch": 0.415,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.017268625895182292,
      "learning_rate": 0.0001,
      "loss": 4.1438,
      "loss/crossentropy": 2.1778156757354736,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20495261996984482,
      "step": 20750
    },
    {
      "epoch": 0.41504,
      "grad_norm": 2.125,
      "grad_norm_var": 0.016877237955729166,
      "learning_rate": 0.0001,
      "loss": 4.2809,
      "loss/crossentropy": 2.368219017982483,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21667053550481796,
      "step": 20752
    },
    {
      "epoch": 0.41508,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.006703440348307292,
      "learning_rate": 0.0001,
      "loss": 3.7872,
      "loss/crossentropy": 1.9143288731575012,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1782715767621994,
      "step": 20754
    },
    {
      "epoch": 0.41512,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.009388987223307292,
      "learning_rate": 0.0001,
      "loss": 4.2937,
      "loss/crossentropy": 1.7776128649711609,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1746697723865509,
      "step": 20756
    },
    {
      "epoch": 0.41516,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.009824371337890625,
      "learning_rate": 0.0001,
      "loss": 3.9991,
      "loss/crossentropy": 2.057144343852997,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18555855751037598,
      "step": 20758
    },
    {
      "epoch": 0.4152,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.012048085530598959,
      "learning_rate": 0.0001,
      "loss": 4.0996,
      "loss/crossentropy": 1.9628003239631653,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2057315558195114,
      "step": 20760
    },
    {
      "epoch": 0.41524,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.01282958984375,
      "learning_rate": 0.0001,
      "loss": 3.9163,
      "loss/crossentropy": 1.8570061326026917,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19530560076236725,
      "step": 20762
    },
    {
      "epoch": 0.41528,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0115386962890625,
      "learning_rate": 0.0001,
      "loss": 3.9201,
      "loss/crossentropy": 1.939897060394287,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1979508325457573,
      "step": 20764
    },
    {
      "epoch": 0.41532,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0116607666015625,
      "learning_rate": 0.0001,
      "loss": 4.0702,
      "loss/crossentropy": 2.15025132894516,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2082752287387848,
      "step": 20766
    },
    {
      "epoch": 0.41536,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.010713704427083333,
      "learning_rate": 0.0001,
      "loss": 4.0037,
      "loss/crossentropy": 2.202960252761841,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20040277391672134,
      "step": 20768
    },
    {
      "epoch": 0.4154,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.008208974202473959,
      "learning_rate": 0.0001,
      "loss": 4.05,
      "loss/crossentropy": 2.107069969177246,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1995261162519455,
      "step": 20770
    },
    {
      "epoch": 0.41544,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.0071103413899739586,
      "learning_rate": 0.0001,
      "loss": 3.8939,
      "loss/crossentropy": 2.287596106529236,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20532196760177612,
      "step": 20772
    },
    {
      "epoch": 0.41548,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.013236236572265626,
      "learning_rate": 0.0001,
      "loss": 3.7519,
      "loss/crossentropy": 1.8234007954597473,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17727909982204437,
      "step": 20774
    },
    {
      "epoch": 0.41552,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.010581207275390626,
      "learning_rate": 0.0001,
      "loss": 3.8238,
      "loss/crossentropy": 1.9223560690879822,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1918986812233925,
      "step": 20776
    },
    {
      "epoch": 0.41556,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.009769694010416666,
      "learning_rate": 0.0001,
      "loss": 3.8487,
      "loss/crossentropy": 2.0554555654525757,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17795050144195557,
      "step": 20778
    },
    {
      "epoch": 0.4156,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.009427642822265625,
      "learning_rate": 0.0001,
      "loss": 3.9899,
      "loss/crossentropy": 2.170193314552307,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20256754010915756,
      "step": 20780
    },
    {
      "epoch": 0.41564,
      "grad_norm": 2.0,
      "grad_norm_var": 0.009732818603515625,
      "learning_rate": 0.0001,
      "loss": 3.8443,
      "loss/crossentropy": 1.704530119895935,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18020445108413696,
      "step": 20782
    },
    {
      "epoch": 0.41568,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.009915924072265625,
      "learning_rate": 0.0001,
      "loss": 4.0143,
      "loss/crossentropy": 2.1245445013046265,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20647235959768295,
      "step": 20784
    },
    {
      "epoch": 0.41572,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.009895579020182291,
      "learning_rate": 0.0001,
      "loss": 4.1806,
      "loss/crossentropy": 2.4078409671783447,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2126438021659851,
      "step": 20786
    },
    {
      "epoch": 0.41576,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.011732737223307291,
      "learning_rate": 0.0001,
      "loss": 4.1084,
      "loss/crossentropy": 2.0577614307403564,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19302819669246674,
      "step": 20788
    },
    {
      "epoch": 0.4158,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0061686197916666664,
      "learning_rate": 0.0001,
      "loss": 4.0234,
      "loss/crossentropy": 2.0301398038864136,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1947554647922516,
      "step": 20790
    },
    {
      "epoch": 0.41584,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0062558492024739586,
      "learning_rate": 0.0001,
      "loss": 3.9092,
      "loss/crossentropy": 2.276142120361328,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21211084723472595,
      "step": 20792
    },
    {
      "epoch": 0.41588,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.006125640869140625,
      "learning_rate": 0.0001,
      "loss": 4.1577,
      "loss/crossentropy": 2.050966262817383,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20556103438138962,
      "step": 20794
    },
    {
      "epoch": 0.41592,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.009218088785807292,
      "learning_rate": 0.0001,
      "loss": 4.1697,
      "loss/crossentropy": 2.135987937450409,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21866349130868912,
      "step": 20796
    },
    {
      "epoch": 0.41596,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0089263916015625,
      "learning_rate": 0.0001,
      "loss": 4.0574,
      "loss/crossentropy": 2.0885995030403137,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2148696705698967,
      "step": 20798
    },
    {
      "epoch": 0.416,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.008471425374348958,
      "learning_rate": 0.0001,
      "loss": 3.9976,
      "loss/crossentropy": 1.8934147953987122,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19176077097654343,
      "step": 20800
    },
    {
      "epoch": 0.41604,
      "grad_norm": 2.0,
      "grad_norm_var": 0.007970937093098958,
      "learning_rate": 0.0001,
      "loss": 4.3367,
      "loss/crossentropy": 2.355128526687622,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21396907418966293,
      "step": 20802
    },
    {
      "epoch": 0.41608,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.005619049072265625,
      "learning_rate": 0.0001,
      "loss": 4.1109,
      "loss/crossentropy": 2.2145700454711914,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21031766384840012,
      "step": 20804
    },
    {
      "epoch": 0.41612,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.006048329671223958,
      "learning_rate": 0.0001,
      "loss": 3.8883,
      "loss/crossentropy": 2.17924165725708,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19039995223283768,
      "step": 20806
    },
    {
      "epoch": 0.41616,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.006017812093098958,
      "learning_rate": 0.0001,
      "loss": 4.0044,
      "loss/crossentropy": 2.2674453258514404,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18743296712636948,
      "step": 20808
    },
    {
      "epoch": 0.4162,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.052711741129557295,
      "learning_rate": 0.0001,
      "loss": 4.1857,
      "loss/crossentropy": 2.331398367881775,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20658230036497116,
      "step": 20810
    },
    {
      "epoch": 0.41624,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.05022354125976562,
      "learning_rate": 0.0001,
      "loss": 4.0019,
      "loss/crossentropy": 1.6092209815979004,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16996397078037262,
      "step": 20812
    },
    {
      "epoch": 0.41628,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.04944661458333333,
      "learning_rate": 0.0001,
      "loss": 4.0452,
      "loss/crossentropy": 2.094128370285034,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19843773543834686,
      "step": 20814
    },
    {
      "epoch": 0.41632,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.050842030843098955,
      "learning_rate": 0.0001,
      "loss": 3.9735,
      "loss/crossentropy": 1.9059642553329468,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20919224619865417,
      "step": 20816
    },
    {
      "epoch": 0.41636,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.050966135660807294,
      "learning_rate": 0.0001,
      "loss": 3.9026,
      "loss/crossentropy": 2.0733126401901245,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2132449895143509,
      "step": 20818
    },
    {
      "epoch": 0.4164,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.05125223795572917,
      "learning_rate": 0.0001,
      "loss": 4.1978,
      "loss/crossentropy": 2.09629487991333,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21850410103797913,
      "step": 20820
    },
    {
      "epoch": 0.41644,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.05614802042643229,
      "learning_rate": 0.0001,
      "loss": 3.6648,
      "loss/crossentropy": 1.7917864322662354,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16684360802173615,
      "step": 20822
    },
    {
      "epoch": 0.41648,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0561187744140625,
      "learning_rate": 0.0001,
      "loss": 3.7587,
      "loss/crossentropy": 1.629518985748291,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16551091521978378,
      "step": 20824
    },
    {
      "epoch": 0.41652,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009476725260416667,
      "learning_rate": 0.0001,
      "loss": 4.2593,
      "loss/crossentropy": 2.2774561643600464,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21336688101291656,
      "step": 20826
    },
    {
      "epoch": 0.41656,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.010684967041015625,
      "learning_rate": 0.0001,
      "loss": 3.7644,
      "loss/crossentropy": 2.0319623947143555,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19169747829437256,
      "step": 20828
    },
    {
      "epoch": 0.4166,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.010261789957682291,
      "learning_rate": 0.0001,
      "loss": 3.7815,
      "loss/crossentropy": 1.8104961514472961,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1863522008061409,
      "step": 20830
    },
    {
      "epoch": 0.41664,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0119781494140625,
      "learning_rate": 0.0001,
      "loss": 3.815,
      "loss/crossentropy": 1.729806661605835,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17772196978330612,
      "step": 20832
    },
    {
      "epoch": 0.41668,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.012116495768229167,
      "learning_rate": 0.0001,
      "loss": 3.9823,
      "loss/crossentropy": 1.9465582966804504,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.193453811109066,
      "step": 20834
    },
    {
      "epoch": 0.41672,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0122222900390625,
      "learning_rate": 0.0001,
      "loss": 4.1078,
      "loss/crossentropy": 1.895602285861969,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18032599240541458,
      "step": 20836
    },
    {
      "epoch": 0.41676,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.009924062093098958,
      "learning_rate": 0.0001,
      "loss": 3.7889,
      "loss/crossentropy": 2.082605481147766,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21816352754831314,
      "step": 20838
    },
    {
      "epoch": 0.4168,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.010628255208333333,
      "learning_rate": 0.0001,
      "loss": 3.9431,
      "loss/crossentropy": 1.7751468420028687,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18468675762414932,
      "step": 20840
    },
    {
      "epoch": 0.41684,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.007838694254557292,
      "learning_rate": 0.0001,
      "loss": 4.1319,
      "loss/crossentropy": 2.2611895203590393,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24100882560014725,
      "step": 20842
    },
    {
      "epoch": 0.41688,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0078125,
      "learning_rate": 0.0001,
      "loss": 4.0711,
      "loss/crossentropy": 2.1387221813201904,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1980198249220848,
      "step": 20844
    },
    {
      "epoch": 0.41692,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.008206939697265625,
      "learning_rate": 0.0001,
      "loss": 4.0165,
      "loss/crossentropy": 2.385176658630371,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22007980942726135,
      "step": 20846
    },
    {
      "epoch": 0.41696,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.005273183186848958,
      "learning_rate": 0.0001,
      "loss": 3.9463,
      "loss/crossentropy": 1.9382571578025818,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19266389310359955,
      "step": 20848
    },
    {
      "epoch": 0.417,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006583404541015625,
      "learning_rate": 0.0001,
      "loss": 4.3108,
      "loss/crossentropy": 1.91634601354599,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2147107571363449,
      "step": 20850
    },
    {
      "epoch": 0.41704,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.012532297770182292,
      "learning_rate": 0.0001,
      "loss": 3.9546,
      "loss/crossentropy": 1.9752032160758972,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20748814195394516,
      "step": 20852
    },
    {
      "epoch": 0.41708,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.011818186442057291,
      "learning_rate": 0.0001,
      "loss": 3.9149,
      "loss/crossentropy": 1.844580054283142,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17454466968774796,
      "step": 20854
    },
    {
      "epoch": 0.41712,
      "grad_norm": 1.875,
      "grad_norm_var": 0.010652669270833333,
      "learning_rate": 0.0001,
      "loss": 4.0963,
      "loss/crossentropy": 2.18678879737854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2004237025976181,
      "step": 20856
    },
    {
      "epoch": 0.41716,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.010583241780598959,
      "learning_rate": 0.0001,
      "loss": 3.855,
      "loss/crossentropy": 1.7560098767280579,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17802055180072784,
      "step": 20858
    },
    {
      "epoch": 0.4172,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.009911092122395833,
      "learning_rate": 0.0001,
      "loss": 4.1007,
      "loss/crossentropy": 1.8778411746025085,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1892327517271042,
      "step": 20860
    },
    {
      "epoch": 0.41724,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.009220123291015625,
      "learning_rate": 0.0001,
      "loss": 3.9034,
      "loss/crossentropy": 1.8801356554031372,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18983188271522522,
      "step": 20862
    },
    {
      "epoch": 0.41728,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.010448201497395834,
      "learning_rate": 0.0001,
      "loss": 3.974,
      "loss/crossentropy": 2.0934754014015198,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17912424355745316,
      "step": 20864
    },
    {
      "epoch": 0.41732,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.010247548421223959,
      "learning_rate": 0.0001,
      "loss": 4.0854,
      "loss/crossentropy": 2.115522801876068,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20105450600385666,
      "step": 20866
    },
    {
      "epoch": 0.41736,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.004622141520182292,
      "learning_rate": 0.0001,
      "loss": 4.0041,
      "loss/crossentropy": 2.1903135776519775,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19662026315927505,
      "step": 20868
    },
    {
      "epoch": 0.4174,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.005794016520182291,
      "learning_rate": 0.0001,
      "loss": 4.2067,
      "loss/crossentropy": 2.1521800756454468,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20861530303955078,
      "step": 20870
    },
    {
      "epoch": 0.41744,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.006145985921223959,
      "learning_rate": 0.0001,
      "loss": 4.4442,
      "loss/crossentropy": 2.0802451968193054,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19014900922775269,
      "step": 20872
    },
    {
      "epoch": 0.41748,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.006091054280598958,
      "learning_rate": 0.0001,
      "loss": 4.008,
      "loss/crossentropy": 1.8639289140701294,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17143192887306213,
      "step": 20874
    },
    {
      "epoch": 0.41752,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.006525675455729167,
      "learning_rate": 0.0001,
      "loss": 4.2595,
      "loss/crossentropy": 2.191206693649292,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2013956382870674,
      "step": 20876
    },
    {
      "epoch": 0.41756,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.006754557291666667,
      "learning_rate": 0.0001,
      "loss": 3.9213,
      "loss/crossentropy": 1.8131752610206604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1932450830936432,
      "step": 20878
    },
    {
      "epoch": 0.4176,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.005018870035807292,
      "learning_rate": 0.0001,
      "loss": 4.0509,
      "loss/crossentropy": 2.1427782773971558,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21525658667087555,
      "step": 20880
    },
    {
      "epoch": 0.41764,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.005956013997395833,
      "learning_rate": 0.0001,
      "loss": 4.1872,
      "loss/crossentropy": 1.997750997543335,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22109957039356232,
      "step": 20882
    },
    {
      "epoch": 0.41768,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.007700347900390625,
      "learning_rate": 0.0001,
      "loss": 4.0445,
      "loss/crossentropy": 2.2799811363220215,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18671181797981262,
      "step": 20884
    },
    {
      "epoch": 0.41772,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.006886545817057292,
      "learning_rate": 0.0001,
      "loss": 3.7884,
      "loss/crossentropy": 1.887694001197815,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19071876257658005,
      "step": 20886
    },
    {
      "epoch": 0.41776,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008235677083333334,
      "learning_rate": 0.0001,
      "loss": 4.0758,
      "loss/crossentropy": 2.0238420367240906,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2015087753534317,
      "step": 20888
    },
    {
      "epoch": 0.4178,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.008481597900390625,
      "learning_rate": 0.0001,
      "loss": 4.3148,
      "loss/crossentropy": 2.5183900594711304,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21474501490592957,
      "step": 20890
    },
    {
      "epoch": 0.41784,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0083160400390625,
      "learning_rate": 0.0001,
      "loss": 3.9086,
      "loss/crossentropy": 2.0730000734329224,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18937845528125763,
      "step": 20892
    },
    {
      "epoch": 0.41788,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.008101145426432291,
      "learning_rate": 0.0001,
      "loss": 3.9845,
      "loss/crossentropy": 2.083210587501526,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19441235065460205,
      "step": 20894
    },
    {
      "epoch": 0.41792,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.010961659749348958,
      "learning_rate": 0.0001,
      "loss": 3.9813,
      "loss/crossentropy": 1.7236113548278809,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1931980401277542,
      "step": 20896
    },
    {
      "epoch": 0.41796,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.008796183268229167,
      "learning_rate": 0.0001,
      "loss": 4.0757,
      "loss/crossentropy": 2.108364462852478,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20710121095180511,
      "step": 20898
    },
    {
      "epoch": 0.418,
      "grad_norm": 1.7421875,
      "grad_norm_var": 0.010179646809895833,
      "learning_rate": 0.0001,
      "loss": 3.8073,
      "loss/crossentropy": 2.10055810213089,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20653685182332993,
      "step": 20900
    },
    {
      "epoch": 0.41804,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.011517079671223958,
      "learning_rate": 0.0001,
      "loss": 4.0977,
      "loss/crossentropy": 2.254370093345642,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21069462597370148,
      "step": 20902
    },
    {
      "epoch": 0.41808,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.010107421875,
      "learning_rate": 0.0001,
      "loss": 3.8429,
      "loss/crossentropy": 2.053595006465912,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19040333479642868,
      "step": 20904
    },
    {
      "epoch": 0.41812,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.014012654622395834,
      "learning_rate": 0.0001,
      "loss": 4.3406,
      "loss/crossentropy": 2.1739301681518555,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19576483964920044,
      "step": 20906
    },
    {
      "epoch": 0.41816,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.01635920206705729,
      "learning_rate": 0.0001,
      "loss": 4.2794,
      "loss/crossentropy": 1.9707902073860168,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20553794503211975,
      "step": 20908
    },
    {
      "epoch": 0.4182,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.017097981770833333,
      "learning_rate": 0.0001,
      "loss": 4.0596,
      "loss/crossentropy": 1.901296079158783,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18205050379037857,
      "step": 20910
    },
    {
      "epoch": 0.41824,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.015388997395833333,
      "learning_rate": 0.0001,
      "loss": 4.3012,
      "loss/crossentropy": 2.2531062364578247,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18042850494384766,
      "step": 20912
    },
    {
      "epoch": 0.41828,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0159088134765625,
      "learning_rate": 0.0001,
      "loss": 3.8839,
      "loss/crossentropy": 1.959227204322815,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19045928120613098,
      "step": 20914
    },
    {
      "epoch": 0.41832,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.014207967122395833,
      "learning_rate": 0.0001,
      "loss": 3.9595,
      "loss/crossentropy": 2.1432504057884216,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1991618126630783,
      "step": 20916
    },
    {
      "epoch": 0.41836,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.013423665364583334,
      "learning_rate": 0.0001,
      "loss": 4.0369,
      "loss/crossentropy": 2.090354800224304,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19416044652462006,
      "step": 20918
    },
    {
      "epoch": 0.4184,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011889394124348958,
      "learning_rate": 0.0001,
      "loss": 4.0133,
      "loss/crossentropy": 2.0165509581565857,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20892751216888428,
      "step": 20920
    },
    {
      "epoch": 0.41844,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.009273020426432292,
      "learning_rate": 0.0001,
      "loss": 4.0852,
      "loss/crossentropy": 2.2627620697021484,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20449956506490707,
      "step": 20922
    },
    {
      "epoch": 0.41848,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.007249959309895833,
      "learning_rate": 0.0001,
      "loss": 3.8586,
      "loss/crossentropy": 1.9771518111228943,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20680005848407745,
      "step": 20924
    },
    {
      "epoch": 0.41852,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.006982421875,
      "learning_rate": 0.0001,
      "loss": 4.1715,
      "loss/crossentropy": 2.2706029415130615,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22013718634843826,
      "step": 20926
    },
    {
      "epoch": 0.41856,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0060503641764322914,
      "learning_rate": 0.0001,
      "loss": 4.1293,
      "loss/crossentropy": 2.071012258529663,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21389423310756683,
      "step": 20928
    },
    {
      "epoch": 0.4186,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0071044921875,
      "learning_rate": 0.0001,
      "loss": 4.3878,
      "loss/crossentropy": 2.084509491920471,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20820914208889008,
      "step": 20930
    },
    {
      "epoch": 0.41864,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.005855051676432291,
      "learning_rate": 0.0001,
      "loss": 3.7038,
      "loss/crossentropy": 2.0449500679969788,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20955926179885864,
      "step": 20932
    },
    {
      "epoch": 0.41868,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.005881500244140625,
      "learning_rate": 0.0001,
      "loss": 4.0819,
      "loss/crossentropy": 2.0860745310783386,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21544789522886276,
      "step": 20934
    },
    {
      "epoch": 0.41872,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0065877278645833336,
      "learning_rate": 0.0001,
      "loss": 3.9627,
      "loss/crossentropy": 2.042892038822174,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20232345163822174,
      "step": 20936
    },
    {
      "epoch": 0.41876,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.006298828125,
      "learning_rate": 0.0001,
      "loss": 4.3137,
      "loss/crossentropy": 2.3840891122817993,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21144797652959824,
      "step": 20938
    },
    {
      "epoch": 0.4188,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.005210113525390625,
      "learning_rate": 0.0001,
      "loss": 4.1256,
      "loss/crossentropy": 1.8122249245643616,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1800767481327057,
      "step": 20940
    },
    {
      "epoch": 0.41884,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0052073160807291664,
      "learning_rate": 0.0001,
      "loss": 4.2324,
      "loss/crossentropy": 2.168284773826599,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2116834670305252,
      "step": 20942
    },
    {
      "epoch": 0.41888,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.006005859375,
      "learning_rate": 0.0001,
      "loss": 4.0847,
      "loss/crossentropy": 2.1386263370513916,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18317822366952896,
      "step": 20944
    },
    {
      "epoch": 0.41892,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.004829915364583334,
      "learning_rate": 0.0001,
      "loss": 3.993,
      "loss/crossentropy": 2.134578227996826,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20232250541448593,
      "step": 20946
    },
    {
      "epoch": 0.41896,
      "grad_norm": 1.875,
      "grad_norm_var": 0.005116526285807292,
      "learning_rate": 0.0001,
      "loss": 4.212,
      "loss/crossentropy": 2.138009011745453,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20763013511896133,
      "step": 20948
    },
    {
      "epoch": 0.419,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.004881795247395833,
      "learning_rate": 0.0001,
      "loss": 3.9959,
      "loss/crossentropy": 2.1796361207962036,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1887281984090805,
      "step": 20950
    },
    {
      "epoch": 0.41904,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.004369099934895833,
      "learning_rate": 0.0001,
      "loss": 3.9328,
      "loss/crossentropy": 1.9848375916481018,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21374072134494781,
      "step": 20952
    },
    {
      "epoch": 0.41908,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.003985341389973958,
      "learning_rate": 0.0001,
      "loss": 4.1199,
      "loss/crossentropy": 1.9993594288825989,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19865376502275467,
      "step": 20954
    },
    {
      "epoch": 0.41912,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.003883616129557292,
      "learning_rate": 0.0001,
      "loss": 4.0882,
      "loss/crossentropy": 1.841040551662445,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1984993740916252,
      "step": 20956
    },
    {
      "epoch": 0.41916,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.007472483317057291,
      "learning_rate": 0.0001,
      "loss": 4.3558,
      "loss/crossentropy": 2.4816187620162964,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2072821408510208,
      "step": 20958
    },
    {
      "epoch": 0.4192,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.009114329020182292,
      "learning_rate": 0.0001,
      "loss": 4.3989,
      "loss/crossentropy": 2.1488978266716003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2619112655520439,
      "step": 20960
    },
    {
      "epoch": 0.41924,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.009251912434895834,
      "learning_rate": 0.0001,
      "loss": 3.8774,
      "loss/crossentropy": 1.8684781193733215,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1915435940027237,
      "step": 20962
    },
    {
      "epoch": 0.41928,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.008599599202473959,
      "learning_rate": 0.0001,
      "loss": 4.0564,
      "loss/crossentropy": 2.2308766841888428,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21091555058956146,
      "step": 20964
    },
    {
      "epoch": 0.41932,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.0086822509765625,
      "learning_rate": 0.0001,
      "loss": 4.016,
      "loss/crossentropy": 2.0642316341400146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18509763479232788,
      "step": 20966
    },
    {
      "epoch": 0.41936,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.008929189046223958,
      "learning_rate": 0.0001,
      "loss": 4.2041,
      "loss/crossentropy": 2.4577912092208862,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22347646206617355,
      "step": 20968
    },
    {
      "epoch": 0.4194,
      "grad_norm": 2.125,
      "grad_norm_var": 0.009563954671223958,
      "learning_rate": 0.0001,
      "loss": 4.4522,
      "loss/crossentropy": 2.0276389122009277,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18190696835517883,
      "step": 20970
    },
    {
      "epoch": 0.41944,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.011171213785807292,
      "learning_rate": 0.0001,
      "loss": 3.7112,
      "loss/crossentropy": 1.920120656490326,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18268628418445587,
      "step": 20972
    },
    {
      "epoch": 0.41948,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.010074615478515625,
      "learning_rate": 0.0001,
      "loss": 3.671,
      "loss/crossentropy": 2.0709950923919678,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20330249518156052,
      "step": 20974
    },
    {
      "epoch": 0.41952,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.0092926025390625,
      "learning_rate": 0.0001,
      "loss": 4.1188,
      "loss/crossentropy": 2.059907555580139,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23229123651981354,
      "step": 20976
    },
    {
      "epoch": 0.41956,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.012190755208333333,
      "learning_rate": 0.0001,
      "loss": 3.8434,
      "loss/crossentropy": 1.9516863226890564,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19371748715639114,
      "step": 20978
    },
    {
      "epoch": 0.4196,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.012181599934895834,
      "learning_rate": 0.0001,
      "loss": 3.8149,
      "loss/crossentropy": 1.8882723450660706,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1830422580242157,
      "step": 20980
    },
    {
      "epoch": 0.41964,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.011864980061848959,
      "learning_rate": 0.0001,
      "loss": 4.2085,
      "loss/crossentropy": 2.3770724534988403,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2055138498544693,
      "step": 20982
    },
    {
      "epoch": 0.41968,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.010221354166666667,
      "learning_rate": 0.0001,
      "loss": 4.0082,
      "loss/crossentropy": 2.302065849304199,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21600044518709183,
      "step": 20984
    },
    {
      "epoch": 0.41972,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.00772705078125,
      "learning_rate": 0.0001,
      "loss": 4.2794,
      "loss/crossentropy": 2.1097174286842346,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2151128277182579,
      "step": 20986
    },
    {
      "epoch": 0.41976,
      "grad_norm": 1.875,
      "grad_norm_var": 0.007765452067057292,
      "learning_rate": 0.0001,
      "loss": 4.0175,
      "loss/crossentropy": 2.1860098838806152,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20798291265964508,
      "step": 20988
    },
    {
      "epoch": 0.4198,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.007258097330729167,
      "learning_rate": 0.0001,
      "loss": 3.7789,
      "loss/crossentropy": 2.0685555934906006,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20412929356098175,
      "step": 20990
    },
    {
      "epoch": 0.41984,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0064046223958333336,
      "learning_rate": 0.0001,
      "loss": 3.9187,
      "loss/crossentropy": 1.886826515197754,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17476174980401993,
      "step": 20992
    },
    {
      "epoch": 0.41988,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.003446197509765625,
      "learning_rate": 0.0001,
      "loss": 3.853,
      "loss/crossentropy": 1.8504652380943298,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18863234668970108,
      "step": 20994
    },
    {
      "epoch": 0.41992,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0035308837890625,
      "learning_rate": 0.0001,
      "loss": 4.0992,
      "loss/crossentropy": 2.396555781364441,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21075168251991272,
      "step": 20996
    },
    {
      "epoch": 0.41996,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.008676910400390625,
      "learning_rate": 0.0001,
      "loss": 4.0947,
      "loss/crossentropy": 2.005809009075165,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2158650904893875,
      "step": 20998
    },
    {
      "epoch": 0.42,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.008660634358723959,
      "learning_rate": 0.0001,
      "loss": 3.8813,
      "loss/crossentropy": 1.8755770325660706,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19334855675697327,
      "step": 21000
    },
    {
      "epoch": 0.42004,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.007989247639973959,
      "learning_rate": 0.0001,
      "loss": 4.1168,
      "loss/crossentropy": 2.4014203548431396,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22790876030921936,
      "step": 21002
    },
    {
      "epoch": 0.42008,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.008983357747395834,
      "learning_rate": 0.0001,
      "loss": 3.7571,
      "loss/crossentropy": 1.959916114807129,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19611585140228271,
      "step": 21004
    },
    {
      "epoch": 0.42012,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.010212961832682292,
      "learning_rate": 0.0001,
      "loss": 4.0505,
      "loss/crossentropy": 2.077211081981659,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19385989010334015,
      "step": 21006
    },
    {
      "epoch": 0.42016,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.017878214518229168,
      "learning_rate": 0.0001,
      "loss": 4.2484,
      "loss/crossentropy": 2.1105176210403442,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.303642213344574,
      "step": 21008
    },
    {
      "epoch": 0.4202,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.01786066691080729,
      "learning_rate": 0.0001,
      "loss": 4.0644,
      "loss/crossentropy": 2.0568641424179077,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2137533351778984,
      "step": 21010
    },
    {
      "epoch": 0.42024,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.018424224853515626,
      "learning_rate": 0.0001,
      "loss": 3.9592,
      "loss/crossentropy": 2.185602903366089,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20309215039014816,
      "step": 21012
    },
    {
      "epoch": 0.42028,
      "grad_norm": 2.0,
      "grad_norm_var": 0.014646148681640625,
      "learning_rate": 0.0001,
      "loss": 4.1891,
      "loss/crossentropy": 1.9848479628562927,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1918111816048622,
      "step": 21014
    },
    {
      "epoch": 0.42032,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.014731597900390626,
      "learning_rate": 0.0001,
      "loss": 4.167,
      "loss/crossentropy": 2.315872311592102,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22417553514242172,
      "step": 21016
    },
    {
      "epoch": 0.42036,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.014798736572265625,
      "learning_rate": 0.0001,
      "loss": 3.9912,
      "loss/crossentropy": 2.1230897903442383,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19763804227113724,
      "step": 21018
    },
    {
      "epoch": 0.4204,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.012947336832682291,
      "learning_rate": 0.0001,
      "loss": 4.2364,
      "loss/crossentropy": 1.9737728834152222,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18397405743598938,
      "step": 21020
    },
    {
      "epoch": 0.42044,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.012064615885416666,
      "learning_rate": 0.0001,
      "loss": 4.0749,
      "loss/crossentropy": 2.1349278688430786,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1923152655363083,
      "step": 21022
    },
    {
      "epoch": 0.42048,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.007242584228515625,
      "learning_rate": 0.0001,
      "loss": 4.1186,
      "loss/crossentropy": 2.110077440738678,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2115027979016304,
      "step": 21024
    },
    {
      "epoch": 0.42052,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.008430989583333333,
      "learning_rate": 0.0001,
      "loss": 4.0164,
      "loss/crossentropy": 2.0237202048301697,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17709830403327942,
      "step": 21026
    },
    {
      "epoch": 0.42056,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.006990559895833333,
      "learning_rate": 0.0001,
      "loss": 4.2527,
      "loss/crossentropy": 2.1712347269058228,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20721115171909332,
      "step": 21028
    },
    {
      "epoch": 0.4206,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.007054646809895833,
      "learning_rate": 0.0001,
      "loss": 4.087,
      "loss/crossentropy": 2.1889408826828003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2252693474292755,
      "step": 21030
    },
    {
      "epoch": 0.42064,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.009051259358723958,
      "learning_rate": 0.0001,
      "loss": 4.2847,
      "loss/crossentropy": 1.9189648032188416,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20265216380357742,
      "step": 21032
    },
    {
      "epoch": 0.42068,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.009008534749348958,
      "learning_rate": 0.0001,
      "loss": 4.2777,
      "loss/crossentropy": 2.335593581199646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21812127530574799,
      "step": 21034
    },
    {
      "epoch": 0.42072,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.010723622639973958,
      "learning_rate": 0.0001,
      "loss": 3.9306,
      "loss/crossentropy": 2.0845232605934143,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1974613294005394,
      "step": 21036
    },
    {
      "epoch": 0.42076,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.010528310139973959,
      "learning_rate": 0.0001,
      "loss": 4.2117,
      "loss/crossentropy": 2.0062036514282227,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18701466172933578,
      "step": 21038
    },
    {
      "epoch": 0.4208,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.006520334879557292,
      "learning_rate": 0.0001,
      "loss": 4.1506,
      "loss/crossentropy": 2.30399227142334,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2108677700161934,
      "step": 21040
    },
    {
      "epoch": 0.42084,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006036122639973958,
      "learning_rate": 0.0001,
      "loss": 4.4162,
      "loss/crossentropy": 2.0211732387542725,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1925763636827469,
      "step": 21042
    },
    {
      "epoch": 0.42088,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.006986490885416667,
      "learning_rate": 0.0001,
      "loss": 4.2366,
      "loss/crossentropy": 2.259310483932495,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19453367590904236,
      "step": 21044
    },
    {
      "epoch": 0.42092,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.006818644205729167,
      "learning_rate": 0.0001,
      "loss": 3.8528,
      "loss/crossentropy": 2.225229859352112,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21418282389640808,
      "step": 21046
    },
    {
      "epoch": 0.42096,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.004667154947916667,
      "learning_rate": 0.0001,
      "loss": 3.993,
      "loss/crossentropy": 2.007095217704773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1860651671886444,
      "step": 21048
    },
    {
      "epoch": 0.421,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.004117838541666667,
      "learning_rate": 0.0001,
      "loss": 3.8646,
      "loss/crossentropy": 2.2098451256752014,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2123607099056244,
      "step": 21050
    },
    {
      "epoch": 0.42104,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0024861653645833334,
      "learning_rate": 0.0001,
      "loss": 3.9383,
      "loss/crossentropy": 2.038383424282074,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19423578679561615,
      "step": 21052
    },
    {
      "epoch": 0.42108,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.0047910054524739586,
      "learning_rate": 0.0001,
      "loss": 3.5858,
      "loss/crossentropy": 1.5344518423080444,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15087325125932693,
      "step": 21054
    },
    {
      "epoch": 0.42112,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0050351460774739586,
      "learning_rate": 0.0001,
      "loss": 3.7398,
      "loss/crossentropy": 2.0475123524665833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19996189326047897,
      "step": 21056
    },
    {
      "epoch": 0.42116,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0047604878743489586,
      "learning_rate": 0.0001,
      "loss": 3.8581,
      "loss/crossentropy": 1.9760377407073975,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20170825719833374,
      "step": 21058
    },
    {
      "epoch": 0.4212,
      "grad_norm": 2.25,
      "grad_norm_var": 0.009871419270833333,
      "learning_rate": 0.0001,
      "loss": 4.0876,
      "loss/crossentropy": 2.0718571543693542,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1987539902329445,
      "step": 21060
    },
    {
      "epoch": 0.42124,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.010908762613932291,
      "learning_rate": 0.0001,
      "loss": 3.7226,
      "loss/crossentropy": 1.694058895111084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15808547288179398,
      "step": 21062
    },
    {
      "epoch": 0.42128,
      "grad_norm": 2.0,
      "grad_norm_var": 0.010827382405598959,
      "learning_rate": 0.0001,
      "loss": 3.9852,
      "loss/crossentropy": 2.024729013442993,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19920113682746887,
      "step": 21064
    },
    {
      "epoch": 0.42132,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.01858495076497396,
      "learning_rate": 0.0001,
      "loss": 4.3691,
      "loss/crossentropy": 2.2473320960998535,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2366955801844597,
      "step": 21066
    },
    {
      "epoch": 0.42136,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.020230865478515624,
      "learning_rate": 0.0001,
      "loss": 4.014,
      "loss/crossentropy": 1.8571689128875732,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.198233962059021,
      "step": 21068
    },
    {
      "epoch": 0.4214,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.018314361572265625,
      "learning_rate": 0.0001,
      "loss": 4.066,
      "loss/crossentropy": 1.7178888320922852,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1756814494729042,
      "step": 21070
    },
    {
      "epoch": 0.42144,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.031840006510416664,
      "learning_rate": 0.0001,
      "loss": 4.0847,
      "loss/crossentropy": 2.5577595233917236,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22383909672498703,
      "step": 21072
    },
    {
      "epoch": 0.42148,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.030926259358723958,
      "learning_rate": 0.0001,
      "loss": 4.1138,
      "loss/crossentropy": 1.8517940640449524,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17597733438014984,
      "step": 21074
    },
    {
      "epoch": 0.42152,
      "grad_norm": 1.7265625,
      "grad_norm_var": 0.033934529622395834,
      "learning_rate": 0.0001,
      "loss": 4.1207,
      "loss/crossentropy": 2.2587300539016724,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20134828239679337,
      "step": 21076
    },
    {
      "epoch": 0.42156,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.04946263631184896,
      "learning_rate": 0.0001,
      "loss": 4.0223,
      "loss/crossentropy": 1.9288156032562256,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2372005432844162,
      "step": 21078
    },
    {
      "epoch": 0.4216,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.051454416910807294,
      "learning_rate": 0.0001,
      "loss": 3.9205,
      "loss/crossentropy": 1.7192566990852356,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19468504190444946,
      "step": 21080
    },
    {
      "epoch": 0.42164,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0494293212890625,
      "learning_rate": 0.0001,
      "loss": 4.1596,
      "loss/crossentropy": 2.2686651945114136,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18164453655481339,
      "step": 21082
    },
    {
      "epoch": 0.42168,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.048685455322265626,
      "learning_rate": 0.0001,
      "loss": 4.1317,
      "loss/crossentropy": 2.090322732925415,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19734591990709305,
      "step": 21084
    },
    {
      "epoch": 0.42172,
      "grad_norm": 2.0,
      "grad_norm_var": 0.047761027018229166,
      "learning_rate": 0.0001,
      "loss": 4.1072,
      "loss/crossentropy": 2.3203768730163574,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18960122764110565,
      "step": 21086
    },
    {
      "epoch": 0.42176,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.04411392211914063,
      "learning_rate": 0.0001,
      "loss": 4.1253,
      "loss/crossentropy": 2.1767812967300415,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20625658333301544,
      "step": 21088
    },
    {
      "epoch": 0.4218,
      "grad_norm": 1.875,
      "grad_norm_var": 0.0453125,
      "learning_rate": 0.0001,
      "loss": 3.9737,
      "loss/crossentropy": 2.1308469772338867,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20404084771871567,
      "step": 21090
    },
    {
      "epoch": 0.42184,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.03766988118489583,
      "learning_rate": 0.0001,
      "loss": 4.0643,
      "loss/crossentropy": 2.34736967086792,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22817441076040268,
      "step": 21092
    },
    {
      "epoch": 0.42188,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.019730631510416666,
      "learning_rate": 0.0001,
      "loss": 3.7592,
      "loss/crossentropy": 1.9515716433525085,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18946269899606705,
      "step": 21094
    },
    {
      "epoch": 0.42192,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.019181315104166666,
      "learning_rate": 0.0001,
      "loss": 4.2118,
      "loss/crossentropy": 2.1175013184547424,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25575874000787735,
      "step": 21096
    },
    {
      "epoch": 0.42196,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.019465128580729168,
      "learning_rate": 0.0001,
      "loss": 3.9934,
      "loss/crossentropy": 1.8839566707611084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1840147227048874,
      "step": 21098
    },
    {
      "epoch": 0.422,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.01761042277018229,
      "learning_rate": 0.0001,
      "loss": 3.9215,
      "loss/crossentropy": 1.8690025806427002,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20930250734090805,
      "step": 21100
    },
    {
      "epoch": 0.42204,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.01693700154622396,
      "learning_rate": 0.0001,
      "loss": 4.0877,
      "loss/crossentropy": 2.0368083119392395,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19566239416599274,
      "step": 21102
    },
    {
      "epoch": 0.42208,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.009384918212890624,
      "learning_rate": 0.0001,
      "loss": 3.9941,
      "loss/crossentropy": 1.5155547261238098,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1755489930510521,
      "step": 21104
    },
    {
      "epoch": 0.42212,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.006259918212890625,
      "learning_rate": 0.0001,
      "loss": 3.9463,
      "loss/crossentropy": 1.850695788860321,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17723418772220612,
      "step": 21106
    },
    {
      "epoch": 0.42216,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.0071441650390625,
      "learning_rate": 0.0001,
      "loss": 3.8903,
      "loss/crossentropy": 2.0554863810539246,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18637049198150635,
      "step": 21108
    },
    {
      "epoch": 0.4222,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0073883056640625,
      "learning_rate": 0.0001,
      "loss": 3.9719,
      "loss/crossentropy": 1.6186088919639587,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17206687480211258,
      "step": 21110
    },
    {
      "epoch": 0.42224,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.006131744384765625,
      "learning_rate": 0.0001,
      "loss": 4.1193,
      "loss/crossentropy": 2.326562762260437,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22168086469173431,
      "step": 21112
    },
    {
      "epoch": 0.42228,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.007670084635416667,
      "learning_rate": 0.0001,
      "loss": 3.8155,
      "loss/crossentropy": 1.8151599764823914,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16791456937789917,
      "step": 21114
    },
    {
      "epoch": 0.42232,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.007249959309895833,
      "learning_rate": 0.0001,
      "loss": 4.2537,
      "loss/crossentropy": 1.8597796559333801,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19767651706933975,
      "step": 21116
    },
    {
      "epoch": 0.42236,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0056884765625,
      "learning_rate": 0.0001,
      "loss": 4.0409,
      "loss/crossentropy": 1.9591187238693237,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18850377202033997,
      "step": 21118
    },
    {
      "epoch": 0.4224,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.007684071858723958,
      "learning_rate": 0.0001,
      "loss": 4.1848,
      "loss/crossentropy": 2.0987173318862915,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2087177261710167,
      "step": 21120
    },
    {
      "epoch": 0.42244,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011010487874348959,
      "learning_rate": 0.0001,
      "loss": 3.96,
      "loss/crossentropy": 1.8050614595413208,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19238876551389694,
      "step": 21122
    },
    {
      "epoch": 0.42248,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010487620035807292,
      "learning_rate": 0.0001,
      "loss": 4.1689,
      "loss/crossentropy": 2.0409955978393555,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19230370223522186,
      "step": 21124
    },
    {
      "epoch": 0.42252,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.009942372639973959,
      "learning_rate": 0.0001,
      "loss": 4.2033,
      "loss/crossentropy": 2.2344201803207397,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21372146904468536,
      "step": 21126
    },
    {
      "epoch": 0.42256,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010033162434895833,
      "learning_rate": 0.0001,
      "loss": 3.9102,
      "loss/crossentropy": 1.905364751815796,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17828336358070374,
      "step": 21128
    },
    {
      "epoch": 0.4226,
      "grad_norm": 8.6875,
      "grad_norm_var": 2.8468658447265627,
      "learning_rate": 0.0001,
      "loss": 4.4863,
      "loss/crossentropy": 2.118414044380188,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20183245092630386,
      "step": 21130
    },
    {
      "epoch": 0.42264,
      "grad_norm": 2.421875,
      "grad_norm_var": 2.826851399739583,
      "learning_rate": 0.0001,
      "loss": 3.9736,
      "loss/crossentropy": 1.8499796390533447,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17906417697668076,
      "step": 21132
    },
    {
      "epoch": 0.42268,
      "grad_norm": 2.0625,
      "grad_norm_var": 2.808918253580729,
      "learning_rate": 0.0001,
      "loss": 3.9669,
      "loss/crossentropy": 2.046007513999939,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20209533721208572,
      "step": 21134
    },
    {
      "epoch": 0.42272,
      "grad_norm": 2.078125,
      "grad_norm_var": 2.7881795247395833,
      "learning_rate": 0.0001,
      "loss": 4.4089,
      "loss/crossentropy": 2.4684035778045654,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23784233629703522,
      "step": 21136
    },
    {
      "epoch": 0.42276,
      "grad_norm": 1.9296875,
      "grad_norm_var": 2.781150054931641,
      "learning_rate": 0.0001,
      "loss": 3.9599,
      "loss/crossentropy": 2.2614521980285645,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20271501690149307,
      "step": 21138
    },
    {
      "epoch": 0.4228,
      "grad_norm": 1.9375,
      "grad_norm_var": 2.778649648030599,
      "learning_rate": 0.0001,
      "loss": 3.995,
      "loss/crossentropy": 1.9676395058631897,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1826155036687851,
      "step": 21140
    },
    {
      "epoch": 0.42284,
      "grad_norm": 2.015625,
      "grad_norm_var": 2.77156982421875,
      "learning_rate": 0.0001,
      "loss": 4.0974,
      "loss/crossentropy": 1.9320645928382874,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19518838077783585,
      "step": 21142
    },
    {
      "epoch": 0.42288,
      "grad_norm": 2.546875,
      "grad_norm_var": 2.751301066080729,
      "learning_rate": 0.0001,
      "loss": 4.2261,
      "loss/crossentropy": 2.0839805603027344,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20529074221849442,
      "step": 21144
    },
    {
      "epoch": 0.42292,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.04561538696289062,
      "learning_rate": 0.0001,
      "loss": 3.6995,
      "loss/crossentropy": 1.7191408276557922,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19476453214883804,
      "step": 21146
    },
    {
      "epoch": 0.42296,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.026151275634765624,
      "learning_rate": 0.0001,
      "loss": 4.3019,
      "loss/crossentropy": 2.3838316202163696,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2231682687997818,
      "step": 21148
    },
    {
      "epoch": 0.423,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.02588678995768229,
      "learning_rate": 0.0001,
      "loss": 4.3851,
      "loss/crossentropy": 2.5376522541046143,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22086318582296371,
      "step": 21150
    },
    {
      "epoch": 0.42304,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.0267486572265625,
      "learning_rate": 0.0001,
      "loss": 4.0375,
      "loss/crossentropy": 2.0386282801628113,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19398655742406845,
      "step": 21152
    },
    {
      "epoch": 0.42308,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.025047810872395833,
      "learning_rate": 0.0001,
      "loss": 4.0863,
      "loss/crossentropy": 1.8140466213226318,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1843537762761116,
      "step": 21154
    },
    {
      "epoch": 0.42312,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.026839192708333334,
      "learning_rate": 0.0001,
      "loss": 3.8526,
      "loss/crossentropy": 1.7746369242668152,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17255257815122604,
      "step": 21156
    },
    {
      "epoch": 0.42316,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0270904541015625,
      "learning_rate": 0.0001,
      "loss": 3.9869,
      "loss/crossentropy": 1.9952461123466492,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18875698745250702,
      "step": 21158
    },
    {
      "epoch": 0.4232,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0079742431640625,
      "learning_rate": 0.0001,
      "loss": 4.1292,
      "loss/crossentropy": 2.1353421211242676,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20545977354049683,
      "step": 21160
    },
    {
      "epoch": 0.42324,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0072662353515625,
      "learning_rate": 0.0001,
      "loss": 3.8585,
      "loss/crossentropy": 1.93438321352005,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2099536955356598,
      "step": 21162
    },
    {
      "epoch": 0.42328,
      "grad_norm": 2.125,
      "grad_norm_var": 0.008714803059895833,
      "learning_rate": 0.0001,
      "loss": 4.1526,
      "loss/crossentropy": 2.085531711578369,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19908609241247177,
      "step": 21164
    },
    {
      "epoch": 0.42332,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.010716756184895834,
      "learning_rate": 0.0001,
      "loss": 3.7828,
      "loss/crossentropy": 1.7960018515586853,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17612508684396744,
      "step": 21166
    },
    {
      "epoch": 0.42336,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.011982981363932292,
      "learning_rate": 0.0001,
      "loss": 4.0211,
      "loss/crossentropy": 2.3417539596557617,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2124033421278,
      "step": 21168
    },
    {
      "epoch": 0.4234,
      "grad_norm": 2.375,
      "grad_norm_var": 0.02384211222330729,
      "learning_rate": 0.0001,
      "loss": 4.0768,
      "loss/crossentropy": 1.7829700708389282,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17676468938589096,
      "step": 21170
    },
    {
      "epoch": 0.42344,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.027815500895182293,
      "learning_rate": 0.0001,
      "loss": 4.0011,
      "loss/crossentropy": 1.8243988156318665,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1778729408979416,
      "step": 21172
    },
    {
      "epoch": 0.42348,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.02774225870768229,
      "learning_rate": 0.0001,
      "loss": 4.1254,
      "loss/crossentropy": 2.0906782150268555,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19730936735868454,
      "step": 21174
    },
    {
      "epoch": 0.42352,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.026041666666666668,
      "learning_rate": 0.0001,
      "loss": 4.0611,
      "loss/crossentropy": 2.0885696411132812,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19815433770418167,
      "step": 21176
    },
    {
      "epoch": 0.42356,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.027513631184895835,
      "learning_rate": 0.0001,
      "loss": 3.7571,
      "loss/crossentropy": 1.9133245944976807,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18601036816835403,
      "step": 21178
    },
    {
      "epoch": 0.4236,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.026071929931640626,
      "learning_rate": 0.0001,
      "loss": 3.9254,
      "loss/crossentropy": 2.3077515363693237,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21091386675834656,
      "step": 21180
    },
    {
      "epoch": 0.42364,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.0247222900390625,
      "learning_rate": 0.0001,
      "loss": 3.9337,
      "loss/crossentropy": 2.170205235481262,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22082476317882538,
      "step": 21182
    },
    {
      "epoch": 0.42368,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0234619140625,
      "learning_rate": 0.0001,
      "loss": 4.1234,
      "loss/crossentropy": 2.053133964538574,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19210617244243622,
      "step": 21184
    },
    {
      "epoch": 0.42372,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.011930084228515625,
      "learning_rate": 0.0001,
      "loss": 4.1882,
      "loss/crossentropy": 2.246233820915222,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2110241949558258,
      "step": 21186
    },
    {
      "epoch": 0.42376,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.005378214518229166,
      "learning_rate": 0.0001,
      "loss": 4.0148,
      "loss/crossentropy": 2.088374972343445,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20536644011735916,
      "step": 21188
    },
    {
      "epoch": 0.4238,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.006105295817057292,
      "learning_rate": 0.0001,
      "loss": 4.08,
      "loss/crossentropy": 2.2355328798294067,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.208805114030838,
      "step": 21190
    },
    {
      "epoch": 0.42384,
      "grad_norm": 2.125,
      "grad_norm_var": 0.008707682291666666,
      "learning_rate": 0.0001,
      "loss": 4.3974,
      "loss/crossentropy": 2.3286044001579285,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21351350098848343,
      "step": 21192
    },
    {
      "epoch": 0.42388,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.008131663004557291,
      "learning_rate": 0.0001,
      "loss": 4.1763,
      "loss/crossentropy": 2.188312590122223,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2166001871228218,
      "step": 21194
    },
    {
      "epoch": 0.42392,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.007873280843098959,
      "learning_rate": 0.0001,
      "loss": 3.9396,
      "loss/crossentropy": 2.1229456663131714,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19442371279001236,
      "step": 21196
    },
    {
      "epoch": 0.42396,
      "grad_norm": 12.0,
      "grad_norm_var": 6.295235188802083,
      "learning_rate": 0.0001,
      "loss": 4.0976,
      "loss/crossentropy": 2.000286102294922,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20243800431489944,
      "step": 21198
    },
    {
      "epoch": 0.424,
      "grad_norm": 2.078125,
      "grad_norm_var": 6.265840657552084,
      "learning_rate": 0.0001,
      "loss": 4.204,
      "loss/crossentropy": 2.0628392100334167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20828108489513397,
      "step": 21200
    },
    {
      "epoch": 0.42404,
      "grad_norm": 2.09375,
      "grad_norm_var": 6.274006144205729,
      "learning_rate": 0.0001,
      "loss": 4.0036,
      "loss/crossentropy": 1.6969141364097595,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20169565826654434,
      "step": 21202
    },
    {
      "epoch": 0.42408,
      "grad_norm": 1.9609375,
      "grad_norm_var": 6.258829752604167,
      "learning_rate": 0.0001,
      "loss": 4.1475,
      "loss/crossentropy": 2.078153133392334,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1974920853972435,
      "step": 21204
    },
    {
      "epoch": 0.42412,
      "grad_norm": 2.03125,
      "grad_norm_var": 6.2670237223307295,
      "learning_rate": 0.0001,
      "loss": 3.7707,
      "loss/crossentropy": 1.9656822681427002,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20239949226379395,
      "step": 21206
    },
    {
      "epoch": 0.42416,
      "grad_norm": 1.890625,
      "grad_norm_var": 6.289948527018229,
      "learning_rate": 0.0001,
      "loss": 4.0624,
      "loss/crossentropy": 1.721840500831604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1692550852894783,
      "step": 21208
    },
    {
      "epoch": 0.4242,
      "grad_norm": 1.9375,
      "grad_norm_var": 6.2876942952473955,
      "learning_rate": 0.0001,
      "loss": 4.0437,
      "loss/crossentropy": 2.093570113182068,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2127639278769493,
      "step": 21210
    },
    {
      "epoch": 0.42424,
      "grad_norm": 2.09375,
      "grad_norm_var": 6.257684071858724,
      "learning_rate": 0.0001,
      "loss": 4.1692,
      "loss/crossentropy": 2.476475954055786,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22028189152479172,
      "step": 21212
    },
    {
      "epoch": 0.42428,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.015653228759765624,
      "learning_rate": 0.0001,
      "loss": 3.8084,
      "loss/crossentropy": 2.1308672428131104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2215510457754135,
      "step": 21214
    },
    {
      "epoch": 0.42432,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.014709218343098959,
      "learning_rate": 0.0001,
      "loss": 3.9183,
      "loss/crossentropy": 1.9697207808494568,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23525278270244598,
      "step": 21216
    },
    {
      "epoch": 0.42436,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.012943267822265625,
      "learning_rate": 0.0001,
      "loss": 4.3044,
      "loss/crossentropy": 2.3466382026672363,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21148498356342316,
      "step": 21218
    },
    {
      "epoch": 0.4244,
      "grad_norm": 2.0,
      "grad_norm_var": 0.013421376546223959,
      "learning_rate": 0.0001,
      "loss": 4.0281,
      "loss/crossentropy": 1.9455206990242004,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18886880576610565,
      "step": 21220
    },
    {
      "epoch": 0.42444,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.010949452718098959,
      "learning_rate": 0.0001,
      "loss": 4.113,
      "loss/crossentropy": 2.030101954936981,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19099503010511398,
      "step": 21222
    },
    {
      "epoch": 0.42448,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0103912353515625,
      "learning_rate": 0.0001,
      "loss": 3.9738,
      "loss/crossentropy": 2.0859753489494324,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20600532740354538,
      "step": 21224
    },
    {
      "epoch": 0.42452,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009749348958333333,
      "learning_rate": 0.0001,
      "loss": 4.0856,
      "loss/crossentropy": 1.9598749279975891,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18913735449314117,
      "step": 21226
    },
    {
      "epoch": 0.42456,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.011791737874348958,
      "learning_rate": 0.0001,
      "loss": 3.8938,
      "loss/crossentropy": 1.8889980912208557,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1892652064561844,
      "step": 21228
    },
    {
      "epoch": 0.4246,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.005224355061848958,
      "learning_rate": 0.0001,
      "loss": 4.0116,
      "loss/crossentropy": 2.0479432940483093,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20084280520677567,
      "step": 21230
    },
    {
      "epoch": 0.42464,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.005407460530598958,
      "learning_rate": 0.0001,
      "loss": 4.1743,
      "loss/crossentropy": 2.2049208879470825,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21012280881404877,
      "step": 21232
    },
    {
      "epoch": 0.42468,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.006945546468098958,
      "learning_rate": 0.0001,
      "loss": 3.8626,
      "loss/crossentropy": 2.1863686442375183,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19185375422239304,
      "step": 21234
    },
    {
      "epoch": 0.42472,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.010564931233723958,
      "learning_rate": 0.0001,
      "loss": 4.0699,
      "loss/crossentropy": 2.144856631755829,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2018456757068634,
      "step": 21236
    },
    {
      "epoch": 0.42476,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.0170806884765625,
      "learning_rate": 0.0001,
      "loss": 4.0695,
      "loss/crossentropy": 2.0092588663101196,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18666698783636093,
      "step": 21238
    },
    {
      "epoch": 0.4248,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.018365224202473957,
      "learning_rate": 0.0001,
      "loss": 3.8831,
      "loss/crossentropy": 1.9071536660194397,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1925549954175949,
      "step": 21240
    },
    {
      "epoch": 0.42484,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.01822509765625,
      "learning_rate": 0.0001,
      "loss": 3.7471,
      "loss/crossentropy": 1.7945414185523987,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18291736394166946,
      "step": 21242
    },
    {
      "epoch": 0.42488,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.017252604166666668,
      "learning_rate": 0.0001,
      "loss": 3.96,
      "loss/crossentropy": 2.0980335474014282,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20030297338962555,
      "step": 21244
    },
    {
      "epoch": 0.42492,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.01800715128580729,
      "learning_rate": 0.0001,
      "loss": 3.8988,
      "loss/crossentropy": 1.8301831483840942,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18173757195472717,
      "step": 21246
    },
    {
      "epoch": 0.42496,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.020734659830729165,
      "learning_rate": 0.0001,
      "loss": 3.8851,
      "loss/crossentropy": 2.1006619930267334,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21144358813762665,
      "step": 21248
    },
    {
      "epoch": 0.425,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.019220987955729168,
      "learning_rate": 0.0001,
      "loss": 4.1271,
      "loss/crossentropy": 2.2240719199180603,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21180906891822815,
      "step": 21250
    },
    {
      "epoch": 0.42504,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.024079386393229166,
      "learning_rate": 0.0001,
      "loss": 4.2143,
      "loss/crossentropy": 2.187661051750183,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19673307240009308,
      "step": 21252
    },
    {
      "epoch": 0.42508,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.07988993326822917,
      "learning_rate": 0.0001,
      "loss": 3.9913,
      "loss/crossentropy": 2.17081356048584,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19375403225421906,
      "step": 21254
    },
    {
      "epoch": 0.42512,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0834673563639323,
      "learning_rate": 0.0001,
      "loss": 3.7945,
      "loss/crossentropy": 1.6309250593185425,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17680996656417847,
      "step": 21256
    },
    {
      "epoch": 0.42516,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.08495992024739583,
      "learning_rate": 0.0001,
      "loss": 3.97,
      "loss/crossentropy": 1.9581794142723083,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18848375976085663,
      "step": 21258
    },
    {
      "epoch": 0.4252,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0839508056640625,
      "learning_rate": 0.0001,
      "loss": 4.1796,
      "loss/crossentropy": 1.9206833839416504,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18920007348060608,
      "step": 21260
    },
    {
      "epoch": 0.42524,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.08524169921875,
      "learning_rate": 0.0001,
      "loss": 3.8909,
      "loss/crossentropy": 1.736355721950531,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17645896971225739,
      "step": 21262
    },
    {
      "epoch": 0.42528,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.0872711181640625,
      "learning_rate": 0.0001,
      "loss": 4.0856,
      "loss/crossentropy": 1.9830248355865479,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18551607429981232,
      "step": 21264
    },
    {
      "epoch": 0.42532,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.08831558227539063,
      "learning_rate": 0.0001,
      "loss": 3.8254,
      "loss/crossentropy": 1.8477718234062195,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18887290358543396,
      "step": 21266
    },
    {
      "epoch": 0.42536,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.08277359008789062,
      "learning_rate": 0.0001,
      "loss": 3.8833,
      "loss/crossentropy": 2.024181544780731,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20996837317943573,
      "step": 21268
    },
    {
      "epoch": 0.4254,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.021930948893229166,
      "learning_rate": 0.0001,
      "loss": 4.0265,
      "loss/crossentropy": 2.153610110282898,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1962708830833435,
      "step": 21270
    },
    {
      "epoch": 0.42544,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.019359334309895834,
      "learning_rate": 0.0001,
      "loss": 4.1288,
      "loss/crossentropy": 2.142420172691345,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2167380526661873,
      "step": 21272
    },
    {
      "epoch": 0.42548,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.017235310872395833,
      "learning_rate": 0.0001,
      "loss": 4.1983,
      "loss/crossentropy": 2.091115117073059,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20703484117984772,
      "step": 21274
    },
    {
      "epoch": 0.42552,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.017235310872395833,
      "learning_rate": 0.0001,
      "loss": 4.1653,
      "loss/crossentropy": 2.0014833211898804,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2172148898243904,
      "step": 21276
    },
    {
      "epoch": 0.42556,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.015340169270833334,
      "learning_rate": 0.0001,
      "loss": 3.9513,
      "loss/crossentropy": 1.8580491542816162,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1861124113202095,
      "step": 21278
    },
    {
      "epoch": 0.4256,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.0071980794270833336,
      "learning_rate": 0.0001,
      "loss": 4.1013,
      "loss/crossentropy": 2.1323113441467285,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1839580088853836,
      "step": 21280
    },
    {
      "epoch": 0.42564,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.005952707926432292,
      "learning_rate": 0.0001,
      "loss": 4.0295,
      "loss/crossentropy": 2.107018530368805,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19513815641403198,
      "step": 21282
    },
    {
      "epoch": 0.42568,
      "grad_norm": 2.0,
      "grad_norm_var": 0.005882771809895834,
      "learning_rate": 0.0001,
      "loss": 3.989,
      "loss/crossentropy": 2.025463044643402,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19870947301387787,
      "step": 21284
    },
    {
      "epoch": 0.42572,
      "grad_norm": 2.125,
      "grad_norm_var": 0.007429758707682292,
      "learning_rate": 0.0001,
      "loss": 4.2469,
      "loss/crossentropy": 2.348206877708435,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2326420098543167,
      "step": 21286
    },
    {
      "epoch": 0.42576,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.005037434895833333,
      "learning_rate": 0.0001,
      "loss": 4.1699,
      "loss/crossentropy": 2.2780548334121704,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1905437633395195,
      "step": 21288
    },
    {
      "epoch": 0.4258,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.004400380452473958,
      "learning_rate": 0.0001,
      "loss": 4.1373,
      "loss/crossentropy": 2.1915369629859924,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19397883117198944,
      "step": 21290
    },
    {
      "epoch": 0.42584,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.004353586832682292,
      "learning_rate": 0.0001,
      "loss": 4.0237,
      "loss/crossentropy": 2.132679283618927,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21279992163181305,
      "step": 21292
    },
    {
      "epoch": 0.42588,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.006058756510416667,
      "learning_rate": 0.0001,
      "loss": 3.9847,
      "loss/crossentropy": 1.860496699810028,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1666220724582672,
      "step": 21294
    },
    {
      "epoch": 0.42592,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.005535634358723959,
      "learning_rate": 0.0001,
      "loss": 4.1154,
      "loss/crossentropy": 2.031413435935974,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1919916793704033,
      "step": 21296
    },
    {
      "epoch": 0.42596,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.005296834309895833,
      "learning_rate": 0.0001,
      "loss": 3.7703,
      "loss/crossentropy": 1.6095055937767029,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1592901274561882,
      "step": 21298
    },
    {
      "epoch": 0.426,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0052886962890625,
      "learning_rate": 0.0001,
      "loss": 4.1652,
      "loss/crossentropy": 2.002205550670624,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18984580785036087,
      "step": 21300
    },
    {
      "epoch": 0.42604,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0029856363932291665,
      "learning_rate": 0.0001,
      "loss": 4.1755,
      "loss/crossentropy": 2.022138476371765,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1904059648513794,
      "step": 21302
    },
    {
      "epoch": 0.42608,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.0036801656087239584,
      "learning_rate": 0.0001,
      "loss": 3.8417,
      "loss/crossentropy": 1.917612910270691,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18372280895709991,
      "step": 21304
    },
    {
      "epoch": 0.42612,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.004426829020182292,
      "learning_rate": 0.0001,
      "loss": 3.8031,
      "loss/crossentropy": 1.827630877494812,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18420201539993286,
      "step": 21306
    },
    {
      "epoch": 0.42616,
      "grad_norm": 1.875,
      "grad_norm_var": 0.004115549723307291,
      "learning_rate": 0.0001,
      "loss": 3.8427,
      "loss/crossentropy": 1.7498629689216614,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1638321429491043,
      "step": 21308
    },
    {
      "epoch": 0.4262,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0036374409993489582,
      "learning_rate": 0.0001,
      "loss": 4.0028,
      "loss/crossentropy": 1.7890136241912842,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18237848579883575,
      "step": 21310
    },
    {
      "epoch": 0.42624,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0033518473307291665,
      "learning_rate": 0.0001,
      "loss": 4.1649,
      "loss/crossentropy": 2.17901873588562,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2104976773262024,
      "step": 21312
    },
    {
      "epoch": 0.42628,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.003987630208333333,
      "learning_rate": 0.0001,
      "loss": 4.2149,
      "loss/crossentropy": 2.322218656539917,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20789335668087006,
      "step": 21314
    },
    {
      "epoch": 0.42632,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.004515584309895833,
      "learning_rate": 0.0001,
      "loss": 4.2208,
      "loss/crossentropy": 2.4203622341156006,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21352334320545197,
      "step": 21316
    },
    {
      "epoch": 0.42636,
      "grad_norm": 2.125,
      "grad_norm_var": 0.008388010660807292,
      "learning_rate": 0.0001,
      "loss": 3.7956,
      "loss/crossentropy": 2.2349472045898438,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2009701356291771,
      "step": 21318
    },
    {
      "epoch": 0.4264,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.009511057535807292,
      "learning_rate": 0.0001,
      "loss": 3.9733,
      "loss/crossentropy": 1.8173826336860657,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1736140102148056,
      "step": 21320
    },
    {
      "epoch": 0.42644,
      "grad_norm": 1.7421875,
      "grad_norm_var": 0.011315663655598959,
      "learning_rate": 0.0001,
      "loss": 3.6665,
      "loss/crossentropy": 2.099972426891327,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19424229115247726,
      "step": 21322
    },
    {
      "epoch": 0.42648,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.06534830729166667,
      "learning_rate": 0.0001,
      "loss": 3.9213,
      "loss/crossentropy": 1.8005958795547485,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17232687771320343,
      "step": 21324
    },
    {
      "epoch": 0.42652,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.06697184244791667,
      "learning_rate": 0.0001,
      "loss": 4.1434,
      "loss/crossentropy": 2.0184829235076904,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19896141439676285,
      "step": 21326
    },
    {
      "epoch": 0.42656,
      "grad_norm": 1.7421875,
      "grad_norm_var": 0.07361551920572916,
      "learning_rate": 0.0001,
      "loss": 3.7018,
      "loss/crossentropy": 2.0207254886627197,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1863861232995987,
      "step": 21328
    },
    {
      "epoch": 0.4266,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.07342020670572917,
      "learning_rate": 0.0001,
      "loss": 4.2953,
      "loss/crossentropy": 2.1894538402557373,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21097677946090698,
      "step": 21330
    },
    {
      "epoch": 0.42664,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.07691650390625,
      "learning_rate": 0.0001,
      "loss": 4.1044,
      "loss/crossentropy": 2.1939873695373535,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1935359239578247,
      "step": 21332
    },
    {
      "epoch": 0.42668,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.07349624633789062,
      "learning_rate": 0.0001,
      "loss": 4.1346,
      "loss/crossentropy": 2.131330966949463,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2001090720295906,
      "step": 21334
    },
    {
      "epoch": 0.42672,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.083154296875,
      "learning_rate": 0.0001,
      "loss": 3.9866,
      "loss/crossentropy": 1.9254841208457947,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20036878436803818,
      "step": 21336
    },
    {
      "epoch": 0.42676,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.07875874837239584,
      "learning_rate": 0.0001,
      "loss": 4.0641,
      "loss/crossentropy": 1.9841803312301636,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19805853068828583,
      "step": 21338
    },
    {
      "epoch": 0.4268,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.02767918904622396,
      "learning_rate": 0.0001,
      "loss": 3.6806,
      "loss/crossentropy": 1.6822729110717773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18236130475997925,
      "step": 21340
    },
    {
      "epoch": 0.42684,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.023789215087890624,
      "learning_rate": 0.0001,
      "loss": 4.1128,
      "loss/crossentropy": 2.0904295444488525,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19326913356781006,
      "step": 21342
    },
    {
      "epoch": 0.42688,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.01951878865559896,
      "learning_rate": 0.0001,
      "loss": 4.0758,
      "loss/crossentropy": 2.111131250858307,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.202858105301857,
      "step": 21344
    },
    {
      "epoch": 0.42692,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.019197591145833335,
      "learning_rate": 0.0001,
      "loss": 4.2061,
      "loss/crossentropy": 2.2793352603912354,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20112870633602142,
      "step": 21346
    },
    {
      "epoch": 0.42696,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.016112263997395834,
      "learning_rate": 0.0001,
      "loss": 4.1173,
      "loss/crossentropy": 2.253583312034607,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21494006365537643,
      "step": 21348
    },
    {
      "epoch": 0.427,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.016532389322916667,
      "learning_rate": 0.0001,
      "loss": 4.2371,
      "loss/crossentropy": 1.9753515124320984,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1828732267022133,
      "step": 21350
    },
    {
      "epoch": 0.42704,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0023455301920572917,
      "learning_rate": 0.0001,
      "loss": 4.0961,
      "loss/crossentropy": 2.110904037952423,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19640487432479858,
      "step": 21352
    },
    {
      "epoch": 0.42708,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0022905985514322918,
      "learning_rate": 0.0001,
      "loss": 4.1961,
      "loss/crossentropy": 2.323657512664795,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2206008955836296,
      "step": 21354
    },
    {
      "epoch": 0.42712,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.003507232666015625,
      "learning_rate": 0.0001,
      "loss": 4.1613,
      "loss/crossentropy": 2.113715887069702,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18136650323867798,
      "step": 21356
    },
    {
      "epoch": 0.42716,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.004207102457682291,
      "learning_rate": 0.0001,
      "loss": 3.9951,
      "loss/crossentropy": 2.096550941467285,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20972034335136414,
      "step": 21358
    },
    {
      "epoch": 0.4272,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.004449208577473958,
      "learning_rate": 0.0001,
      "loss": 4.0398,
      "loss/crossentropy": 2.1959888339042664,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19834591448307037,
      "step": 21360
    },
    {
      "epoch": 0.42724,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.0054351806640625,
      "learning_rate": 0.0001,
      "loss": 4.0003,
      "loss/crossentropy": 2.146053671836853,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18752741813659668,
      "step": 21362
    },
    {
      "epoch": 0.42728,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.005728912353515625,
      "learning_rate": 0.0001,
      "loss": 3.6564,
      "loss/crossentropy": 1.9178010821342468,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20413734018802643,
      "step": 21364
    },
    {
      "epoch": 0.42732,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.005283355712890625,
      "learning_rate": 0.0001,
      "loss": 3.884,
      "loss/crossentropy": 1.8283003568649292,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1882033348083496,
      "step": 21366
    },
    {
      "epoch": 0.42736,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0103759765625,
      "learning_rate": 0.0001,
      "loss": 4.362,
      "loss/crossentropy": 2.1956799030303955,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21325727552175522,
      "step": 21368
    },
    {
      "epoch": 0.4274,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.010381825764973958,
      "learning_rate": 0.0001,
      "loss": 3.9304,
      "loss/crossentropy": 1.8081734776496887,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1818179190158844,
      "step": 21370
    },
    {
      "epoch": 0.42744,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.008699544270833333,
      "learning_rate": 0.0001,
      "loss": 4.3029,
      "loss/crossentropy": 2.283393979072571,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20135364681482315,
      "step": 21372
    },
    {
      "epoch": 0.42748,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.013038889567057291,
      "learning_rate": 0.0001,
      "loss": 3.7118,
      "loss/crossentropy": 1.9896268844604492,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17712793499231339,
      "step": 21374
    },
    {
      "epoch": 0.42752,
      "grad_norm": 2.0,
      "grad_norm_var": 0.017289225260416666,
      "learning_rate": 0.0001,
      "loss": 4.2804,
      "loss/crossentropy": 2.1862595081329346,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19685917347669601,
      "step": 21376
    },
    {
      "epoch": 0.42756,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.016355133056640624,
      "learning_rate": 0.0001,
      "loss": 3.9852,
      "loss/crossentropy": 2.097273588180542,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.204267680644989,
      "step": 21378
    },
    {
      "epoch": 0.4276,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.016031646728515626,
      "learning_rate": 0.0001,
      "loss": 4.0112,
      "loss/crossentropy": 2.0905693769454956,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17958275228738785,
      "step": 21380
    },
    {
      "epoch": 0.42764,
      "grad_norm": 2.0,
      "grad_norm_var": 0.016502888997395833,
      "learning_rate": 0.0001,
      "loss": 4.2815,
      "loss/crossentropy": 2.1066064834594727,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18819285184144974,
      "step": 21382
    },
    {
      "epoch": 0.42768,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0134429931640625,
      "learning_rate": 0.0001,
      "loss": 4.2721,
      "loss/crossentropy": 1.9919481873512268,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19579682499170303,
      "step": 21384
    },
    {
      "epoch": 0.42772,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.01337890625,
      "learning_rate": 0.0001,
      "loss": 3.9186,
      "loss/crossentropy": 1.8611761927604675,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1929011344909668,
      "step": 21386
    },
    {
      "epoch": 0.42776,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.015592193603515625,
      "learning_rate": 0.0001,
      "loss": 3.9346,
      "loss/crossentropy": 2.232940196990967,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1934458464384079,
      "step": 21388
    },
    {
      "epoch": 0.4278,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0073883056640625,
      "learning_rate": 0.0001,
      "loss": 4.2149,
      "loss/crossentropy": 2.2017111778259277,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22515927255153656,
      "step": 21390
    },
    {
      "epoch": 0.42784,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0079498291015625,
      "learning_rate": 0.0001,
      "loss": 4.3437,
      "loss/crossentropy": 2.4802298545837402,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20159347355365753,
      "step": 21392
    },
    {
      "epoch": 0.42788,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0065093994140625,
      "learning_rate": 0.0001,
      "loss": 4.1095,
      "loss/crossentropy": 2.1320372819900513,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19364330172538757,
      "step": 21394
    },
    {
      "epoch": 0.42792,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.0069488525390625,
      "learning_rate": 0.0001,
      "loss": 4.2004,
      "loss/crossentropy": 2.115567684173584,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2150493860244751,
      "step": 21396
    },
    {
      "epoch": 0.42796,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.008259073893229166,
      "learning_rate": 0.0001,
      "loss": 4.4131,
      "loss/crossentropy": 2.3852893114089966,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21371107548475266,
      "step": 21398
    },
    {
      "epoch": 0.428,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.010204060872395834,
      "learning_rate": 0.0001,
      "loss": 3.9377,
      "loss/crossentropy": 1.872315526008606,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17347048223018646,
      "step": 21400
    },
    {
      "epoch": 0.42804,
      "grad_norm": 3.9375,
      "grad_norm_var": 0.23815816243489582,
      "learning_rate": 0.0001,
      "loss": 4.0048,
      "loss/crossentropy": 2.064896881580353,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20905783772468567,
      "step": 21402
    },
    {
      "epoch": 0.42808,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.24268290201822917,
      "learning_rate": 0.0001,
      "loss": 3.6606,
      "loss/crossentropy": 1.90069580078125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19062762707471848,
      "step": 21404
    },
    {
      "epoch": 0.42812,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.24911473592122396,
      "learning_rate": 0.0001,
      "loss": 4.1083,
      "loss/crossentropy": 1.9533473253250122,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17531338334083557,
      "step": 21406
    },
    {
      "epoch": 0.42816,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.25371475219726564,
      "learning_rate": 0.0001,
      "loss": 3.9376,
      "loss/crossentropy": 1.9835584163665771,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19894006848335266,
      "step": 21408
    },
    {
      "epoch": 0.4282,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.26023661295572914,
      "learning_rate": 0.0001,
      "loss": 3.6515,
      "loss/crossentropy": 2.2370243072509766,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20410479605197906,
      "step": 21410
    },
    {
      "epoch": 0.42824,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.2575152079264323,
      "learning_rate": 0.0001,
      "loss": 3.8597,
      "loss/crossentropy": 2.0012041330337524,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18770772218704224,
      "step": 21412
    },
    {
      "epoch": 0.42828,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.2589312235514323,
      "learning_rate": 0.0001,
      "loss": 4.1147,
      "loss/crossentropy": 2.0049465894699097,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19158685207366943,
      "step": 21414
    },
    {
      "epoch": 0.42832,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.2571207682291667,
      "learning_rate": 0.0001,
      "loss": 3.9892,
      "loss/crossentropy": 2.1261045932769775,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21888310462236404,
      "step": 21416
    },
    {
      "epoch": 0.42836,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.007771809895833333,
      "learning_rate": 0.0001,
      "loss": 4.2261,
      "loss/crossentropy": 2.3438754081726074,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22312359511852264,
      "step": 21418
    },
    {
      "epoch": 0.4284,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0075457255045572914,
      "learning_rate": 0.0001,
      "loss": 4.0333,
      "loss/crossentropy": 1.970156729221344,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17968790978193283,
      "step": 21420
    },
    {
      "epoch": 0.42844,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.007356516520182292,
      "learning_rate": 0.0001,
      "loss": 4.1106,
      "loss/crossentropy": 2.0128689408302307,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.190870463848114,
      "step": 21422
    },
    {
      "epoch": 0.42848,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.044755045572916666,
      "learning_rate": 0.0001,
      "loss": 4.0063,
      "loss/crossentropy": 1.766348421573639,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1758265122771263,
      "step": 21424
    },
    {
      "epoch": 0.42852,
      "grad_norm": 2.125,
      "grad_norm_var": 0.041757965087890626,
      "learning_rate": 0.0001,
      "loss": 4.1498,
      "loss/crossentropy": 1.9646649360656738,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2052430659532547,
      "step": 21426
    },
    {
      "epoch": 0.42856,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.042281087239583334,
      "learning_rate": 0.0001,
      "loss": 4.0817,
      "loss/crossentropy": 1.8953626155853271,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1974174976348877,
      "step": 21428
    },
    {
      "epoch": 0.4286,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.04601008097330729,
      "learning_rate": 0.0001,
      "loss": 3.7752,
      "loss/crossentropy": 1.9713319540023804,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18385013192892075,
      "step": 21430
    },
    {
      "epoch": 0.42864,
      "grad_norm": 1.875,
      "grad_norm_var": 0.047501627604166666,
      "learning_rate": 0.0001,
      "loss": 4.1555,
      "loss/crossentropy": 2.2564018964767456,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19498983025550842,
      "step": 21432
    },
    {
      "epoch": 0.42868,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.05091044108072917,
      "learning_rate": 0.0001,
      "loss": 4.1351,
      "loss/crossentropy": 2.1691616773605347,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22087589651346207,
      "step": 21434
    },
    {
      "epoch": 0.42872,
      "grad_norm": 9.1875,
      "grad_norm_var": 3.257096354166667,
      "learning_rate": 0.0001,
      "loss": 4.4825,
      "loss/crossentropy": 1.9714866280555725,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19079908728599548,
      "step": 21436
    },
    {
      "epoch": 0.42876,
      "grad_norm": 2.0,
      "grad_norm_var": 3.2328834533691406,
      "learning_rate": 0.0001,
      "loss": 4.0687,
      "loss/crossentropy": 2.2311829328536987,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19929265975952148,
      "step": 21438
    },
    {
      "epoch": 0.4288,
      "grad_norm": 1.984375,
      "grad_norm_var": 3.231501261393229,
      "learning_rate": 0.0001,
      "loss": 4.0397,
      "loss/crossentropy": 1.9644780158996582,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1768122911453247,
      "step": 21440
    },
    {
      "epoch": 0.42884,
      "grad_norm": 1.796875,
      "grad_norm_var": 3.253226725260417,
      "learning_rate": 0.0001,
      "loss": 3.9993,
      "loss/crossentropy": 1.93829345703125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19605614244937897,
      "step": 21442
    },
    {
      "epoch": 0.42888,
      "grad_norm": 1.9140625,
      "grad_norm_var": 3.2553304036458335,
      "learning_rate": 0.0001,
      "loss": 3.8987,
      "loss/crossentropy": 1.9488537907600403,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20282328873872757,
      "step": 21444
    },
    {
      "epoch": 0.42892,
      "grad_norm": 1.75,
      "grad_norm_var": 3.254095204671224,
      "learning_rate": 0.0001,
      "loss": 3.8275,
      "loss/crossentropy": 1.93411123752594,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18476436287164688,
      "step": 21446
    },
    {
      "epoch": 0.42896,
      "grad_norm": 1.8828125,
      "grad_norm_var": 3.24100341796875,
      "learning_rate": 0.0001,
      "loss": 4.1485,
      "loss/crossentropy": 2.4150885343551636,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21736841648817062,
      "step": 21448
    },
    {
      "epoch": 0.429,
      "grad_norm": 2.078125,
      "grad_norm_var": 3.2536610921223956,
      "learning_rate": 0.0001,
      "loss": 4.227,
      "loss/crossentropy": 2.117435574531555,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20417648553848267,
      "step": 21450
    },
    {
      "epoch": 0.42904,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.014115142822265624,
      "learning_rate": 0.0001,
      "loss": 3.9162,
      "loss/crossentropy": 2.0840513706207275,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20294595509767532,
      "step": 21452
    },
    {
      "epoch": 0.42908,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0087066650390625,
      "learning_rate": 0.0001,
      "loss": 3.9088,
      "loss/crossentropy": 2.1203198432922363,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19258832931518555,
      "step": 21454
    },
    {
      "epoch": 0.42912,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.007503000895182291,
      "learning_rate": 0.0001,
      "loss": 4.1205,
      "loss/crossentropy": 2.1161458492279053,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18999946117401123,
      "step": 21456
    },
    {
      "epoch": 0.42916,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.006876373291015625,
      "learning_rate": 0.0001,
      "loss": 3.9288,
      "loss/crossentropy": 1.7208130955696106,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20022748410701752,
      "step": 21458
    },
    {
      "epoch": 0.4292,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.007940419514973958,
      "learning_rate": 0.0001,
      "loss": 4.1794,
      "loss/crossentropy": 2.0148571729660034,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2059563249349594,
      "step": 21460
    },
    {
      "epoch": 0.42924,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.006615193684895834,
      "learning_rate": 0.0001,
      "loss": 4.0039,
      "loss/crossentropy": 1.985020637512207,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18583452701568604,
      "step": 21462
    },
    {
      "epoch": 0.42928,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0056955973307291664,
      "learning_rate": 0.0001,
      "loss": 3.9623,
      "loss/crossentropy": 1.8767080903053284,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18215155601501465,
      "step": 21464
    },
    {
      "epoch": 0.42932,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0037638346354166665,
      "learning_rate": 0.0001,
      "loss": 4.1592,
      "loss/crossentropy": 2.162920832633972,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20505908131599426,
      "step": 21466
    },
    {
      "epoch": 0.42936,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.00985107421875,
      "learning_rate": 0.0001,
      "loss": 4.0253,
      "loss/crossentropy": 2.329980731010437,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23726581782102585,
      "step": 21468
    },
    {
      "epoch": 0.4294,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.01512451171875,
      "learning_rate": 0.0001,
      "loss": 4.0204,
      "loss/crossentropy": 2.020410180091858,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19389280676841736,
      "step": 21470
    },
    {
      "epoch": 0.42944,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.01613133748372396,
      "learning_rate": 0.0001,
      "loss": 4.0849,
      "loss/crossentropy": 2.157116711139679,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1991463601589203,
      "step": 21472
    },
    {
      "epoch": 0.42948,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.014992014567057291,
      "learning_rate": 0.0001,
      "loss": 4.0913,
      "loss/crossentropy": 2.0108723640441895,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19546864926815033,
      "step": 21474
    },
    {
      "epoch": 0.42952,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0141998291015625,
      "learning_rate": 0.0001,
      "loss": 4.2555,
      "loss/crossentropy": 2.154988646507263,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19204716384410858,
      "step": 21476
    },
    {
      "epoch": 0.42956,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.013549550374348959,
      "learning_rate": 0.0001,
      "loss": 3.7924,
      "loss/crossentropy": 1.9463204145431519,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2041161060333252,
      "step": 21478
    },
    {
      "epoch": 0.4296,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.013459269205729167,
      "learning_rate": 0.0001,
      "loss": 4.1798,
      "loss/crossentropy": 2.3123443126678467,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21649713814258575,
      "step": 21480
    },
    {
      "epoch": 0.42964,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.015160115559895833,
      "learning_rate": 0.0001,
      "loss": 3.8571,
      "loss/crossentropy": 1.658549726009369,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16132741421461105,
      "step": 21482
    },
    {
      "epoch": 0.42968,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.010282389322916667,
      "learning_rate": 0.0001,
      "loss": 3.8965,
      "loss/crossentropy": 1.9251441955566406,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17892278730869293,
      "step": 21484
    },
    {
      "epoch": 0.42972,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.004523722330729166,
      "learning_rate": 0.0001,
      "loss": 4.002,
      "loss/crossentropy": 1.7956182956695557,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17938754707574844,
      "step": 21486
    },
    {
      "epoch": 0.42976,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.004312896728515625,
      "learning_rate": 0.0001,
      "loss": 4.0373,
      "loss/crossentropy": 1.941636562347412,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2159031555056572,
      "step": 21488
    },
    {
      "epoch": 0.4298,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0047108968098958336,
      "learning_rate": 0.0001,
      "loss": 4.2861,
      "loss/crossentropy": 2.4158248901367188,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2346770316362381,
      "step": 21490
    },
    {
      "epoch": 0.42984,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.009110260009765624,
      "learning_rate": 0.0001,
      "loss": 4.0334,
      "loss/crossentropy": 1.749859094619751,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17030366510152817,
      "step": 21492
    },
    {
      "epoch": 0.42988,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.008622233072916667,
      "learning_rate": 0.0001,
      "loss": 4.1944,
      "loss/crossentropy": 2.0735827684402466,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1986066773533821,
      "step": 21494
    },
    {
      "epoch": 0.42992,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.008556874593098958,
      "learning_rate": 0.0001,
      "loss": 3.8175,
      "loss/crossentropy": 1.744678020477295,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16986121982336044,
      "step": 21496
    },
    {
      "epoch": 0.42996,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.007559967041015625,
      "learning_rate": 0.0001,
      "loss": 4.116,
      "loss/crossentropy": 2.060591220855713,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20758916437625885,
      "step": 21498
    },
    {
      "epoch": 0.43,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.0074045817057291664,
      "learning_rate": 0.0001,
      "loss": 3.967,
      "loss/crossentropy": 2.2836644649505615,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20620425045490265,
      "step": 21500
    },
    {
      "epoch": 0.43004,
      "grad_norm": 1.7421875,
      "grad_norm_var": 0.010252888997395833,
      "learning_rate": 0.0001,
      "loss": 3.9827,
      "loss/crossentropy": 2.1133594512939453,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1954261213541031,
      "step": 21502
    },
    {
      "epoch": 0.43008,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0100250244140625,
      "learning_rate": 0.0001,
      "loss": 4.0973,
      "loss/crossentropy": 2.5492414236068726,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23168637603521347,
      "step": 21504
    },
    {
      "epoch": 0.43012,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.0113922119140625,
      "learning_rate": 0.0001,
      "loss": 3.6322,
      "loss/crossentropy": 1.7122142314910889,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16849397122859955,
      "step": 21506
    },
    {
      "epoch": 0.43016,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0077288309733072914,
      "learning_rate": 0.0001,
      "loss": 4.0675,
      "loss/crossentropy": 2.3052932024002075,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21252303570508957,
      "step": 21508
    },
    {
      "epoch": 0.4302,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.009570058186848958,
      "learning_rate": 0.0001,
      "loss": 3.9751,
      "loss/crossentropy": 2.5156666040420532,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26235880702733994,
      "step": 21510
    },
    {
      "epoch": 0.43024,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.009419504801432292,
      "learning_rate": 0.0001,
      "loss": 3.8061,
      "loss/crossentropy": 1.884554922580719,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1867503896355629,
      "step": 21512
    },
    {
      "epoch": 0.43028,
      "grad_norm": 2.25,
      "grad_norm_var": 0.01619440714518229,
      "learning_rate": 0.0001,
      "loss": 4.1092,
      "loss/crossentropy": 2.1004000902175903,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19210458546876907,
      "step": 21514
    },
    {
      "epoch": 0.43032,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.01565526326497396,
      "learning_rate": 0.0001,
      "loss": 4.0146,
      "loss/crossentropy": 1.606240451335907,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15947996824979782,
      "step": 21516
    },
    {
      "epoch": 0.43036,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.013130696614583333,
      "learning_rate": 0.0001,
      "loss": 4.1195,
      "loss/crossentropy": 1.9417288303375244,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19736050814390182,
      "step": 21518
    },
    {
      "epoch": 0.4304,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.014204915364583333,
      "learning_rate": 0.0001,
      "loss": 3.8322,
      "loss/crossentropy": 1.9607113599777222,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17349773645401,
      "step": 21520
    },
    {
      "epoch": 0.43044,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.012105305989583334,
      "learning_rate": 0.0001,
      "loss": 3.9789,
      "loss/crossentropy": 1.9606893062591553,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1726546436548233,
      "step": 21522
    },
    {
      "epoch": 0.43048,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.011915842692057291,
      "learning_rate": 0.0001,
      "loss": 4.1203,
      "loss/crossentropy": 1.9426813125610352,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19272266328334808,
      "step": 21524
    },
    {
      "epoch": 0.43052,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.011533355712890625,
      "learning_rate": 0.0001,
      "loss": 3.8158,
      "loss/crossentropy": 2.1843879222869873,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1986662521958351,
      "step": 21526
    },
    {
      "epoch": 0.43056,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.011146799723307291,
      "learning_rate": 0.0001,
      "loss": 3.8314,
      "loss/crossentropy": 1.7259829640388489,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1759045571088791,
      "step": 21528
    },
    {
      "epoch": 0.4306,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.004670969645182292,
      "learning_rate": 0.0001,
      "loss": 3.7475,
      "loss/crossentropy": 1.7714526653289795,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20490919053554535,
      "step": 21530
    },
    {
      "epoch": 0.43064,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.005631256103515625,
      "learning_rate": 0.0001,
      "loss": 3.905,
      "loss/crossentropy": 1.9385235905647278,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19307191669940948,
      "step": 21532
    },
    {
      "epoch": 0.43068,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.006034088134765625,
      "learning_rate": 0.0001,
      "loss": 4.2582,
      "loss/crossentropy": 1.956704020500183,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24551841616630554,
      "step": 21534
    },
    {
      "epoch": 0.43072,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0048736572265625,
      "learning_rate": 0.0001,
      "loss": 4.0436,
      "loss/crossentropy": 2.0809181332588196,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20348872244358063,
      "step": 21536
    },
    {
      "epoch": 0.43076,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.004589589436848959,
      "learning_rate": 0.0001,
      "loss": 4.1488,
      "loss/crossentropy": 1.9148198366165161,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19790956377983093,
      "step": 21538
    },
    {
      "epoch": 0.4308,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.007207997639973958,
      "learning_rate": 0.0001,
      "loss": 3.9439,
      "loss/crossentropy": 2.0833089351654053,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19176144897937775,
      "step": 21540
    },
    {
      "epoch": 0.43084,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.006376139322916667,
      "learning_rate": 0.0001,
      "loss": 4.0987,
      "loss/crossentropy": 2.1066776514053345,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20604360103607178,
      "step": 21542
    },
    {
      "epoch": 0.43088,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.006248982747395834,
      "learning_rate": 0.0001,
      "loss": 3.9294,
      "loss/crossentropy": 2.125930666923523,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22426339238882065,
      "step": 21544
    },
    {
      "epoch": 0.43092,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0062978108723958336,
      "learning_rate": 0.0001,
      "loss": 3.8652,
      "loss/crossentropy": 1.743666172027588,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1681881994009018,
      "step": 21546
    },
    {
      "epoch": 0.43096,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.00712890625,
      "learning_rate": 0.0001,
      "loss": 3.8952,
      "loss/crossentropy": 1.882837951183319,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18062414973974228,
      "step": 21548
    },
    {
      "epoch": 0.431,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.009137980143229167,
      "learning_rate": 0.0001,
      "loss": 3.6019,
      "loss/crossentropy": 1.8259800672531128,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19174453616142273,
      "step": 21550
    },
    {
      "epoch": 0.43104,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.009199778238932291,
      "learning_rate": 0.0001,
      "loss": 4.0255,
      "loss/crossentropy": 2.3379745483398438,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22536730766296387,
      "step": 21552
    },
    {
      "epoch": 0.43108,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.010560862223307292,
      "learning_rate": 0.0001,
      "loss": 3.9447,
      "loss/crossentropy": 2.123952627182007,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21830175817012787,
      "step": 21554
    },
    {
      "epoch": 0.43112,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.006849924723307292,
      "learning_rate": 0.0001,
      "loss": 4.1353,
      "loss/crossentropy": 2.286398410797119,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20413320511579514,
      "step": 21556
    },
    {
      "epoch": 0.43116,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.006044260660807292,
      "learning_rate": 0.0001,
      "loss": 3.9117,
      "loss/crossentropy": 2.0898618698120117,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2105274796485901,
      "step": 21558
    },
    {
      "epoch": 0.4312,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0067789713541666664,
      "learning_rate": 0.0001,
      "loss": 4.1431,
      "loss/crossentropy": 1.940380573272705,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18282227218151093,
      "step": 21560
    },
    {
      "epoch": 0.43124,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.007657623291015625,
      "learning_rate": 0.0001,
      "loss": 3.9952,
      "loss/crossentropy": 1.6752060055732727,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20306075364351273,
      "step": 21562
    },
    {
      "epoch": 0.43128,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.011091105143229167,
      "learning_rate": 0.0001,
      "loss": 4.3745,
      "loss/crossentropy": 2.234758734703064,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19758003950119019,
      "step": 21564
    },
    {
      "epoch": 0.43132,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.00819091796875,
      "learning_rate": 0.0001,
      "loss": 4.2822,
      "loss/crossentropy": 2.126620352268219,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20688238739967346,
      "step": 21566
    },
    {
      "epoch": 0.43136,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.008084869384765625,
      "learning_rate": 0.0001,
      "loss": 4.1531,
      "loss/crossentropy": 1.9397594332695007,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18199825286865234,
      "step": 21568
    },
    {
      "epoch": 0.4314,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0080322265625,
      "learning_rate": 0.0001,
      "loss": 4.1644,
      "loss/crossentropy": 2.1395343542099,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2074909508228302,
      "step": 21570
    },
    {
      "epoch": 0.43144,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.010827382405598959,
      "learning_rate": 0.0001,
      "loss": 3.7397,
      "loss/crossentropy": 1.9207513332366943,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1881982833147049,
      "step": 21572
    },
    {
      "epoch": 0.43148,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.01856053670247396,
      "learning_rate": 0.0001,
      "loss": 4.1079,
      "loss/crossentropy": 2.0409696102142334,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21291138231754303,
      "step": 21574
    },
    {
      "epoch": 0.43152,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.02153498331705729,
      "learning_rate": 0.0001,
      "loss": 4.0419,
      "loss/crossentropy": 2.0753902792930603,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20471004396677017,
      "step": 21576
    },
    {
      "epoch": 0.43156,
      "grad_norm": 1.875,
      "grad_norm_var": 0.022321573893229165,
      "learning_rate": 0.0001,
      "loss": 4.1617,
      "loss/crossentropy": 2.412545084953308,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21189319342374802,
      "step": 21578
    },
    {
      "epoch": 0.4316,
      "grad_norm": 1.875,
      "grad_norm_var": 0.019850413004557293,
      "learning_rate": 0.0001,
      "loss": 3.9468,
      "loss/crossentropy": 1.9271405339241028,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18710750341415405,
      "step": 21580
    },
    {
      "epoch": 0.43164,
      "grad_norm": 2.0,
      "grad_norm_var": 0.019321441650390625,
      "learning_rate": 0.0001,
      "loss": 4.1681,
      "loss/crossentropy": 2.000905692577362,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17429713159799576,
      "step": 21582
    },
    {
      "epoch": 0.43168,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.019334920247395835,
      "learning_rate": 0.0001,
      "loss": 3.8559,
      "loss/crossentropy": 1.9592137932777405,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.189944788813591,
      "step": 21584
    },
    {
      "epoch": 0.43172,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0188385009765625,
      "learning_rate": 0.0001,
      "loss": 4.1447,
      "loss/crossentropy": 2.283234119415283,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18164613842964172,
      "step": 21586
    },
    {
      "epoch": 0.43176,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0160064697265625,
      "learning_rate": 0.0001,
      "loss": 4.0019,
      "loss/crossentropy": 2.1789008378982544,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2087143510580063,
      "step": 21588
    },
    {
      "epoch": 0.4318,
      "grad_norm": 1.875,
      "grad_norm_var": 0.009000396728515625,
      "learning_rate": 0.0001,
      "loss": 4.0039,
      "loss/crossentropy": 2.192038893699646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19432441890239716,
      "step": 21590
    },
    {
      "epoch": 0.43184,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.005130767822265625,
      "learning_rate": 0.0001,
      "loss": 4.0142,
      "loss/crossentropy": 1.8618733286857605,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18207471072673798,
      "step": 21592
    },
    {
      "epoch": 0.43188,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.006186676025390625,
      "learning_rate": 0.0001,
      "loss": 4.3056,
      "loss/crossentropy": 2.423651933670044,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21819397807121277,
      "step": 21594
    },
    {
      "epoch": 0.43192,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.005765533447265625,
      "learning_rate": 0.0001,
      "loss": 4.0506,
      "loss/crossentropy": 1.7313976287841797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16011521220207214,
      "step": 21596
    },
    {
      "epoch": 0.43196,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0051043192545572914,
      "learning_rate": 0.0001,
      "loss": 4.0915,
      "loss/crossentropy": 2.089974284172058,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20138490200042725,
      "step": 21598
    },
    {
      "epoch": 0.432,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.018895467122395832,
      "learning_rate": 0.0001,
      "loss": 4.3367,
      "loss/crossentropy": 1.9697930812835693,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22859974950551987,
      "step": 21600
    },
    {
      "epoch": 0.43204,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.019832356770833334,
      "learning_rate": 0.0001,
      "loss": 4.0616,
      "loss/crossentropy": 1.7605210542678833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17675022035837173,
      "step": 21602
    },
    {
      "epoch": 0.43208,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.019755045572916668,
      "learning_rate": 0.0001,
      "loss": 3.9467,
      "loss/crossentropy": 1.9884806275367737,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19690094888210297,
      "step": 21604
    },
    {
      "epoch": 0.43212,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.018521881103515624,
      "learning_rate": 0.0001,
      "loss": 3.7705,
      "loss/crossentropy": 1.987808346748352,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20262162387371063,
      "step": 21606
    },
    {
      "epoch": 0.43216,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.026105753580729165,
      "learning_rate": 0.0001,
      "loss": 3.683,
      "loss/crossentropy": 2.220315098762512,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1899535208940506,
      "step": 21608
    },
    {
      "epoch": 0.4322,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.024347941080729168,
      "learning_rate": 0.0001,
      "loss": 4.0442,
      "loss/crossentropy": 2.245596408843994,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21112707257270813,
      "step": 21610
    },
    {
      "epoch": 0.43224,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0252838134765625,
      "learning_rate": 0.0001,
      "loss": 4.2442,
      "loss/crossentropy": 2.3593950271606445,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22111155092716217,
      "step": 21612
    },
    {
      "epoch": 0.43228,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.025721995035807292,
      "learning_rate": 0.0001,
      "loss": 3.7807,
      "loss/crossentropy": 1.7467412948608398,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16833512485027313,
      "step": 21614
    },
    {
      "epoch": 0.43232,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.011328125,
      "learning_rate": 0.0001,
      "loss": 3.815,
      "loss/crossentropy": 2.160380005836487,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22670412063598633,
      "step": 21616
    },
    {
      "epoch": 0.43236,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.012007395426432291,
      "learning_rate": 0.0001,
      "loss": 3.9665,
      "loss/crossentropy": 2.1650896072387695,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19781000167131424,
      "step": 21618
    },
    {
      "epoch": 0.4324,
      "grad_norm": 1.875,
      "grad_norm_var": 0.011987050374348959,
      "learning_rate": 0.0001,
      "loss": 3.8044,
      "loss/crossentropy": 2.028198480606079,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1905156373977661,
      "step": 21620
    },
    {
      "epoch": 0.43244,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.014241282145182292,
      "learning_rate": 0.0001,
      "loss": 3.9015,
      "loss/crossentropy": 2.30538147687912,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21474938094615936,
      "step": 21622
    },
    {
      "epoch": 0.43248,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.009450022379557292,
      "learning_rate": 0.0001,
      "loss": 3.9623,
      "loss/crossentropy": 1.8054092526435852,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1670922413468361,
      "step": 21624
    },
    {
      "epoch": 0.43252,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.011578114827473958,
      "learning_rate": 0.0001,
      "loss": 4.2168,
      "loss/crossentropy": 1.9821457862854004,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19581317901611328,
      "step": 21626
    },
    {
      "epoch": 0.43256,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.011197916666666667,
      "learning_rate": 0.0001,
      "loss": 3.9882,
      "loss/crossentropy": 2.2404789328575134,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19181591272354126,
      "step": 21628
    },
    {
      "epoch": 0.4326,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.013557942708333333,
      "learning_rate": 0.0001,
      "loss": 4.0683,
      "loss/crossentropy": 2.0229339003562927,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2553243041038513,
      "step": 21630
    },
    {
      "epoch": 0.43264,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.011122385660807291,
      "learning_rate": 0.0001,
      "loss": 4.2193,
      "loss/crossentropy": 2.013065457344055,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24048484861850739,
      "step": 21632
    },
    {
      "epoch": 0.43268,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.009173329671223958,
      "learning_rate": 0.0001,
      "loss": 3.9129,
      "loss/crossentropy": 1.9411469101905823,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19341549277305603,
      "step": 21634
    },
    {
      "epoch": 0.43272,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.008558909098307291,
      "learning_rate": 0.0001,
      "loss": 4.0984,
      "loss/crossentropy": 2.1277971267700195,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19953395426273346,
      "step": 21636
    },
    {
      "epoch": 0.43276,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0071795145670572914,
      "learning_rate": 0.0001,
      "loss": 4.2077,
      "loss/crossentropy": 2.1296733617782593,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21117133647203445,
      "step": 21638
    },
    {
      "epoch": 0.4328,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.009376780192057291,
      "learning_rate": 0.0001,
      "loss": 3.8594,
      "loss/crossentropy": 1.6910184025764465,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17567047476768494,
      "step": 21640
    },
    {
      "epoch": 0.43284,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.009845987955729166,
      "learning_rate": 0.0001,
      "loss": 3.9411,
      "loss/crossentropy": 1.9889350533485413,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18229827284812927,
      "step": 21642
    },
    {
      "epoch": 0.43288,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.008931223551432292,
      "learning_rate": 0.0001,
      "loss": 3.8053,
      "loss/crossentropy": 1.885926902294159,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18047264218330383,
      "step": 21644
    },
    {
      "epoch": 0.43292,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.006471506754557292,
      "learning_rate": 0.0001,
      "loss": 3.9383,
      "loss/crossentropy": 1.9706445336341858,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18135320395231247,
      "step": 21646
    },
    {
      "epoch": 0.43296,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.007127888997395833,
      "learning_rate": 0.0001,
      "loss": 3.8972,
      "loss/crossentropy": 1.7751979231834412,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1693567857146263,
      "step": 21648
    },
    {
      "epoch": 0.433,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.006884511311848958,
      "learning_rate": 0.0001,
      "loss": 4.0094,
      "loss/crossentropy": 2.025897264480591,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18943700194358826,
      "step": 21650
    },
    {
      "epoch": 0.43304,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.007627105712890625,
      "learning_rate": 0.0001,
      "loss": 3.8503,
      "loss/crossentropy": 1.7019765973091125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1738589182496071,
      "step": 21652
    },
    {
      "epoch": 0.43308,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.010225168863932292,
      "learning_rate": 0.0001,
      "loss": 4.1418,
      "loss/crossentropy": 1.8927075266838074,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1616983264684677,
      "step": 21654
    },
    {
      "epoch": 0.43312,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.012336222330729167,
      "learning_rate": 0.0001,
      "loss": 4.3299,
      "loss/crossentropy": 2.2224843502044678,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21432733535766602,
      "step": 21656
    },
    {
      "epoch": 0.43316,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.01502685546875,
      "learning_rate": 0.0001,
      "loss": 4.1572,
      "loss/crossentropy": 2.1968295574188232,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20150140672922134,
      "step": 21658
    },
    {
      "epoch": 0.4332,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.015057118733723958,
      "learning_rate": 0.0001,
      "loss": 4.0046,
      "loss/crossentropy": 2.1984314918518066,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21520624309778214,
      "step": 21660
    },
    {
      "epoch": 0.43324,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.01613133748372396,
      "learning_rate": 0.0001,
      "loss": 4.0638,
      "loss/crossentropy": 2.183030843734741,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1994212418794632,
      "step": 21662
    },
    {
      "epoch": 0.43328,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.016353352864583334,
      "learning_rate": 0.0001,
      "loss": 4.108,
      "loss/crossentropy": 2.0716358423233032,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18969116359949112,
      "step": 21664
    },
    {
      "epoch": 0.43332,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.016043853759765626,
      "learning_rate": 0.0001,
      "loss": 4.0988,
      "loss/crossentropy": 1.8427326679229736,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19842079281806946,
      "step": 21666
    },
    {
      "epoch": 0.43336,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.014058430989583334,
      "learning_rate": 0.0001,
      "loss": 4.0744,
      "loss/crossentropy": 2.3021219968795776,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21595575660467148,
      "step": 21668
    },
    {
      "epoch": 0.4334,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.01593195597330729,
      "learning_rate": 0.0001,
      "loss": 3.8216,
      "loss/crossentropy": 2.2828067541122437,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21141938865184784,
      "step": 21670
    },
    {
      "epoch": 0.43344,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.014955393473307292,
      "learning_rate": 0.0001,
      "loss": 4.0502,
      "loss/crossentropy": 2.167429566383362,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22402016073465347,
      "step": 21672
    },
    {
      "epoch": 0.43348,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.010632069905598958,
      "learning_rate": 0.0001,
      "loss": 3.9943,
      "loss/crossentropy": 2.02812397480011,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2061637043952942,
      "step": 21674
    },
    {
      "epoch": 0.43352,
      "grad_norm": 1.7109375,
      "grad_norm_var": 0.013724517822265626,
      "learning_rate": 0.0001,
      "loss": 3.6925,
      "loss/crossentropy": 1.96470308303833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.168222576379776,
      "step": 21676
    },
    {
      "epoch": 0.43356,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.013163248697916666,
      "learning_rate": 0.0001,
      "loss": 4.1567,
      "loss/crossentropy": 2.640696406364441,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22549600154161453,
      "step": 21678
    },
    {
      "epoch": 0.4336,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.010058339436848958,
      "learning_rate": 0.0001,
      "loss": 4.0939,
      "loss/crossentropy": 2.140254855155945,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1973073035478592,
      "step": 21680
    },
    {
      "epoch": 0.43364,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0110595703125,
      "learning_rate": 0.0001,
      "loss": 4.1028,
      "loss/crossentropy": 2.3025078773498535,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19433188438415527,
      "step": 21682
    },
    {
      "epoch": 0.43368,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.012452952067057292,
      "learning_rate": 0.0001,
      "loss": 4.344,
      "loss/crossentropy": 2.1558977365493774,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20599104464054108,
      "step": 21684
    },
    {
      "epoch": 0.43372,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.014780426025390625,
      "learning_rate": 0.0001,
      "loss": 3.7918,
      "loss/crossentropy": 1.8071665167808533,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18443845212459564,
      "step": 21686
    },
    {
      "epoch": 0.43376,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.018477121988932293,
      "learning_rate": 0.0001,
      "loss": 4.0928,
      "loss/crossentropy": 2.41480815410614,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22098398208618164,
      "step": 21688
    },
    {
      "epoch": 0.4338,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.019551595052083332,
      "learning_rate": 0.0001,
      "loss": 3.9046,
      "loss/crossentropy": 2.0506592988967896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17161494493484497,
      "step": 21690
    },
    {
      "epoch": 0.43384,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.018187459309895834,
      "learning_rate": 0.0001,
      "loss": 3.9579,
      "loss/crossentropy": 2.170476496219635,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19826646149158478,
      "step": 21692
    },
    {
      "epoch": 0.43388,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.019665273030598958,
      "learning_rate": 0.0001,
      "loss": 3.9356,
      "loss/crossentropy": 1.6374267935752869,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1581001877784729,
      "step": 21694
    },
    {
      "epoch": 0.43392,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.018944295247395833,
      "learning_rate": 0.0001,
      "loss": 3.9333,
      "loss/crossentropy": 1.9059698581695557,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18373095244169235,
      "step": 21696
    },
    {
      "epoch": 0.43396,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.01762669881184896,
      "learning_rate": 0.0001,
      "loss": 4.2217,
      "loss/crossentropy": 2.238908290863037,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2044297158718109,
      "step": 21698
    },
    {
      "epoch": 0.434,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.01877415974934896,
      "learning_rate": 0.0001,
      "loss": 3.6318,
      "loss/crossentropy": 1.7991122007369995,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18948530405759811,
      "step": 21700
    },
    {
      "epoch": 0.43404,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.01668268839518229,
      "learning_rate": 0.0001,
      "loss": 3.754,
      "loss/crossentropy": 1.773992896080017,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16768701374530792,
      "step": 21702
    },
    {
      "epoch": 0.43408,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.013051096598307292,
      "learning_rate": 0.0001,
      "loss": 4.325,
      "loss/crossentropy": 2.021883547306061,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20071732252836227,
      "step": 21704
    },
    {
      "epoch": 0.43412,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0167144775390625,
      "learning_rate": 0.0001,
      "loss": 4.3162,
      "loss/crossentropy": 2.2198556661605835,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20410272479057312,
      "step": 21706
    },
    {
      "epoch": 0.43416,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.013986968994140625,
      "learning_rate": 0.0001,
      "loss": 4.0144,
      "loss/crossentropy": 2.239061653614044,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20614881813526154,
      "step": 21708
    },
    {
      "epoch": 0.4342,
      "grad_norm": 2.25,
      "grad_norm_var": 0.01849543253580729,
      "learning_rate": 0.0001,
      "loss": 4.0531,
      "loss/crossentropy": 1.720937430858612,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19312319159507751,
      "step": 21710
    },
    {
      "epoch": 0.43424,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.01881103515625,
      "learning_rate": 0.0001,
      "loss": 4.2064,
      "loss/crossentropy": 2.577925205230713,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20975439995527267,
      "step": 21712
    },
    {
      "epoch": 0.43428,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.019634755452473958,
      "learning_rate": 0.0001,
      "loss": 4.2833,
      "loss/crossentropy": 2.3273919820785522,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.203931525349617,
      "step": 21714
    },
    {
      "epoch": 0.43432,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.017316691080729165,
      "learning_rate": 0.0001,
      "loss": 3.7922,
      "loss/crossentropy": 1.6773480772972107,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1546436920762062,
      "step": 21716
    },
    {
      "epoch": 0.43436,
      "grad_norm": 1.875,
      "grad_norm_var": 0.015986887613932292,
      "learning_rate": 0.0001,
      "loss": 4.0342,
      "loss/crossentropy": 2.4566088914871216,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2312512993812561,
      "step": 21718
    },
    {
      "epoch": 0.4344,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.015998331705729167,
      "learning_rate": 0.0001,
      "loss": 3.5272,
      "loss/crossentropy": 1.8777849674224854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18316396325826645,
      "step": 21720
    },
    {
      "epoch": 0.43444,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.011744944254557292,
      "learning_rate": 0.0001,
      "loss": 4.0373,
      "loss/crossentropy": 2.131688177585602,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1945742592215538,
      "step": 21722
    },
    {
      "epoch": 0.43448,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0122222900390625,
      "learning_rate": 0.0001,
      "loss": 3.9186,
      "loss/crossentropy": 1.9808812141418457,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2051435261964798,
      "step": 21724
    },
    {
      "epoch": 0.43452,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.009037017822265625,
      "learning_rate": 0.0001,
      "loss": 3.9247,
      "loss/crossentropy": 2.1240362524986267,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22029808908700943,
      "step": 21726
    },
    {
      "epoch": 0.43456,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0087554931640625,
      "learning_rate": 0.0001,
      "loss": 4.1359,
      "loss/crossentropy": 2.2637228965759277,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24003028869628906,
      "step": 21728
    },
    {
      "epoch": 0.4346,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.0075266520182291664,
      "learning_rate": 0.0001,
      "loss": 3.7561,
      "loss/crossentropy": 2.0770394802093506,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.185364231467247,
      "step": 21730
    },
    {
      "epoch": 0.43464,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0071197509765625,
      "learning_rate": 0.0001,
      "loss": 4.1621,
      "loss/crossentropy": 2.1420366764068604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21707475185394287,
      "step": 21732
    },
    {
      "epoch": 0.43468,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.006940714518229167,
      "learning_rate": 0.0001,
      "loss": 4.0836,
      "loss/crossentropy": 1.8017431497573853,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1667446494102478,
      "step": 21734
    },
    {
      "epoch": 0.43472,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.007490793863932292,
      "learning_rate": 0.0001,
      "loss": 4.0098,
      "loss/crossentropy": 2.215123176574707,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24778465926647186,
      "step": 21736
    },
    {
      "epoch": 0.43476,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.009708658854166666,
      "learning_rate": 0.0001,
      "loss": 3.7596,
      "loss/crossentropy": 1.7275863289833069,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18431701511144638,
      "step": 21738
    },
    {
      "epoch": 0.4348,
      "grad_norm": 2.0,
      "grad_norm_var": 0.009643300374348959,
      "learning_rate": 0.0001,
      "loss": 4.1243,
      "loss/crossentropy": 2.0645371675491333,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19622839987277985,
      "step": 21740
    },
    {
      "epoch": 0.43484,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.006563059488932292,
      "learning_rate": 0.0001,
      "loss": 3.9028,
      "loss/crossentropy": 1.8695591688156128,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18317092955112457,
      "step": 21742
    },
    {
      "epoch": 0.43488,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.010239410400390624,
      "learning_rate": 0.0001,
      "loss": 4.1963,
      "loss/crossentropy": 2.090248942375183,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20329977571964264,
      "step": 21744
    },
    {
      "epoch": 0.43492,
      "grad_norm": 1.875,
      "grad_norm_var": 0.011307779947916667,
      "learning_rate": 0.0001,
      "loss": 3.6347,
      "loss/crossentropy": 1.943733274936676,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20422999560832977,
      "step": 21746
    },
    {
      "epoch": 0.43496,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.012483723958333333,
      "learning_rate": 0.0001,
      "loss": 4.0101,
      "loss/crossentropy": 2.2485480308532715,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20576569437980652,
      "step": 21748
    },
    {
      "epoch": 0.435,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.012569173177083334,
      "learning_rate": 0.0001,
      "loss": 4.1053,
      "loss/crossentropy": 2.184234142303467,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19477711617946625,
      "step": 21750
    },
    {
      "epoch": 0.43504,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.011311848958333334,
      "learning_rate": 0.0001,
      "loss": 4.1796,
      "loss/crossentropy": 2.1742812991142273,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19085688889026642,
      "step": 21752
    },
    {
      "epoch": 0.43508,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.010985310872395833,
      "learning_rate": 0.0001,
      "loss": 4.1924,
      "loss/crossentropy": 2.44822895526886,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2128421813249588,
      "step": 21754
    },
    {
      "epoch": 0.43512,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.01138916015625,
      "learning_rate": 0.0001,
      "loss": 4.1137,
      "loss/crossentropy": 1.8242409229278564,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19045840948820114,
      "step": 21756
    },
    {
      "epoch": 0.43516,
      "grad_norm": 2.0,
      "grad_norm_var": 0.012474568684895833,
      "learning_rate": 0.0001,
      "loss": 4.0617,
      "loss/crossentropy": 2.1707061529159546,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2152707874774933,
      "step": 21758
    },
    {
      "epoch": 0.4352,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.010672760009765626,
      "learning_rate": 0.0001,
      "loss": 4.045,
      "loss/crossentropy": 2.0790776014328003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19175638258457184,
      "step": 21760
    },
    {
      "epoch": 0.43524,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.0095947265625,
      "learning_rate": 0.0001,
      "loss": 3.8814,
      "loss/crossentropy": 1.9847996830940247,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18776872754096985,
      "step": 21762
    },
    {
      "epoch": 0.43528,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0096435546875,
      "learning_rate": 0.0001,
      "loss": 4.0754,
      "loss/crossentropy": 2.122888207435608,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20312964171171188,
      "step": 21764
    },
    {
      "epoch": 0.43532,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0095855712890625,
      "learning_rate": 0.0001,
      "loss": 4.035,
      "loss/crossentropy": 2.1922764778137207,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2057497799396515,
      "step": 21766
    },
    {
      "epoch": 0.43536,
      "grad_norm": 1.7421875,
      "grad_norm_var": 0.013321685791015624,
      "learning_rate": 0.0001,
      "loss": 3.8096,
      "loss/crossentropy": 1.392991840839386,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1425289735198021,
      "step": 21768
    },
    {
      "epoch": 0.4354,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.012115224202473959,
      "learning_rate": 0.0001,
      "loss": 4.1758,
      "loss/crossentropy": 1.9627132415771484,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2007637917995453,
      "step": 21770
    },
    {
      "epoch": 0.43544,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.011374664306640626,
      "learning_rate": 0.0001,
      "loss": 4.0094,
      "loss/crossentropy": 2.04905503988266,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18171894550323486,
      "step": 21772
    },
    {
      "epoch": 0.43548,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.0086090087890625,
      "learning_rate": 0.0001,
      "loss": 3.7465,
      "loss/crossentropy": 2.09994637966156,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1924375221133232,
      "step": 21774
    },
    {
      "epoch": 0.43552,
      "grad_norm": 1.875,
      "grad_norm_var": 0.008565012613932292,
      "learning_rate": 0.0001,
      "loss": 4.2903,
      "loss/crossentropy": 2.381268620491028,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22655388712882996,
      "step": 21776
    },
    {
      "epoch": 0.43556,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.008161417643229167,
      "learning_rate": 0.0001,
      "loss": 4.151,
      "loss/crossentropy": 2.189077138900757,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19720469415187836,
      "step": 21778
    },
    {
      "epoch": 0.4356,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.005381011962890625,
      "learning_rate": 0.0001,
      "loss": 3.7407,
      "loss/crossentropy": 1.691848337650299,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1633242815732956,
      "step": 21780
    },
    {
      "epoch": 0.43564,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.007054646809895833,
      "learning_rate": 0.0001,
      "loss": 4.0392,
      "loss/crossentropy": 2.170955538749695,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20698139071464539,
      "step": 21782
    },
    {
      "epoch": 0.43568,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.00499267578125,
      "learning_rate": 0.0001,
      "loss": 3.9417,
      "loss/crossentropy": 2.0300655364990234,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1849035769701004,
      "step": 21784
    },
    {
      "epoch": 0.43572,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.004443359375,
      "learning_rate": 0.0001,
      "loss": 3.9352,
      "loss/crossentropy": 1.998863697052002,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1871778443455696,
      "step": 21786
    },
    {
      "epoch": 0.43576,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.004257965087890625,
      "learning_rate": 0.0001,
      "loss": 3.9628,
      "loss/crossentropy": 1.7537881731987,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.178309828042984,
      "step": 21788
    },
    {
      "epoch": 0.4358,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.003916168212890625,
      "learning_rate": 0.0001,
      "loss": 3.932,
      "loss/crossentropy": 1.9158823490142822,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18895266950130463,
      "step": 21790
    },
    {
      "epoch": 0.43584,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0035845438639322915,
      "learning_rate": 0.0001,
      "loss": 4.0875,
      "loss/crossentropy": 2.177749514579773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20137280225753784,
      "step": 21792
    },
    {
      "epoch": 0.43588,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.004788970947265625,
      "learning_rate": 0.0001,
      "loss": 4.1789,
      "loss/crossentropy": 2.132981538772583,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20100867003202438,
      "step": 21794
    },
    {
      "epoch": 0.43592,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.004564412434895833,
      "learning_rate": 0.0001,
      "loss": 3.9301,
      "loss/crossentropy": 1.926209807395935,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19080301374197006,
      "step": 21796
    },
    {
      "epoch": 0.43596,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.0040891011555989586,
      "learning_rate": 0.0001,
      "loss": 4.1137,
      "loss/crossentropy": 2.1029208302497864,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2002335861325264,
      "step": 21798
    },
    {
      "epoch": 0.436,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0036333719889322918,
      "learning_rate": 0.0001,
      "loss": 4.1378,
      "loss/crossentropy": 2.0101813673973083,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21301408112049103,
      "step": 21800
    },
    {
      "epoch": 0.43604,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.005250803629557292,
      "learning_rate": 0.0001,
      "loss": 3.8085,
      "loss/crossentropy": 2.2096749544143677,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20425067842006683,
      "step": 21802
    },
    {
      "epoch": 0.43608,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.005236562093098958,
      "learning_rate": 0.0001,
      "loss": 3.9631,
      "loss/crossentropy": 1.7085874676704407,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17989665269851685,
      "step": 21804
    },
    {
      "epoch": 0.43612,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.006086222330729167,
      "learning_rate": 0.0001,
      "loss": 4.2035,
      "loss/crossentropy": 2.082730233669281,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19857076555490494,
      "step": 21806
    },
    {
      "epoch": 0.43616,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0062164306640625,
      "learning_rate": 0.0001,
      "loss": 4.0634,
      "loss/crossentropy": 2.0145097970962524,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1956789717078209,
      "step": 21808
    },
    {
      "epoch": 0.4362,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0051666259765625,
      "learning_rate": 0.0001,
      "loss": 4.31,
      "loss/crossentropy": 2.0774065256118774,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20803070068359375,
      "step": 21810
    },
    {
      "epoch": 0.43624,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.004839833577473958,
      "learning_rate": 0.0001,
      "loss": 3.9442,
      "loss/crossentropy": 2.045994222164154,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19481085240840912,
      "step": 21812
    },
    {
      "epoch": 0.43628,
      "grad_norm": 2.0,
      "grad_norm_var": 0.005570220947265625,
      "learning_rate": 0.0001,
      "loss": 4.119,
      "loss/crossentropy": 1.9371901154518127,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20281973481178284,
      "step": 21814
    },
    {
      "epoch": 0.43632,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.00648193359375,
      "learning_rate": 0.0001,
      "loss": 3.8379,
      "loss/crossentropy": 2.1048339009284973,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1919357031583786,
      "step": 21816
    },
    {
      "epoch": 0.43636,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.005116526285807292,
      "learning_rate": 0.0001,
      "loss": 4.0696,
      "loss/crossentropy": 2.162278175354004,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19293204694986343,
      "step": 21818
    },
    {
      "epoch": 0.4364,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.006034342447916666,
      "learning_rate": 0.0001,
      "loss": 4.0294,
      "loss/crossentropy": 1.9012067317962646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19044938683509827,
      "step": 21820
    },
    {
      "epoch": 0.43644,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.004583485921223958,
      "learning_rate": 0.0001,
      "loss": 3.8538,
      "loss/crossentropy": 1.617782175540924,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1641436219215393,
      "step": 21822
    },
    {
      "epoch": 0.43648,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.006048329671223958,
      "learning_rate": 0.0001,
      "loss": 4.2229,
      "loss/crossentropy": 2.136145055294037,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20208875089883804,
      "step": 21824
    },
    {
      "epoch": 0.43652,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.006475575764973958,
      "learning_rate": 0.0001,
      "loss": 4.0446,
      "loss/crossentropy": 2.1240354776382446,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2052341103553772,
      "step": 21826
    },
    {
      "epoch": 0.43656,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.007020823160807292,
      "learning_rate": 0.0001,
      "loss": 4.0295,
      "loss/crossentropy": 1.8771272897720337,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17846016585826874,
      "step": 21828
    },
    {
      "epoch": 0.4366,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.005985260009765625,
      "learning_rate": 0.0001,
      "loss": 3.9318,
      "loss/crossentropy": 1.8061035871505737,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1792590245604515,
      "step": 21830
    },
    {
      "epoch": 0.43664,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0055084228515625,
      "learning_rate": 0.0001,
      "loss": 4.0204,
      "loss/crossentropy": 2.0216365456581116,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18977319449186325,
      "step": 21832
    },
    {
      "epoch": 0.43668,
      "grad_norm": 1.75,
      "grad_norm_var": 0.008056386311848959,
      "learning_rate": 0.0001,
      "loss": 3.7667,
      "loss/crossentropy": 1.81255304813385,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1740209013223648,
      "step": 21834
    },
    {
      "epoch": 0.43672,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.0087310791015625,
      "learning_rate": 0.0001,
      "loss": 4.0626,
      "loss/crossentropy": 2.2982383966445923,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20780136436223984,
      "step": 21836
    },
    {
      "epoch": 0.43676,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.010037994384765625,
      "learning_rate": 0.0001,
      "loss": 3.9891,
      "loss/crossentropy": 2.0461641550064087,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21413680911064148,
      "step": 21838
    },
    {
      "epoch": 0.4368,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.008341217041015625,
      "learning_rate": 0.0001,
      "loss": 4.1413,
      "loss/crossentropy": 2.1272863149642944,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22310787439346313,
      "step": 21840
    },
    {
      "epoch": 0.43684,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.008133951822916667,
      "learning_rate": 0.0001,
      "loss": 3.9467,
      "loss/crossentropy": 1.9026488661766052,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18013561517000198,
      "step": 21842
    },
    {
      "epoch": 0.43688,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.007635243733723958,
      "learning_rate": 0.0001,
      "loss": 4.2739,
      "loss/crossentropy": 2.1403396129608154,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20329253375530243,
      "step": 21844
    },
    {
      "epoch": 0.43692,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0075103759765625,
      "learning_rate": 0.0001,
      "loss": 4.1922,
      "loss/crossentropy": 2.041237533092499,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20238220691680908,
      "step": 21846
    },
    {
      "epoch": 0.43696,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.008011881510416667,
      "learning_rate": 0.0001,
      "loss": 4.2067,
      "loss/crossentropy": 1.860413134098053,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19062548130750656,
      "step": 21848
    },
    {
      "epoch": 0.437,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.007721964518229167,
      "learning_rate": 0.0001,
      "loss": 3.962,
      "loss/crossentropy": 2.1432868242263794,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1947617009282112,
      "step": 21850
    },
    {
      "epoch": 0.43704,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.007834625244140626,
      "learning_rate": 0.0001,
      "loss": 3.667,
      "loss/crossentropy": 1.5292840003967285,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1465652659535408,
      "step": 21852
    },
    {
      "epoch": 0.43708,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.00523681640625,
      "learning_rate": 0.0001,
      "loss": 3.7604,
      "loss/crossentropy": 1.8826366066932678,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19239384680986404,
      "step": 21854
    },
    {
      "epoch": 0.43712,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.005086008707682292,
      "learning_rate": 0.0001,
      "loss": 4.0817,
      "loss/crossentropy": 1.7915772199630737,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18841900676488876,
      "step": 21856
    },
    {
      "epoch": 0.43716,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0048868815104166664,
      "learning_rate": 0.0001,
      "loss": 3.7986,
      "loss/crossentropy": 1.9425716400146484,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19191939383745193,
      "step": 21858
    },
    {
      "epoch": 0.4372,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.006131744384765625,
      "learning_rate": 0.0001,
      "loss": 3.8735,
      "loss/crossentropy": 2.073697865009308,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19094493240118027,
      "step": 21860
    },
    {
      "epoch": 0.43724,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.046575673421223956,
      "learning_rate": 0.0001,
      "loss": 4.0953,
      "loss/crossentropy": 2.023917257785797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2218250036239624,
      "step": 21862
    },
    {
      "epoch": 0.43728,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.04965413411458333,
      "learning_rate": 0.0001,
      "loss": 4.1522,
      "loss/crossentropy": 1.9710112810134888,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21274644136428833,
      "step": 21864
    },
    {
      "epoch": 0.43732,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.04736328125,
      "learning_rate": 0.0001,
      "loss": 4.0579,
      "loss/crossentropy": 1.9758725762367249,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1855839416384697,
      "step": 21866
    },
    {
      "epoch": 0.43736,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.04616597493489583,
      "learning_rate": 0.0001,
      "loss": 4.0144,
      "loss/crossentropy": 2.0672810077667236,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20418523997068405,
      "step": 21868
    },
    {
      "epoch": 0.4374,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.046000162760416664,
      "learning_rate": 0.0001,
      "loss": 3.6892,
      "loss/crossentropy": 1.8937869668006897,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18653447926044464,
      "step": 21870
    },
    {
      "epoch": 0.43744,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.049006144205729164,
      "learning_rate": 0.0001,
      "loss": 3.8814,
      "loss/crossentropy": 1.6613503694534302,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16891123354434967,
      "step": 21872
    },
    {
      "epoch": 0.43748,
      "grad_norm": 2.0,
      "grad_norm_var": 0.04822489420572917,
      "learning_rate": 0.0001,
      "loss": 4.2159,
      "loss/crossentropy": 2.007514178752899,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2078787311911583,
      "step": 21874
    },
    {
      "epoch": 0.43752,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.043603515625,
      "learning_rate": 0.0001,
      "loss": 3.831,
      "loss/crossentropy": 1.913162350654602,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20729011297225952,
      "step": 21876
    },
    {
      "epoch": 0.43756,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.012849680582682292,
      "learning_rate": 0.0001,
      "loss": 3.9612,
      "loss/crossentropy": 2.1534116864204407,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20183125138282776,
      "step": 21878
    },
    {
      "epoch": 0.4376,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.012841542561848959,
      "learning_rate": 0.0001,
      "loss": 3.8101,
      "loss/crossentropy": 1.9248425960540771,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19415219128131866,
      "step": 21880
    },
    {
      "epoch": 0.43764,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.012790679931640625,
      "learning_rate": 0.0001,
      "loss": 3.7379,
      "loss/crossentropy": 2.095816135406494,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21181143820285797,
      "step": 21882
    },
    {
      "epoch": 0.43768,
      "grad_norm": 2.25,
      "grad_norm_var": 0.015405019124348959,
      "learning_rate": 0.0001,
      "loss": 4.6839,
      "loss/crossentropy": 2.0167580246925354,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21327386051416397,
      "step": 21884
    },
    {
      "epoch": 0.43772,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.019038899739583334,
      "learning_rate": 0.0001,
      "loss": 4.3355,
      "loss/crossentropy": 2.2425005435943604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20246200263500214,
      "step": 21886
    },
    {
      "epoch": 0.43776,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.017232259114583332,
      "learning_rate": 0.0001,
      "loss": 4.3023,
      "loss/crossentropy": 1.9543325901031494,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18824229389429092,
      "step": 21888
    },
    {
      "epoch": 0.4378,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.017365519205729166,
      "learning_rate": 0.0001,
      "loss": 4.0802,
      "loss/crossentropy": 2.2031290531158447,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20949723571538925,
      "step": 21890
    },
    {
      "epoch": 0.43784,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.01901219685872396,
      "learning_rate": 0.0001,
      "loss": 3.728,
      "loss/crossentropy": 1.97504061460495,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19441132247447968,
      "step": 21892
    },
    {
      "epoch": 0.43788,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.018888346354166665,
      "learning_rate": 0.0001,
      "loss": 3.8922,
      "loss/crossentropy": 1.8496057391166687,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18915118277072906,
      "step": 21894
    },
    {
      "epoch": 0.43792,
      "grad_norm": 2.0,
      "grad_norm_var": 0.015116373697916666,
      "learning_rate": 0.0001,
      "loss": 4.0474,
      "loss/crossentropy": 1.9244802594184875,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20679102092981339,
      "step": 21896
    },
    {
      "epoch": 0.43796,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.016902414957682292,
      "learning_rate": 0.0001,
      "loss": 3.9855,
      "loss/crossentropy": 2.1038661003112793,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2014932855963707,
      "step": 21898
    },
    {
      "epoch": 0.438,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.011286417643229166,
      "learning_rate": 0.0001,
      "loss": 4.0416,
      "loss/crossentropy": 2.202482581138611,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20515280216932297,
      "step": 21900
    },
    {
      "epoch": 0.43804,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.008353678385416667,
      "learning_rate": 0.0001,
      "loss": 4.0346,
      "loss/crossentropy": 2.1215643286705017,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18430311977863312,
      "step": 21902
    },
    {
      "epoch": 0.43808,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0086822509765625,
      "learning_rate": 0.0001,
      "loss": 4.1243,
      "loss/crossentropy": 1.585806667804718,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1615592986345291,
      "step": 21904
    },
    {
      "epoch": 0.43812,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.008831532796223958,
      "learning_rate": 0.0001,
      "loss": 4.0288,
      "loss/crossentropy": 2.0006843209266663,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21003154665231705,
      "step": 21906
    },
    {
      "epoch": 0.43816,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.008571116129557292,
      "learning_rate": 0.0001,
      "loss": 3.8523,
      "loss/crossentropy": 1.9077118635177612,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1881355568766594,
      "step": 21908
    },
    {
      "epoch": 0.4382,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0080810546875,
      "learning_rate": 0.0001,
      "loss": 3.7186,
      "loss/crossentropy": 1.752367615699768,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17893216013908386,
      "step": 21910
    },
    {
      "epoch": 0.43824,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0078765869140625,
      "learning_rate": 0.0001,
      "loss": 4.171,
      "loss/crossentropy": 2.1325889825820923,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21129589527845383,
      "step": 21912
    },
    {
      "epoch": 0.43828,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.006949615478515625,
      "learning_rate": 0.0001,
      "loss": 4.3083,
      "loss/crossentropy": 2.630311131477356,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.224892295897007,
      "step": 21914
    },
    {
      "epoch": 0.43832,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.007549794514973959,
      "learning_rate": 0.0001,
      "loss": 4.1696,
      "loss/crossentropy": 2.002493679523468,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19148597121238708,
      "step": 21916
    },
    {
      "epoch": 0.43836,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.006571451822916667,
      "learning_rate": 0.0001,
      "loss": 4.1209,
      "loss/crossentropy": 1.968815267086029,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18172769993543625,
      "step": 21918
    },
    {
      "epoch": 0.4384,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.005402628580729167,
      "learning_rate": 0.0001,
      "loss": 4.1595,
      "loss/crossentropy": 2.2519482374191284,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22669588029384613,
      "step": 21920
    },
    {
      "epoch": 0.43844,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0059315999348958336,
      "learning_rate": 0.0001,
      "loss": 3.9382,
      "loss/crossentropy": 1.9276898503303528,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18415841460227966,
      "step": 21922
    },
    {
      "epoch": 0.43848,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.004941558837890625,
      "learning_rate": 0.0001,
      "loss": 4.1358,
      "loss/crossentropy": 2.1866860389709473,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2306595891714096,
      "step": 21924
    },
    {
      "epoch": 0.43852,
      "grad_norm": 1.875,
      "grad_norm_var": 0.004378000895182292,
      "learning_rate": 0.0001,
      "loss": 3.9259,
      "loss/crossentropy": 1.8013821840286255,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17407341301441193,
      "step": 21926
    },
    {
      "epoch": 0.43856,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.005110422770182292,
      "learning_rate": 0.0001,
      "loss": 3.9644,
      "loss/crossentropy": 2.116545557975769,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19539258629083633,
      "step": 21928
    },
    {
      "epoch": 0.4386,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.006009928385416667,
      "learning_rate": 0.0001,
      "loss": 3.8005,
      "loss/crossentropy": 2.039593815803528,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1793600618839264,
      "step": 21930
    },
    {
      "epoch": 0.43864,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.008385976155598959,
      "learning_rate": 0.0001,
      "loss": 3.9727,
      "loss/crossentropy": 2.0602275133132935,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17620772868394852,
      "step": 21932
    },
    {
      "epoch": 0.43868,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.007024892171223958,
      "learning_rate": 0.0001,
      "loss": 3.7552,
      "loss/crossentropy": 1.9316160082817078,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2026260495185852,
      "step": 21934
    },
    {
      "epoch": 0.43872,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.007004547119140625,
      "learning_rate": 0.0001,
      "loss": 4.0254,
      "loss/crossentropy": 2.152816414833069,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18020687997341156,
      "step": 21936
    },
    {
      "epoch": 0.43876,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.0067626953125,
      "learning_rate": 0.0001,
      "loss": 3.8211,
      "loss/crossentropy": 1.8688368201255798,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17652328312397003,
      "step": 21938
    },
    {
      "epoch": 0.4388,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.006628163655598958,
      "learning_rate": 0.0001,
      "loss": 4.162,
      "loss/crossentropy": 1.8889214992523193,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18263526260852814,
      "step": 21940
    },
    {
      "epoch": 0.43884,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.007869466145833334,
      "learning_rate": 0.0001,
      "loss": 3.8644,
      "loss/crossentropy": 1.9699294567108154,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2170155718922615,
      "step": 21942
    },
    {
      "epoch": 0.43888,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.009642537434895833,
      "learning_rate": 0.0001,
      "loss": 4.1007,
      "loss/crossentropy": 1.7749728560447693,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18438361585140228,
      "step": 21944
    },
    {
      "epoch": 0.43892,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.008910115559895833,
      "learning_rate": 0.0001,
      "loss": 4.0266,
      "loss/crossentropy": 2.1438393592834473,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22740208357572556,
      "step": 21946
    },
    {
      "epoch": 0.43896,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.005663045247395833,
      "learning_rate": 0.0001,
      "loss": 4.0983,
      "loss/crossentropy": 2.1517677307128906,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20089010894298553,
      "step": 21948
    },
    {
      "epoch": 0.439,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.006931304931640625,
      "learning_rate": 0.0001,
      "loss": 4.0286,
      "loss/crossentropy": 2.283053159713745,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19754114747047424,
      "step": 21950
    },
    {
      "epoch": 0.43904,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.008438873291015624,
      "learning_rate": 0.0001,
      "loss": 4.3633,
      "loss/crossentropy": 2.121833860874176,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20339152216911316,
      "step": 21952
    },
    {
      "epoch": 0.43908,
      "grad_norm": 2.0,
      "grad_norm_var": 0.007926177978515626,
      "learning_rate": 0.0001,
      "loss": 4.1399,
      "loss/crossentropy": 2.206274390220642,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19716881215572357,
      "step": 21954
    },
    {
      "epoch": 0.43912,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.012749989827473959,
      "learning_rate": 0.0001,
      "loss": 4.185,
      "loss/crossentropy": 2.1501349210739136,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20559798926115036,
      "step": 21956
    },
    {
      "epoch": 0.43916,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.012088775634765625,
      "learning_rate": 0.0001,
      "loss": 3.9056,
      "loss/crossentropy": 2.0854570269584656,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22488616406917572,
      "step": 21958
    },
    {
      "epoch": 0.4392,
      "grad_norm": 3.265625,
      "grad_norm_var": 0.11881103515625,
      "learning_rate": 0.0001,
      "loss": 3.651,
      "loss/crossentropy": 1.840592384338379,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1934354454278946,
      "step": 21960
    },
    {
      "epoch": 0.43924,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.12266006469726562,
      "learning_rate": 0.0001,
      "loss": 4.013,
      "loss/crossentropy": 2.062163829803467,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19524814933538437,
      "step": 21962
    },
    {
      "epoch": 0.43928,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.1280413309733073,
      "learning_rate": 0.0001,
      "loss": 3.8329,
      "loss/crossentropy": 1.7060028910636902,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17902052402496338,
      "step": 21964
    },
    {
      "epoch": 0.43932,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.12742919921875,
      "learning_rate": 0.0001,
      "loss": 4.0638,
      "loss/crossentropy": 2.246550440788269,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2088882103562355,
      "step": 21966
    },
    {
      "epoch": 0.43936,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.12764053344726561,
      "learning_rate": 0.0001,
      "loss": 4.2095,
      "loss/crossentropy": 2.249360680580139,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20600482821464539,
      "step": 21968
    },
    {
      "epoch": 0.4394,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.13187840779622395,
      "learning_rate": 0.0001,
      "loss": 4.2291,
      "loss/crossentropy": 2.029528558254242,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18438928574323654,
      "step": 21970
    },
    {
      "epoch": 0.43944,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.1287249247233073,
      "learning_rate": 0.0001,
      "loss": 4.0865,
      "loss/crossentropy": 2.0363243222236633,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19402872025966644,
      "step": 21972
    },
    {
      "epoch": 0.43948,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.12544657389322916,
      "learning_rate": 0.0001,
      "loss": 4.2188,
      "loss/crossentropy": 2.1816266775131226,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20053647458553314,
      "step": 21974
    },
    {
      "epoch": 0.43952,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.01646703084309896,
      "learning_rate": 0.0001,
      "loss": 4.2621,
      "loss/crossentropy": 2.452013850212097,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2117229700088501,
      "step": 21976
    },
    {
      "epoch": 0.43956,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0123779296875,
      "learning_rate": 0.0001,
      "loss": 4.0617,
      "loss/crossentropy": 2.1460453271865845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2046569585800171,
      "step": 21978
    },
    {
      "epoch": 0.4396,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.009639485677083334,
      "learning_rate": 0.0001,
      "loss": 4.1805,
      "loss/crossentropy": 2.152291774749756,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19845052808523178,
      "step": 21980
    },
    {
      "epoch": 0.43964,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.007950592041015624,
      "learning_rate": 0.0001,
      "loss": 4.3282,
      "loss/crossentropy": 2.280704140663147,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20648686587810516,
      "step": 21982
    },
    {
      "epoch": 0.43968,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.009004465738932292,
      "learning_rate": 0.0001,
      "loss": 4.146,
      "loss/crossentropy": 2.2323286533355713,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1949191614985466,
      "step": 21984
    },
    {
      "epoch": 0.43972,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006135813395182292,
      "learning_rate": 0.0001,
      "loss": 3.8289,
      "loss/crossentropy": 2.105699062347412,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19488361477851868,
      "step": 21986
    },
    {
      "epoch": 0.43976,
      "grad_norm": 1.6875,
      "grad_norm_var": 0.009834543863932291,
      "learning_rate": 0.0001,
      "loss": 3.731,
      "loss/crossentropy": 1.7268863916397095,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16756988316774368,
      "step": 21988
    },
    {
      "epoch": 0.4398,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.009458160400390625,
      "learning_rate": 0.0001,
      "loss": 4.0178,
      "loss/crossentropy": 2.1496203541755676,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20237596333026886,
      "step": 21990
    },
    {
      "epoch": 0.43984,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.009350331624348958,
      "learning_rate": 0.0001,
      "loss": 4.1835,
      "loss/crossentropy": 2.196579158306122,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20089206099510193,
      "step": 21992
    },
    {
      "epoch": 0.43988,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.011958821614583334,
      "learning_rate": 0.0001,
      "loss": 4.0065,
      "loss/crossentropy": 2.1145724654197693,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21221736818552017,
      "step": 21994
    },
    {
      "epoch": 0.43992,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.012230428059895833,
      "learning_rate": 0.0001,
      "loss": 4.1096,
      "loss/crossentropy": 1.8034029603004456,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18355921655893326,
      "step": 21996
    },
    {
      "epoch": 0.43996,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.012262980143229166,
      "learning_rate": 0.0001,
      "loss": 4.1805,
      "loss/crossentropy": 1.8520742654800415,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18945766985416412,
      "step": 21998
    },
    {
      "epoch": 0.44,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.012962849934895833,
      "learning_rate": 0.0001,
      "loss": 4.4938,
      "loss/crossentropy": 2.0837312936782837,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22002845257520676,
      "step": 22000
    },
    {
      "epoch": 0.44004,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.011470540364583334,
      "learning_rate": 0.0001,
      "loss": 4.0409,
      "loss/crossentropy": 1.9202563166618347,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19138525426387787,
      "step": 22002
    },
    {
      "epoch": 0.44008,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.006506093343098958,
      "learning_rate": 0.0001,
      "loss": 3.9965,
      "loss/crossentropy": 2.0986216068267822,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18914636224508286,
      "step": 22004
    },
    {
      "epoch": 0.44012,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.008955637613932291,
      "learning_rate": 0.0001,
      "loss": 3.5102,
      "loss/crossentropy": 1.675586223602295,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17793113738298416,
      "step": 22006
    },
    {
      "epoch": 0.44016,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0088043212890625,
      "learning_rate": 0.0001,
      "loss": 4.0252,
      "loss/crossentropy": 1.7633379697799683,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17190124094486237,
      "step": 22008
    },
    {
      "epoch": 0.4402,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.008796183268229167,
      "learning_rate": 0.0001,
      "loss": 3.925,
      "loss/crossentropy": 2.0000420212745667,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18909277766942978,
      "step": 22010
    },
    {
      "epoch": 0.44024,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.009691365559895833,
      "learning_rate": 0.0001,
      "loss": 3.8831,
      "loss/crossentropy": 2.2526057958602905,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19274260103702545,
      "step": 22012
    },
    {
      "epoch": 0.44028,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.010261027018229167,
      "learning_rate": 0.0001,
      "loss": 3.9836,
      "loss/crossentropy": 2.1151334643363953,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1992139369249344,
      "step": 22014
    },
    {
      "epoch": 0.44032,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.008546702067057292,
      "learning_rate": 0.0001,
      "loss": 3.932,
      "loss/crossentropy": 2.188230037689209,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.205021433532238,
      "step": 22016
    },
    {
      "epoch": 0.44036,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.008329264322916667,
      "learning_rate": 0.0001,
      "loss": 3.9484,
      "loss/crossentropy": 2.015773594379425,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19697444140911102,
      "step": 22018
    },
    {
      "epoch": 0.4404,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.008882649739583333,
      "learning_rate": 0.0001,
      "loss": 3.7165,
      "loss/crossentropy": 1.724815011024475,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17810098081827164,
      "step": 22020
    },
    {
      "epoch": 0.44044,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.010091145833333334,
      "learning_rate": 0.0001,
      "loss": 4.3719,
      "loss/crossentropy": 2.471455454826355,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20882685482501984,
      "step": 22022
    },
    {
      "epoch": 0.44048,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.009212239583333334,
      "learning_rate": 0.0001,
      "loss": 4.2544,
      "loss/crossentropy": 2.2879083156585693,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20954442769289017,
      "step": 22024
    },
    {
      "epoch": 0.44052,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.008440907796223958,
      "learning_rate": 0.0001,
      "loss": 4.1305,
      "loss/crossentropy": 2.352377772331238,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21374861150979996,
      "step": 22026
    },
    {
      "epoch": 0.44056,
      "grad_norm": 1.875,
      "grad_norm_var": 0.010823313395182292,
      "learning_rate": 0.0001,
      "loss": 3.889,
      "loss/crossentropy": 1.5395016074180603,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17564409226179123,
      "step": 22028
    },
    {
      "epoch": 0.4406,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.009440104166666666,
      "learning_rate": 0.0001,
      "loss": 3.997,
      "loss/crossentropy": 2.079974055290222,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1886643022298813,
      "step": 22030
    },
    {
      "epoch": 0.44064,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.013110097249348958,
      "learning_rate": 0.0001,
      "loss": 4.3671,
      "loss/crossentropy": 2.0840702056884766,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21841587871313095,
      "step": 22032
    },
    {
      "epoch": 0.44068,
      "grad_norm": 2.125,
      "grad_norm_var": 0.014168294270833333,
      "learning_rate": 0.0001,
      "loss": 4.2372,
      "loss/crossentropy": 2.574951171875,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21359464526176453,
      "step": 22034
    },
    {
      "epoch": 0.44072,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.011519368489583333,
      "learning_rate": 0.0001,
      "loss": 4.0261,
      "loss/crossentropy": 1.8530864715576172,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19658754020929337,
      "step": 22036
    },
    {
      "epoch": 0.44076,
      "grad_norm": 1.7109375,
      "grad_norm_var": 0.018289947509765626,
      "learning_rate": 0.0001,
      "loss": 3.7077,
      "loss/crossentropy": 1.9564177989959717,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19047683477401733,
      "step": 22038
    },
    {
      "epoch": 0.4408,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.02148615519205729,
      "learning_rate": 0.0001,
      "loss": 4.2303,
      "loss/crossentropy": 2.2482646703720093,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21484995633363724,
      "step": 22040
    },
    {
      "epoch": 0.44084,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.04890925089518229,
      "learning_rate": 0.0001,
      "loss": 3.9753,
      "loss/crossentropy": 1.9064926505088806,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2042565643787384,
      "step": 22042
    },
    {
      "epoch": 0.44088,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.048164621988932295,
      "learning_rate": 0.0001,
      "loss": 3.8492,
      "loss/crossentropy": 1.9772136211395264,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18002107739448547,
      "step": 22044
    },
    {
      "epoch": 0.44092,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.047240193684895834,
      "learning_rate": 0.0001,
      "loss": 4.0417,
      "loss/crossentropy": 1.8454242944717407,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19727011024951935,
      "step": 22046
    },
    {
      "epoch": 0.44096,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.047584788004557295,
      "learning_rate": 0.0001,
      "loss": 3.8247,
      "loss/crossentropy": 1.815159022808075,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17201630771160126,
      "step": 22048
    },
    {
      "epoch": 0.441,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.05176976521809896,
      "learning_rate": 0.0001,
      "loss": 4.0909,
      "loss/crossentropy": 2.298068404197693,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21187184751033783,
      "step": 22050
    },
    {
      "epoch": 0.44104,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.04794108072916667,
      "learning_rate": 0.0001,
      "loss": 3.9931,
      "loss/crossentropy": 2.4030654430389404,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2233939692378044,
      "step": 22052
    },
    {
      "epoch": 0.44108,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.043981679280598956,
      "learning_rate": 0.0001,
      "loss": 3.924,
      "loss/crossentropy": 2.1413121223449707,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19442930072546005,
      "step": 22054
    },
    {
      "epoch": 0.44112,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0421875,
      "learning_rate": 0.0001,
      "loss": 4.0048,
      "loss/crossentropy": 2.029422342777252,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19413675367832184,
      "step": 22056
    },
    {
      "epoch": 0.44116,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.012010701497395833,
      "learning_rate": 0.0001,
      "loss": 4.3637,
      "loss/crossentropy": 2.3010586500167847,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2198517844080925,
      "step": 22058
    },
    {
      "epoch": 0.4412,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.012353261311848959,
      "learning_rate": 0.0001,
      "loss": 4.0769,
      "loss/crossentropy": 2.0910425782203674,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19917423278093338,
      "step": 22060
    },
    {
      "epoch": 0.44124,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.027766672770182292,
      "learning_rate": 0.0001,
      "loss": 3.7656,
      "loss/crossentropy": 1.948616862297058,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20419400930404663,
      "step": 22062
    },
    {
      "epoch": 0.44128,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.02814509073893229,
      "learning_rate": 0.0001,
      "loss": 4.1349,
      "loss/crossentropy": 2.0931553840637207,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19765160977840424,
      "step": 22064
    },
    {
      "epoch": 0.44132,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.0232421875,
      "learning_rate": 0.0001,
      "loss": 4.1685,
      "loss/crossentropy": 2.0905882120132446,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20447614789009094,
      "step": 22066
    },
    {
      "epoch": 0.44136,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.022861480712890625,
      "learning_rate": 0.0001,
      "loss": 4.1407,
      "loss/crossentropy": 2.365593433380127,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21229872107505798,
      "step": 22068
    },
    {
      "epoch": 0.4414,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.022093709309895834,
      "learning_rate": 0.0001,
      "loss": 4.2306,
      "loss/crossentropy": 2.286802887916565,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20844397693872452,
      "step": 22070
    },
    {
      "epoch": 0.44144,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.021507771809895833,
      "learning_rate": 0.0001,
      "loss": 4.2739,
      "loss/crossentropy": 2.1805503368377686,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20301498472690582,
      "step": 22072
    },
    {
      "epoch": 0.44148,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.020702870686848958,
      "learning_rate": 0.0001,
      "loss": 4.0185,
      "loss/crossentropy": 2.025053381919861,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1959461346268654,
      "step": 22074
    },
    {
      "epoch": 0.44152,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.02149658203125,
      "learning_rate": 0.0001,
      "loss": 4.1134,
      "loss/crossentropy": 2.1158708930015564,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19305311143398285,
      "step": 22076
    },
    {
      "epoch": 0.44156,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0063059488932291664,
      "learning_rate": 0.0001,
      "loss": 3.8749,
      "loss/crossentropy": 1.746639907360077,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18400797247886658,
      "step": 22078
    },
    {
      "epoch": 0.4416,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0041168212890625,
      "learning_rate": 0.0001,
      "loss": 4.3071,
      "loss/crossentropy": 2.3301517963409424,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19662386924028397,
      "step": 22080
    },
    {
      "epoch": 0.44164,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0036272684733072917,
      "learning_rate": 0.0001,
      "loss": 4.1136,
      "loss/crossentropy": 1.9772114753723145,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18933425843715668,
      "step": 22082
    },
    {
      "epoch": 0.44168,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0036069234212239582,
      "learning_rate": 0.0001,
      "loss": 4.0125,
      "loss/crossentropy": 1.9511193633079529,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19440795481204987,
      "step": 22084
    },
    {
      "epoch": 0.44172,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.005785115559895833,
      "learning_rate": 0.0001,
      "loss": 3.6898,
      "loss/crossentropy": 1.7232664823532104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16435310244560242,
      "step": 22086
    },
    {
      "epoch": 0.44176,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.005549875895182291,
      "learning_rate": 0.0001,
      "loss": 4.0631,
      "loss/crossentropy": 2.1858248710632324,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20494847744703293,
      "step": 22088
    },
    {
      "epoch": 0.4418,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.01435546875,
      "learning_rate": 0.0001,
      "loss": 3.9496,
      "loss/crossentropy": 2.4218143224716187,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21809446811676025,
      "step": 22090
    },
    {
      "epoch": 0.44184,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.014168039957682291,
      "learning_rate": 0.0001,
      "loss": 4.0209,
      "loss/crossentropy": 2.133104920387268,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1930398792028427,
      "step": 22092
    },
    {
      "epoch": 0.44188,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.013348134358723958,
      "learning_rate": 0.0001,
      "loss": 4.0959,
      "loss/crossentropy": 2.113961935043335,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2095598503947258,
      "step": 22094
    },
    {
      "epoch": 0.44192,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.0151763916015625,
      "learning_rate": 0.0001,
      "loss": 3.7976,
      "loss/crossentropy": 1.984375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18821635842323303,
      "step": 22096
    },
    {
      "epoch": 0.44196,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0153717041015625,
      "learning_rate": 0.0001,
      "loss": 3.8215,
      "loss/crossentropy": 1.8913013339042664,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1868671551346779,
      "step": 22098
    },
    {
      "epoch": 0.442,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.01610081990559896,
      "learning_rate": 0.0001,
      "loss": 3.8635,
      "loss/crossentropy": 1.9196743369102478,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18776098638772964,
      "step": 22100
    },
    {
      "epoch": 0.44204,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.013134511311848958,
      "learning_rate": 0.0001,
      "loss": 4.1463,
      "loss/crossentropy": 1.929152011871338,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17069555073976517,
      "step": 22102
    },
    {
      "epoch": 0.44208,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0133941650390625,
      "learning_rate": 0.0001,
      "loss": 4.0115,
      "loss/crossentropy": 2.0993363857269287,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19678544998168945,
      "step": 22104
    },
    {
      "epoch": 0.44212,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.004559071858723959,
      "learning_rate": 0.0001,
      "loss": 3.9802,
      "loss/crossentropy": 1.8447982668876648,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18432488292455673,
      "step": 22106
    },
    {
      "epoch": 0.44216,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0041544596354166664,
      "learning_rate": 0.0001,
      "loss": 4.2359,
      "loss/crossentropy": 2.04882550239563,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2129831686615944,
      "step": 22108
    },
    {
      "epoch": 0.4422,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.0047686258951822914,
      "learning_rate": 0.0001,
      "loss": 3.8738,
      "loss/crossentropy": 1.7422301769256592,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18266545236110687,
      "step": 22110
    },
    {
      "epoch": 0.44224,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.006664784749348959,
      "learning_rate": 0.0001,
      "loss": 4.2107,
      "loss/crossentropy": 2.1211363077163696,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22843683511018753,
      "step": 22112
    },
    {
      "epoch": 0.44228,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.008097076416015625,
      "learning_rate": 0.0001,
      "loss": 3.8527,
      "loss/crossentropy": 2.2478936910629272,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21724677085876465,
      "step": 22114
    },
    {
      "epoch": 0.44232,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.0153228759765625,
      "learning_rate": 0.0001,
      "loss": 4.3167,
      "loss/crossentropy": 2.3345694541931152,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20749730616807938,
      "step": 22116
    },
    {
      "epoch": 0.44236,
      "grad_norm": 1.7265625,
      "grad_norm_var": 0.020076243082682292,
      "learning_rate": 0.0001,
      "loss": 3.947,
      "loss/crossentropy": 2.0583457946777344,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18861139565706253,
      "step": 22118
    },
    {
      "epoch": 0.4424,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0246734619140625,
      "learning_rate": 0.0001,
      "loss": 3.7534,
      "loss/crossentropy": 2.06750625371933,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18891629576683044,
      "step": 22120
    },
    {
      "epoch": 0.44244,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.025272623697916666,
      "learning_rate": 0.0001,
      "loss": 4.2192,
      "loss/crossentropy": 1.8823494911193848,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20006988942623138,
      "step": 22122
    },
    {
      "epoch": 0.44248,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.026920572916666666,
      "learning_rate": 0.0001,
      "loss": 4.1285,
      "loss/crossentropy": 2.0374066829681396,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20347502827644348,
      "step": 22124
    },
    {
      "epoch": 0.44252,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.026195271809895834,
      "learning_rate": 0.0001,
      "loss": 3.8961,
      "loss/crossentropy": 1.4977151155471802,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15512564033269882,
      "step": 22126
    },
    {
      "epoch": 0.44256,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.024346669514973957,
      "learning_rate": 0.0001,
      "loss": 3.9462,
      "loss/crossentropy": 1.8102646470069885,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1884431689977646,
      "step": 22128
    },
    {
      "epoch": 0.4426,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.024079386393229166,
      "learning_rate": 0.0001,
      "loss": 4.2244,
      "loss/crossentropy": 2.1330573558807373,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19625350832939148,
      "step": 22130
    },
    {
      "epoch": 0.44264,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0181396484375,
      "learning_rate": 0.0001,
      "loss": 3.7348,
      "loss/crossentropy": 2.1866488456726074,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21527475118637085,
      "step": 22132
    },
    {
      "epoch": 0.44268,
      "grad_norm": 2.0,
      "grad_norm_var": 0.013985188802083333,
      "learning_rate": 0.0001,
      "loss": 4.0073,
      "loss/crossentropy": 2.0211732387542725,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20512627810239792,
      "step": 22134
    },
    {
      "epoch": 0.44272,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.010277303059895833,
      "learning_rate": 0.0001,
      "loss": 3.9338,
      "loss/crossentropy": 1.798271358013153,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1809278130531311,
      "step": 22136
    },
    {
      "epoch": 0.44276,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.008927408854166667,
      "learning_rate": 0.0001,
      "loss": 4.0153,
      "loss/crossentropy": 1.7850923538208008,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16607706993818283,
      "step": 22138
    },
    {
      "epoch": 0.4428,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.012361399332682292,
      "learning_rate": 0.0001,
      "loss": 4.0488,
      "loss/crossentropy": 2.1693010330200195,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18788384646177292,
      "step": 22140
    },
    {
      "epoch": 0.44284,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.013242340087890625,
      "learning_rate": 0.0001,
      "loss": 4.0582,
      "loss/crossentropy": 2.124464750289917,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21157250553369522,
      "step": 22142
    },
    {
      "epoch": 0.44288,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.018416086832682293,
      "learning_rate": 0.0001,
      "loss": 4.2624,
      "loss/crossentropy": 2.1086031794548035,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20727767795324326,
      "step": 22144
    },
    {
      "epoch": 0.44292,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.018000284830729168,
      "learning_rate": 0.0001,
      "loss": 3.8286,
      "loss/crossentropy": 1.8891723155975342,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20070044696331024,
      "step": 22146
    },
    {
      "epoch": 0.44296,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.015173085530598958,
      "learning_rate": 0.0001,
      "loss": 4.1181,
      "loss/crossentropy": 2.1405081748962402,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19909673184156418,
      "step": 22148
    },
    {
      "epoch": 0.443,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0155181884765625,
      "learning_rate": 0.0001,
      "loss": 4.0455,
      "loss/crossentropy": 2.2116858959198,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1970309540629387,
      "step": 22150
    },
    {
      "epoch": 0.44304,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.015183258056640624,
      "learning_rate": 0.0001,
      "loss": 3.8698,
      "loss/crossentropy": 1.9010364413261414,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18096549808979034,
      "step": 22152
    },
    {
      "epoch": 0.44308,
      "grad_norm": 1.875,
      "grad_norm_var": 0.015607706705729167,
      "learning_rate": 0.0001,
      "loss": 4.0256,
      "loss/crossentropy": 2.030808746814728,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20662673562765121,
      "step": 22154
    },
    {
      "epoch": 0.44312,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.018138631184895834,
      "learning_rate": 0.0001,
      "loss": 4.2633,
      "loss/crossentropy": 1.9940236806869507,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2860369235277176,
      "step": 22156
    },
    {
      "epoch": 0.44316,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.01758397420247396,
      "learning_rate": 0.0001,
      "loss": 4.1577,
      "loss/crossentropy": 2.3639464378356934,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20355366170406342,
      "step": 22158
    },
    {
      "epoch": 0.4432,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.013199615478515624,
      "learning_rate": 0.0001,
      "loss": 3.9197,
      "loss/crossentropy": 2.247707962989807,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2214345633983612,
      "step": 22160
    },
    {
      "epoch": 0.44324,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.014696248372395833,
      "learning_rate": 0.0001,
      "loss": 4.3906,
      "loss/crossentropy": 2.462310314178467,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2597040832042694,
      "step": 22162
    },
    {
      "epoch": 0.44328,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.014890289306640625,
      "learning_rate": 0.0001,
      "loss": 4.0932,
      "loss/crossentropy": 1.9838054180145264,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21189701557159424,
      "step": 22164
    },
    {
      "epoch": 0.44332,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.015169270833333333,
      "learning_rate": 0.0001,
      "loss": 3.988,
      "loss/crossentropy": 2.086707830429077,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23454946279525757,
      "step": 22166
    },
    {
      "epoch": 0.44336,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.014924875895182292,
      "learning_rate": 0.0001,
      "loss": 3.8447,
      "loss/crossentropy": 1.9489662051200867,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19790088385343552,
      "step": 22168
    },
    {
      "epoch": 0.4434,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.016532389322916667,
      "learning_rate": 0.0001,
      "loss": 3.7006,
      "loss/crossentropy": 1.9342001676559448,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17832304537296295,
      "step": 22170
    },
    {
      "epoch": 0.44344,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.011139933268229167,
      "learning_rate": 0.0001,
      "loss": 4.353,
      "loss/crossentropy": 2.248290777206421,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21428222954273224,
      "step": 22172
    },
    {
      "epoch": 0.44348,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.009865061442057291,
      "learning_rate": 0.0001,
      "loss": 4.1017,
      "loss/crossentropy": 1.8474311232566833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1795397698879242,
      "step": 22174
    },
    {
      "epoch": 0.44352,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009187571207682292,
      "learning_rate": 0.0001,
      "loss": 4.1804,
      "loss/crossentropy": 1.9687572121620178,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2062787041068077,
      "step": 22176
    },
    {
      "epoch": 0.44356,
      "grad_norm": 1.7421875,
      "grad_norm_var": 0.011625162760416667,
      "learning_rate": 0.0001,
      "loss": 3.8598,
      "loss/crossentropy": 2.4766552448272705,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2046542540192604,
      "step": 22178
    },
    {
      "epoch": 0.4436,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.011618804931640626,
      "learning_rate": 0.0001,
      "loss": 4.1097,
      "loss/crossentropy": 2.2163573503494263,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21187163889408112,
      "step": 22180
    },
    {
      "epoch": 0.44364,
      "grad_norm": 2.0,
      "grad_norm_var": 0.010813140869140625,
      "learning_rate": 0.0001,
      "loss": 4.1702,
      "loss/crossentropy": 2.180974006652832,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20932868123054504,
      "step": 22182
    },
    {
      "epoch": 0.44368,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.014957682291666666,
      "learning_rate": 0.0001,
      "loss": 4.2237,
      "loss/crossentropy": 1.97287255525589,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17987515777349472,
      "step": 22184
    },
    {
      "epoch": 0.44372,
      "grad_norm": 2.0,
      "grad_norm_var": 0.013679758707682291,
      "learning_rate": 0.0001,
      "loss": 4.0676,
      "loss/crossentropy": 1.9375402927398682,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1994931995868683,
      "step": 22186
    },
    {
      "epoch": 0.44376,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.0117340087890625,
      "learning_rate": 0.0001,
      "loss": 3.9908,
      "loss/crossentropy": 2.152758836746216,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19482193887233734,
      "step": 22188
    },
    {
      "epoch": 0.4438,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.012450917561848959,
      "learning_rate": 0.0001,
      "loss": 4.223,
      "loss/crossentropy": 2.29486083984375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24136198312044144,
      "step": 22190
    },
    {
      "epoch": 0.44384,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.015363566080729167,
      "learning_rate": 0.0001,
      "loss": 3.9714,
      "loss/crossentropy": 2.143070697784424,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2122703492641449,
      "step": 22192
    },
    {
      "epoch": 0.44388,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.011229451497395833,
      "learning_rate": 0.0001,
      "loss": 3.8691,
      "loss/crossentropy": 2.0414949655532837,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20614571869373322,
      "step": 22194
    },
    {
      "epoch": 0.44392,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.016951497395833334,
      "learning_rate": 0.0001,
      "loss": 4.244,
      "loss/crossentropy": 2.137809634208679,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22674410045146942,
      "step": 22196
    },
    {
      "epoch": 0.44396,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.01871312459309896,
      "learning_rate": 0.0001,
      "loss": 4.4088,
      "loss/crossentropy": 2.292213559150696,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18514274060726166,
      "step": 22198
    },
    {
      "epoch": 0.444,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.016839345296223957,
      "learning_rate": 0.0001,
      "loss": 4.0419,
      "loss/crossentropy": 2.3552767038345337,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21818525344133377,
      "step": 22200
    },
    {
      "epoch": 0.44404,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.01602961222330729,
      "learning_rate": 0.0001,
      "loss": 4.2912,
      "loss/crossentropy": 2.1532652378082275,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19477012753486633,
      "step": 22202
    },
    {
      "epoch": 0.44408,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.012938435872395833,
      "learning_rate": 0.0001,
      "loss": 3.9805,
      "loss/crossentropy": 1.921963095664978,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16913262009620667,
      "step": 22204
    },
    {
      "epoch": 0.44412,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.015063222249348958,
      "learning_rate": 0.0001,
      "loss": 3.8596,
      "loss/crossentropy": 1.9791225790977478,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18458476662635803,
      "step": 22206
    },
    {
      "epoch": 0.44416,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.014595540364583333,
      "learning_rate": 0.0001,
      "loss": 4.0334,
      "loss/crossentropy": 2.0733718276023865,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2146601676940918,
      "step": 22208
    },
    {
      "epoch": 0.4442,
      "grad_norm": 1.875,
      "grad_norm_var": 0.015830230712890626,
      "learning_rate": 0.0001,
      "loss": 3.7326,
      "loss/crossentropy": 1.5904502272605896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16351275146007538,
      "step": 22210
    },
    {
      "epoch": 0.44424,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.009563954671223958,
      "learning_rate": 0.0001,
      "loss": 4.0214,
      "loss/crossentropy": 2.0840989351272583,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21245559304952621,
      "step": 22212
    },
    {
      "epoch": 0.44428,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.008025868733723959,
      "learning_rate": 0.0001,
      "loss": 3.9995,
      "loss/crossentropy": 2.0513384342193604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19362515211105347,
      "step": 22214
    },
    {
      "epoch": 0.44432,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.010284169514973959,
      "learning_rate": 0.0001,
      "loss": 4.0989,
      "loss/crossentropy": 2.0588608980178833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1839647963643074,
      "step": 22216
    },
    {
      "epoch": 0.44436,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.011742146809895833,
      "learning_rate": 0.0001,
      "loss": 4.0668,
      "loss/crossentropy": 2.0614622831344604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19217892736196518,
      "step": 22218
    },
    {
      "epoch": 0.4444,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.013061269124348959,
      "learning_rate": 0.0001,
      "loss": 4.0823,
      "loss/crossentropy": 1.980837643146515,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1963345631957054,
      "step": 22220
    },
    {
      "epoch": 0.44444,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.014562733968098958,
      "learning_rate": 0.0001,
      "loss": 3.8748,
      "loss/crossentropy": 1.9071148037910461,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20523779094219208,
      "step": 22222
    },
    {
      "epoch": 0.44448,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.014020792643229167,
      "learning_rate": 0.0001,
      "loss": 4.2035,
      "loss/crossentropy": 2.0757131576538086,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1940036118030548,
      "step": 22224
    },
    {
      "epoch": 0.44452,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.015105946858723959,
      "learning_rate": 0.0001,
      "loss": 3.8287,
      "loss/crossentropy": 1.8984442353248596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18971017003059387,
      "step": 22226
    },
    {
      "epoch": 0.44456,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.014281972249348959,
      "learning_rate": 0.0001,
      "loss": 4.1821,
      "loss/crossentropy": 2.2503433227539062,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21632683277130127,
      "step": 22228
    },
    {
      "epoch": 0.4446,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.011771647135416667,
      "learning_rate": 0.0001,
      "loss": 4.2548,
      "loss/crossentropy": 2.264014482498169,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2061793953180313,
      "step": 22230
    },
    {
      "epoch": 0.44464,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.009279123942057292,
      "learning_rate": 0.0001,
      "loss": 4.1289,
      "loss/crossentropy": 2.2728497982025146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20043949782848358,
      "step": 22232
    },
    {
      "epoch": 0.44468,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0100250244140625,
      "learning_rate": 0.0001,
      "loss": 4.199,
      "loss/crossentropy": 2.074263334274292,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2182644009590149,
      "step": 22234
    },
    {
      "epoch": 0.44472,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.008780670166015626,
      "learning_rate": 0.0001,
      "loss": 3.9973,
      "loss/crossentropy": 2.1185666918754578,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1989990770816803,
      "step": 22236
    },
    {
      "epoch": 0.44476,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.007028961181640625,
      "learning_rate": 0.0001,
      "loss": 3.9992,
      "loss/crossentropy": 1.6533132791519165,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17203578352928162,
      "step": 22238
    },
    {
      "epoch": 0.4448,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.008231353759765626,
      "learning_rate": 0.0001,
      "loss": 3.948,
      "loss/crossentropy": 2.2107443809509277,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20056415349245071,
      "step": 22240
    },
    {
      "epoch": 0.44484,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.007938385009765625,
      "learning_rate": 0.0001,
      "loss": 3.793,
      "loss/crossentropy": 1.9579968452453613,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1899353489279747,
      "step": 22242
    },
    {
      "epoch": 0.44488,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.010945638020833334,
      "learning_rate": 0.0001,
      "loss": 4.012,
      "loss/crossentropy": 2.074744164943695,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2323092818260193,
      "step": 22244
    },
    {
      "epoch": 0.44492,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.012308756510416666,
      "learning_rate": 0.0001,
      "loss": 3.8848,
      "loss/crossentropy": 2.2182846069335938,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18586912751197815,
      "step": 22246
    },
    {
      "epoch": 0.44496,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.01761042277018229,
      "learning_rate": 0.0001,
      "loss": 4.3021,
      "loss/crossentropy": 2.2342774868011475,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20816949754953384,
      "step": 22248
    },
    {
      "epoch": 0.445,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.0157135009765625,
      "learning_rate": 0.0001,
      "loss": 3.8011,
      "loss/crossentropy": 2.103771924972534,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20282650738954544,
      "step": 22250
    },
    {
      "epoch": 0.44504,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.017350006103515624,
      "learning_rate": 0.0001,
      "loss": 4.1103,
      "loss/crossentropy": 2.010720193386078,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18631578236818314,
      "step": 22252
    },
    {
      "epoch": 0.44508,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0169097900390625,
      "learning_rate": 0.0001,
      "loss": 3.92,
      "loss/crossentropy": 1.8840174674987793,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19280597567558289,
      "step": 22254
    },
    {
      "epoch": 0.44512,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0151031494140625,
      "learning_rate": 0.0001,
      "loss": 4.0503,
      "loss/crossentropy": 2.404397130012512,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23199105262756348,
      "step": 22256
    },
    {
      "epoch": 0.44516,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.018021392822265624,
      "learning_rate": 0.0001,
      "loss": 3.7078,
      "loss/crossentropy": 2.11636883020401,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20003244280815125,
      "step": 22258
    },
    {
      "epoch": 0.4452,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.018195597330729167,
      "learning_rate": 0.0001,
      "loss": 4.1396,
      "loss/crossentropy": 1.8522619009017944,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16992035508155823,
      "step": 22260
    },
    {
      "epoch": 0.44524,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.01738459269205729,
      "learning_rate": 0.0001,
      "loss": 3.923,
      "loss/crossentropy": 2.243737578392029,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2175471931695938,
      "step": 22262
    },
    {
      "epoch": 0.44528,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.01102294921875,
      "learning_rate": 0.0001,
      "loss": 3.9973,
      "loss/crossentropy": 2.231359362602234,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19454128295183182,
      "step": 22264
    },
    {
      "epoch": 0.44532,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0114898681640625,
      "learning_rate": 0.0001,
      "loss": 4.0373,
      "loss/crossentropy": 2.08256071805954,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2158442661166191,
      "step": 22266
    },
    {
      "epoch": 0.44536,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.010520172119140626,
      "learning_rate": 0.0001,
      "loss": 3.9418,
      "loss/crossentropy": 2.089757025241852,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2029227539896965,
      "step": 22268
    },
    {
      "epoch": 0.4454,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.011017862955729167,
      "learning_rate": 0.0001,
      "loss": 4.1463,
      "loss/crossentropy": 2.0893616676330566,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1958554983139038,
      "step": 22270
    },
    {
      "epoch": 0.44544,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0145751953125,
      "learning_rate": 0.0001,
      "loss": 4.0382,
      "loss/crossentropy": 1.9625884890556335,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20560869574546814,
      "step": 22272
    },
    {
      "epoch": 0.44548,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.012360636393229167,
      "learning_rate": 0.0001,
      "loss": 4.2015,
      "loss/crossentropy": 2.4028927087783813,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21448686718940735,
      "step": 22274
    },
    {
      "epoch": 0.44552,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.010890452067057292,
      "learning_rate": 0.0001,
      "loss": 4.1805,
      "loss/crossentropy": 2.16468608379364,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1987794190645218,
      "step": 22276
    },
    {
      "epoch": 0.44556,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.010326131184895834,
      "learning_rate": 0.0001,
      "loss": 3.9558,
      "loss/crossentropy": 1.992479383945465,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19877834618091583,
      "step": 22278
    },
    {
      "epoch": 0.4456,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.008473459879557292,
      "learning_rate": 0.0001,
      "loss": 3.912,
      "loss/crossentropy": 2.0368640422821045,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20028480142354965,
      "step": 22280
    },
    {
      "epoch": 0.44564,
      "grad_norm": 2.0,
      "grad_norm_var": 0.017753092447916667,
      "learning_rate": 0.0001,
      "loss": 4.4026,
      "loss/crossentropy": 2.252933144569397,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19834356755018234,
      "step": 22282
    },
    {
      "epoch": 0.44568,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.01628392537434896,
      "learning_rate": 0.0001,
      "loss": 3.9787,
      "loss/crossentropy": 1.8675458431243896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1786389946937561,
      "step": 22284
    },
    {
      "epoch": 0.44572,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.01455078125,
      "learning_rate": 0.0001,
      "loss": 4.254,
      "loss/crossentropy": 2.0105971097946167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18405906856060028,
      "step": 22286
    },
    {
      "epoch": 0.44576,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.014598592122395834,
      "learning_rate": 0.0001,
      "loss": 3.8389,
      "loss/crossentropy": 1.8162254095077515,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18923819810152054,
      "step": 22288
    },
    {
      "epoch": 0.4458,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.01656061808268229,
      "learning_rate": 0.0001,
      "loss": 3.8374,
      "loss/crossentropy": 1.7614133954048157,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19005076587200165,
      "step": 22290
    },
    {
      "epoch": 0.44584,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.1847551981608073,
      "learning_rate": 0.0001,
      "loss": 4.135,
      "loss/crossentropy": 2.135009288787842,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20304158329963684,
      "step": 22292
    },
    {
      "epoch": 0.44588,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.18526611328125,
      "learning_rate": 0.0001,
      "loss": 3.9944,
      "loss/crossentropy": 1.897695004940033,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19377534836530685,
      "step": 22294
    },
    {
      "epoch": 0.44592,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.182763671875,
      "learning_rate": 0.0001,
      "loss": 4.1663,
      "loss/crossentropy": 2.0558266043663025,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20582793653011322,
      "step": 22296
    },
    {
      "epoch": 0.44596,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.18102188110351564,
      "learning_rate": 0.0001,
      "loss": 3.9538,
      "loss/crossentropy": 2.329292058944702,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20754558593034744,
      "step": 22298
    },
    {
      "epoch": 0.446,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.18606338500976563,
      "learning_rate": 0.0001,
      "loss": 3.8853,
      "loss/crossentropy": 2.2229275703430176,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20154273509979248,
      "step": 22300
    },
    {
      "epoch": 0.44604,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.18578465779622397,
      "learning_rate": 0.0001,
      "loss": 3.992,
      "loss/crossentropy": 1.9050685167312622,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17552578449249268,
      "step": 22302
    },
    {
      "epoch": 0.44608,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.18623021443684895,
      "learning_rate": 0.0001,
      "loss": 4.2241,
      "loss/crossentropy": 2.139374792575836,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19087333232164383,
      "step": 22304
    },
    {
      "epoch": 0.44612,
      "grad_norm": 2.0,
      "grad_norm_var": 0.18642578125,
      "learning_rate": 0.0001,
      "loss": 4.1013,
      "loss/crossentropy": 1.8093907237052917,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1805548369884491,
      "step": 22306
    },
    {
      "epoch": 0.44616,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.030443318684895835,
      "learning_rate": 0.0001,
      "loss": 3.7124,
      "loss/crossentropy": 1.8334048390388489,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16753460466861725,
      "step": 22308
    },
    {
      "epoch": 0.4462,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.030301920572916665,
      "learning_rate": 0.0001,
      "loss": 4.1018,
      "loss/crossentropy": 2.1181896924972534,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17999915778636932,
      "step": 22310
    },
    {
      "epoch": 0.44624,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.02830785115559896,
      "learning_rate": 0.0001,
      "loss": 3.9938,
      "loss/crossentropy": 2.3377938270568848,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20874116569757462,
      "step": 22312
    },
    {
      "epoch": 0.44628,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.028173828125,
      "learning_rate": 0.0001,
      "loss": 4.3027,
      "loss/crossentropy": 2.2557464838027954,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2159905582666397,
      "step": 22314
    },
    {
      "epoch": 0.44632,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.02478612263997396,
      "learning_rate": 0.0001,
      "loss": 4.1486,
      "loss/crossentropy": 1.690861165523529,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1791190207004547,
      "step": 22316
    },
    {
      "epoch": 0.44636,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.02339452107747396,
      "learning_rate": 0.0001,
      "loss": 3.9995,
      "loss/crossentropy": 2.282811760902405,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2278990000486374,
      "step": 22318
    },
    {
      "epoch": 0.4464,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.023021443684895834,
      "learning_rate": 0.0001,
      "loss": 4.0971,
      "loss/crossentropy": 2.087849497795105,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19879884272813797,
      "step": 22320
    },
    {
      "epoch": 0.44644,
      "grad_norm": 2.0,
      "grad_norm_var": 0.024181874593098958,
      "learning_rate": 0.0001,
      "loss": 4.2878,
      "loss/crossentropy": 1.7355778217315674,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1880098134279251,
      "step": 22322
    },
    {
      "epoch": 0.44648,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.006556955973307291,
      "learning_rate": 0.0001,
      "loss": 3.935,
      "loss/crossentropy": 1.9930670857429504,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.195980504155159,
      "step": 22324
    },
    {
      "epoch": 0.44652,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.006990305582682292,
      "learning_rate": 0.0001,
      "loss": 4.058,
      "loss/crossentropy": 2.2104332447052,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21003258973360062,
      "step": 22326
    },
    {
      "epoch": 0.44656,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.009208170572916667,
      "learning_rate": 0.0001,
      "loss": 3.876,
      "loss/crossentropy": 1.812508225440979,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18001863360404968,
      "step": 22328
    },
    {
      "epoch": 0.4466,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.009992472330729167,
      "learning_rate": 0.0001,
      "loss": 4.2515,
      "loss/crossentropy": 2.0790328979492188,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21385060250759125,
      "step": 22330
    },
    {
      "epoch": 0.44664,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.010206858317057291,
      "learning_rate": 0.0001,
      "loss": 3.9865,
      "loss/crossentropy": 1.778535783290863,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18019723147153854,
      "step": 22332
    },
    {
      "epoch": 0.44668,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.011714680989583334,
      "learning_rate": 0.0001,
      "loss": 4.1055,
      "loss/crossentropy": 2.1208608746528625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2075667828321457,
      "step": 22334
    },
    {
      "epoch": 0.44672,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.013523101806640625,
      "learning_rate": 0.0001,
      "loss": 3.8638,
      "loss/crossentropy": 1.9015109539031982,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18012915551662445,
      "step": 22336
    },
    {
      "epoch": 0.44676,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.01219482421875,
      "learning_rate": 0.0001,
      "loss": 3.9193,
      "loss/crossentropy": 1.844041883945465,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18436504900455475,
      "step": 22338
    },
    {
      "epoch": 0.4468,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.012889607747395834,
      "learning_rate": 0.0001,
      "loss": 3.8859,
      "loss/crossentropy": 1.730314016342163,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16672005504369736,
      "step": 22340
    },
    {
      "epoch": 0.44684,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.06477762858072916,
      "learning_rate": 0.0001,
      "loss": 3.9309,
      "loss/crossentropy": 2.385637402534485,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20245376229286194,
      "step": 22342
    },
    {
      "epoch": 0.44688,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.06330337524414062,
      "learning_rate": 0.0001,
      "loss": 4.0168,
      "loss/crossentropy": 2.033573269844055,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19226811826229095,
      "step": 22344
    },
    {
      "epoch": 0.44692,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.06444498697916666,
      "learning_rate": 0.0001,
      "loss": 3.9613,
      "loss/crossentropy": 2.018698275089264,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19148993492126465,
      "step": 22346
    },
    {
      "epoch": 0.44696,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.06483739217122396,
      "learning_rate": 0.0001,
      "loss": 4.0311,
      "loss/crossentropy": 1.7709164023399353,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18092583864927292,
      "step": 22348
    },
    {
      "epoch": 0.447,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.06496480305989584,
      "learning_rate": 0.0001,
      "loss": 4.1857,
      "loss/crossentropy": 1.9011916518211365,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18185202777385712,
      "step": 22350
    },
    {
      "epoch": 0.44704,
      "grad_norm": 1.875,
      "grad_norm_var": 0.0629547119140625,
      "learning_rate": 0.0001,
      "loss": 4.0509,
      "loss/crossentropy": 1.9298787117004395,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18417780846357346,
      "step": 22352
    },
    {
      "epoch": 0.44708,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0615875244140625,
      "learning_rate": 0.0001,
      "loss": 4.0872,
      "loss/crossentropy": 2.3658339977264404,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2167125567793846,
      "step": 22354
    },
    {
      "epoch": 0.44712,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.06006571451822917,
      "learning_rate": 0.0001,
      "loss": 4.092,
      "loss/crossentropy": 2.150742769241333,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21436245739459991,
      "step": 22356
    },
    {
      "epoch": 0.44716,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.007879384358723958,
      "learning_rate": 0.0001,
      "loss": 4.099,
      "loss/crossentropy": 2.197741746902466,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19714494049549103,
      "step": 22358
    },
    {
      "epoch": 0.4472,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.008286285400390624,
      "learning_rate": 0.0001,
      "loss": 4.0584,
      "loss/crossentropy": 2.11471688747406,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19025350362062454,
      "step": 22360
    },
    {
      "epoch": 0.44724,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0075724283854166664,
      "learning_rate": 0.0001,
      "loss": 3.9815,
      "loss/crossentropy": 2.2634165287017822,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2038334459066391,
      "step": 22362
    },
    {
      "epoch": 0.44728,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.007181549072265625,
      "learning_rate": 0.0001,
      "loss": 4.1587,
      "loss/crossentropy": 2.3305805921554565,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2206403762102127,
      "step": 22364
    },
    {
      "epoch": 0.44732,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0032867431640625,
      "learning_rate": 0.0001,
      "loss": 3.9196,
      "loss/crossentropy": 1.8015141487121582,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1901993378996849,
      "step": 22366
    },
    {
      "epoch": 0.44736,
      "grad_norm": 2.40625,
      "grad_norm_var": 0.017256673177083334,
      "learning_rate": 0.0001,
      "loss": 4.3285,
      "loss/crossentropy": 2.325510263442993,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2076701521873474,
      "step": 22368
    },
    {
      "epoch": 0.4474,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.0187652587890625,
      "learning_rate": 0.0001,
      "loss": 4.0401,
      "loss/crossentropy": 1.9470626711845398,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1854722648859024,
      "step": 22370
    },
    {
      "epoch": 0.44744,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.020442708333333334,
      "learning_rate": 0.0001,
      "loss": 3.9825,
      "loss/crossentropy": 1.7524075508117676,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17917660623788834,
      "step": 22372
    },
    {
      "epoch": 0.44748,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.018480428059895835,
      "learning_rate": 0.0001,
      "loss": 3.5927,
      "loss/crossentropy": 2.0228232741355896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19089559465646744,
      "step": 22374
    },
    {
      "epoch": 0.44752,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.019846343994140626,
      "learning_rate": 0.0001,
      "loss": 4.0478,
      "loss/crossentropy": 2.289384961128235,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21693934500217438,
      "step": 22376
    },
    {
      "epoch": 0.44756,
      "grad_norm": 2.125,
      "grad_norm_var": 0.021996053059895833,
      "learning_rate": 0.0001,
      "loss": 3.6972,
      "loss/crossentropy": 1.9497195482254028,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18409277498722076,
      "step": 22378
    },
    {
      "epoch": 0.4476,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.022004191080729166,
      "learning_rate": 0.0001,
      "loss": 4.2471,
      "loss/crossentropy": 2.184138000011444,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19307169318199158,
      "step": 22380
    },
    {
      "epoch": 0.44764,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.02208226521809896,
      "learning_rate": 0.0001,
      "loss": 4.0119,
      "loss/crossentropy": 2.0050132274627686,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19592129439115524,
      "step": 22382
    },
    {
      "epoch": 0.44768,
      "grad_norm": 1.75,
      "grad_norm_var": 0.012475331624348959,
      "learning_rate": 0.0001,
      "loss": 3.8577,
      "loss/crossentropy": 1.725416898727417,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17788676172494888,
      "step": 22384
    },
    {
      "epoch": 0.44772,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.015103912353515625,
      "learning_rate": 0.0001,
      "loss": 3.8859,
      "loss/crossentropy": 1.8454214930534363,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18375547230243683,
      "step": 22386
    },
    {
      "epoch": 0.44776,
      "grad_norm": 2.0,
      "grad_norm_var": 0.013166300455729167,
      "learning_rate": 0.0001,
      "loss": 4.1856,
      "loss/crossentropy": 2.090963661670685,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20424450188875198,
      "step": 22388
    },
    {
      "epoch": 0.4478,
      "grad_norm": 1.875,
      "grad_norm_var": 0.014296213785807291,
      "learning_rate": 0.0001,
      "loss": 3.5071,
      "loss/crossentropy": 1.7257133722305298,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18389710783958435,
      "step": 22390
    },
    {
      "epoch": 0.44784,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.012153879801432291,
      "learning_rate": 0.0001,
      "loss": 4.2002,
      "loss/crossentropy": 2.0955962538719177,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20340529829263687,
      "step": 22392
    },
    {
      "epoch": 0.44788,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.010103098551432292,
      "learning_rate": 0.0001,
      "loss": 3.965,
      "loss/crossentropy": 1.8837090730667114,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.188998244702816,
      "step": 22394
    },
    {
      "epoch": 0.44792,
      "grad_norm": 1.875,
      "grad_norm_var": 0.010092926025390626,
      "learning_rate": 0.0001,
      "loss": 4.0777,
      "loss/crossentropy": 2.351726531982422,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22628787904977798,
      "step": 22396
    },
    {
      "epoch": 0.44796,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.010107421875,
      "learning_rate": 0.0001,
      "loss": 3.8849,
      "loss/crossentropy": 2.012966811656952,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19502227008342743,
      "step": 22398
    },
    {
      "epoch": 0.448,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0080230712890625,
      "learning_rate": 0.0001,
      "loss": 3.9682,
      "loss/crossentropy": 1.6796467900276184,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17925339192152023,
      "step": 22400
    },
    {
      "epoch": 0.44804,
      "grad_norm": 1.75,
      "grad_norm_var": 0.006696573893229167,
      "learning_rate": 0.0001,
      "loss": 3.7172,
      "loss/crossentropy": 2.1229045391082764,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19038531184196472,
      "step": 22402
    },
    {
      "epoch": 0.44808,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.007372792561848958,
      "learning_rate": 0.0001,
      "loss": 4.1287,
      "loss/crossentropy": 2.251511335372925,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2037583664059639,
      "step": 22404
    },
    {
      "epoch": 0.44812,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.007096099853515625,
      "learning_rate": 0.0001,
      "loss": 3.9975,
      "loss/crossentropy": 1.768878161907196,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.192366823554039,
      "step": 22406
    },
    {
      "epoch": 0.44816,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.008835601806640624,
      "learning_rate": 0.0001,
      "loss": 3.7432,
      "loss/crossentropy": 1.9983880519866943,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1903810203075409,
      "step": 22408
    },
    {
      "epoch": 0.4482,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.007783762613932292,
      "learning_rate": 0.0001,
      "loss": 4.1312,
      "loss/crossentropy": 2.0608294010162354,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20538168400526047,
      "step": 22410
    },
    {
      "epoch": 0.44824,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0072509765625,
      "learning_rate": 0.0001,
      "loss": 4.0167,
      "loss/crossentropy": 2.137459099292755,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2062518671154976,
      "step": 22412
    },
    {
      "epoch": 0.44828,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.007437896728515625,
      "learning_rate": 0.0001,
      "loss": 3.8843,
      "loss/crossentropy": 2.090251922607422,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19352033734321594,
      "step": 22414
    },
    {
      "epoch": 0.44832,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.006359608968098959,
      "learning_rate": 0.0001,
      "loss": 3.9818,
      "loss/crossentropy": 1.8260875344276428,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17446549236774445,
      "step": 22416
    },
    {
      "epoch": 0.44836,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.00504150390625,
      "learning_rate": 0.0001,
      "loss": 4.1228,
      "loss/crossentropy": 2.4598742723464966,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21792489290237427,
      "step": 22418
    },
    {
      "epoch": 0.4484,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.006113433837890625,
      "learning_rate": 0.0001,
      "loss": 4.0063,
      "loss/crossentropy": 2.120418429374695,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20342590659856796,
      "step": 22420
    },
    {
      "epoch": 0.44844,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.007795969645182292,
      "learning_rate": 0.0001,
      "loss": 4.2376,
      "loss/crossentropy": 2.0946252942085266,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19902334362268448,
      "step": 22422
    },
    {
      "epoch": 0.44848,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.007112375895182292,
      "learning_rate": 0.0001,
      "loss": 4.4449,
      "loss/crossentropy": 2.395688056945801,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2158007100224495,
      "step": 22424
    },
    {
      "epoch": 0.44852,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.0090087890625,
      "learning_rate": 0.0001,
      "loss": 4.0946,
      "loss/crossentropy": 2.1429702043533325,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2130357325077057,
      "step": 22426
    },
    {
      "epoch": 0.44856,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.010050201416015625,
      "learning_rate": 0.0001,
      "loss": 4.0105,
      "loss/crossentropy": 2.0070658922195435,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1955135241150856,
      "step": 22428
    },
    {
      "epoch": 0.4486,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.009308878580729167,
      "learning_rate": 0.0001,
      "loss": 4.0902,
      "loss/crossentropy": 2.1505234241485596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20378431677818298,
      "step": 22430
    },
    {
      "epoch": 0.44864,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0091705322265625,
      "learning_rate": 0.0001,
      "loss": 4.3029,
      "loss/crossentropy": 2.4345964193344116,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19363048672676086,
      "step": 22432
    },
    {
      "epoch": 0.44868,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.010811360677083333,
      "learning_rate": 0.0001,
      "loss": 3.9744,
      "loss/crossentropy": 1.922882616519928,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18355388939380646,
      "step": 22434
    },
    {
      "epoch": 0.44872,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.008369954427083333,
      "learning_rate": 0.0001,
      "loss": 3.9232,
      "loss/crossentropy": 2.051816701889038,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21039457619190216,
      "step": 22436
    },
    {
      "epoch": 0.44876,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.009919230143229167,
      "learning_rate": 0.0001,
      "loss": 4.0486,
      "loss/crossentropy": 2.375608444213867,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21171333640813828,
      "step": 22438
    },
    {
      "epoch": 0.4488,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.008937327067057292,
      "learning_rate": 0.0001,
      "loss": 4.2023,
      "loss/crossentropy": 2.02259361743927,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2248649150133133,
      "step": 22440
    },
    {
      "epoch": 0.44884,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.007193756103515625,
      "learning_rate": 0.0001,
      "loss": 4.1429,
      "loss/crossentropy": 2.010511100292206,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19157880544662476,
      "step": 22442
    },
    {
      "epoch": 0.44888,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009030914306640625,
      "learning_rate": 0.0001,
      "loss": 4.4214,
      "loss/crossentropy": 2.3076666593551636,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2125840112566948,
      "step": 22444
    },
    {
      "epoch": 0.44892,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.010227203369140625,
      "learning_rate": 0.0001,
      "loss": 3.772,
      "loss/crossentropy": 1.780324101448059,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16909868270158768,
      "step": 22446
    },
    {
      "epoch": 0.44896,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0105621337890625,
      "learning_rate": 0.0001,
      "loss": 4.2386,
      "loss/crossentropy": 2.1713266372680664,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1884608194231987,
      "step": 22448
    },
    {
      "epoch": 0.449,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.009444173177083333,
      "learning_rate": 0.0001,
      "loss": 4.2591,
      "loss/crossentropy": 2.1273884773254395,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.181084506213665,
      "step": 22450
    },
    {
      "epoch": 0.44904,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.009850819905598959,
      "learning_rate": 0.0001,
      "loss": 4.0999,
      "loss/crossentropy": 1.8685917258262634,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1984790861606598,
      "step": 22452
    },
    {
      "epoch": 0.44908,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.006689453125,
      "learning_rate": 0.0001,
      "loss": 3.9955,
      "loss/crossentropy": 1.8099998831748962,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.186032235622406,
      "step": 22454
    },
    {
      "epoch": 0.44912,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.007063802083333333,
      "learning_rate": 0.0001,
      "loss": 3.8906,
      "loss/crossentropy": 2.111889600753784,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19286638498306274,
      "step": 22456
    },
    {
      "epoch": 0.44916,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.008984120686848958,
      "learning_rate": 0.0001,
      "loss": 3.7561,
      "loss/crossentropy": 2.041996479034424,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18484222888946533,
      "step": 22458
    },
    {
      "epoch": 0.4492,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0056874593098958336,
      "learning_rate": 0.0001,
      "loss": 3.8578,
      "loss/crossentropy": 1.755624771118164,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1807885393500328,
      "step": 22460
    },
    {
      "epoch": 0.44924,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0054840087890625,
      "learning_rate": 0.0001,
      "loss": 3.9186,
      "loss/crossentropy": 2.272615075111389,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22077642381191254,
      "step": 22462
    },
    {
      "epoch": 0.44928,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.0053179423014322914,
      "learning_rate": 0.0001,
      "loss": 3.7631,
      "loss/crossentropy": 2.1574344635009766,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1905694305896759,
      "step": 22464
    },
    {
      "epoch": 0.44932,
      "grad_norm": 2.0,
      "grad_norm_var": 0.008463287353515625,
      "learning_rate": 0.0001,
      "loss": 4.1746,
      "loss/crossentropy": 2.3979402780532837,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21937933564186096,
      "step": 22466
    },
    {
      "epoch": 0.44936,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.009250640869140625,
      "learning_rate": 0.0001,
      "loss": 4.0639,
      "loss/crossentropy": 2.181613326072693,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.219841867685318,
      "step": 22468
    },
    {
      "epoch": 0.4494,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.01041259765625,
      "learning_rate": 0.0001,
      "loss": 4.032,
      "loss/crossentropy": 2.2507534623146057,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20193026959896088,
      "step": 22470
    },
    {
      "epoch": 0.44944,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.013278961181640625,
      "learning_rate": 0.0001,
      "loss": 4.3495,
      "loss/crossentropy": 2.288329839706421,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20958562195301056,
      "step": 22472
    },
    {
      "epoch": 0.44948,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.00838623046875,
      "learning_rate": 0.0001,
      "loss": 3.8597,
      "loss/crossentropy": 2.0730547308921814,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21679005771875381,
      "step": 22474
    },
    {
      "epoch": 0.44952,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.008036041259765625,
      "learning_rate": 0.0001,
      "loss": 4.2922,
      "loss/crossentropy": 2.1609703302383423,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1996021345257759,
      "step": 22476
    },
    {
      "epoch": 0.44956,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.007738240559895833,
      "learning_rate": 0.0001,
      "loss": 4.048,
      "loss/crossentropy": 1.857498288154602,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20019329339265823,
      "step": 22478
    },
    {
      "epoch": 0.4496,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0049468994140625,
      "learning_rate": 0.0001,
      "loss": 4.1772,
      "loss/crossentropy": 1.9688559174537659,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18978293240070343,
      "step": 22480
    },
    {
      "epoch": 0.44964,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.005391184488932292,
      "learning_rate": 0.0001,
      "loss": 4.1759,
      "loss/crossentropy": 2.126068413257599,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19416333734989166,
      "step": 22482
    },
    {
      "epoch": 0.44968,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0064361572265625,
      "learning_rate": 0.0001,
      "loss": 3.9915,
      "loss/crossentropy": 2.0684805512428284,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21337899565696716,
      "step": 22484
    },
    {
      "epoch": 0.44972,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.007669830322265625,
      "learning_rate": 0.0001,
      "loss": 4.0065,
      "loss/crossentropy": 1.965733289718628,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1928843855857849,
      "step": 22486
    },
    {
      "epoch": 0.44976,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0059478759765625,
      "learning_rate": 0.0001,
      "loss": 4.1346,
      "loss/crossentropy": 1.80375075340271,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18274405598640442,
      "step": 22488
    },
    {
      "epoch": 0.4498,
      "grad_norm": 2.0,
      "grad_norm_var": 0.008284505208333333,
      "learning_rate": 0.0001,
      "loss": 3.8954,
      "loss/crossentropy": 1.939602553844452,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18628022074699402,
      "step": 22490
    },
    {
      "epoch": 0.44984,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.009276326497395833,
      "learning_rate": 0.0001,
      "loss": 4.0224,
      "loss/crossentropy": 2.408365488052368,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21072806417942047,
      "step": 22492
    },
    {
      "epoch": 0.44988,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.008961741129557292,
      "learning_rate": 0.0001,
      "loss": 4.0382,
      "loss/crossentropy": 2.1751617789268494,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21095305681228638,
      "step": 22494
    },
    {
      "epoch": 0.44992,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.03264745076497396,
      "learning_rate": 0.0001,
      "loss": 4.2429,
      "loss/crossentropy": 2.3348854780197144,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21335425972938538,
      "step": 22496
    },
    {
      "epoch": 0.44996,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.03258031209309896,
      "learning_rate": 0.0001,
      "loss": 3.9947,
      "loss/crossentropy": 2.096919059753418,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19042403250932693,
      "step": 22498
    },
    {
      "epoch": 0.45,
      "grad_norm": 1.875,
      "grad_norm_var": 0.033056640625,
      "learning_rate": 0.0001,
      "loss": 3.8055,
      "loss/crossentropy": 1.950230062007904,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19603446125984192,
      "step": 22500
    },
    {
      "epoch": 0.45004,
      "grad_norm": 2.0,
      "grad_norm_var": 0.032692209879557295,
      "learning_rate": 0.0001,
      "loss": 4.2118,
      "loss/crossentropy": 2.3010233640670776,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23052462935447693,
      "step": 22502
    },
    {
      "epoch": 0.45008,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.03386408487955729,
      "learning_rate": 0.0001,
      "loss": 3.8238,
      "loss/crossentropy": 2.143756926059723,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19040528684854507,
      "step": 22504
    },
    {
      "epoch": 0.45012,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.03865941365559896,
      "learning_rate": 0.0001,
      "loss": 4.194,
      "loss/crossentropy": 1.9510119557380676,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19400011003017426,
      "step": 22506
    },
    {
      "epoch": 0.45016,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.03951822916666667,
      "learning_rate": 0.0001,
      "loss": 4.0304,
      "loss/crossentropy": 2.2680559158325195,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22951483726501465,
      "step": 22508
    },
    {
      "epoch": 0.4502,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.19219741821289063,
      "learning_rate": 0.0001,
      "loss": 4.0042,
      "loss/crossentropy": 2.3191086053848267,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2179061621427536,
      "step": 22510
    },
    {
      "epoch": 0.45024,
      "grad_norm": 2.125,
      "grad_norm_var": 0.17602310180664063,
      "learning_rate": 0.0001,
      "loss": 4.1409,
      "loss/crossentropy": 2.2202601432800293,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21782507002353668,
      "step": 22512
    },
    {
      "epoch": 0.45028,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.17064208984375,
      "learning_rate": 0.0001,
      "loss": 4.1591,
      "loss/crossentropy": 1.9649160504341125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19763009250164032,
      "step": 22514
    },
    {
      "epoch": 0.45032,
      "grad_norm": 2.125,
      "grad_norm_var": 0.1665435791015625,
      "learning_rate": 0.0001,
      "loss": 4.4044,
      "loss/crossentropy": 2.0819711089134216,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2008282020688057,
      "step": 22516
    },
    {
      "epoch": 0.45036,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.26408589680989586,
      "learning_rate": 0.0001,
      "loss": 4.2564,
      "loss/crossentropy": 2.2396440505981445,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21240604668855667,
      "step": 22518
    },
    {
      "epoch": 0.4504,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.268542226155599,
      "learning_rate": 0.0001,
      "loss": 3.9298,
      "loss/crossentropy": 2.21976238489151,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20444615930318832,
      "step": 22520
    },
    {
      "epoch": 0.45044,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.2693072001139323,
      "learning_rate": 0.0001,
      "loss": 4.075,
      "loss/crossentropy": 2.2542803287506104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2205461636185646,
      "step": 22522
    },
    {
      "epoch": 0.45048,
      "grad_norm": 2.0,
      "grad_norm_var": 0.2702555338541667,
      "learning_rate": 0.0001,
      "loss": 3.9809,
      "loss/crossentropy": 1.7875661253929138,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17205239832401276,
      "step": 22524
    },
    {
      "epoch": 0.45052,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.13923238118489584,
      "learning_rate": 0.0001,
      "loss": 3.875,
      "loss/crossentropy": 2.0036060214042664,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21126192808151245,
      "step": 22526
    },
    {
      "epoch": 0.45056,
      "grad_norm": 2.125,
      "grad_norm_var": 0.13862279256184895,
      "learning_rate": 0.0001,
      "loss": 3.9585,
      "loss/crossentropy": 1.965367078781128,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19563131779432297,
      "step": 22528
    },
    {
      "epoch": 0.4506,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.14239273071289063,
      "learning_rate": 0.0001,
      "loss": 4.158,
      "loss/crossentropy": 1.9605298042297363,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1932133436203003,
      "step": 22530
    },
    {
      "epoch": 0.45064,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.14579671223958332,
      "learning_rate": 0.0001,
      "loss": 3.9343,
      "loss/crossentropy": 2.1047326922416687,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19184139370918274,
      "step": 22532
    },
    {
      "epoch": 0.45068,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.007104237874348958,
      "learning_rate": 0.0001,
      "loss": 4.299,
      "loss/crossentropy": 2.173476457595825,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1873651072382927,
      "step": 22534
    },
    {
      "epoch": 0.45072,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.006379191080729167,
      "learning_rate": 0.0001,
      "loss": 4.0107,
      "loss/crossentropy": 1.928326666355133,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1937912181019783,
      "step": 22536
    },
    {
      "epoch": 0.45076,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.00718994140625,
      "learning_rate": 0.0001,
      "loss": 3.9559,
      "loss/crossentropy": 2.2145062685012817,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21003177762031555,
      "step": 22538
    },
    {
      "epoch": 0.4508,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.008430989583333333,
      "learning_rate": 0.0001,
      "loss": 3.9536,
      "loss/crossentropy": 1.9980989694595337,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18583524972200394,
      "step": 22540
    },
    {
      "epoch": 0.45084,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.008552805582682291,
      "learning_rate": 0.0001,
      "loss": 4.0124,
      "loss/crossentropy": 2.0646828413009644,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20016716420650482,
      "step": 22542
    },
    {
      "epoch": 0.45088,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.006306711832682292,
      "learning_rate": 0.0001,
      "loss": 3.9468,
      "loss/crossentropy": 2.4558991193771362,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1985267549753189,
      "step": 22544
    },
    {
      "epoch": 0.45092,
      "grad_norm": 2.0,
      "grad_norm_var": 0.006589508056640625,
      "learning_rate": 0.0001,
      "loss": 4.1356,
      "loss/crossentropy": 2.140208601951599,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20745448023080826,
      "step": 22546
    },
    {
      "epoch": 0.45096,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0058176676432291664,
      "learning_rate": 0.0001,
      "loss": 4.0322,
      "loss/crossentropy": 2.0344385504722595,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20116931200027466,
      "step": 22548
    },
    {
      "epoch": 0.451,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.007306925455729167,
      "learning_rate": 0.0001,
      "loss": 4.1927,
      "loss/crossentropy": 2.213254153728485,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2145981639623642,
      "step": 22550
    },
    {
      "epoch": 0.45104,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.007260894775390625,
      "learning_rate": 0.0001,
      "loss": 4.0203,
      "loss/crossentropy": 2.2077749967575073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2086869403719902,
      "step": 22552
    },
    {
      "epoch": 0.45108,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0064208984375,
      "learning_rate": 0.0001,
      "loss": 3.9152,
      "loss/crossentropy": 2.131449520587921,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20110997557640076,
      "step": 22554
    },
    {
      "epoch": 0.45112,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.004107411702473958,
      "learning_rate": 0.0001,
      "loss": 4.0194,
      "loss/crossentropy": 1.7281222343444824,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18489576131105423,
      "step": 22556
    },
    {
      "epoch": 0.45116,
      "grad_norm": 3.4375,
      "grad_norm_var": 0.14352188110351563,
      "learning_rate": 0.0001,
      "loss": 3.9235,
      "loss/crossentropy": 1.9907403588294983,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19740281254053116,
      "step": 22558
    },
    {
      "epoch": 0.4512,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.14071451822916667,
      "learning_rate": 0.0001,
      "loss": 4.2471,
      "loss/crossentropy": 2.042698383331299,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20195024460554123,
      "step": 22560
    },
    {
      "epoch": 0.45124,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.14180272420247395,
      "learning_rate": 0.0001,
      "loss": 3.9868,
      "loss/crossentropy": 2.267351269721985,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20345290750265121,
      "step": 22562
    },
    {
      "epoch": 0.45128,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.1418413798014323,
      "learning_rate": 0.0001,
      "loss": 3.8913,
      "loss/crossentropy": 2.031484067440033,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18091221153736115,
      "step": 22564
    },
    {
      "epoch": 0.45132,
      "grad_norm": 2.125,
      "grad_norm_var": 0.14130757649739584,
      "learning_rate": 0.0001,
      "loss": 4.3266,
      "loss/crossentropy": 2.327489733695984,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21997347474098206,
      "step": 22566
    },
    {
      "epoch": 0.45136,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.1402099609375,
      "learning_rate": 0.0001,
      "loss": 4.3752,
      "loss/crossentropy": 2.117951452732086,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1775515079498291,
      "step": 22568
    },
    {
      "epoch": 0.4514,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.1423906962076823,
      "learning_rate": 0.0001,
      "loss": 3.9752,
      "loss/crossentropy": 1.9014524817466736,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19188529253005981,
      "step": 22570
    },
    {
      "epoch": 0.45144,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.13876851399739584,
      "learning_rate": 0.0001,
      "loss": 4.2163,
      "loss/crossentropy": 2.0534915924072266,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20141670107841492,
      "step": 22572
    },
    {
      "epoch": 0.45148,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.009810384114583333,
      "learning_rate": 0.0001,
      "loss": 4.3861,
      "loss/crossentropy": 2.018342673778534,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20101028680801392,
      "step": 22574
    },
    {
      "epoch": 0.45152,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.011755116780598958,
      "learning_rate": 0.0001,
      "loss": 4.0395,
      "loss/crossentropy": 2.000533401966095,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18452423810958862,
      "step": 22576
    },
    {
      "epoch": 0.45156,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.012015533447265626,
      "learning_rate": 0.0001,
      "loss": 3.9407,
      "loss/crossentropy": 2.109315812587738,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20570579916238785,
      "step": 22578
    },
    {
      "epoch": 0.4516,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.011057281494140625,
      "learning_rate": 0.0001,
      "loss": 4.1289,
      "loss/crossentropy": 2.2056689262390137,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.195539653301239,
      "step": 22580
    },
    {
      "epoch": 0.45164,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.010162099202473959,
      "learning_rate": 0.0001,
      "loss": 4.0627,
      "loss/crossentropy": 2.246178388595581,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1960083767771721,
      "step": 22582
    },
    {
      "epoch": 0.45168,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.010176340738932291,
      "learning_rate": 0.0001,
      "loss": 4.1413,
      "loss/crossentropy": 1.8758829832077026,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19170167297124863,
      "step": 22584
    },
    {
      "epoch": 0.45172,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0093994140625,
      "learning_rate": 0.0001,
      "loss": 3.9841,
      "loss/crossentropy": 1.9648401141166687,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1863914281129837,
      "step": 22586
    },
    {
      "epoch": 0.45176,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008339182535807291,
      "learning_rate": 0.0001,
      "loss": 4.2261,
      "loss/crossentropy": 2.2547377347946167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2003096491098404,
      "step": 22588
    },
    {
      "epoch": 0.4518,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.004626210530598958,
      "learning_rate": 0.0001,
      "loss": 3.9569,
      "loss/crossentropy": 1.9784765839576721,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1822790503501892,
      "step": 22590
    },
    {
      "epoch": 0.45184,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0026730855305989584,
      "learning_rate": 0.0001,
      "loss": 4.1475,
      "loss/crossentropy": 1.947198748588562,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18320369720458984,
      "step": 22592
    },
    {
      "epoch": 0.45188,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.007625071207682291,
      "learning_rate": 0.0001,
      "loss": 3.5234,
      "loss/crossentropy": 1.934649407863617,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1809309870004654,
      "step": 22594
    },
    {
      "epoch": 0.45192,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.009439849853515625,
      "learning_rate": 0.0001,
      "loss": 3.9333,
      "loss/crossentropy": 2.064886689186096,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17977996170520782,
      "step": 22596
    },
    {
      "epoch": 0.45196,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.011228179931640625,
      "learning_rate": 0.0001,
      "loss": 3.8852,
      "loss/crossentropy": 1.7243138551712036,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16666193306446075,
      "step": 22598
    },
    {
      "epoch": 0.452,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.01080322265625,
      "learning_rate": 0.0001,
      "loss": 3.9807,
      "loss/crossentropy": 2.0566282272338867,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19942068308591843,
      "step": 22600
    },
    {
      "epoch": 0.45204,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.012078603108723959,
      "learning_rate": 0.0001,
      "loss": 3.9323,
      "loss/crossentropy": 2.108401298522949,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2068444862961769,
      "step": 22602
    },
    {
      "epoch": 0.45208,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.011333974202473958,
      "learning_rate": 0.0001,
      "loss": 3.9746,
      "loss/crossentropy": 2.281686544418335,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20834453403949738,
      "step": 22604
    },
    {
      "epoch": 0.45212,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.012324778238932292,
      "learning_rate": 0.0001,
      "loss": 4.0902,
      "loss/crossentropy": 1.8627066016197205,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18476207554340363,
      "step": 22606
    },
    {
      "epoch": 0.45216,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.011567942301432292,
      "learning_rate": 0.0001,
      "loss": 4.097,
      "loss/crossentropy": 2.46126925945282,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19410762190818787,
      "step": 22608
    },
    {
      "epoch": 0.4522,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.009106190999348958,
      "learning_rate": 0.0001,
      "loss": 3.9222,
      "loss/crossentropy": 1.8764225244522095,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21187759190797806,
      "step": 22610
    },
    {
      "epoch": 0.45224,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.007092030843098959,
      "learning_rate": 0.0001,
      "loss": 3.9213,
      "loss/crossentropy": 1.8535012602806091,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1783105507493019,
      "step": 22612
    },
    {
      "epoch": 0.45228,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0054929097493489586,
      "learning_rate": 0.0001,
      "loss": 4.1948,
      "loss/crossentropy": 1.9735658764839172,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2031656950712204,
      "step": 22614
    },
    {
      "epoch": 0.45232,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.0064389546712239586,
      "learning_rate": 0.0001,
      "loss": 3.8466,
      "loss/crossentropy": 2.005250871181488,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18143242597579956,
      "step": 22616
    },
    {
      "epoch": 0.45236,
      "grad_norm": 4.84375,
      "grad_norm_var": 0.5445149739583334,
      "learning_rate": 0.0001,
      "loss": 4.1072,
      "loss/crossentropy": 2.1731566786766052,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2042900174856186,
      "step": 22618
    },
    {
      "epoch": 0.4524,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.5424496968587239,
      "learning_rate": 0.0001,
      "loss": 3.867,
      "loss/crossentropy": 2.2009552717208862,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20621322095394135,
      "step": 22620
    },
    {
      "epoch": 0.45244,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.5397664388020833,
      "learning_rate": 0.0001,
      "loss": 3.9673,
      "loss/crossentropy": 2.2014458179473877,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1910918429493904,
      "step": 22622
    },
    {
      "epoch": 0.45248,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.538287099202474,
      "learning_rate": 0.0001,
      "loss": 3.8493,
      "loss/crossentropy": 1.9113351702690125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1981104016304016,
      "step": 22624
    },
    {
      "epoch": 0.45252,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.5362223307291667,
      "learning_rate": 0.0001,
      "loss": 3.9984,
      "loss/crossentropy": 1.7521483302116394,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18573005497455597,
      "step": 22626
    },
    {
      "epoch": 0.45256,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.5408111572265625,
      "learning_rate": 0.0001,
      "loss": 3.8505,
      "loss/crossentropy": 2.003620207309723,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1799101158976555,
      "step": 22628
    },
    {
      "epoch": 0.4526,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.5576894124348958,
      "learning_rate": 0.0001,
      "loss": 4.0585,
      "loss/crossentropy": 2.17133104801178,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21434936672449112,
      "step": 22630
    },
    {
      "epoch": 0.45264,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.5505777994791666,
      "learning_rate": 0.0001,
      "loss": 3.8865,
      "loss/crossentropy": 2.113120198249817,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21246708929538727,
      "step": 22632
    },
    {
      "epoch": 0.45268,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.04535319010416667,
      "learning_rate": 0.0001,
      "loss": 3.9273,
      "loss/crossentropy": 1.9400765299797058,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19067071378231049,
      "step": 22634
    },
    {
      "epoch": 0.45272,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.04595133463541667,
      "learning_rate": 0.0001,
      "loss": 4.1057,
      "loss/crossentropy": 2.107842206954956,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1903919279575348,
      "step": 22636
    },
    {
      "epoch": 0.45276,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.046529134114583336,
      "learning_rate": 0.0001,
      "loss": 3.969,
      "loss/crossentropy": 2.093311131000519,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1854369342327118,
      "step": 22638
    },
    {
      "epoch": 0.4528,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.04711278279622396,
      "learning_rate": 0.0001,
      "loss": 4.321,
      "loss/crossentropy": 2.2625142335891724,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2128172516822815,
      "step": 22640
    },
    {
      "epoch": 0.45284,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.04737726847330729,
      "learning_rate": 0.0001,
      "loss": 3.9444,
      "loss/crossentropy": 2.13210928440094,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18905573338270187,
      "step": 22642
    },
    {
      "epoch": 0.45288,
      "grad_norm": 1.75,
      "grad_norm_var": 0.04838841756184896,
      "learning_rate": 0.0001,
      "loss": 3.8324,
      "loss/crossentropy": 2.1068539023399353,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21476633101701736,
      "step": 22644
    },
    {
      "epoch": 0.45292,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.009903971354166667,
      "learning_rate": 0.0001,
      "loss": 4.0148,
      "loss/crossentropy": 1.9843144416809082,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1885334774851799,
      "step": 22646
    },
    {
      "epoch": 0.45296,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.0125396728515625,
      "learning_rate": 0.0001,
      "loss": 4.0819,
      "loss/crossentropy": 2.3613446950912476,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2183755785226822,
      "step": 22648
    },
    {
      "epoch": 0.453,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0125152587890625,
      "learning_rate": 0.0001,
      "loss": 3.9229,
      "loss/crossentropy": 2.2134616374969482,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22959928214550018,
      "step": 22650
    },
    {
      "epoch": 0.45304,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.012532552083333334,
      "learning_rate": 0.0001,
      "loss": 4.0039,
      "loss/crossentropy": 2.332140803337097,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20412423461675644,
      "step": 22652
    },
    {
      "epoch": 0.45308,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.012601470947265625,
      "learning_rate": 0.0001,
      "loss": 4.1368,
      "loss/crossentropy": 1.7886313199996948,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1998751536011696,
      "step": 22654
    },
    {
      "epoch": 0.45312,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.012733713785807291,
      "learning_rate": 0.0001,
      "loss": 3.8701,
      "loss/crossentropy": 1.8666648864746094,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16398675739765167,
      "step": 22656
    },
    {
      "epoch": 0.45316,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.015077463785807292,
      "learning_rate": 0.0001,
      "loss": 3.9209,
      "loss/crossentropy": 1.7308924794197083,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17301977425813675,
      "step": 22658
    },
    {
      "epoch": 0.4532,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.013459269205729167,
      "learning_rate": 0.0001,
      "loss": 4.0125,
      "loss/crossentropy": 2.0344366431236267,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18964581936597824,
      "step": 22660
    },
    {
      "epoch": 0.45324,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.013991038004557291,
      "learning_rate": 0.0001,
      "loss": 4.0698,
      "loss/crossentropy": 2.1261265873908997,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20520442724227905,
      "step": 22662
    },
    {
      "epoch": 0.45328,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.008641560872395834,
      "learning_rate": 0.0001,
      "loss": 4.169,
      "loss/crossentropy": 2.163342595100403,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2064770758152008,
      "step": 22664
    },
    {
      "epoch": 0.45332,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.00960693359375,
      "learning_rate": 0.0001,
      "loss": 4.0267,
      "loss/crossentropy": 2.293798089027405,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21860123425722122,
      "step": 22666
    },
    {
      "epoch": 0.45336,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.009891764322916666,
      "learning_rate": 0.0001,
      "loss": 3.8624,
      "loss/crossentropy": 2.1405990719795227,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19475476443767548,
      "step": 22668
    },
    {
      "epoch": 0.4534,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0090240478515625,
      "learning_rate": 0.0001,
      "loss": 4.0807,
      "loss/crossentropy": 2.4012043476104736,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21024177968502045,
      "step": 22670
    },
    {
      "epoch": 0.45344,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008182525634765625,
      "learning_rate": 0.0001,
      "loss": 4.1348,
      "loss/crossentropy": 1.8321769833564758,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21026848256587982,
      "step": 22672
    },
    {
      "epoch": 0.45348,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.005832672119140625,
      "learning_rate": 0.0001,
      "loss": 3.9761,
      "loss/crossentropy": 1.7738409042358398,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17715871334075928,
      "step": 22674
    },
    {
      "epoch": 0.45352,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.0066070556640625,
      "learning_rate": 0.0001,
      "loss": 4.1336,
      "loss/crossentropy": 2.406354308128357,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20188595354557037,
      "step": 22676
    },
    {
      "epoch": 0.45356,
      "grad_norm": 1.71875,
      "grad_norm_var": 0.0100006103515625,
      "learning_rate": 0.0001,
      "loss": 3.8723,
      "loss/crossentropy": 1.977246344089508,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1766221970319748,
      "step": 22678
    },
    {
      "epoch": 0.4536,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.007922108968098958,
      "learning_rate": 0.0001,
      "loss": 3.9496,
      "loss/crossentropy": 2.2606882452964783,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21319438517093658,
      "step": 22680
    },
    {
      "epoch": 0.45364,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.013313547770182291,
      "learning_rate": 0.0001,
      "loss": 4.3151,
      "loss/crossentropy": 2.212466239929199,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21307764202356339,
      "step": 22682
    },
    {
      "epoch": 0.45368,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.012804921468098958,
      "learning_rate": 0.0001,
      "loss": 4.0279,
      "loss/crossentropy": 1.8246173858642578,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18758808821439743,
      "step": 22684
    },
    {
      "epoch": 0.45372,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.013946278889973959,
      "learning_rate": 0.0001,
      "loss": 3.9354,
      "loss/crossentropy": 2.0831995010375977,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2112889140844345,
      "step": 22686
    },
    {
      "epoch": 0.45376,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.013262685139973958,
      "learning_rate": 0.0001,
      "loss": 4.0611,
      "loss/crossentropy": 1.81759911775589,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17542830109596252,
      "step": 22688
    },
    {
      "epoch": 0.4538,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.014574178059895833,
      "learning_rate": 0.0001,
      "loss": 4.3459,
      "loss/crossentropy": 1.99278324842453,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18871523439884186,
      "step": 22690
    },
    {
      "epoch": 0.45384,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.02394383748372396,
      "learning_rate": 0.0001,
      "loss": 4.2162,
      "loss/crossentropy": 2.050497889518738,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21048897504806519,
      "step": 22692
    },
    {
      "epoch": 0.45388,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.01898981730143229,
      "learning_rate": 0.0001,
      "loss": 3.9088,
      "loss/crossentropy": 1.7781237959861755,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1762016862630844,
      "step": 22694
    },
    {
      "epoch": 0.45392,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.01878840128580729,
      "learning_rate": 0.0001,
      "loss": 3.8821,
      "loss/crossentropy": 1.8609422445297241,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18131760507822037,
      "step": 22696
    },
    {
      "epoch": 0.45396,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.018553670247395834,
      "learning_rate": 0.0001,
      "loss": 4.3945,
      "loss/crossentropy": 2.274345874786377,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22067608684301376,
      "step": 22698
    },
    {
      "epoch": 0.454,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.021036529541015626,
      "learning_rate": 0.0001,
      "loss": 4.1889,
      "loss/crossentropy": 2.0518780946731567,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.190098837018013,
      "step": 22700
    },
    {
      "epoch": 0.45404,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.020654296875,
      "learning_rate": 0.0001,
      "loss": 4.0623,
      "loss/crossentropy": 1.879017412662506,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17030316591262817,
      "step": 22702
    },
    {
      "epoch": 0.45408,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.022944895426432292,
      "learning_rate": 0.0001,
      "loss": 4.0942,
      "loss/crossentropy": 2.090358793735504,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1893768459558487,
      "step": 22704
    },
    {
      "epoch": 0.45412,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.024095662434895835,
      "learning_rate": 0.0001,
      "loss": 4.0252,
      "loss/crossentropy": 1.8985764980316162,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20475559681653976,
      "step": 22706
    },
    {
      "epoch": 0.45416,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.014662424723307291,
      "learning_rate": 0.0001,
      "loss": 4.0392,
      "loss/crossentropy": 2.1758298873901367,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19116609543561935,
      "step": 22708
    },
    {
      "epoch": 0.4542,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.014741770426432292,
      "learning_rate": 0.0001,
      "loss": 3.8379,
      "loss/crossentropy": 1.7963250279426575,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1838328242301941,
      "step": 22710
    },
    {
      "epoch": 0.45424,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.015046183268229167,
      "learning_rate": 0.0001,
      "loss": 4.2924,
      "loss/crossentropy": 2.4492361545562744,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21029194444417953,
      "step": 22712
    },
    {
      "epoch": 0.45428,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0115386962890625,
      "learning_rate": 0.0001,
      "loss": 3.9314,
      "loss/crossentropy": 2.10919725894928,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20362409204244614,
      "step": 22714
    },
    {
      "epoch": 0.45432,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.008713531494140624,
      "learning_rate": 0.0001,
      "loss": 4.0962,
      "loss/crossentropy": 2.132240355014801,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18763454258441925,
      "step": 22716
    },
    {
      "epoch": 0.45436,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.008048502604166667,
      "learning_rate": 0.0001,
      "loss": 4.0287,
      "loss/crossentropy": 2.1648387908935547,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2098705694079399,
      "step": 22718
    },
    {
      "epoch": 0.4544,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.008109283447265626,
      "learning_rate": 0.0001,
      "loss": 3.908,
      "loss/crossentropy": 2.2334399223327637,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21508647501468658,
      "step": 22720
    },
    {
      "epoch": 0.45444,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.006101226806640625,
      "learning_rate": 0.0001,
      "loss": 3.8521,
      "loss/crossentropy": 1.8207709193229675,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1902608647942543,
      "step": 22722
    },
    {
      "epoch": 0.45448,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.007191721598307292,
      "learning_rate": 0.0001,
      "loss": 4.0834,
      "loss/crossentropy": 2.094780683517456,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20928193628787994,
      "step": 22724
    },
    {
      "epoch": 0.45452,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.006791178385416667,
      "learning_rate": 0.0001,
      "loss": 3.8648,
      "loss/crossentropy": 2.164665937423706,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19287051260471344,
      "step": 22726
    },
    {
      "epoch": 0.45456,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0062164306640625,
      "learning_rate": 0.0001,
      "loss": 3.9342,
      "loss/crossentropy": 2.2440634965896606,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18155057728290558,
      "step": 22728
    },
    {
      "epoch": 0.4546,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.008190663655598958,
      "learning_rate": 0.0001,
      "loss": 4.2287,
      "loss/crossentropy": 2.012277126312256,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21760402619838715,
      "step": 22730
    },
    {
      "epoch": 0.45464,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.01033935546875,
      "learning_rate": 0.0001,
      "loss": 4.0072,
      "loss/crossentropy": 2.1217371225357056,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1944856494665146,
      "step": 22732
    },
    {
      "epoch": 0.45468,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.010300445556640624,
      "learning_rate": 0.0001,
      "loss": 3.9806,
      "loss/crossentropy": 2.3193390369415283,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21992851048707962,
      "step": 22734
    },
    {
      "epoch": 0.45472,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.012996164957682292,
      "learning_rate": 0.0001,
      "loss": 4.3389,
      "loss/crossentropy": 2.520304322242737,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.229971744120121,
      "step": 22736
    },
    {
      "epoch": 0.45476,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.012898763020833334,
      "learning_rate": 0.0001,
      "loss": 4.0665,
      "loss/crossentropy": 2.270485758781433,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22266070544719696,
      "step": 22738
    },
    {
      "epoch": 0.4548,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.011945597330729167,
      "learning_rate": 0.0001,
      "loss": 4.1006,
      "loss/crossentropy": 2.046397566795349,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19661714136600494,
      "step": 22740
    },
    {
      "epoch": 0.45484,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.012788645426432292,
      "learning_rate": 0.0001,
      "loss": 4.0564,
      "loss/crossentropy": 1.925121009349823,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19327251613140106,
      "step": 22742
    },
    {
      "epoch": 0.45488,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.01253662109375,
      "learning_rate": 0.0001,
      "loss": 4.0178,
      "loss/crossentropy": 2.1238789558410645,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18549078702926636,
      "step": 22744
    },
    {
      "epoch": 0.45492,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.01082763671875,
      "learning_rate": 0.0001,
      "loss": 3.8122,
      "loss/crossentropy": 1.8756769299507141,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1634177565574646,
      "step": 22746
    },
    {
      "epoch": 0.45496,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.012555948893229167,
      "learning_rate": 0.0001,
      "loss": 4.1419,
      "loss/crossentropy": 2.017587959766388,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.187566876411438,
      "step": 22748
    },
    {
      "epoch": 0.455,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.011735026041666667,
      "learning_rate": 0.0001,
      "loss": 3.8677,
      "loss/crossentropy": 1.9583166241645813,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20180843770503998,
      "step": 22750
    },
    {
      "epoch": 0.45504,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.008259836832682292,
      "learning_rate": 0.0001,
      "loss": 3.8475,
      "loss/crossentropy": 1.9483489990234375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19634968787431717,
      "step": 22752
    },
    {
      "epoch": 0.45508,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.008270009358723959,
      "learning_rate": 0.0001,
      "loss": 3.8734,
      "loss/crossentropy": 1.727650761604309,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17284797132015228,
      "step": 22754
    },
    {
      "epoch": 0.45512,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.009262847900390624,
      "learning_rate": 0.0001,
      "loss": 3.9471,
      "loss/crossentropy": 2.306153416633606,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20535488426685333,
      "step": 22756
    },
    {
      "epoch": 0.45516,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.007755279541015625,
      "learning_rate": 0.0001,
      "loss": 4.0382,
      "loss/crossentropy": 1.8752743601799011,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19387705624103546,
      "step": 22758
    },
    {
      "epoch": 0.4552,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.007941691080729167,
      "learning_rate": 0.0001,
      "loss": 4.2708,
      "loss/crossentropy": 2.4626048803329468,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21318693459033966,
      "step": 22760
    },
    {
      "epoch": 0.45524,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.008503977457682292,
      "learning_rate": 0.0001,
      "loss": 3.7236,
      "loss/crossentropy": 1.968520700931549,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19532684981822968,
      "step": 22762
    },
    {
      "epoch": 0.45528,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0135162353515625,
      "learning_rate": 0.0001,
      "loss": 4.0531,
      "loss/crossentropy": 1.75477135181427,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17470379173755646,
      "step": 22764
    },
    {
      "epoch": 0.45532,
      "grad_norm": 2.0,
      "grad_norm_var": 0.017325846354166667,
      "learning_rate": 0.0001,
      "loss": 4.0837,
      "loss/crossentropy": 2.125413417816162,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21318642050027847,
      "step": 22766
    },
    {
      "epoch": 0.45536,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.01915257771809896,
      "learning_rate": 0.0001,
      "loss": 4.1433,
      "loss/crossentropy": 2.2569944858551025,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.204448863863945,
      "step": 22768
    },
    {
      "epoch": 0.4554,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.018379720052083333,
      "learning_rate": 0.0001,
      "loss": 3.943,
      "loss/crossentropy": 2.219534397125244,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20692889392375946,
      "step": 22770
    },
    {
      "epoch": 0.45544,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.015372467041015626,
      "learning_rate": 0.0001,
      "loss": 4.0119,
      "loss/crossentropy": 2.1480907797813416,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20739484578371048,
      "step": 22772
    },
    {
      "epoch": 0.45548,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.016379547119140626,
      "learning_rate": 0.0001,
      "loss": 3.8875,
      "loss/crossentropy": 1.9118182063102722,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17766325920820236,
      "step": 22774
    },
    {
      "epoch": 0.45552,
      "grad_norm": 2.0,
      "grad_norm_var": 0.018900299072265626,
      "learning_rate": 0.0001,
      "loss": 3.9723,
      "loss/crossentropy": 2.0928712487220764,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19869665801525116,
      "step": 22776
    },
    {
      "epoch": 0.45556,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.022031402587890624,
      "learning_rate": 0.0001,
      "loss": 4.4284,
      "loss/crossentropy": 2.1542623043060303,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22383734583854675,
      "step": 22778
    },
    {
      "epoch": 0.4556,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.019327545166015626,
      "learning_rate": 0.0001,
      "loss": 4.0714,
      "loss/crossentropy": 2.1210632920265198,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19574230164289474,
      "step": 22780
    },
    {
      "epoch": 0.45564,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.01756566365559896,
      "learning_rate": 0.0001,
      "loss": 3.8178,
      "loss/crossentropy": 1.975026547908783,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18365998566150665,
      "step": 22782
    },
    {
      "epoch": 0.45568,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.017549641927083335,
      "learning_rate": 0.0001,
      "loss": 3.4904,
      "loss/crossentropy": 1.5320480465888977,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15212352573871613,
      "step": 22784
    },
    {
      "epoch": 0.45572,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.01838963826497396,
      "learning_rate": 0.0001,
      "loss": 3.9335,
      "loss/crossentropy": 1.8474896550178528,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18486780673265457,
      "step": 22786
    },
    {
      "epoch": 0.45576,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.018553670247395834,
      "learning_rate": 0.0001,
      "loss": 4.037,
      "loss/crossentropy": 1.9554376006126404,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18804097175598145,
      "step": 22788
    },
    {
      "epoch": 0.4558,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.018968709309895835,
      "learning_rate": 0.0001,
      "loss": 4.1759,
      "loss/crossentropy": 2.110785663127899,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21371331810951233,
      "step": 22790
    },
    {
      "epoch": 0.45584,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.016471099853515626,
      "learning_rate": 0.0001,
      "loss": 4.1691,
      "loss/crossentropy": 2.100754976272583,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21454931795597076,
      "step": 22792
    },
    {
      "epoch": 0.45588,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.006845855712890625,
      "learning_rate": 0.0001,
      "loss": 4.0916,
      "loss/crossentropy": 2.1975361704826355,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2107257917523384,
      "step": 22794
    },
    {
      "epoch": 0.45592,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006961822509765625,
      "learning_rate": 0.0001,
      "loss": 4.1367,
      "loss/crossentropy": 2.216245174407959,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22839964926242828,
      "step": 22796
    },
    {
      "epoch": 0.45596,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.007798004150390625,
      "learning_rate": 0.0001,
      "loss": 3.8334,
      "loss/crossentropy": 1.9227718710899353,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18401919305324554,
      "step": 22798
    },
    {
      "epoch": 0.456,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.007348378499348958,
      "learning_rate": 0.0001,
      "loss": 3.6818,
      "loss/crossentropy": 1.916443407535553,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20578119158744812,
      "step": 22800
    },
    {
      "epoch": 0.45604,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006947580973307292,
      "learning_rate": 0.0001,
      "loss": 3.9813,
      "loss/crossentropy": 1.9856197834014893,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20684893429279327,
      "step": 22802
    },
    {
      "epoch": 0.45608,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.00740966796875,
      "learning_rate": 0.0001,
      "loss": 4.2125,
      "loss/crossentropy": 2.284825623035431,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19741405546665192,
      "step": 22804
    },
    {
      "epoch": 0.45612,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.00848388671875,
      "learning_rate": 0.0001,
      "loss": 3.9399,
      "loss/crossentropy": 2.28286874294281,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18636887520551682,
      "step": 22806
    },
    {
      "epoch": 0.45616,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.008121490478515625,
      "learning_rate": 0.0001,
      "loss": 4.0123,
      "loss/crossentropy": 2.1754974722862244,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18408793210983276,
      "step": 22808
    },
    {
      "epoch": 0.4562,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.011668904622395834,
      "learning_rate": 0.0001,
      "loss": 3.9584,
      "loss/crossentropy": 1.9059696793556213,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17439302802085876,
      "step": 22810
    },
    {
      "epoch": 0.45624,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011075846354166667,
      "learning_rate": 0.0001,
      "loss": 4.0839,
      "loss/crossentropy": 2.2676509618759155,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20085486769676208,
      "step": 22812
    },
    {
      "epoch": 0.45628,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.010788726806640624,
      "learning_rate": 0.0001,
      "loss": 3.9465,
      "loss/crossentropy": 2.1320562958717346,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19100098311901093,
      "step": 22814
    },
    {
      "epoch": 0.45632,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0119293212890625,
      "learning_rate": 0.0001,
      "loss": 4.2211,
      "loss/crossentropy": 2.1827250719070435,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1999506950378418,
      "step": 22816
    },
    {
      "epoch": 0.45636,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.01083984375,
      "learning_rate": 0.0001,
      "loss": 3.7179,
      "loss/crossentropy": 1.861267626285553,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19007404148578644,
      "step": 22818
    },
    {
      "epoch": 0.4564,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.014749908447265625,
      "learning_rate": 0.0001,
      "loss": 4.0594,
      "loss/crossentropy": 2.0799012184143066,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22177287936210632,
      "step": 22820
    },
    {
      "epoch": 0.45644,
      "grad_norm": 2.125,
      "grad_norm_var": 0.013742828369140625,
      "learning_rate": 0.0001,
      "loss": 4.2555,
      "loss/crossentropy": 1.847994089126587,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17684108018875122,
      "step": 22822
    },
    {
      "epoch": 0.45648,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.011685943603515625,
      "learning_rate": 0.0001,
      "loss": 4.17,
      "loss/crossentropy": 1.8111347556114197,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1796310842037201,
      "step": 22824
    },
    {
      "epoch": 0.45652,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.011971028645833333,
      "learning_rate": 0.0001,
      "loss": 3.7134,
      "loss/crossentropy": 2.038852334022522,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20186404883861542,
      "step": 22826
    },
    {
      "epoch": 0.45656,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.09888509114583334,
      "learning_rate": 0.0001,
      "loss": 3.6857,
      "loss/crossentropy": 1.6407727003097534,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1637558490037918,
      "step": 22828
    },
    {
      "epoch": 0.4566,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.09705174763997396,
      "learning_rate": 0.0001,
      "loss": 4.2396,
      "loss/crossentropy": 2.240299344062805,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22101856023073196,
      "step": 22830
    },
    {
      "epoch": 0.45664,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.10135269165039062,
      "learning_rate": 0.0001,
      "loss": 3.744,
      "loss/crossentropy": 2.0920748114585876,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1953132450580597,
      "step": 22832
    },
    {
      "epoch": 0.45668,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.09823404947916667,
      "learning_rate": 0.0001,
      "loss": 4.4837,
      "loss/crossentropy": 2.5458264350891113,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22436098754405975,
      "step": 22834
    },
    {
      "epoch": 0.45672,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.1007232666015625,
      "learning_rate": 0.0001,
      "loss": 4.228,
      "loss/crossentropy": 2.2389872074127197,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19549836963415146,
      "step": 22836
    },
    {
      "epoch": 0.45676,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.10216852823893229,
      "learning_rate": 0.0001,
      "loss": 4.0639,
      "loss/crossentropy": 1.943382740020752,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17616666853427887,
      "step": 22838
    },
    {
      "epoch": 0.4568,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.10617574055989583,
      "learning_rate": 0.0001,
      "loss": 3.8807,
      "loss/crossentropy": 2.3038381338119507,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21248196810483932,
      "step": 22840
    },
    {
      "epoch": 0.45684,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.10474446614583334,
      "learning_rate": 0.0001,
      "loss": 4.0004,
      "loss/crossentropy": 2.1889474391937256,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21692103147506714,
      "step": 22842
    },
    {
      "epoch": 0.45688,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.010959625244140625,
      "learning_rate": 0.0001,
      "loss": 3.8093,
      "loss/crossentropy": 1.7324257493019104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1752893403172493,
      "step": 22844
    },
    {
      "epoch": 0.45692,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.009358469645182292,
      "learning_rate": 0.0001,
      "loss": 3.7886,
      "loss/crossentropy": 1.5598450899124146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1739826500415802,
      "step": 22846
    },
    {
      "epoch": 0.45696,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.012699381510416666,
      "learning_rate": 0.0001,
      "loss": 4.1388,
      "loss/crossentropy": 2.629135251045227,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23805684596300125,
      "step": 22848
    },
    {
      "epoch": 0.457,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0129058837890625,
      "learning_rate": 0.0001,
      "loss": 4.0862,
      "loss/crossentropy": 2.2966033220291138,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20307788252830505,
      "step": 22850
    },
    {
      "epoch": 0.45704,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.015372467041015626,
      "learning_rate": 0.0001,
      "loss": 3.8963,
      "loss/crossentropy": 2.111830711364746,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.220965214073658,
      "step": 22852
    },
    {
      "epoch": 0.45708,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.016657511393229168,
      "learning_rate": 0.0001,
      "loss": 3.8466,
      "loss/crossentropy": 1.9285584688186646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1810559406876564,
      "step": 22854
    },
    {
      "epoch": 0.45712,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.014798990885416667,
      "learning_rate": 0.0001,
      "loss": 3.9924,
      "loss/crossentropy": 2.059799551963806,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19754227995872498,
      "step": 22856
    },
    {
      "epoch": 0.45716,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.013330078125,
      "learning_rate": 0.0001,
      "loss": 4.1501,
      "loss/crossentropy": 2.219936490058899,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20215890556573868,
      "step": 22858
    },
    {
      "epoch": 0.4572,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.015541330973307291,
      "learning_rate": 0.0001,
      "loss": 3.6917,
      "loss/crossentropy": 1.639923632144928,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15062974393367767,
      "step": 22860
    },
    {
      "epoch": 0.45724,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.013331858317057292,
      "learning_rate": 0.0001,
      "loss": 3.9225,
      "loss/crossentropy": 2.038732647895813,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19826284795999527,
      "step": 22862
    },
    {
      "epoch": 0.45728,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.012969716389973959,
      "learning_rate": 0.0001,
      "loss": 3.7399,
      "loss/crossentropy": 1.9492093920707703,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18503065407276154,
      "step": 22864
    },
    {
      "epoch": 0.45732,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.011061350504557291,
      "learning_rate": 0.0001,
      "loss": 3.8676,
      "loss/crossentropy": 2.1745853424072266,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20653871446847916,
      "step": 22866
    },
    {
      "epoch": 0.45736,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.008536529541015626,
      "learning_rate": 0.0001,
      "loss": 4.1377,
      "loss/crossentropy": 2.3043514490127563,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20302630215883255,
      "step": 22868
    },
    {
      "epoch": 0.4574,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.006730143229166667,
      "learning_rate": 0.0001,
      "loss": 4.0241,
      "loss/crossentropy": 2.032300651073456,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1972976177930832,
      "step": 22870
    },
    {
      "epoch": 0.45744,
      "grad_norm": 1.75,
      "grad_norm_var": 0.007783762613932292,
      "learning_rate": 0.0001,
      "loss": 3.9334,
      "loss/crossentropy": 1.7591362595558167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1877521052956581,
      "step": 22872
    },
    {
      "epoch": 0.45748,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.010179646809895833,
      "learning_rate": 0.0001,
      "loss": 3.9741,
      "loss/crossentropy": 1.8232863545417786,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19946040213108063,
      "step": 22874
    },
    {
      "epoch": 0.45752,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.007627105712890625,
      "learning_rate": 0.0001,
      "loss": 4.1899,
      "loss/crossentropy": 2.3904630541801453,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21164552122354507,
      "step": 22876
    },
    {
      "epoch": 0.45756,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.007377115885416666,
      "learning_rate": 0.0001,
      "loss": 3.9767,
      "loss/crossentropy": 2.0109705328941345,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19774679094552994,
      "step": 22878
    },
    {
      "epoch": 0.4576,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.006381988525390625,
      "learning_rate": 0.0001,
      "loss": 3.9764,
      "loss/crossentropy": 2.049817442893982,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18809376657009125,
      "step": 22880
    },
    {
      "epoch": 0.45764,
      "grad_norm": 2.25,
      "grad_norm_var": 0.013444010416666667,
      "learning_rate": 0.0001,
      "loss": 3.9405,
      "loss/crossentropy": 1.9604635834693909,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21582353115081787,
      "step": 22882
    },
    {
      "epoch": 0.45768,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.021355946858723957,
      "learning_rate": 0.0001,
      "loss": 4.2544,
      "loss/crossentropy": 2.0224910378456116,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21511054039001465,
      "step": 22884
    },
    {
      "epoch": 0.45772,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.022191365559895832,
      "learning_rate": 0.0001,
      "loss": 4.1981,
      "loss/crossentropy": 2.1743668913841248,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20319290459156036,
      "step": 22886
    },
    {
      "epoch": 0.45776,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.017160797119140626,
      "learning_rate": 0.0001,
      "loss": 4.0452,
      "loss/crossentropy": 2.0692490339279175,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21043430268764496,
      "step": 22888
    },
    {
      "epoch": 0.4578,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.01715672810872396,
      "learning_rate": 0.0001,
      "loss": 3.9975,
      "loss/crossentropy": 2.2976402044296265,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21817487478256226,
      "step": 22890
    },
    {
      "epoch": 0.45784,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.0191162109375,
      "learning_rate": 0.0001,
      "loss": 4.11,
      "loss/crossentropy": 1.9541950225830078,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1726100593805313,
      "step": 22892
    },
    {
      "epoch": 0.45788,
      "grad_norm": 1.7421875,
      "grad_norm_var": 0.022997029622395835,
      "learning_rate": 0.0001,
      "loss": 3.9245,
      "loss/crossentropy": 1.8908615112304688,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17845968157052994,
      "step": 22894
    },
    {
      "epoch": 0.45792,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.023522694905598957,
      "learning_rate": 0.0001,
      "loss": 4.1408,
      "loss/crossentropy": 2.0752804279327393,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17747191339731216,
      "step": 22896
    },
    {
      "epoch": 0.45796,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.02047119140625,
      "learning_rate": 0.0001,
      "loss": 3.6761,
      "loss/crossentropy": 1.7240750789642334,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17918948084115982,
      "step": 22898
    },
    {
      "epoch": 0.458,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.011519114176432291,
      "learning_rate": 0.0001,
      "loss": 4.0632,
      "loss/crossentropy": 2.208250343799591,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1937680020928383,
      "step": 22900
    },
    {
      "epoch": 0.45804,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.008478800455729166,
      "learning_rate": 0.0001,
      "loss": 4.1283,
      "loss/crossentropy": 2.076810359954834,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18243352323770523,
      "step": 22902
    },
    {
      "epoch": 0.45808,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.006624094645182292,
      "learning_rate": 0.0001,
      "loss": 4.2553,
      "loss/crossentropy": 1.929943025112152,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19618961960077286,
      "step": 22904
    },
    {
      "epoch": 0.45812,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0069976806640625,
      "learning_rate": 0.0001,
      "loss": 4.1791,
      "loss/crossentropy": 2.186367392539978,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19272585958242416,
      "step": 22906
    },
    {
      "epoch": 0.45816,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0063385009765625,
      "learning_rate": 0.0001,
      "loss": 4.1156,
      "loss/crossentropy": 2.1537517309188843,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1961517035961151,
      "step": 22908
    },
    {
      "epoch": 0.4582,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.004325103759765625,
      "learning_rate": 0.0001,
      "loss": 4.1084,
      "loss/crossentropy": 2.0141521096229553,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1909116804599762,
      "step": 22910
    },
    {
      "epoch": 0.45824,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.004351552327473958,
      "learning_rate": 0.0001,
      "loss": 4.1536,
      "loss/crossentropy": 1.9720463752746582,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18549032509326935,
      "step": 22912
    },
    {
      "epoch": 0.45828,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.002243804931640625,
      "learning_rate": 0.0001,
      "loss": 4.0425,
      "loss/crossentropy": 1.6521074771881104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.172331303358078,
      "step": 22914
    },
    {
      "epoch": 0.45832,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.013109334309895833,
      "learning_rate": 0.0001,
      "loss": 3.8738,
      "loss/crossentropy": 2.1382288932800293,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19402942061424255,
      "step": 22916
    },
    {
      "epoch": 0.45836,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.013688151041666667,
      "learning_rate": 0.0001,
      "loss": 4.0196,
      "loss/crossentropy": 2.348258137702942,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21549250930547714,
      "step": 22918
    },
    {
      "epoch": 0.4584,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.02235107421875,
      "learning_rate": 0.0001,
      "loss": 4.0723,
      "loss/crossentropy": 1.8907782435417175,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18751342594623566,
      "step": 22920
    },
    {
      "epoch": 0.45844,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0222808837890625,
      "learning_rate": 0.0001,
      "loss": 4.0556,
      "loss/crossentropy": 2.189586043357849,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1954984813928604,
      "step": 22922
    },
    {
      "epoch": 0.45848,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.021491495768229167,
      "learning_rate": 0.0001,
      "loss": 3.8171,
      "loss/crossentropy": 1.8710416555404663,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1760418340563774,
      "step": 22924
    },
    {
      "epoch": 0.45852,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0213775634765625,
      "learning_rate": 0.0001,
      "loss": 4.4274,
      "loss/crossentropy": 2.3559741973876953,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21578038483858109,
      "step": 22926
    },
    {
      "epoch": 0.45856,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.022739410400390625,
      "learning_rate": 0.0001,
      "loss": 3.7592,
      "loss/crossentropy": 1.8588098883628845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18936273455619812,
      "step": 22928
    },
    {
      "epoch": 0.4586,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.024544270833333333,
      "learning_rate": 0.0001,
      "loss": 4.0378,
      "loss/crossentropy": 2.039194941520691,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1917971894145012,
      "step": 22930
    },
    {
      "epoch": 0.45864,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.016739654541015624,
      "learning_rate": 0.0001,
      "loss": 3.7675,
      "loss/crossentropy": 1.7586663365364075,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16597917675971985,
      "step": 22932
    },
    {
      "epoch": 0.45868,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.01702880859375,
      "learning_rate": 0.0001,
      "loss": 3.7682,
      "loss/crossentropy": 2.086498737335205,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21375593543052673,
      "step": 22934
    },
    {
      "epoch": 0.45872,
      "grad_norm": 2.0,
      "grad_norm_var": 0.007853190104166666,
      "learning_rate": 0.0001,
      "loss": 4.0781,
      "loss/crossentropy": 2.14085054397583,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2136429250240326,
      "step": 22936
    },
    {
      "epoch": 0.45876,
      "grad_norm": 2.671875,
      "grad_norm_var": 0.04417724609375,
      "learning_rate": 0.0001,
      "loss": 4.064,
      "loss/crossentropy": 1.7813395261764526,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23206777125597,
      "step": 22938
    },
    {
      "epoch": 0.4588,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.04527587890625,
      "learning_rate": 0.0001,
      "loss": 3.8535,
      "loss/crossentropy": 2.267697334289551,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19420506060123444,
      "step": 22940
    },
    {
      "epoch": 0.45884,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.045328776041666664,
      "learning_rate": 0.0001,
      "loss": 4.1133,
      "loss/crossentropy": 1.9011476039886475,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20060188323259354,
      "step": 22942
    },
    {
      "epoch": 0.45888,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.043822987874348955,
      "learning_rate": 0.0001,
      "loss": 4.0148,
      "loss/crossentropy": 2.0690804719924927,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20410669595003128,
      "step": 22944
    },
    {
      "epoch": 0.45892,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0429443359375,
      "learning_rate": 0.0001,
      "loss": 4.2914,
      "loss/crossentropy": 2.05231511592865,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.200359046459198,
      "step": 22946
    },
    {
      "epoch": 0.45896,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.04160741170247396,
      "learning_rate": 0.0001,
      "loss": 4.2998,
      "loss/crossentropy": 2.167905569076538,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20565193891525269,
      "step": 22948
    },
    {
      "epoch": 0.459,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.041290028889973955,
      "learning_rate": 0.0001,
      "loss": 3.911,
      "loss/crossentropy": 1.8297042846679688,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19305560737848282,
      "step": 22950
    },
    {
      "epoch": 0.45904,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.042335764567057295,
      "learning_rate": 0.0001,
      "loss": 4.2231,
      "loss/crossentropy": 2.1130523681640625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2063598930835724,
      "step": 22952
    },
    {
      "epoch": 0.45908,
      "grad_norm": 2.0,
      "grad_norm_var": 0.011889394124348958,
      "learning_rate": 0.0001,
      "loss": 4.0074,
      "loss/crossentropy": 1.8652898669242859,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19195117056369781,
      "step": 22954
    },
    {
      "epoch": 0.45912,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.009948476155598959,
      "learning_rate": 0.0001,
      "loss": 4.2163,
      "loss/crossentropy": 2.283188223838806,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21027832478284836,
      "step": 22956
    },
    {
      "epoch": 0.45916,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.010990397135416666,
      "learning_rate": 0.0001,
      "loss": 4.1242,
      "loss/crossentropy": 1.9148901104927063,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20047999173402786,
      "step": 22958
    },
    {
      "epoch": 0.4592,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.013936360677083334,
      "learning_rate": 0.0001,
      "loss": 4.1457,
      "loss/crossentropy": 2.0488327741622925,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18601303547620773,
      "step": 22960
    },
    {
      "epoch": 0.45924,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.014810943603515625,
      "learning_rate": 0.0001,
      "loss": 4.0482,
      "loss/crossentropy": 2.316429853439331,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18690580874681473,
      "step": 22962
    },
    {
      "epoch": 0.45928,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.013260650634765624,
      "learning_rate": 0.0001,
      "loss": 3.8958,
      "loss/crossentropy": 2.2469218373298645,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1983572617173195,
      "step": 22964
    },
    {
      "epoch": 0.45932,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.014964803059895834,
      "learning_rate": 0.0001,
      "loss": 3.7772,
      "loss/crossentropy": 1.988997757434845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17721036076545715,
      "step": 22966
    },
    {
      "epoch": 0.45936,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.016259511311848957,
      "learning_rate": 0.0001,
      "loss": 3.6331,
      "loss/crossentropy": 2.0831198692321777,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20350365340709686,
      "step": 22968
    },
    {
      "epoch": 0.4594,
      "grad_norm": 1.875,
      "grad_norm_var": 0.016599273681640624,
      "learning_rate": 0.0001,
      "loss": 4.1053,
      "loss/crossentropy": 1.9105250239372253,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24999283254146576,
      "step": 22970
    },
    {
      "epoch": 0.45944,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.014851633707682292,
      "learning_rate": 0.0001,
      "loss": 3.96,
      "loss/crossentropy": 1.990349531173706,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21390174329280853,
      "step": 22972
    },
    {
      "epoch": 0.45948,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.011759440104166666,
      "learning_rate": 0.0001,
      "loss": 3.981,
      "loss/crossentropy": 1.8016277551651,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16551107168197632,
      "step": 22974
    },
    {
      "epoch": 0.45952,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.008925120035807291,
      "learning_rate": 0.0001,
      "loss": 3.7665,
      "loss/crossentropy": 2.0964725017547607,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18953562527894974,
      "step": 22976
    },
    {
      "epoch": 0.45956,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.008455149332682292,
      "learning_rate": 0.0001,
      "loss": 4.0136,
      "loss/crossentropy": 1.8826870322227478,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19081643223762512,
      "step": 22978
    },
    {
      "epoch": 0.4596,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.009175618489583334,
      "learning_rate": 0.0001,
      "loss": 4.1068,
      "loss/crossentropy": 2.3349568843841553,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2133205682039261,
      "step": 22980
    },
    {
      "epoch": 0.45964,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009912109375,
      "learning_rate": 0.0001,
      "loss": 4.0551,
      "loss/crossentropy": 2.1611928939819336,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18273824453353882,
      "step": 22982
    },
    {
      "epoch": 0.45968,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.008969879150390625,
      "learning_rate": 0.0001,
      "loss": 4.1389,
      "loss/crossentropy": 1.9757606387138367,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20463839918375015,
      "step": 22984
    },
    {
      "epoch": 0.45972,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.008320871988932292,
      "learning_rate": 0.0001,
      "loss": 3.9488,
      "loss/crossentropy": 2.075642466545105,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19987941533327103,
      "step": 22986
    },
    {
      "epoch": 0.45976,
      "grad_norm": 2.0,
      "grad_norm_var": 0.009333292643229166,
      "learning_rate": 0.0001,
      "loss": 3.8588,
      "loss/crossentropy": 1.805062711238861,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16294736042618752,
      "step": 22988
    },
    {
      "epoch": 0.4598,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.008907063802083334,
      "learning_rate": 0.0001,
      "loss": 3.9069,
      "loss/crossentropy": 1.7681823372840881,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16899758577346802,
      "step": 22990
    },
    {
      "epoch": 0.45984,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009837849934895834,
      "learning_rate": 0.0001,
      "loss": 4.1451,
      "loss/crossentropy": 2.2006375789642334,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2165016457438469,
      "step": 22992
    },
    {
      "epoch": 0.45988,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.009822591145833334,
      "learning_rate": 0.0001,
      "loss": 4.0045,
      "loss/crossentropy": 2.141044855117798,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19580233097076416,
      "step": 22994
    },
    {
      "epoch": 0.45992,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.007818349202473958,
      "learning_rate": 0.0001,
      "loss": 4.1228,
      "loss/crossentropy": 1.8378351926803589,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1891382709145546,
      "step": 22996
    },
    {
      "epoch": 0.45996,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.00626220703125,
      "learning_rate": 0.0001,
      "loss": 4.0411,
      "loss/crossentropy": 1.9486631751060486,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18766219913959503,
      "step": 22998
    },
    {
      "epoch": 0.46,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.05397847493489583,
      "learning_rate": 0.0001,
      "loss": 4.0377,
      "loss/crossentropy": 2.2083182334899902,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19954963773488998,
      "step": 23000
    },
    {
      "epoch": 0.46004,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.0544586181640625,
      "learning_rate": 0.0001,
      "loss": 3.8961,
      "loss/crossentropy": 2.163831114768982,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2008308321237564,
      "step": 23002
    },
    {
      "epoch": 0.46008,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.053138987223307295,
      "learning_rate": 0.0001,
      "loss": 4.1704,
      "loss/crossentropy": 2.256836771965027,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20936761796474457,
      "step": 23004
    },
    {
      "epoch": 0.46012,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.051273345947265625,
      "learning_rate": 0.0001,
      "loss": 4.2633,
      "loss/crossentropy": 2.2618257999420166,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2167700231075287,
      "step": 23006
    },
    {
      "epoch": 0.46016,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.08671875,
      "learning_rate": 0.0001,
      "loss": 3.8584,
      "loss/crossentropy": 1.7830750346183777,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17645323276519775,
      "step": 23008
    },
    {
      "epoch": 0.4602,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.08814264933268229,
      "learning_rate": 0.0001,
      "loss": 4.0131,
      "loss/crossentropy": 2.027057111263275,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19191700220108032,
      "step": 23010
    },
    {
      "epoch": 0.46024,
      "grad_norm": 1.875,
      "grad_norm_var": 0.09068094889322917,
      "learning_rate": 0.0001,
      "loss": 3.9818,
      "loss/crossentropy": 2.138069987297058,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1890278086066246,
      "step": 23012
    },
    {
      "epoch": 0.46028,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.09110107421875,
      "learning_rate": 0.0001,
      "loss": 3.928,
      "loss/crossentropy": 1.8553322553634644,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18419906497001648,
      "step": 23014
    },
    {
      "epoch": 0.46032,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.049627431233723956,
      "learning_rate": 0.0001,
      "loss": 3.8015,
      "loss/crossentropy": 2.029839515686035,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17909887433052063,
      "step": 23016
    },
    {
      "epoch": 0.46036,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.048868815104166664,
      "learning_rate": 0.0001,
      "loss": 3.9726,
      "loss/crossentropy": 2.2698041200637817,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.202784962952137,
      "step": 23018
    },
    {
      "epoch": 0.4604,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.05064264933268229,
      "learning_rate": 0.0001,
      "loss": 4.2589,
      "loss/crossentropy": 2.1680142879486084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1947874128818512,
      "step": 23020
    },
    {
      "epoch": 0.46044,
      "grad_norm": 1.875,
      "grad_norm_var": 0.05204976399739583,
      "learning_rate": 0.0001,
      "loss": 4.0885,
      "loss/crossentropy": 2.1033952236175537,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17692237347364426,
      "step": 23022
    },
    {
      "epoch": 0.46048,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0103912353515625,
      "learning_rate": 0.0001,
      "loss": 4.2974,
      "loss/crossentropy": 2.3113337755203247,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2220149040222168,
      "step": 23024
    },
    {
      "epoch": 0.46052,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.009291330973307291,
      "learning_rate": 0.0001,
      "loss": 3.76,
      "loss/crossentropy": 1.4061094522476196,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16114307194948196,
      "step": 23026
    },
    {
      "epoch": 0.46056,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.010811106363932291,
      "learning_rate": 0.0001,
      "loss": 3.9877,
      "loss/crossentropy": 2.14698326587677,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20559164136648178,
      "step": 23028
    },
    {
      "epoch": 0.4606,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.011787668863932291,
      "learning_rate": 0.0001,
      "loss": 3.9314,
      "loss/crossentropy": 2.255793571472168,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18896941095590591,
      "step": 23030
    },
    {
      "epoch": 0.46064,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.01043701171875,
      "learning_rate": 0.0001,
      "loss": 3.9739,
      "loss/crossentropy": 2.186546564102173,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2044816091656685,
      "step": 23032
    },
    {
      "epoch": 0.46068,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0108154296875,
      "learning_rate": 0.0001,
      "loss": 4.175,
      "loss/crossentropy": 2.285028338432312,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20862022787332535,
      "step": 23034
    },
    {
      "epoch": 0.46072,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0055908203125,
      "learning_rate": 0.0001,
      "loss": 4.3081,
      "loss/crossentropy": 2.1346707344055176,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2030143365263939,
      "step": 23036
    },
    {
      "epoch": 0.46076,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0063168843587239586,
      "learning_rate": 0.0001,
      "loss": 4.043,
      "loss/crossentropy": 1.8645261526107788,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1957538053393364,
      "step": 23038
    },
    {
      "epoch": 0.4608,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.006048329671223958,
      "learning_rate": 0.0001,
      "loss": 3.8239,
      "loss/crossentropy": 1.9670527577400208,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1886385902762413,
      "step": 23040
    },
    {
      "epoch": 0.46084,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.006644439697265625,
      "learning_rate": 0.0001,
      "loss": 3.9349,
      "loss/crossentropy": 2.1273213624954224,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1952996477484703,
      "step": 23042
    },
    {
      "epoch": 0.46088,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0060546875,
      "learning_rate": 0.0001,
      "loss": 3.947,
      "loss/crossentropy": 2.1550720930099487,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20368672162294388,
      "step": 23044
    },
    {
      "epoch": 0.46092,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.005830891927083333,
      "learning_rate": 0.0001,
      "loss": 4.054,
      "loss/crossentropy": 2.0871312618255615,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20684774219989777,
      "step": 23046
    },
    {
      "epoch": 0.46096,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.005980428059895833,
      "learning_rate": 0.0001,
      "loss": 3.9013,
      "loss/crossentropy": 2.1256298422813416,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18924856930971146,
      "step": 23048
    },
    {
      "epoch": 0.461,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.014422353108723958,
      "learning_rate": 0.0001,
      "loss": 4.3133,
      "loss/crossentropy": 1.9934388399124146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2362150102853775,
      "step": 23050
    },
    {
      "epoch": 0.46104,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.015478261311848958,
      "learning_rate": 0.0001,
      "loss": 4.0432,
      "loss/crossentropy": 1.9070496559143066,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.198696069419384,
      "step": 23052
    },
    {
      "epoch": 0.46108,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.016403961181640624,
      "learning_rate": 0.0001,
      "loss": 3.647,
      "loss/crossentropy": 2.3180062770843506,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19845713675022125,
      "step": 23054
    },
    {
      "epoch": 0.46112,
      "grad_norm": 2.34375,
      "grad_norm_var": 3.4797027587890623,
      "learning_rate": 0.0001,
      "loss": 3.9427,
      "loss/crossentropy": 1.5877657532691956,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17085347324609756,
      "step": 23056
    },
    {
      "epoch": 0.46116,
      "grad_norm": 2.4375,
      "grad_norm_var": 3.4481992085774738,
      "learning_rate": 0.0001,
      "loss": 4.1123,
      "loss/crossentropy": 1.9232125282287598,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18383993953466415,
      "step": 23058
    },
    {
      "epoch": 0.4612,
      "grad_norm": 1.9765625,
      "grad_norm_var": 3.442277018229167,
      "learning_rate": 0.0001,
      "loss": 4.0018,
      "loss/crossentropy": 1.7219146490097046,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18319600075483322,
      "step": 23060
    },
    {
      "epoch": 0.46124,
      "grad_norm": 2.0625,
      "grad_norm_var": 3.425248209635417,
      "learning_rate": 0.0001,
      "loss": 4.0123,
      "loss/crossentropy": 2.274755835533142,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20030079036951065,
      "step": 23062
    },
    {
      "epoch": 0.46128,
      "grad_norm": 2.015625,
      "grad_norm_var": 3.420116170247396,
      "learning_rate": 0.0001,
      "loss": 3.9746,
      "loss/crossentropy": 2.1552868485450745,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19475380331277847,
      "step": 23064
    },
    {
      "epoch": 0.46132,
      "grad_norm": 1.9140625,
      "grad_norm_var": 3.4470842997233073,
      "learning_rate": 0.0001,
      "loss": 3.7861,
      "loss/crossentropy": 1.7763307094573975,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16649360209703445,
      "step": 23066
    },
    {
      "epoch": 0.46136,
      "grad_norm": 2.0,
      "grad_norm_var": 3.4487945556640627,
      "learning_rate": 0.0001,
      "loss": 4.1083,
      "loss/crossentropy": 1.859923779964447,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17723160982131958,
      "step": 23068
    },
    {
      "epoch": 0.4614,
      "grad_norm": 1.9453125,
      "grad_norm_var": 3.4543690999348957,
      "learning_rate": 0.0001,
      "loss": 3.9948,
      "loss/crossentropy": 1.9055342078208923,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18139074742794037,
      "step": 23070
    },
    {
      "epoch": 0.46144,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.02443415323893229,
      "learning_rate": 0.0001,
      "loss": 4.1256,
      "loss/crossentropy": 2.2504982948303223,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21642805635929108,
      "step": 23072
    },
    {
      "epoch": 0.46148,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.007469685872395834,
      "learning_rate": 0.0001,
      "loss": 4.2309,
      "loss/crossentropy": 2.142970085144043,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19985180348157883,
      "step": 23074
    },
    {
      "epoch": 0.46152,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.0076253255208333336,
      "learning_rate": 0.0001,
      "loss": 3.836,
      "loss/crossentropy": 2.0464539527893066,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20277106016874313,
      "step": 23076
    },
    {
      "epoch": 0.46156,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.007741038004557292,
      "learning_rate": 0.0001,
      "loss": 3.9323,
      "loss/crossentropy": 1.9818042516708374,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2055400162935257,
      "step": 23078
    },
    {
      "epoch": 0.4616,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0063555399576822914,
      "learning_rate": 0.0001,
      "loss": 3.9854,
      "loss/crossentropy": 2.245171070098877,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21404191851615906,
      "step": 23080
    },
    {
      "epoch": 0.46164,
      "grad_norm": 2.0,
      "grad_norm_var": 0.009336090087890625,
      "learning_rate": 0.0001,
      "loss": 3.8036,
      "loss/crossentropy": 1.8196159601211548,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19405725598335266,
      "step": 23082
    },
    {
      "epoch": 0.46168,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0571197509765625,
      "learning_rate": 0.0001,
      "loss": 4.4681,
      "loss/crossentropy": 2.2700140476226807,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2196115106344223,
      "step": 23084
    },
    {
      "epoch": 0.46172,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.05880940755208333,
      "learning_rate": 0.0001,
      "loss": 3.7722,
      "loss/crossentropy": 1.8162717819213867,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19593419879674911,
      "step": 23086
    },
    {
      "epoch": 0.46176,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.057920074462890624,
      "learning_rate": 0.0001,
      "loss": 3.8489,
      "loss/crossentropy": 2.287248969078064,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2079424038529396,
      "step": 23088
    },
    {
      "epoch": 0.4618,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.05727310180664062,
      "learning_rate": 0.0001,
      "loss": 4.1084,
      "loss/crossentropy": 2.004701316356659,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19984427094459534,
      "step": 23090
    },
    {
      "epoch": 0.46184,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.05663655598958333,
      "learning_rate": 0.0001,
      "loss": 4.3167,
      "loss/crossentropy": 2.185683250427246,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.210247203707695,
      "step": 23092
    },
    {
      "epoch": 0.46188,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.055425771077473956,
      "learning_rate": 0.0001,
      "loss": 4.0199,
      "loss/crossentropy": 2.050000488758087,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19505897909402847,
      "step": 23094
    },
    {
      "epoch": 0.46192,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.056703440348307294,
      "learning_rate": 0.0001,
      "loss": 3.9138,
      "loss/crossentropy": 2.0964688062667847,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19070538878440857,
      "step": 23096
    },
    {
      "epoch": 0.46196,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.05205256144205729,
      "learning_rate": 0.0001,
      "loss": 4.1354,
      "loss/crossentropy": 2.02596253156662,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22631053626537323,
      "step": 23098
    },
    {
      "epoch": 0.462,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.004951985677083334,
      "learning_rate": 0.0001,
      "loss": 4.0223,
      "loss/crossentropy": 1.8759450912475586,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18205147981643677,
      "step": 23100
    },
    {
      "epoch": 0.46204,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.004231516520182292,
      "learning_rate": 0.0001,
      "loss": 3.8365,
      "loss/crossentropy": 1.9460791945457458,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17607779800891876,
      "step": 23102
    },
    {
      "epoch": 0.46208,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.011031087239583333,
      "learning_rate": 0.0001,
      "loss": 3.9159,
      "loss/crossentropy": 1.5419431328773499,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15973693877458572,
      "step": 23104
    },
    {
      "epoch": 0.46212,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.014019521077473958,
      "learning_rate": 0.0001,
      "loss": 4.2898,
      "loss/crossentropy": 2.13347589969635,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21076981723308563,
      "step": 23106
    },
    {
      "epoch": 0.46216,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.013948567708333333,
      "learning_rate": 0.0001,
      "loss": 4.2812,
      "loss/crossentropy": 2.199908971786499,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1933242455124855,
      "step": 23108
    },
    {
      "epoch": 0.4622,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.013425445556640625,
      "learning_rate": 0.0001,
      "loss": 3.8662,
      "loss/crossentropy": 2.0715879797935486,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1933552771806717,
      "step": 23110
    },
    {
      "epoch": 0.46224,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0129302978515625,
      "learning_rate": 0.0001,
      "loss": 3.9125,
      "loss/crossentropy": 1.8359448909759521,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17814192920923233,
      "step": 23112
    },
    {
      "epoch": 0.46228,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.014987945556640625,
      "learning_rate": 0.0001,
      "loss": 4.1782,
      "loss/crossentropy": 1.9806716442108154,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18513696640729904,
      "step": 23114
    },
    {
      "epoch": 0.46232,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.014725494384765624,
      "learning_rate": 0.0001,
      "loss": 3.9427,
      "loss/crossentropy": 1.647410809993744,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.177697092294693,
      "step": 23116
    },
    {
      "epoch": 0.46236,
      "grad_norm": 1.734375,
      "grad_norm_var": 0.01885986328125,
      "learning_rate": 0.0001,
      "loss": 3.7102,
      "loss/crossentropy": 2.05259370803833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18241985887289047,
      "step": 23118
    },
    {
      "epoch": 0.4624,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.013874308268229166,
      "learning_rate": 0.0001,
      "loss": 3.9863,
      "loss/crossentropy": 2.229410469532013,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21042390167713165,
      "step": 23120
    },
    {
      "epoch": 0.46244,
      "grad_norm": 1.875,
      "grad_norm_var": 0.010163370768229167,
      "learning_rate": 0.0001,
      "loss": 3.7858,
      "loss/crossentropy": 1.6477417945861816,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15472649037837982,
      "step": 23122
    },
    {
      "epoch": 0.46248,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.009089914957682292,
      "learning_rate": 0.0001,
      "loss": 4.1175,
      "loss/crossentropy": 2.0088080167770386,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18309780955314636,
      "step": 23124
    },
    {
      "epoch": 0.46252,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.009129842122395834,
      "learning_rate": 0.0001,
      "loss": 3.8097,
      "loss/crossentropy": 1.8413895964622498,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1828669160604477,
      "step": 23126
    },
    {
      "epoch": 0.46256,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.009565989176432291,
      "learning_rate": 0.0001,
      "loss": 3.8763,
      "loss/crossentropy": 2.0812554955482483,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1836593672633171,
      "step": 23128
    },
    {
      "epoch": 0.4626,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.006224568684895833,
      "learning_rate": 0.0001,
      "loss": 4.1576,
      "loss/crossentropy": 2.269462764263153,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22045638412237167,
      "step": 23130
    },
    {
      "epoch": 0.46264,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.008898671468098958,
      "learning_rate": 0.0001,
      "loss": 4.203,
      "loss/crossentropy": 2.1176013946533203,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19339244812726974,
      "step": 23132
    },
    {
      "epoch": 0.46268,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.008983357747395834,
      "learning_rate": 0.0001,
      "loss": 3.889,
      "loss/crossentropy": 1.8236181735992432,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18049708008766174,
      "step": 23134
    },
    {
      "epoch": 0.46272,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.007608795166015625,
      "learning_rate": 0.0001,
      "loss": 4.163,
      "loss/crossentropy": 1.7525643706321716,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17853398621082306,
      "step": 23136
    },
    {
      "epoch": 0.46276,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.006135050455729167,
      "learning_rate": 0.0001,
      "loss": 4.3239,
      "loss/crossentropy": 2.2958520650863647,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21427057683467865,
      "step": 23138
    },
    {
      "epoch": 0.4628,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.009016927083333333,
      "learning_rate": 0.0001,
      "loss": 3.6256,
      "loss/crossentropy": 1.8358682990074158,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18813011050224304,
      "step": 23140
    },
    {
      "epoch": 0.46284,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.010667928059895833,
      "learning_rate": 0.0001,
      "loss": 4.3444,
      "loss/crossentropy": 2.4191941022872925,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2317579984664917,
      "step": 23142
    },
    {
      "epoch": 0.46288,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.012452952067057292,
      "learning_rate": 0.0001,
      "loss": 3.9152,
      "loss/crossentropy": 2.2656983137130737,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20870012044906616,
      "step": 23144
    },
    {
      "epoch": 0.46292,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.014347076416015625,
      "learning_rate": 0.0001,
      "loss": 4.0893,
      "loss/crossentropy": 2.2222647666931152,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2072836235165596,
      "step": 23146
    },
    {
      "epoch": 0.46296,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.012410481770833334,
      "learning_rate": 0.0001,
      "loss": 4.0469,
      "loss/crossentropy": 2.001325011253357,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20486044138669968,
      "step": 23148
    },
    {
      "epoch": 0.463,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.013061269124348959,
      "learning_rate": 0.0001,
      "loss": 4.0521,
      "loss/crossentropy": 2.3565926551818848,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19294436275959015,
      "step": 23150
    },
    {
      "epoch": 0.46304,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0134033203125,
      "learning_rate": 0.0001,
      "loss": 3.9513,
      "loss/crossentropy": 1.9006416201591492,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20511633157730103,
      "step": 23152
    },
    {
      "epoch": 0.46308,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.015697224934895834,
      "learning_rate": 0.0001,
      "loss": 3.761,
      "loss/crossentropy": 2.017968237400055,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1877719983458519,
      "step": 23154
    },
    {
      "epoch": 0.46312,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.012833658854166667,
      "learning_rate": 0.0001,
      "loss": 4.0591,
      "loss/crossentropy": 2.093901038169861,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1863752081990242,
      "step": 23156
    },
    {
      "epoch": 0.46316,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011690266927083333,
      "learning_rate": 0.0001,
      "loss": 4.1145,
      "loss/crossentropy": 2.3631211519241333,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2350725531578064,
      "step": 23158
    },
    {
      "epoch": 0.4632,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.007892862955729166,
      "learning_rate": 0.0001,
      "loss": 4.1794,
      "loss/crossentropy": 2.2482752799987793,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18792778253555298,
      "step": 23160
    },
    {
      "epoch": 0.46324,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.004115549723307291,
      "learning_rate": 0.0001,
      "loss": 4.0843,
      "loss/crossentropy": 2.462800145149231,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21113361418247223,
      "step": 23162
    },
    {
      "epoch": 0.46328,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.003639475504557292,
      "learning_rate": 0.0001,
      "loss": 4.1277,
      "loss/crossentropy": 2.2522822618484497,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2163240611553192,
      "step": 23164
    },
    {
      "epoch": 0.46332,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.013444010416666667,
      "learning_rate": 0.0001,
      "loss": 3.9615,
      "loss/crossentropy": 2.0585132837295532,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20750011503696442,
      "step": 23166
    },
    {
      "epoch": 0.46336,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.018357086181640624,
      "learning_rate": 0.0001,
      "loss": 4.0761,
      "loss/crossentropy": 2.0086065530776978,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21273234486579895,
      "step": 23168
    },
    {
      "epoch": 0.4634,
      "grad_norm": 2.125,
      "grad_norm_var": 0.017040761311848958,
      "learning_rate": 0.0001,
      "loss": 3.7175,
      "loss/crossentropy": 1.8222439289093018,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.188787043094635,
      "step": 23170
    },
    {
      "epoch": 0.46344,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.015843709309895832,
      "learning_rate": 0.0001,
      "loss": 4.0286,
      "loss/crossentropy": 1.9914604425430298,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19186841696500778,
      "step": 23172
    },
    {
      "epoch": 0.46348,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.01594823201497396,
      "learning_rate": 0.0001,
      "loss": 4.022,
      "loss/crossentropy": 1.5708445310592651,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16981858015060425,
      "step": 23174
    },
    {
      "epoch": 0.46352,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.015900675455729166,
      "learning_rate": 0.0001,
      "loss": 3.981,
      "loss/crossentropy": 2.170333504676819,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20143602043390274,
      "step": 23176
    },
    {
      "epoch": 0.46356,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.016033681233723958,
      "learning_rate": 0.0001,
      "loss": 4.0689,
      "loss/crossentropy": 1.8889789581298828,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17962591350078583,
      "step": 23178
    },
    {
      "epoch": 0.4636,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.016917928059895834,
      "learning_rate": 0.0001,
      "loss": 3.877,
      "loss/crossentropy": 1.9203835129737854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20256394147872925,
      "step": 23180
    },
    {
      "epoch": 0.46364,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.009715779622395834,
      "learning_rate": 0.0001,
      "loss": 4.1076,
      "loss/crossentropy": 2.2325422763824463,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23236138373613358,
      "step": 23182
    },
    {
      "epoch": 0.46368,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.010133616129557292,
      "learning_rate": 0.0001,
      "loss": 3.6904,
      "loss/crossentropy": 1.5864351987838745,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15730705857276917,
      "step": 23184
    },
    {
      "epoch": 0.46372,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0185211181640625,
      "learning_rate": 0.0001,
      "loss": 4.4229,
      "loss/crossentropy": 2.5526922941207886,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22684484720230103,
      "step": 23186
    },
    {
      "epoch": 0.46376,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.01954930623372396,
      "learning_rate": 0.0001,
      "loss": 3.7298,
      "loss/crossentropy": 1.6752784252166748,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16830473393201828,
      "step": 23188
    },
    {
      "epoch": 0.4638,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.019877115885416668,
      "learning_rate": 0.0001,
      "loss": 4.0933,
      "loss/crossentropy": 1.992654800415039,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.198849655687809,
      "step": 23190
    },
    {
      "epoch": 0.46384,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.021541341145833334,
      "learning_rate": 0.0001,
      "loss": 4.1588,
      "loss/crossentropy": 1.932292878627777,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19009747356176376,
      "step": 23192
    },
    {
      "epoch": 0.46388,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.02211278279622396,
      "learning_rate": 0.0001,
      "loss": 4.2248,
      "loss/crossentropy": 2.362654209136963,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2087061107158661,
      "step": 23194
    },
    {
      "epoch": 0.46392,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.02215754191080729,
      "learning_rate": 0.0001,
      "loss": 3.932,
      "loss/crossentropy": 2.117887258529663,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19435501843690872,
      "step": 23196
    },
    {
      "epoch": 0.46396,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.0228912353515625,
      "learning_rate": 0.0001,
      "loss": 3.7937,
      "loss/crossentropy": 2.164400637149811,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19678650051355362,
      "step": 23198
    },
    {
      "epoch": 0.464,
      "grad_norm": 2.734375,
      "grad_norm_var": 0.05366795857747396,
      "learning_rate": 0.0001,
      "loss": 4.1249,
      "loss/crossentropy": 2.1662251949310303,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19994695484638214,
      "step": 23200
    },
    {
      "epoch": 0.46404,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.04720433553059896,
      "learning_rate": 0.0001,
      "loss": 4.246,
      "loss/crossentropy": 2.082236111164093,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19977488368749619,
      "step": 23202
    },
    {
      "epoch": 0.46408,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.04742838541666667,
      "learning_rate": 0.0001,
      "loss": 4.0368,
      "loss/crossentropy": 2.12766033411026,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18198081851005554,
      "step": 23204
    },
    {
      "epoch": 0.46412,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.047907511393229164,
      "learning_rate": 0.0001,
      "loss": 3.7709,
      "loss/crossentropy": 2.0322113633155823,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1872313991189003,
      "step": 23206
    },
    {
      "epoch": 0.46416,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.04896214803059896,
      "learning_rate": 0.0001,
      "loss": 4.0456,
      "loss/crossentropy": 2.339892268180847,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19740620255470276,
      "step": 23208
    },
    {
      "epoch": 0.4642,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.04987691243489583,
      "learning_rate": 0.0001,
      "loss": 3.7924,
      "loss/crossentropy": 1.989054560661316,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2098187357187271,
      "step": 23210
    },
    {
      "epoch": 0.46424,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.050065104166666666,
      "learning_rate": 0.0001,
      "loss": 3.8527,
      "loss/crossentropy": 2.0670509934425354,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19233432412147522,
      "step": 23212
    },
    {
      "epoch": 0.46428,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.04978815714518229,
      "learning_rate": 0.0001,
      "loss": 4.055,
      "loss/crossentropy": 2.136350452899933,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21299083530902863,
      "step": 23214
    },
    {
      "epoch": 0.46432,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.008333079020182292,
      "learning_rate": 0.0001,
      "loss": 4.2758,
      "loss/crossentropy": 2.1144350171089172,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19434542953968048,
      "step": 23216
    },
    {
      "epoch": 0.46436,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.008605702718098959,
      "learning_rate": 0.0001,
      "loss": 3.8981,
      "loss/crossentropy": 2.307699203491211,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19784536957740784,
      "step": 23218
    },
    {
      "epoch": 0.4644,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.00841064453125,
      "learning_rate": 0.0001,
      "loss": 3.7529,
      "loss/crossentropy": 1.8954175114631653,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18727902323007584,
      "step": 23220
    },
    {
      "epoch": 0.46444,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.011207834879557291,
      "learning_rate": 0.0001,
      "loss": 4.2463,
      "loss/crossentropy": 2.3020440340042114,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2126014232635498,
      "step": 23222
    },
    {
      "epoch": 0.46448,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010628255208333333,
      "learning_rate": 0.0001,
      "loss": 4.1424,
      "loss/crossentropy": 2.1159419417381287,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19804084300994873,
      "step": 23224
    },
    {
      "epoch": 0.46452,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.009422810872395833,
      "learning_rate": 0.0001,
      "loss": 3.7152,
      "loss/crossentropy": 1.7309923768043518,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19001224637031555,
      "step": 23226
    },
    {
      "epoch": 0.46456,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.009273274739583334,
      "learning_rate": 0.0001,
      "loss": 4.0433,
      "loss/crossentropy": 2.2450767755508423,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20024073868989944,
      "step": 23228
    },
    {
      "epoch": 0.4646,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0081695556640625,
      "learning_rate": 0.0001,
      "loss": 4.1005,
      "loss/crossentropy": 1.8837137818336487,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20254365354776382,
      "step": 23230
    },
    {
      "epoch": 0.46464,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.010825347900390626,
      "learning_rate": 0.0001,
      "loss": 3.9611,
      "loss/crossentropy": 1.773855447769165,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17011478543281555,
      "step": 23232
    },
    {
      "epoch": 0.46468,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.017479451497395833,
      "learning_rate": 0.0001,
      "loss": 4.1329,
      "loss/crossentropy": 2.2123981714248657,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2134469673037529,
      "step": 23234
    },
    {
      "epoch": 0.46472,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.015449778238932291,
      "learning_rate": 0.0001,
      "loss": 3.8216,
      "loss/crossentropy": 2.004405915737152,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19117921590805054,
      "step": 23236
    },
    {
      "epoch": 0.46476,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.01505126953125,
      "learning_rate": 0.0001,
      "loss": 4.012,
      "loss/crossentropy": 2.2839083671569824,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20231828093528748,
      "step": 23238
    },
    {
      "epoch": 0.4648,
      "grad_norm": 2.0,
      "grad_norm_var": 0.014769490559895833,
      "learning_rate": 0.0001,
      "loss": 3.9148,
      "loss/crossentropy": 2.196630358695984,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1992904543876648,
      "step": 23240
    },
    {
      "epoch": 0.46484,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.013981119791666666,
      "learning_rate": 0.0001,
      "loss": 4.0372,
      "loss/crossentropy": 1.907668113708496,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16895314306020737,
      "step": 23242
    },
    {
      "epoch": 0.46488,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.01436767578125,
      "learning_rate": 0.0001,
      "loss": 4.2587,
      "loss/crossentropy": 2.1902048587799072,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20193833857774734,
      "step": 23244
    },
    {
      "epoch": 0.46492,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.014427693684895833,
      "learning_rate": 0.0001,
      "loss": 4.084,
      "loss/crossentropy": 2.088558316230774,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21183707565069199,
      "step": 23246
    },
    {
      "epoch": 0.46496,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.018390909830729166,
      "learning_rate": 0.0001,
      "loss": 4.0602,
      "loss/crossentropy": 2.3056023120880127,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21808549761772156,
      "step": 23248
    },
    {
      "epoch": 0.465,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.009235636393229166,
      "learning_rate": 0.0001,
      "loss": 4.2253,
      "loss/crossentropy": 2.255687952041626,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19484181702136993,
      "step": 23250
    },
    {
      "epoch": 0.46504,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.009406534830729167,
      "learning_rate": 0.0001,
      "loss": 4.0991,
      "loss/crossentropy": 2.042704999446869,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20347727835178375,
      "step": 23252
    },
    {
      "epoch": 0.46508,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.009901682535807291,
      "learning_rate": 0.0001,
      "loss": 4.1159,
      "loss/crossentropy": 2.0365681648254395,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19337813556194305,
      "step": 23254
    },
    {
      "epoch": 0.46512,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.010707346598307292,
      "learning_rate": 0.0001,
      "loss": 4.0616,
      "loss/crossentropy": 2.1936534643173218,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2163967713713646,
      "step": 23256
    },
    {
      "epoch": 0.46516,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.01199951171875,
      "learning_rate": 0.0001,
      "loss": 3.9913,
      "loss/crossentropy": 2.116168260574341,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20038608461618423,
      "step": 23258
    },
    {
      "epoch": 0.4652,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.012654622395833334,
      "learning_rate": 0.0001,
      "loss": 4.0231,
      "loss/crossentropy": 2.182048201560974,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19984129816293716,
      "step": 23260
    },
    {
      "epoch": 0.46524,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.018418121337890624,
      "learning_rate": 0.0001,
      "loss": 3.953,
      "loss/crossentropy": 2.1267359256744385,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19032182544469833,
      "step": 23262
    },
    {
      "epoch": 0.46528,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.0109527587890625,
      "learning_rate": 0.0001,
      "loss": 3.8849,
      "loss/crossentropy": 1.763411819934845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1661328673362732,
      "step": 23264
    },
    {
      "epoch": 0.46532,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011279042561848958,
      "learning_rate": 0.0001,
      "loss": 4.1665,
      "loss/crossentropy": 1.908778965473175,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20274333655834198,
      "step": 23266
    },
    {
      "epoch": 0.46536,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.010994466145833333,
      "learning_rate": 0.0001,
      "loss": 4.0938,
      "loss/crossentropy": 2.2249021530151367,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18010813742876053,
      "step": 23268
    },
    {
      "epoch": 0.4654,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.013032786051432292,
      "learning_rate": 0.0001,
      "loss": 4.2546,
      "loss/crossentropy": 2.4366761445999146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22475503385066986,
      "step": 23270
    },
    {
      "epoch": 0.46544,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.013084920247395833,
      "learning_rate": 0.0001,
      "loss": 4.2797,
      "loss/crossentropy": 2.2445744276046753,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18397177010774612,
      "step": 23272
    },
    {
      "epoch": 0.46548,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0141021728515625,
      "learning_rate": 0.0001,
      "loss": 4.5025,
      "loss/crossentropy": 2.1187288761138916,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2064574956893921,
      "step": 23274
    },
    {
      "epoch": 0.46552,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0139892578125,
      "learning_rate": 0.0001,
      "loss": 4.0211,
      "loss/crossentropy": 1.9558793902397156,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1832912713289261,
      "step": 23276
    },
    {
      "epoch": 0.46556,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.009187571207682292,
      "learning_rate": 0.0001,
      "loss": 4.2,
      "loss/crossentropy": 2.125900387763977,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23090820014476776,
      "step": 23278
    },
    {
      "epoch": 0.4656,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.009860992431640625,
      "learning_rate": 0.0001,
      "loss": 4.146,
      "loss/crossentropy": 2.337615966796875,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21365046501159668,
      "step": 23280
    },
    {
      "epoch": 0.46564,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.011482747395833333,
      "learning_rate": 0.0001,
      "loss": 3.8832,
      "loss/crossentropy": 2.0574655532836914,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1987999752163887,
      "step": 23282
    },
    {
      "epoch": 0.46568,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.011250813802083334,
      "learning_rate": 0.0001,
      "loss": 3.8743,
      "loss/crossentropy": 1.8969943523406982,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18804477155208588,
      "step": 23284
    },
    {
      "epoch": 0.46572,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.010005442301432292,
      "learning_rate": 0.0001,
      "loss": 4.2682,
      "loss/crossentropy": 2.218023180961609,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2202347069978714,
      "step": 23286
    },
    {
      "epoch": 0.46576,
      "grad_norm": 2.0,
      "grad_norm_var": 0.010019683837890625,
      "learning_rate": 0.0001,
      "loss": 4.0489,
      "loss/crossentropy": 2.2892422676086426,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21481822431087494,
      "step": 23288
    },
    {
      "epoch": 0.4658,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0074859619140625,
      "learning_rate": 0.0001,
      "loss": 4.1356,
      "loss/crossentropy": 1.9342178106307983,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18514516204595566,
      "step": 23290
    },
    {
      "epoch": 0.46584,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.0075762430826822914,
      "learning_rate": 0.0001,
      "loss": 3.9764,
      "loss/crossentropy": 2.134042203426361,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20004402846097946,
      "step": 23292
    },
    {
      "epoch": 0.46588,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.006422678629557292,
      "learning_rate": 0.0001,
      "loss": 4.3574,
      "loss/crossentropy": 2.075138568878174,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18759854137897491,
      "step": 23294
    },
    {
      "epoch": 0.46592,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.005260976155598959,
      "learning_rate": 0.0001,
      "loss": 4.0339,
      "loss/crossentropy": 2.0511878728866577,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19307290762662888,
      "step": 23296
    },
    {
      "epoch": 0.46596,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.004484049479166667,
      "learning_rate": 0.0001,
      "loss": 4.31,
      "loss/crossentropy": 2.2446112632751465,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2207113802433014,
      "step": 23298
    },
    {
      "epoch": 0.466,
      "grad_norm": 1.875,
      "grad_norm_var": 0.0057769775390625,
      "learning_rate": 0.0001,
      "loss": 4.1328,
      "loss/crossentropy": 2.0350372195243835,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17964959889650345,
      "step": 23300
    },
    {
      "epoch": 0.46604,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.011844635009765625,
      "learning_rate": 0.0001,
      "loss": 4.4,
      "loss/crossentropy": 2.3412392139434814,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23192551732063293,
      "step": 23302
    },
    {
      "epoch": 0.46608,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.013456217447916667,
      "learning_rate": 0.0001,
      "loss": 4.1086,
      "loss/crossentropy": 2.1255269050598145,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22011695802211761,
      "step": 23304
    },
    {
      "epoch": 0.46612,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.013818105061848959,
      "learning_rate": 0.0001,
      "loss": 4.1335,
      "loss/crossentropy": 2.2372154593467712,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20802992582321167,
      "step": 23306
    },
    {
      "epoch": 0.46616,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.013051096598307292,
      "learning_rate": 0.0001,
      "loss": 3.9148,
      "loss/crossentropy": 1.766309678554535,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1731080263853073,
      "step": 23308
    },
    {
      "epoch": 0.4662,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.013354237874348958,
      "learning_rate": 0.0001,
      "loss": 4.2245,
      "loss/crossentropy": 2.035153806209564,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21127209067344666,
      "step": 23310
    },
    {
      "epoch": 0.46624,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.014525349934895833,
      "learning_rate": 0.0001,
      "loss": 3.8472,
      "loss/crossentropy": 2.161786377429962,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2059953212738037,
      "step": 23312
    },
    {
      "epoch": 0.46628,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.015195465087890625,
      "learning_rate": 0.0001,
      "loss": 4.0489,
      "loss/crossentropy": 2.2908066511154175,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20963146537542343,
      "step": 23314
    },
    {
      "epoch": 0.46632,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.015885162353515624,
      "learning_rate": 0.0001,
      "loss": 3.9638,
      "loss/crossentropy": 2.1423317193984985,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2051083892583847,
      "step": 23316
    },
    {
      "epoch": 0.46636,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.009303538004557292,
      "learning_rate": 0.0001,
      "loss": 3.9683,
      "loss/crossentropy": 1.953084647655487,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18355616927146912,
      "step": 23318
    },
    {
      "epoch": 0.4664,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.008348592122395833,
      "learning_rate": 0.0001,
      "loss": 4.4509,
      "loss/crossentropy": 2.3963130712509155,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21937239170074463,
      "step": 23320
    },
    {
      "epoch": 0.46644,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.009419759114583334,
      "learning_rate": 0.0001,
      "loss": 3.8771,
      "loss/crossentropy": 1.991708517074585,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18753455579280853,
      "step": 23322
    },
    {
      "epoch": 0.46648,
      "grad_norm": 2.125,
      "grad_norm_var": 0.013142903645833334,
      "learning_rate": 0.0001,
      "loss": 4.4012,
      "loss/crossentropy": 2.1105082035064697,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21250775456428528,
      "step": 23324
    },
    {
      "epoch": 0.46652,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.011885579427083333,
      "learning_rate": 0.0001,
      "loss": 4.2301,
      "loss/crossentropy": 2.3817760944366455,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22308632731437683,
      "step": 23326
    },
    {
      "epoch": 0.46656,
      "grad_norm": 2.0,
      "grad_norm_var": 0.010762532552083334,
      "learning_rate": 0.0001,
      "loss": 3.9596,
      "loss/crossentropy": 1.9623343348503113,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.189848855137825,
      "step": 23328
    },
    {
      "epoch": 0.4666,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.011156209309895833,
      "learning_rate": 0.0001,
      "loss": 4.0722,
      "loss/crossentropy": 2.2103809118270874,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21060046553611755,
      "step": 23330
    },
    {
      "epoch": 0.46664,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0101715087890625,
      "learning_rate": 0.0001,
      "loss": 3.915,
      "loss/crossentropy": 2.0628533959388733,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19960719347000122,
      "step": 23332
    },
    {
      "epoch": 0.46668,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.01181640625,
      "learning_rate": 0.0001,
      "loss": 3.9823,
      "loss/crossentropy": 1.8410035371780396,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17858797311782837,
      "step": 23334
    },
    {
      "epoch": 0.46672,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.010786946614583333,
      "learning_rate": 0.0001,
      "loss": 4.0117,
      "loss/crossentropy": 2.056548833847046,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1881648674607277,
      "step": 23336
    },
    {
      "epoch": 0.46676,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.012962849934895833,
      "learning_rate": 0.0001,
      "loss": 3.8496,
      "loss/crossentropy": 1.8516274094581604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19123712182044983,
      "step": 23338
    },
    {
      "epoch": 0.4668,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.009137980143229167,
      "learning_rate": 0.0001,
      "loss": 4.1999,
      "loss/crossentropy": 2.1613941192626953,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1937795951962471,
      "step": 23340
    },
    {
      "epoch": 0.46684,
      "grad_norm": 2.0,
      "grad_norm_var": 0.010106404622395834,
      "learning_rate": 0.0001,
      "loss": 4.052,
      "loss/crossentropy": 1.9474772810935974,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17672593146562576,
      "step": 23342
    },
    {
      "epoch": 0.46688,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.01136474609375,
      "learning_rate": 0.0001,
      "loss": 3.9052,
      "loss/crossentropy": 1.8008830547332764,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1621333733201027,
      "step": 23344
    },
    {
      "epoch": 0.46692,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.013768513997395834,
      "learning_rate": 0.0001,
      "loss": 4.385,
      "loss/crossentropy": 2.505362868309021,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21897228807210922,
      "step": 23346
    },
    {
      "epoch": 0.46696,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.013702138264973959,
      "learning_rate": 0.0001,
      "loss": 3.9262,
      "loss/crossentropy": 1.7621804475784302,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18244817852973938,
      "step": 23348
    },
    {
      "epoch": 0.467,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.012630208333333334,
      "learning_rate": 0.0001,
      "loss": 4.0277,
      "loss/crossentropy": 1.8438073992729187,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19016395509243011,
      "step": 23350
    },
    {
      "epoch": 0.46704,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.012630208333333334,
      "learning_rate": 0.0001,
      "loss": 3.7829,
      "loss/crossentropy": 1.9118491411209106,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19326411187648773,
      "step": 23352
    },
    {
      "epoch": 0.46708,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.009806315104166666,
      "learning_rate": 0.0001,
      "loss": 3.9878,
      "loss/crossentropy": 1.911489188671112,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17859935760498047,
      "step": 23354
    },
    {
      "epoch": 0.46712,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.014235178629557291,
      "learning_rate": 0.0001,
      "loss": 4.1033,
      "loss/crossentropy": 2.107007384300232,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20303775370121002,
      "step": 23356
    },
    {
      "epoch": 0.46716,
      "grad_norm": 1.671875,
      "grad_norm_var": 0.017659505208333332,
      "learning_rate": 0.0001,
      "loss": 3.7504,
      "loss/crossentropy": 2.242035746574402,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20925073325634003,
      "step": 23358
    },
    {
      "epoch": 0.4672,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.016364542643229167,
      "learning_rate": 0.0001,
      "loss": 3.9543,
      "loss/crossentropy": 1.9015487432479858,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20830972492694855,
      "step": 23360
    },
    {
      "epoch": 0.46724,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.013916015625,
      "learning_rate": 0.0001,
      "loss": 3.8229,
      "loss/crossentropy": 1.8648836612701416,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17992950975894928,
      "step": 23362
    },
    {
      "epoch": 0.46728,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.012984212239583333,
      "learning_rate": 0.0001,
      "loss": 4.026,
      "loss/crossentropy": 1.7606803178787231,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17884241789579391,
      "step": 23364
    },
    {
      "epoch": 0.46732,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.016635894775390625,
      "learning_rate": 0.0001,
      "loss": 3.9579,
      "loss/crossentropy": 2.115469813346863,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2030588760972023,
      "step": 23366
    },
    {
      "epoch": 0.46736,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.017814127604166667,
      "learning_rate": 0.0001,
      "loss": 3.9706,
      "loss/crossentropy": 1.9341481924057007,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17974573373794556,
      "step": 23368
    },
    {
      "epoch": 0.4674,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.017317454020182293,
      "learning_rate": 0.0001,
      "loss": 4.0494,
      "loss/crossentropy": 2.171218752861023,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20550543069839478,
      "step": 23370
    },
    {
      "epoch": 0.46744,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.013392893473307292,
      "learning_rate": 0.0001,
      "loss": 4.0207,
      "loss/crossentropy": 2.0552384853363037,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19348255544900894,
      "step": 23372
    },
    {
      "epoch": 0.46748,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.010643513997395833,
      "learning_rate": 0.0001,
      "loss": 3.6074,
      "loss/crossentropy": 1.7817274332046509,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16803114861249924,
      "step": 23374
    },
    {
      "epoch": 0.46752,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010245513916015626,
      "learning_rate": 0.0001,
      "loss": 4.2458,
      "loss/crossentropy": 2.0412665009498596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20734214782714844,
      "step": 23376
    },
    {
      "epoch": 0.46756,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.012520090738932291,
      "learning_rate": 0.0001,
      "loss": 4.0561,
      "loss/crossentropy": 2.0934815406799316,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2062682881951332,
      "step": 23378
    },
    {
      "epoch": 0.4676,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.015720367431640625,
      "learning_rate": 0.0001,
      "loss": 3.9964,
      "loss/crossentropy": 2.076676905155182,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20668631047010422,
      "step": 23380
    },
    {
      "epoch": 0.46764,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0129302978515625,
      "learning_rate": 0.0001,
      "loss": 4.2509,
      "loss/crossentropy": 2.185506582260132,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21427924185991287,
      "step": 23382
    },
    {
      "epoch": 0.46768,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.011771392822265626,
      "learning_rate": 0.0001,
      "loss": 3.865,
      "loss/crossentropy": 2.0467058420181274,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19394882023334503,
      "step": 23384
    },
    {
      "epoch": 0.46772,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.013304646809895833,
      "learning_rate": 0.0001,
      "loss": 4.2221,
      "loss/crossentropy": 2.4391517639160156,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22828804701566696,
      "step": 23386
    },
    {
      "epoch": 0.46776,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0128814697265625,
      "learning_rate": 0.0001,
      "loss": 4.1389,
      "loss/crossentropy": 2.2490856647491455,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21296796947717667,
      "step": 23388
    },
    {
      "epoch": 0.4678,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.011810048421223959,
      "learning_rate": 0.0001,
      "loss": 3.7515,
      "loss/crossentropy": 1.5810586214065552,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17569316178560257,
      "step": 23390
    },
    {
      "epoch": 0.46784,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.011805979410807292,
      "learning_rate": 0.0001,
      "loss": 3.9116,
      "loss/crossentropy": 2.069303274154663,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18829242885112762,
      "step": 23392
    },
    {
      "epoch": 0.46788,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.011004384358723958,
      "learning_rate": 0.0001,
      "loss": 4.1711,
      "loss/crossentropy": 1.9410834312438965,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1875254362821579,
      "step": 23394
    },
    {
      "epoch": 0.46792,
      "grad_norm": 1.7421875,
      "grad_norm_var": 0.011498769124348959,
      "learning_rate": 0.0001,
      "loss": 3.697,
      "loss/crossentropy": 2.1052953004837036,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19472668319940567,
      "step": 23396
    },
    {
      "epoch": 0.46796,
      "grad_norm": 1.875,
      "grad_norm_var": 0.011250813802083334,
      "learning_rate": 0.0001,
      "loss": 4.0819,
      "loss/crossentropy": 2.1837843656539917,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20570768415927887,
      "step": 23398
    },
    {
      "epoch": 0.468,
      "grad_norm": 1.71875,
      "grad_norm_var": 0.014229075113932291,
      "learning_rate": 0.0001,
      "loss": 3.7475,
      "loss/crossentropy": 1.980469524860382,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19470104575157166,
      "step": 23400
    },
    {
      "epoch": 0.46804,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.011472320556640625,
      "learning_rate": 0.0001,
      "loss": 3.9931,
      "loss/crossentropy": 1.992879033088684,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19029226154088974,
      "step": 23402
    },
    {
      "epoch": 0.46808,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.010689036051432291,
      "learning_rate": 0.0001,
      "loss": 4.085,
      "loss/crossentropy": 2.5265753269195557,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20238864421844482,
      "step": 23404
    },
    {
      "epoch": 0.46812,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.008943430582682292,
      "learning_rate": 0.0001,
      "loss": 3.9541,
      "loss/crossentropy": 2.1182003021240234,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22061780095100403,
      "step": 23406
    },
    {
      "epoch": 0.46816,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.008381144205729166,
      "learning_rate": 0.0001,
      "loss": 3.9054,
      "loss/crossentropy": 1.8726989030838013,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18806783854961395,
      "step": 23408
    },
    {
      "epoch": 0.4682,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.026371256510416666,
      "learning_rate": 0.0001,
      "loss": 4.0764,
      "loss/crossentropy": 2.3275599479675293,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21061742305755615,
      "step": 23410
    },
    {
      "epoch": 0.46824,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.02622044881184896,
      "learning_rate": 0.0001,
      "loss": 3.8391,
      "loss/crossentropy": 1.9331459999084473,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18973465263843536,
      "step": 23412
    },
    {
      "epoch": 0.46828,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.028043365478515624,
      "learning_rate": 0.0001,
      "loss": 3.7524,
      "loss/crossentropy": 1.9546124339103699,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17560599744319916,
      "step": 23414
    },
    {
      "epoch": 0.46832,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.03088353474934896,
      "learning_rate": 0.0001,
      "loss": 4.2547,
      "loss/crossentropy": 2.5048669576644897,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21813815087080002,
      "step": 23416
    },
    {
      "epoch": 0.46836,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.029930623372395833,
      "learning_rate": 0.0001,
      "loss": 4.0534,
      "loss/crossentropy": 1.8067769408226013,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18293478339910507,
      "step": 23418
    },
    {
      "epoch": 0.4684,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.031981404622395834,
      "learning_rate": 0.0001,
      "loss": 3.7527,
      "loss/crossentropy": 1.6894381642341614,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17101695388555527,
      "step": 23420
    },
    {
      "epoch": 0.46844,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.03349177042643229,
      "learning_rate": 0.0001,
      "loss": 3.6362,
      "loss/crossentropy": 2.3269423246383667,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20582617074251175,
      "step": 23422
    },
    {
      "epoch": 0.46848,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.03337376912434896,
      "learning_rate": 0.0001,
      "loss": 3.9875,
      "loss/crossentropy": 1.9424707293510437,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18869344890117645,
      "step": 23424
    },
    {
      "epoch": 0.46852,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.01929499308268229,
      "learning_rate": 0.0001,
      "loss": 3.927,
      "loss/crossentropy": 1.8979859948158264,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17844338715076447,
      "step": 23426
    },
    {
      "epoch": 0.46856,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0167236328125,
      "learning_rate": 0.0001,
      "loss": 4.3855,
      "loss/crossentropy": 2.2464778423309326,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18301833420991898,
      "step": 23428
    },
    {
      "epoch": 0.4686,
      "grad_norm": 1.7109375,
      "grad_norm_var": 0.02848078409830729,
      "learning_rate": 0.0001,
      "loss": 3.8957,
      "loss/crossentropy": 1.9471614360809326,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20326492190361023,
      "step": 23430
    },
    {
      "epoch": 0.46864,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.02283299763997396,
      "learning_rate": 0.0001,
      "loss": 4.0045,
      "loss/crossentropy": 2.04762601852417,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19779526442289352,
      "step": 23432
    },
    {
      "epoch": 0.46868,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.022607167561848957,
      "learning_rate": 0.0001,
      "loss": 4.1019,
      "loss/crossentropy": 2.1162944436073303,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19175396859645844,
      "step": 23434
    },
    {
      "epoch": 0.46872,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.025567372639973957,
      "learning_rate": 0.0001,
      "loss": 3.9651,
      "loss/crossentropy": 1.6130013465881348,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16156645864248276,
      "step": 23436
    },
    {
      "epoch": 0.46876,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.023423004150390624,
      "learning_rate": 0.0001,
      "loss": 3.9582,
      "loss/crossentropy": 2.0393518805503845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20742176473140717,
      "step": 23438
    },
    {
      "epoch": 0.4688,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.023811848958333333,
      "learning_rate": 0.0001,
      "loss": 3.7042,
      "loss/crossentropy": 1.6487592458724976,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16911844164133072,
      "step": 23440
    },
    {
      "epoch": 0.46884,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.023034413655598957,
      "learning_rate": 0.0001,
      "loss": 3.8824,
      "loss/crossentropy": 1.8171139359474182,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18819218128919601,
      "step": 23442
    },
    {
      "epoch": 0.46888,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.022981770833333335,
      "learning_rate": 0.0001,
      "loss": 4.0849,
      "loss/crossentropy": 1.669542133808136,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18253905326128006,
      "step": 23444
    },
    {
      "epoch": 0.46892,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.008845011393229166,
      "learning_rate": 0.0001,
      "loss": 3.9415,
      "loss/crossentropy": 1.8465211391448975,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.194318987429142,
      "step": 23446
    },
    {
      "epoch": 0.46896,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.011742146809895833,
      "learning_rate": 0.0001,
      "loss": 4.0606,
      "loss/crossentropy": 2.4016542434692383,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2081214338541031,
      "step": 23448
    },
    {
      "epoch": 0.469,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.013692220052083334,
      "learning_rate": 0.0001,
      "loss": 4.062,
      "loss/crossentropy": 2.2425581216812134,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19481270760297775,
      "step": 23450
    },
    {
      "epoch": 0.46904,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.009590657552083333,
      "learning_rate": 0.0001,
      "loss": 3.9642,
      "loss/crossentropy": 1.8975054621696472,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20339980721473694,
      "step": 23452
    },
    {
      "epoch": 0.46908,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.01871312459309896,
      "learning_rate": 0.0001,
      "loss": 4.1964,
      "loss/crossentropy": 2.274298667907715,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22013512253761292,
      "step": 23454
    },
    {
      "epoch": 0.46912,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.019724273681640626,
      "learning_rate": 0.0001,
      "loss": 3.7991,
      "loss/crossentropy": 2.286523461341858,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21875084191560745,
      "step": 23456
    },
    {
      "epoch": 0.46916,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.017465972900390626,
      "learning_rate": 0.0001,
      "loss": 4.1091,
      "loss/crossentropy": 2.199851155281067,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20098759233951569,
      "step": 23458
    },
    {
      "epoch": 0.4692,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.025780232747395833,
      "learning_rate": 0.0001,
      "loss": 3.76,
      "loss/crossentropy": 1.896471917629242,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18724198639392853,
      "step": 23460
    },
    {
      "epoch": 0.46924,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.026879628499348957,
      "learning_rate": 0.0001,
      "loss": 4.0191,
      "loss/crossentropy": 2.0206944942474365,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1795445680618286,
      "step": 23462
    },
    {
      "epoch": 0.46928,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.025911458333333335,
      "learning_rate": 0.0001,
      "loss": 4.1516,
      "loss/crossentropy": 2.230253279209137,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20442913472652435,
      "step": 23464
    },
    {
      "epoch": 0.46932,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.025055948893229166,
      "learning_rate": 0.0001,
      "loss": 4.0306,
      "loss/crossentropy": 2.117598056793213,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20652078092098236,
      "step": 23466
    },
    {
      "epoch": 0.46936,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0245758056640625,
      "learning_rate": 0.0001,
      "loss": 4.0638,
      "loss/crossentropy": 1.6677707433700562,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15297261625528336,
      "step": 23468
    },
    {
      "epoch": 0.4694,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.014875284830729167,
      "learning_rate": 0.0001,
      "loss": 4.0632,
      "loss/crossentropy": 2.039306938648224,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21778366714715958,
      "step": 23470
    },
    {
      "epoch": 0.46944,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.013923136393229167,
      "learning_rate": 0.0001,
      "loss": 3.9302,
      "loss/crossentropy": 1.8116675019264221,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19272665679454803,
      "step": 23472
    },
    {
      "epoch": 0.46948,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.015433502197265626,
      "learning_rate": 0.0001,
      "loss": 4.0605,
      "loss/crossentropy": 2.378050446510315,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21410302817821503,
      "step": 23474
    },
    {
      "epoch": 0.46952,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.005890909830729167,
      "learning_rate": 0.0001,
      "loss": 4.1681,
      "loss/crossentropy": 2.2755852937698364,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20388127863407135,
      "step": 23476
    },
    {
      "epoch": 0.46956,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.005692545572916667,
      "learning_rate": 0.0001,
      "loss": 4.0909,
      "loss/crossentropy": 2.1453245282173157,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20442672073841095,
      "step": 23478
    },
    {
      "epoch": 0.4696,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.003543853759765625,
      "learning_rate": 0.0001,
      "loss": 3.7373,
      "loss/crossentropy": 2.0084391236305237,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18768452107906342,
      "step": 23480
    },
    {
      "epoch": 0.46964,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.002872467041015625,
      "learning_rate": 0.0001,
      "loss": 3.9004,
      "loss/crossentropy": 1.9836713075637817,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1842040792107582,
      "step": 23482
    },
    {
      "epoch": 0.46968,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.004923502604166667,
      "learning_rate": 0.0001,
      "loss": 3.8047,
      "loss/crossentropy": 1.8174407482147217,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1779571697115898,
      "step": 23484
    },
    {
      "epoch": 0.46972,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.010835774739583333,
      "learning_rate": 0.0001,
      "loss": 3.9017,
      "loss/crossentropy": 1.8971520066261292,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19008546322584152,
      "step": 23486
    },
    {
      "epoch": 0.46976,
      "grad_norm": 2.0,
      "grad_norm_var": 0.011356608072916666,
      "learning_rate": 0.0001,
      "loss": 3.7841,
      "loss/crossentropy": 1.9949323534965515,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1878834068775177,
      "step": 23488
    },
    {
      "epoch": 0.4698,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.010212198893229166,
      "learning_rate": 0.0001,
      "loss": 4.0221,
      "loss/crossentropy": 1.7825579047203064,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1911696270108223,
      "step": 23490
    },
    {
      "epoch": 0.46984,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.009422810872395833,
      "learning_rate": 0.0001,
      "loss": 4.0892,
      "loss/crossentropy": 2.305335283279419,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20759187638759613,
      "step": 23492
    },
    {
      "epoch": 0.46988,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.2226715087890625,
      "learning_rate": 0.0001,
      "loss": 3.894,
      "loss/crossentropy": 2.0296601057052612,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1965966895222664,
      "step": 23494
    },
    {
      "epoch": 0.46992,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.22094319661458334,
      "learning_rate": 0.0001,
      "loss": 4.2948,
      "loss/crossentropy": 2.5238767862319946,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21435904502868652,
      "step": 23496
    },
    {
      "epoch": 0.46996,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.21852213541666668,
      "learning_rate": 0.0001,
      "loss": 4.1532,
      "loss/crossentropy": 2.2085973024368286,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1983572095632553,
      "step": 23498
    },
    {
      "epoch": 0.47,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.21304423014322918,
      "learning_rate": 0.0001,
      "loss": 4.2002,
      "loss/crossentropy": 2.257534384727478,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23111815005540848,
      "step": 23500
    },
    {
      "epoch": 0.47004,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.21055094401041666,
      "learning_rate": 0.0001,
      "loss": 4.2338,
      "loss/crossentropy": 2.3113789558410645,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2043687105178833,
      "step": 23502
    },
    {
      "epoch": 0.47008,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.2143267313639323,
      "learning_rate": 0.0001,
      "loss": 4.0727,
      "loss/crossentropy": 2.2028011083602905,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18799518048763275,
      "step": 23504
    },
    {
      "epoch": 0.47012,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.21318257649739583,
      "learning_rate": 0.0001,
      "loss": 4.3289,
      "loss/crossentropy": 2.1186457872390747,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2014438509941101,
      "step": 23506
    },
    {
      "epoch": 0.47016,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.21224136352539064,
      "learning_rate": 0.0001,
      "loss": 3.9821,
      "loss/crossentropy": 2.0399728417396545,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18888195604085922,
      "step": 23508
    },
    {
      "epoch": 0.4702,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.011350250244140625,
      "learning_rate": 0.0001,
      "loss": 3.9898,
      "loss/crossentropy": 1.943885326385498,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20364457368850708,
      "step": 23510
    },
    {
      "epoch": 0.47024,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.011812082926432292,
      "learning_rate": 0.0001,
      "loss": 4.0646,
      "loss/crossentropy": 1.9351946115493774,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18621300905942917,
      "step": 23512
    },
    {
      "epoch": 0.47028,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.014351399739583333,
      "learning_rate": 0.0001,
      "loss": 3.873,
      "loss/crossentropy": 2.0112447142601013,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1884174421429634,
      "step": 23514
    },
    {
      "epoch": 0.47032,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.014227040608723958,
      "learning_rate": 0.0001,
      "loss": 3.9918,
      "loss/crossentropy": 2.164724826812744,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21057604998350143,
      "step": 23516
    },
    {
      "epoch": 0.47036,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0138092041015625,
      "learning_rate": 0.0001,
      "loss": 4.2016,
      "loss/crossentropy": 2.198709011077881,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20399244129657745,
      "step": 23518
    },
    {
      "epoch": 0.4704,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.014792633056640626,
      "learning_rate": 0.0001,
      "loss": 3.8945,
      "loss/crossentropy": 2.3152371644973755,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19689901918172836,
      "step": 23520
    },
    {
      "epoch": 0.47044,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.015242258707682291,
      "learning_rate": 0.0001,
      "loss": 3.9598,
      "loss/crossentropy": 1.8530511260032654,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19973145425319672,
      "step": 23522
    },
    {
      "epoch": 0.47048,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.010894521077473959,
      "learning_rate": 0.0001,
      "loss": 3.9725,
      "loss/crossentropy": 1.6506844758987427,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17004889249801636,
      "step": 23524
    },
    {
      "epoch": 0.47052,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.00750732421875,
      "learning_rate": 0.0001,
      "loss": 4.0036,
      "loss/crossentropy": 1.7448241710662842,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20921830832958221,
      "step": 23526
    },
    {
      "epoch": 0.47056,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0036944071451822918,
      "learning_rate": 0.0001,
      "loss": 3.7118,
      "loss/crossentropy": 1.7023364305496216,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18442079424858093,
      "step": 23528
    },
    {
      "epoch": 0.4706,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.005909983317057292,
      "learning_rate": 0.0001,
      "loss": 3.9572,
      "loss/crossentropy": 2.1476879119873047,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.205117367208004,
      "step": 23530
    },
    {
      "epoch": 0.47064,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.006154123942057292,
      "learning_rate": 0.0001,
      "loss": 4.1135,
      "loss/crossentropy": 1.9776412844657898,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1900978982448578,
      "step": 23532
    },
    {
      "epoch": 0.47068,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.009437815348307291,
      "learning_rate": 0.0001,
      "loss": 4.2694,
      "loss/crossentropy": 2.6421544551849365,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21158932149410248,
      "step": 23534
    },
    {
      "epoch": 0.47072,
      "grad_norm": 2.375,
      "grad_norm_var": 0.020807902018229168,
      "learning_rate": 0.0001,
      "loss": 4.4728,
      "loss/crossentropy": 2.3335670232772827,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25099293887615204,
      "step": 23536
    },
    {
      "epoch": 0.47076,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.02069091796875,
      "learning_rate": 0.0001,
      "loss": 4.2101,
      "loss/crossentropy": 2.309471607208252,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20084770023822784,
      "step": 23538
    },
    {
      "epoch": 0.4708,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.020799763997395835,
      "learning_rate": 0.0001,
      "loss": 3.6632,
      "loss/crossentropy": 1.794599175453186,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17749344557523727,
      "step": 23540
    },
    {
      "epoch": 0.47084,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.022440338134765626,
      "learning_rate": 0.0001,
      "loss": 4.3814,
      "loss/crossentropy": 2.0325594544410706,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19963493198156357,
      "step": 23542
    },
    {
      "epoch": 0.47088,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.018863677978515625,
      "learning_rate": 0.0001,
      "loss": 4.0274,
      "loss/crossentropy": 2.0868377089500427,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20961833745241165,
      "step": 23544
    },
    {
      "epoch": 0.47092,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.02393366495768229,
      "learning_rate": 0.0001,
      "loss": 3.7205,
      "loss/crossentropy": 1.9807188510894775,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1892232447862625,
      "step": 23546
    },
    {
      "epoch": 0.47096,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.023872884114583333,
      "learning_rate": 0.0001,
      "loss": 4.2712,
      "loss/crossentropy": 2.292481303215027,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22137918323278427,
      "step": 23548
    },
    {
      "epoch": 0.471,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.022240193684895833,
      "learning_rate": 0.0001,
      "loss": 4.0832,
      "loss/crossentropy": 2.240318775177002,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2084667906165123,
      "step": 23550
    },
    {
      "epoch": 0.47104,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.019969685872395834,
      "learning_rate": 0.0001,
      "loss": 3.8889,
      "loss/crossentropy": 2.0380293130874634,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2129639983177185,
      "step": 23552
    },
    {
      "epoch": 0.47108,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.01995849609375,
      "learning_rate": 0.0001,
      "loss": 3.9957,
      "loss/crossentropy": 2.296157479286194,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.214280404150486,
      "step": 23554
    },
    {
      "epoch": 0.47112,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.022655995686848958,
      "learning_rate": 0.0001,
      "loss": 4.0575,
      "loss/crossentropy": 2.308936357498169,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20904332399368286,
      "step": 23556
    },
    {
      "epoch": 0.47116,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.01942723592122396,
      "learning_rate": 0.0001,
      "loss": 3.8463,
      "loss/crossentropy": 1.935391128063202,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19526035338640213,
      "step": 23558
    },
    {
      "epoch": 0.4712,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.019022369384765626,
      "learning_rate": 0.0001,
      "loss": 4.0567,
      "loss/crossentropy": 2.21670663356781,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21146119385957718,
      "step": 23560
    },
    {
      "epoch": 0.47124,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.01649958292643229,
      "learning_rate": 0.0001,
      "loss": 3.808,
      "loss/crossentropy": 1.9948694705963135,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18240909278392792,
      "step": 23562
    },
    {
      "epoch": 0.47128,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0165283203125,
      "learning_rate": 0.0001,
      "loss": 4.0053,
      "loss/crossentropy": 2.0435580015182495,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20460917055606842,
      "step": 23564
    },
    {
      "epoch": 0.47132,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.018381500244140626,
      "learning_rate": 0.0001,
      "loss": 3.4991,
      "loss/crossentropy": 2.0021498799324036,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17788298428058624,
      "step": 23566
    },
    {
      "epoch": 0.47136,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008845774332682292,
      "learning_rate": 0.0001,
      "loss": 3.9934,
      "loss/crossentropy": 2.0119062066078186,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1800990328192711,
      "step": 23568
    },
    {
      "epoch": 0.4714,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.008967081705729166,
      "learning_rate": 0.0001,
      "loss": 3.8167,
      "loss/crossentropy": 1.5521536469459534,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15504960715770721,
      "step": 23570
    },
    {
      "epoch": 0.47144,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.004400380452473958,
      "learning_rate": 0.0001,
      "loss": 3.954,
      "loss/crossentropy": 1.9227730631828308,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20652085542678833,
      "step": 23572
    },
    {
      "epoch": 0.47148,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.006078084309895833,
      "learning_rate": 0.0001,
      "loss": 4.075,
      "loss/crossentropy": 2.0273211002349854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18889201432466507,
      "step": 23574
    },
    {
      "epoch": 0.47152,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.006788889567057292,
      "learning_rate": 0.0001,
      "loss": 3.8103,
      "loss/crossentropy": 1.934963881969452,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17783871293067932,
      "step": 23576
    },
    {
      "epoch": 0.47156,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.007020823160807292,
      "learning_rate": 0.0001,
      "loss": 3.8893,
      "loss/crossentropy": 1.9819161295890808,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19646496325731277,
      "step": 23578
    },
    {
      "epoch": 0.4716,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.014387003580729167,
      "learning_rate": 0.0001,
      "loss": 4.1545,
      "loss/crossentropy": 1.9830252528190613,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19861895591020584,
      "step": 23580
    },
    {
      "epoch": 0.47164,
      "grad_norm": 2.125,
      "grad_norm_var": 0.011872355143229167,
      "learning_rate": 0.0001,
      "loss": 4.1332,
      "loss/crossentropy": 1.9679479598999023,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19719085842370987,
      "step": 23582
    },
    {
      "epoch": 0.47168,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.0126708984375,
      "learning_rate": 0.0001,
      "loss": 3.9055,
      "loss/crossentropy": 1.8915583491325378,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1777575984597206,
      "step": 23584
    },
    {
      "epoch": 0.47172,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.011533355712890625,
      "learning_rate": 0.0001,
      "loss": 4.0302,
      "loss/crossentropy": 2.126678943634033,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21122275292873383,
      "step": 23586
    },
    {
      "epoch": 0.47176,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.012109375,
      "learning_rate": 0.0001,
      "loss": 3.8424,
      "loss/crossentropy": 1.8739299774169922,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19388508796691895,
      "step": 23588
    },
    {
      "epoch": 0.4718,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.01207275390625,
      "learning_rate": 0.0001,
      "loss": 4.057,
      "loss/crossentropy": 1.983488917350769,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18411333858966827,
      "step": 23590
    },
    {
      "epoch": 0.47184,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.016796875,
      "learning_rate": 0.0001,
      "loss": 4.4929,
      "loss/crossentropy": 2.127811551094055,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1860193982720375,
      "step": 23592
    },
    {
      "epoch": 0.47188,
      "grad_norm": 2.3125,
      "grad_norm_var": 0.022696940104166667,
      "learning_rate": 0.0001,
      "loss": 4.0036,
      "loss/crossentropy": 1.8028001189231873,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18699438124895096,
      "step": 23594
    },
    {
      "epoch": 0.47192,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.02093683878580729,
      "learning_rate": 0.0001,
      "loss": 4.0754,
      "loss/crossentropy": 2.0716105699539185,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1935330480337143,
      "step": 23596
    },
    {
      "epoch": 0.47196,
      "grad_norm": 2.0,
      "grad_norm_var": 0.020845540364583335,
      "learning_rate": 0.0001,
      "loss": 3.8819,
      "loss/crossentropy": 2.0923121571540833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20516054332256317,
      "step": 23598
    },
    {
      "epoch": 0.472,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.020357259114583335,
      "learning_rate": 0.0001,
      "loss": 4.0871,
      "loss/crossentropy": 2.244894862174988,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21077310293912888,
      "step": 23600
    },
    {
      "epoch": 0.47204,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.02054417928059896,
      "learning_rate": 0.0001,
      "loss": 3.9788,
      "loss/crossentropy": 1.9484725594520569,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16783934831619263,
      "step": 23602
    },
    {
      "epoch": 0.47208,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.19378026326497397,
      "learning_rate": 0.0001,
      "loss": 3.8165,
      "loss/crossentropy": 1.8027321100234985,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21114902943372726,
      "step": 23604
    },
    {
      "epoch": 0.47212,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.18945083618164063,
      "learning_rate": 0.0001,
      "loss": 3.8104,
      "loss/crossentropy": 1.7768787741661072,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1848941519856453,
      "step": 23606
    },
    {
      "epoch": 0.47216,
      "grad_norm": 2.125,
      "grad_norm_var": 0.19102274576822917,
      "learning_rate": 0.0001,
      "loss": 3.8457,
      "loss/crossentropy": 1.8786412477493286,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18932975828647614,
      "step": 23608
    },
    {
      "epoch": 0.4722,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.1960113525390625,
      "learning_rate": 0.0001,
      "loss": 3.9519,
      "loss/crossentropy": 2.0566998720169067,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1795952171087265,
      "step": 23610
    },
    {
      "epoch": 0.47224,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.19468154907226562,
      "learning_rate": 0.0001,
      "loss": 4.0255,
      "loss/crossentropy": 1.9085100293159485,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19264912605285645,
      "step": 23612
    },
    {
      "epoch": 0.47228,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.19574381510416666,
      "learning_rate": 0.0001,
      "loss": 4.0038,
      "loss/crossentropy": 2.0626463294029236,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1947372406721115,
      "step": 23614
    },
    {
      "epoch": 0.47232,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.1929278055826823,
      "learning_rate": 0.0001,
      "loss": 4.144,
      "loss/crossentropy": 2.0861737728118896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21039434522390366,
      "step": 23616
    },
    {
      "epoch": 0.47236,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.19161783854166667,
      "learning_rate": 0.0001,
      "loss": 3.9677,
      "loss/crossentropy": 1.9024608135223389,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18196185678243637,
      "step": 23618
    },
    {
      "epoch": 0.4724,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.006371815999348958,
      "learning_rate": 0.0001,
      "loss": 4.1103,
      "loss/crossentropy": 2.209709107875824,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21473529189825058,
      "step": 23620
    },
    {
      "epoch": 0.47244,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0069010416666666664,
      "learning_rate": 0.0001,
      "loss": 3.9269,
      "loss/crossentropy": 2.1699984073638916,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2023090422153473,
      "step": 23622
    },
    {
      "epoch": 0.47248,
      "grad_norm": 1.875,
      "grad_norm_var": 0.006441243489583333,
      "learning_rate": 0.0001,
      "loss": 3.7869,
      "loss/crossentropy": 1.7027064561843872,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16561312228441238,
      "step": 23624
    },
    {
      "epoch": 0.47252,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.008526357014973958,
      "learning_rate": 0.0001,
      "loss": 3.9753,
      "loss/crossentropy": 2.2230480909347534,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20173655450344086,
      "step": 23626
    },
    {
      "epoch": 0.47256,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.008841705322265626,
      "learning_rate": 0.0001,
      "loss": 3.6368,
      "loss/crossentropy": 1.9015594124794006,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1708713322877884,
      "step": 23628
    },
    {
      "epoch": 0.4726,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.010723622639973958,
      "learning_rate": 0.0001,
      "loss": 4.0118,
      "loss/crossentropy": 2.051177144050598,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1878596916794777,
      "step": 23630
    },
    {
      "epoch": 0.47264,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.013752237955729166,
      "learning_rate": 0.0001,
      "loss": 3.9746,
      "loss/crossentropy": 1.9841215014457703,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1841672658920288,
      "step": 23632
    },
    {
      "epoch": 0.47268,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0148590087890625,
      "learning_rate": 0.0001,
      "loss": 4.1022,
      "loss/crossentropy": 2.0642913579940796,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22168366611003876,
      "step": 23634
    },
    {
      "epoch": 0.47272,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.014899698893229167,
      "learning_rate": 0.0001,
      "loss": 4.0448,
      "loss/crossentropy": 2.149936318397522,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1954735815525055,
      "step": 23636
    },
    {
      "epoch": 0.47276,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.014811197916666666,
      "learning_rate": 0.0001,
      "loss": 3.9646,
      "loss/crossentropy": 2.0555055141448975,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19528509676456451,
      "step": 23638
    },
    {
      "epoch": 0.4728,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.013600413004557292,
      "learning_rate": 0.0001,
      "loss": 4.0768,
      "loss/crossentropy": 1.9836488366127014,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17974190413951874,
      "step": 23640
    },
    {
      "epoch": 0.47284,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.014644368489583334,
      "learning_rate": 0.0001,
      "loss": 4.4381,
      "loss/crossentropy": 2.455438733100891,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21470309793949127,
      "step": 23642
    },
    {
      "epoch": 0.47288,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0148345947265625,
      "learning_rate": 0.0001,
      "loss": 3.9095,
      "loss/crossentropy": 2.0290130376815796,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19266076385974884,
      "step": 23644
    },
    {
      "epoch": 0.47292,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.014312489827473959,
      "learning_rate": 0.0001,
      "loss": 3.8732,
      "loss/crossentropy": 1.9230349659919739,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17801092565059662,
      "step": 23646
    },
    {
      "epoch": 0.47296,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.014452107747395833,
      "learning_rate": 0.0001,
      "loss": 4.1517,
      "loss/crossentropy": 2.015698790550232,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20072294026613235,
      "step": 23648
    },
    {
      "epoch": 0.473,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.014357248942057291,
      "learning_rate": 0.0001,
      "loss": 3.8298,
      "loss/crossentropy": 1.9033147096633911,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18670127540826797,
      "step": 23650
    },
    {
      "epoch": 0.47304,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.014906565348307291,
      "learning_rate": 0.0001,
      "loss": 3.7137,
      "loss/crossentropy": 1.7301848530769348,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1701732650399208,
      "step": 23652
    },
    {
      "epoch": 0.47308,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.016747029622395833,
      "learning_rate": 0.0001,
      "loss": 3.7671,
      "loss/crossentropy": 2.0487022399902344,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18679651618003845,
      "step": 23654
    },
    {
      "epoch": 0.47312,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.017064412434895832,
      "learning_rate": 0.0001,
      "loss": 4.0564,
      "loss/crossentropy": 2.1262341737747192,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21661578118801117,
      "step": 23656
    },
    {
      "epoch": 0.47316,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.011466217041015626,
      "learning_rate": 0.0001,
      "loss": 3.8656,
      "loss/crossentropy": 1.8145031929016113,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1787424087524414,
      "step": 23658
    },
    {
      "epoch": 0.4732,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0114013671875,
      "learning_rate": 0.0001,
      "loss": 3.9916,
      "loss/crossentropy": 2.1354995369911194,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19884371012449265,
      "step": 23660
    },
    {
      "epoch": 0.47324,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.010920206705729166,
      "learning_rate": 0.0001,
      "loss": 3.8533,
      "loss/crossentropy": 1.7008216381072998,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1525547280907631,
      "step": 23662
    },
    {
      "epoch": 0.47328,
      "grad_norm": 2.0,
      "grad_norm_var": 0.006880442301432292,
      "learning_rate": 0.0001,
      "loss": 4.0684,
      "loss/crossentropy": 2.235612154006958,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21687253564596176,
      "step": 23664
    },
    {
      "epoch": 0.47332,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.008819325764973959,
      "learning_rate": 0.0001,
      "loss": 4.3348,
      "loss/crossentropy": 2.375667631626129,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21823541820049286,
      "step": 23666
    },
    {
      "epoch": 0.47336,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.007995351155598959,
      "learning_rate": 0.0001,
      "loss": 4.2375,
      "loss/crossentropy": 2.223970353603363,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21631266921758652,
      "step": 23668
    },
    {
      "epoch": 0.4734,
      "grad_norm": 1.7265625,
      "grad_norm_var": 0.009261067708333333,
      "learning_rate": 0.0001,
      "loss": 3.7413,
      "loss/crossentropy": 1.9337742328643799,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18401625007390976,
      "step": 23670
    },
    {
      "epoch": 0.47344,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.011400349934895833,
      "learning_rate": 0.0001,
      "loss": 4.1189,
      "loss/crossentropy": 2.1500974893569946,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19867678731679916,
      "step": 23672
    },
    {
      "epoch": 0.47348,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.015103912353515625,
      "learning_rate": 0.0001,
      "loss": 4.0475,
      "loss/crossentropy": 1.8808711171150208,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18535293638706207,
      "step": 23674
    },
    {
      "epoch": 0.47352,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0133544921875,
      "learning_rate": 0.0001,
      "loss": 4.1431,
      "loss/crossentropy": 2.1114797592163086,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1959507018327713,
      "step": 23676
    },
    {
      "epoch": 0.47356,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.016141510009765624,
      "learning_rate": 0.0001,
      "loss": 3.8953,
      "loss/crossentropy": 2.2539754509925842,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20711452513933182,
      "step": 23678
    },
    {
      "epoch": 0.4736,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.01567357381184896,
      "learning_rate": 0.0001,
      "loss": 3.9558,
      "loss/crossentropy": 1.950667679309845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18103761970996857,
      "step": 23680
    },
    {
      "epoch": 0.47364,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.015413157145182292,
      "learning_rate": 0.0001,
      "loss": 3.9297,
      "loss/crossentropy": 1.9214385747909546,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19267746806144714,
      "step": 23682
    },
    {
      "epoch": 0.47368,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.0181304931640625,
      "learning_rate": 0.0001,
      "loss": 3.7439,
      "loss/crossentropy": 2.053990364074707,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18003928661346436,
      "step": 23684
    },
    {
      "epoch": 0.47372,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0151275634765625,
      "learning_rate": 0.0001,
      "loss": 3.951,
      "loss/crossentropy": 1.6660608649253845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15546631067991257,
      "step": 23686
    },
    {
      "epoch": 0.47376,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.013250478108723958,
      "learning_rate": 0.0001,
      "loss": 3.8348,
      "loss/crossentropy": 2.1377063989639282,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.202220119535923,
      "step": 23688
    },
    {
      "epoch": 0.4738,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.009293619791666667,
      "learning_rate": 0.0001,
      "loss": 3.8619,
      "loss/crossentropy": 1.9241121411323547,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17836976051330566,
      "step": 23690
    },
    {
      "epoch": 0.47384,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.009159088134765625,
      "learning_rate": 0.0001,
      "loss": 4.2613,
      "loss/crossentropy": 2.158058762550354,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22955617308616638,
      "step": 23692
    },
    {
      "epoch": 0.47388,
      "grad_norm": 2.0,
      "grad_norm_var": 0.006870269775390625,
      "learning_rate": 0.0001,
      "loss": 4.24,
      "loss/crossentropy": 2.138846278190613,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2145211324095726,
      "step": 23694
    },
    {
      "epoch": 0.47392,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.006341298421223958,
      "learning_rate": 0.0001,
      "loss": 3.9193,
      "loss/crossentropy": 2.0770075917243958,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2009606510400772,
      "step": 23696
    },
    {
      "epoch": 0.47396,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.006363932291666667,
      "learning_rate": 0.0001,
      "loss": 3.8684,
      "loss/crossentropy": 1.5739121437072754,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16019296646118164,
      "step": 23698
    },
    {
      "epoch": 0.474,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.005078125,
      "learning_rate": 0.0001,
      "loss": 4.1262,
      "loss/crossentropy": 1.898674726486206,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1942930817604065,
      "step": 23700
    },
    {
      "epoch": 0.47404,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.004776763916015625,
      "learning_rate": 0.0001,
      "loss": 4.0102,
      "loss/crossentropy": 2.2228487730026245,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2093074843287468,
      "step": 23702
    },
    {
      "epoch": 0.47408,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.006390126546223959,
      "learning_rate": 0.0001,
      "loss": 4.1041,
      "loss/crossentropy": 2.1111066341400146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19743914157152176,
      "step": 23704
    },
    {
      "epoch": 0.47412,
      "grad_norm": 3.359375,
      "grad_norm_var": 0.1302886962890625,
      "learning_rate": 0.0001,
      "loss": 4.0682,
      "loss/crossentropy": 1.780187964439392,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18026559054851532,
      "step": 23706
    },
    {
      "epoch": 0.47416,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.13217137654622396,
      "learning_rate": 0.0001,
      "loss": 3.8936,
      "loss/crossentropy": 1.983467936515808,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.188174769282341,
      "step": 23708
    },
    {
      "epoch": 0.4742,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.13393961588541667,
      "learning_rate": 0.0001,
      "loss": 4.0822,
      "loss/crossentropy": 2.0928043723106384,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2048778161406517,
      "step": 23710
    },
    {
      "epoch": 0.47424,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.13646240234375,
      "learning_rate": 0.0001,
      "loss": 4.0171,
      "loss/crossentropy": 1.8958263397216797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17733225226402283,
      "step": 23712
    },
    {
      "epoch": 0.47428,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.1341461181640625,
      "learning_rate": 0.0001,
      "loss": 4.0589,
      "loss/crossentropy": 1.8104780316352844,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1762789711356163,
      "step": 23714
    },
    {
      "epoch": 0.47432,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.1332415262858073,
      "learning_rate": 0.0001,
      "loss": 4.0282,
      "loss/crossentropy": 2.167409896850586,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20198846608400345,
      "step": 23716
    },
    {
      "epoch": 0.47436,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.13642756144205728,
      "learning_rate": 0.0001,
      "loss": 3.9456,
      "loss/crossentropy": 1.8979802131652832,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17944183200597763,
      "step": 23718
    },
    {
      "epoch": 0.4744,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.13585205078125,
      "learning_rate": 0.0001,
      "loss": 4.1639,
      "loss/crossentropy": 2.3538358211517334,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2084932029247284,
      "step": 23720
    },
    {
      "epoch": 0.47444,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.005028279622395834,
      "learning_rate": 0.0001,
      "loss": 4.1931,
      "loss/crossentropy": 2.144857406616211,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20259954035282135,
      "step": 23722
    },
    {
      "epoch": 0.47448,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0050771077473958336,
      "learning_rate": 0.0001,
      "loss": 4.1082,
      "loss/crossentropy": 2.3802285194396973,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2301197499036789,
      "step": 23724
    },
    {
      "epoch": 0.47452,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.006025950113932292,
      "learning_rate": 0.0001,
      "loss": 4.212,
      "loss/crossentropy": 2.5175833702087402,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22316204756498337,
      "step": 23726
    },
    {
      "epoch": 0.47456,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.006514485677083333,
      "learning_rate": 0.0001,
      "loss": 4.3959,
      "loss/crossentropy": 2.266839861869812,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20391426980495453,
      "step": 23728
    },
    {
      "epoch": 0.4746,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.006526438395182291,
      "learning_rate": 0.0001,
      "loss": 3.9426,
      "loss/crossentropy": 1.9993603229522705,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19872139394283295,
      "step": 23730
    },
    {
      "epoch": 0.47464,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.007321929931640625,
      "learning_rate": 0.0001,
      "loss": 4.0236,
      "loss/crossentropy": 1.999217450618744,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18402128666639328,
      "step": 23732
    },
    {
      "epoch": 0.47468,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.006894683837890625,
      "learning_rate": 0.0001,
      "loss": 4.0597,
      "loss/crossentropy": 2.2356200218200684,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19472184777259827,
      "step": 23734
    },
    {
      "epoch": 0.47472,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.0072062174479166664,
      "learning_rate": 0.0001,
      "loss": 3.9894,
      "loss/crossentropy": 1.6930708289146423,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16877874732017517,
      "step": 23736
    },
    {
      "epoch": 0.47476,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.007985178629557292,
      "learning_rate": 0.0001,
      "loss": 3.6418,
      "loss/crossentropy": 2.0427737832069397,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20055241882801056,
      "step": 23738
    },
    {
      "epoch": 0.4748,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.007795206705729167,
      "learning_rate": 0.0001,
      "loss": 4.0536,
      "loss/crossentropy": 2.11184823513031,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1939689666032791,
      "step": 23740
    },
    {
      "epoch": 0.47484,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.008778635660807292,
      "learning_rate": 0.0001,
      "loss": 3.8553,
      "loss/crossentropy": 1.9833735823631287,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16304892301559448,
      "step": 23742
    },
    {
      "epoch": 0.47488,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.006754302978515625,
      "learning_rate": 0.0001,
      "loss": 4.1299,
      "loss/crossentropy": 2.0734696984291077,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18283121287822723,
      "step": 23744
    },
    {
      "epoch": 0.47492,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.00877685546875,
      "learning_rate": 0.0001,
      "loss": 4.0763,
      "loss/crossentropy": 1.7331766486167908,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18056727200746536,
      "step": 23746
    },
    {
      "epoch": 0.47496,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0089752197265625,
      "learning_rate": 0.0001,
      "loss": 4.314,
      "loss/crossentropy": 2.224283456802368,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21115867048501968,
      "step": 23748
    },
    {
      "epoch": 0.475,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.007734934488932292,
      "learning_rate": 0.0001,
      "loss": 3.9618,
      "loss/crossentropy": 1.6858720183372498,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16034550219774246,
      "step": 23750
    },
    {
      "epoch": 0.47504,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.008304850260416666,
      "learning_rate": 0.0001,
      "loss": 4.2015,
      "loss/crossentropy": 2.038428485393524,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2020621821284294,
      "step": 23752
    },
    {
      "epoch": 0.47508,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.007835896809895833,
      "learning_rate": 0.0001,
      "loss": 4.0219,
      "loss/crossentropy": 2.076684832572937,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1903630793094635,
      "step": 23754
    },
    {
      "epoch": 0.47512,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.007665761311848958,
      "learning_rate": 0.0001,
      "loss": 3.7804,
      "loss/crossentropy": 1.7917174100875854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1796514242887497,
      "step": 23756
    },
    {
      "epoch": 0.47516,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006859334309895834,
      "learning_rate": 0.0001,
      "loss": 4.0059,
      "loss/crossentropy": 1.9619473814964294,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18760067224502563,
      "step": 23758
    },
    {
      "epoch": 0.4752,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.007990519205729166,
      "learning_rate": 0.0001,
      "loss": 3.787,
      "loss/crossentropy": 2.083077907562256,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19316518306732178,
      "step": 23760
    },
    {
      "epoch": 0.47524,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.007100168863932292,
      "learning_rate": 0.0001,
      "loss": 3.8462,
      "loss/crossentropy": 1.8045600056648254,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17447172850370407,
      "step": 23762
    },
    {
      "epoch": 0.47528,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.013691965738932292,
      "learning_rate": 0.0001,
      "loss": 3.8125,
      "loss/crossentropy": 2.2159335613250732,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20490044355392456,
      "step": 23764
    },
    {
      "epoch": 0.47532,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.014851633707682292,
      "learning_rate": 0.0001,
      "loss": 3.7541,
      "loss/crossentropy": 2.0102903246879578,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17014086246490479,
      "step": 23766
    },
    {
      "epoch": 0.47536,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.012823232014973958,
      "learning_rate": 0.0001,
      "loss": 4.0154,
      "loss/crossentropy": 2.0524433851242065,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20510738343000412,
      "step": 23768
    },
    {
      "epoch": 0.4754,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.012849934895833333,
      "learning_rate": 0.0001,
      "loss": 3.8245,
      "loss/crossentropy": 1.9695302844047546,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17639748752117157,
      "step": 23770
    },
    {
      "epoch": 0.47544,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.014642079671223959,
      "learning_rate": 0.0001,
      "loss": 4.4496,
      "loss/crossentropy": 2.3865933418273926,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23802988976240158,
      "step": 23772
    },
    {
      "epoch": 0.47548,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.013480631510416667,
      "learning_rate": 0.0001,
      "loss": 3.8708,
      "loss/crossentropy": 1.6841627955436707,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1837886944413185,
      "step": 23774
    },
    {
      "epoch": 0.47552,
      "grad_norm": 1.875,
      "grad_norm_var": 0.0132232666015625,
      "learning_rate": 0.0001,
      "loss": 3.9562,
      "loss/crossentropy": 1.9436610341072083,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2144036665558815,
      "step": 23776
    },
    {
      "epoch": 0.47556,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.013846842447916667,
      "learning_rate": 0.0001,
      "loss": 3.9312,
      "loss/crossentropy": 1.8565305471420288,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19765028357505798,
      "step": 23778
    },
    {
      "epoch": 0.4756,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.007165273030598958,
      "learning_rate": 0.0001,
      "loss": 4.0102,
      "loss/crossentropy": 2.119917571544647,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1896904706954956,
      "step": 23780
    },
    {
      "epoch": 0.47564,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.008892567952473958,
      "learning_rate": 0.0001,
      "loss": 4.0573,
      "loss/crossentropy": 1.9241788387298584,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17011697590351105,
      "step": 23782
    },
    {
      "epoch": 0.47568,
      "grad_norm": 1.875,
      "grad_norm_var": 0.010495758056640625,
      "learning_rate": 0.0001,
      "loss": 3.7454,
      "loss/crossentropy": 1.7912859320640564,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17273171246051788,
      "step": 23784
    },
    {
      "epoch": 0.47572,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0112945556640625,
      "learning_rate": 0.0001,
      "loss": 4.4069,
      "loss/crossentropy": 2.4223393201828003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2194758951663971,
      "step": 23786
    },
    {
      "epoch": 0.47576,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.010416412353515625,
      "learning_rate": 0.0001,
      "loss": 3.7825,
      "loss/crossentropy": 1.9073758721351624,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19918088614940643,
      "step": 23788
    },
    {
      "epoch": 0.4758,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.0115631103515625,
      "learning_rate": 0.0001,
      "loss": 3.4428,
      "loss/crossentropy": 1.5801246762275696,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16602589190006256,
      "step": 23790
    },
    {
      "epoch": 0.47584,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.010892740885416667,
      "learning_rate": 0.0001,
      "loss": 3.7883,
      "loss/crossentropy": 1.847874402999878,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17299149185419083,
      "step": 23792
    },
    {
      "epoch": 0.47588,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.010431925455729166,
      "learning_rate": 0.0001,
      "loss": 3.9211,
      "loss/crossentropy": 2.272279739379883,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19657636433839798,
      "step": 23794
    },
    {
      "epoch": 0.47592,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.014387003580729167,
      "learning_rate": 0.0001,
      "loss": 4.422,
      "loss/crossentropy": 1.9107636213302612,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1978672295808792,
      "step": 23796
    },
    {
      "epoch": 0.47596,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.011374664306640626,
      "learning_rate": 0.0001,
      "loss": 4.2113,
      "loss/crossentropy": 2.384741425514221,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.213532455265522,
      "step": 23798
    },
    {
      "epoch": 0.476,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.019207509358723958,
      "learning_rate": 0.0001,
      "loss": 4.2722,
      "loss/crossentropy": 2.2380464673042297,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1984338015317917,
      "step": 23800
    },
    {
      "epoch": 0.47604,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.019465891520182292,
      "learning_rate": 0.0001,
      "loss": 4.0274,
      "loss/crossentropy": 2.044179916381836,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18904277682304382,
      "step": 23802
    },
    {
      "epoch": 0.47608,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.019425455729166666,
      "learning_rate": 0.0001,
      "loss": 3.9933,
      "loss/crossentropy": 2.3498952388763428,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2706408053636551,
      "step": 23804
    },
    {
      "epoch": 0.47612,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.017710113525390626,
      "learning_rate": 0.0001,
      "loss": 3.7679,
      "loss/crossentropy": 2.2029510736465454,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19871965795755386,
      "step": 23806
    },
    {
      "epoch": 0.47616,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0169830322265625,
      "learning_rate": 0.0001,
      "loss": 3.9891,
      "loss/crossentropy": 1.7823997139930725,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17446641623973846,
      "step": 23808
    },
    {
      "epoch": 0.4762,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.016414388020833334,
      "learning_rate": 0.0001,
      "loss": 4.0756,
      "loss/crossentropy": 1.9810225367546082,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1739773526787758,
      "step": 23810
    },
    {
      "epoch": 0.47624,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.015429433186848958,
      "learning_rate": 0.0001,
      "loss": 4.1135,
      "loss/crossentropy": 2.0867974758148193,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1936039850115776,
      "step": 23812
    },
    {
      "epoch": 0.47628,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.017752838134765626,
      "learning_rate": 0.0001,
      "loss": 3.7694,
      "loss/crossentropy": 1.88890939950943,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18023589998483658,
      "step": 23814
    },
    {
      "epoch": 0.47632,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.007838694254557292,
      "learning_rate": 0.0001,
      "loss": 3.7677,
      "loss/crossentropy": 2.1205832958221436,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20733095705509186,
      "step": 23816
    },
    {
      "epoch": 0.47636,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.009804026285807291,
      "learning_rate": 0.0001,
      "loss": 3.9361,
      "loss/crossentropy": 2.1970152854919434,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2084852010011673,
      "step": 23818
    },
    {
      "epoch": 0.4764,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.011484527587890625,
      "learning_rate": 0.0001,
      "loss": 3.9324,
      "loss/crossentropy": 1.9947285056114197,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.181013822555542,
      "step": 23820
    },
    {
      "epoch": 0.47644,
      "grad_norm": 1.875,
      "grad_norm_var": 0.011205037434895834,
      "learning_rate": 0.0001,
      "loss": 3.7626,
      "loss/crossentropy": 1.8331729769706726,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18118757009506226,
      "step": 23822
    },
    {
      "epoch": 0.47648,
      "grad_norm": 1.875,
      "grad_norm_var": 0.0111083984375,
      "learning_rate": 0.0001,
      "loss": 4.0555,
      "loss/crossentropy": 2.191226840019226,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.202358178794384,
      "step": 23824
    },
    {
      "epoch": 0.47652,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.009348297119140625,
      "learning_rate": 0.0001,
      "loss": 3.9524,
      "loss/crossentropy": 1.9327014088630676,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17946144193410873,
      "step": 23826
    },
    {
      "epoch": 0.47656,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.007482655843098958,
      "learning_rate": 0.0001,
      "loss": 3.8711,
      "loss/crossentropy": 2.2243237495422363,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19515040516853333,
      "step": 23828
    },
    {
      "epoch": 0.4766,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.00675048828125,
      "learning_rate": 0.0001,
      "loss": 4.1158,
      "loss/crossentropy": 2.176102042198181,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19702810794115067,
      "step": 23830
    },
    {
      "epoch": 0.47664,
      "grad_norm": 1.875,
      "grad_norm_var": 0.006197102864583333,
      "learning_rate": 0.0001,
      "loss": 4.0918,
      "loss/crossentropy": 2.1210728883743286,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.222815603017807,
      "step": 23832
    },
    {
      "epoch": 0.47668,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.004134114583333333,
      "learning_rate": 0.0001,
      "loss": 3.9675,
      "loss/crossentropy": 1.9496826529502869,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17903053760528564,
      "step": 23834
    },
    {
      "epoch": 0.47672,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0017575581868489584,
      "learning_rate": 0.0001,
      "loss": 3.9558,
      "loss/crossentropy": 2.139560639858246,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20509754866361618,
      "step": 23836
    },
    {
      "epoch": 0.47676,
      "grad_norm": 1.875,
      "grad_norm_var": 0.0016591389973958333,
      "learning_rate": 0.0001,
      "loss": 4.0937,
      "loss/crossentropy": 2.102014422416687,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19706308841705322,
      "step": 23838
    },
    {
      "epoch": 0.4768,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.0017486572265625,
      "learning_rate": 0.0001,
      "loss": 3.8519,
      "loss/crossentropy": 1.9479581117630005,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20007412880659103,
      "step": 23840
    },
    {
      "epoch": 0.47684,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.003177897135416667,
      "learning_rate": 0.0001,
      "loss": 4.2284,
      "loss/crossentropy": 1.8896448016166687,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19973865151405334,
      "step": 23842
    },
    {
      "epoch": 0.47688,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.0030995686848958332,
      "learning_rate": 0.0001,
      "loss": 4.0693,
      "loss/crossentropy": 2.057901084423065,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19682063907384872,
      "step": 23844
    },
    {
      "epoch": 0.47692,
      "grad_norm": 2.0,
      "grad_norm_var": 0.007355753580729167,
      "learning_rate": 0.0001,
      "loss": 4.2305,
      "loss/crossentropy": 1.9882251024246216,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18287301808595657,
      "step": 23846
    },
    {
      "epoch": 0.47696,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.008421834309895833,
      "learning_rate": 0.0001,
      "loss": 4.1016,
      "loss/crossentropy": 1.9481773972511292,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17307578027248383,
      "step": 23848
    },
    {
      "epoch": 0.477,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.01718928019205729,
      "learning_rate": 0.0001,
      "loss": 3.8738,
      "loss/crossentropy": 1.8696550130844116,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18704131245613098,
      "step": 23850
    },
    {
      "epoch": 0.47704,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.0187652587890625,
      "learning_rate": 0.0001,
      "loss": 3.8426,
      "loss/crossentropy": 2.0851826667785645,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18556270003318787,
      "step": 23852
    },
    {
      "epoch": 0.47708,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.018943023681640626,
      "learning_rate": 0.0001,
      "loss": 4.2501,
      "loss/crossentropy": 2.3411136865615845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.225515678524971,
      "step": 23854
    },
    {
      "epoch": 0.47712,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.01817804972330729,
      "learning_rate": 0.0001,
      "loss": 3.7964,
      "loss/crossentropy": 1.944790780544281,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.195308655500412,
      "step": 23856
    },
    {
      "epoch": 0.47716,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.026594034830729165,
      "learning_rate": 0.0001,
      "loss": 4.004,
      "loss/crossentropy": 1.8672103881835938,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18687407672405243,
      "step": 23858
    },
    {
      "epoch": 0.4772,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.026889801025390625,
      "learning_rate": 0.0001,
      "loss": 3.826,
      "loss/crossentropy": 1.9624406099319458,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19766812026500702,
      "step": 23860
    },
    {
      "epoch": 0.47724,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.026609039306640624,
      "learning_rate": 0.0001,
      "loss": 3.997,
      "loss/crossentropy": 2.1671148538589478,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1916585937142372,
      "step": 23862
    },
    {
      "epoch": 0.47728,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.026668039957682292,
      "learning_rate": 0.0001,
      "loss": 4.0202,
      "loss/crossentropy": 1.8473829627037048,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16666045784950256,
      "step": 23864
    },
    {
      "epoch": 0.47732,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.018182118733723957,
      "learning_rate": 0.0001,
      "loss": 4.0035,
      "loss/crossentropy": 2.0888859033584595,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19743109494447708,
      "step": 23866
    },
    {
      "epoch": 0.47736,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.01679051717122396,
      "learning_rate": 0.0001,
      "loss": 3.8915,
      "loss/crossentropy": 2.0148197412490845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2103613168001175,
      "step": 23868
    },
    {
      "epoch": 0.4774,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.015550740559895833,
      "learning_rate": 0.0001,
      "loss": 3.9269,
      "loss/crossentropy": 1.9780926704406738,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18435538560152054,
      "step": 23870
    },
    {
      "epoch": 0.47744,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.016068522135416666,
      "learning_rate": 0.0001,
      "loss": 3.9308,
      "loss/crossentropy": 2.026561677455902,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19771291315555573,
      "step": 23872
    },
    {
      "epoch": 0.47748,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.005606842041015625,
      "learning_rate": 0.0001,
      "loss": 3.9616,
      "loss/crossentropy": 2.029239237308502,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19737554341554642,
      "step": 23874
    },
    {
      "epoch": 0.47752,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.005360666910807292,
      "learning_rate": 0.0001,
      "loss": 3.9622,
      "loss/crossentropy": 2.32661509513855,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20728177577257156,
      "step": 23876
    },
    {
      "epoch": 0.47756,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.005086008707682292,
      "learning_rate": 0.0001,
      "loss": 3.8965,
      "loss/crossentropy": 2.1897658109664917,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1915959268808365,
      "step": 23878
    },
    {
      "epoch": 0.4776,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.004889933268229166,
      "learning_rate": 0.0001,
      "loss": 4.1431,
      "loss/crossentropy": 2.3252129554748535,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21492170542478561,
      "step": 23880
    },
    {
      "epoch": 0.47764,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.005492146809895833,
      "learning_rate": 0.0001,
      "loss": 3.8796,
      "loss/crossentropy": 1.6142144203186035,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1473793387413025,
      "step": 23882
    },
    {
      "epoch": 0.47768,
      "grad_norm": 2.0,
      "grad_norm_var": 0.005702463785807291,
      "learning_rate": 0.0001,
      "loss": 4.2673,
      "loss/crossentropy": 2.3027801513671875,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21568017452955246,
      "step": 23884
    },
    {
      "epoch": 0.47772,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006791178385416667,
      "learning_rate": 0.0001,
      "loss": 3.9876,
      "loss/crossentropy": 1.7805312871932983,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19376665353775024,
      "step": 23886
    },
    {
      "epoch": 0.47776,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.00482177734375,
      "learning_rate": 0.0001,
      "loss": 4.0035,
      "loss/crossentropy": 2.011800706386566,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1892775595188141,
      "step": 23888
    },
    {
      "epoch": 0.4778,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.005649566650390625,
      "learning_rate": 0.0001,
      "loss": 4.0836,
      "loss/crossentropy": 2.257364869117737,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2021905705332756,
      "step": 23890
    },
    {
      "epoch": 0.47784,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.005641428629557291,
      "learning_rate": 0.0001,
      "loss": 4.0778,
      "loss/crossentropy": 2.211492657661438,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19960176944732666,
      "step": 23892
    },
    {
      "epoch": 0.47788,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006807454427083333,
      "learning_rate": 0.0001,
      "loss": 3.8243,
      "loss/crossentropy": 2.250791549682617,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19187939167022705,
      "step": 23894
    },
    {
      "epoch": 0.47792,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0075927734375,
      "learning_rate": 0.0001,
      "loss": 3.9676,
      "loss/crossentropy": 1.8536748886108398,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1737053319811821,
      "step": 23896
    },
    {
      "epoch": 0.47796,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.014509073893229167,
      "learning_rate": 0.0001,
      "loss": 4.5734,
      "loss/crossentropy": 2.1692601442337036,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19785165041685104,
      "step": 23898
    },
    {
      "epoch": 0.478,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.014826456705729166,
      "learning_rate": 0.0001,
      "loss": 4.0647,
      "loss/crossentropy": 2.229991912841797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22324570268392563,
      "step": 23900
    },
    {
      "epoch": 0.47804,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.014045206705729167,
      "learning_rate": 0.0001,
      "loss": 3.9411,
      "loss/crossentropy": 2.2393418550491333,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19389375299215317,
      "step": 23902
    },
    {
      "epoch": 0.47808,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.014009348551432292,
      "learning_rate": 0.0001,
      "loss": 4.1504,
      "loss/crossentropy": 2.072646141052246,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20844239741563797,
      "step": 23904
    },
    {
      "epoch": 0.47812,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.014202626546223958,
      "learning_rate": 0.0001,
      "loss": 4.1153,
      "loss/crossentropy": 2.3049235343933105,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2123921662569046,
      "step": 23906
    },
    {
      "epoch": 0.47816,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.014964803059895834,
      "learning_rate": 0.0001,
      "loss": 3.7974,
      "loss/crossentropy": 1.9755294919013977,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17440249770879745,
      "step": 23908
    },
    {
      "epoch": 0.4782,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.013402303059895834,
      "learning_rate": 0.0001,
      "loss": 3.8354,
      "loss/crossentropy": 1.8338207602500916,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18320723623037338,
      "step": 23910
    },
    {
      "epoch": 0.47824,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.026493326822916666,
      "learning_rate": 0.0001,
      "loss": 4.3086,
      "loss/crossentropy": 2.079473555088043,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2535192593932152,
      "step": 23912
    },
    {
      "epoch": 0.47828,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.023119099934895835,
      "learning_rate": 0.0001,
      "loss": 3.6006,
      "loss/crossentropy": 1.7429603934288025,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17608244717121124,
      "step": 23914
    },
    {
      "epoch": 0.47832,
      "grad_norm": 1.875,
      "grad_norm_var": 0.023435211181640624,
      "learning_rate": 0.0001,
      "loss": 3.9687,
      "loss/crossentropy": 2.1863731145858765,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18513252586126328,
      "step": 23916
    },
    {
      "epoch": 0.47836,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0236083984375,
      "learning_rate": 0.0001,
      "loss": 3.948,
      "loss/crossentropy": 2.1431294679641724,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22228983789682388,
      "step": 23918
    },
    {
      "epoch": 0.4784,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.023827107747395833,
      "learning_rate": 0.0001,
      "loss": 3.9218,
      "loss/crossentropy": 1.8407437801361084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16755413264036179,
      "step": 23920
    },
    {
      "epoch": 0.47844,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.025229644775390626,
      "learning_rate": 0.0001,
      "loss": 4.2533,
      "loss/crossentropy": 2.0921709537506104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20164211839437485,
      "step": 23922
    },
    {
      "epoch": 0.47848,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.04986979166666667,
      "learning_rate": 0.0001,
      "loss": 3.9155,
      "loss/crossentropy": 1.5495757460594177,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1741114780306816,
      "step": 23924
    },
    {
      "epoch": 0.47852,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.050510406494140625,
      "learning_rate": 0.0001,
      "loss": 3.9084,
      "loss/crossentropy": 2.0054045915603638,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18482672423124313,
      "step": 23926
    },
    {
      "epoch": 0.47856,
      "grad_norm": 3.015625,
      "grad_norm_var": 0.10667724609375,
      "learning_rate": 0.0001,
      "loss": 4.3504,
      "loss/crossentropy": 1.9925623536109924,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19737054407596588,
      "step": 23928
    },
    {
      "epoch": 0.4786,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0996002197265625,
      "learning_rate": 0.0001,
      "loss": 4.1226,
      "loss/crossentropy": 1.9019532203674316,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19580943882465363,
      "step": 23930
    },
    {
      "epoch": 0.47864,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.10124104817708333,
      "learning_rate": 0.0001,
      "loss": 3.969,
      "loss/crossentropy": 1.9665276408195496,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18561270087957382,
      "step": 23932
    },
    {
      "epoch": 0.47868,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.10370992024739584,
      "learning_rate": 0.0001,
      "loss": 4.3574,
      "loss/crossentropy": 2.390560507774353,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.194176584482193,
      "step": 23934
    },
    {
      "epoch": 0.47872,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.10062026977539062,
      "learning_rate": 0.0001,
      "loss": 4.2517,
      "loss/crossentropy": 2.1132951974868774,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1915990635752678,
      "step": 23936
    },
    {
      "epoch": 0.47876,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.1031158447265625,
      "learning_rate": 0.0001,
      "loss": 4.0203,
      "loss/crossentropy": 1.8993237018585205,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18519867211580276,
      "step": 23938
    },
    {
      "epoch": 0.4788,
      "grad_norm": 2.0,
      "grad_norm_var": 0.08478190104166666,
      "learning_rate": 0.0001,
      "loss": 4.546,
      "loss/crossentropy": 2.1111881732940674,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20509092509746552,
      "step": 23940
    },
    {
      "epoch": 0.47884,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.08461812337239584,
      "learning_rate": 0.0001,
      "loss": 4.1067,
      "loss/crossentropy": 2.3452861309051514,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19262470304965973,
      "step": 23942
    },
    {
      "epoch": 0.47888,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.014733632405598959,
      "learning_rate": 0.0001,
      "loss": 3.8591,
      "loss/crossentropy": 2.3471847772598267,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20432017743587494,
      "step": 23944
    },
    {
      "epoch": 0.47892,
      "grad_norm": 1.875,
      "grad_norm_var": 0.01253662109375,
      "learning_rate": 0.0001,
      "loss": 3.9816,
      "loss/crossentropy": 1.9720887541770935,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19780997186899185,
      "step": 23946
    },
    {
      "epoch": 0.47896,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.011124420166015624,
      "learning_rate": 0.0001,
      "loss": 4.2149,
      "loss/crossentropy": 2.311566114425659,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22458520531654358,
      "step": 23948
    },
    {
      "epoch": 0.479,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.011009724934895833,
      "learning_rate": 0.0001,
      "loss": 3.7254,
      "loss/crossentropy": 1.9993168711662292,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1858225166797638,
      "step": 23950
    },
    {
      "epoch": 0.47904,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.010343170166015625,
      "learning_rate": 0.0001,
      "loss": 3.8156,
      "loss/crossentropy": 2.0514512062072754,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19098126888275146,
      "step": 23952
    },
    {
      "epoch": 0.47908,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.013248443603515625,
      "learning_rate": 0.0001,
      "loss": 3.6999,
      "loss/crossentropy": 1.6041250824928284,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15559390932321548,
      "step": 23954
    },
    {
      "epoch": 0.47912,
      "grad_norm": 1.875,
      "grad_norm_var": 0.0064117431640625,
      "learning_rate": 0.0001,
      "loss": 3.946,
      "loss/crossentropy": 1.6981446146965027,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16395021975040436,
      "step": 23956
    },
    {
      "epoch": 0.47916,
      "grad_norm": 1.875,
      "grad_norm_var": 0.007368977864583333,
      "learning_rate": 0.0001,
      "loss": 3.9571,
      "loss/crossentropy": 1.6894282102584839,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18198157846927643,
      "step": 23958
    },
    {
      "epoch": 0.4792,
      "grad_norm": 2.875,
      "grad_norm_var": 0.06331278483072916,
      "learning_rate": 0.0001,
      "loss": 4.1842,
      "loss/crossentropy": 2.1354450583457947,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2053254321217537,
      "step": 23960
    },
    {
      "epoch": 0.47924,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.06447652180989584,
      "learning_rate": 0.0001,
      "loss": 3.6665,
      "loss/crossentropy": 1.8701319098472595,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1739087551832199,
      "step": 23962
    },
    {
      "epoch": 0.47928,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.06447347005208333,
      "learning_rate": 0.0001,
      "loss": 4.0016,
      "loss/crossentropy": 2.1540639400482178,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1914234757423401,
      "step": 23964
    },
    {
      "epoch": 0.47932,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.062170155843098956,
      "learning_rate": 0.0001,
      "loss": 3.9495,
      "loss/crossentropy": 1.6824876070022583,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17504771798849106,
      "step": 23966
    },
    {
      "epoch": 0.47936,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.06093114217122396,
      "learning_rate": 0.0001,
      "loss": 4.0193,
      "loss/crossentropy": 1.893419086933136,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18423987925052643,
      "step": 23968
    },
    {
      "epoch": 0.4794,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.0600738525390625,
      "learning_rate": 0.0001,
      "loss": 4.1371,
      "loss/crossentropy": 1.759018063545227,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1881185621023178,
      "step": 23970
    },
    {
      "epoch": 0.47944,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.062263743082682295,
      "learning_rate": 0.0001,
      "loss": 4.0756,
      "loss/crossentropy": 2.1234898567199707,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2142435535788536,
      "step": 23972
    },
    {
      "epoch": 0.47948,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.06409403483072916,
      "learning_rate": 0.0001,
      "loss": 3.9597,
      "loss/crossentropy": 2.260958671569824,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19538094103336334,
      "step": 23974
    },
    {
      "epoch": 0.47952,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0109039306640625,
      "learning_rate": 0.0001,
      "loss": 4.0813,
      "loss/crossentropy": 2.004380762577057,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1982579454779625,
      "step": 23976
    },
    {
      "epoch": 0.47956,
      "grad_norm": 1.7421875,
      "grad_norm_var": 0.012544759114583333,
      "learning_rate": 0.0001,
      "loss": 3.9318,
      "loss/crossentropy": 2.038970112800598,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19289468228816986,
      "step": 23978
    },
    {
      "epoch": 0.4796,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.013361612955729166,
      "learning_rate": 0.0001,
      "loss": 4.0707,
      "loss/crossentropy": 2.291250705718994,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19521041214466095,
      "step": 23980
    },
    {
      "epoch": 0.47964,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.013537343343098958,
      "learning_rate": 0.0001,
      "loss": 4.2081,
      "loss/crossentropy": 2.03319388628006,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1981820985674858,
      "step": 23982
    },
    {
      "epoch": 0.47968,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.017899322509765624,
      "learning_rate": 0.0001,
      "loss": 4.259,
      "loss/crossentropy": 2.3277642726898193,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22071011364459991,
      "step": 23984
    },
    {
      "epoch": 0.47972,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.017724609375,
      "learning_rate": 0.0001,
      "loss": 4.0444,
      "loss/crossentropy": 2.2068026065826416,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19094634801149368,
      "step": 23986
    },
    {
      "epoch": 0.47976,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.012475331624348959,
      "learning_rate": 0.0001,
      "loss": 4.0603,
      "loss/crossentropy": 2.3219540119171143,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21838166564702988,
      "step": 23988
    },
    {
      "epoch": 0.4798,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.012743123372395833,
      "learning_rate": 0.0001,
      "loss": 3.9701,
      "loss/crossentropy": 1.9068449139595032,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19176269322633743,
      "step": 23990
    },
    {
      "epoch": 0.47984,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.01620661417643229,
      "learning_rate": 0.0001,
      "loss": 3.8448,
      "loss/crossentropy": 1.8598286509513855,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17854613065719604,
      "step": 23992
    },
    {
      "epoch": 0.47988,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.013670857747395833,
      "learning_rate": 0.0001,
      "loss": 4.3274,
      "loss/crossentropy": 2.3834491968154907,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21013744175434113,
      "step": 23994
    },
    {
      "epoch": 0.47992,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.014009602864583333,
      "learning_rate": 0.0001,
      "loss": 3.8976,
      "loss/crossentropy": 1.9524520635604858,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17633382230997086,
      "step": 23996
    },
    {
      "epoch": 0.47996,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.0135162353515625,
      "learning_rate": 0.0001,
      "loss": 3.8024,
      "loss/crossentropy": 1.8082820773124695,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17721424251794815,
      "step": 23998
    },
    {
      "epoch": 0.48,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.0076983133951822914,
      "learning_rate": 0.0001,
      "loss": 4.0181,
      "loss/crossentropy": 2.2108744382858276,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21027880907058716,
      "step": 24000
    },
    {
      "epoch": 0.48004,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.007533518473307291,
      "learning_rate": 0.0001,
      "loss": 4.0885,
      "loss/crossentropy": 2.097791016101837,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1928793340921402,
      "step": 24002
    },
    {
      "epoch": 0.48008,
      "grad_norm": 1.875,
      "grad_norm_var": 0.008429972330729167,
      "learning_rate": 0.0001,
      "loss": 4.1388,
      "loss/crossentropy": 2.2565219402313232,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21726776659488678,
      "step": 24004
    },
    {
      "epoch": 0.48012,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.007706451416015625,
      "learning_rate": 0.0001,
      "loss": 4.0097,
      "loss/crossentropy": 2.260462522506714,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1973881646990776,
      "step": 24006
    },
    {
      "epoch": 0.48016,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.007713826497395834,
      "learning_rate": 0.0001,
      "loss": 4.4655,
      "loss/crossentropy": 2.2929039001464844,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22804804146289825,
      "step": 24008
    },
    {
      "epoch": 0.4802,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.008017730712890626,
      "learning_rate": 0.0001,
      "loss": 3.8324,
      "loss/crossentropy": 2.1556389331817627,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19846376031637192,
      "step": 24010
    },
    {
      "epoch": 0.48024,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.008514149983723959,
      "learning_rate": 0.0001,
      "loss": 4.2922,
      "loss/crossentropy": 2.2187989950180054,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21181489527225494,
      "step": 24012
    },
    {
      "epoch": 0.48028,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.007230631510416667,
      "learning_rate": 0.0001,
      "loss": 3.7714,
      "loss/crossentropy": 1.7474397420883179,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18641416728496552,
      "step": 24014
    },
    {
      "epoch": 0.48032,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.006388346354166667,
      "learning_rate": 0.0001,
      "loss": 4.2139,
      "loss/crossentropy": 2.216716170310974,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20577985048294067,
      "step": 24016
    },
    {
      "epoch": 0.48036,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.007673136393229167,
      "learning_rate": 0.0001,
      "loss": 3.9804,
      "loss/crossentropy": 2.004529654979706,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19065643101930618,
      "step": 24018
    },
    {
      "epoch": 0.4804,
      "grad_norm": 2.0,
      "grad_norm_var": 0.007054646809895833,
      "learning_rate": 0.0001,
      "loss": 4.0908,
      "loss/crossentropy": 2.1064136028289795,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20194996893405914,
      "step": 24020
    },
    {
      "epoch": 0.48044,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.008208974202473959,
      "learning_rate": 0.0001,
      "loss": 4.0358,
      "loss/crossentropy": 1.9716034531593323,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18451768904924393,
      "step": 24022
    },
    {
      "epoch": 0.48048,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0073394775390625,
      "learning_rate": 0.0001,
      "loss": 4.2538,
      "loss/crossentropy": 1.9206894636154175,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21303895115852356,
      "step": 24024
    },
    {
      "epoch": 0.48052,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.00718994140625,
      "learning_rate": 0.0001,
      "loss": 4.0227,
      "loss/crossentropy": 2.2184818983078003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.205161914229393,
      "step": 24026
    },
    {
      "epoch": 0.48056,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.061962890625,
      "learning_rate": 0.0001,
      "loss": 4.0584,
      "loss/crossentropy": 1.6741513013839722,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18154188245534897,
      "step": 24028
    },
    {
      "epoch": 0.4806,
      "grad_norm": 2.96875,
      "grad_norm_var": 0.11687418619791666,
      "learning_rate": 0.0001,
      "loss": 4.0782,
      "loss/crossentropy": 1.9249740839004517,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18673353642225266,
      "step": 24030
    },
    {
      "epoch": 0.48064,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.11699930826822917,
      "learning_rate": 0.0001,
      "loss": 3.9456,
      "loss/crossentropy": 1.7694995403289795,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18005912005901337,
      "step": 24032
    },
    {
      "epoch": 0.48068,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.11251220703125,
      "learning_rate": 0.0001,
      "loss": 4.0384,
      "loss/crossentropy": 1.797115981578827,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15852996706962585,
      "step": 24034
    },
    {
      "epoch": 0.48072,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.11373062133789062,
      "learning_rate": 0.0001,
      "loss": 4.0262,
      "loss/crossentropy": 2.2001808881759644,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20976591855287552,
      "step": 24036
    },
    {
      "epoch": 0.48076,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.11187108357747395,
      "learning_rate": 0.0001,
      "loss": 4.1885,
      "loss/crossentropy": 2.24720299243927,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19293447583913803,
      "step": 24038
    },
    {
      "epoch": 0.4808,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.11860249837239584,
      "learning_rate": 0.0001,
      "loss": 3.8017,
      "loss/crossentropy": 2.0160406827926636,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19429870694875717,
      "step": 24040
    },
    {
      "epoch": 0.48084,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.11473770141601562,
      "learning_rate": 0.0001,
      "loss": 3.7577,
      "loss/crossentropy": 1.6300169229507446,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16134048998355865,
      "step": 24042
    },
    {
      "epoch": 0.48088,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.0718505859375,
      "learning_rate": 0.0001,
      "loss": 3.7797,
      "loss/crossentropy": 1.7445127367973328,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16317147016525269,
      "step": 24044
    },
    {
      "epoch": 0.48092,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.009417470296223958,
      "learning_rate": 0.0001,
      "loss": 3.6375,
      "loss/crossentropy": 1.8674694299697876,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19713745266199112,
      "step": 24046
    },
    {
      "epoch": 0.48096,
      "grad_norm": 1.75,
      "grad_norm_var": 0.01416015625,
      "learning_rate": 0.0001,
      "loss": 3.709,
      "loss/crossentropy": 1.9868064522743225,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17037389427423477,
      "step": 24048
    },
    {
      "epoch": 0.481,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.02504450480143229,
      "learning_rate": 0.0001,
      "loss": 3.9139,
      "loss/crossentropy": 1.8376989960670471,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17913921177387238,
      "step": 24050
    },
    {
      "epoch": 0.48104,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.02818781534830729,
      "learning_rate": 0.0001,
      "loss": 3.9877,
      "loss/crossentropy": 2.100782632827759,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20613420754671097,
      "step": 24052
    },
    {
      "epoch": 0.48108,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.033176422119140625,
      "learning_rate": 0.0001,
      "loss": 4.1303,
      "loss/crossentropy": 2.20209538936615,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1972164735198021,
      "step": 24054
    },
    {
      "epoch": 0.48112,
      "grad_norm": 1.75,
      "grad_norm_var": 0.0370361328125,
      "learning_rate": 0.0001,
      "loss": 4.1346,
      "loss/crossentropy": 2.3585156202316284,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20321515947580338,
      "step": 24056
    },
    {
      "epoch": 0.48116,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.03779678344726563,
      "learning_rate": 0.0001,
      "loss": 4.083,
      "loss/crossentropy": 1.8413453698158264,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18250074982643127,
      "step": 24058
    },
    {
      "epoch": 0.4812,
      "grad_norm": 2.0,
      "grad_norm_var": 0.03765055338541667,
      "learning_rate": 0.0001,
      "loss": 3.9894,
      "loss/crossentropy": 1.975302815437317,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21386513113975525,
      "step": 24060
    },
    {
      "epoch": 0.48124,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.03631998697916667,
      "learning_rate": 0.0001,
      "loss": 4.0553,
      "loss/crossentropy": 2.029150128364563,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18910515308380127,
      "step": 24062
    },
    {
      "epoch": 0.48128,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.031648508707682294,
      "learning_rate": 0.0001,
      "loss": 3.9889,
      "loss/crossentropy": 2.030007481575012,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18395943939685822,
      "step": 24064
    },
    {
      "epoch": 0.48132,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.026610310872395834,
      "learning_rate": 0.0001,
      "loss": 3.9652,
      "loss/crossentropy": 2.0372042059898376,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20342208445072174,
      "step": 24066
    },
    {
      "epoch": 0.48136,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.0240386962890625,
      "learning_rate": 0.0001,
      "loss": 3.6604,
      "loss/crossentropy": 2.1543315649032593,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1886550337076187,
      "step": 24068
    },
    {
      "epoch": 0.4814,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.01649754842122396,
      "learning_rate": 0.0001,
      "loss": 3.8595,
      "loss/crossentropy": 2.125926434993744,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22634702920913696,
      "step": 24070
    },
    {
      "epoch": 0.48144,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.009699503580729166,
      "learning_rate": 0.0001,
      "loss": 3.879,
      "loss/crossentropy": 2.057066559791565,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19772379100322723,
      "step": 24072
    },
    {
      "epoch": 0.48148,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.010151926676432292,
      "learning_rate": 0.0001,
      "loss": 3.8897,
      "loss/crossentropy": 1.6535254120826721,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16015072911977768,
      "step": 24074
    },
    {
      "epoch": 0.48152,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.009471638997395834,
      "learning_rate": 0.0001,
      "loss": 3.8221,
      "loss/crossentropy": 1.9959819912910461,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19117611646652222,
      "step": 24076
    },
    {
      "epoch": 0.48156,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.008251698811848958,
      "learning_rate": 0.0001,
      "loss": 3.9646,
      "loss/crossentropy": 1.8243607878684998,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17835189402103424,
      "step": 24078
    },
    {
      "epoch": 0.4816,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.009779612223307291,
      "learning_rate": 0.0001,
      "loss": 3.8842,
      "loss/crossentropy": 2.0300013422966003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20206968486309052,
      "step": 24080
    },
    {
      "epoch": 0.48164,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0050771077473958336,
      "learning_rate": 0.0001,
      "loss": 3.8706,
      "loss/crossentropy": 1.9207960963249207,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.195884570479393,
      "step": 24082
    },
    {
      "epoch": 0.48168,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.004903157552083333,
      "learning_rate": 0.0001,
      "loss": 3.8254,
      "loss/crossentropy": 1.7780911922454834,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16665860265493393,
      "step": 24084
    },
    {
      "epoch": 0.48172,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.004811350504557292,
      "learning_rate": 0.0001,
      "loss": 4.0063,
      "loss/crossentropy": 2.028922200202942,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19241196662187576,
      "step": 24086
    },
    {
      "epoch": 0.48176,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.004603830973307291,
      "learning_rate": 0.0001,
      "loss": 4.1312,
      "loss/crossentropy": 2.4450284242630005,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2077808380126953,
      "step": 24088
    },
    {
      "epoch": 0.4818,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.006769816080729167,
      "learning_rate": 0.0001,
      "loss": 4.1001,
      "loss/crossentropy": 2.0783454179763794,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20609202980995178,
      "step": 24090
    },
    {
      "epoch": 0.48184,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.006498209635416667,
      "learning_rate": 0.0001,
      "loss": 3.8235,
      "loss/crossentropy": 1.4834896326065063,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1598825305700302,
      "step": 24092
    },
    {
      "epoch": 0.48188,
      "grad_norm": 2.25,
      "grad_norm_var": 0.013226064046223958,
      "learning_rate": 0.0001,
      "loss": 3.9492,
      "loss/crossentropy": 2.1462320685386658,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19592846930027008,
      "step": 24094
    },
    {
      "epoch": 0.48192,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.011735026041666667,
      "learning_rate": 0.0001,
      "loss": 4.1818,
      "loss/crossentropy": 2.197389841079712,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19548462331295013,
      "step": 24096
    },
    {
      "epoch": 0.48196,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0119781494140625,
      "learning_rate": 0.0001,
      "loss": 4.0234,
      "loss/crossentropy": 2.1164830923080444,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1923857405781746,
      "step": 24098
    },
    {
      "epoch": 0.482,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.011234283447265625,
      "learning_rate": 0.0001,
      "loss": 3.9165,
      "loss/crossentropy": 2.2651044130325317,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23053671419620514,
      "step": 24100
    },
    {
      "epoch": 0.48204,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.013155110677083333,
      "learning_rate": 0.0001,
      "loss": 4.0934,
      "loss/crossentropy": 2.0696656107902527,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19399120658636093,
      "step": 24102
    },
    {
      "epoch": 0.48208,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.012741851806640624,
      "learning_rate": 0.0001,
      "loss": 3.9432,
      "loss/crossentropy": 2.0894209146499634,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19454246014356613,
      "step": 24104
    },
    {
      "epoch": 0.48212,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.011844889322916666,
      "learning_rate": 0.0001,
      "loss": 3.7906,
      "loss/crossentropy": 1.9821223020553589,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19387958943843842,
      "step": 24106
    },
    {
      "epoch": 0.48216,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.010949452718098959,
      "learning_rate": 0.0001,
      "loss": 4.2146,
      "loss/crossentropy": 2.172493100166321,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20457924157381058,
      "step": 24108
    },
    {
      "epoch": 0.4822,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.009740956624348958,
      "learning_rate": 0.0001,
      "loss": 3.8386,
      "loss/crossentropy": 2.0507744550704956,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1748208999633789,
      "step": 24110
    },
    {
      "epoch": 0.48224,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.01253662109375,
      "learning_rate": 0.0001,
      "loss": 3.6145,
      "loss/crossentropy": 1.9859300255775452,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1867031380534172,
      "step": 24112
    },
    {
      "epoch": 0.48228,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0148193359375,
      "learning_rate": 0.0001,
      "loss": 4.089,
      "loss/crossentropy": 2.1157150268554688,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.208559550344944,
      "step": 24114
    },
    {
      "epoch": 0.48232,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.015869140625,
      "learning_rate": 0.0001,
      "loss": 4.0411,
      "loss/crossentropy": 2.0446948409080505,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21388032287359238,
      "step": 24116
    },
    {
      "epoch": 0.48236,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0132232666015625,
      "learning_rate": 0.0001,
      "loss": 4.0558,
      "loss/crossentropy": 2.1047087907791138,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18818341195583344,
      "step": 24118
    },
    {
      "epoch": 0.4824,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.013605753580729166,
      "learning_rate": 0.0001,
      "loss": 4.2172,
      "loss/crossentropy": 2.235207200050354,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22523081302642822,
      "step": 24120
    },
    {
      "epoch": 0.48244,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.012068430582682291,
      "learning_rate": 0.0001,
      "loss": 4.0648,
      "loss/crossentropy": 2.4851995706558228,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19662421196699142,
      "step": 24122
    },
    {
      "epoch": 0.48248,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.012980143229166666,
      "learning_rate": 0.0001,
      "loss": 3.9836,
      "loss/crossentropy": 1.8986297249794006,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18801461160182953,
      "step": 24124
    },
    {
      "epoch": 0.48252,
      "grad_norm": 1.703125,
      "grad_norm_var": 0.0140045166015625,
      "learning_rate": 0.0001,
      "loss": 3.8972,
      "loss/crossentropy": 1.7900904417037964,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16809770464897156,
      "step": 24126
    },
    {
      "epoch": 0.48256,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.012914021809895834,
      "learning_rate": 0.0001,
      "loss": 3.8146,
      "loss/crossentropy": 2.057170867919922,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19056524336338043,
      "step": 24128
    },
    {
      "epoch": 0.4826,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.009600575764973958,
      "learning_rate": 0.0001,
      "loss": 4.0454,
      "loss/crossentropy": 2.0708820819854736,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18739734590053558,
      "step": 24130
    },
    {
      "epoch": 0.48264,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009989166259765625,
      "learning_rate": 0.0001,
      "loss": 4.1356,
      "loss/crossentropy": 1.9330047965049744,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18113382160663605,
      "step": 24132
    },
    {
      "epoch": 0.48268,
      "grad_norm": 2.0,
      "grad_norm_var": 0.010178375244140624,
      "learning_rate": 0.0001,
      "loss": 3.9655,
      "loss/crossentropy": 2.076119899749756,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1978432536125183,
      "step": 24134
    },
    {
      "epoch": 0.48272,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010994211832682291,
      "learning_rate": 0.0001,
      "loss": 4.0518,
      "loss/crossentropy": 1.9721894264221191,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19195114821195602,
      "step": 24136
    },
    {
      "epoch": 0.48276,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.010514322916666667,
      "learning_rate": 0.0001,
      "loss": 4.0326,
      "loss/crossentropy": 2.2981468439102173,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2356787919998169,
      "step": 24138
    },
    {
      "epoch": 0.4828,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.014104970296223958,
      "learning_rate": 0.0001,
      "loss": 4.0693,
      "loss/crossentropy": 2.1552239656448364,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20404700934886932,
      "step": 24140
    },
    {
      "epoch": 0.48284,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.010379791259765625,
      "learning_rate": 0.0001,
      "loss": 4.0906,
      "loss/crossentropy": 2.1359177231788635,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20384428650140762,
      "step": 24142
    },
    {
      "epoch": 0.48288,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0084381103515625,
      "learning_rate": 0.0001,
      "loss": 4.2397,
      "loss/crossentropy": 2.340220808982849,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20727048069238663,
      "step": 24144
    },
    {
      "epoch": 0.48292,
      "grad_norm": 1.875,
      "grad_norm_var": 0.009702301025390625,
      "learning_rate": 0.0001,
      "loss": 3.6822,
      "loss/crossentropy": 1.7174909114837646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16191110014915466,
      "step": 24146
    },
    {
      "epoch": 0.48296,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.010876210530598958,
      "learning_rate": 0.0001,
      "loss": 3.8731,
      "loss/crossentropy": 2.0727250576019287,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.182908833026886,
      "step": 24148
    },
    {
      "epoch": 0.483,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0141357421875,
      "learning_rate": 0.0001,
      "loss": 4.2179,
      "loss/crossentropy": 2.073746085166931,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22062638401985168,
      "step": 24150
    },
    {
      "epoch": 0.48304,
      "grad_norm": 4.59375,
      "grad_norm_var": 0.4433143615722656,
      "learning_rate": 0.0001,
      "loss": 3.8633,
      "loss/crossentropy": 1.7323468327522278,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16554639488458633,
      "step": 24152
    },
    {
      "epoch": 0.48308,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.4394365946451823,
      "learning_rate": 0.0001,
      "loss": 4.0216,
      "loss/crossentropy": 2.136180579662323,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.192221999168396,
      "step": 24154
    },
    {
      "epoch": 0.48312,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.4390459696451823,
      "learning_rate": 0.0001,
      "loss": 4.0375,
      "loss/crossentropy": 1.8351938128471375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17663314193487167,
      "step": 24156
    },
    {
      "epoch": 0.48316,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.44324722290039065,
      "learning_rate": 0.0001,
      "loss": 3.9927,
      "loss/crossentropy": 2.02448707818985,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19494900107383728,
      "step": 24158
    },
    {
      "epoch": 0.4832,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.4480865478515625,
      "learning_rate": 0.0001,
      "loss": 3.9197,
      "loss/crossentropy": 2.185999810695648,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1869122013449669,
      "step": 24160
    },
    {
      "epoch": 0.48324,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.4416168212890625,
      "learning_rate": 0.0001,
      "loss": 4.2421,
      "loss/crossentropy": 2.16790235042572,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19200150668621063,
      "step": 24162
    },
    {
      "epoch": 0.48328,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.44467137654622396,
      "learning_rate": 0.0001,
      "loss": 3.7496,
      "loss/crossentropy": 1.5323420763015747,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.14329544454813004,
      "step": 24164
    },
    {
      "epoch": 0.48332,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.4490631103515625,
      "learning_rate": 0.0001,
      "loss": 4.1271,
      "loss/crossentropy": 2.4106760025024414,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2088209092617035,
      "step": 24166
    },
    {
      "epoch": 0.48336,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0195220947265625,
      "learning_rate": 0.0001,
      "loss": 4.1251,
      "loss/crossentropy": 2.2519407272338867,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21344104409217834,
      "step": 24168
    },
    {
      "epoch": 0.4834,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.016771443684895835,
      "learning_rate": 0.0001,
      "loss": 4.129,
      "loss/crossentropy": 2.4721094369888306,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20769134163856506,
      "step": 24170
    },
    {
      "epoch": 0.48344,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.01695734659830729,
      "learning_rate": 0.0001,
      "loss": 4.0163,
      "loss/crossentropy": 1.794982135295868,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18507220596075058,
      "step": 24172
    },
    {
      "epoch": 0.48348,
      "grad_norm": 2.0,
      "grad_norm_var": 0.01649958292643229,
      "learning_rate": 0.0001,
      "loss": 4.0596,
      "loss/crossentropy": 2.086494565010071,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1993468999862671,
      "step": 24174
    },
    {
      "epoch": 0.48352,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.016112263997395834,
      "learning_rate": 0.0001,
      "loss": 4.2193,
      "loss/crossentropy": 1.9496687054634094,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19727850705385208,
      "step": 24176
    },
    {
      "epoch": 0.48356,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.005777740478515625,
      "learning_rate": 0.0001,
      "loss": 4.1632,
      "loss/crossentropy": 2.0036423206329346,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1781500205397606,
      "step": 24178
    },
    {
      "epoch": 0.4836,
      "grad_norm": 2.0,
      "grad_norm_var": 0.013263956705729166,
      "learning_rate": 0.0001,
      "loss": 3.9156,
      "loss/crossentropy": 2.1094807386398315,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1822209656238556,
      "step": 24180
    },
    {
      "epoch": 0.48364,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.013722483317057292,
      "learning_rate": 0.0001,
      "loss": 3.9997,
      "loss/crossentropy": 2.0273566246032715,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21027785539627075,
      "step": 24182
    },
    {
      "epoch": 0.48368,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.014705149332682292,
      "learning_rate": 0.0001,
      "loss": 3.9781,
      "loss/crossentropy": 1.9223063588142395,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1821991503238678,
      "step": 24184
    },
    {
      "epoch": 0.48372,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.015067545572916667,
      "learning_rate": 0.0001,
      "loss": 4.0162,
      "loss/crossentropy": 1.7469323873519897,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19376590102910995,
      "step": 24186
    },
    {
      "epoch": 0.48376,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.017545572916666665,
      "learning_rate": 0.0001,
      "loss": 3.7141,
      "loss/crossentropy": 1.7850856184959412,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17863896489143372,
      "step": 24188
    },
    {
      "epoch": 0.4838,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.01910985310872396,
      "learning_rate": 0.0001,
      "loss": 4.009,
      "loss/crossentropy": 1.7316967844963074,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16793282330036163,
      "step": 24190
    },
    {
      "epoch": 0.48384,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.02093480428059896,
      "learning_rate": 0.0001,
      "loss": 4.0243,
      "loss/crossentropy": 2.3408809900283813,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21008720248937607,
      "step": 24192
    },
    {
      "epoch": 0.48388,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.02029393513997396,
      "learning_rate": 0.0001,
      "loss": 3.7861,
      "loss/crossentropy": 2.0060762763023376,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1912834793329239,
      "step": 24194
    },
    {
      "epoch": 0.48392,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.011189778645833334,
      "learning_rate": 0.0001,
      "loss": 4.1224,
      "loss/crossentropy": 2.1604164838790894,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1941056326031685,
      "step": 24196
    },
    {
      "epoch": 0.48396,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.010371907552083334,
      "learning_rate": 0.0001,
      "loss": 4.2158,
      "loss/crossentropy": 2.0909000635147095,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20140909403562546,
      "step": 24198
    },
    {
      "epoch": 0.484,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.009039052327473958,
      "learning_rate": 0.0001,
      "loss": 3.94,
      "loss/crossentropy": 1.8877050876617432,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18381226062774658,
      "step": 24200
    },
    {
      "epoch": 0.48404,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.008721669514973959,
      "learning_rate": 0.0001,
      "loss": 3.9384,
      "loss/crossentropy": 1.982464611530304,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18099818378686905,
      "step": 24202
    },
    {
      "epoch": 0.48408,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.006481679280598959,
      "learning_rate": 0.0001,
      "loss": 4.1044,
      "loss/crossentropy": 1.9599023461341858,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18184300512075424,
      "step": 24204
    },
    {
      "epoch": 0.48412,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.005110422770182292,
      "learning_rate": 0.0001,
      "loss": 4.1949,
      "loss/crossentropy": 2.4100862741470337,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2156199961900711,
      "step": 24206
    },
    {
      "epoch": 0.48416,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0038653055826822916,
      "learning_rate": 0.0001,
      "loss": 4.4761,
      "loss/crossentropy": 2.2121779322624207,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23106029629707336,
      "step": 24208
    },
    {
      "epoch": 0.4842,
      "grad_norm": 1.6875,
      "grad_norm_var": 0.007331339518229166,
      "learning_rate": 0.0001,
      "loss": 3.8548,
      "loss/crossentropy": 1.9747964143753052,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17455366998910904,
      "step": 24210
    },
    {
      "epoch": 0.48424,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.010008748372395833,
      "learning_rate": 0.0001,
      "loss": 4.3493,
      "loss/crossentropy": 2.3463014364242554,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21605894714593887,
      "step": 24212
    },
    {
      "epoch": 0.48428,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.012626139322916667,
      "learning_rate": 0.0001,
      "loss": 4.1674,
      "loss/crossentropy": 2.1442391872406006,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2003452554345131,
      "step": 24214
    },
    {
      "epoch": 0.48432,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.015803019205729168,
      "learning_rate": 0.0001,
      "loss": 4.2701,
      "loss/crossentropy": 2.281853437423706,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21563008427619934,
      "step": 24216
    },
    {
      "epoch": 0.48436,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.01668065388997396,
      "learning_rate": 0.0001,
      "loss": 4.3137,
      "loss/crossentropy": 2.303357243537903,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23583707213401794,
      "step": 24218
    },
    {
      "epoch": 0.4844,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.016649373372395835,
      "learning_rate": 0.0001,
      "loss": 4.1768,
      "loss/crossentropy": 1.6764637231826782,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1820378378033638,
      "step": 24220
    },
    {
      "epoch": 0.48444,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.01702855428059896,
      "learning_rate": 0.0001,
      "loss": 4.4549,
      "loss/crossentropy": 2.3033594489097595,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21953578293323517,
      "step": 24222
    },
    {
      "epoch": 0.48448,
      "grad_norm": 2.0,
      "grad_norm_var": 0.017354329427083332,
      "learning_rate": 0.0001,
      "loss": 4.0358,
      "loss/crossentropy": 2.1051379442214966,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19566316902637482,
      "step": 24224
    },
    {
      "epoch": 0.48452,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009997304280598958,
      "learning_rate": 0.0001,
      "loss": 4.0489,
      "loss/crossentropy": 2.122231125831604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20219532400369644,
      "step": 24226
    },
    {
      "epoch": 0.48456,
      "grad_norm": 1.7265625,
      "grad_norm_var": 0.013974761962890625,
      "learning_rate": 0.0001,
      "loss": 3.8343,
      "loss/crossentropy": 1.6727787256240845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15022944658994675,
      "step": 24228
    },
    {
      "epoch": 0.4846,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.012410227457682292,
      "learning_rate": 0.0001,
      "loss": 4.1034,
      "loss/crossentropy": 2.0252939462661743,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2014101818203926,
      "step": 24230
    },
    {
      "epoch": 0.48464,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.011213175455729167,
      "learning_rate": 0.0001,
      "loss": 4.003,
      "loss/crossentropy": 2.163161873817444,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20025929063558578,
      "step": 24232
    },
    {
      "epoch": 0.48468,
      "grad_norm": 1.75,
      "grad_norm_var": 0.010895792643229167,
      "learning_rate": 0.0001,
      "loss": 3.7811,
      "loss/crossentropy": 1.9188687205314636,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1916854828596115,
      "step": 24234
    },
    {
      "epoch": 0.48472,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.009870402018229167,
      "learning_rate": 0.0001,
      "loss": 3.9804,
      "loss/crossentropy": 1.820820927619934,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1769949272274971,
      "step": 24236
    },
    {
      "epoch": 0.48476,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.009297434488932292,
      "learning_rate": 0.0001,
      "loss": 3.9483,
      "loss/crossentropy": 2.1655235290527344,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2005934715270996,
      "step": 24238
    },
    {
      "epoch": 0.4848,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.009810384114583333,
      "learning_rate": 0.0001,
      "loss": 4.0458,
      "loss/crossentropy": 1.9468095898628235,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1830679327249527,
      "step": 24240
    },
    {
      "epoch": 0.48484,
      "grad_norm": 2.0,
      "grad_norm_var": 0.009535471598307291,
      "learning_rate": 0.0001,
      "loss": 4.0301,
      "loss/crossentropy": 2.062897562980652,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2143079787492752,
      "step": 24242
    },
    {
      "epoch": 0.48488,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.010389963785807291,
      "learning_rate": 0.0001,
      "loss": 4.1204,
      "loss/crossentropy": 2.0829350352287292,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19584693014621735,
      "step": 24244
    },
    {
      "epoch": 0.48492,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0102447509765625,
      "learning_rate": 0.0001,
      "loss": 3.9724,
      "loss/crossentropy": 2.0563968420028687,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20057959854602814,
      "step": 24246
    },
    {
      "epoch": 0.48496,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.013396962483723959,
      "learning_rate": 0.0001,
      "loss": 4.2274,
      "loss/crossentropy": 2.1221665143966675,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20288699865341187,
      "step": 24248
    },
    {
      "epoch": 0.485,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.010374959309895833,
      "learning_rate": 0.0001,
      "loss": 4.2162,
      "loss/crossentropy": 2.1751914024353027,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20034758746623993,
      "step": 24250
    },
    {
      "epoch": 0.48504,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.010872395833333333,
      "learning_rate": 0.0001,
      "loss": 3.8777,
      "loss/crossentropy": 2.0292373299598694,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18800581246614456,
      "step": 24252
    },
    {
      "epoch": 0.48508,
      "grad_norm": 4.03125,
      "grad_norm_var": 0.275439453125,
      "learning_rate": 0.0001,
      "loss": 4.0572,
      "loss/crossentropy": 2.3077808618545532,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23775576055049896,
      "step": 24254
    },
    {
      "epoch": 0.48512,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.27331441243489585,
      "learning_rate": 0.0001,
      "loss": 4.0708,
      "loss/crossentropy": 1.532023847103119,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1606956273317337,
      "step": 24256
    },
    {
      "epoch": 0.48516,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.26945699055989586,
      "learning_rate": 0.0001,
      "loss": 4.0351,
      "loss/crossentropy": 2.5032970905303955,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21631880104541779,
      "step": 24258
    },
    {
      "epoch": 0.4852,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.27319234212239585,
      "learning_rate": 0.0001,
      "loss": 3.9323,
      "loss/crossentropy": 1.981030285358429,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19134312123060226,
      "step": 24260
    },
    {
      "epoch": 0.48524,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.2728668212890625,
      "learning_rate": 0.0001,
      "loss": 3.7619,
      "loss/crossentropy": 2.031146466732025,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17859568446874619,
      "step": 24262
    },
    {
      "epoch": 0.48528,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.2775917053222656,
      "learning_rate": 0.0001,
      "loss": 4.0923,
      "loss/crossentropy": 2.5168176889419556,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21798798441886902,
      "step": 24264
    },
    {
      "epoch": 0.48532,
      "grad_norm": 4.09375,
      "grad_norm_var": 0.530810292561849,
      "learning_rate": 0.0001,
      "loss": 3.9951,
      "loss/crossentropy": 2.037827789783478,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2093580812215805,
      "step": 24266
    },
    {
      "epoch": 0.48536,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.5302874247233073,
      "learning_rate": 0.0001,
      "loss": 4.0575,
      "loss/crossentropy": 2.17299222946167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18525302410125732,
      "step": 24268
    },
    {
      "epoch": 0.4854,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.2958106994628906,
      "learning_rate": 0.0001,
      "loss": 3.9348,
      "loss/crossentropy": 2.147997260093689,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20097453892230988,
      "step": 24270
    },
    {
      "epoch": 0.48544,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.298583984375,
      "learning_rate": 0.0001,
      "loss": 4.1208,
      "loss/crossentropy": 2.220878005027771,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18916144967079163,
      "step": 24272
    },
    {
      "epoch": 0.48548,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.297265625,
      "learning_rate": 0.0001,
      "loss": 4.0689,
      "loss/crossentropy": 2.3884263038635254,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21515949070453644,
      "step": 24274
    },
    {
      "epoch": 0.48552,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.2965349833170573,
      "learning_rate": 0.0001,
      "loss": 4.091,
      "loss/crossentropy": 2.0658962726593018,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19635243713855743,
      "step": 24276
    },
    {
      "epoch": 0.48556,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.297857411702474,
      "learning_rate": 0.0001,
      "loss": 4.2023,
      "loss/crossentropy": 2.3192285299301147,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22074441611766815,
      "step": 24278
    },
    {
      "epoch": 0.4856,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.30143941243489586,
      "learning_rate": 0.0001,
      "loss": 3.8625,
      "loss/crossentropy": 2.082835614681244,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18948380649089813,
      "step": 24280
    },
    {
      "epoch": 0.48564,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.009039052327473958,
      "learning_rate": 0.0001,
      "loss": 3.8998,
      "loss/crossentropy": 1.7844191193580627,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17573316395282745,
      "step": 24282
    },
    {
      "epoch": 0.48568,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.009161122639973958,
      "learning_rate": 0.0001,
      "loss": 3.8169,
      "loss/crossentropy": 1.9186521768569946,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1827767789363861,
      "step": 24284
    },
    {
      "epoch": 0.48572,
      "grad_norm": 2.296875,
      "grad_norm_var": 3.559098307291667,
      "learning_rate": 0.0001,
      "loss": 4.1294,
      "loss/crossentropy": 2.0445513129234314,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22182191163301468,
      "step": 24286
    },
    {
      "epoch": 0.48576,
      "grad_norm": 2.140625,
      "grad_norm_var": 3.535992177327474,
      "learning_rate": 0.0001,
      "loss": 3.9131,
      "loss/crossentropy": 2.1303210258483887,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20597820729017258,
      "step": 24288
    },
    {
      "epoch": 0.4858,
      "grad_norm": 1.96875,
      "grad_norm_var": 3.544976552327474,
      "learning_rate": 0.0001,
      "loss": 4.0554,
      "loss/crossentropy": 1.9035282135009766,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17603804916143417,
      "step": 24290
    },
    {
      "epoch": 0.48584,
      "grad_norm": 2.109375,
      "grad_norm_var": 3.5285723368326822,
      "learning_rate": 0.0001,
      "loss": 4.0373,
      "loss/crossentropy": 2.101797640323639,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19533278793096542,
      "step": 24292
    },
    {
      "epoch": 0.48588,
      "grad_norm": 1.875,
      "grad_norm_var": 3.5240071614583335,
      "learning_rate": 0.0001,
      "loss": 3.9883,
      "loss/crossentropy": 1.9595980048179626,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17600257694721222,
      "step": 24294
    },
    {
      "epoch": 0.48592,
      "grad_norm": 1.84375,
      "grad_norm_var": 3.506517537434896,
      "learning_rate": 0.0001,
      "loss": 3.8312,
      "loss/crossentropy": 2.020717740058899,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19629748165607452,
      "step": 24296
    },
    {
      "epoch": 0.48596,
      "grad_norm": 1.875,
      "grad_norm_var": 3.5107137044270833,
      "learning_rate": 0.0001,
      "loss": 4.4606,
      "loss/crossentropy": 2.113112688064575,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1987774595618248,
      "step": 24298
    },
    {
      "epoch": 0.486,
      "grad_norm": 1.984375,
      "grad_norm_var": 3.4833943684895834,
      "learning_rate": 0.0001,
      "loss": 4.3874,
      "loss/crossentropy": 2.3140907287597656,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2087773084640503,
      "step": 24300
    },
    {
      "epoch": 0.48604,
      "grad_norm": 2.0,
      "grad_norm_var": 0.012322743733723959,
      "learning_rate": 0.0001,
      "loss": 4.1962,
      "loss/crossentropy": 2.137213349342346,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19433750957250595,
      "step": 24302
    },
    {
      "epoch": 0.48608,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.015026601155598958,
      "learning_rate": 0.0001,
      "loss": 3.7098,
      "loss/crossentropy": 1.8572564125061035,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18543125689029694,
      "step": 24304
    },
    {
      "epoch": 0.48612,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.017389933268229168,
      "learning_rate": 0.0001,
      "loss": 3.8251,
      "loss/crossentropy": 1.991189181804657,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1831136792898178,
      "step": 24306
    },
    {
      "epoch": 0.48616,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.016901652018229168,
      "learning_rate": 0.0001,
      "loss": 3.9742,
      "loss/crossentropy": 1.8540424704551697,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1730886772274971,
      "step": 24308
    },
    {
      "epoch": 0.4862,
      "grad_norm": 1.875,
      "grad_norm_var": 0.016747792561848957,
      "learning_rate": 0.0001,
      "loss": 3.668,
      "loss/crossentropy": 2.0189526677131653,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1916787549853325,
      "step": 24310
    },
    {
      "epoch": 0.48624,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.016071573893229166,
      "learning_rate": 0.0001,
      "loss": 4.0621,
      "loss/crossentropy": 2.2053693532943726,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20035792887210846,
      "step": 24312
    },
    {
      "epoch": 0.48628,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.012189737955729167,
      "learning_rate": 0.0001,
      "loss": 4.0676,
      "loss/crossentropy": 2.2726696729660034,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21893906593322754,
      "step": 24314
    },
    {
      "epoch": 0.48632,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.006940714518229167,
      "learning_rate": 0.0001,
      "loss": 3.6041,
      "loss/crossentropy": 1.9184311032295227,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18804488331079483,
      "step": 24316
    },
    {
      "epoch": 0.48636,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.00714111328125,
      "learning_rate": 0.0001,
      "loss": 4.2603,
      "loss/crossentropy": 2.416733145713806,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21675719320774078,
      "step": 24318
    },
    {
      "epoch": 0.4864,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.00570068359375,
      "learning_rate": 0.0001,
      "loss": 4.1108,
      "loss/crossentropy": 2.2064080238342285,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20753097534179688,
      "step": 24320
    },
    {
      "epoch": 0.48644,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.005324045817057292,
      "learning_rate": 0.0001,
      "loss": 4.1638,
      "loss/crossentropy": 1.9959591031074524,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18936610966920853,
      "step": 24322
    },
    {
      "epoch": 0.48648,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.004613240559895833,
      "learning_rate": 0.0001,
      "loss": 4.0866,
      "loss/crossentropy": 2.14048969745636,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17960359901189804,
      "step": 24324
    },
    {
      "epoch": 0.48652,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.006845855712890625,
      "learning_rate": 0.0001,
      "loss": 4.1387,
      "loss/crossentropy": 1.6789081692695618,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1848226636648178,
      "step": 24326
    },
    {
      "epoch": 0.48656,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0069620768229166664,
      "learning_rate": 0.0001,
      "loss": 4.2576,
      "loss/crossentropy": 2.4790940284729004,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22105763852596283,
      "step": 24328
    },
    {
      "epoch": 0.4866,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.006981404622395834,
      "learning_rate": 0.0001,
      "loss": 4.1167,
      "loss/crossentropy": 1.8698575496673584,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18414634466171265,
      "step": 24330
    },
    {
      "epoch": 0.48664,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0045969645182291664,
      "learning_rate": 0.0001,
      "loss": 4.1848,
      "loss/crossentropy": 2.1648661494255066,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2052561342716217,
      "step": 24332
    },
    {
      "epoch": 0.48668,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.0054107666015625,
      "learning_rate": 0.0001,
      "loss": 3.8961,
      "loss/crossentropy": 1.9554332494735718,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2595558241009712,
      "step": 24334
    },
    {
      "epoch": 0.48672,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.005208079020182292,
      "learning_rate": 0.0001,
      "loss": 3.8727,
      "loss/crossentropy": 2.120553970336914,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20485319197177887,
      "step": 24336
    },
    {
      "epoch": 0.48676,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.006046295166015625,
      "learning_rate": 0.0001,
      "loss": 3.849,
      "loss/crossentropy": 1.7237398028373718,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2036430537700653,
      "step": 24338
    },
    {
      "epoch": 0.4868,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.0088043212890625,
      "learning_rate": 0.0001,
      "loss": 3.9799,
      "loss/crossentropy": 1.7713853120803833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1712682917714119,
      "step": 24340
    },
    {
      "epoch": 0.48684,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.00877685546875,
      "learning_rate": 0.0001,
      "loss": 3.6067,
      "loss/crossentropy": 2.0136027932167053,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18176132440567017,
      "step": 24342
    },
    {
      "epoch": 0.48688,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.007478841145833333,
      "learning_rate": 0.0001,
      "loss": 4.0666,
      "loss/crossentropy": 1.9793068766593933,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1973177120089531,
      "step": 24344
    },
    {
      "epoch": 0.48692,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.011165364583333334,
      "learning_rate": 0.0001,
      "loss": 4.3935,
      "loss/crossentropy": 2.4376041889190674,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22222332656383514,
      "step": 24346
    },
    {
      "epoch": 0.48696,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.010448201497395834,
      "learning_rate": 0.0001,
      "loss": 3.9815,
      "loss/crossentropy": 1.8432115316390991,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1915820688009262,
      "step": 24348
    },
    {
      "epoch": 0.487,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.010530344645182292,
      "learning_rate": 0.0001,
      "loss": 3.8947,
      "loss/crossentropy": 1.8100075125694275,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18175233900547028,
      "step": 24350
    },
    {
      "epoch": 0.48704,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0118072509765625,
      "learning_rate": 0.0001,
      "loss": 4.0207,
      "loss/crossentropy": 2.0997090339660645,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19269437342882156,
      "step": 24352
    },
    {
      "epoch": 0.48708,
      "grad_norm": 2.125,
      "grad_norm_var": 0.013529205322265625,
      "learning_rate": 0.0001,
      "loss": 3.9769,
      "loss/crossentropy": 1.78102046251297,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18288961052894592,
      "step": 24354
    },
    {
      "epoch": 0.48712,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.012227121988932292,
      "learning_rate": 0.0001,
      "loss": 3.9862,
      "loss/crossentropy": 2.006900370121002,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19866903871297836,
      "step": 24356
    },
    {
      "epoch": 0.48716,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.013036855061848958,
      "learning_rate": 0.0001,
      "loss": 3.7675,
      "loss/crossentropy": 1.9674765467643738,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17953628301620483,
      "step": 24358
    },
    {
      "epoch": 0.4872,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.014772288004557292,
      "learning_rate": 0.0001,
      "loss": 3.8801,
      "loss/crossentropy": 1.6254491806030273,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16119451820850372,
      "step": 24360
    },
    {
      "epoch": 0.48724,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.010749308268229167,
      "learning_rate": 0.0001,
      "loss": 4.0639,
      "loss/crossentropy": 2.1395972967147827,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20453596115112305,
      "step": 24362
    },
    {
      "epoch": 0.48728,
      "grad_norm": 2.0,
      "grad_norm_var": 0.011022694905598958,
      "learning_rate": 0.0001,
      "loss": 4.1345,
      "loss/crossentropy": 2.0489049553871155,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2140253186225891,
      "step": 24364
    },
    {
      "epoch": 0.48732,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.011502838134765625,
      "learning_rate": 0.0001,
      "loss": 4.2531,
      "loss/crossentropy": 2.061949372291565,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2023036777973175,
      "step": 24366
    },
    {
      "epoch": 0.48736,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.008658854166666667,
      "learning_rate": 0.0001,
      "loss": 4.0565,
      "loss/crossentropy": 2.2293535470962524,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20210447907447815,
      "step": 24368
    },
    {
      "epoch": 0.4874,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006525675455729167,
      "learning_rate": 0.0001,
      "loss": 4.0853,
      "loss/crossentropy": 2.085119366645813,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19112756848335266,
      "step": 24370
    },
    {
      "epoch": 0.48744,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.012178548177083333,
      "learning_rate": 0.0001,
      "loss": 4.0025,
      "loss/crossentropy": 1.8672300577163696,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19416838884353638,
      "step": 24372
    },
    {
      "epoch": 0.48748,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.015378570556640625,
      "learning_rate": 0.0001,
      "loss": 4.0043,
      "loss/crossentropy": 1.7550761699676514,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23619835823774338,
      "step": 24374
    },
    {
      "epoch": 0.48752,
      "grad_norm": 1.75,
      "grad_norm_var": 0.017061360677083335,
      "learning_rate": 0.0001,
      "loss": 3.8778,
      "loss/crossentropy": 2.199945092201233,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.198745958507061,
      "step": 24376
    },
    {
      "epoch": 0.48756,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.01672948201497396,
      "learning_rate": 0.0001,
      "loss": 3.9917,
      "loss/crossentropy": 2.005652666091919,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20299483090639114,
      "step": 24378
    },
    {
      "epoch": 0.4876,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.01756769816080729,
      "learning_rate": 0.0001,
      "loss": 3.8204,
      "loss/crossentropy": 1.9981390237808228,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19434207677841187,
      "step": 24380
    },
    {
      "epoch": 0.48764,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.017203776041666667,
      "learning_rate": 0.0001,
      "loss": 4.0997,
      "loss/crossentropy": 1.9847465753555298,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2016771137714386,
      "step": 24382
    },
    {
      "epoch": 0.48768,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.017268880208333334,
      "learning_rate": 0.0001,
      "loss": 3.9819,
      "loss/crossentropy": 1.9587423205375671,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20353511720895767,
      "step": 24384
    },
    {
      "epoch": 0.48772,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.017919921875,
      "learning_rate": 0.0001,
      "loss": 3.639,
      "loss/crossentropy": 1.6634628176689148,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18179689347743988,
      "step": 24386
    },
    {
      "epoch": 0.48776,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.011164347330729166,
      "learning_rate": 0.0001,
      "loss": 4.0095,
      "loss/crossentropy": 1.8678483963012695,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18797649443149567,
      "step": 24388
    },
    {
      "epoch": 0.4878,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.034024810791015624,
      "learning_rate": 0.0001,
      "loss": 4.245,
      "loss/crossentropy": 2.2724742889404297,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19732308387756348,
      "step": 24390
    },
    {
      "epoch": 0.48784,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.030918121337890625,
      "learning_rate": 0.0001,
      "loss": 4.1437,
      "loss/crossentropy": 2.2534377574920654,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21449754387140274,
      "step": 24392
    },
    {
      "epoch": 0.48788,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.03006159464518229,
      "learning_rate": 0.0001,
      "loss": 4.4295,
      "loss/crossentropy": 2.3007233142852783,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21122832596302032,
      "step": 24394
    },
    {
      "epoch": 0.48792,
      "grad_norm": 1.6796875,
      "grad_norm_var": 0.036195627848307294,
      "learning_rate": 0.0001,
      "loss": 3.7791,
      "loss/crossentropy": 2.0957794189453125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19196099787950516,
      "step": 24396
    },
    {
      "epoch": 0.48796,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.03947118123372396,
      "learning_rate": 0.0001,
      "loss": 4.2137,
      "loss/crossentropy": 1.8763219118118286,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1841205433011055,
      "step": 24398
    },
    {
      "epoch": 0.488,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.03982747395833333,
      "learning_rate": 0.0001,
      "loss": 3.8093,
      "loss/crossentropy": 2.2737534046173096,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.200559563934803,
      "step": 24400
    },
    {
      "epoch": 0.48804,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.03672866821289063,
      "learning_rate": 0.0001,
      "loss": 3.9472,
      "loss/crossentropy": 2.2588294744491577,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21728847920894623,
      "step": 24402
    },
    {
      "epoch": 0.48808,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.03683039347330729,
      "learning_rate": 0.0001,
      "loss": 4.052,
      "loss/crossentropy": 2.1669589281082153,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2279760167002678,
      "step": 24404
    },
    {
      "epoch": 0.48812,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.014857737223307292,
      "learning_rate": 0.0001,
      "loss": 4.1939,
      "loss/crossentropy": 2.229967176914215,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20311905443668365,
      "step": 24406
    },
    {
      "epoch": 0.48816,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.016242472330729167,
      "learning_rate": 0.0001,
      "loss": 3.8149,
      "loss/crossentropy": 1.7735026478767395,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17489700764417648,
      "step": 24408
    },
    {
      "epoch": 0.4882,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.016880035400390625,
      "learning_rate": 0.0001,
      "loss": 4.0679,
      "loss/crossentropy": 1.6055407524108887,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16866382956504822,
      "step": 24410
    },
    {
      "epoch": 0.48824,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.017826080322265625,
      "learning_rate": 0.0001,
      "loss": 4.1063,
      "loss/crossentropy": 2.0581588745117188,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20155030488967896,
      "step": 24412
    },
    {
      "epoch": 0.48828,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.022802734375,
      "learning_rate": 0.0001,
      "loss": 3.874,
      "loss/crossentropy": 2.0038405656814575,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1854727864265442,
      "step": 24414
    },
    {
      "epoch": 0.48832,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0213134765625,
      "learning_rate": 0.0001,
      "loss": 4.4175,
      "loss/crossentropy": 2.5517786741256714,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22493450343608856,
      "step": 24416
    },
    {
      "epoch": 0.48836,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.021297200520833334,
      "learning_rate": 0.0001,
      "loss": 4.3037,
      "loss/crossentropy": 1.963084876537323,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1878989413380623,
      "step": 24418
    },
    {
      "epoch": 0.4884,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.025023396809895834,
      "learning_rate": 0.0001,
      "loss": 4.3639,
      "loss/crossentropy": 2.093212604522705,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20798558741807938,
      "step": 24420
    },
    {
      "epoch": 0.48844,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.028537750244140625,
      "learning_rate": 0.0001,
      "loss": 3.9645,
      "loss/crossentropy": 2.2910980582237244,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19769180566072464,
      "step": 24422
    },
    {
      "epoch": 0.48848,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.02447509765625,
      "learning_rate": 0.0001,
      "loss": 4.0026,
      "loss/crossentropy": 1.7930738925933838,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18997550755739212,
      "step": 24424
    },
    {
      "epoch": 0.48852,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.02550048828125,
      "learning_rate": 0.0001,
      "loss": 3.9086,
      "loss/crossentropy": 1.7596023678779602,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17135074734687805,
      "step": 24426
    },
    {
      "epoch": 0.48856,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.019059244791666666,
      "learning_rate": 0.0001,
      "loss": 4.2215,
      "loss/crossentropy": 1.6692591905593872,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1735754758119583,
      "step": 24428
    },
    {
      "epoch": 0.4886,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.016015625,
      "learning_rate": 0.0001,
      "loss": 4.1244,
      "loss/crossentropy": 1.9515159130096436,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20981048792600632,
      "step": 24430
    },
    {
      "epoch": 0.48864,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.019451649983723958,
      "learning_rate": 0.0001,
      "loss": 3.9243,
      "loss/crossentropy": 2.218432307243347,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20877013355493546,
      "step": 24432
    },
    {
      "epoch": 0.48868,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.01914240519205729,
      "learning_rate": 0.0001,
      "loss": 3.9418,
      "loss/crossentropy": 1.6877312660217285,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17738080769777298,
      "step": 24434
    },
    {
      "epoch": 0.48872,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.013427480061848959,
      "learning_rate": 0.0001,
      "loss": 4.1303,
      "loss/crossentropy": 2.204097032546997,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2016325369477272,
      "step": 24436
    },
    {
      "epoch": 0.48876,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.012330881754557292,
      "learning_rate": 0.0001,
      "loss": 3.8735,
      "loss/crossentropy": 2.287162959575653,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2044740840792656,
      "step": 24438
    },
    {
      "epoch": 0.4888,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.012743123372395833,
      "learning_rate": 0.0001,
      "loss": 4.0459,
      "loss/crossentropy": 2.0852694511413574,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2053818255662918,
      "step": 24440
    },
    {
      "epoch": 0.48884,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.017170206705729166,
      "learning_rate": 0.0001,
      "loss": 4.075,
      "loss/crossentropy": 2.070200562477112,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1872272714972496,
      "step": 24442
    },
    {
      "epoch": 0.48888,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.01678466796875,
      "learning_rate": 0.0001,
      "loss": 4.2843,
      "loss/crossentropy": 2.034148395061493,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20018816739320755,
      "step": 24444
    },
    {
      "epoch": 0.48892,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.010282389322916667,
      "learning_rate": 0.0001,
      "loss": 3.9187,
      "loss/crossentropy": 2.2302210927009583,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21468773484230042,
      "step": 24446
    },
    {
      "epoch": 0.48896,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.009584299723307292,
      "learning_rate": 0.0001,
      "loss": 3.8918,
      "loss/crossentropy": 1.9789615273475647,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18996840715408325,
      "step": 24448
    },
    {
      "epoch": 0.489,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0099761962890625,
      "learning_rate": 0.0001,
      "loss": 4.2603,
      "loss/crossentropy": 2.1200879216194153,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20140299201011658,
      "step": 24450
    },
    {
      "epoch": 0.48904,
      "grad_norm": 1.734375,
      "grad_norm_var": 0.013475545247395833,
      "learning_rate": 0.0001,
      "loss": 3.6989,
      "loss/crossentropy": 1.9175571203231812,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18238398432731628,
      "step": 24452
    },
    {
      "epoch": 0.48908,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.016527303059895835,
      "learning_rate": 0.0001,
      "loss": 4.1204,
      "loss/crossentropy": 2.414350748062134,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2311321198940277,
      "step": 24454
    },
    {
      "epoch": 0.48912,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.016267903645833335,
      "learning_rate": 0.0001,
      "loss": 4.0023,
      "loss/crossentropy": 2.113545060157776,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20183447748422623,
      "step": 24456
    },
    {
      "epoch": 0.48916,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.035123443603515624,
      "learning_rate": 0.0001,
      "loss": 4.0608,
      "loss/crossentropy": 1.9957387447357178,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.183005228638649,
      "step": 24458
    },
    {
      "epoch": 0.4892,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.034970855712890624,
      "learning_rate": 0.0001,
      "loss": 4.0566,
      "loss/crossentropy": 2.333039402961731,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21214177459478378,
      "step": 24460
    },
    {
      "epoch": 0.48924,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.035302734375,
      "learning_rate": 0.0001,
      "loss": 3.876,
      "loss/crossentropy": 2.0721434950828552,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2012626901268959,
      "step": 24462
    },
    {
      "epoch": 0.48928,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0350006103515625,
      "learning_rate": 0.0001,
      "loss": 3.9045,
      "loss/crossentropy": 1.8990368843078613,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19242222607135773,
      "step": 24464
    },
    {
      "epoch": 0.48932,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0349029541015625,
      "learning_rate": 0.0001,
      "loss": 4.0894,
      "loss/crossentropy": 2.119647741317749,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18154827505350113,
      "step": 24466
    },
    {
      "epoch": 0.48936,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.029679107666015624,
      "learning_rate": 0.0001,
      "loss": 3.9083,
      "loss/crossentropy": 2.0843252539634705,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2219039350748062,
      "step": 24468
    },
    {
      "epoch": 0.4894,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0286773681640625,
      "learning_rate": 0.0001,
      "loss": 3.979,
      "loss/crossentropy": 2.023647725582123,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18391972035169601,
      "step": 24470
    },
    {
      "epoch": 0.48944,
      "grad_norm": 2.0,
      "grad_norm_var": 0.03695475260416667,
      "learning_rate": 0.0001,
      "loss": 4.2427,
      "loss/crossentropy": 2.1299458742141724,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20914346724748611,
      "step": 24472
    },
    {
      "epoch": 0.48948,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.016690826416015624,
      "learning_rate": 0.0001,
      "loss": 3.734,
      "loss/crossentropy": 1.7355242371559143,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1707276776432991,
      "step": 24474
    },
    {
      "epoch": 0.48952,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.020888010660807293,
      "learning_rate": 0.0001,
      "loss": 3.9611,
      "loss/crossentropy": 1.7911944389343262,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19432906806468964,
      "step": 24476
    },
    {
      "epoch": 0.48956,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0211090087890625,
      "learning_rate": 0.0001,
      "loss": 3.7178,
      "loss/crossentropy": 1.824588656425476,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1889827474951744,
      "step": 24478
    },
    {
      "epoch": 0.4896,
      "grad_norm": 1.875,
      "grad_norm_var": 0.020457967122395834,
      "learning_rate": 0.0001,
      "loss": 3.9154,
      "loss/crossentropy": 1.9896376132965088,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2033599317073822,
      "step": 24480
    },
    {
      "epoch": 0.48964,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.020213826497395834,
      "learning_rate": 0.0001,
      "loss": 4.0084,
      "loss/crossentropy": 2.1408406496047974,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20912234485149384,
      "step": 24482
    },
    {
      "epoch": 0.48968,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.02083307902018229,
      "learning_rate": 0.0001,
      "loss": 4.0378,
      "loss/crossentropy": 2.083206295967102,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20860081911087036,
      "step": 24484
    },
    {
      "epoch": 0.48972,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.02238947550455729,
      "learning_rate": 0.0001,
      "loss": 3.676,
      "loss/crossentropy": 1.706899344921112,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1741488128900528,
      "step": 24486
    },
    {
      "epoch": 0.48976,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.011774698893229166,
      "learning_rate": 0.0001,
      "loss": 4.0392,
      "loss/crossentropy": 1.7981711030006409,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1577363833785057,
      "step": 24488
    },
    {
      "epoch": 0.4898,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010716756184895834,
      "learning_rate": 0.0001,
      "loss": 4.0149,
      "loss/crossentropy": 2.3241711854934692,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20463860034942627,
      "step": 24490
    },
    {
      "epoch": 0.48984,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.006810506184895833,
      "learning_rate": 0.0001,
      "loss": 3.8504,
      "loss/crossentropy": 1.8822330832481384,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18153046071529388,
      "step": 24492
    },
    {
      "epoch": 0.48988,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.006406402587890625,
      "learning_rate": 0.0001,
      "loss": 4.2485,
      "loss/crossentropy": 2.4287012815475464,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21238531172275543,
      "step": 24494
    },
    {
      "epoch": 0.48992,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.005783843994140625,
      "learning_rate": 0.0001,
      "loss": 3.898,
      "loss/crossentropy": 2.1026824712753296,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19364942610263824,
      "step": 24496
    },
    {
      "epoch": 0.48996,
      "grad_norm": 1.875,
      "grad_norm_var": 0.00570068359375,
      "learning_rate": 0.0001,
      "loss": 3.8818,
      "loss/crossentropy": 2.045943260192871,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19146864116191864,
      "step": 24498
    },
    {
      "epoch": 0.49,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.004587554931640625,
      "learning_rate": 0.0001,
      "loss": 3.9354,
      "loss/crossentropy": 2.0596802830696106,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18570110201835632,
      "step": 24500
    },
    {
      "epoch": 0.49004,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.005020904541015625,
      "learning_rate": 0.0001,
      "loss": 4.1919,
      "loss/crossentropy": 2.0923487544059753,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19211189448833466,
      "step": 24502
    },
    {
      "epoch": 0.49008,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.007564036051432291,
      "learning_rate": 0.0001,
      "loss": 3.8132,
      "loss/crossentropy": 2.2522445917129517,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19882844388484955,
      "step": 24504
    },
    {
      "epoch": 0.49012,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0066487630208333336,
      "learning_rate": 0.0001,
      "loss": 4.0853,
      "loss/crossentropy": 1.9081300497055054,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18056107312440872,
      "step": 24506
    },
    {
      "epoch": 0.49016,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0062978108723958336,
      "learning_rate": 0.0001,
      "loss": 4.0405,
      "loss/crossentropy": 2.100565791130066,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18689429759979248,
      "step": 24508
    },
    {
      "epoch": 0.4902,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.007562001546223958,
      "learning_rate": 0.0001,
      "loss": 3.9123,
      "loss/crossentropy": 1.9120057821273804,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1958208829164505,
      "step": 24510
    },
    {
      "epoch": 0.49024,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.008888498942057291,
      "learning_rate": 0.0001,
      "loss": 3.9378,
      "loss/crossentropy": 1.812345802783966,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17729748785495758,
      "step": 24512
    },
    {
      "epoch": 0.49028,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.009797922770182292,
      "learning_rate": 0.0001,
      "loss": 3.9652,
      "loss/crossentropy": 1.9347732067108154,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.207149900496006,
      "step": 24514
    },
    {
      "epoch": 0.49032,
      "grad_norm": 1.734375,
      "grad_norm_var": 0.011944325764973958,
      "learning_rate": 0.0001,
      "loss": 3.8159,
      "loss/crossentropy": 1.8915638327598572,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16346803307533264,
      "step": 24516
    },
    {
      "epoch": 0.49036,
      "grad_norm": 1.7265625,
      "grad_norm_var": 0.011934407552083333,
      "learning_rate": 0.0001,
      "loss": 3.7906,
      "loss/crossentropy": 2.2307119369506836,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19432727247476578,
      "step": 24518
    },
    {
      "epoch": 0.4904,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.015240224202473958,
      "learning_rate": 0.0001,
      "loss": 4.012,
      "loss/crossentropy": 2.002243995666504,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2070493996143341,
      "step": 24520
    },
    {
      "epoch": 0.49044,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.017210896809895834,
      "learning_rate": 0.0001,
      "loss": 3.8478,
      "loss/crossentropy": 2.166561722755432,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18929924815893173,
      "step": 24522
    },
    {
      "epoch": 0.49048,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.018619537353515625,
      "learning_rate": 0.0001,
      "loss": 4.1177,
      "loss/crossentropy": 2.1195461750030518,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19457821547985077,
      "step": 24524
    },
    {
      "epoch": 0.49052,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.017878977457682292,
      "learning_rate": 0.0001,
      "loss": 4.195,
      "loss/crossentropy": 2.2752838134765625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21283380687236786,
      "step": 24526
    },
    {
      "epoch": 0.49056,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.015822092692057293,
      "learning_rate": 0.0001,
      "loss": 4.1894,
      "loss/crossentropy": 2.362849473953247,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2148214504122734,
      "step": 24528
    },
    {
      "epoch": 0.4906,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.01643651326497396,
      "learning_rate": 0.0001,
      "loss": 3.965,
      "loss/crossentropy": 1.9928752779960632,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.204628124833107,
      "step": 24530
    },
    {
      "epoch": 0.49064,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.012206013997395833,
      "learning_rate": 0.0001,
      "loss": 4.1465,
      "loss/crossentropy": 2.2063074111938477,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20204229652881622,
      "step": 24532
    },
    {
      "epoch": 0.49068,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008457183837890625,
      "learning_rate": 0.0001,
      "loss": 4.2816,
      "loss/crossentropy": 2.1019493341445923,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18352027982473373,
      "step": 24534
    },
    {
      "epoch": 0.49072,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.006949615478515625,
      "learning_rate": 0.0001,
      "loss": 4.0119,
      "loss/crossentropy": 1.9460110664367676,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18813221156597137,
      "step": 24536
    },
    {
      "epoch": 0.49076,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.006912994384765625,
      "learning_rate": 0.0001,
      "loss": 3.9377,
      "loss/crossentropy": 1.8915529251098633,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.198774054646492,
      "step": 24538
    },
    {
      "epoch": 0.4908,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0067860921223958336,
      "learning_rate": 0.0001,
      "loss": 4.2914,
      "loss/crossentropy": 2.0600665807724,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2067568153142929,
      "step": 24540
    },
    {
      "epoch": 0.49084,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.007347615559895834,
      "learning_rate": 0.0001,
      "loss": 3.8895,
      "loss/crossentropy": 2.3356127738952637,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21175508201122284,
      "step": 24542
    },
    {
      "epoch": 0.49088,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.006626129150390625,
      "learning_rate": 0.0001,
      "loss": 3.7529,
      "loss/crossentropy": 1.7713094353675842,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16650734096765518,
      "step": 24544
    },
    {
      "epoch": 0.49092,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.006514231363932292,
      "learning_rate": 0.0001,
      "loss": 4.1357,
      "loss/crossentropy": 2.307682991027832,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20595747232437134,
      "step": 24546
    },
    {
      "epoch": 0.49096,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.0071044921875,
      "learning_rate": 0.0001,
      "loss": 3.7078,
      "loss/crossentropy": 2.0465540289878845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16928144544363022,
      "step": 24548
    },
    {
      "epoch": 0.491,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.006422678629557292,
      "learning_rate": 0.0001,
      "loss": 3.7281,
      "loss/crossentropy": 1.6479852795600891,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1531681939959526,
      "step": 24550
    },
    {
      "epoch": 0.49104,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.007600911458333333,
      "learning_rate": 0.0001,
      "loss": 3.9933,
      "loss/crossentropy": 2.2218964099884033,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1819760948419571,
      "step": 24552
    },
    {
      "epoch": 0.49108,
      "grad_norm": 1.75,
      "grad_norm_var": 0.009098307291666666,
      "learning_rate": 0.0001,
      "loss": 3.6829,
      "loss/crossentropy": 1.7314581274986267,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1850220039486885,
      "step": 24554
    },
    {
      "epoch": 0.49112,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.007059733072916667,
      "learning_rate": 0.0001,
      "loss": 3.9411,
      "loss/crossentropy": 2.1683263778686523,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18814556300640106,
      "step": 24556
    },
    {
      "epoch": 0.49116,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.007098134358723958,
      "learning_rate": 0.0001,
      "loss": 4.2667,
      "loss/crossentropy": 2.364906072616577,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20846372097730637,
      "step": 24558
    },
    {
      "epoch": 0.4912,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.007710520426432292,
      "learning_rate": 0.0001,
      "loss": 4.1047,
      "loss/crossentropy": 2.4396650791168213,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21760626137256622,
      "step": 24560
    },
    {
      "epoch": 0.49124,
      "grad_norm": 2.375,
      "grad_norm_var": 0.059915924072265626,
      "learning_rate": 0.0001,
      "loss": 4.1424,
      "loss/crossentropy": 2.2420458793640137,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19711492210626602,
      "step": 24562
    },
    {
      "epoch": 0.49128,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.05835367838541667,
      "learning_rate": 0.0001,
      "loss": 3.8802,
      "loss/crossentropy": 1.8962340354919434,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20642152428627014,
      "step": 24564
    },
    {
      "epoch": 0.49132,
      "grad_norm": 2.0,
      "grad_norm_var": 0.05555597941080729,
      "learning_rate": 0.0001,
      "loss": 3.9291,
      "loss/crossentropy": 2.024625241756439,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20574413985013962,
      "step": 24566
    },
    {
      "epoch": 0.49136,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.05478515625,
      "learning_rate": 0.0001,
      "loss": 4.2441,
      "loss/crossentropy": 2.2790380716323853,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19655412435531616,
      "step": 24568
    },
    {
      "epoch": 0.4914,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.04713923136393229,
      "learning_rate": 0.0001,
      "loss": 4.0206,
      "loss/crossentropy": 2.1089435815811157,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18229448795318604,
      "step": 24570
    },
    {
      "epoch": 0.49144,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.04691162109375,
      "learning_rate": 0.0001,
      "loss": 4.2213,
      "loss/crossentropy": 2.242396593093872,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1903126761317253,
      "step": 24572
    },
    {
      "epoch": 0.49148,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.0496002197265625,
      "learning_rate": 0.0001,
      "loss": 3.9656,
      "loss/crossentropy": 2.3436553478240967,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20461376011371613,
      "step": 24574
    },
    {
      "epoch": 0.49152,
      "grad_norm": 2.375,
      "grad_norm_var": 0.05408910115559896,
      "learning_rate": 0.0001,
      "loss": 3.9294,
      "loss/crossentropy": 2.020145893096924,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20228110253810883,
      "step": 24576
    },
    {
      "epoch": 0.49156,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.01663386027018229,
      "learning_rate": 0.0001,
      "loss": 4.1186,
      "loss/crossentropy": 2.0371296405792236,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19508864730596542,
      "step": 24578
    },
    {
      "epoch": 0.4916,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.014544423421223958,
      "learning_rate": 0.0001,
      "loss": 4.2877,
      "loss/crossentropy": 2.078322470188141,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19366976618766785,
      "step": 24580
    },
    {
      "epoch": 0.49164,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.014349110921223958,
      "learning_rate": 0.0001,
      "loss": 4.222,
      "loss/crossentropy": 2.3686896562576294,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21069607138633728,
      "step": 24582
    },
    {
      "epoch": 0.49168,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.01715876261393229,
      "learning_rate": 0.0001,
      "loss": 3.9882,
      "loss/crossentropy": 2.0332735180854797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21546462178230286,
      "step": 24584
    },
    {
      "epoch": 0.49172,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.01779759724934896,
      "learning_rate": 0.0001,
      "loss": 4.1297,
      "loss/crossentropy": 2.0070220232009888,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17601517587900162,
      "step": 24586
    },
    {
      "epoch": 0.49176,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.014888254801432292,
      "learning_rate": 0.0001,
      "loss": 4.0979,
      "loss/crossentropy": 2.1559245586395264,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19782397896051407,
      "step": 24588
    },
    {
      "epoch": 0.4918,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.0160308837890625,
      "learning_rate": 0.0001,
      "loss": 3.8771,
      "loss/crossentropy": 2.173780679702759,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20206089317798615,
      "step": 24590
    },
    {
      "epoch": 0.49184,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.008215077718098958,
      "learning_rate": 0.0001,
      "loss": 3.8911,
      "loss/crossentropy": 1.609294056892395,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16182265430688858,
      "step": 24592
    },
    {
      "epoch": 0.49188,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.08147557576497395,
      "learning_rate": 0.0001,
      "loss": 4.2665,
      "loss/crossentropy": 2.2137110233306885,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27386993169784546,
      "step": 24594
    },
    {
      "epoch": 0.49192,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.08635838826497395,
      "learning_rate": 0.0001,
      "loss": 3.7846,
      "loss/crossentropy": 1.9553529024124146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18842827528715134,
      "step": 24596
    },
    {
      "epoch": 0.49196,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.08915176391601562,
      "learning_rate": 0.0001,
      "loss": 4.0381,
      "loss/crossentropy": 1.8976882696151733,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18534697592258453,
      "step": 24598
    },
    {
      "epoch": 0.492,
      "grad_norm": 2.25,
      "grad_norm_var": 0.09050064086914063,
      "learning_rate": 0.0001,
      "loss": 4.3902,
      "loss/crossentropy": 1.9944769740104675,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20278682559728622,
      "step": 24600
    },
    {
      "epoch": 0.49204,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.094580078125,
      "learning_rate": 0.0001,
      "loss": 3.8038,
      "loss/crossentropy": 2.3396860361099243,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19831668585538864,
      "step": 24602
    },
    {
      "epoch": 0.49208,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.09804280598958333,
      "learning_rate": 0.0001,
      "loss": 3.9544,
      "loss/crossentropy": 1.8764755725860596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19239525496959686,
      "step": 24604
    },
    {
      "epoch": 0.49212,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0951080322265625,
      "learning_rate": 0.0001,
      "loss": 4.0428,
      "loss/crossentropy": 1.9992756843566895,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18613643199205399,
      "step": 24606
    },
    {
      "epoch": 0.49216,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.09273681640625,
      "learning_rate": 0.0001,
      "loss": 4.1538,
      "loss/crossentropy": 2.0907077193260193,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1898733228445053,
      "step": 24608
    },
    {
      "epoch": 0.4922,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.017183430989583335,
      "learning_rate": 0.0001,
      "loss": 3.9356,
      "loss/crossentropy": 1.9387467503547668,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2077641263604164,
      "step": 24610
    },
    {
      "epoch": 0.49224,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.015561676025390625,
      "learning_rate": 0.0001,
      "loss": 3.6729,
      "loss/crossentropy": 1.8329110741615295,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17758768796920776,
      "step": 24612
    },
    {
      "epoch": 0.49228,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.01461181640625,
      "learning_rate": 0.0001,
      "loss": 3.9334,
      "loss/crossentropy": 1.8766810894012451,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18651552498340607,
      "step": 24614
    },
    {
      "epoch": 0.49232,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.009284464518229167,
      "learning_rate": 0.0001,
      "loss": 3.6679,
      "loss/crossentropy": 1.7456438541412354,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16719914227724075,
      "step": 24616
    },
    {
      "epoch": 0.49236,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.007920074462890624,
      "learning_rate": 0.0001,
      "loss": 4.0292,
      "loss/crossentropy": 2.0994815826416016,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20065009593963623,
      "step": 24618
    },
    {
      "epoch": 0.4924,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.007218170166015625,
      "learning_rate": 0.0001,
      "loss": 4.0588,
      "loss/crossentropy": 2.3052616119384766,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2254004403948784,
      "step": 24620
    },
    {
      "epoch": 0.49244,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.006516265869140625,
      "learning_rate": 0.0001,
      "loss": 3.9487,
      "loss/crossentropy": 1.7560802102088928,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17277851700782776,
      "step": 24622
    },
    {
      "epoch": 0.49248,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.007142893473307292,
      "learning_rate": 0.0001,
      "loss": 4.0868,
      "loss/crossentropy": 1.862656593322754,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19756756722927094,
      "step": 24624
    },
    {
      "epoch": 0.49252,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.006959788004557292,
      "learning_rate": 0.0001,
      "loss": 3.9022,
      "loss/crossentropy": 2.0374228954315186,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19410685449838638,
      "step": 24626
    },
    {
      "epoch": 0.49256,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.006315104166666667,
      "learning_rate": 0.0001,
      "loss": 3.9468,
      "loss/crossentropy": 2.2796911001205444,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21644171327352524,
      "step": 24628
    },
    {
      "epoch": 0.4926,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.03423436482747396,
      "learning_rate": 0.0001,
      "loss": 3.9822,
      "loss/crossentropy": 1.9718595743179321,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18669331073760986,
      "step": 24630
    },
    {
      "epoch": 0.49264,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.030427042643229166,
      "learning_rate": 0.0001,
      "loss": 4.3174,
      "loss/crossentropy": 2.0071244835853577,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1994020640850067,
      "step": 24632
    },
    {
      "epoch": 0.49268,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.029759724934895832,
      "learning_rate": 0.0001,
      "loss": 4.3168,
      "loss/crossentropy": 2.2078527212142944,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2190861478447914,
      "step": 24634
    },
    {
      "epoch": 0.49272,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0288482666015625,
      "learning_rate": 0.0001,
      "loss": 4.1319,
      "loss/crossentropy": 2.374260663986206,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2121455818414688,
      "step": 24636
    },
    {
      "epoch": 0.49276,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.02938232421875,
      "learning_rate": 0.0001,
      "loss": 4.0958,
      "loss/crossentropy": 2.139094114303589,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20233627408742905,
      "step": 24638
    },
    {
      "epoch": 0.4928,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0288970947265625,
      "learning_rate": 0.0001,
      "loss": 4.1157,
      "loss/crossentropy": 1.9579105377197266,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2005467787384987,
      "step": 24640
    },
    {
      "epoch": 0.49284,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.025199381510416667,
      "learning_rate": 0.0001,
      "loss": 4.3003,
      "loss/crossentropy": 2.122314691543579,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2005690187215805,
      "step": 24642
    },
    {
      "epoch": 0.49288,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.024094390869140624,
      "learning_rate": 0.0001,
      "loss": 4.3163,
      "loss/crossentropy": 1.9626818299293518,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.196880042552948,
      "step": 24644
    },
    {
      "epoch": 0.49292,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0026446024576822916,
      "learning_rate": 0.0001,
      "loss": 4.1957,
      "loss/crossentropy": 1.99526047706604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18154115974903107,
      "step": 24646
    },
    {
      "epoch": 0.49296,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0028310139973958333,
      "learning_rate": 0.0001,
      "loss": 4.1079,
      "loss/crossentropy": 2.2288442850112915,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19869591295719147,
      "step": 24648
    },
    {
      "epoch": 0.493,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.002756500244140625,
      "learning_rate": 0.0001,
      "loss": 4.0832,
      "loss/crossentropy": 1.8555628061294556,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18529972434043884,
      "step": 24650
    },
    {
      "epoch": 0.49304,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0029042561848958335,
      "learning_rate": 0.0001,
      "loss": 4.0908,
      "loss/crossentropy": 2.1587546467781067,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1936108022928238,
      "step": 24652
    },
    {
      "epoch": 0.49308,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0025469462076822915,
      "learning_rate": 0.0001,
      "loss": 4.2219,
      "loss/crossentropy": 1.8942645192146301,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18769075721502304,
      "step": 24654
    },
    {
      "epoch": 0.49312,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0025266011555989585,
      "learning_rate": 0.0001,
      "loss": 4.0054,
      "loss/crossentropy": 2.334362268447876,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21441897749900818,
      "step": 24656
    },
    {
      "epoch": 0.49316,
      "grad_norm": 2.25,
      "grad_norm_var": 0.011063639322916667,
      "learning_rate": 0.0001,
      "loss": 3.945,
      "loss/crossentropy": 1.9287384748458862,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2295522838830948,
      "step": 24658
    },
    {
      "epoch": 0.4932,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.012111155192057292,
      "learning_rate": 0.0001,
      "loss": 4.1852,
      "loss/crossentropy": 2.212242364883423,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2138296589255333,
      "step": 24660
    },
    {
      "epoch": 0.49324,
      "grad_norm": 2.125,
      "grad_norm_var": 0.014831288655598959,
      "learning_rate": 0.0001,
      "loss": 3.9479,
      "loss/crossentropy": 2.17527437210083,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1837603896856308,
      "step": 24662
    },
    {
      "epoch": 0.49328,
      "grad_norm": 1.875,
      "grad_norm_var": 0.01660741170247396,
      "learning_rate": 0.0001,
      "loss": 4.0985,
      "loss/crossentropy": 2.1768887042999268,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20551389455795288,
      "step": 24664
    },
    {
      "epoch": 0.49332,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.016646321614583334,
      "learning_rate": 0.0001,
      "loss": 3.9299,
      "loss/crossentropy": 1.987557053565979,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1999402493238449,
      "step": 24666
    },
    {
      "epoch": 0.49336,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.0241851806640625,
      "learning_rate": 0.0001,
      "loss": 3.5654,
      "loss/crossentropy": 1.7189387083053589,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16583546251058578,
      "step": 24668
    },
    {
      "epoch": 0.4934,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.026021321614583332,
      "learning_rate": 0.0001,
      "loss": 3.9805,
      "loss/crossentropy": 2.0449089407920837,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21923594176769257,
      "step": 24670
    },
    {
      "epoch": 0.49344,
      "grad_norm": 4.21875,
      "grad_norm_var": 0.34633763631184894,
      "learning_rate": 0.0001,
      "loss": 4.0541,
      "loss/crossentropy": 1.922966718673706,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19492915272712708,
      "step": 24672
    },
    {
      "epoch": 0.49348,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.34245198567708335,
      "learning_rate": 0.0001,
      "loss": 4.1132,
      "loss/crossentropy": 1.8697016835212708,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21420104801654816,
      "step": 24674
    },
    {
      "epoch": 0.49352,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.34196370442708335,
      "learning_rate": 0.0001,
      "loss": 4.2348,
      "loss/crossentropy": 2.2170732021331787,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20673636347055435,
      "step": 24676
    },
    {
      "epoch": 0.49356,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.34133707682291664,
      "learning_rate": 0.0001,
      "loss": 3.7056,
      "loss/crossentropy": 2.0645321011543274,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2113899141550064,
      "step": 24678
    },
    {
      "epoch": 0.4936,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.3439776102701823,
      "learning_rate": 0.0001,
      "loss": 4.0543,
      "loss/crossentropy": 2.0374088883399963,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20146184414625168,
      "step": 24680
    },
    {
      "epoch": 0.49364,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.34632975260416665,
      "learning_rate": 0.0001,
      "loss": 4.0364,
      "loss/crossentropy": 1.9996783137321472,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19109730422496796,
      "step": 24682
    },
    {
      "epoch": 0.49368,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.329974110921224,
      "learning_rate": 0.0001,
      "loss": 4.1988,
      "loss/crossentropy": 2.2255775928497314,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2078489512205124,
      "step": 24684
    },
    {
      "epoch": 0.49372,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.3313385009765625,
      "learning_rate": 0.0001,
      "loss": 4.2153,
      "loss/crossentropy": 1.9860047698020935,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21092459559440613,
      "step": 24686
    },
    {
      "epoch": 0.49376,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.020475260416666665,
      "learning_rate": 0.0001,
      "loss": 4.0701,
      "loss/crossentropy": 1.9905331134796143,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1982455775141716,
      "step": 24688
    },
    {
      "epoch": 0.4938,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.012786610921223959,
      "learning_rate": 0.0001,
      "loss": 4.1729,
      "loss/crossentropy": 2.254474639892578,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20714619755744934,
      "step": 24690
    },
    {
      "epoch": 0.49384,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.011376698811848959,
      "learning_rate": 0.0001,
      "loss": 3.8576,
      "loss/crossentropy": 2.264926791191101,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1946386992931366,
      "step": 24692
    },
    {
      "epoch": 0.49388,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.01116943359375,
      "learning_rate": 0.0001,
      "loss": 3.847,
      "loss/crossentropy": 2.071332335472107,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2170526310801506,
      "step": 24694
    },
    {
      "epoch": 0.49392,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.010746002197265625,
      "learning_rate": 0.0001,
      "loss": 3.8254,
      "loss/crossentropy": 1.973964512348175,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20519066601991653,
      "step": 24696
    },
    {
      "epoch": 0.49396,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.011644490559895833,
      "learning_rate": 0.0001,
      "loss": 3.848,
      "loss/crossentropy": 1.6099820137023926,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15943622589111328,
      "step": 24698
    },
    {
      "epoch": 0.494,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.010542551676432291,
      "learning_rate": 0.0001,
      "loss": 4.1036,
      "loss/crossentropy": 2.1428415775299072,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20248957723379135,
      "step": 24700
    },
    {
      "epoch": 0.49404,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.003742472330729167,
      "learning_rate": 0.0001,
      "loss": 4.3402,
      "loss/crossentropy": 2.3973593711853027,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21721762418746948,
      "step": 24702
    },
    {
      "epoch": 0.49408,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.004219563802083334,
      "learning_rate": 0.0001,
      "loss": 4.2233,
      "loss/crossentropy": 2.184257686138153,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20263968408107758,
      "step": 24704
    },
    {
      "epoch": 0.49412,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0042111714680989586,
      "learning_rate": 0.0001,
      "loss": 3.9782,
      "loss/crossentropy": 1.953734815120697,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18011564016342163,
      "step": 24706
    },
    {
      "epoch": 0.49416,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.00406494140625,
      "learning_rate": 0.0001,
      "loss": 4.0393,
      "loss/crossentropy": 2.0036890506744385,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21608620882034302,
      "step": 24708
    },
    {
      "epoch": 0.4942,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.0047686258951822914,
      "learning_rate": 0.0001,
      "loss": 3.9851,
      "loss/crossentropy": 2.016548752784729,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18857601284980774,
      "step": 24710
    },
    {
      "epoch": 0.49424,
      "grad_norm": 1.875,
      "grad_norm_var": 0.0053179423014322914,
      "learning_rate": 0.0001,
      "loss": 3.9084,
      "loss/crossentropy": 2.0665934085845947,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18890216201543808,
      "step": 24712
    },
    {
      "epoch": 0.49428,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.005216471354166667,
      "learning_rate": 0.0001,
      "loss": 4.2576,
      "loss/crossentropy": 2.102554202079773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19079755991697311,
      "step": 24714
    },
    {
      "epoch": 0.49432,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.007893625895182292,
      "learning_rate": 0.0001,
      "loss": 4.2018,
      "loss/crossentropy": 2.326322555541992,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21783028542995453,
      "step": 24716
    },
    {
      "epoch": 0.49436,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.0073811848958333336,
      "learning_rate": 0.0001,
      "loss": 4.2362,
      "loss/crossentropy": 2.2604973316192627,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19831713289022446,
      "step": 24718
    },
    {
      "epoch": 0.4944,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.00660400390625,
      "learning_rate": 0.0001,
      "loss": 3.859,
      "loss/crossentropy": 2.2417763471603394,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19713596254587173,
      "step": 24720
    },
    {
      "epoch": 0.49444,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.009533437093098958,
      "learning_rate": 0.0001,
      "loss": 4.5059,
      "loss/crossentropy": 2.21281498670578,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2067965790629387,
      "step": 24722
    },
    {
      "epoch": 0.49448,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.009696451822916667,
      "learning_rate": 0.0001,
      "loss": 4.0386,
      "loss/crossentropy": 2.288776397705078,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19350187480449677,
      "step": 24724
    },
    {
      "epoch": 0.49452,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.009528605143229167,
      "learning_rate": 0.0001,
      "loss": 4.0452,
      "loss/crossentropy": 1.8925097584724426,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1952025145292282,
      "step": 24726
    },
    {
      "epoch": 0.49456,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.009287261962890625,
      "learning_rate": 0.0001,
      "loss": 3.8005,
      "loss/crossentropy": 2.2274699211120605,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19862639904022217,
      "step": 24728
    },
    {
      "epoch": 0.4946,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.008671061197916666,
      "learning_rate": 0.0001,
      "loss": 3.9947,
      "loss/crossentropy": 2.028487980365753,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19470887631177902,
      "step": 24730
    },
    {
      "epoch": 0.49464,
      "grad_norm": 1.75,
      "grad_norm_var": 0.0082672119140625,
      "learning_rate": 0.0001,
      "loss": 4.2435,
      "loss/crossentropy": 2.413878560066223,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20918180793523788,
      "step": 24732
    },
    {
      "epoch": 0.49468,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.007478841145833333,
      "learning_rate": 0.0001,
      "loss": 4.0814,
      "loss/crossentropy": 2.2201125621795654,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19350255280733109,
      "step": 24734
    },
    {
      "epoch": 0.49472,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.006859334309895834,
      "learning_rate": 0.0001,
      "loss": 4.0363,
      "loss/crossentropy": 2.4426982402801514,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20905796438455582,
      "step": 24736
    },
    {
      "epoch": 0.49476,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0044667561848958336,
      "learning_rate": 0.0001,
      "loss": 3.811,
      "loss/crossentropy": 1.586302399635315,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1673523262143135,
      "step": 24738
    },
    {
      "epoch": 0.4948,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.009822591145833334,
      "learning_rate": 0.0001,
      "loss": 3.8759,
      "loss/crossentropy": 2.0435025691986084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17974509298801422,
      "step": 24740
    },
    {
      "epoch": 0.49484,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.009916178385416667,
      "learning_rate": 0.0001,
      "loss": 4.288,
      "loss/crossentropy": 2.2075798511505127,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2002267688512802,
      "step": 24742
    },
    {
      "epoch": 0.49488,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.011563873291015625,
      "learning_rate": 0.0001,
      "loss": 3.6645,
      "loss/crossentropy": 1.8993717432022095,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17360415309667587,
      "step": 24744
    },
    {
      "epoch": 0.49492,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.013024648030598959,
      "learning_rate": 0.0001,
      "loss": 3.7726,
      "loss/crossentropy": 1.8296250700950623,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1821255162358284,
      "step": 24746
    },
    {
      "epoch": 0.49496,
      "grad_norm": 2.125,
      "grad_norm_var": 0.013911692301432292,
      "learning_rate": 0.0001,
      "loss": 3.9835,
      "loss/crossentropy": 1.8268097043037415,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1724962294101715,
      "step": 24748
    },
    {
      "epoch": 0.495,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.014411417643229167,
      "learning_rate": 0.0001,
      "loss": 3.9835,
      "loss/crossentropy": 1.9587016701698303,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.193455271422863,
      "step": 24750
    },
    {
      "epoch": 0.49504,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.014811197916666666,
      "learning_rate": 0.0001,
      "loss": 4.0863,
      "loss/crossentropy": 2.1292134523391724,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20238783955574036,
      "step": 24752
    },
    {
      "epoch": 0.49508,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.01566136678059896,
      "learning_rate": 0.0001,
      "loss": 3.9682,
      "loss/crossentropy": 2.0145905017852783,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2276977151632309,
      "step": 24754
    },
    {
      "epoch": 0.49512,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011028798421223958,
      "learning_rate": 0.0001,
      "loss": 4.0566,
      "loss/crossentropy": 1.9815504550933838,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1814296990633011,
      "step": 24756
    },
    {
      "epoch": 0.49516,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.012532552083333334,
      "learning_rate": 0.0001,
      "loss": 3.7894,
      "loss/crossentropy": 1.8275295495986938,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16582081466913223,
      "step": 24758
    },
    {
      "epoch": 0.4952,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.009859212239583333,
      "learning_rate": 0.0001,
      "loss": 3.9488,
      "loss/crossentropy": 1.9509209990501404,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19137832522392273,
      "step": 24760
    },
    {
      "epoch": 0.49524,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.007433827718098958,
      "learning_rate": 0.0001,
      "loss": 4.405,
      "loss/crossentropy": 2.255669593811035,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20520294457674026,
      "step": 24762
    },
    {
      "epoch": 0.49528,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.006624348958333333,
      "learning_rate": 0.0001,
      "loss": 3.9608,
      "loss/crossentropy": 1.902245819568634,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1721034198999405,
      "step": 24764
    },
    {
      "epoch": 0.49532,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.007085927327473958,
      "learning_rate": 0.0001,
      "loss": 4.033,
      "loss/crossentropy": 2.0003857016563416,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1873132660984993,
      "step": 24766
    },
    {
      "epoch": 0.49536,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.01024169921875,
      "learning_rate": 0.0001,
      "loss": 4.1782,
      "loss/crossentropy": 2.3280882835388184,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21413114666938782,
      "step": 24768
    },
    {
      "epoch": 0.4954,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.008882649739583333,
      "learning_rate": 0.0001,
      "loss": 4.0517,
      "loss/crossentropy": 2.0396437644958496,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18958552181720734,
      "step": 24770
    },
    {
      "epoch": 0.49544,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.009012603759765625,
      "learning_rate": 0.0001,
      "loss": 3.9254,
      "loss/crossentropy": 2.247615098953247,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21501318365335464,
      "step": 24772
    },
    {
      "epoch": 0.49548,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.008479817708333334,
      "learning_rate": 0.0001,
      "loss": 3.8623,
      "loss/crossentropy": 2.0479390621185303,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20071326941251755,
      "step": 24774
    },
    {
      "epoch": 0.49552,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.008591461181640624,
      "learning_rate": 0.0001,
      "loss": 4.2585,
      "loss/crossentropy": 2.129696846008301,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2128293514251709,
      "step": 24776
    },
    {
      "epoch": 0.49556,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.008805338541666667,
      "learning_rate": 0.0001,
      "loss": 4.1479,
      "loss/crossentropy": 2.0912004709243774,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24846868962049484,
      "step": 24778
    },
    {
      "epoch": 0.4956,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.008934529622395833,
      "learning_rate": 0.0001,
      "loss": 4.1595,
      "loss/crossentropy": 2.310052990913391,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2025884911417961,
      "step": 24780
    },
    {
      "epoch": 0.49564,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.010886383056640626,
      "learning_rate": 0.0001,
      "loss": 4.2532,
      "loss/crossentropy": 2.3318560123443604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21703048795461655,
      "step": 24782
    },
    {
      "epoch": 0.49568,
      "grad_norm": 2.0,
      "grad_norm_var": 0.009791819254557292,
      "learning_rate": 0.0001,
      "loss": 3.7083,
      "loss/crossentropy": 1.9909458756446838,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18664265424013138,
      "step": 24784
    },
    {
      "epoch": 0.49572,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.009814198811848958,
      "learning_rate": 0.0001,
      "loss": 4.34,
      "loss/crossentropy": 2.530202269554138,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20007390528917313,
      "step": 24786
    },
    {
      "epoch": 0.49576,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.009382120768229167,
      "learning_rate": 0.0001,
      "loss": 4.1865,
      "loss/crossentropy": 2.286345958709717,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20716369152069092,
      "step": 24788
    },
    {
      "epoch": 0.4958,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.008768717447916666,
      "learning_rate": 0.0001,
      "loss": 4.06,
      "loss/crossentropy": 2.105964183807373,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1934148371219635,
      "step": 24790
    },
    {
      "epoch": 0.49584,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.009618123372395834,
      "learning_rate": 0.0001,
      "loss": 3.8057,
      "loss/crossentropy": 1.8528069853782654,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16931568086147308,
      "step": 24792
    },
    {
      "epoch": 0.49588,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.008699544270833333,
      "learning_rate": 0.0001,
      "loss": 3.9729,
      "loss/crossentropy": 1.8847370743751526,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16615931689739227,
      "step": 24794
    },
    {
      "epoch": 0.49592,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.009040323893229167,
      "learning_rate": 0.0001,
      "loss": 4.2483,
      "loss/crossentropy": 2.159093201160431,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20933274924755096,
      "step": 24796
    },
    {
      "epoch": 0.49596,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.007995351155598959,
      "learning_rate": 0.0001,
      "loss": 4.2635,
      "loss/crossentropy": 2.1289132833480835,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19792458415031433,
      "step": 24798
    },
    {
      "epoch": 0.496,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.0071523030598958336,
      "learning_rate": 0.0001,
      "loss": 4.0548,
      "loss/crossentropy": 2.0613357424736023,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18525776267051697,
      "step": 24800
    },
    {
      "epoch": 0.49604,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0070953369140625,
      "learning_rate": 0.0001,
      "loss": 4.0807,
      "loss/crossentropy": 1.7627623081207275,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16891766339540482,
      "step": 24802
    },
    {
      "epoch": 0.49608,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.008446248372395833,
      "learning_rate": 0.0001,
      "loss": 4.0642,
      "loss/crossentropy": 1.77890944480896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24872711300849915,
      "step": 24804
    },
    {
      "epoch": 0.49612,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.009291330973307291,
      "learning_rate": 0.0001,
      "loss": 4.0468,
      "loss/crossentropy": 2.240827202796936,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20014363527297974,
      "step": 24806
    },
    {
      "epoch": 0.49616,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.009968058268229166,
      "learning_rate": 0.0001,
      "loss": 3.8452,
      "loss/crossentropy": 1.7963212132453918,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16260015219449997,
      "step": 24808
    },
    {
      "epoch": 0.4962,
      "grad_norm": 2.125,
      "grad_norm_var": 0.011681874593098959,
      "learning_rate": 0.0001,
      "loss": 4.0605,
      "loss/crossentropy": 2.1720168590545654,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18297959119081497,
      "step": 24810
    },
    {
      "epoch": 0.49624,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.010849761962890624,
      "learning_rate": 0.0001,
      "loss": 3.8772,
      "loss/crossentropy": 2.0715248584747314,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20910069346427917,
      "step": 24812
    },
    {
      "epoch": 0.49628,
      "grad_norm": 2.0,
      "grad_norm_var": 0.009203084309895833,
      "learning_rate": 0.0001,
      "loss": 3.8442,
      "loss/crossentropy": 1.8676977157592773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19553124904632568,
      "step": 24814
    },
    {
      "epoch": 0.49632,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.008809407552083334,
      "learning_rate": 0.0001,
      "loss": 4.1918,
      "loss/crossentropy": 2.383392572402954,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20431752502918243,
      "step": 24816
    },
    {
      "epoch": 0.49636,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.013744099934895834,
      "learning_rate": 0.0001,
      "loss": 4.1956,
      "loss/crossentropy": 1.8039666414260864,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17365024238824844,
      "step": 24818
    },
    {
      "epoch": 0.4964,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.012967936197916667,
      "learning_rate": 0.0001,
      "loss": 3.7321,
      "loss/crossentropy": 1.803081214427948,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.188148595392704,
      "step": 24820
    },
    {
      "epoch": 0.49644,
      "grad_norm": 1.7421875,
      "grad_norm_var": 0.014137522379557291,
      "learning_rate": 0.0001,
      "loss": 3.8016,
      "loss/crossentropy": 1.9158729910850525,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17063472419977188,
      "step": 24822
    },
    {
      "epoch": 0.49648,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.01593805948893229,
      "learning_rate": 0.0001,
      "loss": 3.8044,
      "loss/crossentropy": 2.013060450553894,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.192637138068676,
      "step": 24824
    },
    {
      "epoch": 0.49652,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.015126291910807292,
      "learning_rate": 0.0001,
      "loss": 3.8618,
      "loss/crossentropy": 2.2182289361953735,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2003488838672638,
      "step": 24826
    },
    {
      "epoch": 0.49656,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.0152008056640625,
      "learning_rate": 0.0001,
      "loss": 3.7418,
      "loss/crossentropy": 1.9512917399406433,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17393754422664642,
      "step": 24828
    },
    {
      "epoch": 0.4966,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.017986806233723958,
      "learning_rate": 0.0001,
      "loss": 4.2311,
      "loss/crossentropy": 2.3564621210098267,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22488048672676086,
      "step": 24830
    },
    {
      "epoch": 0.49664,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0183502197265625,
      "learning_rate": 0.0001,
      "loss": 4.0941,
      "loss/crossentropy": 2.3347015380859375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2045806273818016,
      "step": 24832
    },
    {
      "epoch": 0.49668,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0120513916015625,
      "learning_rate": 0.0001,
      "loss": 4.0312,
      "loss/crossentropy": 1.8157051801681519,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16906943917274475,
      "step": 24834
    },
    {
      "epoch": 0.49672,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.011903635660807292,
      "learning_rate": 0.0001,
      "loss": 4.1453,
      "loss/crossentropy": 2.068196475505829,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20066522806882858,
      "step": 24836
    },
    {
      "epoch": 0.49676,
      "grad_norm": 1.875,
      "grad_norm_var": 0.010198720296223958,
      "learning_rate": 0.0001,
      "loss": 3.9633,
      "loss/crossentropy": 1.771648347377777,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15898138284683228,
      "step": 24838
    },
    {
      "epoch": 0.4968,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008267974853515625,
      "learning_rate": 0.0001,
      "loss": 3.8167,
      "loss/crossentropy": 2.206678628921509,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22059056907892227,
      "step": 24840
    },
    {
      "epoch": 0.49684,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006502278645833333,
      "learning_rate": 0.0001,
      "loss": 4.0938,
      "loss/crossentropy": 2.3368433713912964,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20108889043331146,
      "step": 24842
    },
    {
      "epoch": 0.49688,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.007039388020833333,
      "learning_rate": 0.0001,
      "loss": 3.6263,
      "loss/crossentropy": 2.1412705183029175,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19969124346971512,
      "step": 24844
    },
    {
      "epoch": 0.49692,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0048215230305989586,
      "learning_rate": 0.0001,
      "loss": 4.0444,
      "loss/crossentropy": 1.9018915295600891,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17174704372882843,
      "step": 24846
    },
    {
      "epoch": 0.49696,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.0062652587890625,
      "learning_rate": 0.0001,
      "loss": 4.3229,
      "loss/crossentropy": 2.3042339086532593,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20182086527347565,
      "step": 24848
    },
    {
      "epoch": 0.497,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.009948476155598959,
      "learning_rate": 0.0001,
      "loss": 3.899,
      "loss/crossentropy": 2.0412787199020386,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19640880823135376,
      "step": 24850
    },
    {
      "epoch": 0.49704,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.010065714518229166,
      "learning_rate": 0.0001,
      "loss": 3.76,
      "loss/crossentropy": 1.9754884243011475,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17998956143856049,
      "step": 24852
    },
    {
      "epoch": 0.49708,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010038248697916667,
      "learning_rate": 0.0001,
      "loss": 4.1763,
      "loss/crossentropy": 2.060430645942688,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21152494102716446,
      "step": 24854
    },
    {
      "epoch": 0.49712,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.010375722249348959,
      "learning_rate": 0.0001,
      "loss": 3.7536,
      "loss/crossentropy": 1.6876602172851562,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1664302945137024,
      "step": 24856
    },
    {
      "epoch": 0.49716,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.009663645426432292,
      "learning_rate": 0.0001,
      "loss": 3.9341,
      "loss/crossentropy": 1.9389417171478271,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19122107326984406,
      "step": 24858
    },
    {
      "epoch": 0.4972,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.010752105712890625,
      "learning_rate": 0.0001,
      "loss": 3.905,
      "loss/crossentropy": 2.069887936115265,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20993375778198242,
      "step": 24860
    },
    {
      "epoch": 0.49724,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0125640869140625,
      "learning_rate": 0.0001,
      "loss": 4.2935,
      "loss/crossentropy": 2.1173367500305176,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19083526730537415,
      "step": 24862
    },
    {
      "epoch": 0.49728,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.011138661702473959,
      "learning_rate": 0.0001,
      "loss": 4.2694,
      "loss/crossentropy": 2.0459975004196167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20265580713748932,
      "step": 24864
    },
    {
      "epoch": 0.49732,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009471638997395834,
      "learning_rate": 0.0001,
      "loss": 3.9095,
      "loss/crossentropy": 1.9144663214683533,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18860550224781036,
      "step": 24866
    },
    {
      "epoch": 0.49736,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.010762278238932292,
      "learning_rate": 0.0001,
      "loss": 3.8107,
      "loss/crossentropy": 1.6619080901145935,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17097505927085876,
      "step": 24868
    },
    {
      "epoch": 0.4974,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.014192708333333333,
      "learning_rate": 0.0001,
      "loss": 4.2346,
      "loss/crossentropy": 2.031949281692505,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21831347048282623,
      "step": 24870
    },
    {
      "epoch": 0.49744,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.014869944254557291,
      "learning_rate": 0.0001,
      "loss": 4.098,
      "loss/crossentropy": 2.330108165740967,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21315570175647736,
      "step": 24872
    },
    {
      "epoch": 0.49748,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.014876302083333333,
      "learning_rate": 0.0001,
      "loss": 3.8719,
      "loss/crossentropy": 2.301952600479126,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18241943418979645,
      "step": 24874
    },
    {
      "epoch": 0.49752,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.013634999593098959,
      "learning_rate": 0.0001,
      "loss": 4.3927,
      "loss/crossentropy": 2.3587260246276855,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19822461903095245,
      "step": 24876
    },
    {
      "epoch": 0.49756,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.012729644775390625,
      "learning_rate": 0.0001,
      "loss": 3.7801,
      "loss/crossentropy": 1.8675512671470642,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19025489687919617,
      "step": 24878
    },
    {
      "epoch": 0.4976,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.027367146809895833,
      "learning_rate": 0.0001,
      "loss": 3.8605,
      "loss/crossentropy": 1.9007975459098816,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18459592014551163,
      "step": 24880
    },
    {
      "epoch": 0.49764,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.026009114583333333,
      "learning_rate": 0.0001,
      "loss": 4.1207,
      "loss/crossentropy": 1.9351030588150024,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17839118093252182,
      "step": 24882
    },
    {
      "epoch": 0.49768,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.02781956990559896,
      "learning_rate": 0.0001,
      "loss": 4.2804,
      "loss/crossentropy": 2.2661246061325073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19518450647592545,
      "step": 24884
    },
    {
      "epoch": 0.49772,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.025333658854166666,
      "learning_rate": 0.0001,
      "loss": 4.2568,
      "loss/crossentropy": 2.1555867791175842,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20517798513174057,
      "step": 24886
    },
    {
      "epoch": 0.49776,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.023067220052083334,
      "learning_rate": 0.0001,
      "loss": 4.0291,
      "loss/crossentropy": 2.1638144850730896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18970053642988205,
      "step": 24888
    },
    {
      "epoch": 0.4978,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.023280588785807292,
      "learning_rate": 0.0001,
      "loss": 3.9658,
      "loss/crossentropy": 2.230544328689575,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20187361538410187,
      "step": 24890
    },
    {
      "epoch": 0.49784,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.02308349609375,
      "learning_rate": 0.0001,
      "loss": 4.1406,
      "loss/crossentropy": 2.1005473136901855,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20453108847141266,
      "step": 24892
    },
    {
      "epoch": 0.49788,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0233642578125,
      "learning_rate": 0.0001,
      "loss": 3.898,
      "loss/crossentropy": 2.056541621685028,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18511568009853363,
      "step": 24894
    },
    {
      "epoch": 0.49792,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.011407216389973959,
      "learning_rate": 0.0001,
      "loss": 4.2019,
      "loss/crossentropy": 2.151825726032257,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19390379637479782,
      "step": 24896
    },
    {
      "epoch": 0.49796,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011726633707682291,
      "learning_rate": 0.0001,
      "loss": 4.159,
      "loss/crossentropy": 2.2062729597091675,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21959953010082245,
      "step": 24898
    },
    {
      "epoch": 0.498,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.007835896809895833,
      "learning_rate": 0.0001,
      "loss": 4.0506,
      "loss/crossentropy": 2.11991286277771,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20495782047510147,
      "step": 24900
    },
    {
      "epoch": 0.49804,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.007785797119140625,
      "learning_rate": 0.0001,
      "loss": 3.8555,
      "loss/crossentropy": 1.9586367011070251,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18644800782203674,
      "step": 24902
    },
    {
      "epoch": 0.49808,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.008665974934895833,
      "learning_rate": 0.0001,
      "loss": 3.987,
      "loss/crossentropy": 2.030204951763153,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1938316822052002,
      "step": 24904
    },
    {
      "epoch": 0.49812,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.00999755859375,
      "learning_rate": 0.0001,
      "loss": 4.2644,
      "loss/crossentropy": 2.063572645187378,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19029874354600906,
      "step": 24906
    },
    {
      "epoch": 0.49816,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.010081990559895834,
      "learning_rate": 0.0001,
      "loss": 3.8608,
      "loss/crossentropy": 1.8356643319129944,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18317610025405884,
      "step": 24908
    },
    {
      "epoch": 0.4982,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.08252665201822916,
      "learning_rate": 0.0001,
      "loss": 4.1617,
      "loss/crossentropy": 1.8856282830238342,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1754850372672081,
      "step": 24910
    },
    {
      "epoch": 0.49824,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.08241780598958333,
      "learning_rate": 0.0001,
      "loss": 3.9649,
      "loss/crossentropy": 1.8674440383911133,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1805490255355835,
      "step": 24912
    },
    {
      "epoch": 0.49828,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.08478978474934896,
      "learning_rate": 0.0001,
      "loss": 3.839,
      "loss/crossentropy": 1.8774593472480774,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18495067209005356,
      "step": 24914
    },
    {
      "epoch": 0.49832,
      "grad_norm": 1.71875,
      "grad_norm_var": 0.08945210774739583,
      "learning_rate": 0.0001,
      "loss": 3.9121,
      "loss/crossentropy": 2.059852659702301,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19449050724506378,
      "step": 24916
    },
    {
      "epoch": 0.49836,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.08824869791666666,
      "learning_rate": 0.0001,
      "loss": 4.2468,
      "loss/crossentropy": 2.3840794563293457,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23490352928638458,
      "step": 24918
    },
    {
      "epoch": 0.4984,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.08859049479166667,
      "learning_rate": 0.0001,
      "loss": 4.1537,
      "loss/crossentropy": 2.136437773704529,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19557444751262665,
      "step": 24920
    },
    {
      "epoch": 0.49844,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.08821614583333333,
      "learning_rate": 0.0001,
      "loss": 4.3658,
      "loss/crossentropy": 2.4357622861862183,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21140636503696442,
      "step": 24922
    },
    {
      "epoch": 0.49848,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.08875223795572916,
      "learning_rate": 0.0001,
      "loss": 3.9085,
      "loss/crossentropy": 2.1498345136642456,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18728043884038925,
      "step": 24924
    },
    {
      "epoch": 0.49852,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.012580362955729167,
      "learning_rate": 0.0001,
      "loss": 4.0771,
      "loss/crossentropy": 2.131583571434021,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21287915110588074,
      "step": 24926
    },
    {
      "epoch": 0.49856,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.012505849202473959,
      "learning_rate": 0.0001,
      "loss": 4.1492,
      "loss/crossentropy": 2.434916377067566,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21438215672969818,
      "step": 24928
    },
    {
      "epoch": 0.4986,
      "grad_norm": 1.6875,
      "grad_norm_var": 0.016397857666015626,
      "learning_rate": 0.0001,
      "loss": 3.8951,
      "loss/crossentropy": 1.9909663796424866,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16714239120483398,
      "step": 24930
    },
    {
      "epoch": 0.49864,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.01343994140625,
      "learning_rate": 0.0001,
      "loss": 4.0229,
      "loss/crossentropy": 1.9362955689430237,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18765835464000702,
      "step": 24932
    },
    {
      "epoch": 0.49868,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.011527252197265626,
      "learning_rate": 0.0001,
      "loss": 3.888,
      "loss/crossentropy": 1.909186065196991,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19517959654331207,
      "step": 24934
    },
    {
      "epoch": 0.49872,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.010350545247395834,
      "learning_rate": 0.0001,
      "loss": 4.0523,
      "loss/crossentropy": 1.9746766686439514,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19376099109649658,
      "step": 24936
    },
    {
      "epoch": 0.49876,
      "grad_norm": 2.0,
      "grad_norm_var": 0.006844075520833334,
      "learning_rate": 0.0001,
      "loss": 4.0004,
      "loss/crossentropy": 2.0698294639587402,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1993079036474228,
      "step": 24938
    },
    {
      "epoch": 0.4988,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.007453409830729166,
      "learning_rate": 0.0001,
      "loss": 4.1234,
      "loss/crossentropy": 1.8996255993843079,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1822480708360672,
      "step": 24940
    },
    {
      "epoch": 0.49884,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0114898681640625,
      "learning_rate": 0.0001,
      "loss": 4.4206,
      "loss/crossentropy": 1.9174728989601135,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18220385164022446,
      "step": 24942
    },
    {
      "epoch": 0.49888,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.0126953125,
      "learning_rate": 0.0001,
      "loss": 3.6162,
      "loss/crossentropy": 1.827277660369873,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17845244705677032,
      "step": 24944
    },
    {
      "epoch": 0.49892,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.009264882405598958,
      "learning_rate": 0.0001,
      "loss": 4.1165,
      "loss/crossentropy": 2.0837597846984863,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19145962595939636,
      "step": 24946
    },
    {
      "epoch": 0.49896,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.008915201822916666,
      "learning_rate": 0.0001,
      "loss": 3.969,
      "loss/crossentropy": 2.417944073677063,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22484175115823746,
      "step": 24948
    },
    {
      "epoch": 0.499,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009639485677083334,
      "learning_rate": 0.0001,
      "loss": 4.0408,
      "loss/crossentropy": 2.338744640350342,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2132393717765808,
      "step": 24950
    },
    {
      "epoch": 0.49904,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.009559885660807291,
      "learning_rate": 0.0001,
      "loss": 3.9549,
      "loss/crossentropy": 2.1383684277534485,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19005178660154343,
      "step": 24952
    },
    {
      "epoch": 0.49908,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.010731760660807292,
      "learning_rate": 0.0001,
      "loss": 4.1188,
      "loss/crossentropy": 1.8223644495010376,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17134226113557816,
      "step": 24954
    },
    {
      "epoch": 0.49912,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.017228190104166666,
      "learning_rate": 0.0001,
      "loss": 3.9886,
      "loss/crossentropy": 2.09305202960968,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19109848141670227,
      "step": 24956
    },
    {
      "epoch": 0.49916,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.017844390869140626,
      "learning_rate": 0.0001,
      "loss": 3.7533,
      "loss/crossentropy": 1.9903115034103394,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18018385022878647,
      "step": 24958
    },
    {
      "epoch": 0.4992,
      "grad_norm": 2.0,
      "grad_norm_var": 0.02005182902018229,
      "learning_rate": 0.0001,
      "loss": 4.0391,
      "loss/crossentropy": 1.9048819541931152,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18057531118392944,
      "step": 24960
    },
    {
      "epoch": 0.49924,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.019505818684895832,
      "learning_rate": 0.0001,
      "loss": 4.0654,
      "loss/crossentropy": 2.045142412185669,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18374620378017426,
      "step": 24962
    },
    {
      "epoch": 0.49928,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.019798787434895833,
      "learning_rate": 0.0001,
      "loss": 4.0365,
      "loss/crossentropy": 2.094703733921051,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21248038858175278,
      "step": 24964
    },
    {
      "epoch": 0.49932,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.020359039306640625,
      "learning_rate": 0.0001,
      "loss": 3.7111,
      "loss/crossentropy": 2.113132894039154,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19428254663944244,
      "step": 24966
    },
    {
      "epoch": 0.49936,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.019123331705729166,
      "learning_rate": 0.0001,
      "loss": 4.2238,
      "loss/crossentropy": 2.0182060599327087,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22048161923885345,
      "step": 24968
    },
    {
      "epoch": 0.4994,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.01933568318684896,
      "learning_rate": 0.0001,
      "loss": 4.0755,
      "loss/crossentropy": 2.141752541065216,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.198372982442379,
      "step": 24970
    },
    {
      "epoch": 0.49944,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.012737782796223958,
      "learning_rate": 0.0001,
      "loss": 4.0035,
      "loss/crossentropy": 1.9773234724998474,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18223125487565994,
      "step": 24972
    },
    {
      "epoch": 0.49948,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.009464518229166666,
      "learning_rate": 0.0001,
      "loss": 3.9405,
      "loss/crossentropy": 1.7909184098243713,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17887402325868607,
      "step": 24974
    },
    {
      "epoch": 0.49952,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0048411051432291664,
      "learning_rate": 0.0001,
      "loss": 4.0024,
      "loss/crossentropy": 1.9185696840286255,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19022425264120102,
      "step": 24976
    },
    {
      "epoch": 0.49956,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.005472819010416667,
      "learning_rate": 0.0001,
      "loss": 4.2593,
      "loss/crossentropy": 1.7053416967391968,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18504460155963898,
      "step": 24978
    },
    {
      "epoch": 0.4996,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.006156158447265625,
      "learning_rate": 0.0001,
      "loss": 4.2626,
      "loss/crossentropy": 2.1260892152786255,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1958574280142784,
      "step": 24980
    },
    {
      "epoch": 0.49964,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.005537923177083333,
      "learning_rate": 0.0001,
      "loss": 3.9854,
      "loss/crossentropy": 1.9965519905090332,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20814893394708633,
      "step": 24982
    },
    {
      "epoch": 0.49968,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.3696734110514323,
      "learning_rate": 0.0001,
      "loss": 4.1443,
      "loss/crossentropy": 1.882089376449585,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23063261061906815,
      "step": 24984
    },
    {
      "epoch": 0.49972,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.36444498697916666,
      "learning_rate": 0.0001,
      "loss": 3.9484,
      "loss/crossentropy": 2.0816534757614136,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1818612888455391,
      "step": 24986
    },
    {
      "epoch": 0.49976,
      "grad_norm": 1.875,
      "grad_norm_var": 0.3668365478515625,
      "learning_rate": 0.0001,
      "loss": 3.794,
      "loss/crossentropy": 1.7514638304710388,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18518543243408203,
      "step": 24988
    },
    {
      "epoch": 0.4998,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.37027969360351565,
      "learning_rate": 0.0001,
      "loss": 4.095,
      "loss/crossentropy": 2.438727021217346,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2042185142636299,
      "step": 24990
    },
    {
      "epoch": 0.49984,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.3688087463378906,
      "learning_rate": 0.0001,
      "loss": 4.1072,
      "loss/crossentropy": 2.0413439869880676,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19952382892370224,
      "step": 24992
    },
    {
      "epoch": 0.49988,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.3713612874348958,
      "learning_rate": 0.0001,
      "loss": 4.1332,
      "loss/crossentropy": 2.1857372522354126,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18842580169439316,
      "step": 24994
    },
    {
      "epoch": 0.49992,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.36927261352539065,
      "learning_rate": 0.0001,
      "loss": 4.1855,
      "loss/crossentropy": 1.9340556859970093,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24346761405467987,
      "step": 24996
    },
    {
      "epoch": 0.49996,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.3735389709472656,
      "learning_rate": 0.0001,
      "loss": 3.9177,
      "loss/crossentropy": 1.9736055731773376,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20419831573963165,
      "step": 24998
    },
    {
      "epoch": 0.5,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0122314453125,
      "learning_rate": 0.0001,
      "loss": 4.0167,
      "loss/crossentropy": 1.8731598258018494,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20324288308620453,
      "step": 25000
    },
    {
      "epoch": 0.50004,
      "grad_norm": 2.0,
      "grad_norm_var": 0.012645467122395834,
      "learning_rate": 0.0001,
      "loss": 4.1379,
      "loss/crossentropy": 1.9114505648612976,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19572290033102036,
      "step": 25002
    },
    {
      "epoch": 0.50008,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.016022745768229166,
      "learning_rate": 0.0001,
      "loss": 4.0389,
      "loss/crossentropy": 2.0608550310134888,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20759478956460953,
      "step": 25004
    },
    {
      "epoch": 0.50012,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.015529123942057292,
      "learning_rate": 0.0001,
      "loss": 4.0434,
      "loss/crossentropy": 2.0759811401367188,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1895374357700348,
      "step": 25006
    },
    {
      "epoch": 0.50016,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.012859853108723958,
      "learning_rate": 0.0001,
      "loss": 3.7772,
      "loss/crossentropy": 2.139121353626251,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19550149887800217,
      "step": 25008
    },
    {
      "epoch": 0.5002,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.013410441080729167,
      "learning_rate": 0.0001,
      "loss": 3.9163,
      "loss/crossentropy": 1.8311315774917603,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17756005376577377,
      "step": 25010
    },
    {
      "epoch": 0.50024,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.012311808268229167,
      "learning_rate": 0.0001,
      "loss": 3.9897,
      "loss/crossentropy": 2.0824968814849854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21674959361553192,
      "step": 25012
    },
    {
      "epoch": 0.50028,
      "grad_norm": 2.0,
      "grad_norm_var": 0.018317667643229167,
      "learning_rate": 0.0001,
      "loss": 4.4713,
      "loss/crossentropy": 2.3479214906692505,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22297757118940353,
      "step": 25014
    },
    {
      "epoch": 0.50032,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.017682902018229165,
      "learning_rate": 0.0001,
      "loss": 4.2473,
      "loss/crossentropy": 2.477561354637146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2347298264503479,
      "step": 25016
    },
    {
      "epoch": 0.50036,
      "grad_norm": 1.875,
      "grad_norm_var": 0.020369466145833334,
      "learning_rate": 0.0001,
      "loss": 3.797,
      "loss/crossentropy": 1.7736787796020508,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1656038835644722,
      "step": 25018
    },
    {
      "epoch": 0.5004,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.017130533854166668,
      "learning_rate": 0.0001,
      "loss": 4.1537,
      "loss/crossentropy": 2.2568663358688354,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19755709171295166,
      "step": 25020
    },
    {
      "epoch": 0.50044,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.018668619791666667,
      "learning_rate": 0.0001,
      "loss": 3.9279,
      "loss/crossentropy": 1.961386263370514,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2079472467303276,
      "step": 25022
    },
    {
      "epoch": 0.50048,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.01898981730143229,
      "learning_rate": 0.0001,
      "loss": 3.7884,
      "loss/crossentropy": 1.75395268201828,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16586285829544067,
      "step": 25024
    },
    {
      "epoch": 0.50052,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0171142578125,
      "learning_rate": 0.0001,
      "loss": 3.9074,
      "loss/crossentropy": 2.1727951169013977,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21714362502098083,
      "step": 25026
    },
    {
      "epoch": 0.50056,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.017765299479166666,
      "learning_rate": 0.0001,
      "loss": 4.276,
      "loss/crossentropy": 2.147822380065918,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19961202144622803,
      "step": 25028
    },
    {
      "epoch": 0.5006,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.012775675455729166,
      "learning_rate": 0.0001,
      "loss": 4.1055,
      "loss/crossentropy": 2.0300870537757874,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19732433557510376,
      "step": 25030
    },
    {
      "epoch": 0.50064,
      "grad_norm": 1.875,
      "grad_norm_var": 0.019828287760416667,
      "learning_rate": 0.0001,
      "loss": 4.0727,
      "loss/crossentropy": 1.840386688709259,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18403710424900055,
      "step": 25032
    },
    {
      "epoch": 0.50068,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.01875,
      "learning_rate": 0.0001,
      "loss": 3.8442,
      "loss/crossentropy": 1.9718754291534424,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19007521867752075,
      "step": 25034
    },
    {
      "epoch": 0.50072,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.022468058268229167,
      "learning_rate": 0.0001,
      "loss": 3.8137,
      "loss/crossentropy": 1.9550666809082031,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1944015771150589,
      "step": 25036
    },
    {
      "epoch": 0.50076,
      "grad_norm": 1.75,
      "grad_norm_var": 0.024397786458333334,
      "learning_rate": 0.0001,
      "loss": 3.7467,
      "loss/crossentropy": 1.9608187079429626,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1774805188179016,
      "step": 25038
    },
    {
      "epoch": 0.5008,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0225494384765625,
      "learning_rate": 0.0001,
      "loss": 3.7607,
      "loss/crossentropy": 1.6980991959571838,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16579563170671463,
      "step": 25040
    },
    {
      "epoch": 0.50084,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.02269261678059896,
      "learning_rate": 0.0001,
      "loss": 3.9008,
      "loss/crossentropy": 1.8796940445899963,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1857006549835205,
      "step": 25042
    },
    {
      "epoch": 0.50088,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.02159398396809896,
      "learning_rate": 0.0001,
      "loss": 3.7862,
      "loss/crossentropy": 1.8667701482772827,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1920718252658844,
      "step": 25044
    },
    {
      "epoch": 0.50092,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.019676717122395833,
      "learning_rate": 0.0001,
      "loss": 4.11,
      "loss/crossentropy": 2.142220616340637,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20785054564476013,
      "step": 25046
    },
    {
      "epoch": 0.50096,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.011185709635416667,
      "learning_rate": 0.0001,
      "loss": 4.1763,
      "loss/crossentropy": 2.1965246200561523,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2001591995358467,
      "step": 25048
    },
    {
      "epoch": 0.501,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.01103515625,
      "learning_rate": 0.0001,
      "loss": 3.9254,
      "loss/crossentropy": 2.1812084913253784,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21115607023239136,
      "step": 25050
    },
    {
      "epoch": 0.50104,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.00670166015625,
      "learning_rate": 0.0001,
      "loss": 4.1107,
      "loss/crossentropy": 1.8218246698379517,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20282388478517532,
      "step": 25052
    },
    {
      "epoch": 0.50108,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0037750244140625,
      "learning_rate": 0.0001,
      "loss": 3.9292,
      "loss/crossentropy": 2.2070316076278687,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21164244413375854,
      "step": 25054
    },
    {
      "epoch": 0.50112,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.00509033203125,
      "learning_rate": 0.0001,
      "loss": 4.0604,
      "loss/crossentropy": 2.1380701065063477,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20642387121915817,
      "step": 25056
    },
    {
      "epoch": 0.50116,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.004671223958333333,
      "learning_rate": 0.0001,
      "loss": 4.188,
      "loss/crossentropy": 2.2019251585006714,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20543986558914185,
      "step": 25058
    },
    {
      "epoch": 0.5012,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.006583658854166666,
      "learning_rate": 0.0001,
      "loss": 3.9042,
      "loss/crossentropy": 2.0563461780548096,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18677344918251038,
      "step": 25060
    },
    {
      "epoch": 0.50124,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0067291259765625,
      "learning_rate": 0.0001,
      "loss": 4.0295,
      "loss/crossentropy": 1.788195252418518,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18303587287664413,
      "step": 25062
    },
    {
      "epoch": 0.50128,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0079254150390625,
      "learning_rate": 0.0001,
      "loss": 3.7124,
      "loss/crossentropy": 1.888840913772583,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1736098676919937,
      "step": 25064
    },
    {
      "epoch": 0.50132,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.006139882405598958,
      "learning_rate": 0.0001,
      "loss": 4.0185,
      "loss/crossentropy": 2.0963358283042908,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1944149062037468,
      "step": 25066
    },
    {
      "epoch": 0.50136,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.006322224934895833,
      "learning_rate": 0.0001,
      "loss": 4.0357,
      "loss/crossentropy": 2.3116443157196045,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22342295944690704,
      "step": 25068
    },
    {
      "epoch": 0.5014,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.006266276041666667,
      "learning_rate": 0.0001,
      "loss": 4.2353,
      "loss/crossentropy": 2.251197099685669,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25324299186468124,
      "step": 25070
    },
    {
      "epoch": 0.50144,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.004613240559895833,
      "learning_rate": 0.0001,
      "loss": 3.7142,
      "loss/crossentropy": 2.2114810943603516,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18877273797988892,
      "step": 25072
    },
    {
      "epoch": 0.50148,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.004923248291015625,
      "learning_rate": 0.0001,
      "loss": 4.1174,
      "loss/crossentropy": 2.266746401786804,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21522346138954163,
      "step": 25074
    },
    {
      "epoch": 0.50152,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.004809315999348958,
      "learning_rate": 0.0001,
      "loss": 4.2369,
      "loss/crossentropy": 1.922059416770935,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17906102538108826,
      "step": 25076
    },
    {
      "epoch": 0.50156,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.005891672770182292,
      "learning_rate": 0.0001,
      "loss": 4.0695,
      "loss/crossentropy": 2.077875077724457,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18431179225444794,
      "step": 25078
    },
    {
      "epoch": 0.5016,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.006135050455729167,
      "learning_rate": 0.0001,
      "loss": 3.7519,
      "loss/crossentropy": 1.8861806988716125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17609698325395584,
      "step": 25080
    },
    {
      "epoch": 0.50164,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.008302561442057292,
      "learning_rate": 0.0001,
      "loss": 4.0371,
      "loss/crossentropy": 2.1382817029953003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21308781206607819,
      "step": 25082
    },
    {
      "epoch": 0.50168,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008656565348307292,
      "learning_rate": 0.0001,
      "loss": 3.9133,
      "loss/crossentropy": 1.9955537915229797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20031572878360748,
      "step": 25084
    },
    {
      "epoch": 0.50172,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0108551025390625,
      "learning_rate": 0.0001,
      "loss": 4.0816,
      "loss/crossentropy": 2.2564213275909424,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19504930824041367,
      "step": 25086
    },
    {
      "epoch": 0.50176,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.01778132120768229,
      "learning_rate": 0.0001,
      "loss": 4.2432,
      "loss/crossentropy": 2.388614296913147,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2021147906780243,
      "step": 25088
    },
    {
      "epoch": 0.5018,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.016764322916666668,
      "learning_rate": 0.0001,
      "loss": 4.0045,
      "loss/crossentropy": 2.5410468578338623,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23654652386903763,
      "step": 25090
    },
    {
      "epoch": 0.50184,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.016255696614583332,
      "learning_rate": 0.0001,
      "loss": 4.0248,
      "loss/crossentropy": 1.9850627183914185,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1861669197678566,
      "step": 25092
    },
    {
      "epoch": 0.50188,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.01573486328125,
      "learning_rate": 0.0001,
      "loss": 4.2318,
      "loss/crossentropy": 2.3069719076156616,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22062674164772034,
      "step": 25094
    },
    {
      "epoch": 0.50192,
      "grad_norm": 1.875,
      "grad_norm_var": 0.013960520426432291,
      "learning_rate": 0.0001,
      "loss": 3.9187,
      "loss/crossentropy": 1.7462975978851318,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17929460108280182,
      "step": 25096
    },
    {
      "epoch": 0.50196,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.012870025634765626,
      "learning_rate": 0.0001,
      "loss": 4.0647,
      "loss/crossentropy": 1.8277402520179749,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17240776866674423,
      "step": 25098
    },
    {
      "epoch": 0.502,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.013978830973307292,
      "learning_rate": 0.0001,
      "loss": 4.0331,
      "loss/crossentropy": 2.2530910968780518,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20934444665908813,
      "step": 25100
    },
    {
      "epoch": 0.50204,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0118316650390625,
      "learning_rate": 0.0001,
      "loss": 3.9368,
      "loss/crossentropy": 1.8995028138160706,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18956587463617325,
      "step": 25102
    },
    {
      "epoch": 0.50208,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.0044553120930989586,
      "learning_rate": 0.0001,
      "loss": 3.8491,
      "loss/crossentropy": 1.9500588774681091,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18196535110473633,
      "step": 25104
    },
    {
      "epoch": 0.50212,
      "grad_norm": 1.875,
      "grad_norm_var": 0.0037750244140625,
      "learning_rate": 0.0001,
      "loss": 3.8702,
      "loss/crossentropy": 1.794252336025238,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17875805497169495,
      "step": 25106
    },
    {
      "epoch": 0.50216,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.004124959309895833,
      "learning_rate": 0.0001,
      "loss": 3.9752,
      "loss/crossentropy": 2.282618999481201,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18928487598896027,
      "step": 25108
    },
    {
      "epoch": 0.5022,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0036173502604166666,
      "learning_rate": 0.0001,
      "loss": 4.1439,
      "loss/crossentropy": 1.8668266534805298,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1754981055855751,
      "step": 25110
    },
    {
      "epoch": 0.50224,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0038859049479166665,
      "learning_rate": 0.0001,
      "loss": 4.0509,
      "loss/crossentropy": 2.416601300239563,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21391690522432327,
      "step": 25112
    },
    {
      "epoch": 0.50228,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.003311920166015625,
      "learning_rate": 0.0001,
      "loss": 3.914,
      "loss/crossentropy": 1.9013903141021729,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16768474876880646,
      "step": 25114
    },
    {
      "epoch": 0.50232,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0035807291666666665,
      "learning_rate": 0.0001,
      "loss": 3.8587,
      "loss/crossentropy": 2.090600550174713,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1963217630982399,
      "step": 25116
    },
    {
      "epoch": 0.50236,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.003929646809895834,
      "learning_rate": 0.0001,
      "loss": 4.0404,
      "loss/crossentropy": 1.938132882118225,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17899858951568604,
      "step": 25118
    },
    {
      "epoch": 0.5024,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0031613667805989584,
      "learning_rate": 0.0001,
      "loss": 4.1818,
      "loss/crossentropy": 2.2119863033294678,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20339767634868622,
      "step": 25120
    },
    {
      "epoch": 0.50244,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0031809488932291667,
      "learning_rate": 0.0001,
      "loss": 4.2186,
      "loss/crossentropy": 2.35983407497406,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2063140720129013,
      "step": 25122
    },
    {
      "epoch": 0.50248,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.003830718994140625,
      "learning_rate": 0.0001,
      "loss": 4.2016,
      "loss/crossentropy": 2.0203242897987366,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19055970013141632,
      "step": 25124
    },
    {
      "epoch": 0.50252,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.0050961812337239586,
      "learning_rate": 0.0001,
      "loss": 3.6522,
      "loss/crossentropy": 1.7058457732200623,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1857994720339775,
      "step": 25126
    },
    {
      "epoch": 0.50256,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.0054514567057291664,
      "learning_rate": 0.0001,
      "loss": 3.7193,
      "loss/crossentropy": 1.720770239830017,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1738799437880516,
      "step": 25128
    },
    {
      "epoch": 0.5026,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.006247711181640625,
      "learning_rate": 0.0001,
      "loss": 3.8967,
      "loss/crossentropy": 1.677247405052185,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18985359370708466,
      "step": 25130
    },
    {
      "epoch": 0.50264,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0058977762858072914,
      "learning_rate": 0.0001,
      "loss": 4.0737,
      "loss/crossentropy": 2.1808619499206543,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19232448935508728,
      "step": 25132
    },
    {
      "epoch": 0.50268,
      "grad_norm": 1.6953125,
      "grad_norm_var": 0.008512369791666667,
      "learning_rate": 0.0001,
      "loss": 3.8655,
      "loss/crossentropy": 1.8475935459136963,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18105845898389816,
      "step": 25134
    },
    {
      "epoch": 0.50272,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.012052154541015625,
      "learning_rate": 0.0001,
      "loss": 3.8568,
      "loss/crossentropy": 2.0687568187713623,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1997617706656456,
      "step": 25136
    },
    {
      "epoch": 0.50276,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.01256103515625,
      "learning_rate": 0.0001,
      "loss": 4.29,
      "loss/crossentropy": 2.405083179473877,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22054974734783173,
      "step": 25138
    },
    {
      "epoch": 0.5028,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.014037068684895833,
      "learning_rate": 0.0001,
      "loss": 4.0613,
      "loss/crossentropy": 1.917975127696991,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20000945031642914,
      "step": 25140
    },
    {
      "epoch": 0.50284,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.017010243733723958,
      "learning_rate": 0.0001,
      "loss": 4.1259,
      "loss/crossentropy": 2.270500898361206,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21464426815509796,
      "step": 25142
    },
    {
      "epoch": 0.50288,
      "grad_norm": 1.875,
      "grad_norm_var": 0.017145792643229168,
      "learning_rate": 0.0001,
      "loss": 4.0141,
      "loss/crossentropy": 2.0085031390190125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20128140598535538,
      "step": 25144
    },
    {
      "epoch": 0.50292,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0150299072265625,
      "learning_rate": 0.0001,
      "loss": 4.1969,
      "loss/crossentropy": 2.206787347793579,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19804110378026962,
      "step": 25146
    },
    {
      "epoch": 0.50296,
      "grad_norm": 1.875,
      "grad_norm_var": 0.015428670247395833,
      "learning_rate": 0.0001,
      "loss": 3.8519,
      "loss/crossentropy": 1.7760102152824402,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18732204288244247,
      "step": 25148
    },
    {
      "epoch": 0.503,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.011439768473307292,
      "learning_rate": 0.0001,
      "loss": 4.1334,
      "loss/crossentropy": 1.9429696202278137,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21402037143707275,
      "step": 25150
    },
    {
      "epoch": 0.50304,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.007315826416015625,
      "learning_rate": 0.0001,
      "loss": 3.8488,
      "loss/crossentropy": 1.6982505321502686,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16420165449380875,
      "step": 25152
    },
    {
      "epoch": 0.50308,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0076324462890625,
      "learning_rate": 0.0001,
      "loss": 4.0041,
      "loss/crossentropy": 1.998267948627472,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.193763867020607,
      "step": 25154
    },
    {
      "epoch": 0.50312,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.009917958577473959,
      "learning_rate": 0.0001,
      "loss": 4.0845,
      "loss/crossentropy": 2.4127193689346313,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2271365001797676,
      "step": 25156
    },
    {
      "epoch": 0.50316,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.0094390869140625,
      "learning_rate": 0.0001,
      "loss": 3.8656,
      "loss/crossentropy": 1.5740959644317627,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17892491817474365,
      "step": 25158
    },
    {
      "epoch": 0.5032,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.010205078125,
      "learning_rate": 0.0001,
      "loss": 4.079,
      "loss/crossentropy": 2.1531224250793457,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20502899587154388,
      "step": 25160
    },
    {
      "epoch": 0.50324,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.0105621337890625,
      "learning_rate": 0.0001,
      "loss": 3.9895,
      "loss/crossentropy": 2.024933695793152,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21745888888835907,
      "step": 25162
    },
    {
      "epoch": 0.50328,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.010465240478515625,
      "learning_rate": 0.0001,
      "loss": 3.7801,
      "loss/crossentropy": 1.7188761234283447,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1754877120256424,
      "step": 25164
    },
    {
      "epoch": 0.50332,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.009642537434895833,
      "learning_rate": 0.0001,
      "loss": 3.9803,
      "loss/crossentropy": 2.1078373193740845,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1812133714556694,
      "step": 25166
    },
    {
      "epoch": 0.50336,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.010949452718098959,
      "learning_rate": 0.0001,
      "loss": 3.753,
      "loss/crossentropy": 1.793417751789093,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1709349974989891,
      "step": 25168
    },
    {
      "epoch": 0.5034,
      "grad_norm": 1.734375,
      "grad_norm_var": 0.014115142822265624,
      "learning_rate": 0.0001,
      "loss": 3.2689,
      "loss/crossentropy": 1.357922375202179,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.14589300006628036,
      "step": 25170
    },
    {
      "epoch": 0.50344,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.014247385660807292,
      "learning_rate": 0.0001,
      "loss": 4.4387,
      "loss/crossentropy": 2.3469592332839966,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20105785876512527,
      "step": 25172
    },
    {
      "epoch": 0.50348,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.014159138997395833,
      "learning_rate": 0.0001,
      "loss": 3.8041,
      "loss/crossentropy": 2.1891257762908936,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21113110333681107,
      "step": 25174
    },
    {
      "epoch": 0.50352,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.012878163655598959,
      "learning_rate": 0.0001,
      "loss": 3.9236,
      "loss/crossentropy": 2.201194167137146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21105756610631943,
      "step": 25176
    },
    {
      "epoch": 0.50356,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.012702433268229167,
      "learning_rate": 0.0001,
      "loss": 4.0474,
      "loss/crossentropy": 1.8287039995193481,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18074452877044678,
      "step": 25178
    },
    {
      "epoch": 0.5036,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.012813313802083334,
      "learning_rate": 0.0001,
      "loss": 4.1655,
      "loss/crossentropy": 1.8915085792541504,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1997976452112198,
      "step": 25180
    },
    {
      "epoch": 0.50364,
      "grad_norm": 1.7109375,
      "grad_norm_var": 0.014509836832682291,
      "learning_rate": 0.0001,
      "loss": 3.8989,
      "loss/crossentropy": 2.102576494216919,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18678200989961624,
      "step": 25182
    },
    {
      "epoch": 0.50368,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.014898427327473958,
      "learning_rate": 0.0001,
      "loss": 4.3278,
      "loss/crossentropy": 2.1603941917419434,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20021572709083557,
      "step": 25184
    },
    {
      "epoch": 0.50372,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.011263020833333333,
      "learning_rate": 0.0001,
      "loss": 3.6356,
      "loss/crossentropy": 1.8718876838684082,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18388725817203522,
      "step": 25186
    },
    {
      "epoch": 0.50376,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0069272359212239586,
      "learning_rate": 0.0001,
      "loss": 4.1273,
      "loss/crossentropy": 2.2126933336257935,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20829670876264572,
      "step": 25188
    },
    {
      "epoch": 0.5038,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.0072743733723958336,
      "learning_rate": 0.0001,
      "loss": 4.0426,
      "loss/crossentropy": 2.1955376863479614,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19484563171863556,
      "step": 25190
    },
    {
      "epoch": 0.50384,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.0067535400390625,
      "learning_rate": 0.0001,
      "loss": 4.0132,
      "loss/crossentropy": 1.9294677376747131,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18266455084085464,
      "step": 25192
    },
    {
      "epoch": 0.50388,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.012851715087890625,
      "learning_rate": 0.0001,
      "loss": 4.1586,
      "loss/crossentropy": 1.867956280708313,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2331952303647995,
      "step": 25194
    },
    {
      "epoch": 0.50392,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.012984212239583333,
      "learning_rate": 0.0001,
      "loss": 3.8322,
      "loss/crossentropy": 1.7810456156730652,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1834517866373062,
      "step": 25196
    },
    {
      "epoch": 0.50396,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.010711415608723959,
      "learning_rate": 0.0001,
      "loss": 3.7718,
      "loss/crossentropy": 2.148942291736603,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2065306007862091,
      "step": 25198
    },
    {
      "epoch": 0.504,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.011291249593098959,
      "learning_rate": 0.0001,
      "loss": 3.9163,
      "loss/crossentropy": 1.730657696723938,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18555008620023727,
      "step": 25200
    },
    {
      "epoch": 0.50404,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.014190419514973959,
      "learning_rate": 0.0001,
      "loss": 3.9587,
      "loss/crossentropy": 2.2561585903167725,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22614946961402893,
      "step": 25202
    },
    {
      "epoch": 0.50408,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.014324696858723958,
      "learning_rate": 0.0001,
      "loss": 4.1299,
      "loss/crossentropy": 2.2798372507095337,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19820939004421234,
      "step": 25204
    },
    {
      "epoch": 0.50412,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.014582316080729166,
      "learning_rate": 0.0001,
      "loss": 4.1232,
      "loss/crossentropy": 2.2171024680137634,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22784201800823212,
      "step": 25206
    },
    {
      "epoch": 0.50416,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.016283162434895835,
      "learning_rate": 0.0001,
      "loss": 3.8819,
      "loss/crossentropy": 1.867311179637909,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19023532420396805,
      "step": 25208
    },
    {
      "epoch": 0.5042,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.009506988525390624,
      "learning_rate": 0.0001,
      "loss": 3.843,
      "loss/crossentropy": 2.0274672508239746,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19995202869176865,
      "step": 25210
    },
    {
      "epoch": 0.50424,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.009663899739583334,
      "learning_rate": 0.0001,
      "loss": 3.7862,
      "loss/crossentropy": 1.88859623670578,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1842176765203476,
      "step": 25212
    },
    {
      "epoch": 0.50428,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.008786773681640625,
      "learning_rate": 0.0001,
      "loss": 3.9548,
      "loss/crossentropy": 1.917616069316864,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19126974791288376,
      "step": 25214
    },
    {
      "epoch": 0.50432,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0074615478515625,
      "learning_rate": 0.0001,
      "loss": 3.928,
      "loss/crossentropy": 2.111130714416504,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20154303312301636,
      "step": 25216
    },
    {
      "epoch": 0.50436,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.004857381184895833,
      "learning_rate": 0.0001,
      "loss": 3.9316,
      "loss/crossentropy": 2.0559436678886414,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1823289915919304,
      "step": 25218
    },
    {
      "epoch": 0.5044,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.007225545247395834,
      "learning_rate": 0.0001,
      "loss": 4.2238,
      "loss/crossentropy": 1.8830538392066956,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17736117541790009,
      "step": 25220
    },
    {
      "epoch": 0.50444,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.009488932291666667,
      "learning_rate": 0.0001,
      "loss": 4.1422,
      "loss/crossentropy": 1.9821715354919434,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20550980418920517,
      "step": 25222
    },
    {
      "epoch": 0.50448,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008670806884765625,
      "learning_rate": 0.0001,
      "loss": 4.0223,
      "loss/crossentropy": 2.182245671749115,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2306300848722458,
      "step": 25224
    },
    {
      "epoch": 0.50452,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.0091217041015625,
      "learning_rate": 0.0001,
      "loss": 3.9751,
      "loss/crossentropy": 1.6160125136375427,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15688229352235794,
      "step": 25226
    },
    {
      "epoch": 0.50456,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.008988189697265624,
      "learning_rate": 0.0001,
      "loss": 4.037,
      "loss/crossentropy": 1.8484386801719666,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19160201400518417,
      "step": 25228
    },
    {
      "epoch": 0.5046,
      "grad_norm": 2.0,
      "grad_norm_var": 0.008438873291015624,
      "learning_rate": 0.0001,
      "loss": 4.2143,
      "loss/crossentropy": 2.153999447822571,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2077987790107727,
      "step": 25230
    },
    {
      "epoch": 0.50464,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.008487701416015625,
      "learning_rate": 0.0001,
      "loss": 4.0174,
      "loss/crossentropy": 2.1414119601249695,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2047003135085106,
      "step": 25232
    },
    {
      "epoch": 0.50468,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.00701904296875,
      "learning_rate": 0.0001,
      "loss": 4.1841,
      "loss/crossentropy": 2.128756880760193,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19793596118688583,
      "step": 25234
    },
    {
      "epoch": 0.50472,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.008160146077473958,
      "learning_rate": 0.0001,
      "loss": 4.0356,
      "loss/crossentropy": 1.9649195075035095,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19532814621925354,
      "step": 25236
    },
    {
      "epoch": 0.50476,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.006158192952473958,
      "learning_rate": 0.0001,
      "loss": 4.3162,
      "loss/crossentropy": 2.165442705154419,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19320187717676163,
      "step": 25238
    },
    {
      "epoch": 0.5048,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.005909983317057292,
      "learning_rate": 0.0001,
      "loss": 4.0444,
      "loss/crossentropy": 2.172460913658142,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18116996437311172,
      "step": 25240
    },
    {
      "epoch": 0.50484,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.0055501302083333336,
      "learning_rate": 0.0001,
      "loss": 3.9613,
      "loss/crossentropy": 1.9772083163261414,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18840450048446655,
      "step": 25242
    },
    {
      "epoch": 0.50488,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.005952962239583333,
      "learning_rate": 0.0001,
      "loss": 3.8147,
      "loss/crossentropy": 1.9985793828964233,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19273578375577927,
      "step": 25244
    },
    {
      "epoch": 0.50492,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.00513916015625,
      "learning_rate": 0.0001,
      "loss": 4.0365,
      "loss/crossentropy": 2.339000701904297,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2071395143866539,
      "step": 25246
    },
    {
      "epoch": 0.50496,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.006056467692057292,
      "learning_rate": 0.0001,
      "loss": 4.2748,
      "loss/crossentropy": 2.1699774861335754,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2179349735379219,
      "step": 25248
    },
    {
      "epoch": 0.505,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.0066569010416666664,
      "learning_rate": 0.0001,
      "loss": 3.9062,
      "loss/crossentropy": 1.9445035457611084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17794214189052582,
      "step": 25250
    },
    {
      "epoch": 0.50504,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0077626546223958336,
      "learning_rate": 0.0001,
      "loss": 4.3328,
      "loss/crossentropy": 2.325773239135742,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20460912585258484,
      "step": 25252
    },
    {
      "epoch": 0.50508,
      "grad_norm": 21.75,
      "grad_norm_var": 24.616644032796223,
      "learning_rate": 0.0001,
      "loss": 4.5896,
      "loss/crossentropy": 2.308157444000244,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2063561975955963,
      "step": 25254
    },
    {
      "epoch": 0.50512,
      "grad_norm": 2.1875,
      "grad_norm_var": 24.524991607666017,
      "learning_rate": 0.0001,
      "loss": 4.1727,
      "loss/crossentropy": 2.1321988105773926,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19514495134353638,
      "step": 25256
    },
    {
      "epoch": 0.50516,
      "grad_norm": 2.03125,
      "grad_norm_var": 24.486148834228516,
      "learning_rate": 0.0001,
      "loss": 3.9135,
      "loss/crossentropy": 1.7947803735733032,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18287408351898193,
      "step": 25258
    },
    {
      "epoch": 0.5052,
      "grad_norm": 1.8046875,
      "grad_norm_var": 24.50101318359375,
      "learning_rate": 0.0001,
      "loss": 4.0195,
      "loss/crossentropy": 2.0636664628982544,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20408915728330612,
      "step": 25260
    },
    {
      "epoch": 0.50524,
      "grad_norm": 1.9296875,
      "grad_norm_var": 24.510210927327474,
      "learning_rate": 0.0001,
      "loss": 3.904,
      "loss/crossentropy": 2.113875925540924,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19412121921777725,
      "step": 25262
    },
    {
      "epoch": 0.50528,
      "grad_norm": 1.8671875,
      "grad_norm_var": 24.563250478108724,
      "learning_rate": 0.0001,
      "loss": 3.9354,
      "loss/crossentropy": 1.9069438576698303,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18107398599386215,
      "step": 25264
    },
    {
      "epoch": 0.50532,
      "grad_norm": 1.90625,
      "grad_norm_var": 24.5382687886556,
      "learning_rate": 0.0001,
      "loss": 3.7792,
      "loss/crossentropy": 1.9241713881492615,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18617494404315948,
      "step": 25266
    },
    {
      "epoch": 0.50536,
      "grad_norm": 1.984375,
      "grad_norm_var": 24.575275675455728,
      "learning_rate": 0.0001,
      "loss": 3.8934,
      "loss/crossentropy": 1.9778280854225159,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19652489572763443,
      "step": 25268
    },
    {
      "epoch": 0.5054,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.014729817708333334,
      "learning_rate": 0.0001,
      "loss": 3.7332,
      "loss/crossentropy": 2.0369997024536133,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18344805389642715,
      "step": 25270
    },
    {
      "epoch": 0.50544,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.007176717122395833,
      "learning_rate": 0.0001,
      "loss": 4.0336,
      "loss/crossentropy": 1.9891296029090881,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19600480049848557,
      "step": 25272
    },
    {
      "epoch": 0.50548,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.005619049072265625,
      "learning_rate": 0.0001,
      "loss": 4.024,
      "loss/crossentropy": 2.325801730155945,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21493100374937057,
      "step": 25274
    },
    {
      "epoch": 0.50552,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.005736287434895833,
      "learning_rate": 0.0001,
      "loss": 4.067,
      "loss/crossentropy": 1.9277620315551758,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2140570506453514,
      "step": 25276
    },
    {
      "epoch": 0.50556,
      "grad_norm": 1.875,
      "grad_norm_var": 0.005763498942057291,
      "learning_rate": 0.0001,
      "loss": 3.9703,
      "loss/crossentropy": 2.3552772998809814,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21981871128082275,
      "step": 25278
    },
    {
      "epoch": 0.5056,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.005338287353515625,
      "learning_rate": 0.0001,
      "loss": 3.7174,
      "loss/crossentropy": 1.7559251189231873,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1645895466208458,
      "step": 25280
    },
    {
      "epoch": 0.50564,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.005149078369140625,
      "learning_rate": 0.0001,
      "loss": 4.0222,
      "loss/crossentropy": 1.8345605731010437,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18997152149677277,
      "step": 25282
    },
    {
      "epoch": 0.50568,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.005150349934895834,
      "learning_rate": 0.0001,
      "loss": 4.0908,
      "loss/crossentropy": 2.297994017601013,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2172359898686409,
      "step": 25284
    },
    {
      "epoch": 0.50572,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.004239908854166667,
      "learning_rate": 0.0001,
      "loss": 3.9462,
      "loss/crossentropy": 1.9640920758247375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1819777637720108,
      "step": 25286
    },
    {
      "epoch": 0.50576,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.004957834879557292,
      "learning_rate": 0.0001,
      "loss": 3.9544,
      "loss/crossentropy": 2.0045265555381775,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20263119041919708,
      "step": 25288
    },
    {
      "epoch": 0.5058,
      "grad_norm": 1.75,
      "grad_norm_var": 0.006058756510416667,
      "learning_rate": 0.0001,
      "loss": 3.8583,
      "loss/crossentropy": 2.1333796977996826,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20355874300003052,
      "step": 25290
    },
    {
      "epoch": 0.50584,
      "grad_norm": 2.0,
      "grad_norm_var": 0.006827799479166666,
      "learning_rate": 0.0001,
      "loss": 4.3003,
      "loss/crossentropy": 2.1814208030700684,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19890426099300385,
      "step": 25292
    },
    {
      "epoch": 0.50588,
      "grad_norm": 2.0,
      "grad_norm_var": 0.007283528645833333,
      "learning_rate": 0.0001,
      "loss": 4.4133,
      "loss/crossentropy": 2.1642476320266724,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1955343708395958,
      "step": 25294
    },
    {
      "epoch": 0.50592,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.006498209635416667,
      "learning_rate": 0.0001,
      "loss": 4.1446,
      "loss/crossentropy": 2.126586079597473,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19216856360435486,
      "step": 25296
    },
    {
      "epoch": 0.50596,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.007094065348307292,
      "learning_rate": 0.0001,
      "loss": 4.2278,
      "loss/crossentropy": 2.4168988466262817,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21824205666780472,
      "step": 25298
    },
    {
      "epoch": 0.506,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.005887858072916667,
      "learning_rate": 0.0001,
      "loss": 4.103,
      "loss/crossentropy": 2.013881802558899,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2011483684182167,
      "step": 25300
    },
    {
      "epoch": 0.50604,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.007964833577473959,
      "learning_rate": 0.0001,
      "loss": 4.0249,
      "loss/crossentropy": 2.18127703666687,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19816970825195312,
      "step": 25302
    },
    {
      "epoch": 0.50608,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.006249745686848958,
      "learning_rate": 0.0001,
      "loss": 4.0909,
      "loss/crossentropy": 1.8370496034622192,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18162214756011963,
      "step": 25304
    },
    {
      "epoch": 0.50612,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0038998921712239585,
      "learning_rate": 0.0001,
      "loss": 4.0421,
      "loss/crossentropy": 2.12531840801239,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21469102054834366,
      "step": 25306
    },
    {
      "epoch": 0.50616,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.004961903889973958,
      "learning_rate": 0.0001,
      "loss": 3.95,
      "loss/crossentropy": 1.956727385520935,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18601331859827042,
      "step": 25308
    },
    {
      "epoch": 0.5062,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.016161092122395835,
      "learning_rate": 0.0001,
      "loss": 3.9688,
      "loss/crossentropy": 2.232389807701111,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19958854466676712,
      "step": 25310
    },
    {
      "epoch": 0.50624,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.016280110677083334,
      "learning_rate": 0.0001,
      "loss": 4.0164,
      "loss/crossentropy": 2.0312620997428894,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19724415987730026,
      "step": 25312
    },
    {
      "epoch": 0.50628,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.01625544230143229,
      "learning_rate": 0.0001,
      "loss": 4.232,
      "loss/crossentropy": 2.344046115875244,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20940567553043365,
      "step": 25314
    },
    {
      "epoch": 0.50632,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.017014312744140624,
      "learning_rate": 0.0001,
      "loss": 4.046,
      "loss/crossentropy": 1.99763023853302,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1835399493575096,
      "step": 25316
    },
    {
      "epoch": 0.50636,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.017044830322265624,
      "learning_rate": 0.0001,
      "loss": 3.6584,
      "loss/crossentropy": 1.9649406671524048,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19432833045721054,
      "step": 25318
    },
    {
      "epoch": 0.5064,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.016434478759765624,
      "learning_rate": 0.0001,
      "loss": 3.9027,
      "loss/crossentropy": 1.8670902848243713,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1731145828962326,
      "step": 25320
    },
    {
      "epoch": 0.50644,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.016706339518229165,
      "learning_rate": 0.0001,
      "loss": 4.0066,
      "loss/crossentropy": 2.0195581912994385,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19497420638799667,
      "step": 25322
    },
    {
      "epoch": 0.50648,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.016290028889973957,
      "learning_rate": 0.0001,
      "loss": 4.1156,
      "loss/crossentropy": 2.1221953630447388,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20753790438175201,
      "step": 25324
    },
    {
      "epoch": 0.50652,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.007169596354166667,
      "learning_rate": 0.0001,
      "loss": 4.2117,
      "loss/crossentropy": 2.1407066583633423,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1934654265642166,
      "step": 25326
    },
    {
      "epoch": 0.50656,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.007917277018229167,
      "learning_rate": 0.0001,
      "loss": 3.9074,
      "loss/crossentropy": 1.956154465675354,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1854771375656128,
      "step": 25328
    },
    {
      "epoch": 0.5066,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.007450103759765625,
      "learning_rate": 0.0001,
      "loss": 4.0652,
      "loss/crossentropy": 2.0948599576950073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19326548278331757,
      "step": 25330
    },
    {
      "epoch": 0.50664,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.006705729166666666,
      "learning_rate": 0.0001,
      "loss": 4.0609,
      "loss/crossentropy": 2.137898027896881,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20896823704242706,
      "step": 25332
    },
    {
      "epoch": 0.50668,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.005859375,
      "learning_rate": 0.0001,
      "loss": 3.8687,
      "loss/crossentropy": 1.6471683382987976,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1814308613538742,
      "step": 25334
    },
    {
      "epoch": 0.50672,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.007165273030598958,
      "learning_rate": 0.0001,
      "loss": 3.7461,
      "loss/crossentropy": 1.861536979675293,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17766699194908142,
      "step": 25336
    },
    {
      "epoch": 0.50676,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.006974029541015625,
      "learning_rate": 0.0001,
      "loss": 3.9319,
      "loss/crossentropy": 2.312160909175873,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20586217939853668,
      "step": 25338
    },
    {
      "epoch": 0.5068,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.005293528238932292,
      "learning_rate": 0.0001,
      "loss": 4.1102,
      "loss/crossentropy": 1.8415343165397644,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20047077536582947,
      "step": 25340
    },
    {
      "epoch": 0.50684,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0033925374348958335,
      "learning_rate": 0.0001,
      "loss": 4.0579,
      "loss/crossentropy": 1.94148850440979,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19869551062583923,
      "step": 25342
    },
    {
      "epoch": 0.50688,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.002559407552083333,
      "learning_rate": 0.0001,
      "loss": 4.0379,
      "loss/crossentropy": 2.280241370201111,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19969766587018967,
      "step": 25344
    },
    {
      "epoch": 0.50692,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.0026140848795572916,
      "learning_rate": 0.0001,
      "loss": 4.1569,
      "loss/crossentropy": 2.0140087604522705,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20252882689237595,
      "step": 25346
    },
    {
      "epoch": 0.50696,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.0073201497395833336,
      "learning_rate": 0.0001,
      "loss": 3.9593,
      "loss/crossentropy": 2.186760902404785,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20365024358034134,
      "step": 25348
    },
    {
      "epoch": 0.507,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.007161458333333333,
      "learning_rate": 0.0001,
      "loss": 4.2392,
      "loss/crossentropy": 2.324553370475769,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20509789884090424,
      "step": 25350
    },
    {
      "epoch": 0.50704,
      "grad_norm": 2.0,
      "grad_norm_var": 0.006638336181640625,
      "learning_rate": 0.0001,
      "loss": 4.255,
      "loss/crossentropy": 2.2305015325546265,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21707269549369812,
      "step": 25352
    },
    {
      "epoch": 0.50708,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.006864166259765625,
      "learning_rate": 0.0001,
      "loss": 4.0548,
      "loss/crossentropy": 2.1571671962738037,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20072072744369507,
      "step": 25354
    },
    {
      "epoch": 0.50712,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0072825113932291664,
      "learning_rate": 0.0001,
      "loss": 4.0577,
      "loss/crossentropy": 2.014027178287506,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1823829561471939,
      "step": 25356
    },
    {
      "epoch": 0.50716,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.008304595947265625,
      "learning_rate": 0.0001,
      "loss": 3.7066,
      "loss/crossentropy": 2.306033492088318,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2326786294579506,
      "step": 25358
    },
    {
      "epoch": 0.5072,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.008335113525390625,
      "learning_rate": 0.0001,
      "loss": 3.9867,
      "loss/crossentropy": 2.1029749512672424,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19578313827514648,
      "step": 25360
    },
    {
      "epoch": 0.50724,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.009242502848307292,
      "learning_rate": 0.0001,
      "loss": 3.9696,
      "loss/crossentropy": 2.244466543197632,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20059070736169815,
      "step": 25362
    },
    {
      "epoch": 0.50728,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.006858062744140625,
      "learning_rate": 0.0001,
      "loss": 4.1295,
      "loss/crossentropy": 2.3815391063690186,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21556073427200317,
      "step": 25364
    },
    {
      "epoch": 0.50732,
      "grad_norm": 2.0,
      "grad_norm_var": 0.013767242431640625,
      "learning_rate": 0.0001,
      "loss": 4.0822,
      "loss/crossentropy": 1.997973620891571,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20311636477708817,
      "step": 25366
    },
    {
      "epoch": 0.50736,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.016845703125,
      "learning_rate": 0.0001,
      "loss": 3.8043,
      "loss/crossentropy": 2.2644081115722656,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2022029459476471,
      "step": 25368
    },
    {
      "epoch": 0.5074,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.01632868448893229,
      "learning_rate": 0.0001,
      "loss": 3.9491,
      "loss/crossentropy": 1.7542709112167358,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15834683179855347,
      "step": 25370
    },
    {
      "epoch": 0.50744,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.016502888997395833,
      "learning_rate": 0.0001,
      "loss": 4.0981,
      "loss/crossentropy": 2.2310311794281006,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21090705692768097,
      "step": 25372
    },
    {
      "epoch": 0.50748,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.014977773030598959,
      "learning_rate": 0.0001,
      "loss": 3.9221,
      "loss/crossentropy": 1.9018943905830383,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1821354478597641,
      "step": 25374
    },
    {
      "epoch": 0.50752,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.01602961222330729,
      "learning_rate": 0.0001,
      "loss": 4.0671,
      "loss/crossentropy": 2.0701186656951904,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19244033098220825,
      "step": 25376
    },
    {
      "epoch": 0.50756,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.01553955078125,
      "learning_rate": 0.0001,
      "loss": 3.7639,
      "loss/crossentropy": 2.0421605110168457,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1847202628850937,
      "step": 25378
    },
    {
      "epoch": 0.5076,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.014314524332682292,
      "learning_rate": 0.0001,
      "loss": 3.9923,
      "loss/crossentropy": 1.9375264048576355,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17147281020879745,
      "step": 25380
    },
    {
      "epoch": 0.50764,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.0062408447265625,
      "learning_rate": 0.0001,
      "loss": 4.0086,
      "loss/crossentropy": 1.9593093395233154,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19151991605758667,
      "step": 25382
    },
    {
      "epoch": 0.50768,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.007500966389973958,
      "learning_rate": 0.0001,
      "loss": 3.9897,
      "loss/crossentropy": 1.9283559322357178,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18128316849470139,
      "step": 25384
    },
    {
      "epoch": 0.50772,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010970052083333333,
      "learning_rate": 0.0001,
      "loss": 4.385,
      "loss/crossentropy": 1.912258267402649,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18579021841287613,
      "step": 25386
    },
    {
      "epoch": 0.50776,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.010619099934895833,
      "learning_rate": 0.0001,
      "loss": 4.0037,
      "loss/crossentropy": 2.161900043487549,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2025725245475769,
      "step": 25388
    },
    {
      "epoch": 0.5078,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.010733795166015626,
      "learning_rate": 0.0001,
      "loss": 4.2077,
      "loss/crossentropy": 2.1753687858581543,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21103990823030472,
      "step": 25390
    },
    {
      "epoch": 0.50784,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.010581207275390626,
      "learning_rate": 0.0001,
      "loss": 4.2122,
      "loss/crossentropy": 1.869983971118927,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18909727036952972,
      "step": 25392
    },
    {
      "epoch": 0.50788,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0122711181640625,
      "learning_rate": 0.0001,
      "loss": 4.0602,
      "loss/crossentropy": 2.0557621121406555,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18932487815618515,
      "step": 25394
    },
    {
      "epoch": 0.50792,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011946360270182291,
      "learning_rate": 0.0001,
      "loss": 4.0104,
      "loss/crossentropy": 1.865005910396576,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2251267060637474,
      "step": 25396
    },
    {
      "epoch": 0.50796,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.011503092447916667,
      "learning_rate": 0.0001,
      "loss": 3.836,
      "loss/crossentropy": 1.7247771620750427,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16510912030935287,
      "step": 25398
    },
    {
      "epoch": 0.508,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.009266916910807292,
      "learning_rate": 0.0001,
      "loss": 4.0276,
      "loss/crossentropy": 2.110623359680176,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18747511506080627,
      "step": 25400
    },
    {
      "epoch": 0.50804,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.008058420817057292,
      "learning_rate": 0.0001,
      "loss": 4.1451,
      "loss/crossentropy": 2.0501877069473267,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19226385653018951,
      "step": 25402
    },
    {
      "epoch": 0.50808,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.008119455973307292,
      "learning_rate": 0.0001,
      "loss": 3.8899,
      "loss/crossentropy": 2.479986786842346,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23350444436073303,
      "step": 25404
    },
    {
      "epoch": 0.50812,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.008133951822916667,
      "learning_rate": 0.0001,
      "loss": 4.0184,
      "loss/crossentropy": 2.215437889099121,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19325988739728928,
      "step": 25406
    },
    {
      "epoch": 0.50816,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.008485666910807292,
      "learning_rate": 0.0001,
      "loss": 3.8584,
      "loss/crossentropy": 2.1168838143348694,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18737324327230453,
      "step": 25408
    },
    {
      "epoch": 0.5082,
      "grad_norm": 2.0,
      "grad_norm_var": 0.007173411051432292,
      "learning_rate": 0.0001,
      "loss": 3.9339,
      "loss/crossentropy": 1.9043878316879272,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19250840693712234,
      "step": 25410
    },
    {
      "epoch": 0.50824,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.010158030192057292,
      "learning_rate": 0.0001,
      "loss": 4.2541,
      "loss/crossentropy": 2.411430239677429,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22225366532802582,
      "step": 25412
    },
    {
      "epoch": 0.50828,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.009091949462890625,
      "learning_rate": 0.0001,
      "loss": 3.9909,
      "loss/crossentropy": 1.843587577342987,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18295426666736603,
      "step": 25414
    },
    {
      "epoch": 0.50832,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.008733876546223958,
      "learning_rate": 0.0001,
      "loss": 3.94,
      "loss/crossentropy": 1.8655884861946106,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17755136638879776,
      "step": 25416
    },
    {
      "epoch": 0.50836,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.007176717122395833,
      "learning_rate": 0.0001,
      "loss": 3.9523,
      "loss/crossentropy": 1.8799603581428528,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18243838846683502,
      "step": 25418
    },
    {
      "epoch": 0.5084,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.007226308186848958,
      "learning_rate": 0.0001,
      "loss": 3.9171,
      "loss/crossentropy": 2.2428990602493286,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21759599447250366,
      "step": 25420
    },
    {
      "epoch": 0.50844,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.008265940348307292,
      "learning_rate": 0.0001,
      "loss": 3.9152,
      "loss/crossentropy": 1.9947530031204224,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1956867277622223,
      "step": 25422
    },
    {
      "epoch": 0.50848,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.006359608968098959,
      "learning_rate": 0.0001,
      "loss": 3.9553,
      "loss/crossentropy": 2.0679984092712402,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21864020824432373,
      "step": 25424
    },
    {
      "epoch": 0.50852,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.005712890625,
      "learning_rate": 0.0001,
      "loss": 4.0127,
      "loss/crossentropy": 1.9529247879981995,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1891709715127945,
      "step": 25426
    },
    {
      "epoch": 0.50856,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.006308746337890625,
      "learning_rate": 0.0001,
      "loss": 4.0049,
      "loss/crossentropy": 2.1328742504119873,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21699222922325134,
      "step": 25428
    },
    {
      "epoch": 0.5086,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.00760498046875,
      "learning_rate": 0.0001,
      "loss": 4.0057,
      "loss/crossentropy": 1.8275007009506226,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1621478796005249,
      "step": 25430
    },
    {
      "epoch": 0.50864,
      "grad_norm": 1.7265625,
      "grad_norm_var": 0.010374959309895833,
      "learning_rate": 0.0001,
      "loss": 3.7843,
      "loss/crossentropy": 2.134227454662323,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19917849451303482,
      "step": 25432
    },
    {
      "epoch": 0.50868,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010499827067057292,
      "learning_rate": 0.0001,
      "loss": 4.099,
      "loss/crossentropy": 2.040915012359619,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1758517250418663,
      "step": 25434
    },
    {
      "epoch": 0.50872,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.012247721354166666,
      "learning_rate": 0.0001,
      "loss": 4.0379,
      "loss/crossentropy": 1.9219905734062195,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26709431409835815,
      "step": 25436
    },
    {
      "epoch": 0.50876,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.011683909098307292,
      "learning_rate": 0.0001,
      "loss": 3.8866,
      "loss/crossentropy": 1.8520461320877075,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17154939472675323,
      "step": 25438
    },
    {
      "epoch": 0.5088,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.011891428629557292,
      "learning_rate": 0.0001,
      "loss": 3.9908,
      "loss/crossentropy": 2.091896414756775,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19294072687625885,
      "step": 25440
    },
    {
      "epoch": 0.50884,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.012029774983723958,
      "learning_rate": 0.0001,
      "loss": 3.8595,
      "loss/crossentropy": 2.209986925125122,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19535083323717117,
      "step": 25442
    },
    {
      "epoch": 0.50888,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.010643513997395833,
      "learning_rate": 0.0001,
      "loss": 3.5984,
      "loss/crossentropy": 1.7103394865989685,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17907221615314484,
      "step": 25444
    },
    {
      "epoch": 0.50892,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.0099365234375,
      "learning_rate": 0.0001,
      "loss": 4.0083,
      "loss/crossentropy": 2.185406744480133,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19615823030471802,
      "step": 25446
    },
    {
      "epoch": 0.50896,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.009317779541015625,
      "learning_rate": 0.0001,
      "loss": 4.0897,
      "loss/crossentropy": 1.7030528783798218,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19345925003290176,
      "step": 25448
    },
    {
      "epoch": 0.509,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.0098541259765625,
      "learning_rate": 0.0001,
      "loss": 3.8739,
      "loss/crossentropy": 1.9207965731620789,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18073637783527374,
      "step": 25450
    },
    {
      "epoch": 0.50904,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.008307902018229167,
      "learning_rate": 0.0001,
      "loss": 3.8217,
      "loss/crossentropy": 1.8818482160568237,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18431610614061356,
      "step": 25452
    },
    {
      "epoch": 0.50908,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.009965006510416667,
      "learning_rate": 0.0001,
      "loss": 3.8117,
      "loss/crossentropy": 2.2623773217201233,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20106880366802216,
      "step": 25454
    },
    {
      "epoch": 0.50912,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.010782623291015625,
      "learning_rate": 0.0001,
      "loss": 4.1155,
      "loss/crossentropy": 2.303765892982483,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19978389143943787,
      "step": 25456
    },
    {
      "epoch": 0.50916,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.012824503580729167,
      "learning_rate": 0.0001,
      "loss": 4.0437,
      "loss/crossentropy": 2.1355656385421753,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20279624313116074,
      "step": 25458
    },
    {
      "epoch": 0.5092,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.012621815999348958,
      "learning_rate": 0.0001,
      "loss": 3.9021,
      "loss/crossentropy": 1.7096583843231201,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16103574633598328,
      "step": 25460
    },
    {
      "epoch": 0.50924,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.012889607747395834,
      "learning_rate": 0.0001,
      "loss": 3.9125,
      "loss/crossentropy": 1.7230717539787292,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16039150953292847,
      "step": 25462
    },
    {
      "epoch": 0.50928,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.024662017822265625,
      "learning_rate": 0.0001,
      "loss": 4.0199,
      "loss/crossentropy": 1.983295977115631,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20290112495422363,
      "step": 25464
    },
    {
      "epoch": 0.50932,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.024812825520833335,
      "learning_rate": 0.0001,
      "loss": 3.8669,
      "loss/crossentropy": 1.7826906442642212,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1826622262597084,
      "step": 25466
    },
    {
      "epoch": 0.50936,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.024472808837890624,
      "learning_rate": 0.0001,
      "loss": 3.6992,
      "loss/crossentropy": 1.7718060612678528,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1686256304383278,
      "step": 25468
    },
    {
      "epoch": 0.5094,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.021605428059895834,
      "learning_rate": 0.0001,
      "loss": 3.7736,
      "loss/crossentropy": 1.9546560645103455,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1890355497598648,
      "step": 25470
    },
    {
      "epoch": 0.50944,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.024265289306640625,
      "learning_rate": 0.0001,
      "loss": 3.7519,
      "loss/crossentropy": 2.2464581727981567,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21081877499818802,
      "step": 25472
    },
    {
      "epoch": 0.50948,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.022810872395833334,
      "learning_rate": 0.0001,
      "loss": 4.0546,
      "loss/crossentropy": 2.150168299674988,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20787308365106583,
      "step": 25474
    },
    {
      "epoch": 0.50952,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.020169830322265624,
      "learning_rate": 0.0001,
      "loss": 3.8762,
      "loss/crossentropy": 2.08556866645813,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18290850520133972,
      "step": 25476
    },
    {
      "epoch": 0.50956,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0203125,
      "learning_rate": 0.0001,
      "loss": 4.1563,
      "loss/crossentropy": 2.174665331840515,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2028878778219223,
      "step": 25478
    },
    {
      "epoch": 0.5096,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.0035764058430989582,
      "learning_rate": 0.0001,
      "loss": 3.9699,
      "loss/crossentropy": 1.9720661640167236,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19616466760635376,
      "step": 25480
    },
    {
      "epoch": 0.50964,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0045562744140625,
      "learning_rate": 0.0001,
      "loss": 3.9704,
      "loss/crossentropy": 2.2741299867630005,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2136249840259552,
      "step": 25482
    },
    {
      "epoch": 0.50968,
      "grad_norm": 1.875,
      "grad_norm_var": 0.004620107014973959,
      "learning_rate": 0.0001,
      "loss": 3.9,
      "loss/crossentropy": 1.8212140798568726,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16650617122650146,
      "step": 25484
    },
    {
      "epoch": 0.50972,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0072174072265625,
      "learning_rate": 0.0001,
      "loss": 3.9479,
      "loss/crossentropy": 2.25445818901062,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21185942739248276,
      "step": 25486
    },
    {
      "epoch": 0.50976,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0061838785807291664,
      "learning_rate": 0.0001,
      "loss": 4.273,
      "loss/crossentropy": 2.198422074317932,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19693002849817276,
      "step": 25488
    },
    {
      "epoch": 0.5098,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.007177480061848958,
      "learning_rate": 0.0001,
      "loss": 4.0788,
      "loss/crossentropy": 2.0654536485671997,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20444564521312714,
      "step": 25490
    },
    {
      "epoch": 0.50984,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.007889811197916667,
      "learning_rate": 0.0001,
      "loss": 4.1771,
      "loss/crossentropy": 1.9605732560157776,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1785297840833664,
      "step": 25492
    },
    {
      "epoch": 0.50988,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.009968821207682292,
      "learning_rate": 0.0001,
      "loss": 4.2828,
      "loss/crossentropy": 2.128088355064392,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2063232958316803,
      "step": 25494
    },
    {
      "epoch": 0.50992,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.011039225260416667,
      "learning_rate": 0.0001,
      "loss": 3.7328,
      "loss/crossentropy": 1.9981979727745056,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17392293363809586,
      "step": 25496
    },
    {
      "epoch": 0.50996,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.014243316650390626,
      "learning_rate": 0.0001,
      "loss": 4.3475,
      "loss/crossentropy": 2.4065998792648315,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20214569568634033,
      "step": 25498
    },
    {
      "epoch": 0.51,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.01361083984375,
      "learning_rate": 0.0001,
      "loss": 3.9476,
      "loss/crossentropy": 2.154718041419983,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19910430163145065,
      "step": 25500
    },
    {
      "epoch": 0.51004,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.011042277018229166,
      "learning_rate": 0.0001,
      "loss": 4.0985,
      "loss/crossentropy": 2.22087424993515,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20276758074760437,
      "step": 25502
    },
    {
      "epoch": 0.51008,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.014336903889973959,
      "learning_rate": 0.0001,
      "loss": 3.6317,
      "loss/crossentropy": 1.9433262944221497,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1835595816373825,
      "step": 25504
    },
    {
      "epoch": 0.51012,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.014625803629557291,
      "learning_rate": 0.0001,
      "loss": 3.8672,
      "loss/crossentropy": 1.814663290977478,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1932072937488556,
      "step": 25506
    },
    {
      "epoch": 0.51016,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0143463134765625,
      "learning_rate": 0.0001,
      "loss": 3.9603,
      "loss/crossentropy": 1.8567258715629578,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20248033851385117,
      "step": 25508
    },
    {
      "epoch": 0.5102,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.013232167561848958,
      "learning_rate": 0.0001,
      "loss": 4.1419,
      "loss/crossentropy": 2.056326985359192,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19645985960960388,
      "step": 25510
    },
    {
      "epoch": 0.51024,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.010884348551432292,
      "learning_rate": 0.0001,
      "loss": 4.2054,
      "loss/crossentropy": 2.068827450275421,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19381123036146164,
      "step": 25512
    },
    {
      "epoch": 0.51028,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.008898671468098958,
      "learning_rate": 0.0001,
      "loss": 3.9707,
      "loss/crossentropy": 2.4085018634796143,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20001942664384842,
      "step": 25514
    },
    {
      "epoch": 0.51032,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.008056386311848959,
      "learning_rate": 0.0001,
      "loss": 4.0348,
      "loss/crossentropy": 1.9481959342956543,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17734962701797485,
      "step": 25516
    },
    {
      "epoch": 0.51036,
      "grad_norm": 1.7265625,
      "grad_norm_var": 0.011722819010416666,
      "learning_rate": 0.0001,
      "loss": 3.6203,
      "loss/crossentropy": 1.9250916838645935,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1881168857216835,
      "step": 25518
    },
    {
      "epoch": 0.5104,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0101715087890625,
      "learning_rate": 0.0001,
      "loss": 4.2853,
      "loss/crossentropy": 2.4334945678710938,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22089337557554245,
      "step": 25520
    },
    {
      "epoch": 0.51044,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.012629954020182292,
      "learning_rate": 0.0001,
      "loss": 3.9618,
      "loss/crossentropy": 1.8872219324111938,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17582690715789795,
      "step": 25522
    },
    {
      "epoch": 0.51048,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.011156209309895833,
      "learning_rate": 0.0001,
      "loss": 4.0339,
      "loss/crossentropy": 2.0595308542251587,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20405854284763336,
      "step": 25524
    },
    {
      "epoch": 0.51052,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.012819163004557292,
      "learning_rate": 0.0001,
      "loss": 4.0668,
      "loss/crossentropy": 1.7874937653541565,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21105076372623444,
      "step": 25526
    },
    {
      "epoch": 0.51056,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.014076487223307291,
      "learning_rate": 0.0001,
      "loss": 3.8664,
      "loss/crossentropy": 1.866211712360382,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18459182977676392,
      "step": 25528
    },
    {
      "epoch": 0.5106,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.015641021728515624,
      "learning_rate": 0.0001,
      "loss": 4.0479,
      "loss/crossentropy": 1.8424429893493652,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17495467513799667,
      "step": 25530
    },
    {
      "epoch": 0.51064,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.015632120768229167,
      "learning_rate": 0.0001,
      "loss": 4.1848,
      "loss/crossentropy": 2.3827977180480957,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2094314694404602,
      "step": 25532
    },
    {
      "epoch": 0.51068,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.00950927734375,
      "learning_rate": 0.0001,
      "loss": 3.9321,
      "loss/crossentropy": 2.0699292421340942,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20574011653661728,
      "step": 25534
    },
    {
      "epoch": 0.51072,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0102447509765625,
      "learning_rate": 0.0001,
      "loss": 3.7784,
      "loss/crossentropy": 2.058907687664032,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1945437341928482,
      "step": 25536
    },
    {
      "epoch": 0.51076,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0090484619140625,
      "learning_rate": 0.0001,
      "loss": 4.0229,
      "loss/crossentropy": 2.138782501220703,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19527214765548706,
      "step": 25538
    },
    {
      "epoch": 0.5108,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.00936279296875,
      "learning_rate": 0.0001,
      "loss": 3.9339,
      "loss/crossentropy": 1.9123243689537048,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18704041838645935,
      "step": 25540
    },
    {
      "epoch": 0.51084,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.008056640625,
      "learning_rate": 0.0001,
      "loss": 4.2846,
      "loss/crossentropy": 2.5264101028442383,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23050238192081451,
      "step": 25542
    },
    {
      "epoch": 0.51088,
      "grad_norm": 1.875,
      "grad_norm_var": 0.005826822916666667,
      "learning_rate": 0.0001,
      "loss": 4.0812,
      "loss/crossentropy": 1.9210030436515808,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19104420393705368,
      "step": 25544
    },
    {
      "epoch": 0.51092,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.006809234619140625,
      "learning_rate": 0.0001,
      "loss": 3.6386,
      "loss/crossentropy": 1.7989648580551147,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17732901126146317,
      "step": 25546
    },
    {
      "epoch": 0.51096,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.006086985270182292,
      "learning_rate": 0.0001,
      "loss": 4.0115,
      "loss/crossentropy": 1.7673185467720032,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19276081770658493,
      "step": 25548
    },
    {
      "epoch": 0.511,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.005586496988932292,
      "learning_rate": 0.0001,
      "loss": 3.9068,
      "loss/crossentropy": 2.451088309288025,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21039804071187973,
      "step": 25550
    },
    {
      "epoch": 0.51104,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.004813385009765625,
      "learning_rate": 0.0001,
      "loss": 3.9767,
      "loss/crossentropy": 2.0937219858169556,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21383897960186005,
      "step": 25552
    },
    {
      "epoch": 0.51108,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.007010650634765625,
      "learning_rate": 0.0001,
      "loss": 3.8091,
      "loss/crossentropy": 1.7688068747520447,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17263571172952652,
      "step": 25554
    },
    {
      "epoch": 0.51112,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.008137003580729166,
      "learning_rate": 0.0001,
      "loss": 4.1427,
      "loss/crossentropy": 2.165677785873413,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19934290647506714,
      "step": 25556
    },
    {
      "epoch": 0.51116,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.009197743733723958,
      "learning_rate": 0.0001,
      "loss": 3.8542,
      "loss/crossentropy": 1.7860172986984253,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18735045939683914,
      "step": 25558
    },
    {
      "epoch": 0.5112,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.012165323893229166,
      "learning_rate": 0.0001,
      "loss": 4.0364,
      "loss/crossentropy": 2.2124353647232056,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27785858511924744,
      "step": 25560
    },
    {
      "epoch": 0.51124,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.01620457967122396,
      "learning_rate": 0.0001,
      "loss": 4.0802,
      "loss/crossentropy": 1.8342975974082947,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1646983027458191,
      "step": 25562
    },
    {
      "epoch": 0.51128,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.01895319620768229,
      "learning_rate": 0.0001,
      "loss": 3.7718,
      "loss/crossentropy": 2.0271248817443848,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18964418768882751,
      "step": 25564
    },
    {
      "epoch": 0.51132,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.01892267862955729,
      "learning_rate": 0.0001,
      "loss": 3.8739,
      "loss/crossentropy": 2.1983633041381836,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2039782777428627,
      "step": 25566
    },
    {
      "epoch": 0.51136,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.01982421875,
      "learning_rate": 0.0001,
      "loss": 3.9194,
      "loss/crossentropy": 1.804263949394226,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17312633991241455,
      "step": 25568
    },
    {
      "epoch": 0.5114,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.019624837239583335,
      "learning_rate": 0.0001,
      "loss": 4.2465,
      "loss/crossentropy": 1.7486448287963867,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16695599257946014,
      "step": 25570
    },
    {
      "epoch": 0.51144,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.02077204386393229,
      "learning_rate": 0.0001,
      "loss": 3.7398,
      "loss/crossentropy": 1.9466286301612854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1805291324853897,
      "step": 25572
    },
    {
      "epoch": 0.51148,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.020703125,
      "learning_rate": 0.0001,
      "loss": 4.1362,
      "loss/crossentropy": 2.203175187110901,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21553437411785126,
      "step": 25574
    },
    {
      "epoch": 0.51152,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.01708958943684896,
      "learning_rate": 0.0001,
      "loss": 4.1116,
      "loss/crossentropy": 1.8743736743927002,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19245517253875732,
      "step": 25576
    },
    {
      "epoch": 0.51156,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.008211008707682292,
      "learning_rate": 0.0001,
      "loss": 3.8857,
      "loss/crossentropy": 1.866178572177887,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18233061581850052,
      "step": 25578
    },
    {
      "epoch": 0.5116,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.007420857747395833,
      "learning_rate": 0.0001,
      "loss": 3.8674,
      "loss/crossentropy": 2.0799529552459717,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19485916197299957,
      "step": 25580
    },
    {
      "epoch": 0.51164,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.007222493489583333,
      "learning_rate": 0.0001,
      "loss": 3.9318,
      "loss/crossentropy": 1.7319077253341675,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16157583892345428,
      "step": 25582
    },
    {
      "epoch": 0.51168,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.0073893229166666664,
      "learning_rate": 0.0001,
      "loss": 3.7334,
      "loss/crossentropy": 2.0670501589775085,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18853918462991714,
      "step": 25584
    },
    {
      "epoch": 0.51172,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.004671223958333333,
      "learning_rate": 0.0001,
      "loss": 4.0737,
      "loss/crossentropy": 2.0605591535568237,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1799873411655426,
      "step": 25586
    },
    {
      "epoch": 0.51176,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.004288482666015625,
      "learning_rate": 0.0001,
      "loss": 3.9416,
      "loss/crossentropy": 1.9844827055931091,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19314195960760117,
      "step": 25588
    },
    {
      "epoch": 0.5118,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.0035906473795572916,
      "learning_rate": 0.0001,
      "loss": 3.9044,
      "loss/crossentropy": 2.292548894882202,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19697485864162445,
      "step": 25590
    },
    {
      "epoch": 0.51184,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.004400380452473958,
      "learning_rate": 0.0001,
      "loss": 4.1964,
      "loss/crossentropy": 2.472001791000366,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20994122326374054,
      "step": 25592
    },
    {
      "epoch": 0.51188,
      "grad_norm": 2.25,
      "grad_norm_var": 0.011283365885416667,
      "learning_rate": 0.0001,
      "loss": 3.9873,
      "loss/crossentropy": 1.3704800009727478,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1426696479320526,
      "step": 25594
    },
    {
      "epoch": 0.51192,
      "grad_norm": 1.875,
      "grad_norm_var": 0.0113922119140625,
      "learning_rate": 0.0001,
      "loss": 3.9129,
      "loss/crossentropy": 1.6338598132133484,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17765308916568756,
      "step": 25596
    },
    {
      "epoch": 0.51196,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.011717732747395833,
      "learning_rate": 0.0001,
      "loss": 4.1234,
      "loss/crossentropy": 1.8394725322723389,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1996341496706009,
      "step": 25598
    },
    {
      "epoch": 0.512,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.012443033854166667,
      "learning_rate": 0.0001,
      "loss": 3.9978,
      "loss/crossentropy": 1.9061697125434875,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18420706689357758,
      "step": 25600
    },
    {
      "epoch": 0.51204,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.012426503499348958,
      "learning_rate": 0.0001,
      "loss": 4.0949,
      "loss/crossentropy": 2.3584840297698975,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21293433010578156,
      "step": 25602
    },
    {
      "epoch": 0.51208,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.012251536051432291,
      "learning_rate": 0.0001,
      "loss": 3.8624,
      "loss/crossentropy": 1.923392653465271,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18330930918455124,
      "step": 25604
    },
    {
      "epoch": 0.51212,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.011934153238932292,
      "learning_rate": 0.0001,
      "loss": 4.027,
      "loss/crossentropy": 2.2249737977981567,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2061079815030098,
      "step": 25606
    },
    {
      "epoch": 0.51216,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.013505045572916667,
      "learning_rate": 0.0001,
      "loss": 3.8274,
      "loss/crossentropy": 1.7640107870101929,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1722632348537445,
      "step": 25608
    },
    {
      "epoch": 0.5122,
      "grad_norm": 2.0,
      "grad_norm_var": 0.006278483072916666,
      "learning_rate": 0.0001,
      "loss": 4.0634,
      "loss/crossentropy": 2.0654106736183167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1959560066461563,
      "step": 25610
    },
    {
      "epoch": 0.51224,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.006685384114583333,
      "learning_rate": 0.0001,
      "loss": 3.857,
      "loss/crossentropy": 1.8277021646499634,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1689283475279808,
      "step": 25612
    },
    {
      "epoch": 0.51228,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0066802978515625,
      "learning_rate": 0.0001,
      "loss": 3.9565,
      "loss/crossentropy": 2.2842466831207275,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2116750180721283,
      "step": 25614
    },
    {
      "epoch": 0.51232,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.01710205078125,
      "learning_rate": 0.0001,
      "loss": 3.7537,
      "loss/crossentropy": 1.8136274218559265,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18518541753292084,
      "step": 25616
    },
    {
      "epoch": 0.51236,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.017277018229166666,
      "learning_rate": 0.0001,
      "loss": 3.9369,
      "loss/crossentropy": 2.0048798322677612,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1975068524479866,
      "step": 25618
    },
    {
      "epoch": 0.5124,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.016745758056640626,
      "learning_rate": 0.0001,
      "loss": 3.7449,
      "loss/crossentropy": 1.7246285676956177,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17372817546129227,
      "step": 25620
    },
    {
      "epoch": 0.51244,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0177978515625,
      "learning_rate": 0.0001,
      "loss": 4.2149,
      "loss/crossentropy": 2.160382390022278,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20257163047790527,
      "step": 25622
    },
    {
      "epoch": 0.51248,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.015604400634765625,
      "learning_rate": 0.0001,
      "loss": 4.0976,
      "loss/crossentropy": 2.1341055631637573,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18860094994306564,
      "step": 25624
    },
    {
      "epoch": 0.51252,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.01613337198893229,
      "learning_rate": 0.0001,
      "loss": 3.9408,
      "loss/crossentropy": 2.1212183237075806,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18820258975028992,
      "step": 25626
    },
    {
      "epoch": 0.51256,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.017045084635416666,
      "learning_rate": 0.0001,
      "loss": 3.88,
      "loss/crossentropy": 2.0193417072296143,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18116050213575363,
      "step": 25628
    },
    {
      "epoch": 0.5126,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.017472330729166666,
      "learning_rate": 0.0001,
      "loss": 4.0049,
      "loss/crossentropy": 2.2681472301483154,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21637655794620514,
      "step": 25630
    },
    {
      "epoch": 0.51264,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.006444295247395833,
      "learning_rate": 0.0001,
      "loss": 4.1065,
      "loss/crossentropy": 1.879043698310852,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1792997345328331,
      "step": 25632
    },
    {
      "epoch": 0.51268,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.005980428059895833,
      "learning_rate": 0.0001,
      "loss": 3.9104,
      "loss/crossentropy": 1.8325838446617126,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18299763649702072,
      "step": 25634
    },
    {
      "epoch": 0.51272,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.006241607666015625,
      "learning_rate": 0.0001,
      "loss": 3.9663,
      "loss/crossentropy": 2.1776668429374695,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21143290400505066,
      "step": 25636
    },
    {
      "epoch": 0.51276,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.003932444254557291,
      "learning_rate": 0.0001,
      "loss": 3.7604,
      "loss/crossentropy": 2.010956645011902,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18642369657754898,
      "step": 25638
    },
    {
      "epoch": 0.5128,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.0038401285807291665,
      "learning_rate": 0.0001,
      "loss": 3.9791,
      "loss/crossentropy": 2.3246684074401855,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20379284024238586,
      "step": 25640
    },
    {
      "epoch": 0.51284,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.008740234375,
      "learning_rate": 0.0001,
      "loss": 3.9221,
      "loss/crossentropy": 2.1043150424957275,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1939583420753479,
      "step": 25642
    },
    {
      "epoch": 0.51288,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.008308664957682291,
      "learning_rate": 0.0001,
      "loss": 3.8916,
      "loss/crossentropy": 1.947311520576477,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19082298129796982,
      "step": 25644
    },
    {
      "epoch": 0.51292,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.008581288655598958,
      "learning_rate": 0.0001,
      "loss": 3.991,
      "loss/crossentropy": 2.189740777015686,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19475838541984558,
      "step": 25646
    },
    {
      "epoch": 0.51296,
      "grad_norm": 1.875,
      "grad_norm_var": 0.007420857747395833,
      "learning_rate": 0.0001,
      "loss": 4.0033,
      "loss/crossentropy": 1.6541009545326233,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16929760575294495,
      "step": 25648
    },
    {
      "epoch": 0.513,
      "grad_norm": 2.125,
      "grad_norm_var": 0.011130523681640626,
      "learning_rate": 0.0001,
      "loss": 4.1202,
      "loss/crossentropy": 2.102361261844635,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20186372101306915,
      "step": 25650
    },
    {
      "epoch": 0.51304,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.016169230143229168,
      "learning_rate": 0.0001,
      "loss": 4.3193,
      "loss/crossentropy": 2.1970421075820923,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2268364280462265,
      "step": 25652
    },
    {
      "epoch": 0.51308,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.01594823201497396,
      "learning_rate": 0.0001,
      "loss": 4.0125,
      "loss/crossentropy": 1.9391701817512512,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19213733822107315,
      "step": 25654
    },
    {
      "epoch": 0.51312,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.015453084309895834,
      "learning_rate": 0.0001,
      "loss": 4.1499,
      "loss/crossentropy": 1.9313429594039917,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.184333935379982,
      "step": 25656
    },
    {
      "epoch": 0.51316,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.03003107706705729,
      "learning_rate": 0.0001,
      "loss": 4.0926,
      "loss/crossentropy": 2.221387028694153,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2127813696861267,
      "step": 25658
    },
    {
      "epoch": 0.5132,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.0320220947265625,
      "learning_rate": 0.0001,
      "loss": 4.2681,
      "loss/crossentropy": 2.0020928978919983,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1746811792254448,
      "step": 25660
    },
    {
      "epoch": 0.51324,
      "grad_norm": 3.421875,
      "grad_norm_var": 0.15138320922851561,
      "learning_rate": 0.0001,
      "loss": 3.8316,
      "loss/crossentropy": 1.960661768913269,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19446442276239395,
      "step": 25662
    },
    {
      "epoch": 0.51328,
      "grad_norm": 2.125,
      "grad_norm_var": 0.1433428446451823,
      "learning_rate": 0.0001,
      "loss": 3.7593,
      "loss/crossentropy": 1.6914892792701721,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17224331945180893,
      "step": 25664
    },
    {
      "epoch": 0.51332,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.14908447265625,
      "learning_rate": 0.0001,
      "loss": 3.7995,
      "loss/crossentropy": 1.8427329063415527,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18707136064767838,
      "step": 25666
    },
    {
      "epoch": 0.51336,
      "grad_norm": 2.453125,
      "grad_norm_var": 0.1630938212076823,
      "learning_rate": 0.0001,
      "loss": 3.7455,
      "loss/crossentropy": 1.7277203798294067,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1778879463672638,
      "step": 25668
    },
    {
      "epoch": 0.5134,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.16569417317708332,
      "learning_rate": 0.0001,
      "loss": 3.817,
      "loss/crossentropy": 1.9606621861457825,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19271928071975708,
      "step": 25670
    },
    {
      "epoch": 0.51344,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.1670000712076823,
      "learning_rate": 0.0001,
      "loss": 3.7844,
      "loss/crossentropy": 2.004323959350586,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18009979277849197,
      "step": 25672
    },
    {
      "epoch": 0.51348,
      "grad_norm": 2.59375,
      "grad_norm_var": 0.1731035868326823,
      "learning_rate": 0.0001,
      "loss": 4.1429,
      "loss/crossentropy": 2.0127062797546387,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20746992528438568,
      "step": 25674
    },
    {
      "epoch": 0.51352,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.181298828125,
      "learning_rate": 0.0001,
      "loss": 3.9813,
      "loss/crossentropy": 2.240972399711609,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19375959038734436,
      "step": 25676
    },
    {
      "epoch": 0.51356,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.06014404296875,
      "learning_rate": 0.0001,
      "loss": 4.1652,
      "loss/crossentropy": 2.1942732334136963,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21413519978523254,
      "step": 25678
    },
    {
      "epoch": 0.5136,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.061522420247395834,
      "learning_rate": 0.0001,
      "loss": 4.0474,
      "loss/crossentropy": 2.0388938188552856,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1887573003768921,
      "step": 25680
    },
    {
      "epoch": 0.51364,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.059458160400390626,
      "learning_rate": 0.0001,
      "loss": 4.0529,
      "loss/crossentropy": 2.147862672805786,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1947481632232666,
      "step": 25682
    },
    {
      "epoch": 0.51368,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.052308909098307294,
      "learning_rate": 0.0001,
      "loss": 4.0428,
      "loss/crossentropy": 1.7891934514045715,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2200366035103798,
      "step": 25684
    },
    {
      "epoch": 0.51372,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.047973378499348955,
      "learning_rate": 0.0001,
      "loss": 4.1498,
      "loss/crossentropy": 2.0402196645736694,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20540674775838852,
      "step": 25686
    },
    {
      "epoch": 0.51376,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.044535064697265626,
      "learning_rate": 0.0001,
      "loss": 3.809,
      "loss/crossentropy": 1.9355202913284302,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19018185883760452,
      "step": 25688
    },
    {
      "epoch": 0.5138,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.021648915608723958,
      "learning_rate": 0.0001,
      "loss": 3.8918,
      "loss/crossentropy": 2.140886068344116,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20067927986383438,
      "step": 25690
    },
    {
      "epoch": 0.51384,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.021149698893229166,
      "learning_rate": 0.0001,
      "loss": 3.7235,
      "loss/crossentropy": 1.9624481201171875,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.191593699157238,
      "step": 25692
    },
    {
      "epoch": 0.51388,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.021834055582682293,
      "learning_rate": 0.0001,
      "loss": 4.1363,
      "loss/crossentropy": 2.2475873231887817,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1987176612019539,
      "step": 25694
    },
    {
      "epoch": 0.51392,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.020857747395833334,
      "learning_rate": 0.0001,
      "loss": 4.0871,
      "loss/crossentropy": 2.336251735687256,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2216811552643776,
      "step": 25696
    },
    {
      "epoch": 0.51396,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.019893391927083334,
      "learning_rate": 0.0001,
      "loss": 3.9574,
      "loss/crossentropy": 1.9058191776275635,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20057585835456848,
      "step": 25698
    },
    {
      "epoch": 0.514,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.009749094645182291,
      "learning_rate": 0.0001,
      "loss": 3.8912,
      "loss/crossentropy": 1.8164226412773132,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1772766411304474,
      "step": 25700
    },
    {
      "epoch": 0.51404,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.010578409830729166,
      "learning_rate": 0.0001,
      "loss": 4.1467,
      "loss/crossentropy": 1.9770846962928772,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19005891680717468,
      "step": 25702
    },
    {
      "epoch": 0.51408,
      "grad_norm": 3.5625,
      "grad_norm_var": 0.1743242899576823,
      "learning_rate": 0.0001,
      "loss": 3.859,
      "loss/crossentropy": 1.9907479286193848,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19942636042833328,
      "step": 25704
    },
    {
      "epoch": 0.51412,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.17293472290039064,
      "learning_rate": 0.0001,
      "loss": 3.7331,
      "loss/crossentropy": 1.803969383239746,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17413488030433655,
      "step": 25706
    },
    {
      "epoch": 0.51416,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.1740386962890625,
      "learning_rate": 0.0001,
      "loss": 4.0005,
      "loss/crossentropy": 1.9947617053985596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18278180807828903,
      "step": 25708
    },
    {
      "epoch": 0.5142,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.17277399698893228,
      "learning_rate": 0.0001,
      "loss": 3.9845,
      "loss/crossentropy": 2.102198004722595,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19085385650396347,
      "step": 25710
    },
    {
      "epoch": 0.51424,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.1716631571451823,
      "learning_rate": 0.0001,
      "loss": 4.2193,
      "loss/crossentropy": 2.087872624397278,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18291640281677246,
      "step": 25712
    },
    {
      "epoch": 0.51428,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.17193094889322916,
      "learning_rate": 0.0001,
      "loss": 4.1705,
      "loss/crossentropy": 2.1515949964523315,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19207611680030823,
      "step": 25714
    },
    {
      "epoch": 0.51432,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.1703814188639323,
      "learning_rate": 0.0001,
      "loss": 4.1645,
      "loss/crossentropy": 2.2736594676971436,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21448197960853577,
      "step": 25716
    },
    {
      "epoch": 0.51436,
      "grad_norm": 2.0,
      "grad_norm_var": 0.16901219685872396,
      "learning_rate": 0.0001,
      "loss": 4.3332,
      "loss/crossentropy": 1.8625124096870422,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1784079223871231,
      "step": 25718
    },
    {
      "epoch": 0.5144,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.005600738525390625,
      "learning_rate": 0.0001,
      "loss": 3.9551,
      "loss/crossentropy": 2.2445143461227417,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20068903267383575,
      "step": 25720
    },
    {
      "epoch": 0.51444,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.005924224853515625,
      "learning_rate": 0.0001,
      "loss": 3.7845,
      "loss/crossentropy": 1.7019882202148438,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17819605767726898,
      "step": 25722
    },
    {
      "epoch": 0.51448,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.016523996988932293,
      "learning_rate": 0.0001,
      "loss": 3.8062,
      "loss/crossentropy": 2.077397406101227,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20366767793893814,
      "step": 25724
    },
    {
      "epoch": 0.51452,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0159423828125,
      "learning_rate": 0.0001,
      "loss": 4.2385,
      "loss/crossentropy": 2.1213592290878296,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19402803480625153,
      "step": 25726
    },
    {
      "epoch": 0.51456,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.016110992431640624,
      "learning_rate": 0.0001,
      "loss": 3.6229,
      "loss/crossentropy": 1.7625560760498047,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17147068679332733,
      "step": 25728
    },
    {
      "epoch": 0.5146,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.02178955078125,
      "learning_rate": 0.0001,
      "loss": 3.8784,
      "loss/crossentropy": 2.108244776725769,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20061994343996048,
      "step": 25730
    },
    {
      "epoch": 0.51464,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.023787180582682293,
      "learning_rate": 0.0001,
      "loss": 3.7433,
      "loss/crossentropy": 2.0263237953186035,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17062099277973175,
      "step": 25732
    },
    {
      "epoch": 0.51468,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.023337554931640626,
      "learning_rate": 0.0001,
      "loss": 3.9004,
      "loss/crossentropy": 2.3399864435195923,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22683630138635635,
      "step": 25734
    },
    {
      "epoch": 0.51472,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.024761708577473958,
      "learning_rate": 0.0001,
      "loss": 3.9724,
      "loss/crossentropy": 2.179167151451111,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2077057957649231,
      "step": 25736
    },
    {
      "epoch": 0.51476,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.027428944905598957,
      "learning_rate": 0.0001,
      "loss": 3.5788,
      "loss/crossentropy": 1.7649011611938477,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17765354365110397,
      "step": 25738
    },
    {
      "epoch": 0.5148,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.03184789021809896,
      "learning_rate": 0.0001,
      "loss": 4.0858,
      "loss/crossentropy": 2.01451712846756,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20021513849496841,
      "step": 25740
    },
    {
      "epoch": 0.51484,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.03407567342122396,
      "learning_rate": 0.0001,
      "loss": 3.8967,
      "loss/crossentropy": 2.043434739112854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18407747894525528,
      "step": 25742
    },
    {
      "epoch": 0.51488,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0338531494140625,
      "learning_rate": 0.0001,
      "loss": 3.9306,
      "loss/crossentropy": 1.7236329913139343,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16566471755504608,
      "step": 25744
    },
    {
      "epoch": 0.51492,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.028107706705729166,
      "learning_rate": 0.0001,
      "loss": 3.9946,
      "loss/crossentropy": 1.9381142258644104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18159441649913788,
      "step": 25746
    },
    {
      "epoch": 0.51496,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.025341542561848958,
      "learning_rate": 0.0001,
      "loss": 4.1442,
      "loss/crossentropy": 2.262045383453369,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20239067822694778,
      "step": 25748
    },
    {
      "epoch": 0.515,
      "grad_norm": 2.0,
      "grad_norm_var": 0.026682281494140626,
      "learning_rate": 0.0001,
      "loss": 4.0358,
      "loss/crossentropy": 1.980510652065277,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1910887509584427,
      "step": 25750
    },
    {
      "epoch": 0.51504,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.024494425455729166,
      "learning_rate": 0.0001,
      "loss": 3.9608,
      "loss/crossentropy": 2.1935293674468994,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19832123070955276,
      "step": 25752
    },
    {
      "epoch": 0.51508,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.023835245768229166,
      "learning_rate": 0.0001,
      "loss": 4.1082,
      "loss/crossentropy": 1.912952721118927,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1726488322019577,
      "step": 25754
    },
    {
      "epoch": 0.51512,
      "grad_norm": 1.734375,
      "grad_norm_var": 0.0124664306640625,
      "learning_rate": 0.0001,
      "loss": 3.6737,
      "loss/crossentropy": 1.7230817675590515,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16161160171031952,
      "step": 25756
    },
    {
      "epoch": 0.51516,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.010846964518229167,
      "learning_rate": 0.0001,
      "loss": 3.8654,
      "loss/crossentropy": 1.7760237455368042,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1803196743130684,
      "step": 25758
    },
    {
      "epoch": 0.5152,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.011126454671223958,
      "learning_rate": 0.0001,
      "loss": 4.0545,
      "loss/crossentropy": 2.3100990056991577,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2151026725769043,
      "step": 25760
    },
    {
      "epoch": 0.51524,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.010957845052083333,
      "learning_rate": 0.0001,
      "loss": 4.2432,
      "loss/crossentropy": 2.1021112203598022,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18987688422203064,
      "step": 25762
    },
    {
      "epoch": 0.51528,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.011139933268229167,
      "learning_rate": 0.0001,
      "loss": 3.798,
      "loss/crossentropy": 1.9850506782531738,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19890005141496658,
      "step": 25764
    },
    {
      "epoch": 0.51532,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.010253651936848959,
      "learning_rate": 0.0001,
      "loss": 4.2499,
      "loss/crossentropy": 2.37569797039032,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20863597840070724,
      "step": 25766
    },
    {
      "epoch": 0.51536,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011994425455729167,
      "learning_rate": 0.0001,
      "loss": 4.202,
      "loss/crossentropy": 2.3777748346328735,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22870129346847534,
      "step": 25768
    },
    {
      "epoch": 0.5154,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.009114329020182292,
      "learning_rate": 0.0001,
      "loss": 4.2291,
      "loss/crossentropy": 2.071221649646759,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19465549290180206,
      "step": 25770
    },
    {
      "epoch": 0.51544,
      "grad_norm": 1.7265625,
      "grad_norm_var": 0.009650675455729167,
      "learning_rate": 0.0001,
      "loss": 3.8449,
      "loss/crossentropy": 2.2113492488861084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19282811880111694,
      "step": 25772
    },
    {
      "epoch": 0.51548,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.0171539306640625,
      "learning_rate": 0.0001,
      "loss": 4.0252,
      "loss/crossentropy": 1.8800004124641418,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1772722378373146,
      "step": 25774
    },
    {
      "epoch": 0.51552,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.01837946573893229,
      "learning_rate": 0.0001,
      "loss": 3.8032,
      "loss/crossentropy": 1.9569828510284424,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1836630329489708,
      "step": 25776
    },
    {
      "epoch": 0.51556,
      "grad_norm": 2.125,
      "grad_norm_var": 0.020114898681640625,
      "learning_rate": 0.0001,
      "loss": 4.2328,
      "loss/crossentropy": 2.035541355609894,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19203660637140274,
      "step": 25778
    },
    {
      "epoch": 0.5156,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.0198638916015625,
      "learning_rate": 0.0001,
      "loss": 3.9283,
      "loss/crossentropy": 2.2393122911453247,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18675968796014786,
      "step": 25780
    },
    {
      "epoch": 0.51564,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.019632975260416668,
      "learning_rate": 0.0001,
      "loss": 4.0009,
      "loss/crossentropy": 2.0540746450424194,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20924384146928787,
      "step": 25782
    },
    {
      "epoch": 0.51568,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.01869481404622396,
      "learning_rate": 0.0001,
      "loss": 4.2234,
      "loss/crossentropy": 2.168312907218933,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21717043220996857,
      "step": 25784
    },
    {
      "epoch": 0.51572,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.017252349853515626,
      "learning_rate": 0.0001,
      "loss": 3.8948,
      "loss/crossentropy": 1.9122300744056702,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18530257046222687,
      "step": 25786
    },
    {
      "epoch": 0.51576,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.018314615885416666,
      "learning_rate": 0.0001,
      "loss": 3.7462,
      "loss/crossentropy": 1.813349425792694,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17834128439426422,
      "step": 25788
    },
    {
      "epoch": 0.5158,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.011604563395182291,
      "learning_rate": 0.0001,
      "loss": 4.197,
      "loss/crossentropy": 2.217802047729492,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1857040449976921,
      "step": 25790
    },
    {
      "epoch": 0.51584,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.011132558186848959,
      "learning_rate": 0.0001,
      "loss": 3.8139,
      "loss/crossentropy": 2.0986403822898865,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1917191743850708,
      "step": 25792
    },
    {
      "epoch": 0.51588,
      "grad_norm": 2.125,
      "grad_norm_var": 0.011250813802083334,
      "learning_rate": 0.0001,
      "loss": 4.1166,
      "loss/crossentropy": 1.7962473034858704,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1671363189816475,
      "step": 25794
    },
    {
      "epoch": 0.51592,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.012215169270833333,
      "learning_rate": 0.0001,
      "loss": 3.7758,
      "loss/crossentropy": 2.001879632472992,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19479414075613022,
      "step": 25796
    },
    {
      "epoch": 0.51596,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.012018839518229166,
      "learning_rate": 0.0001,
      "loss": 3.9258,
      "loss/crossentropy": 2.228678584098816,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20541846752166748,
      "step": 25798
    },
    {
      "epoch": 0.516,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.013679758707682291,
      "learning_rate": 0.0001,
      "loss": 4.1654,
      "loss/crossentropy": 2.283624291419983,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23651224374771118,
      "step": 25800
    },
    {
      "epoch": 0.51604,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.015006510416666667,
      "learning_rate": 0.0001,
      "loss": 3.9617,
      "loss/crossentropy": 1.9860345125198364,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1878475695848465,
      "step": 25802
    },
    {
      "epoch": 0.51608,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.01031494140625,
      "learning_rate": 0.0001,
      "loss": 3.9197,
      "loss/crossentropy": 2.1123459935188293,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22042006999254227,
      "step": 25804
    },
    {
      "epoch": 0.51612,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009886678059895833,
      "learning_rate": 0.0001,
      "loss": 3.9532,
      "loss/crossentropy": 2.092814803123474,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2060900628566742,
      "step": 25806
    },
    {
      "epoch": 0.51616,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0083404541015625,
      "learning_rate": 0.0001,
      "loss": 3.8515,
      "loss/crossentropy": 1.9071126580238342,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18779786676168442,
      "step": 25808
    },
    {
      "epoch": 0.5162,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.006819661458333333,
      "learning_rate": 0.0001,
      "loss": 3.9823,
      "loss/crossentropy": 1.9593830704689026,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17265263199806213,
      "step": 25810
    },
    {
      "epoch": 0.51624,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.00438232421875,
      "learning_rate": 0.0001,
      "loss": 4.0247,
      "loss/crossentropy": 1.8603730201721191,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18761274218559265,
      "step": 25812
    },
    {
      "epoch": 0.51628,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.007425689697265625,
      "learning_rate": 0.0001,
      "loss": 3.7386,
      "loss/crossentropy": 1.8858801126480103,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.186118483543396,
      "step": 25814
    },
    {
      "epoch": 0.51632,
      "grad_norm": 2.796875,
      "grad_norm_var": 0.051513671875,
      "learning_rate": 0.0001,
      "loss": 4.0368,
      "loss/crossentropy": 2.1853126287460327,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23933759331703186,
      "step": 25816
    },
    {
      "epoch": 0.51636,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.05208104451497396,
      "learning_rate": 0.0001,
      "loss": 4.123,
      "loss/crossentropy": 2.0599172115325928,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19681133329868317,
      "step": 25818
    },
    {
      "epoch": 0.5164,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.055663045247395834,
      "learning_rate": 0.0001,
      "loss": 4.2483,
      "loss/crossentropy": 2.371991515159607,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2135465070605278,
      "step": 25820
    },
    {
      "epoch": 0.51644,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.05572077433268229,
      "learning_rate": 0.0001,
      "loss": 4.2967,
      "loss/crossentropy": 2.1175169944763184,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1994188129901886,
      "step": 25822
    },
    {
      "epoch": 0.51648,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.059357706705729166,
      "learning_rate": 0.0001,
      "loss": 3.735,
      "loss/crossentropy": 2.074811100959778,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20690523087978363,
      "step": 25824
    },
    {
      "epoch": 0.51652,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.05923258463541667,
      "learning_rate": 0.0001,
      "loss": 3.9436,
      "loss/crossentropy": 2.041485607624054,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19222356379032135,
      "step": 25826
    },
    {
      "epoch": 0.51656,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.060212961832682294,
      "learning_rate": 0.0001,
      "loss": 3.9312,
      "loss/crossentropy": 1.9386745691299438,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20328567177057266,
      "step": 25828
    },
    {
      "epoch": 0.5166,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.055692291259765624,
      "learning_rate": 0.0001,
      "loss": 4.1459,
      "loss/crossentropy": 1.8701601028442383,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17632923275232315,
      "step": 25830
    },
    {
      "epoch": 0.51664,
      "grad_norm": 2.125,
      "grad_norm_var": 0.012333170572916666,
      "learning_rate": 0.0001,
      "loss": 4.0424,
      "loss/crossentropy": 2.258900284767151,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2108033150434494,
      "step": 25832
    },
    {
      "epoch": 0.51668,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.016806793212890626,
      "learning_rate": 0.0001,
      "loss": 4.4649,
      "loss/crossentropy": 2.10329806804657,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19907134026288986,
      "step": 25834
    },
    {
      "epoch": 0.51672,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.014208984375,
      "learning_rate": 0.0001,
      "loss": 3.9527,
      "loss/crossentropy": 1.692655324935913,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17038661241531372,
      "step": 25836
    },
    {
      "epoch": 0.51676,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.01628392537434896,
      "learning_rate": 0.0001,
      "loss": 4.122,
      "loss/crossentropy": 2.0541951060295105,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20199306309223175,
      "step": 25838
    },
    {
      "epoch": 0.5168,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.014721425374348958,
      "learning_rate": 0.0001,
      "loss": 4.2206,
      "loss/crossentropy": 2.0732097029685974,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21288836747407913,
      "step": 25840
    },
    {
      "epoch": 0.51684,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.013939412434895833,
      "learning_rate": 0.0001,
      "loss": 4.2017,
      "loss/crossentropy": 2.282121777534485,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21497543156147003,
      "step": 25842
    },
    {
      "epoch": 0.51688,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0136383056640625,
      "learning_rate": 0.0001,
      "loss": 3.978,
      "loss/crossentropy": 1.8445320129394531,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17342890799045563,
      "step": 25844
    },
    {
      "epoch": 0.51692,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.0162109375,
      "learning_rate": 0.0001,
      "loss": 3.9638,
      "loss/crossentropy": 1.9874401092529297,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18226970732212067,
      "step": 25846
    },
    {
      "epoch": 0.51696,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.015290323893229167,
      "learning_rate": 0.0001,
      "loss": 4.0861,
      "loss/crossentropy": 2.0648937225341797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19047797471284866,
      "step": 25848
    },
    {
      "epoch": 0.517,
      "grad_norm": 9.125,
      "grad_norm_var": 3.220908355712891,
      "learning_rate": 0.0001,
      "loss": 4.2993,
      "loss/crossentropy": 2.3565629720687866,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20815564692020416,
      "step": 25850
    },
    {
      "epoch": 0.51704,
      "grad_norm": 2.15625,
      "grad_norm_var": 3.200935872395833,
      "learning_rate": 0.0001,
      "loss": 4.2516,
      "loss/crossentropy": 1.9872604608535767,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20761053264141083,
      "step": 25852
    },
    {
      "epoch": 0.51708,
      "grad_norm": 1.921875,
      "grad_norm_var": 3.206274159749349,
      "learning_rate": 0.0001,
      "loss": 3.8383,
      "loss/crossentropy": 1.6661525964736938,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16651833057403564,
      "step": 25854
    },
    {
      "epoch": 0.51712,
      "grad_norm": 2.03125,
      "grad_norm_var": 3.218633778889974,
      "learning_rate": 0.0001,
      "loss": 4.2274,
      "loss/crossentropy": 2.043734908103943,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1980709508061409,
      "step": 25856
    },
    {
      "epoch": 0.51716,
      "grad_norm": 1.921875,
      "grad_norm_var": 3.2231727600097657,
      "learning_rate": 0.0001,
      "loss": 3.9739,
      "loss/crossentropy": 1.897942066192627,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18658392876386642,
      "step": 25858
    },
    {
      "epoch": 0.5172,
      "grad_norm": 1.9609375,
      "grad_norm_var": 3.216633097330729,
      "learning_rate": 0.0001,
      "loss": 4.1149,
      "loss/crossentropy": 2.1246655583381653,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21123212575912476,
      "step": 25860
    },
    {
      "epoch": 0.51724,
      "grad_norm": 2.03125,
      "grad_norm_var": 3.201690419514974,
      "learning_rate": 0.0001,
      "loss": 4.0062,
      "loss/crossentropy": 2.000667631626129,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2007341906428337,
      "step": 25862
    },
    {
      "epoch": 0.51728,
      "grad_norm": 2.015625,
      "grad_norm_var": 3.202071126302083,
      "learning_rate": 0.0001,
      "loss": 3.987,
      "loss/crossentropy": 2.09254252910614,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19557519257068634,
      "step": 25864
    },
    {
      "epoch": 0.51732,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.012505849202473959,
      "learning_rate": 0.0001,
      "loss": 4.1372,
      "loss/crossentropy": 2.0801188945770264,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20193548500537872,
      "step": 25866
    },
    {
      "epoch": 0.51736,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.004166412353515625,
      "learning_rate": 0.0001,
      "loss": 3.9061,
      "loss/crossentropy": 2.0977261662483215,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2003898024559021,
      "step": 25868
    },
    {
      "epoch": 0.5174,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.005246734619140625,
      "learning_rate": 0.0001,
      "loss": 3.7979,
      "loss/crossentropy": 1.5647385120391846,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.14602012932300568,
      "step": 25870
    },
    {
      "epoch": 0.51744,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0049631754557291664,
      "learning_rate": 0.0001,
      "loss": 3.8804,
      "loss/crossentropy": 1.9746862649917603,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19539950788021088,
      "step": 25872
    },
    {
      "epoch": 0.51748,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.005721028645833333,
      "learning_rate": 0.0001,
      "loss": 4.143,
      "loss/crossentropy": 2.08547705411911,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1976427659392357,
      "step": 25874
    },
    {
      "epoch": 0.51752,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.007647450764973958,
      "learning_rate": 0.0001,
      "loss": 4.017,
      "loss/crossentropy": 2.040421783924103,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20465777814388275,
      "step": 25876
    },
    {
      "epoch": 0.51756,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.008272043863932292,
      "learning_rate": 0.0001,
      "loss": 3.7525,
      "loss/crossentropy": 1.8793463706970215,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17017312347888947,
      "step": 25878
    },
    {
      "epoch": 0.5176,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.007903798421223959,
      "learning_rate": 0.0001,
      "loss": 4.0791,
      "loss/crossentropy": 2.263822913169861,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2253909707069397,
      "step": 25880
    },
    {
      "epoch": 0.51764,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.01856689453125,
      "learning_rate": 0.0001,
      "loss": 4.2281,
      "loss/crossentropy": 1.8976882100105286,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1699293926358223,
      "step": 25882
    },
    {
      "epoch": 0.51768,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.016641998291015626,
      "learning_rate": 0.0001,
      "loss": 4.0898,
      "loss/crossentropy": 2.102656662464142,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.189642496407032,
      "step": 25884
    },
    {
      "epoch": 0.51772,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.0206451416015625,
      "learning_rate": 0.0001,
      "loss": 4.0169,
      "loss/crossentropy": 2.1792502403259277,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2173495665192604,
      "step": 25886
    },
    {
      "epoch": 0.51776,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.021286773681640624,
      "learning_rate": 0.0001,
      "loss": 4.2587,
      "loss/crossentropy": 2.3405139446258545,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20445296168327332,
      "step": 25888
    },
    {
      "epoch": 0.5178,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0225830078125,
      "learning_rate": 0.0001,
      "loss": 3.8529,
      "loss/crossentropy": 2.116323173046112,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.204654723405838,
      "step": 25890
    },
    {
      "epoch": 0.51784,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.021628570556640626,
      "learning_rate": 0.0001,
      "loss": 3.9595,
      "loss/crossentropy": 1.9406288266181946,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19154663383960724,
      "step": 25892
    },
    {
      "epoch": 0.51788,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.0339263916015625,
      "learning_rate": 0.0001,
      "loss": 4.0066,
      "loss/crossentropy": 2.008155882358551,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18465732783079147,
      "step": 25894
    },
    {
      "epoch": 0.51792,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.03385594685872396,
      "learning_rate": 0.0001,
      "loss": 4.2731,
      "loss/crossentropy": 2.2472530007362366,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22532539069652557,
      "step": 25896
    },
    {
      "epoch": 0.51796,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.028238932291666668,
      "learning_rate": 0.0001,
      "loss": 3.9856,
      "loss/crossentropy": 2.0776681900024414,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18579304963350296,
      "step": 25898
    },
    {
      "epoch": 0.518,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.029670206705729167,
      "learning_rate": 0.0001,
      "loss": 3.8088,
      "loss/crossentropy": 2.053288221359253,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1885019615292549,
      "step": 25900
    },
    {
      "epoch": 0.51804,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.02557551066080729,
      "learning_rate": 0.0001,
      "loss": 4.0077,
      "loss/crossentropy": 1.6223698258399963,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21323902904987335,
      "step": 25902
    },
    {
      "epoch": 0.51808,
      "grad_norm": 1.71875,
      "grad_norm_var": 0.03012873331705729,
      "learning_rate": 0.0001,
      "loss": 3.8248,
      "loss/crossentropy": 2.012439727783203,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17938166856765747,
      "step": 25904
    },
    {
      "epoch": 0.51812,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.02967096964518229,
      "learning_rate": 0.0001,
      "loss": 4.0141,
      "loss/crossentropy": 1.975797176361084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2021034210920334,
      "step": 25906
    },
    {
      "epoch": 0.51816,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.03150126139322917,
      "learning_rate": 0.0001,
      "loss": 4.0165,
      "loss/crossentropy": 2.1858623027801514,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22130713611841202,
      "step": 25908
    },
    {
      "epoch": 0.5182,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.0142486572265625,
      "learning_rate": 0.0001,
      "loss": 3.9008,
      "loss/crossentropy": 1.8949698209762573,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19013772159814835,
      "step": 25910
    },
    {
      "epoch": 0.51824,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.013266754150390626,
      "learning_rate": 0.0001,
      "loss": 3.9872,
      "loss/crossentropy": 1.9657301902770996,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19463873654603958,
      "step": 25912
    },
    {
      "epoch": 0.51828,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0107818603515625,
      "learning_rate": 0.0001,
      "loss": 3.9089,
      "loss/crossentropy": 1.9067611694335938,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1767619624733925,
      "step": 25914
    },
    {
      "epoch": 0.51832,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.012032063802083333,
      "learning_rate": 0.0001,
      "loss": 3.8141,
      "loss/crossentropy": 1.9040088653564453,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19490251690149307,
      "step": 25916
    },
    {
      "epoch": 0.51836,
      "grad_norm": 1.6796875,
      "grad_norm_var": 0.014776357014973958,
      "learning_rate": 0.0001,
      "loss": 3.8649,
      "loss/crossentropy": 1.9055203795433044,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1863974630832672,
      "step": 25918
    },
    {
      "epoch": 0.5184,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.012507883707682292,
      "learning_rate": 0.0001,
      "loss": 3.8156,
      "loss/crossentropy": 2.01656973361969,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1821809709072113,
      "step": 25920
    },
    {
      "epoch": 0.51844,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.012629191080729166,
      "learning_rate": 0.0001,
      "loss": 3.9567,
      "loss/crossentropy": 1.9760606288909912,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18721874803304672,
      "step": 25922
    },
    {
      "epoch": 0.51848,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.008579254150390625,
      "learning_rate": 0.0001,
      "loss": 3.8369,
      "loss/crossentropy": 2.002072513103485,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2038237452507019,
      "step": 25924
    },
    {
      "epoch": 0.51852,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.008454386393229167,
      "learning_rate": 0.0001,
      "loss": 4.0081,
      "loss/crossentropy": 2.329070210456848,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19514091312885284,
      "step": 25926
    },
    {
      "epoch": 0.51856,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.008243815104166666,
      "learning_rate": 0.0001,
      "loss": 3.8548,
      "loss/crossentropy": 2.008804202079773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18007662147283554,
      "step": 25928
    },
    {
      "epoch": 0.5186,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.00882568359375,
      "learning_rate": 0.0001,
      "loss": 3.8927,
      "loss/crossentropy": 1.7008295059204102,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17658023536205292,
      "step": 25930
    },
    {
      "epoch": 0.51864,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.009455362955729166,
      "learning_rate": 0.0001,
      "loss": 4.2386,
      "loss/crossentropy": 2.215006470680237,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20486676692962646,
      "step": 25932
    },
    {
      "epoch": 0.51868,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.006404368082682291,
      "learning_rate": 0.0001,
      "loss": 3.6252,
      "loss/crossentropy": 1.5741249322891235,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15962044149637222,
      "step": 25934
    },
    {
      "epoch": 0.51872,
      "grad_norm": 1.7421875,
      "grad_norm_var": 0.0077545166015625,
      "learning_rate": 0.0001,
      "loss": 3.8169,
      "loss/crossentropy": 1.8923559784889221,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1873473823070526,
      "step": 25936
    },
    {
      "epoch": 0.51876,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0076067606608072914,
      "learning_rate": 0.0001,
      "loss": 3.8542,
      "loss/crossentropy": 1.7315371632575989,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17056572437286377,
      "step": 25938
    },
    {
      "epoch": 0.5188,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.007469685872395834,
      "learning_rate": 0.0001,
      "loss": 3.9419,
      "loss/crossentropy": 1.9019699096679688,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19772474467754364,
      "step": 25940
    },
    {
      "epoch": 0.51884,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.006068674723307291,
      "learning_rate": 0.0001,
      "loss": 3.9864,
      "loss/crossentropy": 2.0523456931114197,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19947312027215958,
      "step": 25942
    },
    {
      "epoch": 0.51888,
      "grad_norm": 2.0,
      "grad_norm_var": 0.005785115559895833,
      "learning_rate": 0.0001,
      "loss": 4.046,
      "loss/crossentropy": 1.9278483390808105,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19414179772138596,
      "step": 25944
    },
    {
      "epoch": 0.51892,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.004874674479166666,
      "learning_rate": 0.0001,
      "loss": 3.8827,
      "loss/crossentropy": 2.059605300426483,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18402428179979324,
      "step": 25946
    },
    {
      "epoch": 0.51896,
      "grad_norm": 2.46875,
      "grad_norm_var": 0.023219553629557292,
      "learning_rate": 0.0001,
      "loss": 4.34,
      "loss/crossentropy": 2.2166699171066284,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21071894466876984,
      "step": 25948
    },
    {
      "epoch": 0.519,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.03566665649414062,
      "learning_rate": 0.0001,
      "loss": 4.3082,
      "loss/crossentropy": 1.8669118881225586,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2712375670671463,
      "step": 25950
    },
    {
      "epoch": 0.51904,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.03169530232747396,
      "learning_rate": 0.0001,
      "loss": 4.1025,
      "loss/crossentropy": 1.9831582307815552,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18969787657260895,
      "step": 25952
    },
    {
      "epoch": 0.51908,
      "grad_norm": 3.421875,
      "grad_norm_var": 0.15215835571289063,
      "learning_rate": 0.0001,
      "loss": 4.2287,
      "loss/crossentropy": 2.092381715774536,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.31603462994098663,
      "step": 25954
    },
    {
      "epoch": 0.51912,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.15208231608072917,
      "learning_rate": 0.0001,
      "loss": 3.7941,
      "loss/crossentropy": 2.083071291446686,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18124651163816452,
      "step": 25956
    },
    {
      "epoch": 0.51916,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.1500017801920573,
      "learning_rate": 0.0001,
      "loss": 4.2498,
      "loss/crossentropy": 2.280423641204834,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21027068048715591,
      "step": 25958
    },
    {
      "epoch": 0.5192,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.15642471313476564,
      "learning_rate": 0.0001,
      "loss": 4.0311,
      "loss/crossentropy": 2.176509916782379,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1987079679965973,
      "step": 25960
    },
    {
      "epoch": 0.51924,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.16028416951497396,
      "learning_rate": 0.0001,
      "loss": 3.9031,
      "loss/crossentropy": 1.918852150440216,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17551065981388092,
      "step": 25962
    },
    {
      "epoch": 0.51928,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.15350748697916666,
      "learning_rate": 0.0001,
      "loss": 3.8734,
      "loss/crossentropy": 2.238860845565796,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2066703960299492,
      "step": 25964
    },
    {
      "epoch": 0.51932,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.1447771708170573,
      "learning_rate": 0.0001,
      "loss": 4.261,
      "loss/crossentropy": 2.1192798614501953,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21037639677524567,
      "step": 25966
    },
    {
      "epoch": 0.51936,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.14727554321289063,
      "learning_rate": 0.0001,
      "loss": 3.9908,
      "loss/crossentropy": 2.0698190927505493,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19720114767551422,
      "step": 25968
    },
    {
      "epoch": 0.5194,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.0139801025390625,
      "learning_rate": 0.0001,
      "loss": 4.3106,
      "loss/crossentropy": 2.273471713066101,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21805588901042938,
      "step": 25970
    },
    {
      "epoch": 0.51944,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.0140869140625,
      "learning_rate": 0.0001,
      "loss": 3.9038,
      "loss/crossentropy": 2.1107550263404846,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19864343851804733,
      "step": 25972
    },
    {
      "epoch": 0.51948,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.013566080729166667,
      "learning_rate": 0.0001,
      "loss": 4.0438,
      "loss/crossentropy": 1.9644930362701416,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18844127655029297,
      "step": 25974
    },
    {
      "epoch": 0.51952,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0092926025390625,
      "learning_rate": 0.0001,
      "loss": 4.1062,
      "loss/crossentropy": 2.120868682861328,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19970117509365082,
      "step": 25976
    },
    {
      "epoch": 0.51956,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.007201131184895833,
      "learning_rate": 0.0001,
      "loss": 3.8065,
      "loss/crossentropy": 1.6433513164520264,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15953880548477173,
      "step": 25978
    },
    {
      "epoch": 0.5196,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.009043121337890625,
      "learning_rate": 0.0001,
      "loss": 4.1572,
      "loss/crossentropy": 2.0936105847358704,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19382526725530624,
      "step": 25980
    },
    {
      "epoch": 0.51964,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.008234659830729166,
      "learning_rate": 0.0001,
      "loss": 3.9524,
      "loss/crossentropy": 2.3531426191329956,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20965352654457092,
      "step": 25982
    },
    {
      "epoch": 0.51968,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.006493123372395834,
      "learning_rate": 0.0001,
      "loss": 4.1327,
      "loss/crossentropy": 2.342974007129669,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20701958984136581,
      "step": 25984
    },
    {
      "epoch": 0.51972,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0052154541015625,
      "learning_rate": 0.0001,
      "loss": 4.3142,
      "loss/crossentropy": 2.193662643432617,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2072567343711853,
      "step": 25986
    },
    {
      "epoch": 0.51976,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.004809315999348958,
      "learning_rate": 0.0001,
      "loss": 4.1056,
      "loss/crossentropy": 1.8900989294052124,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1910807117819786,
      "step": 25988
    },
    {
      "epoch": 0.5198,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.005387369791666667,
      "learning_rate": 0.0001,
      "loss": 4.3945,
      "loss/crossentropy": 2.243148624897003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20235490798950195,
      "step": 25990
    },
    {
      "epoch": 0.51984,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.007513173421223958,
      "learning_rate": 0.0001,
      "loss": 3.9088,
      "loss/crossentropy": 2.105653166770935,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20262588560581207,
      "step": 25992
    },
    {
      "epoch": 0.51988,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0073811848958333336,
      "learning_rate": 0.0001,
      "loss": 4.0318,
      "loss/crossentropy": 1.9967975616455078,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18423616141080856,
      "step": 25994
    },
    {
      "epoch": 0.51992,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.009565989176432291,
      "learning_rate": 0.0001,
      "loss": 4.1648,
      "loss/crossentropy": 2.357354164123535,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2126748189330101,
      "step": 25996
    },
    {
      "epoch": 0.51996,
      "grad_norm": 1.671875,
      "grad_norm_var": 0.0246002197265625,
      "learning_rate": 0.0001,
      "loss": 3.6427,
      "loss/crossentropy": 2.22074818611145,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.187141053378582,
      "step": 25998
    },
    {
      "epoch": 0.52,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0246002197265625,
      "learning_rate": 0.0001,
      "loss": 4.1297,
      "loss/crossentropy": 2.0902404189109802,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18575181812047958,
      "step": 26000
    },
    {
      "epoch": 0.52004,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.025172678629557292,
      "learning_rate": 0.0001,
      "loss": 4.2075,
      "loss/crossentropy": 1.9274433851242065,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.178068108856678,
      "step": 26002
    },
    {
      "epoch": 0.52008,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.026362864176432292,
      "learning_rate": 0.0001,
      "loss": 4.0935,
      "loss/crossentropy": 2.3539743423461914,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2101089358329773,
      "step": 26004
    },
    {
      "epoch": 0.52012,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.027264149983723958,
      "learning_rate": 0.0001,
      "loss": 3.9747,
      "loss/crossentropy": 1.7968198657035828,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17364778369665146,
      "step": 26006
    },
    {
      "epoch": 0.52016,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.025731404622395832,
      "learning_rate": 0.0001,
      "loss": 4.0823,
      "loss/crossentropy": 1.7845246195793152,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18205135315656662,
      "step": 26008
    },
    {
      "epoch": 0.5202,
      "grad_norm": 1.875,
      "grad_norm_var": 0.027034250895182292,
      "learning_rate": 0.0001,
      "loss": 3.8676,
      "loss/crossentropy": 2.309031844139099,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21052244305610657,
      "step": 26010
    },
    {
      "epoch": 0.52024,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.024491373697916666,
      "learning_rate": 0.0001,
      "loss": 3.9956,
      "loss/crossentropy": 1.9549171328544617,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2348322793841362,
      "step": 26012
    },
    {
      "epoch": 0.52028,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009211222330729166,
      "learning_rate": 0.0001,
      "loss": 4.3597,
      "loss/crossentropy": 2.5377613306045532,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23388546705245972,
      "step": 26014
    },
    {
      "epoch": 0.52032,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009912109375,
      "learning_rate": 0.0001,
      "loss": 4.244,
      "loss/crossentropy": 2.1748557090759277,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20094042271375656,
      "step": 26016
    },
    {
      "epoch": 0.52036,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.007661946614583333,
      "learning_rate": 0.0001,
      "loss": 3.8643,
      "loss/crossentropy": 1.4638828039169312,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15480295568704605,
      "step": 26018
    },
    {
      "epoch": 0.5204,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.011408487955729166,
      "learning_rate": 0.0001,
      "loss": 4.041,
      "loss/crossentropy": 2.058153450489044,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19334695488214493,
      "step": 26020
    },
    {
      "epoch": 0.52044,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.011543782552083333,
      "learning_rate": 0.0001,
      "loss": 4.1254,
      "loss/crossentropy": 2.0739980936050415,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19739702343940735,
      "step": 26022
    },
    {
      "epoch": 0.52048,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0109619140625,
      "learning_rate": 0.0001,
      "loss": 4.1202,
      "loss/crossentropy": 2.0489348769187927,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2435540407896042,
      "step": 26024
    },
    {
      "epoch": 0.52052,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.010138956705729167,
      "learning_rate": 0.0001,
      "loss": 4.1301,
      "loss/crossentropy": 1.9589496850967407,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1817004606127739,
      "step": 26026
    },
    {
      "epoch": 0.52056,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.009032185872395833,
      "learning_rate": 0.0001,
      "loss": 4.1931,
      "loss/crossentropy": 2.2213997840881348,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20844081044197083,
      "step": 26028
    },
    {
      "epoch": 0.5206,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.010863240559895833,
      "learning_rate": 0.0001,
      "loss": 3.9974,
      "loss/crossentropy": 1.920683741569519,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19115915894508362,
      "step": 26030
    },
    {
      "epoch": 0.52064,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.012599436442057292,
      "learning_rate": 0.0001,
      "loss": 4.011,
      "loss/crossentropy": 2.142166316509247,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18952738493680954,
      "step": 26032
    },
    {
      "epoch": 0.52068,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.013067372639973958,
      "learning_rate": 0.0001,
      "loss": 4.2289,
      "loss/crossentropy": 2.1924999952316284,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21429342031478882,
      "step": 26034
    },
    {
      "epoch": 0.52072,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.010874176025390625,
      "learning_rate": 0.0001,
      "loss": 3.7799,
      "loss/crossentropy": 2.1277804374694824,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19686074554920197,
      "step": 26036
    },
    {
      "epoch": 0.52076,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.010959625244140625,
      "learning_rate": 0.0001,
      "loss": 3.8865,
      "loss/crossentropy": 2.2193630933761597,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20152346789836884,
      "step": 26038
    },
    {
      "epoch": 0.5208,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.010545857747395833,
      "learning_rate": 0.0001,
      "loss": 4.1349,
      "loss/crossentropy": 1.8031319975852966,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18107134103775024,
      "step": 26040
    },
    {
      "epoch": 0.52084,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.009032185872395833,
      "learning_rate": 0.0001,
      "loss": 4.0376,
      "loss/crossentropy": 1.9713850021362305,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17920061200857162,
      "step": 26042
    },
    {
      "epoch": 0.52088,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.00533447265625,
      "learning_rate": 0.0001,
      "loss": 4.142,
      "loss/crossentropy": 2.5888466835021973,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2081139236688614,
      "step": 26044
    },
    {
      "epoch": 0.52092,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.004811350504557292,
      "learning_rate": 0.0001,
      "loss": 4.1217,
      "loss/crossentropy": 2.4226022958755493,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20813634991645813,
      "step": 26046
    },
    {
      "epoch": 0.52096,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.005248006184895833,
      "learning_rate": 0.0001,
      "loss": 4.1199,
      "loss/crossentropy": 2.179656744003296,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19288229197263718,
      "step": 26048
    },
    {
      "epoch": 0.521,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.0041463216145833336,
      "learning_rate": 0.0001,
      "loss": 4.0571,
      "loss/crossentropy": 2.014236092567444,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1917470395565033,
      "step": 26050
    },
    {
      "epoch": 0.52104,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0035519917805989583,
      "learning_rate": 0.0001,
      "loss": 3.7101,
      "loss/crossentropy": 2.12498939037323,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18156899511814117,
      "step": 26052
    },
    {
      "epoch": 0.52108,
      "grad_norm": 1.875,
      "grad_norm_var": 0.003342437744140625,
      "learning_rate": 0.0001,
      "loss": 3.9443,
      "loss/crossentropy": 1.8441026210784912,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18120328336954117,
      "step": 26054
    },
    {
      "epoch": 0.52112,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.008603668212890625,
      "learning_rate": 0.0001,
      "loss": 3.9958,
      "loss/crossentropy": 2.047904908657074,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19617904722690582,
      "step": 26056
    },
    {
      "epoch": 0.52116,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.008697255452473959,
      "learning_rate": 0.0001,
      "loss": 4.0111,
      "loss/crossentropy": 1.7961552739143372,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16928328573703766,
      "step": 26058
    },
    {
      "epoch": 0.5212,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.008763631184895834,
      "learning_rate": 0.0001,
      "loss": 4.0501,
      "loss/crossentropy": 2.0813616514205933,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18402273952960968,
      "step": 26060
    },
    {
      "epoch": 0.52124,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.008632151285807292,
      "learning_rate": 0.0001,
      "loss": 3.8866,
      "loss/crossentropy": 2.13175368309021,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19714613258838654,
      "step": 26062
    },
    {
      "epoch": 0.52128,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.008640289306640625,
      "learning_rate": 0.0001,
      "loss": 3.9793,
      "loss/crossentropy": 2.0052719712257385,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20111151784658432,
      "step": 26064
    },
    {
      "epoch": 0.52132,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.013216145833333333,
      "learning_rate": 0.0001,
      "loss": 4.2311,
      "loss/crossentropy": 1.939625859260559,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20675888657569885,
      "step": 26066
    },
    {
      "epoch": 0.52136,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.013255818684895834,
      "learning_rate": 0.0001,
      "loss": 3.7256,
      "loss/crossentropy": 2.208404779434204,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18266795575618744,
      "step": 26068
    },
    {
      "epoch": 0.5214,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.0123687744140625,
      "learning_rate": 0.0001,
      "loss": 3.8233,
      "loss/crossentropy": 1.8239786028862,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18409103155136108,
      "step": 26070
    },
    {
      "epoch": 0.52144,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.007743326822916666,
      "learning_rate": 0.0001,
      "loss": 4.0984,
      "loss/crossentropy": 1.951551616191864,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17273114621639252,
      "step": 26072
    },
    {
      "epoch": 0.52148,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.007899729410807292,
      "learning_rate": 0.0001,
      "loss": 3.6726,
      "loss/crossentropy": 1.6747522354125977,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1755337342619896,
      "step": 26074
    },
    {
      "epoch": 0.52152,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.007661946614583333,
      "learning_rate": 0.0001,
      "loss": 4.0389,
      "loss/crossentropy": 1.954121172428131,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2060263231396675,
      "step": 26076
    },
    {
      "epoch": 0.52156,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.007757314046223958,
      "learning_rate": 0.0001,
      "loss": 4.0388,
      "loss/crossentropy": 2.025652050971985,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19741995632648468,
      "step": 26078
    },
    {
      "epoch": 0.5216,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.010249582926432292,
      "learning_rate": 0.0001,
      "loss": 4.0048,
      "loss/crossentropy": 2.150280773639679,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2016005516052246,
      "step": 26080
    },
    {
      "epoch": 0.52164,
      "grad_norm": 2.125,
      "grad_norm_var": 0.009211985270182292,
      "learning_rate": 0.0001,
      "loss": 4.0257,
      "loss/crossentropy": 1.9364630579948425,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17605994641780853,
      "step": 26082
    },
    {
      "epoch": 0.52168,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.009226226806640625,
      "learning_rate": 0.0001,
      "loss": 4.1845,
      "loss/crossentropy": 2.195487380027771,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19364381581544876,
      "step": 26084
    },
    {
      "epoch": 0.52172,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.0098785400390625,
      "learning_rate": 0.0001,
      "loss": 3.7514,
      "loss/crossentropy": 1.6146498918533325,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1726212576031685,
      "step": 26086
    },
    {
      "epoch": 0.52176,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.009968821207682292,
      "learning_rate": 0.0001,
      "loss": 3.8782,
      "loss/crossentropy": 2.2554049491882324,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20600321888923645,
      "step": 26088
    },
    {
      "epoch": 0.5218,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.010210927327473958,
      "learning_rate": 0.0001,
      "loss": 3.8424,
      "loss/crossentropy": 2.0821692943573,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20989617705345154,
      "step": 26090
    },
    {
      "epoch": 0.52184,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.009765625,
      "learning_rate": 0.0001,
      "loss": 4.1895,
      "loss/crossentropy": 2.114021897315979,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20276522636413574,
      "step": 26092
    },
    {
      "epoch": 0.52188,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.009897613525390625,
      "learning_rate": 0.0001,
      "loss": 3.9856,
      "loss/crossentropy": 1.7443894147872925,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15722404420375824,
      "step": 26094
    },
    {
      "epoch": 0.52192,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.010796864827473959,
      "learning_rate": 0.0001,
      "loss": 3.5758,
      "loss/crossentropy": 1.603552222251892,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.14188075065612793,
      "step": 26096
    },
    {
      "epoch": 0.52196,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0070269266764322914,
      "learning_rate": 0.0001,
      "loss": 3.9047,
      "loss/crossentropy": 1.7003676295280457,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1696496084332466,
      "step": 26098
    },
    {
      "epoch": 0.522,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.0069964090983072914,
      "learning_rate": 0.0001,
      "loss": 3.9792,
      "loss/crossentropy": 2.1242096424102783,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1936848685145378,
      "step": 26100
    },
    {
      "epoch": 0.52204,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0076416015625,
      "learning_rate": 0.0001,
      "loss": 3.9315,
      "loss/crossentropy": 2.012674570083618,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2068016678094864,
      "step": 26102
    },
    {
      "epoch": 0.52208,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.014304351806640626,
      "learning_rate": 0.0001,
      "loss": 4.0618,
      "loss/crossentropy": 2.0136077404022217,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1968667060136795,
      "step": 26104
    },
    {
      "epoch": 0.52212,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.0141021728515625,
      "learning_rate": 0.0001,
      "loss": 3.9797,
      "loss/crossentropy": 1.8880528211593628,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18080192804336548,
      "step": 26106
    },
    {
      "epoch": 0.52216,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.014216105143229166,
      "learning_rate": 0.0001,
      "loss": 4.0981,
      "loss/crossentropy": 2.109636068344116,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19920819997787476,
      "step": 26108
    },
    {
      "epoch": 0.5222,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.015236155192057291,
      "learning_rate": 0.0001,
      "loss": 4.0566,
      "loss/crossentropy": 1.8923597931861877,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18053779006004333,
      "step": 26110
    },
    {
      "epoch": 0.52224,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.011324055989583333,
      "learning_rate": 0.0001,
      "loss": 4.0756,
      "loss/crossentropy": 2.179795265197754,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18849624693393707,
      "step": 26112
    },
    {
      "epoch": 0.52228,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.011789703369140625,
      "learning_rate": 0.0001,
      "loss": 3.9964,
      "loss/crossentropy": 2.0764212608337402,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18905582278966904,
      "step": 26114
    },
    {
      "epoch": 0.52232,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.012634023030598959,
      "learning_rate": 0.0001,
      "loss": 4.0006,
      "loss/crossentropy": 2.2309343814849854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2156553491950035,
      "step": 26116
    },
    {
      "epoch": 0.52236,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.012261708577473959,
      "learning_rate": 0.0001,
      "loss": 3.8689,
      "loss/crossentropy": 1.8764967918395996,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17533813416957855,
      "step": 26118
    },
    {
      "epoch": 0.5224,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0075266520182291664,
      "learning_rate": 0.0001,
      "loss": 3.916,
      "loss/crossentropy": 2.1870853900909424,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1984037682414055,
      "step": 26120
    },
    {
      "epoch": 0.52244,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.007838694254557292,
      "learning_rate": 0.0001,
      "loss": 3.716,
      "loss/crossentropy": 2.113897442817688,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18819265812635422,
      "step": 26122
    },
    {
      "epoch": 0.52248,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.008056386311848959,
      "learning_rate": 0.0001,
      "loss": 4.1003,
      "loss/crossentropy": 1.9980355501174927,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20408682525157928,
      "step": 26124
    },
    {
      "epoch": 0.52252,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.007784016927083333,
      "learning_rate": 0.0001,
      "loss": 4.0452,
      "loss/crossentropy": 2.0322210788726807,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19906745105981827,
      "step": 26126
    },
    {
      "epoch": 0.52256,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.008304595947265625,
      "learning_rate": 0.0001,
      "loss": 3.9786,
      "loss/crossentropy": 2.0122641921043396,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2028798833489418,
      "step": 26128
    },
    {
      "epoch": 0.5226,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.007352447509765625,
      "learning_rate": 0.0001,
      "loss": 4.0595,
      "loss/crossentropy": 2.0169489979743958,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20758477598428726,
      "step": 26130
    },
    {
      "epoch": 0.52264,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.006754302978515625,
      "learning_rate": 0.0001,
      "loss": 4.2196,
      "loss/crossentropy": 2.45787513256073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2269362136721611,
      "step": 26132
    },
    {
      "epoch": 0.52268,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.0064198811848958336,
      "learning_rate": 0.0001,
      "loss": 3.9921,
      "loss/crossentropy": 2.0325412154197693,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20839164406061172,
      "step": 26134
    },
    {
      "epoch": 0.52272,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.006815338134765625,
      "learning_rate": 0.0001,
      "loss": 4.1983,
      "loss/crossentropy": 2.1984806060791016,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21555981040000916,
      "step": 26136
    },
    {
      "epoch": 0.52276,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.005440266927083334,
      "learning_rate": 0.0001,
      "loss": 3.9955,
      "loss/crossentropy": 1.9628766775131226,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2089555338025093,
      "step": 26138
    },
    {
      "epoch": 0.5228,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.0069882710774739586,
      "learning_rate": 0.0001,
      "loss": 3.9137,
      "loss/crossentropy": 2.1061203479766846,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18837912380695343,
      "step": 26140
    },
    {
      "epoch": 0.52284,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.014485677083333334,
      "learning_rate": 0.0001,
      "loss": 3.6926,
      "loss/crossentropy": 2.120006561279297,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19504809379577637,
      "step": 26142
    },
    {
      "epoch": 0.52288,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.013749186197916667,
      "learning_rate": 0.0001,
      "loss": 3.8771,
      "loss/crossentropy": 2.0000118613243103,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19644330441951752,
      "step": 26144
    },
    {
      "epoch": 0.52292,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.014411417643229167,
      "learning_rate": 0.0001,
      "loss": 4.2381,
      "loss/crossentropy": 2.261639356613159,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21112587302923203,
      "step": 26146
    },
    {
      "epoch": 0.52296,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.013392893473307292,
      "learning_rate": 0.0001,
      "loss": 4.0808,
      "loss/crossentropy": 2.203276216983795,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19632557779550552,
      "step": 26148
    },
    {
      "epoch": 0.523,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0129302978515625,
      "learning_rate": 0.0001,
      "loss": 4.1363,
      "loss/crossentropy": 2.1430857181549072,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18741771578788757,
      "step": 26150
    },
    {
      "epoch": 0.52304,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.012581380208333333,
      "learning_rate": 0.0001,
      "loss": 3.9574,
      "loss/crossentropy": 2.094003438949585,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19085614383220673,
      "step": 26152
    },
    {
      "epoch": 0.52308,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.011800130208333334,
      "learning_rate": 0.0001,
      "loss": 3.9145,
      "loss/crossentropy": 1.9885526299476624,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17914890497922897,
      "step": 26154
    },
    {
      "epoch": 0.52312,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.012263743082682292,
      "learning_rate": 0.0001,
      "loss": 3.9942,
      "loss/crossentropy": 1.885224461555481,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1779332235455513,
      "step": 26156
    },
    {
      "epoch": 0.52316,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.007411448160807291,
      "learning_rate": 0.0001,
      "loss": 3.9081,
      "loss/crossentropy": 1.8236491084098816,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17156238853931427,
      "step": 26158
    },
    {
      "epoch": 0.5232,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.008685048421223958,
      "learning_rate": 0.0001,
      "loss": 3.8614,
      "loss/crossentropy": 1.7451134324073792,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16748512536287308,
      "step": 26160
    },
    {
      "epoch": 0.52324,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.00872802734375,
      "learning_rate": 0.0001,
      "loss": 4.0804,
      "loss/crossentropy": 2.408850908279419,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20247718691825867,
      "step": 26162
    },
    {
      "epoch": 0.52328,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0099273681640625,
      "learning_rate": 0.0001,
      "loss": 4.0908,
      "loss/crossentropy": 2.265584349632263,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20462538301944733,
      "step": 26164
    },
    {
      "epoch": 0.52332,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.008676910400390625,
      "learning_rate": 0.0001,
      "loss": 3.7925,
      "loss/crossentropy": 1.8766103982925415,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1740010604262352,
      "step": 26166
    },
    {
      "epoch": 0.52336,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.009413401285807291,
      "learning_rate": 0.0001,
      "loss": 3.9389,
      "loss/crossentropy": 1.858881413936615,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.176437109708786,
      "step": 26168
    },
    {
      "epoch": 0.5234,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.008907063802083334,
      "learning_rate": 0.0001,
      "loss": 3.9833,
      "loss/crossentropy": 1.733948528766632,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.183179572224617,
      "step": 26170
    },
    {
      "epoch": 0.52344,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.017923990885416668,
      "learning_rate": 0.0001,
      "loss": 4.4352,
      "loss/crossentropy": 1.7623996138572693,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.3073955178260803,
      "step": 26172
    },
    {
      "epoch": 0.52348,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.017470041910807293,
      "learning_rate": 0.0001,
      "loss": 4.0571,
      "loss/crossentropy": 1.9229013919830322,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19095296412706375,
      "step": 26174
    },
    {
      "epoch": 0.52352,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.014867146809895834,
      "learning_rate": 0.0001,
      "loss": 4.1139,
      "loss/crossentropy": 2.155863106250763,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1953117996454239,
      "step": 26176
    },
    {
      "epoch": 0.52356,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.01754735310872396,
      "learning_rate": 0.0001,
      "loss": 3.8574,
      "loss/crossentropy": 1.8662053942680359,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17596308141946793,
      "step": 26178
    },
    {
      "epoch": 0.5236,
      "grad_norm": 2.4375,
      "grad_norm_var": 0.03288548787434896,
      "learning_rate": 0.0001,
      "loss": 4.2924,
      "loss/crossentropy": 2.0425168871879578,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2088511660695076,
      "step": 26180
    },
    {
      "epoch": 0.52364,
      "grad_norm": 1.875,
      "grad_norm_var": 0.031689453125,
      "learning_rate": 0.0001,
      "loss": 4.0537,
      "loss/crossentropy": 2.3064836263656616,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20652327686548233,
      "step": 26182
    },
    {
      "epoch": 0.52368,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.035065714518229166,
      "learning_rate": 0.0001,
      "loss": 4.2602,
      "loss/crossentropy": 2.437976837158203,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21545469760894775,
      "step": 26184
    },
    {
      "epoch": 0.52372,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.034645334879557295,
      "learning_rate": 0.0001,
      "loss": 3.8111,
      "loss/crossentropy": 1.8655226826667786,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19221200793981552,
      "step": 26186
    },
    {
      "epoch": 0.52376,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.027339426676432292,
      "learning_rate": 0.0001,
      "loss": 3.9068,
      "loss/crossentropy": 1.5981062650680542,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15811648964881897,
      "step": 26188
    },
    {
      "epoch": 0.5238,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.02739232381184896,
      "learning_rate": 0.0001,
      "loss": 3.6361,
      "loss/crossentropy": 2.0123122334480286,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21117129176855087,
      "step": 26190
    },
    {
      "epoch": 0.52384,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.02658869425455729,
      "learning_rate": 0.0001,
      "loss": 4.2715,
      "loss/crossentropy": 1.8640416860580444,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17403434216976166,
      "step": 26192
    },
    {
      "epoch": 0.52388,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.026683553059895834,
      "learning_rate": 0.0001,
      "loss": 3.9331,
      "loss/crossentropy": 1.9721081256866455,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18515276163816452,
      "step": 26194
    },
    {
      "epoch": 0.52392,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0124176025390625,
      "learning_rate": 0.0001,
      "loss": 3.9945,
      "loss/crossentropy": 1.9967116713523865,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1789851263165474,
      "step": 26196
    },
    {
      "epoch": 0.52396,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.012467193603515624,
      "learning_rate": 0.0001,
      "loss": 4.0541,
      "loss/crossentropy": 1.9825578331947327,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1994011253118515,
      "step": 26198
    },
    {
      "epoch": 0.524,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.011922200520833334,
      "learning_rate": 0.0001,
      "loss": 3.8301,
      "loss/crossentropy": 1.9351627230644226,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18274720758199692,
      "step": 26200
    },
    {
      "epoch": 0.52404,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.011970011393229167,
      "learning_rate": 0.0001,
      "loss": 3.7656,
      "loss/crossentropy": 1.9799736738204956,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18502405285835266,
      "step": 26202
    },
    {
      "epoch": 0.52408,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.011896769205729166,
      "learning_rate": 0.0001,
      "loss": 3.8567,
      "loss/crossentropy": 1.6783319115638733,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.14768389612436295,
      "step": 26204
    },
    {
      "epoch": 0.52412,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.022709147135416666,
      "learning_rate": 0.0001,
      "loss": 3.8827,
      "loss/crossentropy": 2.167922616004944,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2303374707698822,
      "step": 26206
    },
    {
      "epoch": 0.52416,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.022359212239583332,
      "learning_rate": 0.0001,
      "loss": 3.9447,
      "loss/crossentropy": 2.075382351875305,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19902942329645157,
      "step": 26208
    },
    {
      "epoch": 0.5242,
      "grad_norm": 1.6796875,
      "grad_norm_var": 0.021203358968098957,
      "learning_rate": 0.0001,
      "loss": 3.7008,
      "loss/crossentropy": 1.9230342507362366,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18066919595003128,
      "step": 26210
    },
    {
      "epoch": 0.52424,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.021134440104166666,
      "learning_rate": 0.0001,
      "loss": 3.7475,
      "loss/crossentropy": 2.000435531139374,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19018910825252533,
      "step": 26212
    },
    {
      "epoch": 0.52428,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.025569407145182292,
      "learning_rate": 0.0001,
      "loss": 4.1926,
      "loss/crossentropy": 1.887748658657074,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17914170771837234,
      "step": 26214
    },
    {
      "epoch": 0.52432,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.023911285400390624,
      "learning_rate": 0.0001,
      "loss": 3.8331,
      "loss/crossentropy": 1.9090191721916199,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18189192563295364,
      "step": 26216
    },
    {
      "epoch": 0.52436,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.023331705729166666,
      "learning_rate": 0.0001,
      "loss": 3.9697,
      "loss/crossentropy": 1.9368041157722473,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18426424264907837,
      "step": 26218
    },
    {
      "epoch": 0.5244,
      "grad_norm": 1.875,
      "grad_norm_var": 0.022761027018229168,
      "learning_rate": 0.0001,
      "loss": 3.7137,
      "loss/crossentropy": 2.1554033160209656,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20169231295585632,
      "step": 26220
    },
    {
      "epoch": 0.52444,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.011848704020182291,
      "learning_rate": 0.0001,
      "loss": 3.78,
      "loss/crossentropy": 2.013387084007263,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1921246349811554,
      "step": 26222
    },
    {
      "epoch": 0.52448,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.011848704020182291,
      "learning_rate": 0.0001,
      "loss": 4.045,
      "loss/crossentropy": 2.166366934776306,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19481656700372696,
      "step": 26224
    },
    {
      "epoch": 0.52452,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.012669881184895834,
      "learning_rate": 0.0001,
      "loss": 3.9517,
      "loss/crossentropy": 2.15239155292511,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20820968598127365,
      "step": 26226
    },
    {
      "epoch": 0.52456,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.014229329427083333,
      "learning_rate": 0.0001,
      "loss": 3.9075,
      "loss/crossentropy": 2.3724613189697266,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21621037274599075,
      "step": 26228
    },
    {
      "epoch": 0.5246,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.008888498942057291,
      "learning_rate": 0.0001,
      "loss": 3.9089,
      "loss/crossentropy": 2.34514319896698,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21700077503919601,
      "step": 26230
    },
    {
      "epoch": 0.52464,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.009085845947265626,
      "learning_rate": 0.0001,
      "loss": 3.8283,
      "loss/crossentropy": 1.7633844017982483,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17453230917453766,
      "step": 26232
    },
    {
      "epoch": 0.52468,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.010135650634765625,
      "learning_rate": 0.0001,
      "loss": 3.6374,
      "loss/crossentropy": 1.952212929725647,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18606504052877426,
      "step": 26234
    },
    {
      "epoch": 0.52472,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.012851715087890625,
      "learning_rate": 0.0001,
      "loss": 4.4356,
      "loss/crossentropy": 2.2077181935310364,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19942647963762283,
      "step": 26236
    },
    {
      "epoch": 0.52476,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.014029947916666667,
      "learning_rate": 0.0001,
      "loss": 4.2066,
      "loss/crossentropy": 2.110091209411621,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2050892785191536,
      "step": 26238
    },
    {
      "epoch": 0.5248,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.014284006754557292,
      "learning_rate": 0.0001,
      "loss": 3.9062,
      "loss/crossentropy": 2.2912864685058594,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2043801248073578,
      "step": 26240
    },
    {
      "epoch": 0.52484,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.010396067301432292,
      "learning_rate": 0.0001,
      "loss": 3.9453,
      "loss/crossentropy": 2.280913293361664,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21790295839309692,
      "step": 26242
    },
    {
      "epoch": 0.52488,
      "grad_norm": 1.875,
      "grad_norm_var": 0.008568318684895833,
      "learning_rate": 0.0001,
      "loss": 3.9073,
      "loss/crossentropy": 1.9681018590927124,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18275773525238037,
      "step": 26244
    },
    {
      "epoch": 0.52492,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.009073893229166666,
      "learning_rate": 0.0001,
      "loss": 4.0666,
      "loss/crossentropy": 2.1007355451583862,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20663000643253326,
      "step": 26246
    },
    {
      "epoch": 0.52496,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.009029134114583334,
      "learning_rate": 0.0001,
      "loss": 4.0375,
      "loss/crossentropy": 2.1250157952308655,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20055226981639862,
      "step": 26248
    },
    {
      "epoch": 0.525,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.008451334635416667,
      "learning_rate": 0.0001,
      "loss": 3.8399,
      "loss/crossentropy": 1.8575809597969055,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18729300796985626,
      "step": 26250
    },
    {
      "epoch": 0.52504,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.006150054931640625,
      "learning_rate": 0.0001,
      "loss": 4.0213,
      "loss/crossentropy": 2.172927498817444,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20007719844579697,
      "step": 26252
    },
    {
      "epoch": 0.52508,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.005271148681640625,
      "learning_rate": 0.0001,
      "loss": 3.8717,
      "loss/crossentropy": 2.200989246368408,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2174500823020935,
      "step": 26254
    },
    {
      "epoch": 0.52512,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.004835764567057292,
      "learning_rate": 0.0001,
      "loss": 4.051,
      "loss/crossentropy": 2.1007824540138245,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20543181151151657,
      "step": 26256
    },
    {
      "epoch": 0.52516,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.3579241434733073,
      "learning_rate": 0.0001,
      "loss": 3.9813,
      "loss/crossentropy": 1.8371055722236633,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18277855217456818,
      "step": 26258
    },
    {
      "epoch": 0.5252,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.3564849853515625,
      "learning_rate": 0.0001,
      "loss": 3.8955,
      "loss/crossentropy": 1.9994609355926514,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19775213301181793,
      "step": 26260
    },
    {
      "epoch": 0.52524,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.3601519266764323,
      "learning_rate": 0.0001,
      "loss": 3.8842,
      "loss/crossentropy": 1.9619916677474976,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1894378885626793,
      "step": 26262
    },
    {
      "epoch": 0.52528,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.3581438700358073,
      "learning_rate": 0.0001,
      "loss": 4.0634,
      "loss/crossentropy": 1.7938214540481567,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18398133665323257,
      "step": 26264
    },
    {
      "epoch": 0.52532,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.3526995340983073,
      "learning_rate": 0.0001,
      "loss": 4.1002,
      "loss/crossentropy": 2.0497565865516663,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19529610127210617,
      "step": 26266
    },
    {
      "epoch": 0.52536,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.3564715067545573,
      "learning_rate": 0.0001,
      "loss": 3.8681,
      "loss/crossentropy": 2.2085641622543335,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2090958058834076,
      "step": 26268
    },
    {
      "epoch": 0.5254,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.3512034098307292,
      "learning_rate": 0.0001,
      "loss": 4.1566,
      "loss/crossentropy": 2.143643379211426,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21305573731660843,
      "step": 26270
    },
    {
      "epoch": 0.52544,
      "grad_norm": 2.34375,
      "grad_norm_var": 0.35494969685872396,
      "learning_rate": 0.0001,
      "loss": 4.1428,
      "loss/crossentropy": 2.2753764390945435,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21875588595867157,
      "step": 26272
    },
    {
      "epoch": 0.52548,
      "grad_norm": 2.0,
      "grad_norm_var": 0.09860610961914062,
      "learning_rate": 0.0001,
      "loss": 4.1114,
      "loss/crossentropy": 2.0325489044189453,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19430440664291382,
      "step": 26274
    },
    {
      "epoch": 0.52552,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0970293680826823,
      "learning_rate": 0.0001,
      "loss": 3.9278,
      "loss/crossentropy": 2.3086185455322266,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21185901761054993,
      "step": 26276
    },
    {
      "epoch": 0.52556,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0931060791015625,
      "learning_rate": 0.0001,
      "loss": 4.1219,
      "loss/crossentropy": 2.1545599699020386,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2019871026277542,
      "step": 26278
    },
    {
      "epoch": 0.5256,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.08904622395833334,
      "learning_rate": 0.0001,
      "loss": 4.0154,
      "loss/crossentropy": 1.8528248071670532,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17894673347473145,
      "step": 26280
    },
    {
      "epoch": 0.52564,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.09781901041666667,
      "learning_rate": 0.0001,
      "loss": 3.6732,
      "loss/crossentropy": 1.7371985912322998,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17898088693618774,
      "step": 26282
    },
    {
      "epoch": 0.52568,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.10242284138997396,
      "learning_rate": 0.0001,
      "loss": 3.7005,
      "loss/crossentropy": 1.8484544157981873,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16372547298669815,
      "step": 26284
    },
    {
      "epoch": 0.52572,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.10460586547851562,
      "learning_rate": 0.0001,
      "loss": 3.9015,
      "loss/crossentropy": 2.2279897928237915,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19722852110862732,
      "step": 26286
    },
    {
      "epoch": 0.52576,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.09019749959309896,
      "learning_rate": 0.0001,
      "loss": 4.0875,
      "loss/crossentropy": 1.9138891100883484,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17364701628684998,
      "step": 26288
    },
    {
      "epoch": 0.5258,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.010786946614583333,
      "learning_rate": 0.0001,
      "loss": 3.8941,
      "loss/crossentropy": 2.062381386756897,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20237115025520325,
      "step": 26290
    },
    {
      "epoch": 0.52584,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.010548655192057292,
      "learning_rate": 0.0001,
      "loss": 4.0351,
      "loss/crossentropy": 2.094722032546997,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1814640611410141,
      "step": 26292
    },
    {
      "epoch": 0.52588,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010334269205729166,
      "learning_rate": 0.0001,
      "loss": 4.2114,
      "loss/crossentropy": 1.6870167255401611,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19745031744241714,
      "step": 26294
    },
    {
      "epoch": 0.52592,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.007062784830729167,
      "learning_rate": 0.0001,
      "loss": 3.7982,
      "loss/crossentropy": 1.688876748085022,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17806339263916016,
      "step": 26296
    },
    {
      "epoch": 0.52596,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.006898752848307292,
      "learning_rate": 0.0001,
      "loss": 3.8256,
      "loss/crossentropy": 2.0577162504196167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18965084105730057,
      "step": 26298
    },
    {
      "epoch": 0.526,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.005057779947916666,
      "learning_rate": 0.0001,
      "loss": 3.8858,
      "loss/crossentropy": 1.9294962882995605,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19645613431930542,
      "step": 26300
    },
    {
      "epoch": 0.52604,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0063435872395833336,
      "learning_rate": 0.0001,
      "loss": 3.9,
      "loss/crossentropy": 2.1160236597061157,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20182175934314728,
      "step": 26302
    },
    {
      "epoch": 0.52608,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.005232747395833333,
      "learning_rate": 0.0001,
      "loss": 3.9438,
      "loss/crossentropy": 2.0521216988563538,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19732891023159027,
      "step": 26304
    },
    {
      "epoch": 0.52612,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.004789225260416667,
      "learning_rate": 0.0001,
      "loss": 4.1944,
      "loss/crossentropy": 2.3100136518478394,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20355679094791412,
      "step": 26306
    },
    {
      "epoch": 0.52616,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0065081278483072914,
      "learning_rate": 0.0001,
      "loss": 4.1077,
      "loss/crossentropy": 2.2635756731033325,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21735866367816925,
      "step": 26308
    },
    {
      "epoch": 0.5262,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.008512115478515625,
      "learning_rate": 0.0001,
      "loss": 4.0985,
      "loss/crossentropy": 1.8445220589637756,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18094825744628906,
      "step": 26310
    },
    {
      "epoch": 0.52624,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.007651519775390625,
      "learning_rate": 0.0001,
      "loss": 3.9866,
      "loss/crossentropy": 2.174062728881836,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19913697242736816,
      "step": 26312
    },
    {
      "epoch": 0.52628,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.008565266927083334,
      "learning_rate": 0.0001,
      "loss": 3.8556,
      "loss/crossentropy": 2.1039488911628723,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1815500482916832,
      "step": 26314
    },
    {
      "epoch": 0.52632,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.008491770426432291,
      "learning_rate": 0.0001,
      "loss": 4.3284,
      "loss/crossentropy": 2.134014844894409,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20855064690113068,
      "step": 26316
    },
    {
      "epoch": 0.52636,
      "grad_norm": 1.875,
      "grad_norm_var": 0.008050282796223959,
      "learning_rate": 0.0001,
      "loss": 3.9977,
      "loss/crossentropy": 1.7635084390640259,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16970381140708923,
      "step": 26318
    },
    {
      "epoch": 0.5264,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0078122456868489586,
      "learning_rate": 0.0001,
      "loss": 4.0349,
      "loss/crossentropy": 2.4526535272598267,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23066413402557373,
      "step": 26320
    },
    {
      "epoch": 0.52644,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.008210245768229167,
      "learning_rate": 0.0001,
      "loss": 4.1903,
      "loss/crossentropy": 2.073263466358185,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19519560784101486,
      "step": 26322
    },
    {
      "epoch": 0.52648,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.007401529947916667,
      "learning_rate": 0.0001,
      "loss": 3.9606,
      "loss/crossentropy": 1.9899149537086487,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18811341375112534,
      "step": 26324
    },
    {
      "epoch": 0.52652,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.013899739583333333,
      "learning_rate": 0.0001,
      "loss": 3.9753,
      "loss/crossentropy": 2.039364755153656,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18913507461547852,
      "step": 26326
    },
    {
      "epoch": 0.52656,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.014096832275390625,
      "learning_rate": 0.0001,
      "loss": 3.9944,
      "loss/crossentropy": 2.076085150241852,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21231256425380707,
      "step": 26328
    },
    {
      "epoch": 0.5266,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.016331990559895832,
      "learning_rate": 0.0001,
      "loss": 3.8768,
      "loss/crossentropy": 2.1222329139709473,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18735431134700775,
      "step": 26330
    },
    {
      "epoch": 0.52664,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.01644872029622396,
      "learning_rate": 0.0001,
      "loss": 4.206,
      "loss/crossentropy": 2.490665316581726,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19682377576828003,
      "step": 26332
    },
    {
      "epoch": 0.52668,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.01778132120768229,
      "learning_rate": 0.0001,
      "loss": 4.0541,
      "loss/crossentropy": 2.2802486419677734,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20574340224266052,
      "step": 26334
    },
    {
      "epoch": 0.52672,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.019925944010416665,
      "learning_rate": 0.0001,
      "loss": 3.9517,
      "loss/crossentropy": 2.094264805316925,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18822843581438065,
      "step": 26336
    },
    {
      "epoch": 0.52676,
      "grad_norm": 2.0,
      "grad_norm_var": 0.020151519775390626,
      "learning_rate": 0.0001,
      "loss": 3.7813,
      "loss/crossentropy": 1.9270477294921875,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1901463121175766,
      "step": 26338
    },
    {
      "epoch": 0.5268,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.018046061197916668,
      "learning_rate": 0.0001,
      "loss": 4.1199,
      "loss/crossentropy": 2.1571450233459473,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1919422224164009,
      "step": 26340
    },
    {
      "epoch": 0.52684,
      "grad_norm": 1.875,
      "grad_norm_var": 0.008902994791666667,
      "learning_rate": 0.0001,
      "loss": 3.9607,
      "loss/crossentropy": 1.9483349323272705,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.188851498067379,
      "step": 26342
    },
    {
      "epoch": 0.52688,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.009401194254557292,
      "learning_rate": 0.0001,
      "loss": 3.9986,
      "loss/crossentropy": 2.0212563276290894,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19506627321243286,
      "step": 26344
    },
    {
      "epoch": 0.52692,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.004689280192057292,
      "learning_rate": 0.0001,
      "loss": 3.9598,
      "loss/crossentropy": 1.9661552906036377,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18422409892082214,
      "step": 26346
    },
    {
      "epoch": 0.52696,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0053212483723958336,
      "learning_rate": 0.0001,
      "loss": 3.851,
      "loss/crossentropy": 2.0267988443374634,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18864717334508896,
      "step": 26348
    },
    {
      "epoch": 0.527,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.006468709309895833,
      "learning_rate": 0.0001,
      "loss": 3.8491,
      "loss/crossentropy": 1.874852180480957,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17451918125152588,
      "step": 26350
    },
    {
      "epoch": 0.52704,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0056722005208333336,
      "learning_rate": 0.0001,
      "loss": 3.8164,
      "loss/crossentropy": 1.9450770020484924,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19308660924434662,
      "step": 26352
    },
    {
      "epoch": 0.52708,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.004589589436848959,
      "learning_rate": 0.0001,
      "loss": 3.9731,
      "loss/crossentropy": 1.8964659571647644,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18824025243520737,
      "step": 26354
    },
    {
      "epoch": 0.52712,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0045166015625,
      "learning_rate": 0.0001,
      "loss": 3.9693,
      "loss/crossentropy": 1.856788158416748,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17411265522241592,
      "step": 26356
    },
    {
      "epoch": 0.52716,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.005098470052083333,
      "learning_rate": 0.0001,
      "loss": 3.963,
      "loss/crossentropy": 1.6551550030708313,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1649954915046692,
      "step": 26358
    },
    {
      "epoch": 0.5272,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.005791982014973958,
      "learning_rate": 0.0001,
      "loss": 3.7678,
      "loss/crossentropy": 1.9796473383903503,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1766190305352211,
      "step": 26360
    },
    {
      "epoch": 0.52724,
      "grad_norm": 1.875,
      "grad_norm_var": 0.0056955973307291664,
      "learning_rate": 0.0001,
      "loss": 3.9551,
      "loss/crossentropy": 1.9023584723472595,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18641258776187897,
      "step": 26362
    },
    {
      "epoch": 0.52728,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.005069732666015625,
      "learning_rate": 0.0001,
      "loss": 4.1641,
      "loss/crossentropy": 2.0332838892936707,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18603450059890747,
      "step": 26364
    },
    {
      "epoch": 0.52732,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.003580474853515625,
      "learning_rate": 0.0001,
      "loss": 3.8799,
      "loss/crossentropy": 2.0670453310012817,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1904071420431137,
      "step": 26366
    },
    {
      "epoch": 0.52736,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.004162343343098959,
      "learning_rate": 0.0001,
      "loss": 3.7713,
      "loss/crossentropy": 1.9684784412384033,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19187042117118835,
      "step": 26368
    },
    {
      "epoch": 0.5274,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.003979237874348959,
      "learning_rate": 0.0001,
      "loss": 4.0018,
      "loss/crossentropy": 1.9178830981254578,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1805613711476326,
      "step": 26370
    },
    {
      "epoch": 0.52744,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.005496978759765625,
      "learning_rate": 0.0001,
      "loss": 4.1603,
      "loss/crossentropy": 2.094612956047058,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2008626013994217,
      "step": 26372
    },
    {
      "epoch": 0.52748,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.005102284749348958,
      "learning_rate": 0.0001,
      "loss": 3.8223,
      "loss/crossentropy": 2.0617056488990784,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19100335985422134,
      "step": 26374
    },
    {
      "epoch": 0.52752,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.00411376953125,
      "learning_rate": 0.0001,
      "loss": 3.8758,
      "loss/crossentropy": 2.1675769090652466,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1890057846903801,
      "step": 26376
    },
    {
      "epoch": 0.52756,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.005600738525390625,
      "learning_rate": 0.0001,
      "loss": 4.0096,
      "loss/crossentropy": 2.245623469352722,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20736746490001678,
      "step": 26378
    },
    {
      "epoch": 0.5276,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0055735270182291664,
      "learning_rate": 0.0001,
      "loss": 4.1111,
      "loss/crossentropy": 2.107472836971283,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1996273696422577,
      "step": 26380
    },
    {
      "epoch": 0.52764,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.006083170572916667,
      "learning_rate": 0.0001,
      "loss": 3.7678,
      "loss/crossentropy": 1.6452732682228088,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16541734337806702,
      "step": 26382
    },
    {
      "epoch": 0.52768,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.020611317952473958,
      "learning_rate": 0.0001,
      "loss": 4.1566,
      "loss/crossentropy": 2.0002782940864563,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.27171435952186584,
      "step": 26384
    },
    {
      "epoch": 0.52772,
      "grad_norm": 2.0,
      "grad_norm_var": 0.018888092041015624,
      "learning_rate": 0.0001,
      "loss": 4.2397,
      "loss/crossentropy": 2.0969172716140747,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20492591708898544,
      "step": 26386
    },
    {
      "epoch": 0.52776,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.019510904947916668,
      "learning_rate": 0.0001,
      "loss": 3.8078,
      "loss/crossentropy": 2.238045036792755,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20098427683115005,
      "step": 26388
    },
    {
      "epoch": 0.5278,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.021022288004557292,
      "learning_rate": 0.0001,
      "loss": 4.0391,
      "loss/crossentropy": 1.9958226680755615,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19130828231573105,
      "step": 26390
    },
    {
      "epoch": 0.52784,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.020702107747395834,
      "learning_rate": 0.0001,
      "loss": 4.1161,
      "loss/crossentropy": 2.176322102546692,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20082970708608627,
      "step": 26392
    },
    {
      "epoch": 0.52788,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0212066650390625,
      "learning_rate": 0.0001,
      "loss": 4.0363,
      "loss/crossentropy": 1.9507973790168762,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17588113993406296,
      "step": 26394
    },
    {
      "epoch": 0.52792,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.022013346354166668,
      "learning_rate": 0.0001,
      "loss": 3.9508,
      "loss/crossentropy": 2.0576651096343994,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19712203741073608,
      "step": 26396
    },
    {
      "epoch": 0.52796,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.022663370768229166,
      "learning_rate": 0.0001,
      "loss": 3.9035,
      "loss/crossentropy": 2.030472457408905,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18470944464206696,
      "step": 26398
    },
    {
      "epoch": 0.528,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.011378733317057292,
      "learning_rate": 0.0001,
      "loss": 3.9675,
      "loss/crossentropy": 2.0267462730407715,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19596907496452332,
      "step": 26400
    },
    {
      "epoch": 0.52804,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.013927968343098958,
      "learning_rate": 0.0001,
      "loss": 3.7917,
      "loss/crossentropy": 2.087961494922638,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19787786155939102,
      "step": 26402
    },
    {
      "epoch": 0.52808,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.0133941650390625,
      "learning_rate": 0.0001,
      "loss": 3.7384,
      "loss/crossentropy": 1.6107558608055115,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16004415601491928,
      "step": 26404
    },
    {
      "epoch": 0.52812,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010249837239583334,
      "learning_rate": 0.0001,
      "loss": 4.0522,
      "loss/crossentropy": 1.8615078926086426,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1997840330004692,
      "step": 26406
    },
    {
      "epoch": 0.52816,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.009528605143229167,
      "learning_rate": 0.0001,
      "loss": 4.0445,
      "loss/crossentropy": 2.165738344192505,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2002716213464737,
      "step": 26408
    },
    {
      "epoch": 0.5282,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.009700520833333334,
      "learning_rate": 0.0001,
      "loss": 3.9418,
      "loss/crossentropy": 2.039876639842987,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19271554052829742,
      "step": 26410
    },
    {
      "epoch": 0.52824,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.008763631184895834,
      "learning_rate": 0.0001,
      "loss": 3.9841,
      "loss/crossentropy": 2.114060878753662,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19228192418813705,
      "step": 26412
    },
    {
      "epoch": 0.52828,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.006998697916666667,
      "learning_rate": 0.0001,
      "loss": 4.0182,
      "loss/crossentropy": 2.1510127782821655,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19021940976381302,
      "step": 26414
    },
    {
      "epoch": 0.52832,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.004986317952473959,
      "learning_rate": 0.0001,
      "loss": 3.9933,
      "loss/crossentropy": 1.9514253735542297,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19038037955760956,
      "step": 26416
    },
    {
      "epoch": 0.52836,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.0032867431640625,
      "learning_rate": 0.0001,
      "loss": 3.9041,
      "loss/crossentropy": 1.8573151230812073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17533313482999802,
      "step": 26418
    },
    {
      "epoch": 0.5284,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.004084269205729167,
      "learning_rate": 0.0001,
      "loss": 4.1181,
      "loss/crossentropy": 2.2349034547805786,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1933293491601944,
      "step": 26420
    },
    {
      "epoch": 0.52844,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0031084696451822917,
      "learning_rate": 0.0001,
      "loss": 4.127,
      "loss/crossentropy": 1.9790935516357422,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1930369809269905,
      "step": 26422
    },
    {
      "epoch": 0.52848,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.002399698893229167,
      "learning_rate": 0.0001,
      "loss": 3.9531,
      "loss/crossentropy": 1.9758057594299316,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19405999034643173,
      "step": 26424
    },
    {
      "epoch": 0.52852,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.0030637105305989583,
      "learning_rate": 0.0001,
      "loss": 4.0303,
      "loss/crossentropy": 1.9666665196418762,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19631757587194443,
      "step": 26426
    },
    {
      "epoch": 0.52856,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.0037717183430989584,
      "learning_rate": 0.0001,
      "loss": 3.8257,
      "loss/crossentropy": 1.7752289175987244,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16831018775701523,
      "step": 26428
    },
    {
      "epoch": 0.5286,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.005293782552083333,
      "learning_rate": 0.0001,
      "loss": 4.0879,
      "loss/crossentropy": 2.1496593952178955,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1875729113817215,
      "step": 26430
    },
    {
      "epoch": 0.52864,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.005926259358723958,
      "learning_rate": 0.0001,
      "loss": 4.0407,
      "loss/crossentropy": 1.9487584829330444,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18598007410764694,
      "step": 26432
    },
    {
      "epoch": 0.52868,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.009886678059895833,
      "learning_rate": 0.0001,
      "loss": 3.9064,
      "loss/crossentropy": 1.8842872977256775,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1913917362689972,
      "step": 26434
    },
    {
      "epoch": 0.52872,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.009913889567057292,
      "learning_rate": 0.0001,
      "loss": 4.1857,
      "loss/crossentropy": 2.1598324179649353,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19371075928211212,
      "step": 26436
    },
    {
      "epoch": 0.52876,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.010228474934895834,
      "learning_rate": 0.0001,
      "loss": 4.0243,
      "loss/crossentropy": 1.993024468421936,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19384337216615677,
      "step": 26438
    },
    {
      "epoch": 0.5288,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.011631011962890625,
      "learning_rate": 0.0001,
      "loss": 3.9132,
      "loss/crossentropy": 2.083270490169525,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1818510740995407,
      "step": 26440
    },
    {
      "epoch": 0.52884,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011761220296223958,
      "learning_rate": 0.0001,
      "loss": 4.0291,
      "loss/crossentropy": 2.136199116706848,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19123348593711853,
      "step": 26442
    },
    {
      "epoch": 0.52888,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.010335286458333334,
      "learning_rate": 0.0001,
      "loss": 4.2448,
      "loss/crossentropy": 2.405175805091858,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21493228524923325,
      "step": 26444
    },
    {
      "epoch": 0.52892,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.010900624593098958,
      "learning_rate": 0.0001,
      "loss": 3.9793,
      "loss/crossentropy": 2.0925610065460205,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20184864103794098,
      "step": 26446
    },
    {
      "epoch": 0.52896,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.012116495768229167,
      "learning_rate": 0.0001,
      "loss": 4.1213,
      "loss/crossentropy": 2.1476112604141235,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.206947922706604,
      "step": 26448
    },
    {
      "epoch": 0.529,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.009333292643229166,
      "learning_rate": 0.0001,
      "loss": 3.9592,
      "loss/crossentropy": 1.7512850165367126,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15776671469211578,
      "step": 26450
    },
    {
      "epoch": 0.52904,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.009505208333333333,
      "learning_rate": 0.0001,
      "loss": 4.0763,
      "loss/crossentropy": 1.766721785068512,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18328477442264557,
      "step": 26452
    },
    {
      "epoch": 0.52908,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.009178670247395833,
      "learning_rate": 0.0001,
      "loss": 4.0028,
      "loss/crossentropy": 1.9713833928108215,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19086521863937378,
      "step": 26454
    },
    {
      "epoch": 0.52912,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.009212239583333334,
      "learning_rate": 0.0001,
      "loss": 3.7999,
      "loss/crossentropy": 1.8375617861747742,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1799859181046486,
      "step": 26456
    },
    {
      "epoch": 0.52916,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.009368642171223959,
      "learning_rate": 0.0001,
      "loss": 4.0928,
      "loss/crossentropy": 2.3689894676208496,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21843338012695312,
      "step": 26458
    },
    {
      "epoch": 0.5292,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.008896636962890624,
      "learning_rate": 0.0001,
      "loss": 3.8911,
      "loss/crossentropy": 2.2842042446136475,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20215284824371338,
      "step": 26460
    },
    {
      "epoch": 0.52924,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.008405558268229167,
      "learning_rate": 0.0001,
      "loss": 3.7891,
      "loss/crossentropy": 2.0660494565963745,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18394764512777328,
      "step": 26462
    },
    {
      "epoch": 0.52928,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.009537506103515624,
      "learning_rate": 0.0001,
      "loss": 4.097,
      "loss/crossentropy": 2.101405918598175,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2051784247159958,
      "step": 26464
    },
    {
      "epoch": 0.52932,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.00943603515625,
      "learning_rate": 0.0001,
      "loss": 3.8075,
      "loss/crossentropy": 1.8293753862380981,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18773258477449417,
      "step": 26466
    },
    {
      "epoch": 0.52936,
      "grad_norm": 1.703125,
      "grad_norm_var": 0.011115519205729167,
      "learning_rate": 0.0001,
      "loss": 3.5724,
      "loss/crossentropy": 1.9455705881118774,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17760945856571198,
      "step": 26468
    },
    {
      "epoch": 0.5294,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.011508941650390625,
      "learning_rate": 0.0001,
      "loss": 3.9305,
      "loss/crossentropy": 1.750687837600708,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18026860058307648,
      "step": 26470
    },
    {
      "epoch": 0.52944,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.010375722249348959,
      "learning_rate": 0.0001,
      "loss": 4.0931,
      "loss/crossentropy": 2.189963698387146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19402040541172028,
      "step": 26472
    },
    {
      "epoch": 0.52948,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.009549713134765625,
      "learning_rate": 0.0001,
      "loss": 4.2266,
      "loss/crossentropy": 1.8729415535926819,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20855195820331573,
      "step": 26474
    },
    {
      "epoch": 0.52952,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.0097412109375,
      "learning_rate": 0.0001,
      "loss": 3.5935,
      "loss/crossentropy": 1.8071807622909546,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16968178004026413,
      "step": 26476
    },
    {
      "epoch": 0.52956,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.010257720947265625,
      "learning_rate": 0.0001,
      "loss": 3.9832,
      "loss/crossentropy": 2.241515278816223,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20128827542066574,
      "step": 26478
    },
    {
      "epoch": 0.5296,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0062978108723958336,
      "learning_rate": 0.0001,
      "loss": 3.9792,
      "loss/crossentropy": 2.2276766300201416,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.201935775578022,
      "step": 26480
    },
    {
      "epoch": 0.52964,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0061604817708333336,
      "learning_rate": 0.0001,
      "loss": 3.8511,
      "loss/crossentropy": 2.2751588821411133,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21366826444864273,
      "step": 26482
    },
    {
      "epoch": 0.52968,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.00335693359375,
      "learning_rate": 0.0001,
      "loss": 4.1496,
      "loss/crossentropy": 2.2164629697799683,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2294466495513916,
      "step": 26484
    },
    {
      "epoch": 0.52972,
      "grad_norm": 1.6328125,
      "grad_norm_var": 0.008131663004557291,
      "learning_rate": 0.0001,
      "loss": 3.7879,
      "loss/crossentropy": 1.8609917163848877,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1908324807882309,
      "step": 26486
    },
    {
      "epoch": 0.52976,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.0085601806640625,
      "learning_rate": 0.0001,
      "loss": 4.066,
      "loss/crossentropy": 2.050028145313263,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18777167797088623,
      "step": 26488
    },
    {
      "epoch": 0.5298,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.011329905192057291,
      "learning_rate": 0.0001,
      "loss": 4.2666,
      "loss/crossentropy": 2.398528814315796,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22428229451179504,
      "step": 26490
    },
    {
      "epoch": 0.52984,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.012889607747395834,
      "learning_rate": 0.0001,
      "loss": 3.8208,
      "loss/crossentropy": 2.0193370580673218,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19870522618293762,
      "step": 26492
    },
    {
      "epoch": 0.52988,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.012369791666666666,
      "learning_rate": 0.0001,
      "loss": 3.9186,
      "loss/crossentropy": 2.1982868313789368,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19849159568548203,
      "step": 26494
    },
    {
      "epoch": 0.52992,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.013405100504557291,
      "learning_rate": 0.0001,
      "loss": 3.8017,
      "loss/crossentropy": 1.8915838599205017,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16818702220916748,
      "step": 26496
    },
    {
      "epoch": 0.52996,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.012596638997395833,
      "learning_rate": 0.0001,
      "loss": 3.8179,
      "loss/crossentropy": 1.9891789555549622,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18337423354387283,
      "step": 26498
    },
    {
      "epoch": 0.53,
      "grad_norm": 2.125,
      "grad_norm_var": 0.015571848551432291,
      "learning_rate": 0.0001,
      "loss": 4.1336,
      "loss/crossentropy": 2.4187783002853394,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20852266252040863,
      "step": 26500
    },
    {
      "epoch": 0.53004,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.01514892578125,
      "learning_rate": 0.0001,
      "loss": 3.4934,
      "loss/crossentropy": 1.742977499961853,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17425427585840225,
      "step": 26502
    },
    {
      "epoch": 0.53008,
      "grad_norm": 1.875,
      "grad_norm_var": 0.017000325520833335,
      "learning_rate": 0.0001,
      "loss": 4.0615,
      "loss/crossentropy": 1.8980122208595276,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16612428426742554,
      "step": 26504
    },
    {
      "epoch": 0.53012,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.013993072509765624,
      "learning_rate": 0.0001,
      "loss": 3.9807,
      "loss/crossentropy": 1.7788900136947632,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15797217935323715,
      "step": 26506
    },
    {
      "epoch": 0.53016,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.011578114827473958,
      "learning_rate": 0.0001,
      "loss": 4.0657,
      "loss/crossentropy": 1.8558028936386108,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18111377209424973,
      "step": 26508
    },
    {
      "epoch": 0.5302,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.012886555989583333,
      "learning_rate": 0.0001,
      "loss": 4.1489,
      "loss/crossentropy": 2.2158637046813965,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20664983242750168,
      "step": 26510
    },
    {
      "epoch": 0.53024,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.012347157796223958,
      "learning_rate": 0.0001,
      "loss": 4.1,
      "loss/crossentropy": 2.159493863582611,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18167608231306076,
      "step": 26512
    },
    {
      "epoch": 0.53028,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.012717437744140626,
      "learning_rate": 0.0001,
      "loss": 3.9015,
      "loss/crossentropy": 2.328041195869446,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1777345836162567,
      "step": 26514
    },
    {
      "epoch": 0.53032,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0093902587890625,
      "learning_rate": 0.0001,
      "loss": 3.7341,
      "loss/crossentropy": 1.724794864654541,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1699744239449501,
      "step": 26516
    },
    {
      "epoch": 0.53036,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.006198883056640625,
      "learning_rate": 0.0001,
      "loss": 4.0218,
      "loss/crossentropy": 1.96088445186615,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1849687099456787,
      "step": 26518
    },
    {
      "epoch": 0.5304,
      "grad_norm": 1.875,
      "grad_norm_var": 0.0041656494140625,
      "learning_rate": 0.0001,
      "loss": 3.9326,
      "loss/crossentropy": 2.150395691394806,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20122019946575165,
      "step": 26520
    },
    {
      "epoch": 0.53044,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.004296875,
      "learning_rate": 0.0001,
      "loss": 3.9209,
      "loss/crossentropy": 2.016223907470703,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18350518494844437,
      "step": 26522
    },
    {
      "epoch": 0.53048,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0060791015625,
      "learning_rate": 0.0001,
      "loss": 4.3091,
      "loss/crossentropy": 2.334647059440613,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20854850858449936,
      "step": 26524
    },
    {
      "epoch": 0.53052,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0053059895833333336,
      "learning_rate": 0.0001,
      "loss": 3.8457,
      "loss/crossentropy": 1.928149163722992,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18799126893281937,
      "step": 26526
    },
    {
      "epoch": 0.53056,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.005224355061848958,
      "learning_rate": 0.0001,
      "loss": 3.8411,
      "loss/crossentropy": 2.297117590904236,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20247841626405716,
      "step": 26528
    },
    {
      "epoch": 0.5306,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.025925445556640624,
      "learning_rate": 0.0001,
      "loss": 3.8426,
      "loss/crossentropy": 1.6292110085487366,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1556340456008911,
      "step": 26530
    },
    {
      "epoch": 0.53064,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.023819732666015624,
      "learning_rate": 0.0001,
      "loss": 3.9924,
      "loss/crossentropy": 2.3576741218566895,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22244800627231598,
      "step": 26532
    },
    {
      "epoch": 0.53068,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.024812825520833335,
      "learning_rate": 0.0001,
      "loss": 3.8956,
      "loss/crossentropy": 2.412485718727112,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1838645040988922,
      "step": 26534
    },
    {
      "epoch": 0.53072,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.023276519775390626,
      "learning_rate": 0.0001,
      "loss": 3.9826,
      "loss/crossentropy": 2.1326744556427,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19059032201766968,
      "step": 26536
    },
    {
      "epoch": 0.53076,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.023726399739583334,
      "learning_rate": 0.0001,
      "loss": 3.8374,
      "loss/crossentropy": 1.8022708892822266,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17919087409973145,
      "step": 26538
    },
    {
      "epoch": 0.5308,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.023563385009765625,
      "learning_rate": 0.0001,
      "loss": 3.8724,
      "loss/crossentropy": 2.1001075506210327,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1887490525841713,
      "step": 26540
    },
    {
      "epoch": 0.53084,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.023860677083333334,
      "learning_rate": 0.0001,
      "loss": 3.8678,
      "loss/crossentropy": 1.9068372249603271,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1752641201019287,
      "step": 26542
    },
    {
      "epoch": 0.53088,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.023787180582682293,
      "learning_rate": 0.0001,
      "loss": 4.072,
      "loss/crossentropy": 1.8628905415534973,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17757906019687653,
      "step": 26544
    },
    {
      "epoch": 0.53092,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0041196187337239586,
      "learning_rate": 0.0001,
      "loss": 4.1044,
      "loss/crossentropy": 1.9989616870880127,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1890716478228569,
      "step": 26546
    },
    {
      "epoch": 0.53096,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.004624176025390625,
      "learning_rate": 0.0001,
      "loss": 3.8013,
      "loss/crossentropy": 2.063919186592102,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18740086257457733,
      "step": 26548
    },
    {
      "epoch": 0.531,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.004355621337890625,
      "learning_rate": 0.0001,
      "loss": 4.0148,
      "loss/crossentropy": 1.9364339709281921,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18698105216026306,
      "step": 26550
    },
    {
      "epoch": 0.53104,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.0037859598795572918,
      "learning_rate": 0.0001,
      "loss": 4.1554,
      "loss/crossentropy": 2.063125193119049,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1881653591990471,
      "step": 26552
    },
    {
      "epoch": 0.53108,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.003570302327473958,
      "learning_rate": 0.0001,
      "loss": 3.9373,
      "loss/crossentropy": 1.9567288160324097,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19221889972686768,
      "step": 26554
    },
    {
      "epoch": 0.53112,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.004367828369140625,
      "learning_rate": 0.0001,
      "loss": 4.0455,
      "loss/crossentropy": 2.177999496459961,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2061704844236374,
      "step": 26556
    },
    {
      "epoch": 0.53116,
      "grad_norm": 1.7421875,
      "grad_norm_var": 0.006052398681640625,
      "learning_rate": 0.0001,
      "loss": 3.7498,
      "loss/crossentropy": 1.956209123134613,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16483356803655624,
      "step": 26558
    },
    {
      "epoch": 0.5312,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.006082916259765625,
      "learning_rate": 0.0001,
      "loss": 3.9295,
      "loss/crossentropy": 1.778043806552887,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15840457379817963,
      "step": 26560
    },
    {
      "epoch": 0.53124,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0064849853515625,
      "learning_rate": 0.0001,
      "loss": 4.2495,
      "loss/crossentropy": 2.5685311555862427,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21522855013608932,
      "step": 26562
    },
    {
      "epoch": 0.53128,
      "grad_norm": 1.875,
      "grad_norm_var": 0.006259918212890625,
      "learning_rate": 0.0001,
      "loss": 4.0887,
      "loss/crossentropy": 2.054154694080353,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18618051707744598,
      "step": 26564
    },
    {
      "epoch": 0.53132,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.011114247639973958,
      "learning_rate": 0.0001,
      "loss": 4.1652,
      "loss/crossentropy": 2.0676876306533813,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19200553745031357,
      "step": 26566
    },
    {
      "epoch": 0.53136,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.010817209879557291,
      "learning_rate": 0.0001,
      "loss": 3.9219,
      "loss/crossentropy": 2.081417202949524,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19634481519460678,
      "step": 26568
    },
    {
      "epoch": 0.5314,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.010400136311848959,
      "learning_rate": 0.0001,
      "loss": 3.911,
      "loss/crossentropy": 2.0342991948127747,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19060657918453217,
      "step": 26570
    },
    {
      "epoch": 0.53144,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.012132771809895833,
      "learning_rate": 0.0001,
      "loss": 3.8132,
      "loss/crossentropy": 2.2454493641853333,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17674832791090012,
      "step": 26572
    },
    {
      "epoch": 0.53148,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.010927073160807292,
      "learning_rate": 0.0001,
      "loss": 3.9746,
      "loss/crossentropy": 2.1805293560028076,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19897499680519104,
      "step": 26574
    },
    {
      "epoch": 0.53152,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0107086181640625,
      "learning_rate": 0.0001,
      "loss": 4.0728,
      "loss/crossentropy": 2.074836492538452,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20241302251815796,
      "step": 26576
    },
    {
      "epoch": 0.53156,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.010380045572916666,
      "learning_rate": 0.0001,
      "loss": 4.0942,
      "loss/crossentropy": 2.341915488243103,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19814413785934448,
      "step": 26578
    },
    {
      "epoch": 0.5316,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.010994211832682291,
      "learning_rate": 0.0001,
      "loss": 3.8701,
      "loss/crossentropy": 1.690186619758606,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16645215451717377,
      "step": 26580
    },
    {
      "epoch": 0.53164,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.005614217122395833,
      "learning_rate": 0.0001,
      "loss": 3.7241,
      "loss/crossentropy": 2.0770626068115234,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19572289288043976,
      "step": 26582
    },
    {
      "epoch": 0.53168,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.006351470947265625,
      "learning_rate": 0.0001,
      "loss": 4.0342,
      "loss/crossentropy": 2.411499500274658,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21515025198459625,
      "step": 26584
    },
    {
      "epoch": 0.53172,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.0062558492024739586,
      "learning_rate": 0.0001,
      "loss": 4.1189,
      "loss/crossentropy": 2.0363634824752808,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18869787454605103,
      "step": 26586
    },
    {
      "epoch": 0.53176,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.004858144124348958,
      "learning_rate": 0.0001,
      "loss": 4.049,
      "loss/crossentropy": 2.203519582748413,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20460474491119385,
      "step": 26588
    },
    {
      "epoch": 0.5318,
      "grad_norm": 2.0,
      "grad_norm_var": 0.004491170247395833,
      "learning_rate": 0.0001,
      "loss": 3.8633,
      "loss/crossentropy": 1.9883779883384705,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21096121519804,
      "step": 26590
    },
    {
      "epoch": 0.53184,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.005332183837890625,
      "learning_rate": 0.0001,
      "loss": 4.1148,
      "loss/crossentropy": 2.209134042263031,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23714564740657806,
      "step": 26592
    },
    {
      "epoch": 0.53188,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.005125935872395833,
      "learning_rate": 0.0001,
      "loss": 3.9585,
      "loss/crossentropy": 2.3128318786621094,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20354144275188446,
      "step": 26594
    },
    {
      "epoch": 0.53192,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.006624348958333333,
      "learning_rate": 0.0001,
      "loss": 4.2532,
      "loss/crossentropy": 2.2282800674438477,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2072424292564392,
      "step": 26596
    },
    {
      "epoch": 0.53196,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0056111653645833336,
      "learning_rate": 0.0001,
      "loss": 4.0443,
      "loss/crossentropy": 2.086778998374939,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18629200011491776,
      "step": 26598
    },
    {
      "epoch": 0.532,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.005773671468098958,
      "learning_rate": 0.0001,
      "loss": 3.8206,
      "loss/crossentropy": 2.12203449010849,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1955840140581131,
      "step": 26600
    },
    {
      "epoch": 0.53204,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.007008616129557292,
      "learning_rate": 0.0001,
      "loss": 3.7792,
      "loss/crossentropy": 1.987316608428955,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18615661561489105,
      "step": 26602
    },
    {
      "epoch": 0.53208,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.007228342692057291,
      "learning_rate": 0.0001,
      "loss": 4.1107,
      "loss/crossentropy": 2.0519716143608093,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1968577429652214,
      "step": 26604
    },
    {
      "epoch": 0.53212,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.007020823160807292,
      "learning_rate": 0.0001,
      "loss": 4.1097,
      "loss/crossentropy": 2.2337719202041626,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19842206686735153,
      "step": 26606
    },
    {
      "epoch": 0.53216,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0071489969889322914,
      "learning_rate": 0.0001,
      "loss": 3.7935,
      "loss/crossentropy": 2.2014262080192566,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2037125900387764,
      "step": 26608
    },
    {
      "epoch": 0.5322,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.0175689697265625,
      "learning_rate": 0.0001,
      "loss": 3.8532,
      "loss/crossentropy": 1.7964777946472168,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17876359075307846,
      "step": 26610
    },
    {
      "epoch": 0.53224,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.0155914306640625,
      "learning_rate": 0.0001,
      "loss": 3.9521,
      "loss/crossentropy": 1.9710991978645325,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18646740168333054,
      "step": 26612
    },
    {
      "epoch": 0.53228,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.016495513916015624,
      "learning_rate": 0.0001,
      "loss": 3.8435,
      "loss/crossentropy": 2.039173901081085,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19768649339675903,
      "step": 26614
    },
    {
      "epoch": 0.53232,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0170318603515625,
      "learning_rate": 0.0001,
      "loss": 3.9751,
      "loss/crossentropy": 2.2037198543548584,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2104790136218071,
      "step": 26616
    },
    {
      "epoch": 0.53236,
      "grad_norm": 2.0,
      "grad_norm_var": 0.01741943359375,
      "learning_rate": 0.0001,
      "loss": 3.9441,
      "loss/crossentropy": 2.1893996596336365,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21086306124925613,
      "step": 26618
    },
    {
      "epoch": 0.5324,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.0174560546875,
      "learning_rate": 0.0001,
      "loss": 3.9598,
      "loss/crossentropy": 2.0324689745903015,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20213106274604797,
      "step": 26620
    },
    {
      "epoch": 0.53244,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.018244425455729168,
      "learning_rate": 0.0001,
      "loss": 3.8941,
      "loss/crossentropy": 2.191510498523712,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20318301022052765,
      "step": 26622
    },
    {
      "epoch": 0.53248,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.018741861979166666,
      "learning_rate": 0.0001,
      "loss": 4.131,
      "loss/crossentropy": 2.2092620134353638,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2033141329884529,
      "step": 26624
    },
    {
      "epoch": 0.53252,
      "grad_norm": 1.875,
      "grad_norm_var": 0.008796946207682291,
      "learning_rate": 0.0001,
      "loss": 3.8481,
      "loss/crossentropy": 2.078882932662964,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1927858516573906,
      "step": 26626
    },
    {
      "epoch": 0.53256,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.010599517822265625,
      "learning_rate": 0.0001,
      "loss": 4.3066,
      "loss/crossentropy": 2.1759738326072693,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21904072910547256,
      "step": 26628
    },
    {
      "epoch": 0.5326,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.01015625,
      "learning_rate": 0.0001,
      "loss": 3.7002,
      "loss/crossentropy": 2.0541751980781555,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17661282420158386,
      "step": 26630
    },
    {
      "epoch": 0.53264,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.010394032796223958,
      "learning_rate": 0.0001,
      "loss": 3.7789,
      "loss/crossentropy": 1.8126670122146606,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16475193202495575,
      "step": 26632
    },
    {
      "epoch": 0.53268,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.009065500895182292,
      "learning_rate": 0.0001,
      "loss": 3.9798,
      "loss/crossentropy": 1.8467394709587097,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17031905055046082,
      "step": 26634
    },
    {
      "epoch": 0.53272,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.008516184488932292,
      "learning_rate": 0.0001,
      "loss": 3.8392,
      "loss/crossentropy": 1.7537733912467957,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17307621240615845,
      "step": 26636
    },
    {
      "epoch": 0.53276,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.009273274739583334,
      "learning_rate": 0.0001,
      "loss": 4.2491,
      "loss/crossentropy": 2.2367827892303467,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21301713585853577,
      "step": 26638
    },
    {
      "epoch": 0.5328,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.009419504801432292,
      "learning_rate": 0.0001,
      "loss": 3.9927,
      "loss/crossentropy": 1.9437520503997803,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19594744592905045,
      "step": 26640
    },
    {
      "epoch": 0.53284,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0102691650390625,
      "learning_rate": 0.0001,
      "loss": 4.1308,
      "loss/crossentropy": 1.9712047576904297,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2043192759156227,
      "step": 26642
    },
    {
      "epoch": 0.53288,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.0079498291015625,
      "learning_rate": 0.0001,
      "loss": 3.916,
      "loss/crossentropy": 1.951958179473877,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2054751068353653,
      "step": 26644
    },
    {
      "epoch": 0.53292,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0067291259765625,
      "learning_rate": 0.0001,
      "loss": 3.929,
      "loss/crossentropy": 2.0811157822608948,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19759366661310196,
      "step": 26646
    },
    {
      "epoch": 0.53296,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.06748428344726562,
      "learning_rate": 0.0001,
      "loss": 4.0009,
      "loss/crossentropy": 2.0309654474258423,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1910555213689804,
      "step": 26648
    },
    {
      "epoch": 0.533,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.07044448852539062,
      "learning_rate": 0.0001,
      "loss": 3.7709,
      "loss/crossentropy": 2.003150999546051,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19752421230077744,
      "step": 26650
    },
    {
      "epoch": 0.53304,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0694091796875,
      "learning_rate": 0.0001,
      "loss": 3.9324,
      "loss/crossentropy": 1.788662612438202,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17759086191654205,
      "step": 26652
    },
    {
      "epoch": 0.53308,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.07455240885416667,
      "learning_rate": 0.0001,
      "loss": 3.835,
      "loss/crossentropy": 1.7692713737487793,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16581754386425018,
      "step": 26654
    },
    {
      "epoch": 0.53312,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.07297337849934896,
      "learning_rate": 0.0001,
      "loss": 3.798,
      "loss/crossentropy": 1.8395215272903442,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18920384347438812,
      "step": 26656
    },
    {
      "epoch": 0.53316,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.07563247680664062,
      "learning_rate": 0.0001,
      "loss": 4.0133,
      "loss/crossentropy": 2.0375067591667175,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19706164300441742,
      "step": 26658
    },
    {
      "epoch": 0.5332,
      "grad_norm": 2.0,
      "grad_norm_var": 0.076025390625,
      "learning_rate": 0.0001,
      "loss": 3.9964,
      "loss/crossentropy": 2.101609468460083,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20092131942510605,
      "step": 26660
    },
    {
      "epoch": 0.53324,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.07556864420572916,
      "learning_rate": 0.0001,
      "loss": 3.8518,
      "loss/crossentropy": 1.9724875092506409,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19354254007339478,
      "step": 26662
    },
    {
      "epoch": 0.53328,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.015000152587890624,
      "learning_rate": 0.0001,
      "loss": 4.199,
      "loss/crossentropy": 1.9491792917251587,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21028251200914383,
      "step": 26664
    },
    {
      "epoch": 0.53332,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.012133534749348958,
      "learning_rate": 0.0001,
      "loss": 3.8784,
      "loss/crossentropy": 2.0189873576164246,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20224104076623917,
      "step": 26666
    },
    {
      "epoch": 0.53336,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.01177978515625,
      "learning_rate": 0.0001,
      "loss": 3.6177,
      "loss/crossentropy": 2.0183663368225098,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20128542184829712,
      "step": 26668
    },
    {
      "epoch": 0.5334,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.01939697265625,
      "learning_rate": 0.0001,
      "loss": 4.2613,
      "loss/crossentropy": 2.323311448097229,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2202368974685669,
      "step": 26670
    },
    {
      "epoch": 0.53344,
      "grad_norm": 2.0,
      "grad_norm_var": 0.01764094034830729,
      "learning_rate": 0.0001,
      "loss": 4.2713,
      "loss/crossentropy": 2.14883291721344,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21048131585121155,
      "step": 26672
    },
    {
      "epoch": 0.53348,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.015313466389973959,
      "learning_rate": 0.0001,
      "loss": 4.2169,
      "loss/crossentropy": 2.0525609254837036,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19780127704143524,
      "step": 26674
    },
    {
      "epoch": 0.53352,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.016900380452473957,
      "learning_rate": 0.0001,
      "loss": 4.0969,
      "loss/crossentropy": 2.125749707221985,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18475912511348724,
      "step": 26676
    },
    {
      "epoch": 0.53356,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.016044108072916667,
      "learning_rate": 0.0001,
      "loss": 4.1139,
      "loss/crossentropy": 2.3169859647750854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21877627819776535,
      "step": 26678
    },
    {
      "epoch": 0.5336,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.014525349934895833,
      "learning_rate": 0.0001,
      "loss": 3.9182,
      "loss/crossentropy": 2.263182818889618,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20379970222711563,
      "step": 26680
    },
    {
      "epoch": 0.53364,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.014737701416015625,
      "learning_rate": 0.0001,
      "loss": 4.1722,
      "loss/crossentropy": 2.2744826078414917,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20668719708919525,
      "step": 26682
    },
    {
      "epoch": 0.53368,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.014872233072916666,
      "learning_rate": 0.0001,
      "loss": 3.9593,
      "loss/crossentropy": 2.2624911665916443,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2010696530342102,
      "step": 26684
    },
    {
      "epoch": 0.53372,
      "grad_norm": 1.875,
      "grad_norm_var": 0.005456288655598958,
      "learning_rate": 0.0001,
      "loss": 3.7412,
      "loss/crossentropy": 2.2623918056488037,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20084036886692047,
      "step": 26686
    },
    {
      "epoch": 0.53376,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.013960774739583333,
      "learning_rate": 0.0001,
      "loss": 4.2368,
      "loss/crossentropy": 2.116790533065796,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22413402050733566,
      "step": 26688
    },
    {
      "epoch": 0.5338,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.013881174723307292,
      "learning_rate": 0.0001,
      "loss": 3.8506,
      "loss/crossentropy": 1.9280251860618591,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19084317982196808,
      "step": 26690
    },
    {
      "epoch": 0.53384,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.013490549723307292,
      "learning_rate": 0.0001,
      "loss": 3.727,
      "loss/crossentropy": 1.8203374743461609,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16682762652635574,
      "step": 26692
    },
    {
      "epoch": 0.53388,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.013337961832682292,
      "learning_rate": 0.0001,
      "loss": 3.7433,
      "loss/crossentropy": 1.9362438321113586,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1947326734662056,
      "step": 26694
    },
    {
      "epoch": 0.53392,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.015148671468098958,
      "learning_rate": 0.0001,
      "loss": 4.1108,
      "loss/crossentropy": 1.945202112197876,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2301146537065506,
      "step": 26696
    },
    {
      "epoch": 0.53396,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.016087849934895832,
      "learning_rate": 0.0001,
      "loss": 3.8197,
      "loss/crossentropy": 1.986995279788971,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18129996210336685,
      "step": 26698
    },
    {
      "epoch": 0.534,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.01623713175455729,
      "learning_rate": 0.0001,
      "loss": 3.9488,
      "loss/crossentropy": 2.157664656639099,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2106902152299881,
      "step": 26700
    },
    {
      "epoch": 0.53404,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.014839680989583333,
      "learning_rate": 0.0001,
      "loss": 4.1374,
      "loss/crossentropy": 1.9703331589698792,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21851468831300735,
      "step": 26702
    },
    {
      "epoch": 0.53408,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.008160146077473958,
      "learning_rate": 0.0001,
      "loss": 3.9243,
      "loss/crossentropy": 2.1339242458343506,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1987123116850853,
      "step": 26704
    },
    {
      "epoch": 0.53412,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0085357666015625,
      "learning_rate": 0.0001,
      "loss": 4.0897,
      "loss/crossentropy": 1.9810134768486023,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18423764407634735,
      "step": 26706
    },
    {
      "epoch": 0.53416,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.008695475260416667,
      "learning_rate": 0.0001,
      "loss": 4.0445,
      "loss/crossentropy": 1.8842402696609497,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19414211809635162,
      "step": 26708
    },
    {
      "epoch": 0.5342,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0121490478515625,
      "learning_rate": 0.0001,
      "loss": 4.0123,
      "loss/crossentropy": 2.0668699741363525,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19577328860759735,
      "step": 26710
    },
    {
      "epoch": 0.53424,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.009276326497395833,
      "learning_rate": 0.0001,
      "loss": 3.913,
      "loss/crossentropy": 2.2768534421920776,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21711596846580505,
      "step": 26712
    },
    {
      "epoch": 0.53428,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.008872222900390626,
      "learning_rate": 0.0001,
      "loss": 4.0675,
      "loss/crossentropy": 2.1378949880599976,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20244380831718445,
      "step": 26714
    },
    {
      "epoch": 0.53432,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.009798177083333333,
      "learning_rate": 0.0001,
      "loss": 4.0259,
      "loss/crossentropy": 2.4403127431869507,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2065192013978958,
      "step": 26716
    },
    {
      "epoch": 0.53436,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.008324178059895833,
      "learning_rate": 0.0001,
      "loss": 3.8731,
      "loss/crossentropy": 1.8811705708503723,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1881936937570572,
      "step": 26718
    },
    {
      "epoch": 0.5344,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.010682932535807292,
      "learning_rate": 0.0001,
      "loss": 4.3268,
      "loss/crossentropy": 2.331605315208435,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20130542665719986,
      "step": 26720
    },
    {
      "epoch": 0.53444,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.011763509114583333,
      "learning_rate": 0.0001,
      "loss": 3.6066,
      "loss/crossentropy": 1.521777868270874,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1611887440085411,
      "step": 26722
    },
    {
      "epoch": 0.53448,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.011905924479166666,
      "learning_rate": 0.0001,
      "loss": 4.0351,
      "loss/crossentropy": 2.0702260732650757,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21350041031837463,
      "step": 26724
    },
    {
      "epoch": 0.53452,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008760579427083333,
      "learning_rate": 0.0001,
      "loss": 4.2144,
      "loss/crossentropy": 2.2656288146972656,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1988365724682808,
      "step": 26726
    },
    {
      "epoch": 0.53456,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.008821614583333333,
      "learning_rate": 0.0001,
      "loss": 3.9917,
      "loss/crossentropy": 1.8514603972434998,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19230767339468002,
      "step": 26728
    },
    {
      "epoch": 0.5346,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0082672119140625,
      "learning_rate": 0.0001,
      "loss": 3.6906,
      "loss/crossentropy": 1.682263970375061,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19245991110801697,
      "step": 26730
    },
    {
      "epoch": 0.53464,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.008778635660807292,
      "learning_rate": 0.0001,
      "loss": 3.9556,
      "loss/crossentropy": 2.0426923036575317,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19547611474990845,
      "step": 26732
    },
    {
      "epoch": 0.53468,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.007826487223307291,
      "learning_rate": 0.0001,
      "loss": 3.8705,
      "loss/crossentropy": 2.0580697059631348,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17934934794902802,
      "step": 26734
    },
    {
      "epoch": 0.53472,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0060791015625,
      "learning_rate": 0.0001,
      "loss": 4.0613,
      "loss/crossentropy": 2.2542420625686646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21669819951057434,
      "step": 26736
    },
    {
      "epoch": 0.53476,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.004740142822265625,
      "learning_rate": 0.0001,
      "loss": 3.8354,
      "loss/crossentropy": 2.0720590949058533,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18977323174476624,
      "step": 26738
    },
    {
      "epoch": 0.5348,
      "grad_norm": 1.875,
      "grad_norm_var": 0.007134755452473958,
      "learning_rate": 0.0001,
      "loss": 3.7576,
      "loss/crossentropy": 2.1534887552261353,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1833745241165161,
      "step": 26740
    },
    {
      "epoch": 0.53484,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008381907145182292,
      "learning_rate": 0.0001,
      "loss": 3.9765,
      "loss/crossentropy": 1.4169192910194397,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.14363554865121841,
      "step": 26742
    },
    {
      "epoch": 0.53488,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.007340240478515625,
      "learning_rate": 0.0001,
      "loss": 4.046,
      "loss/crossentropy": 2.149993062019348,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20160876959562302,
      "step": 26744
    },
    {
      "epoch": 0.53492,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.007433827718098958,
      "learning_rate": 0.0001,
      "loss": 4.0038,
      "loss/crossentropy": 2.032483458518982,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18361316621303558,
      "step": 26746
    },
    {
      "epoch": 0.53496,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.007979329427083333,
      "learning_rate": 0.0001,
      "loss": 4.0003,
      "loss/crossentropy": 2.253453016281128,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20721124857664108,
      "step": 26748
    },
    {
      "epoch": 0.535,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.009528605143229167,
      "learning_rate": 0.0001,
      "loss": 3.983,
      "loss/crossentropy": 1.8833884000778198,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19324089586734772,
      "step": 26750
    },
    {
      "epoch": 0.53504,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.011427561442057291,
      "learning_rate": 0.0001,
      "loss": 4.1681,
      "loss/crossentropy": 2.146417558193207,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19708352535963058,
      "step": 26752
    },
    {
      "epoch": 0.53508,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.013248443603515625,
      "learning_rate": 0.0001,
      "loss": 4.1508,
      "loss/crossentropy": 1.9810682535171509,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1777956709265709,
      "step": 26754
    },
    {
      "epoch": 0.53512,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.010426584879557292,
      "learning_rate": 0.0001,
      "loss": 3.9092,
      "loss/crossentropy": 2.0470046401023865,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1866726651787758,
      "step": 26756
    },
    {
      "epoch": 0.53516,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.009287261962890625,
      "learning_rate": 0.0001,
      "loss": 3.9476,
      "loss/crossentropy": 2.0760093927383423,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18907316029071808,
      "step": 26758
    },
    {
      "epoch": 0.5352,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011110178629557292,
      "learning_rate": 0.0001,
      "loss": 3.8118,
      "loss/crossentropy": 1.7074882984161377,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16364683210849762,
      "step": 26760
    },
    {
      "epoch": 0.53524,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.011614735921223958,
      "learning_rate": 0.0001,
      "loss": 4.196,
      "loss/crossentropy": 1.925625503063202,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18412255495786667,
      "step": 26762
    },
    {
      "epoch": 0.53528,
      "grad_norm": 2.0,
      "grad_norm_var": 0.012962849934895833,
      "learning_rate": 0.0001,
      "loss": 3.9354,
      "loss/crossentropy": 2.03251188993454,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1895642802119255,
      "step": 26764
    },
    {
      "epoch": 0.53532,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.013012440999348958,
      "learning_rate": 0.0001,
      "loss": 3.9456,
      "loss/crossentropy": 1.9689412117004395,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17626702040433884,
      "step": 26766
    },
    {
      "epoch": 0.53536,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.011864980061848959,
      "learning_rate": 0.0001,
      "loss": 3.9408,
      "loss/crossentropy": 1.7644684314727783,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18325244635343552,
      "step": 26768
    },
    {
      "epoch": 0.5354,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.009544881184895833,
      "learning_rate": 0.0001,
      "loss": 3.8692,
      "loss/crossentropy": 1.9543325304985046,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18265419453382492,
      "step": 26770
    },
    {
      "epoch": 0.53544,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.012703196207682291,
      "learning_rate": 0.0001,
      "loss": 4.2303,
      "loss/crossentropy": 2.107349157333374,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18807468563318253,
      "step": 26772
    },
    {
      "epoch": 0.53548,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.022900390625,
      "learning_rate": 0.0001,
      "loss": 4.1627,
      "loss/crossentropy": 2.0298044085502625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1947350949048996,
      "step": 26774
    },
    {
      "epoch": 0.53552,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.020491536458333334,
      "learning_rate": 0.0001,
      "loss": 4.06,
      "loss/crossentropy": 2.1605560183525085,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2067297026515007,
      "step": 26776
    },
    {
      "epoch": 0.53556,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.021762847900390625,
      "learning_rate": 0.0001,
      "loss": 4.2464,
      "loss/crossentropy": 1.9689620733261108,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24544911086559296,
      "step": 26778
    },
    {
      "epoch": 0.5356,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.018723297119140624,
      "learning_rate": 0.0001,
      "loss": 3.93,
      "loss/crossentropy": 1.7339588403701782,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21119988709688187,
      "step": 26780
    },
    {
      "epoch": 0.53564,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.018260701497395834,
      "learning_rate": 0.0001,
      "loss": 3.8853,
      "loss/crossentropy": 2.3095227479934692,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19727592915296555,
      "step": 26782
    },
    {
      "epoch": 0.53568,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.017661285400390626,
      "learning_rate": 0.0001,
      "loss": 3.9575,
      "loss/crossentropy": 2.1668676137924194,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18963289260864258,
      "step": 26784
    },
    {
      "epoch": 0.53572,
      "grad_norm": 2.0,
      "grad_norm_var": 0.018683878580729167,
      "learning_rate": 0.0001,
      "loss": 3.7399,
      "loss/crossentropy": 1.7347453236579895,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1699044108390808,
      "step": 26786
    },
    {
      "epoch": 0.53576,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.0186431884765625,
      "learning_rate": 0.0001,
      "loss": 3.9689,
      "loss/crossentropy": 2.144263744354248,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21406567096710205,
      "step": 26788
    },
    {
      "epoch": 0.5358,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.008202870686848959,
      "learning_rate": 0.0001,
      "loss": 4.0654,
      "loss/crossentropy": 2.2909047603607178,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19608308374881744,
      "step": 26790
    },
    {
      "epoch": 0.53584,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0089752197265625,
      "learning_rate": 0.0001,
      "loss": 3.748,
      "loss/crossentropy": 2.0192511677742004,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19014007598161697,
      "step": 26792
    },
    {
      "epoch": 0.53588,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.005879720052083333,
      "learning_rate": 0.0001,
      "loss": 4.1532,
      "loss/crossentropy": 1.848621129989624,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16638639569282532,
      "step": 26794
    },
    {
      "epoch": 0.53592,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.008128865559895834,
      "learning_rate": 0.0001,
      "loss": 3.8812,
      "loss/crossentropy": 1.8177622556686401,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17412111163139343,
      "step": 26796
    },
    {
      "epoch": 0.53596,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.009490712483723959,
      "learning_rate": 0.0001,
      "loss": 3.8414,
      "loss/crossentropy": 1.8192716240882874,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1820041686296463,
      "step": 26798
    },
    {
      "epoch": 0.536,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.037534332275390624,
      "learning_rate": 0.0001,
      "loss": 3.9625,
      "loss/crossentropy": 2.037019371986389,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19533763080835342,
      "step": 26800
    },
    {
      "epoch": 0.53604,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.03748779296875,
      "learning_rate": 0.0001,
      "loss": 4.1231,
      "loss/crossentropy": 2.3284034729003906,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20415876060724258,
      "step": 26802
    },
    {
      "epoch": 0.53608,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.04108861287434896,
      "learning_rate": 0.0001,
      "loss": 4.5056,
      "loss/crossentropy": 1.9130100011825562,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19624194502830505,
      "step": 26804
    },
    {
      "epoch": 0.53612,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.04149754842122396,
      "learning_rate": 0.0001,
      "loss": 3.9375,
      "loss/crossentropy": 1.9683635234832764,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2047107145190239,
      "step": 26806
    },
    {
      "epoch": 0.53616,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0410552978515625,
      "learning_rate": 0.0001,
      "loss": 4.0838,
      "loss/crossentropy": 2.3120559453964233,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2024022340774536,
      "step": 26808
    },
    {
      "epoch": 0.5362,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0410552978515625,
      "learning_rate": 0.0001,
      "loss": 3.9571,
      "loss/crossentropy": 2.0975415110588074,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19205840677022934,
      "step": 26810
    },
    {
      "epoch": 0.53624,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.03737564086914062,
      "learning_rate": 0.0001,
      "loss": 3.9933,
      "loss/crossentropy": 1.995516061782837,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18409030139446259,
      "step": 26812
    },
    {
      "epoch": 0.53628,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.03712539672851563,
      "learning_rate": 0.0001,
      "loss": 3.5401,
      "loss/crossentropy": 1.9188540577888489,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17965808510780334,
      "step": 26814
    },
    {
      "epoch": 0.53632,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.03156636555989583,
      "learning_rate": 0.0001,
      "loss": 4.127,
      "loss/crossentropy": 2.1765600442886353,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20042569190263748,
      "step": 26816
    },
    {
      "epoch": 0.53636,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.031243642171223957,
      "learning_rate": 0.0001,
      "loss": 4.1306,
      "loss/crossentropy": 2.2123221158981323,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19921905547380447,
      "step": 26818
    },
    {
      "epoch": 0.5364,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.027950032552083334,
      "learning_rate": 0.0001,
      "loss": 4.091,
      "loss/crossentropy": 2.130329966545105,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20274868607521057,
      "step": 26820
    },
    {
      "epoch": 0.53644,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.02783788045247396,
      "learning_rate": 0.0001,
      "loss": 3.9039,
      "loss/crossentropy": 1.8405003547668457,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1754842922091484,
      "step": 26822
    },
    {
      "epoch": 0.53648,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.027611287434895833,
      "learning_rate": 0.0001,
      "loss": 4.1288,
      "loss/crossentropy": 2.1880215406417847,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1983429118990898,
      "step": 26824
    },
    {
      "epoch": 0.53652,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.027486165364583332,
      "learning_rate": 0.0001,
      "loss": 4.0219,
      "loss/crossentropy": 2.1088255643844604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2141232043504715,
      "step": 26826
    },
    {
      "epoch": 0.53656,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.02906494140625,
      "learning_rate": 0.0001,
      "loss": 3.8428,
      "loss/crossentropy": 2.2226059436798096,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20068518072366714,
      "step": 26828
    },
    {
      "epoch": 0.5366,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.02655614217122396,
      "learning_rate": 0.0001,
      "loss": 3.9276,
      "loss/crossentropy": 2.0935966968536377,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20954880118370056,
      "step": 26830
    },
    {
      "epoch": 0.53664,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.0034238179524739582,
      "learning_rate": 0.0001,
      "loss": 4.1073,
      "loss/crossentropy": 1.9899848699569702,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19131190329790115,
      "step": 26832
    },
    {
      "epoch": 0.53668,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.002854156494140625,
      "learning_rate": 0.0001,
      "loss": 4.0719,
      "loss/crossentropy": 1.9950811862945557,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20007413625717163,
      "step": 26834
    },
    {
      "epoch": 0.53672,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.003763580322265625,
      "learning_rate": 0.0001,
      "loss": 3.6452,
      "loss/crossentropy": 2.1717607975006104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19856677949428558,
      "step": 26836
    },
    {
      "epoch": 0.53676,
      "grad_norm": 2.0,
      "grad_norm_var": 0.003780110677083333,
      "learning_rate": 0.0001,
      "loss": 4.1077,
      "loss/crossentropy": 2.0720032453536987,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17657846212387085,
      "step": 26838
    },
    {
      "epoch": 0.5368,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.005541737874348958,
      "learning_rate": 0.0001,
      "loss": 3.776,
      "loss/crossentropy": 2.0533303022384644,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19667675346136093,
      "step": 26840
    },
    {
      "epoch": 0.53684,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.005844879150390625,
      "learning_rate": 0.0001,
      "loss": 3.9664,
      "loss/crossentropy": 1.8576315641403198,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17336532473564148,
      "step": 26842
    },
    {
      "epoch": 0.53688,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0067179361979166664,
      "learning_rate": 0.0001,
      "loss": 4.1348,
      "loss/crossentropy": 2.250246524810791,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19317224621772766,
      "step": 26844
    },
    {
      "epoch": 0.53692,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.006239573160807292,
      "learning_rate": 0.0001,
      "loss": 4.2467,
      "loss/crossentropy": 2.241440773010254,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20358380675315857,
      "step": 26846
    },
    {
      "epoch": 0.53696,
      "grad_norm": 2.125,
      "grad_norm_var": 0.009332021077473959,
      "learning_rate": 0.0001,
      "loss": 4.1142,
      "loss/crossentropy": 1.8086896538734436,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17357632517814636,
      "step": 26848
    },
    {
      "epoch": 0.537,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.010695139567057291,
      "learning_rate": 0.0001,
      "loss": 4.3621,
      "loss/crossentropy": 2.396283984184265,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21695053577423096,
      "step": 26850
    },
    {
      "epoch": 0.53704,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.010477447509765625,
      "learning_rate": 0.0001,
      "loss": 4.2781,
      "loss/crossentropy": 2.1522679328918457,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21720080822706223,
      "step": 26852
    },
    {
      "epoch": 0.53708,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.011763254801432291,
      "learning_rate": 0.0001,
      "loss": 3.8039,
      "loss/crossentropy": 1.5082102417945862,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15677820146083832,
      "step": 26854
    },
    {
      "epoch": 0.53712,
      "grad_norm": 1.875,
      "grad_norm_var": 0.014855702718098959,
      "learning_rate": 0.0001,
      "loss": 4.1837,
      "loss/crossentropy": 2.1900625228881836,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19950998574495316,
      "step": 26856
    },
    {
      "epoch": 0.53716,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.015409088134765625,
      "learning_rate": 0.0001,
      "loss": 4.2357,
      "loss/crossentropy": 1.978192925453186,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19752255082130432,
      "step": 26858
    },
    {
      "epoch": 0.5372,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.015962727864583335,
      "learning_rate": 0.0001,
      "loss": 4.0597,
      "loss/crossentropy": 1.9195253252983093,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2122282087802887,
      "step": 26860
    },
    {
      "epoch": 0.53724,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.015913899739583334,
      "learning_rate": 0.0001,
      "loss": 4.0172,
      "loss/crossentropy": 1.7090142369270325,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1740838885307312,
      "step": 26862
    },
    {
      "epoch": 0.53728,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.029788970947265625,
      "learning_rate": 0.0001,
      "loss": 3.9981,
      "loss/crossentropy": 1.9377044439315796,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18080267310142517,
      "step": 26864
    },
    {
      "epoch": 0.53732,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.029499308268229166,
      "learning_rate": 0.0001,
      "loss": 3.7442,
      "loss/crossentropy": 1.635703206062317,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16993890702724457,
      "step": 26866
    },
    {
      "epoch": 0.53736,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.03104426066080729,
      "learning_rate": 0.0001,
      "loss": 3.6425,
      "loss/crossentropy": 2.10710072517395,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2069242298603058,
      "step": 26868
    },
    {
      "epoch": 0.5374,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.030543772379557292,
      "learning_rate": 0.0001,
      "loss": 3.9258,
      "loss/crossentropy": 1.9034352898597717,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17317350208759308,
      "step": 26870
    },
    {
      "epoch": 0.53744,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.027032216389973957,
      "learning_rate": 0.0001,
      "loss": 3.7471,
      "loss/crossentropy": 1.97059965133667,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18057716637849808,
      "step": 26872
    },
    {
      "epoch": 0.53748,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.02578913370768229,
      "learning_rate": 0.0001,
      "loss": 3.92,
      "loss/crossentropy": 2.11410790681839,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19710883498191833,
      "step": 26874
    },
    {
      "epoch": 0.53752,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.026652018229166668,
      "learning_rate": 0.0001,
      "loss": 3.7229,
      "loss/crossentropy": 2.1530507802963257,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18232395499944687,
      "step": 26876
    },
    {
      "epoch": 0.53756,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.027286783854166666,
      "learning_rate": 0.0001,
      "loss": 3.9919,
      "loss/crossentropy": 1.9864270687103271,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18189801275730133,
      "step": 26878
    },
    {
      "epoch": 0.5376,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.004610188802083333,
      "learning_rate": 0.0001,
      "loss": 3.8101,
      "loss/crossentropy": 1.7933497428894043,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16698282212018967,
      "step": 26880
    },
    {
      "epoch": 0.53764,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.005102284749348958,
      "learning_rate": 0.0001,
      "loss": 4.096,
      "loss/crossentropy": 2.1818475127220154,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2031894475221634,
      "step": 26882
    },
    {
      "epoch": 0.53768,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.004980214436848958,
      "learning_rate": 0.0001,
      "loss": 3.8372,
      "loss/crossentropy": 1.7670655250549316,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19976353645324707,
      "step": 26884
    },
    {
      "epoch": 0.53772,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.007252756754557292,
      "learning_rate": 0.0001,
      "loss": 3.7664,
      "loss/crossentropy": 1.9625222086906433,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1887579709291458,
      "step": 26886
    },
    {
      "epoch": 0.53776,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.006082916259765625,
      "learning_rate": 0.0001,
      "loss": 4.065,
      "loss/crossentropy": 1.989456593990326,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18352140486240387,
      "step": 26888
    },
    {
      "epoch": 0.5378,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0054595947265625,
      "learning_rate": 0.0001,
      "loss": 3.8264,
      "loss/crossentropy": 1.9268362522125244,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18406879901885986,
      "step": 26890
    },
    {
      "epoch": 0.53784,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.005515289306640625,
      "learning_rate": 0.0001,
      "loss": 4.0215,
      "loss/crossentropy": 2.3704354763031006,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20472833514213562,
      "step": 26892
    },
    {
      "epoch": 0.53788,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.006831614176432291,
      "learning_rate": 0.0001,
      "loss": 4.0136,
      "loss/crossentropy": 2.2468762397766113,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20067602396011353,
      "step": 26894
    },
    {
      "epoch": 0.53792,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.007456207275390625,
      "learning_rate": 0.0001,
      "loss": 4.0154,
      "loss/crossentropy": 1.9968597888946533,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19497767835855484,
      "step": 26896
    },
    {
      "epoch": 0.53796,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008861287434895834,
      "learning_rate": 0.0001,
      "loss": 4.3752,
      "loss/crossentropy": 2.3659461736679077,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2061542198061943,
      "step": 26898
    },
    {
      "epoch": 0.538,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.00928955078125,
      "learning_rate": 0.0001,
      "loss": 3.7333,
      "loss/crossentropy": 2.1831624507904053,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18195770680904388,
      "step": 26900
    },
    {
      "epoch": 0.53804,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.007419586181640625,
      "learning_rate": 0.0001,
      "loss": 3.9325,
      "loss/crossentropy": 2.011473059654236,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17340296506881714,
      "step": 26902
    },
    {
      "epoch": 0.53808,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.009751129150390624,
      "learning_rate": 0.0001,
      "loss": 4.2062,
      "loss/crossentropy": 2.0382936000823975,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20788908749818802,
      "step": 26904
    },
    {
      "epoch": 0.53812,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011530558268229166,
      "learning_rate": 0.0001,
      "loss": 3.7917,
      "loss/crossentropy": 1.9090477228164673,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1654892861843109,
      "step": 26906
    },
    {
      "epoch": 0.53816,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.013850657145182292,
      "learning_rate": 0.0001,
      "loss": 4.1095,
      "loss/crossentropy": 2.1782519817352295,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20165212452411652,
      "step": 26908
    },
    {
      "epoch": 0.5382,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0138336181640625,
      "learning_rate": 0.0001,
      "loss": 4.0269,
      "loss/crossentropy": 2.3456512689590454,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19297856092453003,
      "step": 26910
    },
    {
      "epoch": 0.53824,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.013108062744140624,
      "learning_rate": 0.0001,
      "loss": 4.1148,
      "loss/crossentropy": 2.1863032579421997,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19647317379713058,
      "step": 26912
    },
    {
      "epoch": 0.53828,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.013187408447265625,
      "learning_rate": 0.0001,
      "loss": 3.8884,
      "loss/crossentropy": 1.824396550655365,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18155476450920105,
      "step": 26914
    },
    {
      "epoch": 0.53832,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.013630930582682292,
      "learning_rate": 0.0001,
      "loss": 3.9332,
      "loss/crossentropy": 1.913030207157135,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16873834282159805,
      "step": 26916
    },
    {
      "epoch": 0.53836,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.012780507405598959,
      "learning_rate": 0.0001,
      "loss": 4.0736,
      "loss/crossentropy": 1.791024386882782,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17709940671920776,
      "step": 26918
    },
    {
      "epoch": 0.5384,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.010857899983723959,
      "learning_rate": 0.0001,
      "loss": 4.0551,
      "loss/crossentropy": 1.9327716827392578,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17906151711940765,
      "step": 26920
    },
    {
      "epoch": 0.53844,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.010290273030598958,
      "learning_rate": 0.0001,
      "loss": 3.9922,
      "loss/crossentropy": 2.1072241067886353,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19492276012897491,
      "step": 26922
    },
    {
      "epoch": 0.53848,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.009993235270182291,
      "learning_rate": 0.0001,
      "loss": 4.039,
      "loss/crossentropy": 2.024174690246582,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2018662467598915,
      "step": 26924
    },
    {
      "epoch": 0.53852,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.00892333984375,
      "learning_rate": 0.0001,
      "loss": 3.988,
      "loss/crossentropy": 1.9384547472000122,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1663040593266487,
      "step": 26926
    },
    {
      "epoch": 0.53856,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.007692209879557292,
      "learning_rate": 0.0001,
      "loss": 3.7386,
      "loss/crossentropy": 2.0045499205589294,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18697284162044525,
      "step": 26928
    },
    {
      "epoch": 0.5386,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.008038075764973958,
      "learning_rate": 0.0001,
      "loss": 4.1255,
      "loss/crossentropy": 2.3768097162246704,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20385072380304337,
      "step": 26930
    },
    {
      "epoch": 0.53864,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.008097330729166666,
      "learning_rate": 0.0001,
      "loss": 3.6845,
      "loss/crossentropy": 1.8940687775611877,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1758279949426651,
      "step": 26932
    },
    {
      "epoch": 0.53868,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.008503214518229166,
      "learning_rate": 0.0001,
      "loss": 4.1272,
      "loss/crossentropy": 2.258841872215271,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18384167551994324,
      "step": 26934
    },
    {
      "epoch": 0.53872,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.010115305582682291,
      "learning_rate": 0.0001,
      "loss": 4.1708,
      "loss/crossentropy": 2.3876583576202393,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20442500710487366,
      "step": 26936
    },
    {
      "epoch": 0.53876,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.010106404622395834,
      "learning_rate": 0.0001,
      "loss": 3.975,
      "loss/crossentropy": 2.3199750185012817,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1975177749991417,
      "step": 26938
    },
    {
      "epoch": 0.5388,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.006365712483723958,
      "learning_rate": 0.0001,
      "loss": 4.1408,
      "loss/crossentropy": 2.2277639508247375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1962454915046692,
      "step": 26940
    },
    {
      "epoch": 0.53884,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.009844716389973958,
      "learning_rate": 0.0001,
      "loss": 4.1696,
      "loss/crossentropy": 2.2041778564453125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19852803647518158,
      "step": 26942
    },
    {
      "epoch": 0.53888,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.009924062093098958,
      "learning_rate": 0.0001,
      "loss": 3.844,
      "loss/crossentropy": 1.6400386095046997,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15487553179264069,
      "step": 26944
    },
    {
      "epoch": 0.53892,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.012059529622395834,
      "learning_rate": 0.0001,
      "loss": 4.0689,
      "loss/crossentropy": 2.4118189811706543,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22375066578388214,
      "step": 26946
    },
    {
      "epoch": 0.53896,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.014253489176432292,
      "learning_rate": 0.0001,
      "loss": 3.932,
      "loss/crossentropy": 2.238221287727356,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18950769305229187,
      "step": 26948
    },
    {
      "epoch": 0.539,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.0158935546875,
      "learning_rate": 0.0001,
      "loss": 4.0809,
      "loss/crossentropy": 2.2348607182502747,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1920975297689438,
      "step": 26950
    },
    {
      "epoch": 0.53904,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.015954335530598957,
      "learning_rate": 0.0001,
      "loss": 4.0146,
      "loss/crossentropy": 2.2838883996009827,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2128254398703575,
      "step": 26952
    },
    {
      "epoch": 0.53908,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0150787353515625,
      "learning_rate": 0.0001,
      "loss": 3.7944,
      "loss/crossentropy": 2.183197498321533,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2030157521367073,
      "step": 26954
    },
    {
      "epoch": 0.53912,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.014579010009765626,
      "learning_rate": 0.0001,
      "loss": 4.1249,
      "loss/crossentropy": 2.132546067237854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2098417431116104,
      "step": 26956
    },
    {
      "epoch": 0.53916,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.011839803059895833,
      "learning_rate": 0.0001,
      "loss": 3.9566,
      "loss/crossentropy": 1.867222011089325,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18416466563940048,
      "step": 26958
    },
    {
      "epoch": 0.5392,
      "grad_norm": 1.875,
      "grad_norm_var": 0.013710276285807291,
      "learning_rate": 0.0001,
      "loss": 4.3045,
      "loss/crossentropy": 2.0456870198249817,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19690076261758804,
      "step": 26960
    },
    {
      "epoch": 0.53924,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.017731730143229166,
      "learning_rate": 0.0001,
      "loss": 4.207,
      "loss/crossentropy": 2.387152314186096,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2672813981771469,
      "step": 26962
    },
    {
      "epoch": 0.53928,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.014679972330729167,
      "learning_rate": 0.0001,
      "loss": 3.8385,
      "loss/crossentropy": 1.7455761432647705,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.158515103161335,
      "step": 26964
    },
    {
      "epoch": 0.53932,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.013643391927083333,
      "learning_rate": 0.0001,
      "loss": 4.0564,
      "loss/crossentropy": 2.0408239364624023,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18679992854595184,
      "step": 26966
    },
    {
      "epoch": 0.53936,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.012555948893229167,
      "learning_rate": 0.0001,
      "loss": 4.1616,
      "loss/crossentropy": 2.1810312271118164,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18764056265354156,
      "step": 26968
    },
    {
      "epoch": 0.5394,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.012015787760416667,
      "learning_rate": 0.0001,
      "loss": 3.9197,
      "loss/crossentropy": 2.152927279472351,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18063102662563324,
      "step": 26970
    },
    {
      "epoch": 0.53944,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.01220703125,
      "learning_rate": 0.0001,
      "loss": 3.912,
      "loss/crossentropy": 2.006937623023987,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17757008224725723,
      "step": 26972
    },
    {
      "epoch": 0.53948,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.01268310546875,
      "learning_rate": 0.0001,
      "loss": 4.0045,
      "loss/crossentropy": 1.7032560110092163,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16759414970874786,
      "step": 26974
    },
    {
      "epoch": 0.53952,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.014506022135416666,
      "learning_rate": 0.0001,
      "loss": 4.0903,
      "loss/crossentropy": 2.0685030817985535,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18894075602293015,
      "step": 26976
    },
    {
      "epoch": 0.53956,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.010015614827473958,
      "learning_rate": 0.0001,
      "loss": 3.5637,
      "loss/crossentropy": 1.7113747000694275,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16482854634523392,
      "step": 26978
    },
    {
      "epoch": 0.5396,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.009208170572916667,
      "learning_rate": 0.0001,
      "loss": 3.8915,
      "loss/crossentropy": 1.9814255237579346,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19485389441251755,
      "step": 26980
    },
    {
      "epoch": 0.53964,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.009466298421223958,
      "learning_rate": 0.0001,
      "loss": 3.7662,
      "loss/crossentropy": 1.825324296951294,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.172153040766716,
      "step": 26982
    },
    {
      "epoch": 0.53968,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.008837636311848958,
      "learning_rate": 0.0001,
      "loss": 4.0153,
      "loss/crossentropy": 2.093530297279358,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1956939622759819,
      "step": 26984
    },
    {
      "epoch": 0.53972,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.010716756184895834,
      "learning_rate": 0.0001,
      "loss": 4.3494,
      "loss/crossentropy": 2.333137631416321,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21523389220237732,
      "step": 26986
    },
    {
      "epoch": 0.53976,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.010814412434895834,
      "learning_rate": 0.0001,
      "loss": 3.8208,
      "loss/crossentropy": 1.6317270994186401,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16613581776618958,
      "step": 26988
    },
    {
      "epoch": 0.5398,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.010560862223307292,
      "learning_rate": 0.0001,
      "loss": 3.7877,
      "loss/crossentropy": 2.286810517311096,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21348290145397186,
      "step": 26990
    },
    {
      "epoch": 0.53984,
      "grad_norm": 1.6875,
      "grad_norm_var": 0.008597819010416667,
      "learning_rate": 0.0001,
      "loss": 3.8261,
      "loss/crossentropy": 2.2280890941619873,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20495467633008957,
      "step": 26992
    },
    {
      "epoch": 0.53988,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.0084625244140625,
      "learning_rate": 0.0001,
      "loss": 3.7743,
      "loss/crossentropy": 1.4591253399848938,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1521049663424492,
      "step": 26994
    },
    {
      "epoch": 0.53992,
      "grad_norm": 1.875,
      "grad_norm_var": 0.0085205078125,
      "learning_rate": 0.0001,
      "loss": 4.0559,
      "loss/crossentropy": 2.236023187637329,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2155672013759613,
      "step": 26996
    },
    {
      "epoch": 0.53996,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0094390869140625,
      "learning_rate": 0.0001,
      "loss": 4.1202,
      "loss/crossentropy": 2.0742298364639282,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17755182832479477,
      "step": 26998
    },
    {
      "epoch": 0.54,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.009562174479166666,
      "learning_rate": 0.0001,
      "loss": 3.8773,
      "loss/crossentropy": 1.9153380990028381,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1838880032300949,
      "step": 27000
    },
    {
      "epoch": 0.54004,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.009349568684895834,
      "learning_rate": 0.0001,
      "loss": 4.062,
      "loss/crossentropy": 2.2510672211647034,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2028658166527748,
      "step": 27002
    },
    {
      "epoch": 0.54008,
      "grad_norm": 1.75,
      "grad_norm_var": 0.011016591389973959,
      "learning_rate": 0.0001,
      "loss": 3.5163,
      "loss/crossentropy": 1.9110174179077148,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18953163921833038,
      "step": 27004
    },
    {
      "epoch": 0.54012,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.012604777018229167,
      "learning_rate": 0.0001,
      "loss": 3.9874,
      "loss/crossentropy": 1.8102795481681824,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16844668984413147,
      "step": 27006
    },
    {
      "epoch": 0.54016,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.009797159830729167,
      "learning_rate": 0.0001,
      "loss": 4.1755,
      "loss/crossentropy": 1.7649898529052734,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18175919353961945,
      "step": 27008
    },
    {
      "epoch": 0.5402,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.011104075113932292,
      "learning_rate": 0.0001,
      "loss": 4.0751,
      "loss/crossentropy": 2.199475884437561,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19509385526180267,
      "step": 27010
    },
    {
      "epoch": 0.54024,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.010935211181640625,
      "learning_rate": 0.0001,
      "loss": 4.1241,
      "loss/crossentropy": 2.313786506652832,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2032681107521057,
      "step": 27012
    },
    {
      "epoch": 0.54028,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.009992472330729167,
      "learning_rate": 0.0001,
      "loss": 4.1953,
      "loss/crossentropy": 2.040158271789551,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20921558886766434,
      "step": 27014
    },
    {
      "epoch": 0.54032,
      "grad_norm": 1.875,
      "grad_norm_var": 0.010871378580729167,
      "learning_rate": 0.0001,
      "loss": 3.7967,
      "loss/crossentropy": 2.0283621549606323,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19085510820150375,
      "step": 27016
    },
    {
      "epoch": 0.54036,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.015941365559895834,
      "learning_rate": 0.0001,
      "loss": 3.9703,
      "loss/crossentropy": 2.1031923294067383,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19315888732671738,
      "step": 27018
    },
    {
      "epoch": 0.5404,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.013142903645833334,
      "learning_rate": 0.0001,
      "loss": 3.7665,
      "loss/crossentropy": 1.8264936804771423,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17999880015850067,
      "step": 27020
    },
    {
      "epoch": 0.54044,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.011531321207682292,
      "learning_rate": 0.0001,
      "loss": 3.7928,
      "loss/crossentropy": 2.127643585205078,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20216452330350876,
      "step": 27022
    },
    {
      "epoch": 0.54048,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0120849609375,
      "learning_rate": 0.0001,
      "loss": 3.9311,
      "loss/crossentropy": 1.8567208647727966,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17688161879777908,
      "step": 27024
    },
    {
      "epoch": 0.54052,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0108062744140625,
      "learning_rate": 0.0001,
      "loss": 3.8285,
      "loss/crossentropy": 2.0056475400924683,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16437891870737076,
      "step": 27026
    },
    {
      "epoch": 0.54056,
      "grad_norm": 1.875,
      "grad_norm_var": 0.012023671468098959,
      "learning_rate": 0.0001,
      "loss": 4.0068,
      "loss/crossentropy": 2.2776039838790894,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21126240491867065,
      "step": 27028
    },
    {
      "epoch": 0.5406,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.012674967447916666,
      "learning_rate": 0.0001,
      "loss": 4.2638,
      "loss/crossentropy": 2.2637280225753784,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19725032150745392,
      "step": 27030
    },
    {
      "epoch": 0.54064,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.011895497639973959,
      "learning_rate": 0.0001,
      "loss": 3.837,
      "loss/crossentropy": 1.6967254281044006,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1950267255306244,
      "step": 27032
    },
    {
      "epoch": 0.54068,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.008591461181640624,
      "learning_rate": 0.0001,
      "loss": 4.3259,
      "loss/crossentropy": 2.2444742918014526,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20126602053642273,
      "step": 27034
    },
    {
      "epoch": 0.54072,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.008656565348307292,
      "learning_rate": 0.0001,
      "loss": 3.8887,
      "loss/crossentropy": 2.216984808444977,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18849068880081177,
      "step": 27036
    },
    {
      "epoch": 0.54076,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.008296712239583334,
      "learning_rate": 0.0001,
      "loss": 3.9656,
      "loss/crossentropy": 2.089816451072693,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2007404863834381,
      "step": 27038
    },
    {
      "epoch": 0.5408,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.0071370442708333336,
      "learning_rate": 0.0001,
      "loss": 3.7989,
      "loss/crossentropy": 1.839013695716858,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17485745251178741,
      "step": 27040
    },
    {
      "epoch": 0.54084,
      "grad_norm": 2.0,
      "grad_norm_var": 0.007393391927083334,
      "learning_rate": 0.0001,
      "loss": 4.3015,
      "loss/crossentropy": 2.3915692567825317,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21270561963319778,
      "step": 27042
    },
    {
      "epoch": 0.54088,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0087799072265625,
      "learning_rate": 0.0001,
      "loss": 3.8979,
      "loss/crossentropy": 2.2086989879608154,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21337078511714935,
      "step": 27044
    },
    {
      "epoch": 0.54092,
      "grad_norm": 2.0,
      "grad_norm_var": 0.008971913655598959,
      "learning_rate": 0.0001,
      "loss": 4.241,
      "loss/crossentropy": 1.8932998776435852,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1783016324043274,
      "step": 27046
    },
    {
      "epoch": 0.54096,
      "grad_norm": 1.875,
      "grad_norm_var": 0.008479563395182292,
      "learning_rate": 0.0001,
      "loss": 3.9551,
      "loss/crossentropy": 1.9314876198768616,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1857961043715477,
      "step": 27048
    },
    {
      "epoch": 0.541,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.0074371337890625,
      "learning_rate": 0.0001,
      "loss": 3.875,
      "loss/crossentropy": 1.616044044494629,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16122639179229736,
      "step": 27050
    },
    {
      "epoch": 0.54104,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.007572174072265625,
      "learning_rate": 0.0001,
      "loss": 3.9375,
      "loss/crossentropy": 1.7567220330238342,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18351546674966812,
      "step": 27052
    },
    {
      "epoch": 0.54108,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.007350413004557291,
      "learning_rate": 0.0001,
      "loss": 3.9276,
      "loss/crossentropy": 1.8721659779548645,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19143584370613098,
      "step": 27054
    },
    {
      "epoch": 0.54112,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0079254150390625,
      "learning_rate": 0.0001,
      "loss": 3.8402,
      "loss/crossentropy": 1.8633847832679749,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18027572333812714,
      "step": 27056
    },
    {
      "epoch": 0.54116,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.008455149332682292,
      "learning_rate": 0.0001,
      "loss": 4.0355,
      "loss/crossentropy": 2.0588165521621704,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19255611300468445,
      "step": 27058
    },
    {
      "epoch": 0.5412,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0056304931640625,
      "learning_rate": 0.0001,
      "loss": 4.1375,
      "loss/crossentropy": 2.316554546356201,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2137911319732666,
      "step": 27060
    },
    {
      "epoch": 0.54124,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.0047910054524739586,
      "learning_rate": 0.0001,
      "loss": 4.0224,
      "loss/crossentropy": 2.1319636702537537,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21494153887033463,
      "step": 27062
    },
    {
      "epoch": 0.54128,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.005326334635416667,
      "learning_rate": 0.0001,
      "loss": 3.8054,
      "loss/crossentropy": 1.9028087854385376,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1869627833366394,
      "step": 27064
    },
    {
      "epoch": 0.54132,
      "grad_norm": 1.875,
      "grad_norm_var": 0.0044097900390625,
      "learning_rate": 0.0001,
      "loss": 3.9521,
      "loss/crossentropy": 2.0914413928985596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2090655192732811,
      "step": 27066
    },
    {
      "epoch": 0.54136,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.004367828369140625,
      "learning_rate": 0.0001,
      "loss": 3.9667,
      "loss/crossentropy": 2.1949799060821533,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19384562969207764,
      "step": 27068
    },
    {
      "epoch": 0.5414,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0045562744140625,
      "learning_rate": 0.0001,
      "loss": 4.0931,
      "loss/crossentropy": 2.076161026954651,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17761556804180145,
      "step": 27070
    },
    {
      "epoch": 0.54144,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.004874674479166666,
      "learning_rate": 0.0001,
      "loss": 3.9357,
      "loss/crossentropy": 2.0287702679634094,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18184736371040344,
      "step": 27072
    },
    {
      "epoch": 0.54148,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.004914347330729167,
      "learning_rate": 0.0001,
      "loss": 3.7579,
      "loss/crossentropy": 2.2093619108200073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20473447442054749,
      "step": 27074
    },
    {
      "epoch": 0.54152,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.004646809895833334,
      "learning_rate": 0.0001,
      "loss": 4.0884,
      "loss/crossentropy": 2.076124429702759,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19069521129131317,
      "step": 27076
    },
    {
      "epoch": 0.54156,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.004303995768229167,
      "learning_rate": 0.0001,
      "loss": 3.7931,
      "loss/crossentropy": 2.014284610748291,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18654431402683258,
      "step": 27078
    },
    {
      "epoch": 0.5416,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.0044329325358072914,
      "learning_rate": 0.0001,
      "loss": 4.0599,
      "loss/crossentropy": 2.0898516178131104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19269923120737076,
      "step": 27080
    },
    {
      "epoch": 0.54164,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.015819295247395834,
      "learning_rate": 0.0001,
      "loss": 4.4848,
      "loss/crossentropy": 2.1016663908958435,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2058946192264557,
      "step": 27082
    },
    {
      "epoch": 0.54168,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.020896148681640626,
      "learning_rate": 0.0001,
      "loss": 3.8787,
      "loss/crossentropy": 1.6725799441337585,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1853618249297142,
      "step": 27084
    },
    {
      "epoch": 0.54172,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.020531209309895833,
      "learning_rate": 0.0001,
      "loss": 4.1032,
      "loss/crossentropy": 2.414761781692505,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2319842129945755,
      "step": 27086
    },
    {
      "epoch": 0.54176,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.018344879150390625,
      "learning_rate": 0.0001,
      "loss": 4.0617,
      "loss/crossentropy": 1.887232780456543,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1807008758187294,
      "step": 27088
    },
    {
      "epoch": 0.5418,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.0190093994140625,
      "learning_rate": 0.0001,
      "loss": 4.3199,
      "loss/crossentropy": 2.1659968495368958,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2109128087759018,
      "step": 27090
    },
    {
      "epoch": 0.54184,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.02069269816080729,
      "learning_rate": 0.0001,
      "loss": 4.3623,
      "loss/crossentropy": 2.134926438331604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.200565405189991,
      "step": 27092
    },
    {
      "epoch": 0.54188,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.016621907552083332,
      "learning_rate": 0.0001,
      "loss": 4.3347,
      "loss/crossentropy": 2.3204580545425415,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21586615592241287,
      "step": 27094
    },
    {
      "epoch": 0.54192,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.016462198893229165,
      "learning_rate": 0.0001,
      "loss": 3.8415,
      "loss/crossentropy": 2.1035609245300293,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20287606865167618,
      "step": 27096
    },
    {
      "epoch": 0.54196,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0115631103515625,
      "learning_rate": 0.0001,
      "loss": 4.0134,
      "loss/crossentropy": 1.9996371865272522,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1981046125292778,
      "step": 27098
    },
    {
      "epoch": 0.542,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.011220041910807292,
      "learning_rate": 0.0001,
      "loss": 4.4227,
      "loss/crossentropy": 2.2769532203674316,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19060392677783966,
      "step": 27100
    },
    {
      "epoch": 0.54204,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.014684804280598958,
      "learning_rate": 0.0001,
      "loss": 3.92,
      "loss/crossentropy": 1.8792597651481628,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17460587620735168,
      "step": 27102
    },
    {
      "epoch": 0.54208,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.01631647745768229,
      "learning_rate": 0.0001,
      "loss": 3.9027,
      "loss/crossentropy": 1.7762881517410278,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15890707820653915,
      "step": 27104
    },
    {
      "epoch": 0.54212,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.016196441650390626,
      "learning_rate": 0.0001,
      "loss": 3.9892,
      "loss/crossentropy": 2.125525116920471,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22518931329250336,
      "step": 27106
    },
    {
      "epoch": 0.54216,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.012033843994140625,
      "learning_rate": 0.0001,
      "loss": 3.7171,
      "loss/crossentropy": 2.1541160345077515,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1856529340147972,
      "step": 27108
    },
    {
      "epoch": 0.5422,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.010927073160807292,
      "learning_rate": 0.0001,
      "loss": 4.0158,
      "loss/crossentropy": 1.8441925048828125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19040126353502274,
      "step": 27110
    },
    {
      "epoch": 0.54224,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.011260732014973959,
      "learning_rate": 0.0001,
      "loss": 3.9886,
      "loss/crossentropy": 1.9131816625595093,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19395362585783005,
      "step": 27112
    },
    {
      "epoch": 0.54228,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.015254465738932292,
      "learning_rate": 0.0001,
      "loss": 4.0498,
      "loss/crossentropy": 1.9133941531181335,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23238787055015564,
      "step": 27114
    },
    {
      "epoch": 0.54232,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.011253865559895833,
      "learning_rate": 0.0001,
      "loss": 4.1971,
      "loss/crossentropy": 2.2706873416900635,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21193598210811615,
      "step": 27116
    },
    {
      "epoch": 0.54236,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.009822336832682292,
      "learning_rate": 0.0001,
      "loss": 3.9732,
      "loss/crossentropy": 1.7896053791046143,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1722758710384369,
      "step": 27118
    },
    {
      "epoch": 0.5424,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.012636057535807292,
      "learning_rate": 0.0001,
      "loss": 4.3074,
      "loss/crossentropy": 2.3640060424804688,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21565549075603485,
      "step": 27120
    },
    {
      "epoch": 0.54244,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.012857818603515625,
      "learning_rate": 0.0001,
      "loss": 3.8032,
      "loss/crossentropy": 1.6995807886123657,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15929770469665527,
      "step": 27122
    },
    {
      "epoch": 0.54248,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.012607574462890625,
      "learning_rate": 0.0001,
      "loss": 3.9518,
      "loss/crossentropy": 1.6559955477714539,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1741805449128151,
      "step": 27124
    },
    {
      "epoch": 0.54252,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.012451171875,
      "learning_rate": 0.0001,
      "loss": 3.8393,
      "loss/crossentropy": 1.9256713390350342,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23186514526605606,
      "step": 27126
    },
    {
      "epoch": 0.54256,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.011486562093098958,
      "learning_rate": 0.0001,
      "loss": 3.8599,
      "loss/crossentropy": 2.0060980319976807,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20487740635871887,
      "step": 27128
    },
    {
      "epoch": 0.5426,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.009545644124348959,
      "learning_rate": 0.0001,
      "loss": 3.6376,
      "loss/crossentropy": 1.8419195413589478,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17364958673715591,
      "step": 27130
    },
    {
      "epoch": 0.54264,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0100250244140625,
      "learning_rate": 0.0001,
      "loss": 4.1161,
      "loss/crossentropy": 1.9890483617782593,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21333064138889313,
      "step": 27132
    },
    {
      "epoch": 0.54268,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.009639485677083334,
      "learning_rate": 0.0001,
      "loss": 4.1288,
      "loss/crossentropy": 2.2758660316467285,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21215695142745972,
      "step": 27134
    },
    {
      "epoch": 0.54272,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.006086985270182292,
      "learning_rate": 0.0001,
      "loss": 4.1358,
      "loss/crossentropy": 2.273390769958496,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20087362825870514,
      "step": 27136
    },
    {
      "epoch": 0.54276,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.005322265625,
      "learning_rate": 0.0001,
      "loss": 4.0575,
      "loss/crossentropy": 2.446821928024292,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20775765180587769,
      "step": 27138
    },
    {
      "epoch": 0.5428,
      "grad_norm": 1.875,
      "grad_norm_var": 0.006449127197265625,
      "learning_rate": 0.0001,
      "loss": 4.0756,
      "loss/crossentropy": 2.391844630241394,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2062644362449646,
      "step": 27140
    },
    {
      "epoch": 0.54284,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.007216135660807292,
      "learning_rate": 0.0001,
      "loss": 3.9436,
      "loss/crossentropy": 2.112763822078705,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17983781546354294,
      "step": 27142
    },
    {
      "epoch": 0.54288,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.006624094645182292,
      "learning_rate": 0.0001,
      "loss": 4.2086,
      "loss/crossentropy": 2.184209704399109,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20391154289245605,
      "step": 27144
    },
    {
      "epoch": 0.54292,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.005806477864583334,
      "learning_rate": 0.0001,
      "loss": 3.9503,
      "loss/crossentropy": 1.911209523677826,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19172395020723343,
      "step": 27146
    },
    {
      "epoch": 0.54296,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.00419921875,
      "learning_rate": 0.0001,
      "loss": 3.9281,
      "loss/crossentropy": 2.132554292678833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1887652799487114,
      "step": 27148
    },
    {
      "epoch": 0.543,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.010130818684895833,
      "learning_rate": 0.0001,
      "loss": 4.1577,
      "loss/crossentropy": 2.4312938451766968,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20902784168720245,
      "step": 27150
    },
    {
      "epoch": 0.54304,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.011909993489583333,
      "learning_rate": 0.0001,
      "loss": 4.088,
      "loss/crossentropy": 1.857543170452118,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1920638307929039,
      "step": 27152
    },
    {
      "epoch": 0.54308,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.01109619140625,
      "learning_rate": 0.0001,
      "loss": 4.035,
      "loss/crossentropy": 1.8170040845870972,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1604015827178955,
      "step": 27154
    },
    {
      "epoch": 0.54312,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010815175374348958,
      "learning_rate": 0.0001,
      "loss": 3.9593,
      "loss/crossentropy": 1.8490120768547058,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19839002192020416,
      "step": 27156
    },
    {
      "epoch": 0.54316,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.009954579671223958,
      "learning_rate": 0.0001,
      "loss": 3.9611,
      "loss/crossentropy": 2.1401009559631348,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20167674124240875,
      "step": 27158
    },
    {
      "epoch": 0.5432,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0108306884765625,
      "learning_rate": 0.0001,
      "loss": 4.1829,
      "loss/crossentropy": 2.216339647769928,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2046881765127182,
      "step": 27160
    },
    {
      "epoch": 0.54324,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.011112213134765625,
      "learning_rate": 0.0001,
      "loss": 3.9202,
      "loss/crossentropy": 2.2559473514556885,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20772842317819595,
      "step": 27162
    },
    {
      "epoch": 0.54328,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.010811360677083333,
      "learning_rate": 0.0001,
      "loss": 4.0261,
      "loss/crossentropy": 1.9810506105422974,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18164421617984772,
      "step": 27164
    },
    {
      "epoch": 0.54332,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0047271728515625,
      "learning_rate": 0.0001,
      "loss": 3.9554,
      "loss/crossentropy": 2.2585933208465576,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20920690149068832,
      "step": 27166
    },
    {
      "epoch": 0.54336,
      "grad_norm": 1.75,
      "grad_norm_var": 0.005524698893229167,
      "learning_rate": 0.0001,
      "loss": 3.6078,
      "loss/crossentropy": 1.8078173995018005,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16720515489578247,
      "step": 27168
    },
    {
      "epoch": 0.5434,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.005678049723307292,
      "learning_rate": 0.0001,
      "loss": 3.9103,
      "loss/crossentropy": 1.7529219388961792,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17757199704647064,
      "step": 27170
    },
    {
      "epoch": 0.54344,
      "grad_norm": 1.875,
      "grad_norm_var": 0.004713694254557292,
      "learning_rate": 0.0001,
      "loss": 3.8833,
      "loss/crossentropy": 2.0651434659957886,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2342069447040558,
      "step": 27172
    },
    {
      "epoch": 0.54348,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.005125935872395833,
      "learning_rate": 0.0001,
      "loss": 3.7693,
      "loss/crossentropy": 1.7311297059059143,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17116959393024445,
      "step": 27174
    },
    {
      "epoch": 0.54352,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.0042803446451822914,
      "learning_rate": 0.0001,
      "loss": 3.9136,
      "loss/crossentropy": 1.65495365858078,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1756131500005722,
      "step": 27176
    },
    {
      "epoch": 0.54356,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.0039713541666666664,
      "learning_rate": 0.0001,
      "loss": 3.8718,
      "loss/crossentropy": 2.1134061217308044,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19070632755756378,
      "step": 27178
    },
    {
      "epoch": 0.5436,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0037261962890625,
      "learning_rate": 0.0001,
      "loss": 3.8315,
      "loss/crossentropy": 1.8080366849899292,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1905672550201416,
      "step": 27180
    },
    {
      "epoch": 0.54364,
      "grad_norm": 2.0,
      "grad_norm_var": 0.004550933837890625,
      "learning_rate": 0.0001,
      "loss": 4.1646,
      "loss/crossentropy": 2.1257325410842896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20127511024475098,
      "step": 27182
    },
    {
      "epoch": 0.54368,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0030263264973958335,
      "learning_rate": 0.0001,
      "loss": 4.2156,
      "loss/crossentropy": 2.3125778436660767,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21088162064552307,
      "step": 27184
    },
    {
      "epoch": 0.54372,
      "grad_norm": 2.125,
      "grad_norm_var": 0.005777740478515625,
      "learning_rate": 0.0001,
      "loss": 4.14,
      "loss/crossentropy": 2.21799373626709,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2073218673467636,
      "step": 27186
    },
    {
      "epoch": 0.54376,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.005686187744140625,
      "learning_rate": 0.0001,
      "loss": 3.8577,
      "loss/crossentropy": 1.8872935771942139,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17742664366960526,
      "step": 27188
    },
    {
      "epoch": 0.5438,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0054433186848958336,
      "learning_rate": 0.0001,
      "loss": 3.7985,
      "loss/crossentropy": 1.6956111788749695,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17246189713478088,
      "step": 27190
    },
    {
      "epoch": 0.54384,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0057065327962239586,
      "learning_rate": 0.0001,
      "loss": 4.0475,
      "loss/crossentropy": 2.0606382489204407,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2049379125237465,
      "step": 27192
    },
    {
      "epoch": 0.54388,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.005092112223307291,
      "learning_rate": 0.0001,
      "loss": 3.9601,
      "loss/crossentropy": 2.1031752824783325,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19281570613384247,
      "step": 27194
    },
    {
      "epoch": 0.54392,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.006306711832682292,
      "learning_rate": 0.0001,
      "loss": 4.1294,
      "loss/crossentropy": 2.113471269607544,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17625518888235092,
      "step": 27196
    },
    {
      "epoch": 0.54396,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0061279296875,
      "learning_rate": 0.0001,
      "loss": 4.1399,
      "loss/crossentropy": 2.0233633518218994,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19023866951465607,
      "step": 27198
    },
    {
      "epoch": 0.544,
      "grad_norm": 2.0,
      "grad_norm_var": 0.006453196207682292,
      "learning_rate": 0.0001,
      "loss": 4.0167,
      "loss/crossentropy": 2.1072750091552734,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18891653418540955,
      "step": 27200
    },
    {
      "epoch": 0.54404,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.011250559488932292,
      "learning_rate": 0.0001,
      "loss": 4.0871,
      "loss/crossentropy": 2.1586121320724487,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19431254267692566,
      "step": 27202
    },
    {
      "epoch": 0.54408,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.01123046875,
      "learning_rate": 0.0001,
      "loss": 3.9215,
      "loss/crossentropy": 1.817845344543457,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17477234452962875,
      "step": 27204
    },
    {
      "epoch": 0.54412,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.0123687744140625,
      "learning_rate": 0.0001,
      "loss": 3.7602,
      "loss/crossentropy": 2.0410149693489075,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19700051844120026,
      "step": 27206
    },
    {
      "epoch": 0.54416,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.011563873291015625,
      "learning_rate": 0.0001,
      "loss": 4.1133,
      "loss/crossentropy": 2.096368670463562,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20101771503686905,
      "step": 27208
    },
    {
      "epoch": 0.5442,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.013061269124348959,
      "learning_rate": 0.0001,
      "loss": 4.0634,
      "loss/crossentropy": 2.3180510997772217,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23763899505138397,
      "step": 27210
    },
    {
      "epoch": 0.54424,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.011986287434895833,
      "learning_rate": 0.0001,
      "loss": 4.0617,
      "loss/crossentropy": 2.2700339555740356,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21923137456178665,
      "step": 27212
    },
    {
      "epoch": 0.54428,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.011659495035807292,
      "learning_rate": 0.0001,
      "loss": 3.8744,
      "loss/crossentropy": 2.0254225730895996,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1879374161362648,
      "step": 27214
    },
    {
      "epoch": 0.54432,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.012401326497395834,
      "learning_rate": 0.0001,
      "loss": 4.0095,
      "loss/crossentropy": 2.0617589950561523,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20201390981674194,
      "step": 27216
    },
    {
      "epoch": 0.54436,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.005147298177083333,
      "learning_rate": 0.0001,
      "loss": 4.1374,
      "loss/crossentropy": 2.1432749032974243,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21799273788928986,
      "step": 27218
    },
    {
      "epoch": 0.5444,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.005622355143229166,
      "learning_rate": 0.0001,
      "loss": 4.2461,
      "loss/crossentropy": 1.948439359664917,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1876726821064949,
      "step": 27220
    },
    {
      "epoch": 0.54444,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.004518381754557292,
      "learning_rate": 0.0001,
      "loss": 4.0501,
      "loss/crossentropy": 2.2085187435150146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2106829285621643,
      "step": 27222
    },
    {
      "epoch": 0.54448,
      "grad_norm": 2.0,
      "grad_norm_var": 0.005475870768229167,
      "learning_rate": 0.0001,
      "loss": 3.8899,
      "loss/crossentropy": 2.0642993450164795,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19151441007852554,
      "step": 27224
    },
    {
      "epoch": 0.54452,
      "grad_norm": 2.125,
      "grad_norm_var": 0.009427897135416667,
      "learning_rate": 0.0001,
      "loss": 3.8693,
      "loss/crossentropy": 1.8351057767868042,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17997722327709198,
      "step": 27226
    },
    {
      "epoch": 0.54456,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.009903971354166667,
      "learning_rate": 0.0001,
      "loss": 3.8896,
      "loss/crossentropy": 2.033097803592682,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18373706936836243,
      "step": 27228
    },
    {
      "epoch": 0.5446,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.009643300374348959,
      "learning_rate": 0.0001,
      "loss": 4.076,
      "loss/crossentropy": 1.9884595274925232,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19420110434293747,
      "step": 27230
    },
    {
      "epoch": 0.54464,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.009258778889973958,
      "learning_rate": 0.0001,
      "loss": 3.7903,
      "loss/crossentropy": 1.9873828887939453,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19645929336547852,
      "step": 27232
    },
    {
      "epoch": 0.54468,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.010298411051432291,
      "learning_rate": 0.0001,
      "loss": 3.8454,
      "loss/crossentropy": 1.9592986702919006,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20220571756362915,
      "step": 27234
    },
    {
      "epoch": 0.54472,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.009480794270833334,
      "learning_rate": 0.0001,
      "loss": 3.8467,
      "loss/crossentropy": 2.0207183957099915,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18422196060419083,
      "step": 27236
    },
    {
      "epoch": 0.54476,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.009740193684895834,
      "learning_rate": 0.0001,
      "loss": 3.753,
      "loss/crossentropy": 1.8570082187652588,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17869313061237335,
      "step": 27238
    },
    {
      "epoch": 0.5448,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.008882649739583333,
      "learning_rate": 0.0001,
      "loss": 3.7697,
      "loss/crossentropy": 2.043560802936554,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18643952161073685,
      "step": 27240
    },
    {
      "epoch": 0.54484,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0052874247233072914,
      "learning_rate": 0.0001,
      "loss": 4.0459,
      "loss/crossentropy": 2.2474546432495117,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2210846170783043,
      "step": 27242
    },
    {
      "epoch": 0.54488,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.005882771809895834,
      "learning_rate": 0.0001,
      "loss": 4.0702,
      "loss/crossentropy": 2.4019227027893066,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20284153521060944,
      "step": 27244
    },
    {
      "epoch": 0.54492,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.006367746988932292,
      "learning_rate": 0.0001,
      "loss": 3.9849,
      "loss/crossentropy": 2.4402183294296265,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2066793590784073,
      "step": 27246
    },
    {
      "epoch": 0.54496,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.006623331705729167,
      "learning_rate": 0.0001,
      "loss": 3.7691,
      "loss/crossentropy": 1.6989346742630005,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1731812208890915,
      "step": 27248
    },
    {
      "epoch": 0.545,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.007126617431640625,
      "learning_rate": 0.0001,
      "loss": 3.7411,
      "loss/crossentropy": 1.658067524433136,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17113492637872696,
      "step": 27250
    },
    {
      "epoch": 0.54504,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.007287343343098958,
      "learning_rate": 0.0001,
      "loss": 4.0801,
      "loss/crossentropy": 2.032026529312134,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18906760960817337,
      "step": 27252
    },
    {
      "epoch": 0.54508,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0065185546875,
      "learning_rate": 0.0001,
      "loss": 4.1494,
      "loss/crossentropy": 2.0466246008872986,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1963697373867035,
      "step": 27254
    },
    {
      "epoch": 0.54512,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.006906890869140625,
      "learning_rate": 0.0001,
      "loss": 4.1727,
      "loss/crossentropy": 2.199449300765991,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19188550859689713,
      "step": 27256
    },
    {
      "epoch": 0.54516,
      "grad_norm": 1.875,
      "grad_norm_var": 0.007542928059895833,
      "learning_rate": 0.0001,
      "loss": 4.1578,
      "loss/crossentropy": 2.10872745513916,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20023853331804276,
      "step": 27258
    },
    {
      "epoch": 0.5452,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.007010904947916666,
      "learning_rate": 0.0001,
      "loss": 4.1923,
      "loss/crossentropy": 2.2712149620056152,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19052613526582718,
      "step": 27260
    },
    {
      "epoch": 0.54524,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.008156077067057291,
      "learning_rate": 0.0001,
      "loss": 3.6014,
      "loss/crossentropy": 1.61397385597229,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.155337356030941,
      "step": 27262
    },
    {
      "epoch": 0.54528,
      "grad_norm": 2.0,
      "grad_norm_var": 0.008194986979166667,
      "learning_rate": 0.0001,
      "loss": 4.1433,
      "loss/crossentropy": 1.8988584876060486,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17318273335695267,
      "step": 27264
    },
    {
      "epoch": 0.54532,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.007298787434895833,
      "learning_rate": 0.0001,
      "loss": 4.1592,
      "loss/crossentropy": 1.9996931552886963,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18785406649112701,
      "step": 27266
    },
    {
      "epoch": 0.54536,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.007462565104166667,
      "learning_rate": 0.0001,
      "loss": 4.2391,
      "loss/crossentropy": 2.1527179479599,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20097015798091888,
      "step": 27268
    },
    {
      "epoch": 0.5454,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008609771728515625,
      "learning_rate": 0.0001,
      "loss": 4.051,
      "loss/crossentropy": 2.1786444187164307,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20732732862234116,
      "step": 27270
    },
    {
      "epoch": 0.54544,
      "grad_norm": 1.875,
      "grad_norm_var": 0.007673136393229167,
      "learning_rate": 0.0001,
      "loss": 3.8147,
      "loss/crossentropy": 1.8255316019058228,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18340704590082169,
      "step": 27272
    },
    {
      "epoch": 0.54548,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.008046213785807292,
      "learning_rate": 0.0001,
      "loss": 4.0119,
      "loss/crossentropy": 2.1311771273612976,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.206636942923069,
      "step": 27274
    },
    {
      "epoch": 0.54552,
      "grad_norm": 1.7265625,
      "grad_norm_var": 0.010113271077473958,
      "learning_rate": 0.0001,
      "loss": 3.752,
      "loss/crossentropy": 1.7570669651031494,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15514526516199112,
      "step": 27276
    },
    {
      "epoch": 0.54556,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.00963134765625,
      "learning_rate": 0.0001,
      "loss": 4.3659,
      "loss/crossentropy": 2.407896041870117,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22130534052848816,
      "step": 27278
    },
    {
      "epoch": 0.5456,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.009891764322916666,
      "learning_rate": 0.0001,
      "loss": 4.1536,
      "loss/crossentropy": 2.1037758588790894,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23098163306713104,
      "step": 27280
    },
    {
      "epoch": 0.54564,
      "grad_norm": 1.875,
      "grad_norm_var": 0.008847808837890625,
      "learning_rate": 0.0001,
      "loss": 4.1011,
      "loss/crossentropy": 2.213571786880493,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.210727259516716,
      "step": 27282
    },
    {
      "epoch": 0.54568,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.00955810546875,
      "learning_rate": 0.0001,
      "loss": 3.997,
      "loss/crossentropy": 1.9279372096061707,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19715701043605804,
      "step": 27284
    },
    {
      "epoch": 0.54572,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.009372711181640625,
      "learning_rate": 0.0001,
      "loss": 4.0212,
      "loss/crossentropy": 2.257995128631592,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20844868570566177,
      "step": 27286
    },
    {
      "epoch": 0.54576,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.009694163004557292,
      "learning_rate": 0.0001,
      "loss": 3.8984,
      "loss/crossentropy": 1.7289800643920898,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1757895052433014,
      "step": 27288
    },
    {
      "epoch": 0.5458,
      "grad_norm": 2.0,
      "grad_norm_var": 0.009930165608723958,
      "learning_rate": 0.0001,
      "loss": 3.767,
      "loss/crossentropy": 1.9156250953674316,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17408546805381775,
      "step": 27290
    },
    {
      "epoch": 0.54584,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.007330067952473958,
      "learning_rate": 0.0001,
      "loss": 4.153,
      "loss/crossentropy": 2.2218767404556274,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20564305782318115,
      "step": 27292
    },
    {
      "epoch": 0.54588,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.007767740885416667,
      "learning_rate": 0.0001,
      "loss": 4.3436,
      "loss/crossentropy": 2.2322014570236206,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19707325100898743,
      "step": 27294
    },
    {
      "epoch": 0.54592,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.007575480143229166,
      "learning_rate": 0.0001,
      "loss": 3.9286,
      "loss/crossentropy": 1.9559943675994873,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2009485438466072,
      "step": 27296
    },
    {
      "epoch": 0.54596,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.007496897379557292,
      "learning_rate": 0.0001,
      "loss": 3.8224,
      "loss/crossentropy": 2.322990119457245,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21545371413230896,
      "step": 27298
    },
    {
      "epoch": 0.546,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.007698567708333334,
      "learning_rate": 0.0001,
      "loss": 3.7273,
      "loss/crossentropy": 1.7752234935760498,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18764031678438187,
      "step": 27300
    },
    {
      "epoch": 0.54604,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.008317057291666667,
      "learning_rate": 0.0001,
      "loss": 4.1671,
      "loss/crossentropy": 2.2797446250915527,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20618843287229538,
      "step": 27302
    },
    {
      "epoch": 0.54608,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.009468332926432291,
      "learning_rate": 0.0001,
      "loss": 4.0499,
      "loss/crossentropy": 2.1392215490341187,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19076371937990189,
      "step": 27304
    },
    {
      "epoch": 0.54612,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.006852213541666667,
      "learning_rate": 0.0001,
      "loss": 3.9362,
      "loss/crossentropy": 2.065139055252075,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20819596201181412,
      "step": 27306
    },
    {
      "epoch": 0.54616,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.007281239827473958,
      "learning_rate": 0.0001,
      "loss": 3.8854,
      "loss/crossentropy": 1.90589839220047,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18194814771413803,
      "step": 27308
    },
    {
      "epoch": 0.5462,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.0071489969889322914,
      "learning_rate": 0.0001,
      "loss": 4.0056,
      "loss/crossentropy": 2.18922221660614,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18913178890943527,
      "step": 27310
    },
    {
      "epoch": 0.54624,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.006357574462890625,
      "learning_rate": 0.0001,
      "loss": 3.9098,
      "loss/crossentropy": 2.3256269693374634,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20674752444028854,
      "step": 27312
    },
    {
      "epoch": 0.54628,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.006859334309895834,
      "learning_rate": 0.0001,
      "loss": 3.8007,
      "loss/crossentropy": 2.2145326733589172,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18575318902730942,
      "step": 27314
    },
    {
      "epoch": 0.54632,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0065419514973958336,
      "learning_rate": 0.0001,
      "loss": 4.27,
      "loss/crossentropy": 2.326914429664612,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2115549072623253,
      "step": 27316
    },
    {
      "epoch": 0.54636,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.006551106770833333,
      "learning_rate": 0.0001,
      "loss": 4.1064,
      "loss/crossentropy": 2.1162266731262207,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1951146125793457,
      "step": 27318
    },
    {
      "epoch": 0.5464,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.027858225504557292,
      "learning_rate": 0.0001,
      "loss": 3.929,
      "loss/crossentropy": 1.96768057346344,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19022534787654877,
      "step": 27320
    },
    {
      "epoch": 0.54644,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.028368123372395835,
      "learning_rate": 0.0001,
      "loss": 3.7685,
      "loss/crossentropy": 1.6591554880142212,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16152790188789368,
      "step": 27322
    },
    {
      "epoch": 0.54648,
      "grad_norm": 1.734375,
      "grad_norm_var": 0.030863444010416668,
      "learning_rate": 0.0001,
      "loss": 3.8354,
      "loss/crossentropy": 2.3317655324935913,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20369040220975876,
      "step": 27324
    },
    {
      "epoch": 0.54652,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.032871246337890625,
      "learning_rate": 0.0001,
      "loss": 3.808,
      "loss/crossentropy": 1.9729547500610352,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18059556931257248,
      "step": 27326
    },
    {
      "epoch": 0.54656,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.03397216796875,
      "learning_rate": 0.0001,
      "loss": 3.9546,
      "loss/crossentropy": 2.0512842535972595,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19131098687648773,
      "step": 27328
    },
    {
      "epoch": 0.5466,
      "grad_norm": 2.125,
      "grad_norm_var": 0.0363433837890625,
      "learning_rate": 0.0001,
      "loss": 4.1688,
      "loss/crossentropy": 2.385029435157776,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2064640149474144,
      "step": 27330
    },
    {
      "epoch": 0.54664,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0443603515625,
      "learning_rate": 0.0001,
      "loss": 4.2089,
      "loss/crossentropy": 2.069806933403015,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19368857145309448,
      "step": 27332
    },
    {
      "epoch": 0.54668,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.043314615885416664,
      "learning_rate": 0.0001,
      "loss": 3.9046,
      "loss/crossentropy": 1.7913403511047363,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18388576805591583,
      "step": 27334
    },
    {
      "epoch": 0.54672,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.01974665323893229,
      "learning_rate": 0.0001,
      "loss": 3.9618,
      "loss/crossentropy": 1.8095470070838928,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16660822927951813,
      "step": 27336
    },
    {
      "epoch": 0.54676,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.019551595052083332,
      "learning_rate": 0.0001,
      "loss": 3.9662,
      "loss/crossentropy": 1.9614168405532837,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1717158928513527,
      "step": 27338
    },
    {
      "epoch": 0.5468,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0172607421875,
      "learning_rate": 0.0001,
      "loss": 4.1897,
      "loss/crossentropy": 1.9303128719329834,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18384890258312225,
      "step": 27340
    },
    {
      "epoch": 0.54684,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.014204915364583333,
      "learning_rate": 0.0001,
      "loss": 3.8205,
      "loss/crossentropy": 1.675184428691864,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17097727209329605,
      "step": 27342
    },
    {
      "epoch": 0.54688,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.013331858317057292,
      "learning_rate": 0.0001,
      "loss": 3.817,
      "loss/crossentropy": 2.0593711137771606,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22880251705646515,
      "step": 27344
    },
    {
      "epoch": 0.54692,
      "grad_norm": 1.875,
      "grad_norm_var": 0.011424763997395834,
      "learning_rate": 0.0001,
      "loss": 3.7013,
      "loss/crossentropy": 1.79487544298172,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1764550730586052,
      "step": 27346
    },
    {
      "epoch": 0.54696,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.00430908203125,
      "learning_rate": 0.0001,
      "loss": 3.7599,
      "loss/crossentropy": 2.2460497617721558,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20630930364131927,
      "step": 27348
    },
    {
      "epoch": 0.547,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.024339803059895835,
      "learning_rate": 0.0001,
      "loss": 3.9512,
      "loss/crossentropy": 2.056776225566864,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1956087201833725,
      "step": 27350
    },
    {
      "epoch": 0.54704,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0264556884765625,
      "learning_rate": 0.0001,
      "loss": 4.0642,
      "loss/crossentropy": 2.1221381425857544,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19081637263298035,
      "step": 27352
    },
    {
      "epoch": 0.54708,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.025994618733723957,
      "learning_rate": 0.0001,
      "loss": 4.2032,
      "loss/crossentropy": 2.1899205446243286,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19173882901668549,
      "step": 27354
    },
    {
      "epoch": 0.54712,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.029012044270833332,
      "learning_rate": 0.0001,
      "loss": 4.3297,
      "loss/crossentropy": 1.9936851859092712,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2355274260044098,
      "step": 27356
    },
    {
      "epoch": 0.54716,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.02979100545247396,
      "learning_rate": 0.0001,
      "loss": 3.882,
      "loss/crossentropy": 1.7802082300186157,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17976941913366318,
      "step": 27358
    },
    {
      "epoch": 0.5472,
      "grad_norm": 2.0,
      "grad_norm_var": 0.029808553059895833,
      "learning_rate": 0.0001,
      "loss": 3.9911,
      "loss/crossentropy": 2.1092435121536255,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18953461945056915,
      "step": 27360
    },
    {
      "epoch": 0.54724,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.031676991780598955,
      "learning_rate": 0.0001,
      "loss": 3.735,
      "loss/crossentropy": 1.9264041185379028,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1695871204137802,
      "step": 27362
    },
    {
      "epoch": 0.54728,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.028816731770833333,
      "learning_rate": 0.0001,
      "loss": 3.724,
      "loss/crossentropy": 2.0451253056526184,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19317582994699478,
      "step": 27364
    },
    {
      "epoch": 0.54732,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.011218007405598958,
      "learning_rate": 0.0001,
      "loss": 4.0508,
      "loss/crossentropy": 2.084929406642914,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20370616763830185,
      "step": 27366
    },
    {
      "epoch": 0.54736,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.008654530843098958,
      "learning_rate": 0.0001,
      "loss": 3.9552,
      "loss/crossentropy": 2.1887794733047485,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21116343885660172,
      "step": 27368
    },
    {
      "epoch": 0.5474,
      "grad_norm": 1.875,
      "grad_norm_var": 0.011229451497395833,
      "learning_rate": 0.0001,
      "loss": 3.7472,
      "loss/crossentropy": 1.9667014479637146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17529867589473724,
      "step": 27370
    },
    {
      "epoch": 0.54744,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.005346425374348958,
      "learning_rate": 0.0001,
      "loss": 4.1537,
      "loss/crossentropy": 2.351253390312195,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23024218529462814,
      "step": 27372
    },
    {
      "epoch": 0.54748,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.006089019775390625,
      "learning_rate": 0.0001,
      "loss": 3.9757,
      "loss/crossentropy": 1.9090047478675842,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17822134494781494,
      "step": 27374
    },
    {
      "epoch": 0.54752,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.005516560872395834,
      "learning_rate": 0.0001,
      "loss": 3.9657,
      "loss/crossentropy": 2.1193515062332153,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1953936368227005,
      "step": 27376
    },
    {
      "epoch": 0.54756,
      "grad_norm": 1.875,
      "grad_norm_var": 0.005230458577473959,
      "learning_rate": 0.0001,
      "loss": 3.9614,
      "loss/crossentropy": 1.8148779273033142,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18782274425029755,
      "step": 27378
    },
    {
      "epoch": 0.5476,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0066640218098958336,
      "learning_rate": 0.0001,
      "loss": 3.9958,
      "loss/crossentropy": 1.8987022042274475,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1779102012515068,
      "step": 27380
    },
    {
      "epoch": 0.54764,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.0071604410807291664,
      "learning_rate": 0.0001,
      "loss": 3.8742,
      "loss/crossentropy": 1.7504222989082336,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.173823744058609,
      "step": 27382
    },
    {
      "epoch": 0.54768,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.008239491780598959,
      "learning_rate": 0.0001,
      "loss": 3.8653,
      "loss/crossentropy": 1.9609686732292175,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18690945953130722,
      "step": 27384
    },
    {
      "epoch": 0.54772,
      "grad_norm": 1.875,
      "grad_norm_var": 0.006639607747395833,
      "learning_rate": 0.0001,
      "loss": 3.8654,
      "loss/crossentropy": 1.613632321357727,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15942876040935516,
      "step": 27386
    },
    {
      "epoch": 0.54776,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.006379191080729167,
      "learning_rate": 0.0001,
      "loss": 3.9697,
      "loss/crossentropy": 2.052034914493561,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18169156461954117,
      "step": 27388
    },
    {
      "epoch": 0.5478,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.005509185791015625,
      "learning_rate": 0.0001,
      "loss": 3.8938,
      "loss/crossentropy": 1.9778786301612854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21691841632127762,
      "step": 27390
    },
    {
      "epoch": 0.54784,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010416412353515625,
      "learning_rate": 0.0001,
      "loss": 4.2751,
      "loss/crossentropy": 2.272818922996521,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.26315005123615265,
      "step": 27392
    },
    {
      "epoch": 0.54788,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.010335286458333334,
      "learning_rate": 0.0001,
      "loss": 3.9688,
      "loss/crossentropy": 1.8213757872581482,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1901681274175644,
      "step": 27394
    },
    {
      "epoch": 0.54792,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0095855712890625,
      "learning_rate": 0.0001,
      "loss": 4.131,
      "loss/crossentropy": 2.0346380472183228,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18924283981323242,
      "step": 27396
    },
    {
      "epoch": 0.54796,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.009738922119140625,
      "learning_rate": 0.0001,
      "loss": 4.1628,
      "loss/crossentropy": 1.9133538603782654,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2124219387769699,
      "step": 27398
    },
    {
      "epoch": 0.548,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.008540852864583334,
      "learning_rate": 0.0001,
      "loss": 4.0782,
      "loss/crossentropy": 2.141755998134613,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20012597739696503,
      "step": 27400
    },
    {
      "epoch": 0.54804,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.008104451497395833,
      "learning_rate": 0.0001,
      "loss": 3.9103,
      "loss/crossentropy": 2.031430244445801,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1892971396446228,
      "step": 27402
    },
    {
      "epoch": 0.54808,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0066220601399739586,
      "learning_rate": 0.0001,
      "loss": 4.1763,
      "loss/crossentropy": 2.3186575174331665,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.231618270277977,
      "step": 27404
    },
    {
      "epoch": 0.54812,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.007496897379557292,
      "learning_rate": 0.0001,
      "loss": 3.7874,
      "loss/crossentropy": 1.719622254371643,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1646794229745865,
      "step": 27406
    },
    {
      "epoch": 0.54816,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.008379872639973958,
      "learning_rate": 0.0001,
      "loss": 4.5104,
      "loss/crossentropy": 2.099879503250122,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20306643843650818,
      "step": 27408
    },
    {
      "epoch": 0.5482,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008553059895833333,
      "learning_rate": 0.0001,
      "loss": 4.1404,
      "loss/crossentropy": 2.104840338230133,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20727242529392242,
      "step": 27410
    },
    {
      "epoch": 0.54824,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.009256744384765625,
      "learning_rate": 0.0001,
      "loss": 4.026,
      "loss/crossentropy": 2.3337541818618774,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19424156099557877,
      "step": 27412
    },
    {
      "epoch": 0.54828,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.01055908203125,
      "learning_rate": 0.0001,
      "loss": 3.8915,
      "loss/crossentropy": 1.751102864742279,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1715778112411499,
      "step": 27414
    },
    {
      "epoch": 0.54832,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0109771728515625,
      "learning_rate": 0.0001,
      "loss": 4.2044,
      "loss/crossentropy": 2.2091389894485474,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2181248739361763,
      "step": 27416
    },
    {
      "epoch": 0.54836,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.011226145426432292,
      "learning_rate": 0.0001,
      "loss": 4.0839,
      "loss/crossentropy": 2.1870399713516235,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20490305125713348,
      "step": 27418
    },
    {
      "epoch": 0.5484,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.011344146728515626,
      "learning_rate": 0.0001,
      "loss": 4.0857,
      "loss/crossentropy": 2.4501689672470093,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22984202951192856,
      "step": 27420
    },
    {
      "epoch": 0.54844,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.011771392822265626,
      "learning_rate": 0.0001,
      "loss": 3.9126,
      "loss/crossentropy": 1.8921697735786438,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1749446839094162,
      "step": 27422
    },
    {
      "epoch": 0.54848,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.008036041259765625,
      "learning_rate": 0.0001,
      "loss": 3.8426,
      "loss/crossentropy": 1.6973050236701965,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1689457967877388,
      "step": 27424
    },
    {
      "epoch": 0.54852,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.006257883707682292,
      "learning_rate": 0.0001,
      "loss": 3.9369,
      "loss/crossentropy": 1.7454912066459656,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17953843623399734,
      "step": 27426
    },
    {
      "epoch": 0.54856,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.0097412109375,
      "learning_rate": 0.0001,
      "loss": 4.2128,
      "loss/crossentropy": 2.327135920524597,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2150314748287201,
      "step": 27428
    },
    {
      "epoch": 0.5486,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.008746083577473958,
      "learning_rate": 0.0001,
      "loss": 4.1777,
      "loss/crossentropy": 2.273073196411133,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20920293778181076,
      "step": 27430
    },
    {
      "epoch": 0.54864,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0080718994140625,
      "learning_rate": 0.0001,
      "loss": 3.9236,
      "loss/crossentropy": 1.474639356136322,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16699142009019852,
      "step": 27432
    },
    {
      "epoch": 0.54868,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.01053466796875,
      "learning_rate": 0.0001,
      "loss": 4.1162,
      "loss/crossentropy": 2.116548180580139,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.225176103413105,
      "step": 27434
    },
    {
      "epoch": 0.54872,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.013060506184895833,
      "learning_rate": 0.0001,
      "loss": 3.7837,
      "loss/crossentropy": 2.1261587142944336,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20171497762203217,
      "step": 27436
    },
    {
      "epoch": 0.54876,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.012074534098307292,
      "learning_rate": 0.0001,
      "loss": 4.1137,
      "loss/crossentropy": 2.0924129486083984,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2012445107102394,
      "step": 27438
    },
    {
      "epoch": 0.5488,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.011008453369140626,
      "learning_rate": 0.0001,
      "loss": 3.934,
      "loss/crossentropy": 1.7499247789382935,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17569267749786377,
      "step": 27440
    },
    {
      "epoch": 0.54884,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0103424072265625,
      "learning_rate": 0.0001,
      "loss": 3.7889,
      "loss/crossentropy": 2.0465195178985596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17263885587453842,
      "step": 27442
    },
    {
      "epoch": 0.54888,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0085113525390625,
      "learning_rate": 0.0001,
      "loss": 3.873,
      "loss/crossentropy": 1.8351019620895386,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1929614618420601,
      "step": 27444
    },
    {
      "epoch": 0.54892,
      "grad_norm": 1.875,
      "grad_norm_var": 0.008119455973307292,
      "learning_rate": 0.0001,
      "loss": 3.7952,
      "loss/crossentropy": 1.8877743482589722,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18555018305778503,
      "step": 27446
    },
    {
      "epoch": 0.54896,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.009757486979166667,
      "learning_rate": 0.0001,
      "loss": 3.7006,
      "loss/crossentropy": 1.7358683943748474,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18054651468992233,
      "step": 27448
    },
    {
      "epoch": 0.549,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.009178670247395833,
      "learning_rate": 0.0001,
      "loss": 3.9719,
      "loss/crossentropy": 2.0366458892822266,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2056865617632866,
      "step": 27450
    },
    {
      "epoch": 0.54904,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0085113525390625,
      "learning_rate": 0.0001,
      "loss": 4.0996,
      "loss/crossentropy": 2.078625202178955,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1962544023990631,
      "step": 27452
    },
    {
      "epoch": 0.54908,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.009590657552083333,
      "learning_rate": 0.0001,
      "loss": 3.6335,
      "loss/crossentropy": 1.6732112765312195,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1725183129310608,
      "step": 27454
    },
    {
      "epoch": 0.54912,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.009419759114583334,
      "learning_rate": 0.0001,
      "loss": 3.513,
      "loss/crossentropy": 1.8087647557258606,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1810692474246025,
      "step": 27456
    },
    {
      "epoch": 0.54916,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0110504150390625,
      "learning_rate": 0.0001,
      "loss": 4.2636,
      "loss/crossentropy": 2.087713837623596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20880009979009628,
      "step": 27458
    },
    {
      "epoch": 0.5492,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010367838541666667,
      "learning_rate": 0.0001,
      "loss": 4.1659,
      "loss/crossentropy": 1.8009834289550781,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19154398143291473,
      "step": 27460
    },
    {
      "epoch": 0.54924,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.010994211832682291,
      "learning_rate": 0.0001,
      "loss": 4.1084,
      "loss/crossentropy": 2.272579550743103,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21466071158647537,
      "step": 27462
    },
    {
      "epoch": 0.54928,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.010518391927083334,
      "learning_rate": 0.0001,
      "loss": 3.9836,
      "loss/crossentropy": 2.001379668712616,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1841510310769081,
      "step": 27464
    },
    {
      "epoch": 0.54932,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.008973948160807292,
      "learning_rate": 0.0001,
      "loss": 4.2364,
      "loss/crossentropy": 2.197210907936096,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20746250450611115,
      "step": 27466
    },
    {
      "epoch": 0.54936,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008153279622395834,
      "learning_rate": 0.0001,
      "loss": 4.0393,
      "loss/crossentropy": 1.6839552521705627,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19475190341472626,
      "step": 27468
    },
    {
      "epoch": 0.5494,
      "grad_norm": 2.640625,
      "grad_norm_var": 0.03849995930989583,
      "learning_rate": 0.0001,
      "loss": 3.8291,
      "loss/crossentropy": 1.8280132412910461,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19640739262104034,
      "step": 27470
    },
    {
      "epoch": 0.54944,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.03960367838541667,
      "learning_rate": 0.0001,
      "loss": 3.8258,
      "loss/crossentropy": 2.082128942012787,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20284345746040344,
      "step": 27472
    },
    {
      "epoch": 0.54948,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.04104181925455729,
      "learning_rate": 0.0001,
      "loss": 4.1464,
      "loss/crossentropy": 2.346489429473877,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1972089186310768,
      "step": 27474
    },
    {
      "epoch": 0.54952,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.041751861572265625,
      "learning_rate": 0.0001,
      "loss": 4.1133,
      "loss/crossentropy": 2.431715250015259,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2505635917186737,
      "step": 27476
    },
    {
      "epoch": 0.54956,
      "grad_norm": 2.0,
      "grad_norm_var": 0.03922907511393229,
      "learning_rate": 0.0001,
      "loss": 4.1596,
      "loss/crossentropy": 2.012192726135254,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20177678763866425,
      "step": 27478
    },
    {
      "epoch": 0.5496,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.040726725260416666,
      "learning_rate": 0.0001,
      "loss": 3.9623,
      "loss/crossentropy": 2.1662270426750183,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1866678223013878,
      "step": 27480
    },
    {
      "epoch": 0.54964,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.04033381144205729,
      "learning_rate": 0.0001,
      "loss": 4.1946,
      "loss/crossentropy": 2.331666111946106,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2118167132139206,
      "step": 27482
    },
    {
      "epoch": 0.54968,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.042429351806640626,
      "learning_rate": 0.0001,
      "loss": 4.084,
      "loss/crossentropy": 2.371376156806946,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20595970004796982,
      "step": 27484
    },
    {
      "epoch": 0.54972,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.008780670166015626,
      "learning_rate": 0.0001,
      "loss": 3.9568,
      "loss/crossentropy": 1.8372794389724731,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18964159488677979,
      "step": 27486
    },
    {
      "epoch": 0.54976,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.007865142822265626,
      "learning_rate": 0.0001,
      "loss": 3.9367,
      "loss/crossentropy": 1.9252594709396362,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19232141971588135,
      "step": 27488
    },
    {
      "epoch": 0.5498,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.00655517578125,
      "learning_rate": 0.0001,
      "loss": 3.9779,
      "loss/crossentropy": 1.9566543102264404,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20439066737890244,
      "step": 27490
    },
    {
      "epoch": 0.54984,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.006776682535807292,
      "learning_rate": 0.0001,
      "loss": 3.8192,
      "loss/crossentropy": 1.8082122802734375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1651742085814476,
      "step": 27492
    },
    {
      "epoch": 0.54988,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.006834920247395833,
      "learning_rate": 0.0001,
      "loss": 4.0115,
      "loss/crossentropy": 1.9120944738388062,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.181972436606884,
      "step": 27494
    },
    {
      "epoch": 0.54992,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.005812327067057292,
      "learning_rate": 0.0001,
      "loss": 3.9873,
      "loss/crossentropy": 2.134087026119232,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19440361112356186,
      "step": 27496
    },
    {
      "epoch": 0.54996,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.007039388020833333,
      "learning_rate": 0.0001,
      "loss": 3.6912,
      "loss/crossentropy": 1.8519493341445923,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16762658208608627,
      "step": 27498
    },
    {
      "epoch": 0.55,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.0065093994140625,
      "learning_rate": 0.0001,
      "loss": 3.6277,
      "loss/crossentropy": 2.10863196849823,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21393945068120956,
      "step": 27500
    },
    {
      "epoch": 0.55004,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.011252593994140626,
      "learning_rate": 0.0001,
      "loss": 4.3002,
      "loss/crossentropy": 2.226866364479065,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21370790153741837,
      "step": 27502
    },
    {
      "epoch": 0.55008,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.011749013264973959,
      "learning_rate": 0.0001,
      "loss": 3.8337,
      "loss/crossentropy": 1.5916427373886108,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16826944053173065,
      "step": 27504
    },
    {
      "epoch": 0.55012,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.011677805582682292,
      "learning_rate": 0.0001,
      "loss": 4.2064,
      "loss/crossentropy": 2.212289810180664,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20235298573970795,
      "step": 27506
    },
    {
      "epoch": 0.55016,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.009698232014973959,
      "learning_rate": 0.0001,
      "loss": 3.8529,
      "loss/crossentropy": 1.9228017926216125,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19821560382843018,
      "step": 27508
    },
    {
      "epoch": 0.5502,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.010247548421223959,
      "learning_rate": 0.0001,
      "loss": 3.9431,
      "loss/crossentropy": 1.8887941241264343,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20294177532196045,
      "step": 27510
    },
    {
      "epoch": 0.55024,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.010456339518229166,
      "learning_rate": 0.0001,
      "loss": 4.2666,
      "loss/crossentropy": 2.2699393033981323,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22084251046180725,
      "step": 27512
    },
    {
      "epoch": 0.55028,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008101145426432291,
      "learning_rate": 0.0001,
      "loss": 4.0852,
      "loss/crossentropy": 1.9771258234977722,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1795499324798584,
      "step": 27514
    },
    {
      "epoch": 0.55032,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.007990519205729166,
      "learning_rate": 0.0001,
      "loss": 3.7655,
      "loss/crossentropy": 2.0192587971687317,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19595889747142792,
      "step": 27516
    },
    {
      "epoch": 0.55036,
      "grad_norm": 1.8671875,
      "grad_norm_var": 5.339900461832682,
      "learning_rate": 0.0001,
      "loss": 3.9772,
      "loss/crossentropy": 1.9080755710601807,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18093246221542358,
      "step": 27518
    },
    {
      "epoch": 0.5504,
      "grad_norm": 2.0,
      "grad_norm_var": 5.307067616780599,
      "learning_rate": 0.0001,
      "loss": 4.2234,
      "loss/crossentropy": 2.237337589263916,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21287637203931808,
      "step": 27520
    },
    {
      "epoch": 0.55044,
      "grad_norm": 1.8203125,
      "grad_norm_var": 5.324763743082682,
      "learning_rate": 0.0001,
      "loss": 3.8267,
      "loss/crossentropy": 1.986120343208313,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16889339685440063,
      "step": 27522
    },
    {
      "epoch": 0.55048,
      "grad_norm": 1.9140625,
      "grad_norm_var": 5.322076161702474,
      "learning_rate": 0.0001,
      "loss": 4.0203,
      "loss/crossentropy": 1.9601141810417175,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20309537649154663,
      "step": 27524
    },
    {
      "epoch": 0.55052,
      "grad_norm": 1.921875,
      "grad_norm_var": 5.332348378499349,
      "learning_rate": 0.0001,
      "loss": 4.0248,
      "loss/crossentropy": 1.8901299238204956,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18104882538318634,
      "step": 27526
    },
    {
      "epoch": 0.55056,
      "grad_norm": 1.890625,
      "grad_norm_var": 5.3486480712890625,
      "learning_rate": 0.0001,
      "loss": 4.1152,
      "loss/crossentropy": 2.204539656639099,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20814384520053864,
      "step": 27528
    },
    {
      "epoch": 0.5506,
      "grad_norm": 1.6796875,
      "grad_norm_var": 5.393595377604167,
      "learning_rate": 0.0001,
      "loss": 3.6623,
      "loss/crossentropy": 1.8643839955329895,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1773769110441208,
      "step": 27530
    },
    {
      "epoch": 0.55064,
      "grad_norm": 1.8359375,
      "grad_norm_var": 5.410757446289063,
      "learning_rate": 0.0001,
      "loss": 3.5404,
      "loss/crossentropy": 1.66867595911026,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1539328545331955,
      "step": 27532
    },
    {
      "epoch": 0.55068,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.0146484375,
      "learning_rate": 0.0001,
      "loss": 3.87,
      "loss/crossentropy": 1.8830199241638184,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18702629208564758,
      "step": 27534
    },
    {
      "epoch": 0.55072,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.007793935139973959,
      "learning_rate": 0.0001,
      "loss": 4.0369,
      "loss/crossentropy": 1.8840887546539307,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17654889076948166,
      "step": 27536
    },
    {
      "epoch": 0.55076,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.008196767171223958,
      "learning_rate": 0.0001,
      "loss": 4.2076,
      "loss/crossentropy": 1.9453116655349731,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1726330816745758,
      "step": 27538
    },
    {
      "epoch": 0.5508,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.008133951822916667,
      "learning_rate": 0.0001,
      "loss": 3.9739,
      "loss/crossentropy": 2.0022284388542175,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18195754289627075,
      "step": 27540
    },
    {
      "epoch": 0.55084,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0090240478515625,
      "learning_rate": 0.0001,
      "loss": 4.2032,
      "loss/crossentropy": 2.164666533470154,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20948568731546402,
      "step": 27542
    },
    {
      "epoch": 0.55088,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.00947265625,
      "learning_rate": 0.0001,
      "loss": 4.0897,
      "loss/crossentropy": 1.8899564146995544,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1793677881360054,
      "step": 27544
    },
    {
      "epoch": 0.55092,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.008648427327473958,
      "learning_rate": 0.0001,
      "loss": 4.0669,
      "loss/crossentropy": 1.914944589138031,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21741337329149246,
      "step": 27546
    },
    {
      "epoch": 0.55096,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0091552734375,
      "learning_rate": 0.0001,
      "loss": 3.9413,
      "loss/crossentropy": 1.8026766180992126,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1790105178952217,
      "step": 27548
    },
    {
      "epoch": 0.551,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0085601806640625,
      "learning_rate": 0.0001,
      "loss": 4.2072,
      "loss/crossentropy": 2.1936656832695007,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21415145695209503,
      "step": 27550
    },
    {
      "epoch": 0.55104,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0077056884765625,
      "learning_rate": 0.0001,
      "loss": 4.0292,
      "loss/crossentropy": 1.841277301311493,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19467134028673172,
      "step": 27552
    },
    {
      "epoch": 0.55108,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0077056884765625,
      "learning_rate": 0.0001,
      "loss": 4.1539,
      "loss/crossentropy": 1.9853840470314026,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18696378916502,
      "step": 27554
    },
    {
      "epoch": 0.55112,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.006740061442057291,
      "learning_rate": 0.0001,
      "loss": 4.0032,
      "loss/crossentropy": 2.213033676147461,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19526077806949615,
      "step": 27556
    },
    {
      "epoch": 0.55116,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.00665283203125,
      "learning_rate": 0.0001,
      "loss": 3.9773,
      "loss/crossentropy": 1.8020763397216797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1804225742816925,
      "step": 27558
    },
    {
      "epoch": 0.5512,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.005838775634765625,
      "learning_rate": 0.0001,
      "loss": 4.0551,
      "loss/crossentropy": 1.899497926235199,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19439035654067993,
      "step": 27560
    },
    {
      "epoch": 0.55124,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.00552978515625,
      "learning_rate": 0.0001,
      "loss": 3.9789,
      "loss/crossentropy": 1.9123604893684387,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18431273847818375,
      "step": 27562
    },
    {
      "epoch": 0.55128,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.004510243733723958,
      "learning_rate": 0.0001,
      "loss": 4.3084,
      "loss/crossentropy": 2.3199894428253174,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21417556703090668,
      "step": 27564
    },
    {
      "epoch": 0.55132,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.0055653889973958336,
      "learning_rate": 0.0001,
      "loss": 3.8603,
      "loss/crossentropy": 2.3096247911453247,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19787238538265228,
      "step": 27566
    },
    {
      "epoch": 0.55136,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.005861155192057292,
      "learning_rate": 0.0001,
      "loss": 4.083,
      "loss/crossentropy": 2.2241322994232178,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21322394907474518,
      "step": 27568
    },
    {
      "epoch": 0.5514,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.005850982666015625,
      "learning_rate": 0.0001,
      "loss": 3.8552,
      "loss/crossentropy": 2.3283156156539917,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21560409665107727,
      "step": 27570
    },
    {
      "epoch": 0.55144,
      "grad_norm": 2.0,
      "grad_norm_var": 0.005352528889973959,
      "learning_rate": 0.0001,
      "loss": 4.0709,
      "loss/crossentropy": 2.07872211933136,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20099805295467377,
      "step": 27572
    },
    {
      "epoch": 0.55148,
      "grad_norm": 2.0,
      "grad_norm_var": 0.005273183186848958,
      "learning_rate": 0.0001,
      "loss": 4.0868,
      "loss/crossentropy": 2.1145899295806885,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19824448227882385,
      "step": 27574
    },
    {
      "epoch": 0.55152,
      "grad_norm": 1.640625,
      "grad_norm_var": 0.012674967447916666,
      "learning_rate": 0.0001,
      "loss": 3.4628,
      "loss/crossentropy": 1.5884234309196472,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1501767933368683,
      "step": 27576
    },
    {
      "epoch": 0.55156,
      "grad_norm": 1.875,
      "grad_norm_var": 0.0121002197265625,
      "learning_rate": 0.0001,
      "loss": 3.6905,
      "loss/crossentropy": 1.9045534133911133,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19776541739702225,
      "step": 27578
    },
    {
      "epoch": 0.5516,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.010337066650390626,
      "learning_rate": 0.0001,
      "loss": 3.862,
      "loss/crossentropy": 1.6315965056419373,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15820547938346863,
      "step": 27580
    },
    {
      "epoch": 0.55164,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.011790974934895834,
      "learning_rate": 0.0001,
      "loss": 4.298,
      "loss/crossentropy": 2.1012359261512756,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19510812312364578,
      "step": 27582
    },
    {
      "epoch": 0.55168,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.012261708577473959,
      "learning_rate": 0.0001,
      "loss": 3.8833,
      "loss/crossentropy": 1.8365219831466675,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16663482040166855,
      "step": 27584
    },
    {
      "epoch": 0.55172,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.012181599934895834,
      "learning_rate": 0.0001,
      "loss": 3.6868,
      "loss/crossentropy": 1.6274245977401733,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1516888439655304,
      "step": 27586
    },
    {
      "epoch": 0.55176,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.011529286702473959,
      "learning_rate": 0.0001,
      "loss": 3.8555,
      "loss/crossentropy": 1.6476022601127625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17268501967191696,
      "step": 27588
    },
    {
      "epoch": 0.5518,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.010400390625,
      "learning_rate": 0.0001,
      "loss": 3.7764,
      "loss/crossentropy": 2.135971188545227,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20169296115636826,
      "step": 27590
    },
    {
      "epoch": 0.55184,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.006640625,
      "learning_rate": 0.0001,
      "loss": 3.7435,
      "loss/crossentropy": 2.2080432176589966,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20463210344314575,
      "step": 27592
    },
    {
      "epoch": 0.55188,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008131663004557291,
      "learning_rate": 0.0001,
      "loss": 4.1494,
      "loss/crossentropy": 1.953204333782196,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18644939363002777,
      "step": 27594
    },
    {
      "epoch": 0.55192,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.007429758707682292,
      "learning_rate": 0.0001,
      "loss": 4.1851,
      "loss/crossentropy": 1.7578163146972656,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1816823109984398,
      "step": 27596
    },
    {
      "epoch": 0.55196,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.006193033854166667,
      "learning_rate": 0.0001,
      "loss": 3.7816,
      "loss/crossentropy": 1.9401060938835144,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1864597350358963,
      "step": 27598
    },
    {
      "epoch": 0.552,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.009455362955729166,
      "learning_rate": 0.0001,
      "loss": 3.8935,
      "loss/crossentropy": 2.03184574842453,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1829465553164482,
      "step": 27600
    },
    {
      "epoch": 0.55204,
      "grad_norm": 1.7421875,
      "grad_norm_var": 0.010365549723307292,
      "learning_rate": 0.0001,
      "loss": 3.6649,
      "loss/crossentropy": 2.15334689617157,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18497572094202042,
      "step": 27602
    },
    {
      "epoch": 0.55208,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.012918853759765625,
      "learning_rate": 0.0001,
      "loss": 4.0528,
      "loss/crossentropy": 1.894173800945282,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1757444217801094,
      "step": 27604
    },
    {
      "epoch": 0.55212,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0134765625,
      "learning_rate": 0.0001,
      "loss": 3.9785,
      "loss/crossentropy": 1.8364281058311462,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17648981511592865,
      "step": 27606
    },
    {
      "epoch": 0.55216,
      "grad_norm": 1.875,
      "grad_norm_var": 0.012882232666015625,
      "learning_rate": 0.0001,
      "loss": 3.8651,
      "loss/crossentropy": 1.5523585081100464,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1650981828570366,
      "step": 27608
    },
    {
      "epoch": 0.5522,
      "grad_norm": 2.0,
      "grad_norm_var": 0.013887532552083333,
      "learning_rate": 0.0001,
      "loss": 3.8205,
      "loss/crossentropy": 2.1311700344085693,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19313734769821167,
      "step": 27610
    },
    {
      "epoch": 0.55224,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.014149729410807292,
      "learning_rate": 0.0001,
      "loss": 3.8613,
      "loss/crossentropy": 2.23338782787323,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21325203776359558,
      "step": 27612
    },
    {
      "epoch": 0.55228,
      "grad_norm": 1.75,
      "grad_norm_var": 0.013736724853515625,
      "learning_rate": 0.0001,
      "loss": 3.7978,
      "loss/crossentropy": 2.0709031224250793,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17793143540620804,
      "step": 27614
    },
    {
      "epoch": 0.55232,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.010505167643229167,
      "learning_rate": 0.0001,
      "loss": 3.8325,
      "loss/crossentropy": 1.758750557899475,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1758880391716957,
      "step": 27616
    },
    {
      "epoch": 0.55236,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.010353342692057291,
      "learning_rate": 0.0001,
      "loss": 3.695,
      "loss/crossentropy": 1.8869126439094543,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16803206503391266,
      "step": 27618
    },
    {
      "epoch": 0.5524,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.010982004801432292,
      "learning_rate": 0.0001,
      "loss": 4.1633,
      "loss/crossentropy": 2.5184292793273926,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22967734932899475,
      "step": 27620
    },
    {
      "epoch": 0.55244,
      "grad_norm": 3.859375,
      "grad_norm_var": 0.25396906534830727,
      "learning_rate": 0.0001,
      "loss": 3.9171,
      "loss/crossentropy": 2.2767695784568787,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21888408064842224,
      "step": 27622
    },
    {
      "epoch": 0.55248,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.2516009012858073,
      "learning_rate": 0.0001,
      "loss": 4.0942,
      "loss/crossentropy": 2.015432834625244,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19384412467479706,
      "step": 27624
    },
    {
      "epoch": 0.55252,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.2524879455566406,
      "learning_rate": 0.0001,
      "loss": 4.1693,
      "loss/crossentropy": 1.9935010075569153,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20147496461868286,
      "step": 27626
    },
    {
      "epoch": 0.55256,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.24866714477539062,
      "learning_rate": 0.0001,
      "loss": 4.1237,
      "loss/crossentropy": 2.0798122882843018,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19859570264816284,
      "step": 27628
    },
    {
      "epoch": 0.5526,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.24353001912434896,
      "learning_rate": 0.0001,
      "loss": 3.7006,
      "loss/crossentropy": 1.871802031993866,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1913897469639778,
      "step": 27630
    },
    {
      "epoch": 0.55264,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.2379351298014323,
      "learning_rate": 0.0001,
      "loss": 3.9546,
      "loss/crossentropy": 2.184646248817444,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1939481720328331,
      "step": 27632
    },
    {
      "epoch": 0.55268,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.23611831665039062,
      "learning_rate": 0.0001,
      "loss": 3.8362,
      "loss/crossentropy": 2.0267680883407593,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20360219478607178,
      "step": 27634
    },
    {
      "epoch": 0.55272,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.2367754618326823,
      "learning_rate": 0.0001,
      "loss": 4.2263,
      "loss/crossentropy": 2.2218815088272095,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20947173982858658,
      "step": 27636
    },
    {
      "epoch": 0.55276,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.010217030843098959,
      "learning_rate": 0.0001,
      "loss": 4.2844,
      "loss/crossentropy": 1.982936441898346,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1881447434425354,
      "step": 27638
    },
    {
      "epoch": 0.5528,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.008601888020833334,
      "learning_rate": 0.0001,
      "loss": 3.8908,
      "loss/crossentropy": 1.8088389039039612,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1668427437543869,
      "step": 27640
    },
    {
      "epoch": 0.55284,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.008735911051432291,
      "learning_rate": 0.0001,
      "loss": 4.214,
      "loss/crossentropy": 2.0886768102645874,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1919582262635231,
      "step": 27642
    },
    {
      "epoch": 0.55288,
      "grad_norm": 2.90625,
      "grad_norm_var": 0.06743748982747395,
      "learning_rate": 0.0001,
      "loss": 4.3581,
      "loss/crossentropy": 2.0147945880889893,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20268514752388,
      "step": 27644
    },
    {
      "epoch": 0.55292,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.06584447224934896,
      "learning_rate": 0.0001,
      "loss": 4.0147,
      "loss/crossentropy": 1.8306025266647339,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1949317753314972,
      "step": 27646
    },
    {
      "epoch": 0.55296,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.06684137980143229,
      "learning_rate": 0.0001,
      "loss": 4.0944,
      "loss/crossentropy": 2.201099991798401,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19578266888856888,
      "step": 27648
    },
    {
      "epoch": 0.553,
      "grad_norm": 1.875,
      "grad_norm_var": 0.06663792928059896,
      "learning_rate": 0.0001,
      "loss": 3.9902,
      "loss/crossentropy": 2.159493327140808,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18864253163337708,
      "step": 27650
    },
    {
      "epoch": 0.55304,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.06830215454101562,
      "learning_rate": 0.0001,
      "loss": 3.9129,
      "loss/crossentropy": 2.0548607110977173,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18820471316576004,
      "step": 27652
    },
    {
      "epoch": 0.55308,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.07153701782226562,
      "learning_rate": 0.0001,
      "loss": 3.4452,
      "loss/crossentropy": 1.8324944972991943,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19259792566299438,
      "step": 27654
    },
    {
      "epoch": 0.55312,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.10803120930989583,
      "learning_rate": 0.0001,
      "loss": 4.2944,
      "loss/crossentropy": 1.8857054114341736,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1885119453072548,
      "step": 27656
    },
    {
      "epoch": 0.55316,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.10672200520833333,
      "learning_rate": 0.0001,
      "loss": 4.1616,
      "loss/crossentropy": 2.0037535429000854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1955013945698738,
      "step": 27658
    },
    {
      "epoch": 0.5532,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.05010579427083333,
      "learning_rate": 0.0001,
      "loss": 3.8527,
      "loss/crossentropy": 1.945756196975708,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17360417544841766,
      "step": 27660
    },
    {
      "epoch": 0.55324,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.05109456380208333,
      "learning_rate": 0.0001,
      "loss": 3.8767,
      "loss/crossentropy": 1.8675792217254639,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17810215801000595,
      "step": 27662
    },
    {
      "epoch": 0.55328,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.05110041300455729,
      "learning_rate": 0.0001,
      "loss": 4.1631,
      "loss/crossentropy": 1.9741141200065613,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17763611674308777,
      "step": 27664
    },
    {
      "epoch": 0.55332,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.05176493326822917,
      "learning_rate": 0.0001,
      "loss": 3.9846,
      "loss/crossentropy": 2.123018801212311,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21710921823978424,
      "step": 27666
    },
    {
      "epoch": 0.55336,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.05128758748372396,
      "learning_rate": 0.0001,
      "loss": 3.7929,
      "loss/crossentropy": 2.043752074241638,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1959235817193985,
      "step": 27668
    },
    {
      "epoch": 0.5534,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.04549560546875,
      "learning_rate": 0.0001,
      "loss": 4.0634,
      "loss/crossentropy": 1.9128360748291016,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18556588888168335,
      "step": 27670
    },
    {
      "epoch": 0.55344,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.008585611979166666,
      "learning_rate": 0.0001,
      "loss": 4.2678,
      "loss/crossentropy": 2.1376100182533264,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1985430046916008,
      "step": 27672
    },
    {
      "epoch": 0.55348,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0088775634765625,
      "learning_rate": 0.0001,
      "loss": 4.0457,
      "loss/crossentropy": 2.106353759765625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20115973055362701,
      "step": 27674
    },
    {
      "epoch": 0.55352,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.008316802978515624,
      "learning_rate": 0.0001,
      "loss": 4.0802,
      "loss/crossentropy": 2.5206472873687744,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21297650784254074,
      "step": 27676
    },
    {
      "epoch": 0.55356,
      "grad_norm": 1.875,
      "grad_norm_var": 0.008229319254557292,
      "learning_rate": 0.0001,
      "loss": 3.9164,
      "loss/crossentropy": 1.8287059664726257,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16045409440994263,
      "step": 27678
    },
    {
      "epoch": 0.5536,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.007610829671223959,
      "learning_rate": 0.0001,
      "loss": 3.9755,
      "loss/crossentropy": 2.103208303451538,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20481672883033752,
      "step": 27680
    },
    {
      "epoch": 0.55364,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.006689453125,
      "learning_rate": 0.0001,
      "loss": 4.0925,
      "loss/crossentropy": 1.6746403574943542,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16627443581819534,
      "step": 27682
    },
    {
      "epoch": 0.55368,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.005954742431640625,
      "learning_rate": 0.0001,
      "loss": 3.9342,
      "loss/crossentropy": 1.9239201545715332,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19345758855342865,
      "step": 27684
    },
    {
      "epoch": 0.55372,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.006009674072265625,
      "learning_rate": 0.0001,
      "loss": 3.9512,
      "loss/crossentropy": 2.052439272403717,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18695925921201706,
      "step": 27686
    },
    {
      "epoch": 0.55376,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.005476633707682292,
      "learning_rate": 0.0001,
      "loss": 4.3674,
      "loss/crossentropy": 2.4474085569381714,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21558378636837006,
      "step": 27688
    },
    {
      "epoch": 0.5538,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.006062825520833333,
      "learning_rate": 0.0001,
      "loss": 3.6432,
      "loss/crossentropy": 1.8521843552589417,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16763968020677567,
      "step": 27690
    },
    {
      "epoch": 0.55384,
      "grad_norm": 1.875,
      "grad_norm_var": 0.0060455322265625,
      "learning_rate": 0.0001,
      "loss": 3.9086,
      "loss/crossentropy": 1.888016939163208,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.185853011906147,
      "step": 27692
    },
    {
      "epoch": 0.55388,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.005830637613932292,
      "learning_rate": 0.0001,
      "loss": 3.8378,
      "loss/crossentropy": 1.7538975477218628,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1854776367545128,
      "step": 27694
    },
    {
      "epoch": 0.55392,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.005936431884765625,
      "learning_rate": 0.0001,
      "loss": 4.1631,
      "loss/crossentropy": 2.259602427482605,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21480850875377655,
      "step": 27696
    },
    {
      "epoch": 0.55396,
      "grad_norm": 1.875,
      "grad_norm_var": 0.006860097249348958,
      "learning_rate": 0.0001,
      "loss": 4.0021,
      "loss/crossentropy": 2.0374972820281982,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20546026527881622,
      "step": 27698
    },
    {
      "epoch": 0.554,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.006392415364583333,
      "learning_rate": 0.0001,
      "loss": 4.1568,
      "loss/crossentropy": 2.15248441696167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19681835174560547,
      "step": 27700
    },
    {
      "epoch": 0.55404,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.0070798238118489586,
      "learning_rate": 0.0001,
      "loss": 3.7793,
      "loss/crossentropy": 1.8983227610588074,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18331478536128998,
      "step": 27702
    },
    {
      "epoch": 0.55408,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.00849609375,
      "learning_rate": 0.0001,
      "loss": 3.6701,
      "loss/crossentropy": 1.7995752692222595,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17722319066524506,
      "step": 27704
    },
    {
      "epoch": 0.55412,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.00760498046875,
      "learning_rate": 0.0001,
      "loss": 4.0987,
      "loss/crossentropy": 1.89347642660141,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17802709341049194,
      "step": 27706
    },
    {
      "epoch": 0.55416,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.008343251546223958,
      "learning_rate": 0.0001,
      "loss": 3.8648,
      "loss/crossentropy": 2.024496912956238,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18508757650852203,
      "step": 27708
    },
    {
      "epoch": 0.5542,
      "grad_norm": 1.7265625,
      "grad_norm_var": 0.010731760660807292,
      "learning_rate": 0.0001,
      "loss": 3.7977,
      "loss/crossentropy": 2.0142337679862976,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18837066739797592,
      "step": 27710
    },
    {
      "epoch": 0.55424,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.009913889567057292,
      "learning_rate": 0.0001,
      "loss": 3.9401,
      "loss/crossentropy": 1.75287264585495,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1797751560807228,
      "step": 27712
    },
    {
      "epoch": 0.55428,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.007861073811848958,
      "learning_rate": 0.0001,
      "loss": 4.2331,
      "loss/crossentropy": 2.204139769077301,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17446620762348175,
      "step": 27714
    },
    {
      "epoch": 0.55432,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.009317779541015625,
      "learning_rate": 0.0001,
      "loss": 4.1056,
      "loss/crossentropy": 2.0878941416740417,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19087567925453186,
      "step": 27716
    },
    {
      "epoch": 0.55436,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0128570556640625,
      "learning_rate": 0.0001,
      "loss": 4.2309,
      "loss/crossentropy": 2.2289849519729614,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21995704621076584,
      "step": 27718
    },
    {
      "epoch": 0.5544,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.013140614827473958,
      "learning_rate": 0.0001,
      "loss": 4.1908,
      "loss/crossentropy": 2.2673051357269287,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19668636471033096,
      "step": 27720
    },
    {
      "epoch": 0.55444,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0144439697265625,
      "learning_rate": 0.0001,
      "loss": 3.9146,
      "loss/crossentropy": 1.7722193598747253,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17487196624279022,
      "step": 27722
    },
    {
      "epoch": 0.55448,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0147613525390625,
      "learning_rate": 0.0001,
      "loss": 4.2427,
      "loss/crossentropy": 1.8388479351997375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20283987373113632,
      "step": 27724
    },
    {
      "epoch": 0.55452,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.010334269205729166,
      "learning_rate": 0.0001,
      "loss": 4.1243,
      "loss/crossentropy": 2.2758948802948,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20532135665416718,
      "step": 27726
    },
    {
      "epoch": 0.55456,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.009037017822265625,
      "learning_rate": 0.0001,
      "loss": 4.0048,
      "loss/crossentropy": 2.272206664085388,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20994101464748383,
      "step": 27728
    },
    {
      "epoch": 0.5546,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.011824289957682291,
      "learning_rate": 0.0001,
      "loss": 4.1122,
      "loss/crossentropy": 2.0169429779052734,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21998245269060135,
      "step": 27730
    },
    {
      "epoch": 0.55464,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.011668904622395834,
      "learning_rate": 0.0001,
      "loss": 4.0199,
      "loss/crossentropy": 2.0700154900550842,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20395687222480774,
      "step": 27732
    },
    {
      "epoch": 0.55468,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.010237630208333333,
      "learning_rate": 0.0001,
      "loss": 4.2295,
      "loss/crossentropy": 2.2826544046401978,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1890336349606514,
      "step": 27734
    },
    {
      "epoch": 0.55472,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.010503896077473958,
      "learning_rate": 0.0001,
      "loss": 3.8089,
      "loss/crossentropy": 2.2162158489227295,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2044035568833351,
      "step": 27736
    },
    {
      "epoch": 0.55476,
      "grad_norm": 2.0,
      "grad_norm_var": 0.009894816080729167,
      "learning_rate": 0.0001,
      "loss": 4.2302,
      "loss/crossentropy": 1.8466681838035583,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18051490187644958,
      "step": 27738
    },
    {
      "epoch": 0.5548,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.010469309488932292,
      "learning_rate": 0.0001,
      "loss": 3.8307,
      "loss/crossentropy": 2.19679594039917,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19721779227256775,
      "step": 27740
    },
    {
      "epoch": 0.55484,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.010109202067057291,
      "learning_rate": 0.0001,
      "loss": 4.1634,
      "loss/crossentropy": 2.0091158151626587,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19828540831804276,
      "step": 27742
    },
    {
      "epoch": 0.55488,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.010261027018229167,
      "learning_rate": 0.0001,
      "loss": 4.1095,
      "loss/crossentropy": 1.9434176087379456,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20168793946504593,
      "step": 27744
    },
    {
      "epoch": 0.55492,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.0070302327473958336,
      "learning_rate": 0.0001,
      "loss": 3.9754,
      "loss/crossentropy": 2.365121006965637,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20621124655008316,
      "step": 27746
    },
    {
      "epoch": 0.55496,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.011714426676432292,
      "learning_rate": 0.0001,
      "loss": 4.1344,
      "loss/crossentropy": 2.4154850244522095,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21647262573242188,
      "step": 27748
    },
    {
      "epoch": 0.555,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.01280517578125,
      "learning_rate": 0.0001,
      "loss": 3.8326,
      "loss/crossentropy": 1.9805672764778137,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1818150281906128,
      "step": 27750
    },
    {
      "epoch": 0.55504,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.011188761393229166,
      "learning_rate": 0.0001,
      "loss": 4.0105,
      "loss/crossentropy": 2.1163841485977173,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20045197010040283,
      "step": 27752
    },
    {
      "epoch": 0.55508,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010008748372395833,
      "learning_rate": 0.0001,
      "loss": 3.8754,
      "loss/crossentropy": 2.3555898666381836,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19626620411872864,
      "step": 27754
    },
    {
      "epoch": 0.55512,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.007883453369140625,
      "learning_rate": 0.0001,
      "loss": 3.8355,
      "loss/crossentropy": 1.8587325811386108,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18491162359714508,
      "step": 27756
    },
    {
      "epoch": 0.55516,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.009138743082682291,
      "learning_rate": 0.0001,
      "loss": 3.9636,
      "loss/crossentropy": 1.9335539937019348,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18707312643527985,
      "step": 27758
    },
    {
      "epoch": 0.5552,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.010953776041666667,
      "learning_rate": 0.0001,
      "loss": 4.0691,
      "loss/crossentropy": 1.9938113689422607,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1832147091627121,
      "step": 27760
    },
    {
      "epoch": 0.55524,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.010106404622395834,
      "learning_rate": 0.0001,
      "loss": 4.1229,
      "loss/crossentropy": 1.796451210975647,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16813044995069504,
      "step": 27762
    },
    {
      "epoch": 0.55528,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.0072672526041666664,
      "learning_rate": 0.0001,
      "loss": 3.9291,
      "loss/crossentropy": 2.2282769680023193,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20037836581468582,
      "step": 27764
    },
    {
      "epoch": 0.55532,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.007088216145833334,
      "learning_rate": 0.0001,
      "loss": 4.0021,
      "loss/crossentropy": 2.3429712057113647,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21040766686201096,
      "step": 27766
    },
    {
      "epoch": 0.55536,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.006932576497395833,
      "learning_rate": 0.0001,
      "loss": 4.0902,
      "loss/crossentropy": 2.24351704120636,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19055639952421188,
      "step": 27768
    },
    {
      "epoch": 0.5554,
      "grad_norm": 2.0,
      "grad_norm_var": 0.007120768229166667,
      "learning_rate": 0.0001,
      "loss": 4.1444,
      "loss/crossentropy": 2.190573811531067,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1991223245859146,
      "step": 27770
    },
    {
      "epoch": 0.55544,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.008925120035807291,
      "learning_rate": 0.0001,
      "loss": 3.9567,
      "loss/crossentropy": 1.9847629070281982,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17740985751152039,
      "step": 27772
    },
    {
      "epoch": 0.55548,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.008565012613932292,
      "learning_rate": 0.0001,
      "loss": 3.8379,
      "loss/crossentropy": 2.0242174863815308,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18693319708108902,
      "step": 27774
    },
    {
      "epoch": 0.55552,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.008141835530598959,
      "learning_rate": 0.0001,
      "loss": 3.8359,
      "loss/crossentropy": 1.7930986881256104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17298821359872818,
      "step": 27776
    },
    {
      "epoch": 0.55556,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.009227498372395834,
      "learning_rate": 0.0001,
      "loss": 3.7851,
      "loss/crossentropy": 1.967544972896576,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1983192041516304,
      "step": 27778
    },
    {
      "epoch": 0.5556,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008280436197916666,
      "learning_rate": 0.0001,
      "loss": 4.2846,
      "loss/crossentropy": 2.353300929069519,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2090948149561882,
      "step": 27780
    },
    {
      "epoch": 0.55564,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.013826243082682292,
      "learning_rate": 0.0001,
      "loss": 4.0386,
      "loss/crossentropy": 2.1349263191223145,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19604195654392242,
      "step": 27782
    },
    {
      "epoch": 0.55568,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.013890584309895834,
      "learning_rate": 0.0001,
      "loss": 3.9092,
      "loss/crossentropy": 2.1047816276550293,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20156846940517426,
      "step": 27784
    },
    {
      "epoch": 0.55572,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.013213857014973959,
      "learning_rate": 0.0001,
      "loss": 3.8338,
      "loss/crossentropy": 1.9704868793487549,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18116163462400436,
      "step": 27786
    },
    {
      "epoch": 0.55576,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.013628896077473958,
      "learning_rate": 0.0001,
      "loss": 3.9997,
      "loss/crossentropy": 1.906118392944336,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18277255445718765,
      "step": 27788
    },
    {
      "epoch": 0.5558,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.013449859619140626,
      "learning_rate": 0.0001,
      "loss": 4.2688,
      "loss/crossentropy": 2.0571773648262024,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1843668520450592,
      "step": 27790
    },
    {
      "epoch": 0.55584,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.01390380859375,
      "learning_rate": 0.0001,
      "loss": 3.8647,
      "loss/crossentropy": 1.9324238300323486,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16451232880353928,
      "step": 27792
    },
    {
      "epoch": 0.55588,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.012589263916015624,
      "learning_rate": 0.0001,
      "loss": 4.1593,
      "loss/crossentropy": 2.27968692779541,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21742059290409088,
      "step": 27794
    },
    {
      "epoch": 0.55592,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.012804921468098958,
      "learning_rate": 0.0001,
      "loss": 4.3922,
      "loss/crossentropy": 2.20041286945343,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22774580121040344,
      "step": 27796
    },
    {
      "epoch": 0.55596,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.006239573160807292,
      "learning_rate": 0.0001,
      "loss": 3.9747,
      "loss/crossentropy": 2.258635997772217,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18686214089393616,
      "step": 27798
    },
    {
      "epoch": 0.556,
      "grad_norm": 1.875,
      "grad_norm_var": 0.006217193603515625,
      "learning_rate": 0.0001,
      "loss": 3.8514,
      "loss/crossentropy": 1.8728525638580322,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1724954918026924,
      "step": 27800
    },
    {
      "epoch": 0.55604,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.009789021809895833,
      "learning_rate": 0.0001,
      "loss": 4.3107,
      "loss/crossentropy": 1.7146453261375427,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18469443172216415,
      "step": 27802
    },
    {
      "epoch": 0.55608,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.009015909830729167,
      "learning_rate": 0.0001,
      "loss": 3.9967,
      "loss/crossentropy": 2.194948673248291,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1963738277554512,
      "step": 27804
    },
    {
      "epoch": 0.55612,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.0107177734375,
      "learning_rate": 0.0001,
      "loss": 3.7152,
      "loss/crossentropy": 1.7860903143882751,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16485057771205902,
      "step": 27806
    },
    {
      "epoch": 0.55616,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.011246490478515624,
      "learning_rate": 0.0001,
      "loss": 4.0281,
      "loss/crossentropy": 2.1505147218704224,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2449631690979004,
      "step": 27808
    },
    {
      "epoch": 0.5562,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011139933268229167,
      "learning_rate": 0.0001,
      "loss": 4.1753,
      "loss/crossentropy": 2.2144479751586914,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19644592702388763,
      "step": 27810
    },
    {
      "epoch": 0.55624,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.010796864827473959,
      "learning_rate": 0.0001,
      "loss": 4.1869,
      "loss/crossentropy": 2.2427414655685425,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20603332668542862,
      "step": 27812
    },
    {
      "epoch": 0.55628,
      "grad_norm": 1.6875,
      "grad_norm_var": 0.013533528645833333,
      "learning_rate": 0.0001,
      "loss": 3.8191,
      "loss/crossentropy": 1.9676510691642761,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16527333855628967,
      "step": 27814
    },
    {
      "epoch": 0.55632,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.014025624593098958,
      "learning_rate": 0.0001,
      "loss": 3.823,
      "loss/crossentropy": 1.9767807722091675,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1865728572010994,
      "step": 27816
    },
    {
      "epoch": 0.55636,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.008782704671223959,
      "learning_rate": 0.0001,
      "loss": 3.7126,
      "loss/crossentropy": 1.9406518936157227,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16727876663208008,
      "step": 27818
    },
    {
      "epoch": 0.5564,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.010550689697265626,
      "learning_rate": 0.0001,
      "loss": 3.9657,
      "loss/crossentropy": 2.0927204489707947,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2055317759513855,
      "step": 27820
    },
    {
      "epoch": 0.55644,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0100006103515625,
      "learning_rate": 0.0001,
      "loss": 3.825,
      "loss/crossentropy": 1.9616308212280273,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1885625645518303,
      "step": 27822
    },
    {
      "epoch": 0.55648,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.008890533447265625,
      "learning_rate": 0.0001,
      "loss": 3.7727,
      "loss/crossentropy": 1.7788268327713013,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1750224232673645,
      "step": 27824
    },
    {
      "epoch": 0.55652,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.009391021728515626,
      "learning_rate": 0.0001,
      "loss": 3.8066,
      "loss/crossentropy": 2.006068170070648,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18516544997692108,
      "step": 27826
    },
    {
      "epoch": 0.55656,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.012995402018229166,
      "learning_rate": 0.0001,
      "loss": 3.8937,
      "loss/crossentropy": 2.046180784702301,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17838794738054276,
      "step": 27828
    },
    {
      "epoch": 0.5566,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.010526275634765625,
      "learning_rate": 0.0001,
      "loss": 3.8813,
      "loss/crossentropy": 1.9033851027488708,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18221035599708557,
      "step": 27830
    },
    {
      "epoch": 0.55664,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.009415435791015624,
      "learning_rate": 0.0001,
      "loss": 4.128,
      "loss/crossentropy": 2.1586283445358276,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20454445481300354,
      "step": 27832
    },
    {
      "epoch": 0.55668,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.009645334879557292,
      "learning_rate": 0.0001,
      "loss": 4.0045,
      "loss/crossentropy": 2.383496642112732,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19787979125976562,
      "step": 27834
    },
    {
      "epoch": 0.55672,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.009150950113932292,
      "learning_rate": 0.0001,
      "loss": 3.9134,
      "loss/crossentropy": 2.052733063697815,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1916261911392212,
      "step": 27836
    },
    {
      "epoch": 0.55676,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.009281412760416666,
      "learning_rate": 0.0001,
      "loss": 4.2263,
      "loss/crossentropy": 2.1427736282348633,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1837182566523552,
      "step": 27838
    },
    {
      "epoch": 0.5568,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.009651438395182291,
      "learning_rate": 0.0001,
      "loss": 3.8118,
      "loss/crossentropy": 2.186544895172119,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20473113656044006,
      "step": 27840
    },
    {
      "epoch": 0.55684,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.007845052083333333,
      "learning_rate": 0.0001,
      "loss": 3.9411,
      "loss/crossentropy": 2.012458860874176,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20121898502111435,
      "step": 27842
    },
    {
      "epoch": 0.55688,
      "grad_norm": 2.0,
      "grad_norm_var": 0.00859375,
      "learning_rate": 0.0001,
      "loss": 3.8415,
      "loss/crossentropy": 1.9217159748077393,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1689399629831314,
      "step": 27844
    },
    {
      "epoch": 0.55692,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.007258097330729167,
      "learning_rate": 0.0001,
      "loss": 3.9051,
      "loss/crossentropy": 2.1690521240234375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21002651005983353,
      "step": 27846
    },
    {
      "epoch": 0.55696,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.006992340087890625,
      "learning_rate": 0.0001,
      "loss": 3.9794,
      "loss/crossentropy": 2.3649327754974365,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20836569368839264,
      "step": 27848
    },
    {
      "epoch": 0.557,
      "grad_norm": 2.390625,
      "grad_norm_var": 0.021735636393229167,
      "learning_rate": 0.0001,
      "loss": 4.1079,
      "loss/crossentropy": 2.118848145008087,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2462398260831833,
      "step": 27850
    },
    {
      "epoch": 0.55704,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.022043609619140626,
      "learning_rate": 0.0001,
      "loss": 4.1345,
      "loss/crossentropy": 1.9461410641670227,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1836184561252594,
      "step": 27852
    },
    {
      "epoch": 0.55708,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.021817779541015624,
      "learning_rate": 0.0001,
      "loss": 4.0911,
      "loss/crossentropy": 2.225842833518982,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19738706201314926,
      "step": 27854
    },
    {
      "epoch": 0.55712,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.020702870686848958,
      "learning_rate": 0.0001,
      "loss": 4.2437,
      "loss/crossentropy": 2.379333972930908,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21520506590604782,
      "step": 27856
    },
    {
      "epoch": 0.55716,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.020957183837890626,
      "learning_rate": 0.0001,
      "loss": 3.94,
      "loss/crossentropy": 1.789103090763092,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16888972371816635,
      "step": 27858
    },
    {
      "epoch": 0.5572,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.017464192708333333,
      "learning_rate": 0.0001,
      "loss": 4.1865,
      "loss/crossentropy": 2.1485098600387573,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2049245834350586,
      "step": 27860
    },
    {
      "epoch": 0.55724,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.017185211181640625,
      "learning_rate": 0.0001,
      "loss": 4.1494,
      "loss/crossentropy": 1.8173152804374695,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17991861701011658,
      "step": 27862
    },
    {
      "epoch": 0.55728,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.016206868489583335,
      "learning_rate": 0.0001,
      "loss": 4.0881,
      "loss/crossentropy": 2.1297428011894226,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19571492820978165,
      "step": 27864
    },
    {
      "epoch": 0.55732,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.007212066650390625,
      "learning_rate": 0.0001,
      "loss": 3.9824,
      "loss/crossentropy": 2.038254678249359,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1918657273054123,
      "step": 27866
    },
    {
      "epoch": 0.55736,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.006790924072265625,
      "learning_rate": 0.0001,
      "loss": 4.2425,
      "loss/crossentropy": 2.1835550665855408,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19663546979427338,
      "step": 27868
    },
    {
      "epoch": 0.5574,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.006021881103515625,
      "learning_rate": 0.0001,
      "loss": 4.097,
      "loss/crossentropy": 2.0005786418914795,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20361479371786118,
      "step": 27870
    },
    {
      "epoch": 0.55744,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.006640625,
      "learning_rate": 0.0001,
      "loss": 4.036,
      "loss/crossentropy": 1.9897403717041016,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18811679631471634,
      "step": 27872
    },
    {
      "epoch": 0.55748,
      "grad_norm": 2.625,
      "grad_norm_var": 0.035042063395182295,
      "learning_rate": 0.0001,
      "loss": 3.8279,
      "loss/crossentropy": 1.8575212359428406,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18323712795972824,
      "step": 27874
    },
    {
      "epoch": 0.55752,
      "grad_norm": 2.296875,
      "grad_norm_var": 0.040537261962890626,
      "learning_rate": 0.0001,
      "loss": 3.8482,
      "loss/crossentropy": 2.1971237659454346,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20962104946374893,
      "step": 27876
    },
    {
      "epoch": 0.55756,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.04003473917643229,
      "learning_rate": 0.0001,
      "loss": 4.0675,
      "loss/crossentropy": 2.1657967567443848,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19974523782730103,
      "step": 27878
    },
    {
      "epoch": 0.5576,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.04272028605143229,
      "learning_rate": 0.0001,
      "loss": 3.9154,
      "loss/crossentropy": 1.988546371459961,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2359297126531601,
      "step": 27880
    },
    {
      "epoch": 0.55764,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0413330078125,
      "learning_rate": 0.0001,
      "loss": 4.3125,
      "loss/crossentropy": 1.979064404964447,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1763293370604515,
      "step": 27882
    },
    {
      "epoch": 0.55768,
      "grad_norm": 2.0,
      "grad_norm_var": 0.04175516764322917,
      "learning_rate": 0.0001,
      "loss": 4.307,
      "loss/crossentropy": 2.3738937377929688,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21125850826501846,
      "step": 27884
    },
    {
      "epoch": 0.55772,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.04346110026041667,
      "learning_rate": 0.0001,
      "loss": 4.0331,
      "loss/crossentropy": 1.9243283867835999,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22126196324825287,
      "step": 27886
    },
    {
      "epoch": 0.55776,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0432373046875,
      "learning_rate": 0.0001,
      "loss": 3.9017,
      "loss/crossentropy": 2.0033947229385376,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17997342348098755,
      "step": 27888
    },
    {
      "epoch": 0.5578,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.015396881103515624,
      "learning_rate": 0.0001,
      "loss": 4.0113,
      "loss/crossentropy": 2.1850863695144653,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20201555639505386,
      "step": 27890
    },
    {
      "epoch": 0.55784,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.008487955729166666,
      "learning_rate": 0.0001,
      "loss": 3.8585,
      "loss/crossentropy": 1.9696857333183289,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20272113382816315,
      "step": 27892
    },
    {
      "epoch": 0.55788,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.009234364827473958,
      "learning_rate": 0.0001,
      "loss": 4.2124,
      "loss/crossentropy": 2.014931857585907,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2670851871371269,
      "step": 27894
    },
    {
      "epoch": 0.55792,
      "grad_norm": 2.8125,
      "grad_norm_var": 0.05494359334309896,
      "learning_rate": 0.0001,
      "loss": 3.91,
      "loss/crossentropy": 1.7789946794509888,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16623269766569138,
      "step": 27896
    },
    {
      "epoch": 0.55796,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.05752766927083333,
      "learning_rate": 0.0001,
      "loss": 3.9915,
      "loss/crossentropy": 1.9287369847297668,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17834855616092682,
      "step": 27898
    },
    {
      "epoch": 0.558,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0579254150390625,
      "learning_rate": 0.0001,
      "loss": 4.0939,
      "loss/crossentropy": 2.2972252368927,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19952362030744553,
      "step": 27900
    },
    {
      "epoch": 0.55804,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.05802993774414063,
      "learning_rate": 0.0001,
      "loss": 4.0678,
      "loss/crossentropy": 2.307905077934265,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21856702864170074,
      "step": 27902
    },
    {
      "epoch": 0.55808,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.06280008951822917,
      "learning_rate": 0.0001,
      "loss": 3.9421,
      "loss/crossentropy": 2.0781975984573364,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20248684287071228,
      "step": 27904
    },
    {
      "epoch": 0.55812,
      "grad_norm": 2.125,
      "grad_norm_var": 0.06393229166666667,
      "learning_rate": 0.0001,
      "loss": 4.0949,
      "loss/crossentropy": 2.2437129616737366,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20828180015087128,
      "step": 27906
    },
    {
      "epoch": 0.55816,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.06414286295572917,
      "learning_rate": 0.0001,
      "loss": 3.6518,
      "loss/crossentropy": 1.8736275434494019,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1924392208456993,
      "step": 27908
    },
    {
      "epoch": 0.5582,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.06412734985351562,
      "learning_rate": 0.0001,
      "loss": 3.874,
      "loss/crossentropy": 1.7928629517555237,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1721597984433174,
      "step": 27910
    },
    {
      "epoch": 0.55824,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.0142486572265625,
      "learning_rate": 0.0001,
      "loss": 3.9806,
      "loss/crossentropy": 1.8429313898086548,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18428131937980652,
      "step": 27912
    },
    {
      "epoch": 0.55828,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.015417226155598958,
      "learning_rate": 0.0001,
      "loss": 3.8051,
      "loss/crossentropy": 1.7473148107528687,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1703418493270874,
      "step": 27914
    },
    {
      "epoch": 0.55832,
      "grad_norm": 2.0,
      "grad_norm_var": 0.015712229410807292,
      "learning_rate": 0.0001,
      "loss": 4.1174,
      "loss/crossentropy": 1.9749281406402588,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19364860653877258,
      "step": 27916
    },
    {
      "epoch": 0.55836,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.014564768473307291,
      "learning_rate": 0.0001,
      "loss": 3.9925,
      "loss/crossentropy": 2.0153204202651978,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18824293464422226,
      "step": 27918
    },
    {
      "epoch": 0.5584,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.008925120035807291,
      "learning_rate": 0.0001,
      "loss": 3.9048,
      "loss/crossentropy": 1.984444797039032,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17791762948036194,
      "step": 27920
    },
    {
      "epoch": 0.55844,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.007846832275390625,
      "learning_rate": 0.0001,
      "loss": 4.0684,
      "loss/crossentropy": 2.1440568566322327,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.277907058596611,
      "step": 27922
    },
    {
      "epoch": 0.55848,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.01706517537434896,
      "learning_rate": 0.0001,
      "loss": 4.1565,
      "loss/crossentropy": 2.261883854866028,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21542102098464966,
      "step": 27924
    },
    {
      "epoch": 0.55852,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.017647298177083333,
      "learning_rate": 0.0001,
      "loss": 4.1513,
      "loss/crossentropy": 2.0659791827201843,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17906638234853745,
      "step": 27926
    },
    {
      "epoch": 0.55856,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.016893513997395835,
      "learning_rate": 0.0001,
      "loss": 4.1226,
      "loss/crossentropy": 2.536762237548828,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20740076154470444,
      "step": 27928
    },
    {
      "epoch": 0.5586,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.014314778645833333,
      "learning_rate": 0.0001,
      "loss": 3.741,
      "loss/crossentropy": 1.8174605965614319,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17254772782325745,
      "step": 27930
    },
    {
      "epoch": 0.55864,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.015575154622395834,
      "learning_rate": 0.0001,
      "loss": 3.9703,
      "loss/crossentropy": 2.2474478483200073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22224698960781097,
      "step": 27932
    },
    {
      "epoch": 0.55868,
      "grad_norm": 2.125,
      "grad_norm_var": 0.016844685872395834,
      "learning_rate": 0.0001,
      "loss": 4.0159,
      "loss/crossentropy": 1.9788220524787903,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19506333768367767,
      "step": 27934
    },
    {
      "epoch": 0.55872,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.016670735677083333,
      "learning_rate": 0.0001,
      "loss": 4.0164,
      "loss/crossentropy": 1.9576459527015686,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17929108440876007,
      "step": 27936
    },
    {
      "epoch": 0.55876,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.015987141927083334,
      "learning_rate": 0.0001,
      "loss": 4.2099,
      "loss/crossentropy": 2.2682799100875854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22949229925870895,
      "step": 27938
    },
    {
      "epoch": 0.5588,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.009235636393229166,
      "learning_rate": 0.0001,
      "loss": 3.9841,
      "loss/crossentropy": 2.0454965829849243,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19698018580675125,
      "step": 27940
    },
    {
      "epoch": 0.55884,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.010729726155598958,
      "learning_rate": 0.0001,
      "loss": 4.0611,
      "loss/crossentropy": 2.3304296731948853,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22486522793769836,
      "step": 27942
    },
    {
      "epoch": 0.55888,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.010538736979166666,
      "learning_rate": 0.0001,
      "loss": 3.787,
      "loss/crossentropy": 1.7931689620018005,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16503961384296417,
      "step": 27944
    },
    {
      "epoch": 0.55892,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.011124674479166667,
      "learning_rate": 0.0001,
      "loss": 3.7788,
      "loss/crossentropy": 1.8899441957473755,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18204578012228012,
      "step": 27946
    },
    {
      "epoch": 0.55896,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0106109619140625,
      "learning_rate": 0.0001,
      "loss": 3.9515,
      "loss/crossentropy": 1.9887003302574158,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19222819805145264,
      "step": 27948
    },
    {
      "epoch": 0.559,
      "grad_norm": 1.875,
      "grad_norm_var": 0.0088531494140625,
      "learning_rate": 0.0001,
      "loss": 3.9696,
      "loss/crossentropy": 2.097675323486328,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1877489537000656,
      "step": 27950
    },
    {
      "epoch": 0.55904,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.009248860677083333,
      "learning_rate": 0.0001,
      "loss": 3.9556,
      "loss/crossentropy": 1.801637887954712,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15989546477794647,
      "step": 27952
    },
    {
      "epoch": 0.55908,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.008756256103515625,
      "learning_rate": 0.0001,
      "loss": 4.141,
      "loss/crossentropy": 2.1653464436531067,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17995328456163406,
      "step": 27954
    },
    {
      "epoch": 0.55912,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.009506988525390624,
      "learning_rate": 0.0001,
      "loss": 3.8267,
      "loss/crossentropy": 2.370779275894165,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22155411541461945,
      "step": 27956
    },
    {
      "epoch": 0.55916,
      "grad_norm": 1.875,
      "grad_norm_var": 0.008055623372395833,
      "learning_rate": 0.0001,
      "loss": 3.9821,
      "loss/crossentropy": 1.9569392204284668,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20606116950511932,
      "step": 27958
    },
    {
      "epoch": 0.5592,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.009919230143229167,
      "learning_rate": 0.0001,
      "loss": 3.9857,
      "loss/crossentropy": 1.8667678833007812,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18434254080057144,
      "step": 27960
    },
    {
      "epoch": 0.55924,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.009171549479166667,
      "learning_rate": 0.0001,
      "loss": 4.0802,
      "loss/crossentropy": 2.4322283267974854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21796736866235733,
      "step": 27962
    },
    {
      "epoch": 0.55928,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0070220947265625,
      "learning_rate": 0.0001,
      "loss": 3.8252,
      "loss/crossentropy": 1.7290849685668945,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17558299005031586,
      "step": 27964
    },
    {
      "epoch": 0.55932,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.006430816650390625,
      "learning_rate": 0.0001,
      "loss": 3.8143,
      "loss/crossentropy": 1.9967953562736511,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19310998916625977,
      "step": 27966
    },
    {
      "epoch": 0.55936,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.006685129801432292,
      "learning_rate": 0.0001,
      "loss": 3.9863,
      "loss/crossentropy": 2.027643382549286,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1989550217986107,
      "step": 27968
    },
    {
      "epoch": 0.5594,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.016312408447265624,
      "learning_rate": 0.0001,
      "loss": 3.8968,
      "loss/crossentropy": 2.057853937149048,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21117058396339417,
      "step": 27970
    },
    {
      "epoch": 0.55944,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.015122222900390624,
      "learning_rate": 0.0001,
      "loss": 4.1212,
      "loss/crossentropy": 2.113202214241028,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19248170405626297,
      "step": 27972
    },
    {
      "epoch": 0.55948,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.014454905192057292,
      "learning_rate": 0.0001,
      "loss": 4.0971,
      "loss/crossentropy": 2.04659241437912,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2064732015132904,
      "step": 27974
    },
    {
      "epoch": 0.55952,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.012943522135416666,
      "learning_rate": 0.0001,
      "loss": 4.0117,
      "loss/crossentropy": 2.046691119670868,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19065889716148376,
      "step": 27976
    },
    {
      "epoch": 0.55956,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.014945475260416667,
      "learning_rate": 0.0001,
      "loss": 3.7466,
      "loss/crossentropy": 1.896644115447998,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1838160902261734,
      "step": 27978
    },
    {
      "epoch": 0.5596,
      "grad_norm": 2.0,
      "grad_norm_var": 0.015411122639973959,
      "learning_rate": 0.0001,
      "loss": 3.9367,
      "loss/crossentropy": 1.9928682446479797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17411044239997864,
      "step": 27980
    },
    {
      "epoch": 0.55964,
      "grad_norm": 1.875,
      "grad_norm_var": 0.015843709309895832,
      "learning_rate": 0.0001,
      "loss": 3.9616,
      "loss/crossentropy": 1.9449793100357056,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18892540037631989,
      "step": 27982
    },
    {
      "epoch": 0.55968,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0159912109375,
      "learning_rate": 0.0001,
      "loss": 3.8687,
      "loss/crossentropy": 1.949626863002777,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1996856927871704,
      "step": 27984
    },
    {
      "epoch": 0.55972,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.0056793212890625,
      "learning_rate": 0.0001,
      "loss": 4.0761,
      "loss/crossentropy": 2.116423010826111,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18394631892442703,
      "step": 27986
    },
    {
      "epoch": 0.55976,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.006320953369140625,
      "learning_rate": 0.0001,
      "loss": 3.9168,
      "loss/crossentropy": 2.151313066482544,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1895233690738678,
      "step": 27988
    },
    {
      "epoch": 0.5598,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.007739003499348958,
      "learning_rate": 0.0001,
      "loss": 3.8336,
      "loss/crossentropy": 2.1060383319854736,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21665210276842117,
      "step": 27990
    },
    {
      "epoch": 0.55984,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009528605143229167,
      "learning_rate": 0.0001,
      "loss": 3.9715,
      "loss/crossentropy": 2.201907217502594,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20985589921474457,
      "step": 27992
    },
    {
      "epoch": 0.55988,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0083251953125,
      "learning_rate": 0.0001,
      "loss": 3.9765,
      "loss/crossentropy": 2.0219600796699524,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2357076108455658,
      "step": 27994
    },
    {
      "epoch": 0.55992,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.008166249593098958,
      "learning_rate": 0.0001,
      "loss": 4.1518,
      "loss/crossentropy": 2.006410300731659,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20221157371997833,
      "step": 27996
    },
    {
      "epoch": 0.55996,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009886678059895833,
      "learning_rate": 0.0001,
      "loss": 3.842,
      "loss/crossentropy": 1.7009567618370056,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16861556470394135,
      "step": 27998
    },
    {
      "epoch": 0.56,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.010587310791015625,
      "learning_rate": 0.0001,
      "loss": 3.6914,
      "loss/crossentropy": 1.7348475456237793,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15609163790941238,
      "step": 28000
    },
    {
      "epoch": 0.56004,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.011087799072265625,
      "learning_rate": 0.0001,
      "loss": 4.0131,
      "loss/crossentropy": 2.253269076347351,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21236062049865723,
      "step": 28002
    },
    {
      "epoch": 0.56008,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.01077880859375,
      "learning_rate": 0.0001,
      "loss": 3.6868,
      "loss/crossentropy": 1.5210962891578674,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15905095636844635,
      "step": 28004
    },
    {
      "epoch": 0.56012,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.011017862955729167,
      "learning_rate": 0.0001,
      "loss": 4.201,
      "loss/crossentropy": 2.201940894126892,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.206693597137928,
      "step": 28006
    },
    {
      "epoch": 0.56016,
      "grad_norm": 2.0,
      "grad_norm_var": 0.010367584228515626,
      "learning_rate": 0.0001,
      "loss": 3.979,
      "loss/crossentropy": 2.0980228185653687,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19499729573726654,
      "step": 28008
    },
    {
      "epoch": 0.5602,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.011413319905598959,
      "learning_rate": 0.0001,
      "loss": 4.0716,
      "loss/crossentropy": 2.0373584032058716,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17089952528476715,
      "step": 28010
    },
    {
      "epoch": 0.56024,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.013291168212890624,
      "learning_rate": 0.0001,
      "loss": 4.3161,
      "loss/crossentropy": 2.4253649711608887,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2154974490404129,
      "step": 28012
    },
    {
      "epoch": 0.56028,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.011405181884765626,
      "learning_rate": 0.0001,
      "loss": 3.9914,
      "loss/crossentropy": 2.121450662612915,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19490933418273926,
      "step": 28014
    },
    {
      "epoch": 0.56032,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.013916015625,
      "learning_rate": 0.0001,
      "loss": 3.9158,
      "loss/crossentropy": 1.9926584959030151,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19274179637432098,
      "step": 28016
    },
    {
      "epoch": 0.56036,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.015559641520182292,
      "learning_rate": 0.0001,
      "loss": 3.9918,
      "loss/crossentropy": 1.6820538640022278,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17553947865962982,
      "step": 28018
    },
    {
      "epoch": 0.5604,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.013496653238932291,
      "learning_rate": 0.0001,
      "loss": 4.2007,
      "loss/crossentropy": 2.278828501701355,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22723327577114105,
      "step": 28020
    },
    {
      "epoch": 0.56044,
      "grad_norm": 1.7421875,
      "grad_norm_var": 0.016682942708333332,
      "learning_rate": 0.0001,
      "loss": 3.4084,
      "loss/crossentropy": 1.860534906387329,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17910753190517426,
      "step": 28022
    },
    {
      "epoch": 0.56048,
      "grad_norm": 1.734375,
      "grad_norm_var": 0.019945271809895835,
      "learning_rate": 0.0001,
      "loss": 3.6443,
      "loss/crossentropy": 2.094952940940857,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19640950858592987,
      "step": 28024
    },
    {
      "epoch": 0.56052,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.03429133097330729,
      "learning_rate": 0.0001,
      "loss": 3.9637,
      "loss/crossentropy": 2.2802677154541016,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2223518192768097,
      "step": 28026
    },
    {
      "epoch": 0.56056,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.03408177693684896,
      "learning_rate": 0.0001,
      "loss": 4.2298,
      "loss/crossentropy": 2.382766842842102,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22707599401474,
      "step": 28028
    },
    {
      "epoch": 0.5606,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.033512115478515625,
      "learning_rate": 0.0001,
      "loss": 3.984,
      "loss/crossentropy": 1.861901044845581,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17228157818317413,
      "step": 28030
    },
    {
      "epoch": 0.56064,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.030454254150390624,
      "learning_rate": 0.0001,
      "loss": 3.8515,
      "loss/crossentropy": 1.8863429427146912,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17535988986492157,
      "step": 28032
    },
    {
      "epoch": 0.56068,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.02935765584309896,
      "learning_rate": 0.0001,
      "loss": 4.3787,
      "loss/crossentropy": 2.0987571477890015,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2073696255683899,
      "step": 28034
    },
    {
      "epoch": 0.56072,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.03867975870768229,
      "learning_rate": 0.0001,
      "loss": 3.9042,
      "loss/crossentropy": 1.6989398002624512,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1658734381198883,
      "step": 28036
    },
    {
      "epoch": 0.56076,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.030586751302083333,
      "learning_rate": 0.0001,
      "loss": 3.8771,
      "loss/crossentropy": 2.007630228996277,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17940524220466614,
      "step": 28038
    },
    {
      "epoch": 0.5608,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.028888956705729166,
      "learning_rate": 0.0001,
      "loss": 3.8596,
      "loss/crossentropy": 2.2327693700790405,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22857680916786194,
      "step": 28040
    },
    {
      "epoch": 0.56084,
      "grad_norm": 1.875,
      "grad_norm_var": 0.016524251302083334,
      "learning_rate": 0.0001,
      "loss": 3.9398,
      "loss/crossentropy": 2.051284074783325,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18832950294017792,
      "step": 28042
    },
    {
      "epoch": 0.56088,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.016627756754557292,
      "learning_rate": 0.0001,
      "loss": 3.9606,
      "loss/crossentropy": 1.9091659784317017,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17789962142705917,
      "step": 28044
    },
    {
      "epoch": 0.56092,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.05832112630208333,
      "learning_rate": 0.0001,
      "loss": 3.7684,
      "loss/crossentropy": 1.8703528046607971,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24716350436210632,
      "step": 28046
    },
    {
      "epoch": 0.56096,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.060031890869140625,
      "learning_rate": 0.0001,
      "loss": 3.7816,
      "loss/crossentropy": 1.9286105036735535,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17276924848556519,
      "step": 28048
    },
    {
      "epoch": 0.561,
      "grad_norm": 2.125,
      "grad_norm_var": 0.06181208292643229,
      "learning_rate": 0.0001,
      "loss": 4.2004,
      "loss/crossentropy": 2.115488290786743,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20674347877502441,
      "step": 28050
    },
    {
      "epoch": 0.56104,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.058426920572916666,
      "learning_rate": 0.0001,
      "loss": 3.6918,
      "loss/crossentropy": 2.1754192113876343,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19405043870210648,
      "step": 28052
    },
    {
      "epoch": 0.56108,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.08551025390625,
      "learning_rate": 0.0001,
      "loss": 4.2559,
      "loss/crossentropy": 2.378780484199524,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19838006794452667,
      "step": 28054
    },
    {
      "epoch": 0.56112,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.08651504516601563,
      "learning_rate": 0.0001,
      "loss": 4.1649,
      "loss/crossentropy": 1.8579109907150269,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19068627059459686,
      "step": 28056
    },
    {
      "epoch": 0.56116,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.1222564697265625,
      "learning_rate": 0.0001,
      "loss": 4.17,
      "loss/crossentropy": 2.190861463546753,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1977207213640213,
      "step": 28058
    },
    {
      "epoch": 0.5612,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.15713297526041667,
      "learning_rate": 0.0001,
      "loss": 3.8599,
      "loss/crossentropy": 1.8596174716949463,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18201489746570587,
      "step": 28060
    },
    {
      "epoch": 0.56124,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.1685808817545573,
      "learning_rate": 0.0001,
      "loss": 4.105,
      "loss/crossentropy": 1.8212996125221252,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16964350640773773,
      "step": 28062
    },
    {
      "epoch": 0.56128,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.15612691243489582,
      "learning_rate": 0.0001,
      "loss": 4.0088,
      "loss/crossentropy": 2.1002501845359802,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20258799940347672,
      "step": 28064
    },
    {
      "epoch": 0.56132,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.18886617024739583,
      "learning_rate": 0.0001,
      "loss": 4.0179,
      "loss/crossentropy": 2.1405937671661377,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2100813016295433,
      "step": 28066
    },
    {
      "epoch": 0.56136,
      "grad_norm": 2.90625,
      "grad_norm_var": 0.22138264973958333,
      "learning_rate": 0.0001,
      "loss": 3.7586,
      "loss/crossentropy": 2.23690402507782,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19980955868959427,
      "step": 28068
    },
    {
      "epoch": 0.5614,
      "grad_norm": 3.015625,
      "grad_norm_var": 0.30380452473958336,
      "learning_rate": 0.0001,
      "loss": 3.5867,
      "loss/crossentropy": 1.875457525253296,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16666759550571442,
      "step": 28070
    },
    {
      "epoch": 0.56144,
      "grad_norm": 2.78125,
      "grad_norm_var": 0.26413141886393227,
      "learning_rate": 0.0001,
      "loss": 4.0021,
      "loss/crossentropy": 2.1469756960868835,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19378764182329178,
      "step": 28072
    },
    {
      "epoch": 0.56148,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.25313212076822916,
      "learning_rate": 0.0001,
      "loss": 4.0959,
      "loss/crossentropy": 2.156083106994629,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20279116183519363,
      "step": 28074
    },
    {
      "epoch": 0.56152,
      "grad_norm": 2.0,
      "grad_norm_var": 0.27230631510416664,
      "learning_rate": 0.0001,
      "loss": 3.9681,
      "loss/crossentropy": 1.9321695566177368,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1937977820634842,
      "step": 28076
    },
    {
      "epoch": 0.56156,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.29947077433268227,
      "learning_rate": 0.0001,
      "loss": 3.6465,
      "loss/crossentropy": 2.005180776119232,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18012206256389618,
      "step": 28078
    },
    {
      "epoch": 0.5616,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.35347391764322916,
      "learning_rate": 0.0001,
      "loss": 3.7987,
      "loss/crossentropy": 2.040120303630829,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17059049755334854,
      "step": 28080
    },
    {
      "epoch": 0.56164,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.34440511067708335,
      "learning_rate": 0.0001,
      "loss": 4.3903,
      "loss/crossentropy": 2.5171138048171997,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21068025380373,
      "step": 28082
    },
    {
      "epoch": 0.56168,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.2691070556640625,
      "learning_rate": 0.0001,
      "loss": 4.252,
      "loss/crossentropy": 2.1827582120895386,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1995992213487625,
      "step": 28084
    },
    {
      "epoch": 0.56172,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.09042154947916667,
      "learning_rate": 0.0001,
      "loss": 4.1365,
      "loss/crossentropy": 2.0082287192344666,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23266522586345673,
      "step": 28086
    },
    {
      "epoch": 0.56176,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.024853261311848958,
      "learning_rate": 0.0001,
      "loss": 3.8143,
      "loss/crossentropy": 1.59010910987854,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15921758115291595,
      "step": 28088
    },
    {
      "epoch": 0.5618,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.012132771809895833,
      "learning_rate": 0.0001,
      "loss": 4.0431,
      "loss/crossentropy": 2.3045564889907837,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2091711461544037,
      "step": 28090
    },
    {
      "epoch": 0.56184,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.0129547119140625,
      "learning_rate": 0.0001,
      "loss": 3.8729,
      "loss/crossentropy": 1.9903671741485596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19884900748729706,
      "step": 28092
    },
    {
      "epoch": 0.56188,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.011970011393229167,
      "learning_rate": 0.0001,
      "loss": 3.9903,
      "loss/crossentropy": 2.3190979957580566,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20256339758634567,
      "step": 28094
    },
    {
      "epoch": 0.56192,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.009983062744140625,
      "learning_rate": 0.0001,
      "loss": 4.0022,
      "loss/crossentropy": 1.8695592880249023,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17132485657930374,
      "step": 28096
    },
    {
      "epoch": 0.56196,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.008253733317057291,
      "learning_rate": 0.0001,
      "loss": 3.7549,
      "loss/crossentropy": 1.9170012474060059,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18364953994750977,
      "step": 28098
    },
    {
      "epoch": 0.562,
      "grad_norm": 1.6328125,
      "grad_norm_var": 0.011749013264973959,
      "learning_rate": 0.0001,
      "loss": 3.6666,
      "loss/crossentropy": 1.7711463570594788,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16884273290634155,
      "step": 28100
    },
    {
      "epoch": 0.56204,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0090728759765625,
      "learning_rate": 0.0001,
      "loss": 4.1265,
      "loss/crossentropy": 2.216683030128479,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19033723324537277,
      "step": 28102
    },
    {
      "epoch": 0.56208,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.008666737874348959,
      "learning_rate": 0.0001,
      "loss": 3.7862,
      "loss/crossentropy": 2.068448841571808,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18903378397226334,
      "step": 28104
    },
    {
      "epoch": 0.56212,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.008378092447916667,
      "learning_rate": 0.0001,
      "loss": 3.9428,
      "loss/crossentropy": 1.9194504618644714,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17733828723430634,
      "step": 28106
    },
    {
      "epoch": 0.56216,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.00836181640625,
      "learning_rate": 0.0001,
      "loss": 3.8966,
      "loss/crossentropy": 1.8640123009681702,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18660252541303635,
      "step": 28108
    },
    {
      "epoch": 0.5622,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.006628163655598958,
      "learning_rate": 0.0001,
      "loss": 3.883,
      "loss/crossentropy": 2.084853768348694,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1897662729024887,
      "step": 28110
    },
    {
      "epoch": 0.56224,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.006681060791015625,
      "learning_rate": 0.0001,
      "loss": 3.7912,
      "loss/crossentropy": 1.9047400951385498,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16695564985275269,
      "step": 28112
    },
    {
      "epoch": 0.56228,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.007616170247395833,
      "learning_rate": 0.0001,
      "loss": 3.9128,
      "loss/crossentropy": 1.9586027264595032,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18182067573070526,
      "step": 28114
    },
    {
      "epoch": 0.56232,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.007448069254557292,
      "learning_rate": 0.0001,
      "loss": 4.0318,
      "loss/crossentropy": 2.0719743371009827,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18775883316993713,
      "step": 28116
    },
    {
      "epoch": 0.56236,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.014070638020833333,
      "learning_rate": 0.0001,
      "loss": 4.5828,
      "loss/crossentropy": 2.2210047245025635,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.32780784368515015,
      "step": 28118
    },
    {
      "epoch": 0.5624,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0147125244140625,
      "learning_rate": 0.0001,
      "loss": 3.9965,
      "loss/crossentropy": 2.212351083755493,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19030366092920303,
      "step": 28120
    },
    {
      "epoch": 0.56244,
      "grad_norm": 2.0,
      "grad_norm_var": 0.014939117431640624,
      "learning_rate": 0.0001,
      "loss": 4.1715,
      "loss/crossentropy": 2.2713009119033813,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21552074700593948,
      "step": 28122
    },
    {
      "epoch": 0.56248,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.014192708333333333,
      "learning_rate": 0.0001,
      "loss": 3.8242,
      "loss/crossentropy": 1.872189700603485,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20009242743253708,
      "step": 28124
    },
    {
      "epoch": 0.56252,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.015203603108723958,
      "learning_rate": 0.0001,
      "loss": 3.9798,
      "loss/crossentropy": 2.0979323983192444,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19020158797502518,
      "step": 28126
    },
    {
      "epoch": 0.56256,
      "grad_norm": 2.0,
      "grad_norm_var": 0.010937245686848958,
      "learning_rate": 0.0001,
      "loss": 4.1726,
      "loss/crossentropy": 1.9983355402946472,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18428711593151093,
      "step": 28128
    },
    {
      "epoch": 0.5626,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.011864217122395833,
      "learning_rate": 0.0001,
      "loss": 4.0696,
      "loss/crossentropy": 1.784355342388153,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23605111241340637,
      "step": 28130
    },
    {
      "epoch": 0.56264,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.01231689453125,
      "learning_rate": 0.0001,
      "loss": 3.932,
      "loss/crossentropy": 1.719801664352417,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16245415434241295,
      "step": 28132
    },
    {
      "epoch": 0.56268,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.00958251953125,
      "learning_rate": 0.0001,
      "loss": 3.8402,
      "loss/crossentropy": 1.8063556551933289,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17704912275075912,
      "step": 28134
    },
    {
      "epoch": 0.56272,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.007832845052083334,
      "learning_rate": 0.0001,
      "loss": 4.0469,
      "loss/crossentropy": 1.9270595908164978,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1939307227730751,
      "step": 28136
    },
    {
      "epoch": 0.56276,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.008060709635416666,
      "learning_rate": 0.0001,
      "loss": 3.9754,
      "loss/crossentropy": 1.9221222400665283,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1753731295466423,
      "step": 28138
    },
    {
      "epoch": 0.5628,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.00994873046875,
      "learning_rate": 0.0001,
      "loss": 3.7678,
      "loss/crossentropy": 1.718525469303131,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1854087859392166,
      "step": 28140
    },
    {
      "epoch": 0.56284,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.009200032552083333,
      "learning_rate": 0.0001,
      "loss": 4.0674,
      "loss/crossentropy": 1.9418652057647705,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19125580787658691,
      "step": 28142
    },
    {
      "epoch": 0.56288,
      "grad_norm": 1.875,
      "grad_norm_var": 0.008587392171223958,
      "learning_rate": 0.0001,
      "loss": 4.1413,
      "loss/crossentropy": 2.100765645503998,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19208013266324997,
      "step": 28144
    },
    {
      "epoch": 0.56292,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.010849761962890624,
      "learning_rate": 0.0001,
      "loss": 3.8,
      "loss/crossentropy": 1.9104883074760437,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1662292778491974,
      "step": 28146
    },
    {
      "epoch": 0.56296,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.010092926025390626,
      "learning_rate": 0.0001,
      "loss": 3.8448,
      "loss/crossentropy": 2.059969484806061,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18414444476366043,
      "step": 28148
    },
    {
      "epoch": 0.563,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.010392252604166667,
      "learning_rate": 0.0001,
      "loss": 3.8721,
      "loss/crossentropy": 1.9335008263587952,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18294576555490494,
      "step": 28150
    },
    {
      "epoch": 0.56304,
      "grad_norm": 2.703125,
      "grad_norm_var": 0.05110575358072917,
      "learning_rate": 0.0001,
      "loss": 4.3341,
      "loss/crossentropy": 2.1855704188346863,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19496884942054749,
      "step": 28152
    },
    {
      "epoch": 0.56308,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.058176422119140626,
      "learning_rate": 0.0001,
      "loss": 4.2066,
      "loss/crossentropy": 2.1697583198547363,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19855469465255737,
      "step": 28154
    },
    {
      "epoch": 0.56312,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.058176422119140626,
      "learning_rate": 0.0001,
      "loss": 3.833,
      "loss/crossentropy": 1.926409661769867,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18937255442142487,
      "step": 28156
    },
    {
      "epoch": 0.56316,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.059456125895182295,
      "learning_rate": 0.0001,
      "loss": 4.1422,
      "loss/crossentropy": 1.900004506111145,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.190530426800251,
      "step": 28158
    },
    {
      "epoch": 0.5632,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.06049982706705729,
      "learning_rate": 0.0001,
      "loss": 4.0405,
      "loss/crossentropy": 1.8822137117385864,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1802232339978218,
      "step": 28160
    },
    {
      "epoch": 0.56324,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.0831207275390625,
      "learning_rate": 0.0001,
      "loss": 3.9198,
      "loss/crossentropy": 2.088913321495056,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2068777084350586,
      "step": 28162
    },
    {
      "epoch": 0.56328,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.08334935506184896,
      "learning_rate": 0.0001,
      "loss": 3.967,
      "loss/crossentropy": 2.0997453927993774,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21025776118040085,
      "step": 28164
    },
    {
      "epoch": 0.56332,
      "grad_norm": 1.875,
      "grad_norm_var": 0.08046061197916667,
      "learning_rate": 0.0001,
      "loss": 3.9585,
      "loss/crossentropy": 2.19319224357605,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1984698325395584,
      "step": 28166
    },
    {
      "epoch": 0.56336,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.04764989217122396,
      "learning_rate": 0.0001,
      "loss": 4.0751,
      "loss/crossentropy": 2.022126793861389,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1894361823797226,
      "step": 28168
    },
    {
      "epoch": 0.5634,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.04308242797851562,
      "learning_rate": 0.0001,
      "loss": 3.9752,
      "loss/crossentropy": 2.0495948791503906,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18251951038837433,
      "step": 28170
    },
    {
      "epoch": 0.56344,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.04134928385416667,
      "learning_rate": 0.0001,
      "loss": 4.1206,
      "loss/crossentropy": 1.9946674704551697,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18497124314308167,
      "step": 28172
    },
    {
      "epoch": 0.56348,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.03991063435872396,
      "learning_rate": 0.0001,
      "loss": 3.6823,
      "loss/crossentropy": 1.970799744129181,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20478563010692596,
      "step": 28174
    },
    {
      "epoch": 0.56352,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.039606730143229164,
      "learning_rate": 0.0001,
      "loss": 3.6153,
      "loss/crossentropy": 1.9632895588874817,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1864907443523407,
      "step": 28176
    },
    {
      "epoch": 0.56356,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.005150349934895834,
      "learning_rate": 0.0001,
      "loss": 3.8629,
      "loss/crossentropy": 1.9328609108924866,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19433043897151947,
      "step": 28178
    },
    {
      "epoch": 0.5636,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.004727935791015625,
      "learning_rate": 0.0001,
      "loss": 3.9304,
      "loss/crossentropy": 2.1020684242248535,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18058212101459503,
      "step": 28180
    },
    {
      "epoch": 0.56364,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.0033322652180989582,
      "learning_rate": 0.0001,
      "loss": 3.8489,
      "loss/crossentropy": 2.008388578891754,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.189447820186615,
      "step": 28182
    },
    {
      "epoch": 0.56368,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.004073079427083333,
      "learning_rate": 0.0001,
      "loss": 3.554,
      "loss/crossentropy": 1.790539801120758,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16334787011146545,
      "step": 28184
    },
    {
      "epoch": 0.56372,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.004439290364583333,
      "learning_rate": 0.0001,
      "loss": 3.7392,
      "loss/crossentropy": 1.9603918194770813,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18403412401676178,
      "step": 28186
    },
    {
      "epoch": 0.56376,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.0047760009765625,
      "learning_rate": 0.0001,
      "loss": 3.7194,
      "loss/crossentropy": 2.04629784822464,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18315870314836502,
      "step": 28188
    },
    {
      "epoch": 0.5638,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.007950592041015624,
      "learning_rate": 0.0001,
      "loss": 3.8053,
      "loss/crossentropy": 2.2571674585342407,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19582515209913254,
      "step": 28190
    },
    {
      "epoch": 0.56384,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.008668772379557292,
      "learning_rate": 0.0001,
      "loss": 4.0982,
      "loss/crossentropy": 1.923226773738861,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19587361812591553,
      "step": 28192
    },
    {
      "epoch": 0.56388,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.007527669270833333,
      "learning_rate": 0.0001,
      "loss": 4.0072,
      "loss/crossentropy": 2.167626738548279,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20034413039684296,
      "step": 28194
    },
    {
      "epoch": 0.56392,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.007462310791015625,
      "learning_rate": 0.0001,
      "loss": 3.9982,
      "loss/crossentropy": 2.1942092180252075,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21775193512439728,
      "step": 28196
    },
    {
      "epoch": 0.56396,
      "grad_norm": 2.0,
      "grad_norm_var": 0.007950592041015624,
      "learning_rate": 0.0001,
      "loss": 4.1772,
      "loss/crossentropy": 1.789182722568512,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19279392063617706,
      "step": 28198
    },
    {
      "epoch": 0.564,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0064656575520833336,
      "learning_rate": 0.0001,
      "loss": 4.1771,
      "loss/crossentropy": 1.8911787271499634,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18866229057312012,
      "step": 28200
    },
    {
      "epoch": 0.56404,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.006091054280598958,
      "learning_rate": 0.0001,
      "loss": 3.9862,
      "loss/crossentropy": 2.0467012524604797,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21637209504842758,
      "step": 28202
    },
    {
      "epoch": 0.56408,
      "grad_norm": 2.15625,
      "grad_norm_var": 0.007541656494140625,
      "learning_rate": 0.0001,
      "loss": 3.9004,
      "loss/crossentropy": 2.183765947818756,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19495883584022522,
      "step": 28204
    },
    {
      "epoch": 0.56412,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.004489898681640625,
      "learning_rate": 0.0001,
      "loss": 4.0853,
      "loss/crossentropy": 2.237957239151001,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20132584869861603,
      "step": 28206
    },
    {
      "epoch": 0.56416,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.005549112955729167,
      "learning_rate": 0.0001,
      "loss": 3.7728,
      "loss/crossentropy": 1.9930571913719177,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18800070136785507,
      "step": 28208
    },
    {
      "epoch": 0.5642,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.005915323893229167,
      "learning_rate": 0.0001,
      "loss": 3.8866,
      "loss/crossentropy": 1.9499656558036804,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19010086357593536,
      "step": 28210
    },
    {
      "epoch": 0.56424,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.005708567301432292,
      "learning_rate": 0.0001,
      "loss": 4.1938,
      "loss/crossentropy": 2.115884840488434,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19502697885036469,
      "step": 28212
    },
    {
      "epoch": 0.56428,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.005940755208333333,
      "learning_rate": 0.0001,
      "loss": 4.1859,
      "loss/crossentropy": 2.1863423585891724,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2202358990907669,
      "step": 28214
    },
    {
      "epoch": 0.56432,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.007661946614583333,
      "learning_rate": 0.0001,
      "loss": 3.6467,
      "loss/crossentropy": 1.860621988773346,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18784625828266144,
      "step": 28216
    },
    {
      "epoch": 0.56436,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.007338205973307292,
      "learning_rate": 0.0001,
      "loss": 4.055,
      "loss/crossentropy": 1.7891228199005127,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1822507157921791,
      "step": 28218
    },
    {
      "epoch": 0.5644,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.0050771077473958336,
      "learning_rate": 0.0001,
      "loss": 3.9942,
      "loss/crossentropy": 2.050603687763214,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20126069337129593,
      "step": 28220
    },
    {
      "epoch": 0.56444,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.005350748697916667,
      "learning_rate": 0.0001,
      "loss": 3.9056,
      "loss/crossentropy": 2.3879984617233276,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19843067973852158,
      "step": 28222
    },
    {
      "epoch": 0.56448,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.005158487955729167,
      "learning_rate": 0.0001,
      "loss": 4.0579,
      "loss/crossentropy": 2.0878185033798218,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18253600597381592,
      "step": 28224
    },
    {
      "epoch": 0.56452,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0050432840983072914,
      "learning_rate": 0.0001,
      "loss": 3.9793,
      "loss/crossentropy": 2.1168408393859863,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21165993809700012,
      "step": 28226
    },
    {
      "epoch": 0.56456,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.0061948140462239586,
      "learning_rate": 0.0001,
      "loss": 3.8824,
      "loss/crossentropy": 1.8579602241516113,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18875548988580704,
      "step": 28228
    },
    {
      "epoch": 0.5646,
      "grad_norm": 1.7109375,
      "grad_norm_var": 0.007323201497395833,
      "learning_rate": 0.0001,
      "loss": 3.7804,
      "loss/crossentropy": 2.0095511078834534,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1720530390739441,
      "step": 28230
    },
    {
      "epoch": 0.56464,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0074765523274739586,
      "learning_rate": 0.0001,
      "loss": 3.841,
      "loss/crossentropy": 1.8599479794502258,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17767678201198578,
      "step": 28232
    },
    {
      "epoch": 0.56468,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.007746378580729167,
      "learning_rate": 0.0001,
      "loss": 4.2437,
      "loss/crossentropy": 2.11702424287796,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1897270455956459,
      "step": 28234
    },
    {
      "epoch": 0.56472,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.006180826822916667,
      "learning_rate": 0.0001,
      "loss": 3.9554,
      "loss/crossentropy": 1.8956347703933716,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1801730990409851,
      "step": 28236
    },
    {
      "epoch": 0.56476,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.006766764322916666,
      "learning_rate": 0.0001,
      "loss": 3.9988,
      "loss/crossentropy": 2.044249713420868,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17439774423837662,
      "step": 28238
    },
    {
      "epoch": 0.5648,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.006534576416015625,
      "learning_rate": 0.0001,
      "loss": 3.9429,
      "loss/crossentropy": 1.8406980633735657,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18690447509288788,
      "step": 28240
    },
    {
      "epoch": 0.56484,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.03523661295572917,
      "learning_rate": 0.0001,
      "loss": 3.6977,
      "loss/crossentropy": 1.8951767683029175,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18628580123186111,
      "step": 28242
    },
    {
      "epoch": 0.56488,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.03559951782226563,
      "learning_rate": 0.0001,
      "loss": 3.638,
      "loss/crossentropy": 1.9350382685661316,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19427261501550674,
      "step": 28244
    },
    {
      "epoch": 0.56492,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.032698567708333334,
      "learning_rate": 0.0001,
      "loss": 3.997,
      "loss/crossentropy": 2.370202422142029,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2236226201057434,
      "step": 28246
    },
    {
      "epoch": 0.56496,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.032163238525390624,
      "learning_rate": 0.0001,
      "loss": 4.0368,
      "loss/crossentropy": 2.1876412630081177,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2005016803741455,
      "step": 28248
    },
    {
      "epoch": 0.565,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0342193603515625,
      "learning_rate": 0.0001,
      "loss": 3.9678,
      "loss/crossentropy": 2.1329512000083923,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20921162515878677,
      "step": 28250
    },
    {
      "epoch": 0.56504,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.040036773681640624,
      "learning_rate": 0.0001,
      "loss": 3.8688,
      "loss/crossentropy": 1.8053449392318726,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17606808245182037,
      "step": 28252
    },
    {
      "epoch": 0.56508,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.04079360961914062,
      "learning_rate": 0.0001,
      "loss": 3.9478,
      "loss/crossentropy": 2.0177281498908997,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19240254908800125,
      "step": 28254
    },
    {
      "epoch": 0.56512,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.04112930297851562,
      "learning_rate": 0.0001,
      "loss": 3.9164,
      "loss/crossentropy": 1.9807183146476746,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1726047471165657,
      "step": 28256
    },
    {
      "epoch": 0.56516,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.012345377604166667,
      "learning_rate": 0.0001,
      "loss": 4.0006,
      "loss/crossentropy": 2.0992757081985474,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19170843064785004,
      "step": 28258
    },
    {
      "epoch": 0.5652,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.01220703125,
      "learning_rate": 0.0001,
      "loss": 3.6902,
      "loss/crossentropy": 1.865010380744934,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18145494908094406,
      "step": 28260
    },
    {
      "epoch": 0.56524,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.012788899739583333,
      "learning_rate": 0.0001,
      "loss": 3.7781,
      "loss/crossentropy": 2.060899078845978,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17389538139104843,
      "step": 28262
    },
    {
      "epoch": 0.56528,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.013911946614583334,
      "learning_rate": 0.0001,
      "loss": 3.8494,
      "loss/crossentropy": 1.7959403991699219,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17851080000400543,
      "step": 28264
    },
    {
      "epoch": 0.56532,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.012410227457682292,
      "learning_rate": 0.0001,
      "loss": 3.8456,
      "loss/crossentropy": 2.169707775115967,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19611390680074692,
      "step": 28266
    },
    {
      "epoch": 0.56536,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.015697224934895834,
      "learning_rate": 0.0001,
      "loss": 4.1967,
      "loss/crossentropy": 2.322705864906311,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20728319883346558,
      "step": 28268
    },
    {
      "epoch": 0.5654,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.013423665364583334,
      "learning_rate": 0.0001,
      "loss": 3.8118,
      "loss/crossentropy": 1.9398122429847717,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19211900979280472,
      "step": 28270
    },
    {
      "epoch": 0.56544,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.013435618082682291,
      "learning_rate": 0.0001,
      "loss": 3.9892,
      "loss/crossentropy": 1.7359251976013184,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16433238983154297,
      "step": 28272
    },
    {
      "epoch": 0.56548,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.014973958333333334,
      "learning_rate": 0.0001,
      "loss": 4.1366,
      "loss/crossentropy": 2.3107075691223145,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2127886563539505,
      "step": 28274
    },
    {
      "epoch": 0.56552,
      "grad_norm": 1.734375,
      "grad_norm_var": 0.01906305948893229,
      "learning_rate": 0.0001,
      "loss": 3.5307,
      "loss/crossentropy": 1.5368210673332214,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15861748158931732,
      "step": 28276
    },
    {
      "epoch": 0.56556,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.01902033487955729,
      "learning_rate": 0.0001,
      "loss": 4.2099,
      "loss/crossentropy": 2.145693838596344,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1863054633140564,
      "step": 28278
    },
    {
      "epoch": 0.5656,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.018464152018229166,
      "learning_rate": 0.0001,
      "loss": 3.9928,
      "loss/crossentropy": 2.3174896240234375,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1882578581571579,
      "step": 28280
    },
    {
      "epoch": 0.56564,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.01803766886393229,
      "learning_rate": 0.0001,
      "loss": 4.2401,
      "loss/crossentropy": 2.3332719802856445,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2089502438902855,
      "step": 28282
    },
    {
      "epoch": 0.56568,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.008066558837890625,
      "learning_rate": 0.0001,
      "loss": 3.9758,
      "loss/crossentropy": 2.3516300916671753,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21649855375289917,
      "step": 28284
    },
    {
      "epoch": 0.56572,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.0067138671875,
      "learning_rate": 0.0001,
      "loss": 3.9705,
      "loss/crossentropy": 1.9031076431274414,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18855175375938416,
      "step": 28286
    },
    {
      "epoch": 0.56576,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.007100423177083333,
      "learning_rate": 0.0001,
      "loss": 4.1005,
      "loss/crossentropy": 1.5520136952400208,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1816611886024475,
      "step": 28288
    },
    {
      "epoch": 0.5658,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.006894683837890625,
      "learning_rate": 0.0001,
      "loss": 3.7001,
      "loss/crossentropy": 1.9759097695350647,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19612447917461395,
      "step": 28290
    },
    {
      "epoch": 0.56584,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.005163319905598958,
      "learning_rate": 0.0001,
      "loss": 4.0389,
      "loss/crossentropy": 2.1442651748657227,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18353651463985443,
      "step": 28292
    },
    {
      "epoch": 0.56588,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.005914052327473958,
      "learning_rate": 0.0001,
      "loss": 4.0021,
      "loss/crossentropy": 1.8098361492156982,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18010258674621582,
      "step": 28294
    },
    {
      "epoch": 0.56592,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0052886962890625,
      "learning_rate": 0.0001,
      "loss": 3.9866,
      "loss/crossentropy": 2.2412387132644653,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21006706357002258,
      "step": 28296
    },
    {
      "epoch": 0.56596,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.006461334228515625,
      "learning_rate": 0.0001,
      "loss": 4.07,
      "loss/crossentropy": 1.927619457244873,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1915675550699234,
      "step": 28298
    },
    {
      "epoch": 0.566,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.007502237955729167,
      "learning_rate": 0.0001,
      "loss": 3.9184,
      "loss/crossentropy": 2.317517042160034,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19702450931072235,
      "step": 28300
    },
    {
      "epoch": 0.56604,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.006514231363932292,
      "learning_rate": 0.0001,
      "loss": 4.0558,
      "loss/crossentropy": 2.02554988861084,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20614077150821686,
      "step": 28302
    },
    {
      "epoch": 0.56608,
      "grad_norm": 1.875,
      "grad_norm_var": 0.006359608968098959,
      "learning_rate": 0.0001,
      "loss": 3.9733,
      "loss/crossentropy": 2.2094991207122803,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18375249207019806,
      "step": 28304
    },
    {
      "epoch": 0.56612,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.0048601786295572914,
      "learning_rate": 0.0001,
      "loss": 3.956,
      "loss/crossentropy": 1.938852310180664,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19518058001995087,
      "step": 28306
    },
    {
      "epoch": 0.56616,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.0065081278483072914,
      "learning_rate": 0.0001,
      "loss": 4.0979,
      "loss/crossentropy": 2.0719003677368164,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19132573902606964,
      "step": 28308
    },
    {
      "epoch": 0.5662,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0059588114420572914,
      "learning_rate": 0.0001,
      "loss": 3.838,
      "loss/crossentropy": 2.01191109418869,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19400015473365784,
      "step": 28310
    },
    {
      "epoch": 0.56624,
      "grad_norm": 1.875,
      "grad_norm_var": 0.005782063802083333,
      "learning_rate": 0.0001,
      "loss": 3.9532,
      "loss/crossentropy": 2.302769422531128,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21603811532258987,
      "step": 28312
    },
    {
      "epoch": 0.56628,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.006254069010416667,
      "learning_rate": 0.0001,
      "loss": 3.9965,
      "loss/crossentropy": 1.8251231908798218,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1660621464252472,
      "step": 28314
    },
    {
      "epoch": 0.56632,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.0079254150390625,
      "learning_rate": 0.0001,
      "loss": 4.2141,
      "loss/crossentropy": 2.0656734704971313,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19783338904380798,
      "step": 28316
    },
    {
      "epoch": 0.56636,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.008485666910807292,
      "learning_rate": 0.0001,
      "loss": 4.1956,
      "loss/crossentropy": 1.949280321598053,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19571927189826965,
      "step": 28318
    },
    {
      "epoch": 0.5664,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.008571116129557292,
      "learning_rate": 0.0001,
      "loss": 4.0139,
      "loss/crossentropy": 1.801173746585846,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17420585453510284,
      "step": 28320
    },
    {
      "epoch": 0.56644,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.008519490559895834,
      "learning_rate": 0.0001,
      "loss": 3.8094,
      "loss/crossentropy": 2.000474214553833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19203417748212814,
      "step": 28322
    },
    {
      "epoch": 0.56648,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.006681315104166667,
      "learning_rate": 0.0001,
      "loss": 4.0042,
      "loss/crossentropy": 1.9475297331809998,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19463783502578735,
      "step": 28324
    },
    {
      "epoch": 0.56652,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.005965169270833333,
      "learning_rate": 0.0001,
      "loss": 3.9425,
      "loss/crossentropy": 2.029974043369293,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1769544631242752,
      "step": 28326
    },
    {
      "epoch": 0.56656,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.0058258056640625,
      "learning_rate": 0.0001,
      "loss": 3.995,
      "loss/crossentropy": 1.9521240592002869,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.189279705286026,
      "step": 28328
    },
    {
      "epoch": 0.5666,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.006136067708333333,
      "learning_rate": 0.0001,
      "loss": 4.1978,
      "loss/crossentropy": 2.128006398677826,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22940929979085922,
      "step": 28330
    },
    {
      "epoch": 0.56664,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.005234527587890625,
      "learning_rate": 0.0001,
      "loss": 3.9098,
      "loss/crossentropy": 2.3887888193130493,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2106098011136055,
      "step": 28332
    },
    {
      "epoch": 0.56668,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.006639607747395833,
      "learning_rate": 0.0001,
      "loss": 3.9625,
      "loss/crossentropy": 1.9730157256126404,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2026069611310959,
      "step": 28334
    },
    {
      "epoch": 0.56672,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.006856028238932292,
      "learning_rate": 0.0001,
      "loss": 4.0934,
      "loss/crossentropy": 2.384614109992981,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21988043189048767,
      "step": 28336
    },
    {
      "epoch": 0.56676,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.008388010660807292,
      "learning_rate": 0.0001,
      "loss": 3.9947,
      "loss/crossentropy": 1.843708097934723,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18971703946590424,
      "step": 28338
    },
    {
      "epoch": 0.5668,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009897613525390625,
      "learning_rate": 0.0001,
      "loss": 3.953,
      "loss/crossentropy": 2.1084065437316895,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19351261109113693,
      "step": 28340
    },
    {
      "epoch": 0.56684,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.011321767171223959,
      "learning_rate": 0.0001,
      "loss": 3.742,
      "loss/crossentropy": 1.7598699927330017,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18556873500347137,
      "step": 28342
    },
    {
      "epoch": 0.56688,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0115875244140625,
      "learning_rate": 0.0001,
      "loss": 3.8072,
      "loss/crossentropy": 1.952016532421112,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1828347072005272,
      "step": 28344
    },
    {
      "epoch": 0.56692,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.010286458333333333,
      "learning_rate": 0.0001,
      "loss": 3.939,
      "loss/crossentropy": 1.9486799240112305,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19223074615001678,
      "step": 28346
    },
    {
      "epoch": 0.56696,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.009197743733723958,
      "learning_rate": 0.0001,
      "loss": 3.9421,
      "loss/crossentropy": 1.9944785237312317,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19656967371702194,
      "step": 28348
    },
    {
      "epoch": 0.567,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.007911936442057291,
      "learning_rate": 0.0001,
      "loss": 3.5903,
      "loss/crossentropy": 2.055353820323944,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19104135036468506,
      "step": 28350
    },
    {
      "epoch": 0.56704,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.007096354166666667,
      "learning_rate": 0.0001,
      "loss": 3.8961,
      "loss/crossentropy": 1.9901525974273682,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20019636303186417,
      "step": 28352
    },
    {
      "epoch": 0.56708,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.005418904622395833,
      "learning_rate": 0.0001,
      "loss": 4.0514,
      "loss/crossentropy": 2.246248483657837,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20701756328344345,
      "step": 28354
    },
    {
      "epoch": 0.56712,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.011136881510416667,
      "learning_rate": 0.0001,
      "loss": 3.8456,
      "loss/crossentropy": 1.9263428449630737,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19890911877155304,
      "step": 28356
    },
    {
      "epoch": 0.56716,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.009338124593098959,
      "learning_rate": 0.0001,
      "loss": 4.0555,
      "loss/crossentropy": 2.2119653820991516,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20472905784845352,
      "step": 28358
    },
    {
      "epoch": 0.5672,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.010406239827473959,
      "learning_rate": 0.0001,
      "loss": 4.2023,
      "loss/crossentropy": 2.1126968264579773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18353600800037384,
      "step": 28360
    },
    {
      "epoch": 0.56724,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.010929361979166666,
      "learning_rate": 0.0001,
      "loss": 3.8412,
      "loss/crossentropy": 1.7391886115074158,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1703544184565544,
      "step": 28362
    },
    {
      "epoch": 0.56728,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.010689036051432291,
      "learning_rate": 0.0001,
      "loss": 3.8372,
      "loss/crossentropy": 1.8684263229370117,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18913953006267548,
      "step": 28364
    },
    {
      "epoch": 0.56732,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.010699208577473958,
      "learning_rate": 0.0001,
      "loss": 3.7079,
      "loss/crossentropy": 1.971863567829132,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.188376322388649,
      "step": 28366
    },
    {
      "epoch": 0.56736,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010827382405598959,
      "learning_rate": 0.0001,
      "loss": 4.1637,
      "loss/crossentropy": 1.8890693187713623,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18029199540615082,
      "step": 28368
    },
    {
      "epoch": 0.5674,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.011861165364583334,
      "learning_rate": 0.0001,
      "loss": 4.2677,
      "loss/crossentropy": 1.9506273865699768,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19081934541463852,
      "step": 28370
    },
    {
      "epoch": 0.56744,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.006463368733723958,
      "learning_rate": 0.0001,
      "loss": 4.0055,
      "loss/crossentropy": 1.937731921672821,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21456565707921982,
      "step": 28372
    },
    {
      "epoch": 0.56748,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.006891886393229167,
      "learning_rate": 0.0001,
      "loss": 4.0904,
      "loss/crossentropy": 2.0715887546539307,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21278052031993866,
      "step": 28374
    },
    {
      "epoch": 0.56752,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.007413736979166667,
      "learning_rate": 0.0001,
      "loss": 3.9901,
      "loss/crossentropy": 2.2458627223968506,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20296980440616608,
      "step": 28376
    },
    {
      "epoch": 0.56756,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.006886545817057292,
      "learning_rate": 0.0001,
      "loss": 4.178,
      "loss/crossentropy": 2.3213919401168823,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19989027082920074,
      "step": 28378
    },
    {
      "epoch": 0.5676,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.007500966389973958,
      "learning_rate": 0.0001,
      "loss": 4.2319,
      "loss/crossentropy": 2.543839931488037,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22476375102996826,
      "step": 28380
    },
    {
      "epoch": 0.56764,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.006084950764973959,
      "learning_rate": 0.0001,
      "loss": 3.9426,
      "loss/crossentropy": 2.1060808897018433,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1993144229054451,
      "step": 28382
    },
    {
      "epoch": 0.56768,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0074859619140625,
      "learning_rate": 0.0001,
      "loss": 3.8902,
      "loss/crossentropy": 1.9778028726577759,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18509671837091446,
      "step": 28384
    },
    {
      "epoch": 0.56772,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.010445149739583333,
      "learning_rate": 0.0001,
      "loss": 3.7404,
      "loss/crossentropy": 1.7300578951835632,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17200996726751328,
      "step": 28386
    },
    {
      "epoch": 0.56776,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.011352284749348959,
      "learning_rate": 0.0001,
      "loss": 3.8029,
      "loss/crossentropy": 1.87715744972229,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1691356599330902,
      "step": 28388
    },
    {
      "epoch": 0.5678,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.01263427734375,
      "learning_rate": 0.0001,
      "loss": 3.7688,
      "loss/crossentropy": 1.9529326558113098,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18868190050125122,
      "step": 28390
    },
    {
      "epoch": 0.56784,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.012654622395833334,
      "learning_rate": 0.0001,
      "loss": 3.9265,
      "loss/crossentropy": 1.9922877550125122,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18494701385498047,
      "step": 28392
    },
    {
      "epoch": 0.56788,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.014012654622395834,
      "learning_rate": 0.0001,
      "loss": 4.0745,
      "loss/crossentropy": 2.1247864961624146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20046664774417877,
      "step": 28394
    },
    {
      "epoch": 0.56792,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.019334920247395835,
      "learning_rate": 0.0001,
      "loss": 3.9056,
      "loss/crossentropy": 2.106461524963379,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1846928596496582,
      "step": 28396
    },
    {
      "epoch": 0.56796,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.02042210896809896,
      "learning_rate": 0.0001,
      "loss": 4.0709,
      "loss/crossentropy": 2.5050116777420044,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23650984466075897,
      "step": 28398
    },
    {
      "epoch": 0.568,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.01956151326497396,
      "learning_rate": 0.0001,
      "loss": 3.9026,
      "loss/crossentropy": 1.8201875686645508,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18172843009233475,
      "step": 28400
    },
    {
      "epoch": 0.56804,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.014680735270182292,
      "learning_rate": 0.0001,
      "loss": 3.7716,
      "loss/crossentropy": 1.748970091342926,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1714978739619255,
      "step": 28402
    },
    {
      "epoch": 0.56808,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.013703409830729167,
      "learning_rate": 0.0001,
      "loss": 3.9609,
      "loss/crossentropy": 2.027329981327057,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16597037762403488,
      "step": 28404
    },
    {
      "epoch": 0.56812,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.012400054931640625,
      "learning_rate": 0.0001,
      "loss": 3.8902,
      "loss/crossentropy": 1.8651673197746277,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17772310227155685,
      "step": 28406
    },
    {
      "epoch": 0.56816,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.011869049072265625,
      "learning_rate": 0.0001,
      "loss": 4.0313,
      "loss/crossentropy": 2.1574735045433044,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2000775933265686,
      "step": 28408
    },
    {
      "epoch": 0.5682,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.010861968994140625,
      "learning_rate": 0.0001,
      "loss": 4.1171,
      "loss/crossentropy": 2.1159602403640747,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19108104705810547,
      "step": 28410
    },
    {
      "epoch": 0.56824,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.008829498291015625,
      "learning_rate": 0.0001,
      "loss": 4.229,
      "loss/crossentropy": 2.203667163848877,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2056882083415985,
      "step": 28412
    },
    {
      "epoch": 0.56828,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.008381907145182292,
      "learning_rate": 0.0001,
      "loss": 4.0476,
      "loss/crossentropy": 2.0054327845573425,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1914311721920967,
      "step": 28414
    },
    {
      "epoch": 0.56832,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.008489735921223958,
      "learning_rate": 0.0001,
      "loss": 4.1249,
      "loss/crossentropy": 2.199601411819458,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2020196169614792,
      "step": 28416
    },
    {
      "epoch": 0.56836,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.007682291666666666,
      "learning_rate": 0.0001,
      "loss": 4.1068,
      "loss/crossentropy": 1.84443199634552,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17476340383291245,
      "step": 28418
    },
    {
      "epoch": 0.5684,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.007624308268229167,
      "learning_rate": 0.0001,
      "loss": 3.9032,
      "loss/crossentropy": 2.2852976322174072,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21395886689424515,
      "step": 28420
    },
    {
      "epoch": 0.56844,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.009956614176432291,
      "learning_rate": 0.0001,
      "loss": 4.0763,
      "loss/crossentropy": 2.2443678975105286,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19366547465324402,
      "step": 28422
    },
    {
      "epoch": 0.56848,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.012784830729166667,
      "learning_rate": 0.0001,
      "loss": 4.2711,
      "loss/crossentropy": 2.055034577846527,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1914302259683609,
      "step": 28424
    },
    {
      "epoch": 0.56852,
      "grad_norm": 1.875,
      "grad_norm_var": 0.012245432535807291,
      "learning_rate": 0.0001,
      "loss": 4.0782,
      "loss/crossentropy": 2.33309006690979,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2132418155670166,
      "step": 28426
    },
    {
      "epoch": 0.56856,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.009246571858723959,
      "learning_rate": 0.0001,
      "loss": 4.2069,
      "loss/crossentropy": 2.4401670694351196,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20467859506607056,
      "step": 28428
    },
    {
      "epoch": 0.5686,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.015778605143229166,
      "learning_rate": 0.0001,
      "loss": 3.9526,
      "loss/crossentropy": 1.8660191297531128,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23643852770328522,
      "step": 28430
    },
    {
      "epoch": 0.56864,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.016080729166666665,
      "learning_rate": 0.0001,
      "loss": 4.1835,
      "loss/crossentropy": 2.026413381099701,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.25309374928474426,
      "step": 28432
    },
    {
      "epoch": 0.56868,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.01834691365559896,
      "learning_rate": 0.0001,
      "loss": 4.0417,
      "loss/crossentropy": 2.1234898567199707,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19204584509134293,
      "step": 28434
    },
    {
      "epoch": 0.56872,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.017292277018229166,
      "learning_rate": 0.0001,
      "loss": 4.1999,
      "loss/crossentropy": 2.3347290754318237,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21934974193572998,
      "step": 28436
    },
    {
      "epoch": 0.56876,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.016076405843098957,
      "learning_rate": 0.0001,
      "loss": 3.8796,
      "loss/crossentropy": 2.0287744402885437,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1834561824798584,
      "step": 28438
    },
    {
      "epoch": 0.5688,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.014654286702473958,
      "learning_rate": 0.0001,
      "loss": 4.0872,
      "loss/crossentropy": 1.782648503780365,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1810266673564911,
      "step": 28440
    },
    {
      "epoch": 0.56884,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.016139475504557292,
      "learning_rate": 0.0001,
      "loss": 4.0876,
      "loss/crossentropy": 2.2251389622688293,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2041720598936081,
      "step": 28442
    },
    {
      "epoch": 0.56888,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.018846638997395835,
      "learning_rate": 0.0001,
      "loss": 3.7876,
      "loss/crossentropy": 1.9368031024932861,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1828494518995285,
      "step": 28444
    },
    {
      "epoch": 0.56892,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.011763254801432291,
      "learning_rate": 0.0001,
      "loss": 3.9775,
      "loss/crossentropy": 2.0573925971984863,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.192618265748024,
      "step": 28446
    },
    {
      "epoch": 0.56896,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0088775634765625,
      "learning_rate": 0.0001,
      "loss": 3.8321,
      "loss/crossentropy": 1.9373971223831177,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20026732236146927,
      "step": 28448
    },
    {
      "epoch": 0.569,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.007391103108723958,
      "learning_rate": 0.0001,
      "loss": 3.8771,
      "loss/crossentropy": 1.877286970615387,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17163537442684174,
      "step": 28450
    },
    {
      "epoch": 0.56904,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0054931640625,
      "learning_rate": 0.0001,
      "loss": 3.8502,
      "loss/crossentropy": 2.1928617358207703,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20396625250577927,
      "step": 28452
    },
    {
      "epoch": 0.56908,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.008373006184895834,
      "learning_rate": 0.0001,
      "loss": 3.9737,
      "loss/crossentropy": 2.087548613548279,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2330121174454689,
      "step": 28454
    },
    {
      "epoch": 0.56912,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.008275349934895834,
      "learning_rate": 0.0001,
      "loss": 4.24,
      "loss/crossentropy": 2.0506752729415894,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19728151708841324,
      "step": 28456
    },
    {
      "epoch": 0.56916,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.006174468994140625,
      "learning_rate": 0.0001,
      "loss": 4.1038,
      "loss/crossentropy": 2.3582634925842285,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21210015565156937,
      "step": 28458
    },
    {
      "epoch": 0.5692,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0051422119140625,
      "learning_rate": 0.0001,
      "loss": 3.8762,
      "loss/crossentropy": 2.3920403718948364,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20361161977052689,
      "step": 28460
    },
    {
      "epoch": 0.56924,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0048248291015625,
      "learning_rate": 0.0001,
      "loss": 3.9837,
      "loss/crossentropy": 2.077300548553467,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18526732921600342,
      "step": 28462
    },
    {
      "epoch": 0.56928,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.0059967041015625,
      "learning_rate": 0.0001,
      "loss": 3.7625,
      "loss/crossentropy": 2.0746008157730103,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.176588736474514,
      "step": 28464
    },
    {
      "epoch": 0.56932,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.005008697509765625,
      "learning_rate": 0.0001,
      "loss": 4.0275,
      "loss/crossentropy": 2.651300311088562,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22577688097953796,
      "step": 28466
    },
    {
      "epoch": 0.56936,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.006477864583333334,
      "learning_rate": 0.0001,
      "loss": 3.9464,
      "loss/crossentropy": 2.169211745262146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20319853723049164,
      "step": 28468
    },
    {
      "epoch": 0.5694,
      "grad_norm": 1.7109375,
      "grad_norm_var": 0.005673980712890625,
      "learning_rate": 0.0001,
      "loss": 3.7715,
      "loss/crossentropy": 2.16380512714386,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1928708478808403,
      "step": 28470
    },
    {
      "epoch": 0.56944,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0054094950358072914,
      "learning_rate": 0.0001,
      "loss": 3.8378,
      "loss/crossentropy": 1.780498206615448,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1736089140176773,
      "step": 28472
    },
    {
      "epoch": 0.56948,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.005597941080729167,
      "learning_rate": 0.0001,
      "loss": 4.1348,
      "loss/crossentropy": 1.9082182049751282,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17953625321388245,
      "step": 28474
    },
    {
      "epoch": 0.56952,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.005499013264973958,
      "learning_rate": 0.0001,
      "loss": 3.8997,
      "loss/crossentropy": 1.6533525586128235,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16805779933929443,
      "step": 28476
    },
    {
      "epoch": 0.56956,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.006882476806640625,
      "learning_rate": 0.0001,
      "loss": 4.0518,
      "loss/crossentropy": 2.0916685461997986,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1956561803817749,
      "step": 28478
    },
    {
      "epoch": 0.5696,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0078277587890625,
      "learning_rate": 0.0001,
      "loss": 4.1282,
      "loss/crossentropy": 1.9846032857894897,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17997410148382187,
      "step": 28480
    },
    {
      "epoch": 0.56964,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.008396148681640625,
      "learning_rate": 0.0001,
      "loss": 4.0715,
      "loss/crossentropy": 1.9905300736427307,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17957545816898346,
      "step": 28482
    },
    {
      "epoch": 0.56968,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.007356516520182292,
      "learning_rate": 0.0001,
      "loss": 3.7545,
      "loss/crossentropy": 1.9274362921714783,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20025887340307236,
      "step": 28484
    },
    {
      "epoch": 0.56972,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.003981272379557292,
      "learning_rate": 0.0001,
      "loss": 3.8606,
      "loss/crossentropy": 1.7590890526771545,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17347681522369385,
      "step": 28486
    },
    {
      "epoch": 0.56976,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.003979237874348959,
      "learning_rate": 0.0001,
      "loss": 3.9737,
      "loss/crossentropy": 1.8959112763404846,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18316560983657837,
      "step": 28488
    },
    {
      "epoch": 0.5698,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.004292805989583333,
      "learning_rate": 0.0001,
      "loss": 3.9393,
      "loss/crossentropy": 2.108347535133362,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20547642558813095,
      "step": 28490
    },
    {
      "epoch": 0.56984,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.003482818603515625,
      "learning_rate": 0.0001,
      "loss": 3.8703,
      "loss/crossentropy": 1.9661864638328552,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1860649138689041,
      "step": 28492
    },
    {
      "epoch": 0.56988,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0029111226399739583,
      "learning_rate": 0.0001,
      "loss": 3.8467,
      "loss/crossentropy": 1.7892447710037231,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16265498101711273,
      "step": 28494
    },
    {
      "epoch": 0.56992,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.0030507405598958334,
      "learning_rate": 0.0001,
      "loss": 4.1363,
      "loss/crossentropy": 2.2188740968704224,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20954828709363937,
      "step": 28496
    },
    {
      "epoch": 0.56996,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.005028279622395834,
      "learning_rate": 0.0001,
      "loss": 4.1339,
      "loss/crossentropy": 2.1535292863845825,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19683580845594406,
      "step": 28498
    },
    {
      "epoch": 0.57,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0048258463541666664,
      "learning_rate": 0.0001,
      "loss": 3.8687,
      "loss/crossentropy": 2.170483350753784,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20434408634901047,
      "step": 28500
    },
    {
      "epoch": 0.57004,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.004735310872395833,
      "learning_rate": 0.0001,
      "loss": 3.9846,
      "loss/crossentropy": 1.9062520265579224,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18132874369621277,
      "step": 28502
    },
    {
      "epoch": 0.57008,
      "grad_norm": 2.0,
      "grad_norm_var": 0.005014801025390625,
      "learning_rate": 0.0001,
      "loss": 3.9361,
      "loss/crossentropy": 1.9402993321418762,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1883842498064041,
      "step": 28504
    },
    {
      "epoch": 0.57012,
      "grad_norm": 2.25,
      "grad_norm_var": 0.010516103108723958,
      "learning_rate": 0.0001,
      "loss": 4.2954,
      "loss/crossentropy": 2.1244900226593018,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19686578959226608,
      "step": 28506
    },
    {
      "epoch": 0.57016,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.010432688395182292,
      "learning_rate": 0.0001,
      "loss": 4.213,
      "loss/crossentropy": 2.255519151687622,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19818376004695892,
      "step": 28508
    },
    {
      "epoch": 0.5702,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.017838541666666666,
      "learning_rate": 0.0001,
      "loss": 4.3414,
      "loss/crossentropy": 1.69990873336792,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16047322005033493,
      "step": 28510
    },
    {
      "epoch": 0.57024,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.01759618123372396,
      "learning_rate": 0.0001,
      "loss": 4.0451,
      "loss/crossentropy": 2.0938282012939453,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19069606065750122,
      "step": 28512
    },
    {
      "epoch": 0.57028,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.015746053059895834,
      "learning_rate": 0.0001,
      "loss": 3.9285,
      "loss/crossentropy": 2.119936943054199,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20535118132829666,
      "step": 28514
    },
    {
      "epoch": 0.57032,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.015364583333333333,
      "learning_rate": 0.0001,
      "loss": 4.1106,
      "loss/crossentropy": 2.124122440814972,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19785825163125992,
      "step": 28516
    },
    {
      "epoch": 0.57036,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.018342081705729166,
      "learning_rate": 0.0001,
      "loss": 4.025,
      "loss/crossentropy": 2.275521993637085,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.197978176176548,
      "step": 28518
    },
    {
      "epoch": 0.5704,
      "grad_norm": 1.75,
      "grad_norm_var": 0.021089680989583335,
      "learning_rate": 0.0001,
      "loss": 3.8305,
      "loss/crossentropy": 1.9788197875022888,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17468783259391785,
      "step": 28520
    },
    {
      "epoch": 0.57044,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.015799713134765626,
      "learning_rate": 0.0001,
      "loss": 3.9497,
      "loss/crossentropy": 1.9069678783416748,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17532987147569656,
      "step": 28522
    },
    {
      "epoch": 0.57048,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.015705362955729166,
      "learning_rate": 0.0001,
      "loss": 3.9084,
      "loss/crossentropy": 1.7438762784004211,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15763580799102783,
      "step": 28524
    },
    {
      "epoch": 0.57052,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.008280436197916666,
      "learning_rate": 0.0001,
      "loss": 4.0142,
      "loss/crossentropy": 2.0234435200691223,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17508037388324738,
      "step": 28526
    },
    {
      "epoch": 0.57056,
      "grad_norm": 2.0,
      "grad_norm_var": 0.008571116129557292,
      "learning_rate": 0.0001,
      "loss": 4.1993,
      "loss/crossentropy": 2.2856411933898926,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20183924585580826,
      "step": 28528
    },
    {
      "epoch": 0.5706,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.010358683268229167,
      "learning_rate": 0.0001,
      "loss": 3.9569,
      "loss/crossentropy": 2.201618194580078,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2216714471578598,
      "step": 28530
    },
    {
      "epoch": 0.57064,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.011348215738932292,
      "learning_rate": 0.0001,
      "loss": 4.017,
      "loss/crossentropy": 1.9202881455421448,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18147829920053482,
      "step": 28532
    },
    {
      "epoch": 0.57068,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.009733072916666667,
      "learning_rate": 0.0001,
      "loss": 3.7967,
      "loss/crossentropy": 1.5407178401947021,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.14478188008069992,
      "step": 28534
    },
    {
      "epoch": 0.57072,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.010920206705729166,
      "learning_rate": 0.0001,
      "loss": 3.8766,
      "loss/crossentropy": 1.836295247077942,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18802417069673538,
      "step": 28536
    },
    {
      "epoch": 0.57076,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.013553619384765625,
      "learning_rate": 0.0001,
      "loss": 4.1791,
      "loss/crossentropy": 2.043636739253998,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19008664041757584,
      "step": 28538
    },
    {
      "epoch": 0.5708,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.014745076497395834,
      "learning_rate": 0.0001,
      "loss": 4.1531,
      "loss/crossentropy": 1.8939302563667297,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18854600936174393,
      "step": 28540
    },
    {
      "epoch": 0.57084,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.012658437093098959,
      "learning_rate": 0.0001,
      "loss": 4.0664,
      "loss/crossentropy": 2.1033613085746765,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17389168590307236,
      "step": 28542
    },
    {
      "epoch": 0.57088,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.013230133056640624,
      "learning_rate": 0.0001,
      "loss": 4.1976,
      "loss/crossentropy": 2.245096266269684,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21308264881372452,
      "step": 28544
    },
    {
      "epoch": 0.57092,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.012910715738932292,
      "learning_rate": 0.0001,
      "loss": 3.8638,
      "loss/crossentropy": 1.6646681427955627,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16192668676376343,
      "step": 28546
    },
    {
      "epoch": 0.57096,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.011922200520833334,
      "learning_rate": 0.0001,
      "loss": 3.8962,
      "loss/crossentropy": 2.29884135723114,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17613109946250916,
      "step": 28548
    },
    {
      "epoch": 0.571,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.01628392537434896,
      "learning_rate": 0.0001,
      "loss": 4.0125,
      "loss/crossentropy": 2.10237193107605,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1935042291879654,
      "step": 28550
    },
    {
      "epoch": 0.57104,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.015048980712890625,
      "learning_rate": 0.0001,
      "loss": 4.0384,
      "loss/crossentropy": 2.1097005009651184,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1907399743795395,
      "step": 28552
    },
    {
      "epoch": 0.57108,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.012824503580729167,
      "learning_rate": 0.0001,
      "loss": 4.155,
      "loss/crossentropy": 2.1058011054992676,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19711777567863464,
      "step": 28554
    },
    {
      "epoch": 0.57112,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.011659495035807292,
      "learning_rate": 0.0001,
      "loss": 3.9603,
      "loss/crossentropy": 2.0466136932373047,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18761292099952698,
      "step": 28556
    },
    {
      "epoch": 0.57116,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.011759440104166666,
      "learning_rate": 0.0001,
      "loss": 3.6357,
      "loss/crossentropy": 1.6006923913955688,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17703057825565338,
      "step": 28558
    },
    {
      "epoch": 0.5712,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.011183420817057291,
      "learning_rate": 0.0001,
      "loss": 3.8007,
      "loss/crossentropy": 1.884728193283081,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18712394684553146,
      "step": 28560
    },
    {
      "epoch": 0.57124,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.009718577067057291,
      "learning_rate": 0.0001,
      "loss": 3.8565,
      "loss/crossentropy": 1.8759589791297913,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19130942225456238,
      "step": 28562
    },
    {
      "epoch": 0.57128,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.009785715738932292,
      "learning_rate": 0.0001,
      "loss": 4.1392,
      "loss/crossentropy": 1.9973544478416443,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18966401368379593,
      "step": 28564
    },
    {
      "epoch": 0.57132,
      "grad_norm": 1.75,
      "grad_norm_var": 0.007368723551432292,
      "learning_rate": 0.0001,
      "loss": 3.9515,
      "loss/crossentropy": 2.1246695518493652,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1938970983028412,
      "step": 28566
    },
    {
      "epoch": 0.57136,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0065093994140625,
      "learning_rate": 0.0001,
      "loss": 3.7797,
      "loss/crossentropy": 1.5528571605682373,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15729191899299622,
      "step": 28568
    },
    {
      "epoch": 0.5714,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0068023681640625,
      "learning_rate": 0.0001,
      "loss": 3.995,
      "loss/crossentropy": 2.0078552961349487,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19063813984394073,
      "step": 28570
    },
    {
      "epoch": 0.57144,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0072021484375,
      "learning_rate": 0.0001,
      "loss": 4.0105,
      "loss/crossentropy": 1.8582743406295776,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16606975346803665,
      "step": 28572
    },
    {
      "epoch": 0.57148,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.008784739176432292,
      "learning_rate": 0.0001,
      "loss": 3.7379,
      "loss/crossentropy": 2.1142138242721558,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19289004802703857,
      "step": 28574
    },
    {
      "epoch": 0.57152,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.008147939046223959,
      "learning_rate": 0.0001,
      "loss": 3.9771,
      "loss/crossentropy": 2.0852617621421814,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18875133991241455,
      "step": 28576
    },
    {
      "epoch": 0.57156,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.007541656494140625,
      "learning_rate": 0.0001,
      "loss": 4.0243,
      "loss/crossentropy": 2.1352654695510864,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19411734491586685,
      "step": 28578
    },
    {
      "epoch": 0.5716,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.0069048563639322914,
      "learning_rate": 0.0001,
      "loss": 3.5985,
      "loss/crossentropy": 1.7001954317092896,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17129848897457123,
      "step": 28580
    },
    {
      "epoch": 0.57164,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.004894765218098959,
      "learning_rate": 0.0001,
      "loss": 3.7744,
      "loss/crossentropy": 1.8788430094718933,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17039231956005096,
      "step": 28582
    },
    {
      "epoch": 0.57168,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.005307769775390625,
      "learning_rate": 0.0001,
      "loss": 4.0533,
      "loss/crossentropy": 1.9935740232467651,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18447662144899368,
      "step": 28584
    },
    {
      "epoch": 0.57172,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0067779541015625,
      "learning_rate": 0.0001,
      "loss": 3.652,
      "loss/crossentropy": 1.6966716051101685,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17215370386838913,
      "step": 28586
    },
    {
      "epoch": 0.57176,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.006841786702473958,
      "learning_rate": 0.0001,
      "loss": 3.6792,
      "loss/crossentropy": 1.8064610958099365,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18994714319705963,
      "step": 28588
    },
    {
      "epoch": 0.5718,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.006956990559895833,
      "learning_rate": 0.0001,
      "loss": 3.8408,
      "loss/crossentropy": 1.8688351511955261,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17651986330747604,
      "step": 28590
    },
    {
      "epoch": 0.57184,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.023482259114583334,
      "learning_rate": 0.0001,
      "loss": 3.9822,
      "loss/crossentropy": 1.9788597226142883,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21657729148864746,
      "step": 28592
    },
    {
      "epoch": 0.57188,
      "grad_norm": 2.125,
      "grad_norm_var": 0.02662938435872396,
      "learning_rate": 0.0001,
      "loss": 4.2794,
      "loss/crossentropy": 2.2449204325675964,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20486489683389664,
      "step": 28594
    },
    {
      "epoch": 0.57192,
      "grad_norm": 1.875,
      "grad_norm_var": 0.02296727498372396,
      "learning_rate": 0.0001,
      "loss": 3.9333,
      "loss/crossentropy": 1.9045695662498474,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19815212488174438,
      "step": 28596
    },
    {
      "epoch": 0.57196,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.025742340087890624,
      "learning_rate": 0.0001,
      "loss": 3.7332,
      "loss/crossentropy": 1.6905238628387451,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1561586558818817,
      "step": 28598
    },
    {
      "epoch": 0.572,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.03126220703125,
      "learning_rate": 0.0001,
      "loss": 3.7146,
      "loss/crossentropy": 1.788037657737732,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17239058762788773,
      "step": 28600
    },
    {
      "epoch": 0.57204,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.02777684529622396,
      "learning_rate": 0.0001,
      "loss": 4.0529,
      "loss/crossentropy": 1.9274045825004578,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20560631155967712,
      "step": 28602
    },
    {
      "epoch": 0.57208,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.028820546468098958,
      "learning_rate": 0.0001,
      "loss": 3.8056,
      "loss/crossentropy": 1.909210741519928,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17594803869724274,
      "step": 28604
    },
    {
      "epoch": 0.57212,
      "grad_norm": 2.125,
      "grad_norm_var": 0.028362782796223958,
      "learning_rate": 0.0001,
      "loss": 3.9912,
      "loss/crossentropy": 2.0221698880195618,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1869770884513855,
      "step": 28606
    },
    {
      "epoch": 0.57216,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.01813329060872396,
      "learning_rate": 0.0001,
      "loss": 4.184,
      "loss/crossentropy": 2.235592007637024,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17893031239509583,
      "step": 28608
    },
    {
      "epoch": 0.5722,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.028303019205729165,
      "learning_rate": 0.0001,
      "loss": 4.0714,
      "loss/crossentropy": 2.098126530647278,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2068413719534874,
      "step": 28610
    },
    {
      "epoch": 0.57224,
      "grad_norm": 1.75,
      "grad_norm_var": 0.03432184855143229,
      "learning_rate": 0.0001,
      "loss": 3.7392,
      "loss/crossentropy": 1.97664475440979,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18118216842412949,
      "step": 28612
    },
    {
      "epoch": 0.57228,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.03205540974934896,
      "learning_rate": 0.0001,
      "loss": 3.9675,
      "loss/crossentropy": 1.9696056246757507,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18983419984579086,
      "step": 28614
    },
    {
      "epoch": 0.57232,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0686767578125,
      "learning_rate": 0.0001,
      "loss": 4.0106,
      "loss/crossentropy": 2.1281611919403076,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22855115681886673,
      "step": 28616
    },
    {
      "epoch": 0.57236,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.07779032389322917,
      "learning_rate": 0.0001,
      "loss": 3.987,
      "loss/crossentropy": 1.7333523631095886,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1650746390223503,
      "step": 28618
    },
    {
      "epoch": 0.5724,
      "grad_norm": 2.71875,
      "grad_norm_var": 0.10811258951822916,
      "learning_rate": 0.0001,
      "loss": 4.2372,
      "loss/crossentropy": 2.021788716316223,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19636931270360947,
      "step": 28620
    },
    {
      "epoch": 0.57244,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.1073394775390625,
      "learning_rate": 0.0001,
      "loss": 3.8144,
      "loss/crossentropy": 1.7764569520950317,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18557444959878922,
      "step": 28622
    },
    {
      "epoch": 0.57248,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.10955301920572917,
      "learning_rate": 0.0001,
      "loss": 3.8214,
      "loss/crossentropy": 1.8160730004310608,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1741533875465393,
      "step": 28624
    },
    {
      "epoch": 0.57252,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.1038225809733073,
      "learning_rate": 0.0001,
      "loss": 3.7493,
      "loss/crossentropy": 1.7295884490013123,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17627893388271332,
      "step": 28626
    },
    {
      "epoch": 0.57256,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.08949381510416667,
      "learning_rate": 0.0001,
      "loss": 3.9968,
      "loss/crossentropy": 1.9967339038848877,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1878109648823738,
      "step": 28628
    },
    {
      "epoch": 0.5726,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.09168294270833334,
      "learning_rate": 0.0001,
      "loss": 3.9292,
      "loss/crossentropy": 2.4338849782943726,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2211342379450798,
      "step": 28630
    },
    {
      "epoch": 0.57264,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.057566070556640626,
      "learning_rate": 0.0001,
      "loss": 3.928,
      "loss/crossentropy": 2.0511388778686523,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2051515057682991,
      "step": 28632
    },
    {
      "epoch": 0.57268,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.05156021118164063,
      "learning_rate": 0.0001,
      "loss": 3.5661,
      "loss/crossentropy": 1.7425475716590881,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18100523948669434,
      "step": 28634
    },
    {
      "epoch": 0.57272,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.004343414306640625,
      "learning_rate": 0.0001,
      "loss": 4.2029,
      "loss/crossentropy": 2.0718079805374146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1951262131333351,
      "step": 28636
    },
    {
      "epoch": 0.57276,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.004514312744140625,
      "learning_rate": 0.0001,
      "loss": 4.0589,
      "loss/crossentropy": 1.9455790519714355,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2024966925382614,
      "step": 28638
    },
    {
      "epoch": 0.5728,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.006075032552083333,
      "learning_rate": 0.0001,
      "loss": 3.8238,
      "loss/crossentropy": 1.9923955798149109,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17989075928926468,
      "step": 28640
    },
    {
      "epoch": 0.57284,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.005708567301432292,
      "learning_rate": 0.0001,
      "loss": 4.0637,
      "loss/crossentropy": 2.178214907646179,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1973334327340126,
      "step": 28642
    },
    {
      "epoch": 0.57288,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.004349517822265625,
      "learning_rate": 0.0001,
      "loss": 3.9935,
      "loss/crossentropy": 1.881864607334137,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17736243456602097,
      "step": 28644
    },
    {
      "epoch": 0.57292,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.004988352457682292,
      "learning_rate": 0.0001,
      "loss": 4.2611,
      "loss/crossentropy": 2.140734553337097,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2056664153933525,
      "step": 28646
    },
    {
      "epoch": 0.57296,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0049479166666666664,
      "learning_rate": 0.0001,
      "loss": 4.1077,
      "loss/crossentropy": 2.2030951976776123,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1997833326458931,
      "step": 28648
    },
    {
      "epoch": 0.573,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0045888264973958336,
      "learning_rate": 0.0001,
      "loss": 4.1237,
      "loss/crossentropy": 1.8152233958244324,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17497697472572327,
      "step": 28650
    },
    {
      "epoch": 0.57304,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.008056386311848959,
      "learning_rate": 0.0001,
      "loss": 4.4633,
      "loss/crossentropy": 2.2969506978988647,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22079680860042572,
      "step": 28652
    },
    {
      "epoch": 0.57308,
      "grad_norm": 1.875,
      "grad_norm_var": 0.010550689697265626,
      "learning_rate": 0.0001,
      "loss": 4.0007,
      "loss/crossentropy": 2.058579981327057,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1992625817656517,
      "step": 28654
    },
    {
      "epoch": 0.57312,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0073964436848958336,
      "learning_rate": 0.0001,
      "loss": 3.9863,
      "loss/crossentropy": 1.9630435109138489,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19192881137132645,
      "step": 28656
    },
    {
      "epoch": 0.57316,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.008408355712890624,
      "learning_rate": 0.0001,
      "loss": 3.9874,
      "loss/crossentropy": 2.089448630809784,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.185981385409832,
      "step": 28658
    },
    {
      "epoch": 0.5732,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0088287353515625,
      "learning_rate": 0.0001,
      "loss": 3.8705,
      "loss/crossentropy": 2.1264119148254395,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18827120959758759,
      "step": 28660
    },
    {
      "epoch": 0.57324,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.0121978759765625,
      "learning_rate": 0.0001,
      "loss": 3.8047,
      "loss/crossentropy": 2.121288537979126,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1772882044315338,
      "step": 28662
    },
    {
      "epoch": 0.57328,
      "grad_norm": 2.484375,
      "grad_norm_var": 0.03190689086914063,
      "learning_rate": 0.0001,
      "loss": 4.1513,
      "loss/crossentropy": 2.0339037775993347,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1962498426437378,
      "step": 28664
    },
    {
      "epoch": 0.57332,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.03292210896809896,
      "learning_rate": 0.0001,
      "loss": 4.0585,
      "loss/crossentropy": 2.1422963738441467,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18338894098997116,
      "step": 28666
    },
    {
      "epoch": 0.57336,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.030590565999348958,
      "learning_rate": 0.0001,
      "loss": 4.2466,
      "loss/crossentropy": 2.029167354106903,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20171435922384262,
      "step": 28668
    },
    {
      "epoch": 0.5734,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.032225545247395834,
      "learning_rate": 0.0001,
      "loss": 3.918,
      "loss/crossentropy": 1.7628564238548279,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19051892310380936,
      "step": 28670
    },
    {
      "epoch": 0.57344,
      "grad_norm": 1.734375,
      "grad_norm_var": 0.03588053385416667,
      "learning_rate": 0.0001,
      "loss": 3.7368,
      "loss/crossentropy": 2.0292606949806213,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17968103289604187,
      "step": 28672
    },
    {
      "epoch": 0.57348,
      "grad_norm": 2.125,
      "grad_norm_var": 0.044130198160807294,
      "learning_rate": 0.0001,
      "loss": 4.2414,
      "loss/crossentropy": 1.9167731404304504,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22261708974838257,
      "step": 28674
    },
    {
      "epoch": 0.57352,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.04302164713541667,
      "learning_rate": 0.0001,
      "loss": 3.9668,
      "loss/crossentropy": 1.9671185612678528,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19196172803640366,
      "step": 28676
    },
    {
      "epoch": 0.57356,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.037666575113932295,
      "learning_rate": 0.0001,
      "loss": 3.8129,
      "loss/crossentropy": 2.0191508531570435,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18491507321596146,
      "step": 28678
    },
    {
      "epoch": 0.5736,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.023209635416666666,
      "learning_rate": 0.0001,
      "loss": 4.0662,
      "loss/crossentropy": 1.9060558676719666,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18243618309497833,
      "step": 28680
    },
    {
      "epoch": 0.57364,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0218658447265625,
      "learning_rate": 0.0001,
      "loss": 4.0068,
      "loss/crossentropy": 1.867979109287262,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23452743887901306,
      "step": 28682
    },
    {
      "epoch": 0.57368,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.021354166666666667,
      "learning_rate": 0.0001,
      "loss": 4.079,
      "loss/crossentropy": 1.6199312806129456,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16022612154483795,
      "step": 28684
    },
    {
      "epoch": 0.57372,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0176422119140625,
      "learning_rate": 0.0001,
      "loss": 4.191,
      "loss/crossentropy": 2.5148130655288696,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2360077053308487,
      "step": 28686
    },
    {
      "epoch": 0.57376,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.015126291910807292,
      "learning_rate": 0.0001,
      "loss": 4.032,
      "loss/crossentropy": 1.9117471575737,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2010490596294403,
      "step": 28688
    },
    {
      "epoch": 0.5738,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.006821441650390625,
      "learning_rate": 0.0001,
      "loss": 4.0699,
      "loss/crossentropy": 2.079534113407135,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18718824535608292,
      "step": 28690
    },
    {
      "epoch": 0.57384,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.00633544921875,
      "learning_rate": 0.0001,
      "loss": 3.97,
      "loss/crossentropy": 2.187251031398773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1929212063550949,
      "step": 28692
    },
    {
      "epoch": 0.57388,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.006023915608723959,
      "learning_rate": 0.0001,
      "loss": 4.0509,
      "loss/crossentropy": 1.9140017628669739,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1726428121328354,
      "step": 28694
    },
    {
      "epoch": 0.57392,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0047515869140625,
      "learning_rate": 0.0001,
      "loss": 3.9645,
      "loss/crossentropy": 1.530074417591095,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15649892389774323,
      "step": 28696
    },
    {
      "epoch": 0.57396,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.006998443603515625,
      "learning_rate": 0.0001,
      "loss": 4.1736,
      "loss/crossentropy": 2.2588948011398315,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20315733551979065,
      "step": 28698
    },
    {
      "epoch": 0.574,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0094390869140625,
      "learning_rate": 0.0001,
      "loss": 4.0446,
      "loss/crossentropy": 2.1702714562416077,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24278688430786133,
      "step": 28700
    },
    {
      "epoch": 0.57404,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.009789021809895833,
      "learning_rate": 0.0001,
      "loss": 3.9904,
      "loss/crossentropy": 2.1350772380828857,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21515338122844696,
      "step": 28702
    },
    {
      "epoch": 0.57408,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0086822509765625,
      "learning_rate": 0.0001,
      "loss": 3.9282,
      "loss/crossentropy": 1.9809751510620117,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18557599931955338,
      "step": 28704
    },
    {
      "epoch": 0.57412,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.006658681233723958,
      "learning_rate": 0.0001,
      "loss": 3.9628,
      "loss/crossentropy": 1.7658061981201172,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16570701450109482,
      "step": 28706
    },
    {
      "epoch": 0.57416,
      "grad_norm": 1.7421875,
      "grad_norm_var": 0.010188547770182292,
      "learning_rate": 0.0001,
      "loss": 3.8666,
      "loss/crossentropy": 1.787250578403473,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18032344430685043,
      "step": 28708
    },
    {
      "epoch": 0.5742,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.011082967122395834,
      "learning_rate": 0.0001,
      "loss": 4.1878,
      "loss/crossentropy": 2.361757516860962,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2135041281580925,
      "step": 28710
    },
    {
      "epoch": 0.57424,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.010921223958333334,
      "learning_rate": 0.0001,
      "loss": 4.1266,
      "loss/crossentropy": 2.0687750577926636,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20218269526958466,
      "step": 28712
    },
    {
      "epoch": 0.57428,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.010587310791015625,
      "learning_rate": 0.0001,
      "loss": 3.8437,
      "loss/crossentropy": 1.816366970539093,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1572190448641777,
      "step": 28714
    },
    {
      "epoch": 0.57432,
      "grad_norm": 1.7265625,
      "grad_norm_var": 0.010138956705729167,
      "learning_rate": 0.0001,
      "loss": 3.8273,
      "loss/crossentropy": 2.2949939966201782,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18894164264202118,
      "step": 28716
    },
    {
      "epoch": 0.57436,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.008680979410807291,
      "learning_rate": 0.0001,
      "loss": 3.9153,
      "loss/crossentropy": 2.249353289604187,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21117348968982697,
      "step": 28718
    },
    {
      "epoch": 0.5744,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.008902740478515626,
      "learning_rate": 0.0001,
      "loss": 4.2787,
      "loss/crossentropy": 2.3134714365005493,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21170897036790848,
      "step": 28720
    },
    {
      "epoch": 0.57444,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.009748331705729167,
      "learning_rate": 0.0001,
      "loss": 3.8791,
      "loss/crossentropy": 2.241877317428589,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2200472578406334,
      "step": 28722
    },
    {
      "epoch": 0.57448,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.007844034830729167,
      "learning_rate": 0.0001,
      "loss": 4.2115,
      "loss/crossentropy": 2.39408540725708,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2025068998336792,
      "step": 28724
    },
    {
      "epoch": 0.57452,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.008063761393229167,
      "learning_rate": 0.0001,
      "loss": 3.9963,
      "loss/crossentropy": 1.9735210537910461,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17944949120283127,
      "step": 28726
    },
    {
      "epoch": 0.57456,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.009040323893229167,
      "learning_rate": 0.0001,
      "loss": 3.9401,
      "loss/crossentropy": 2.028856635093689,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19235961139202118,
      "step": 28728
    },
    {
      "epoch": 0.5746,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.006992340087890625,
      "learning_rate": 0.0001,
      "loss": 4.3698,
      "loss/crossentropy": 2.18750536441803,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18273723125457764,
      "step": 28730
    },
    {
      "epoch": 0.57464,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.004378255208333333,
      "learning_rate": 0.0001,
      "loss": 4.0532,
      "loss/crossentropy": 2.127722382545471,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2066717892885208,
      "step": 28732
    },
    {
      "epoch": 0.57468,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.004255930582682292,
      "learning_rate": 0.0001,
      "loss": 3.8228,
      "loss/crossentropy": 2.1561193466186523,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19151438027620316,
      "step": 28734
    },
    {
      "epoch": 0.57472,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.004463704427083334,
      "learning_rate": 0.0001,
      "loss": 3.9893,
      "loss/crossentropy": 2.3356776237487793,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20846813172101974,
      "step": 28736
    },
    {
      "epoch": 0.57476,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.005182902018229167,
      "learning_rate": 0.0001,
      "loss": 4.14,
      "loss/crossentropy": 2.024761378765106,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19134287536144257,
      "step": 28738
    },
    {
      "epoch": 0.5748,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.006060536702473958,
      "learning_rate": 0.0001,
      "loss": 3.8726,
      "loss/crossentropy": 2.242386519908905,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20453084260225296,
      "step": 28740
    },
    {
      "epoch": 0.57484,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.005472819010416667,
      "learning_rate": 0.0001,
      "loss": 3.9335,
      "loss/crossentropy": 1.7191591262817383,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17990755289793015,
      "step": 28742
    },
    {
      "epoch": 0.57488,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.004571278889973958,
      "learning_rate": 0.0001,
      "loss": 3.7335,
      "loss/crossentropy": 2.1018252968788147,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20789441466331482,
      "step": 28744
    },
    {
      "epoch": 0.57492,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0042803446451822914,
      "learning_rate": 0.0001,
      "loss": 3.8906,
      "loss/crossentropy": 1.9056601524353027,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18688670545816422,
      "step": 28746
    },
    {
      "epoch": 0.57496,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.00360107421875,
      "learning_rate": 0.0001,
      "loss": 3.8521,
      "loss/crossentropy": 2.0531609058380127,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1931629702448845,
      "step": 28748
    },
    {
      "epoch": 0.575,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.003684234619140625,
      "learning_rate": 0.0001,
      "loss": 3.9218,
      "loss/crossentropy": 1.89992356300354,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1808946281671524,
      "step": 28750
    },
    {
      "epoch": 0.57504,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.003739166259765625,
      "learning_rate": 0.0001,
      "loss": 3.8245,
      "loss/crossentropy": 2.2009544372558594,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20267173647880554,
      "step": 28752
    },
    {
      "epoch": 0.57508,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0043108622233072914,
      "learning_rate": 0.0001,
      "loss": 4.2622,
      "loss/crossentropy": 2.009132504463196,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20399124920368195,
      "step": 28754
    },
    {
      "epoch": 0.57512,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.004365793863932292,
      "learning_rate": 0.0001,
      "loss": 3.7545,
      "loss/crossentropy": 1.9875890612602234,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20523500442504883,
      "step": 28756
    },
    {
      "epoch": 0.57516,
      "grad_norm": 5.40625,
      "grad_norm_var": 0.7703570048014323,
      "learning_rate": 0.0001,
      "loss": 4.0484,
      "loss/crossentropy": 1.8446847200393677,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16559410095214844,
      "step": 28758
    },
    {
      "epoch": 0.5752,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.7632118225097656,
      "learning_rate": 0.0001,
      "loss": 4.1006,
      "loss/crossentropy": 2.147071599960327,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19807270914316177,
      "step": 28760
    },
    {
      "epoch": 0.57524,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.760791015625,
      "learning_rate": 0.0001,
      "loss": 4.1881,
      "loss/crossentropy": 2.046617865562439,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18792100995779037,
      "step": 28762
    },
    {
      "epoch": 0.57528,
      "grad_norm": 2.0,
      "grad_norm_var": 0.7528297424316406,
      "learning_rate": 0.0001,
      "loss": 4.0231,
      "loss/crossentropy": 1.6951394081115723,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1635662019252777,
      "step": 28764
    },
    {
      "epoch": 0.57532,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.7440752665201823,
      "learning_rate": 0.0001,
      "loss": 3.8999,
      "loss/crossentropy": 1.7923210263252258,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16349300742149353,
      "step": 28766
    },
    {
      "epoch": 0.57536,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.7344011942545573,
      "learning_rate": 0.0001,
      "loss": 4.2475,
      "loss/crossentropy": 2.0918190479278564,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20321150124073029,
      "step": 28768
    },
    {
      "epoch": 0.5754,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.7387451171875,
      "learning_rate": 0.0001,
      "loss": 4.1483,
      "loss/crossentropy": 1.691568374633789,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18507587909698486,
      "step": 28770
    },
    {
      "epoch": 0.57544,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.7320757548014323,
      "learning_rate": 0.0001,
      "loss": 4.0857,
      "loss/crossentropy": 2.019860029220581,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19651245325803757,
      "step": 28772
    },
    {
      "epoch": 0.57548,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.01715672810872396,
      "learning_rate": 0.0001,
      "loss": 4.0712,
      "loss/crossentropy": 1.9527531862258911,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19377586245536804,
      "step": 28774
    },
    {
      "epoch": 0.57552,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.019486236572265624,
      "learning_rate": 0.0001,
      "loss": 4.0374,
      "loss/crossentropy": 2.105517864227295,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1907084435224533,
      "step": 28776
    },
    {
      "epoch": 0.57556,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.014277903238932292,
      "learning_rate": 0.0001,
      "loss": 3.88,
      "loss/crossentropy": 1.804184377193451,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17274542152881622,
      "step": 28778
    },
    {
      "epoch": 0.5756,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.014102935791015625,
      "learning_rate": 0.0001,
      "loss": 4.0615,
      "loss/crossentropy": 2.3283534049987793,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21980226039886475,
      "step": 28780
    },
    {
      "epoch": 0.57564,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.010727691650390624,
      "learning_rate": 0.0001,
      "loss": 3.8824,
      "loss/crossentropy": 2.054227828979492,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1968243345618248,
      "step": 28782
    },
    {
      "epoch": 0.57568,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.007614898681640625,
      "learning_rate": 0.0001,
      "loss": 3.9686,
      "loss/crossentropy": 2.2226120233535767,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20916803926229477,
      "step": 28784
    },
    {
      "epoch": 0.57572,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.007905832926432292,
      "learning_rate": 0.0001,
      "loss": 4.007,
      "loss/crossentropy": 1.9371869564056396,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21197544038295746,
      "step": 28786
    },
    {
      "epoch": 0.57576,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.017122395833333335,
      "learning_rate": 0.0001,
      "loss": 3.6572,
      "loss/crossentropy": 1.6595491766929626,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1565944254398346,
      "step": 28788
    },
    {
      "epoch": 0.5758,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.015276845296223958,
      "learning_rate": 0.0001,
      "loss": 3.9956,
      "loss/crossentropy": 1.7383070588111877,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1768334060907364,
      "step": 28790
    },
    {
      "epoch": 0.57584,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.015508778889973958,
      "learning_rate": 0.0001,
      "loss": 4.0105,
      "loss/crossentropy": 1.9087260365486145,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1958516761660576,
      "step": 28792
    },
    {
      "epoch": 0.57588,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.014460245768229166,
      "learning_rate": 0.0001,
      "loss": 3.9552,
      "loss/crossentropy": 2.1638959646224976,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20037583261728287,
      "step": 28794
    },
    {
      "epoch": 0.57592,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.015132649739583334,
      "learning_rate": 0.0001,
      "loss": 4.1782,
      "loss/crossentropy": 2.067569851875305,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20235508680343628,
      "step": 28796
    },
    {
      "epoch": 0.57596,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.014891560872395833,
      "learning_rate": 0.0001,
      "loss": 4.0691,
      "loss/crossentropy": 2.377517819404602,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21966589242219925,
      "step": 28798
    },
    {
      "epoch": 0.576,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.016556549072265624,
      "learning_rate": 0.0001,
      "loss": 3.8553,
      "loss/crossentropy": 1.8835238814353943,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17692726850509644,
      "step": 28800
    },
    {
      "epoch": 0.57604,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.017093912760416666,
      "learning_rate": 0.0001,
      "loss": 4.0865,
      "loss/crossentropy": 2.177576720714569,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1855052188038826,
      "step": 28802
    },
    {
      "epoch": 0.57608,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.007881673177083333,
      "learning_rate": 0.0001,
      "loss": 4.09,
      "loss/crossentropy": 2.2061915397644043,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2123064547777176,
      "step": 28804
    },
    {
      "epoch": 0.57612,
      "grad_norm": 1.7421875,
      "grad_norm_var": 0.010692342122395834,
      "learning_rate": 0.0001,
      "loss": 3.9471,
      "loss/crossentropy": 1.9637818932533264,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20487510412931442,
      "step": 28806
    },
    {
      "epoch": 0.57616,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.014399973551432292,
      "learning_rate": 0.0001,
      "loss": 4.0394,
      "loss/crossentropy": 2.1481754183769226,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1974523589015007,
      "step": 28808
    },
    {
      "epoch": 0.5762,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0139312744140625,
      "learning_rate": 0.0001,
      "loss": 3.9125,
      "loss/crossentropy": 1.907672941684723,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18901971727609634,
      "step": 28810
    },
    {
      "epoch": 0.57624,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0115875244140625,
      "learning_rate": 0.0001,
      "loss": 4.0425,
      "loss/crossentropy": 2.2241225838661194,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1882694587111473,
      "step": 28812
    },
    {
      "epoch": 0.57628,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.01248779296875,
      "learning_rate": 0.0001,
      "loss": 4.1874,
      "loss/crossentropy": 2.150957465171814,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2233213186264038,
      "step": 28814
    },
    {
      "epoch": 0.57632,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.011494954427083334,
      "learning_rate": 0.0001,
      "loss": 4.0973,
      "loss/crossentropy": 1.9427680373191833,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19366063922643661,
      "step": 28816
    },
    {
      "epoch": 0.57636,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.012027740478515625,
      "learning_rate": 0.0001,
      "loss": 3.7749,
      "loss/crossentropy": 1.7898414731025696,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18277223408222198,
      "step": 28818
    },
    {
      "epoch": 0.5764,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.03846435546875,
      "learning_rate": 0.0001,
      "loss": 4.0501,
      "loss/crossentropy": 2.0117841362953186,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17678485065698624,
      "step": 28820
    },
    {
      "epoch": 0.57644,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.03804423014322917,
      "learning_rate": 0.0001,
      "loss": 3.749,
      "loss/crossentropy": 2.0270848274230957,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17247303575277328,
      "step": 28822
    },
    {
      "epoch": 0.57648,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0336822509765625,
      "learning_rate": 0.0001,
      "loss": 3.9248,
      "loss/crossentropy": 1.917984962463379,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1722751259803772,
      "step": 28824
    },
    {
      "epoch": 0.57652,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.03392512003580729,
      "learning_rate": 0.0001,
      "loss": 4.1579,
      "loss/crossentropy": 1.8189431428909302,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18306910246610641,
      "step": 28826
    },
    {
      "epoch": 0.57656,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.03524144490559896,
      "learning_rate": 0.0001,
      "loss": 4.0638,
      "loss/crossentropy": 1.8207548260688782,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1782800853252411,
      "step": 28828
    },
    {
      "epoch": 0.5766,
      "grad_norm": 1.703125,
      "grad_norm_var": 0.038266754150390624,
      "learning_rate": 0.0001,
      "loss": 4.0089,
      "loss/crossentropy": 2.1244252920150757,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18752482533454895,
      "step": 28830
    },
    {
      "epoch": 0.57664,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.038426717122395836,
      "learning_rate": 0.0001,
      "loss": 4.1127,
      "loss/crossentropy": 2.435193181037903,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23042550683021545,
      "step": 28832
    },
    {
      "epoch": 0.57668,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.03916803995768229,
      "learning_rate": 0.0001,
      "loss": 3.6812,
      "loss/crossentropy": 2.1272794008255005,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19795158505439758,
      "step": 28834
    },
    {
      "epoch": 0.57672,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.009814453125,
      "learning_rate": 0.0001,
      "loss": 4.08,
      "loss/crossentropy": 1.7166744470596313,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17582352459430695,
      "step": 28836
    },
    {
      "epoch": 0.57676,
      "grad_norm": 1.7421875,
      "grad_norm_var": 0.011248524983723958,
      "learning_rate": 0.0001,
      "loss": 3.85,
      "loss/crossentropy": 2.066642463207245,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19613336026668549,
      "step": 28838
    },
    {
      "epoch": 0.5768,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.0114166259765625,
      "learning_rate": 0.0001,
      "loss": 3.8214,
      "loss/crossentropy": 1.9082713723182678,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.173607736825943,
      "step": 28840
    },
    {
      "epoch": 0.57684,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.0119781494140625,
      "learning_rate": 0.0001,
      "loss": 3.8438,
      "loss/crossentropy": 2.0809885263442993,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1808517575263977,
      "step": 28842
    },
    {
      "epoch": 0.57688,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.011246490478515624,
      "learning_rate": 0.0001,
      "loss": 3.7981,
      "loss/crossentropy": 1.8057249784469604,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17345206439495087,
      "step": 28844
    },
    {
      "epoch": 0.57692,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.012627919514973959,
      "learning_rate": 0.0001,
      "loss": 4.2121,
      "loss/crossentropy": 2.3364038467407227,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2242283821105957,
      "step": 28846
    },
    {
      "epoch": 0.57696,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.014387003580729167,
      "learning_rate": 0.0001,
      "loss": 4.2341,
      "loss/crossentropy": 2.256643772125244,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20197833329439163,
      "step": 28848
    },
    {
      "epoch": 0.577,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.013270823160807292,
      "learning_rate": 0.0001,
      "loss": 4.0897,
      "loss/crossentropy": 2.2531803846359253,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19735650718212128,
      "step": 28850
    },
    {
      "epoch": 0.57704,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.011897786458333334,
      "learning_rate": 0.0001,
      "loss": 4.1416,
      "loss/crossentropy": 2.1737340688705444,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2048204094171524,
      "step": 28852
    },
    {
      "epoch": 0.57708,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0110595703125,
      "learning_rate": 0.0001,
      "loss": 4.2294,
      "loss/crossentropy": 2.2098671197891235,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21505644917488098,
      "step": 28854
    },
    {
      "epoch": 0.57712,
      "grad_norm": 1.875,
      "grad_norm_var": 0.011149088541666666,
      "learning_rate": 0.0001,
      "loss": 3.7808,
      "loss/crossentropy": 2.0007169246673584,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17798644304275513,
      "step": 28856
    },
    {
      "epoch": 0.57716,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.0106597900390625,
      "learning_rate": 0.0001,
      "loss": 3.6825,
      "loss/crossentropy": 1.9002285599708557,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18565426766872406,
      "step": 28858
    },
    {
      "epoch": 0.5772,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.008833567301432291,
      "learning_rate": 0.0001,
      "loss": 3.8333,
      "loss/crossentropy": 1.9257296323776245,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18187644332647324,
      "step": 28860
    },
    {
      "epoch": 0.57724,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.007868448893229166,
      "learning_rate": 0.0001,
      "loss": 4.204,
      "loss/crossentropy": 2.0777525305747986,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20168712735176086,
      "step": 28862
    },
    {
      "epoch": 0.57728,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.0074127197265625,
      "learning_rate": 0.0001,
      "loss": 3.9328,
      "loss/crossentropy": 2.1630115509033203,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19102369993925095,
      "step": 28864
    },
    {
      "epoch": 0.57732,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.0077301025390625,
      "learning_rate": 0.0001,
      "loss": 3.7329,
      "loss/crossentropy": 1.8166091442108154,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1680990755558014,
      "step": 28866
    },
    {
      "epoch": 0.57736,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.009419759114583334,
      "learning_rate": 0.0001,
      "loss": 4.1253,
      "loss/crossentropy": 2.073417544364929,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20478951185941696,
      "step": 28868
    },
    {
      "epoch": 0.5774,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.007184855143229167,
      "learning_rate": 0.0001,
      "loss": 3.9782,
      "loss/crossentropy": 1.9039225578308105,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17434976249933243,
      "step": 28870
    },
    {
      "epoch": 0.57744,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.0077301025390625,
      "learning_rate": 0.0001,
      "loss": 4.0646,
      "loss/crossentropy": 2.3139874935150146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20631124079227448,
      "step": 28872
    },
    {
      "epoch": 0.57748,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.007279459635416667,
      "learning_rate": 0.0001,
      "loss": 3.8964,
      "loss/crossentropy": 2.1889963150024414,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22081536054611206,
      "step": 28874
    },
    {
      "epoch": 0.57752,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.007283274332682292,
      "learning_rate": 0.0001,
      "loss": 3.9964,
      "loss/crossentropy": 2.1492413878440857,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1974084973335266,
      "step": 28876
    },
    {
      "epoch": 0.57756,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.007201131184895833,
      "learning_rate": 0.0001,
      "loss": 3.751,
      "loss/crossentropy": 1.8974568843841553,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1662147268652916,
      "step": 28878
    },
    {
      "epoch": 0.5776,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.008093007405598958,
      "learning_rate": 0.0001,
      "loss": 3.8792,
      "loss/crossentropy": 2.0488428473472595,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1929314285516739,
      "step": 28880
    },
    {
      "epoch": 0.57764,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.007664998372395833,
      "learning_rate": 0.0001,
      "loss": 3.9943,
      "loss/crossentropy": 1.9983501434326172,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19050447642803192,
      "step": 28882
    },
    {
      "epoch": 0.57768,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0066640218098958336,
      "learning_rate": 0.0001,
      "loss": 3.8174,
      "loss/crossentropy": 1.8651137948036194,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18415149301290512,
      "step": 28884
    },
    {
      "epoch": 0.57772,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.007055409749348958,
      "learning_rate": 0.0001,
      "loss": 3.8707,
      "loss/crossentropy": 2.0147623419761658,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1963946893811226,
      "step": 28886
    },
    {
      "epoch": 0.57776,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.23955052693684895,
      "learning_rate": 0.0001,
      "loss": 4.1033,
      "loss/crossentropy": 2.057682752609253,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22113247215747833,
      "step": 28888
    },
    {
      "epoch": 0.5778,
      "grad_norm": 1.875,
      "grad_norm_var": 0.239501953125,
      "learning_rate": 0.0001,
      "loss": 4.2273,
      "loss/crossentropy": 2.0742294788360596,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2198052555322647,
      "step": 28890
    },
    {
      "epoch": 0.57784,
      "grad_norm": 2.0,
      "grad_norm_var": 0.23716201782226562,
      "learning_rate": 0.0001,
      "loss": 4.0652,
      "loss/crossentropy": 2.1344255208969116,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19126100838184357,
      "step": 28892
    },
    {
      "epoch": 0.57788,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.23585611979166668,
      "learning_rate": 0.0001,
      "loss": 3.8575,
      "loss/crossentropy": 1.8000599145889282,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17296022176742554,
      "step": 28894
    },
    {
      "epoch": 0.57792,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.2333941141764323,
      "learning_rate": 0.0001,
      "loss": 4.0821,
      "loss/crossentropy": 2.2380261421203613,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18993133306503296,
      "step": 28896
    },
    {
      "epoch": 0.57796,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.23112360636393228,
      "learning_rate": 0.0001,
      "loss": 4.1168,
      "loss/crossentropy": 2.066701650619507,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1944030448794365,
      "step": 28898
    },
    {
      "epoch": 0.578,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.22492268880208333,
      "learning_rate": 0.0001,
      "loss": 4.2338,
      "loss/crossentropy": 2.2941386699676514,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2139548882842064,
      "step": 28900
    },
    {
      "epoch": 0.57804,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.22176920572916667,
      "learning_rate": 0.0001,
      "loss": 3.8767,
      "loss/crossentropy": 2.206367015838623,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1887761801481247,
      "step": 28902
    },
    {
      "epoch": 0.57808,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.004473622639973958,
      "learning_rate": 0.0001,
      "loss": 4.1728,
      "loss/crossentropy": 2.176335871219635,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2016155868768692,
      "step": 28904
    },
    {
      "epoch": 0.57812,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.015022532145182291,
      "learning_rate": 0.0001,
      "loss": 4.2275,
      "loss/crossentropy": 2.2069336771965027,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20764954388141632,
      "step": 28906
    },
    {
      "epoch": 0.57816,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.0157379150390625,
      "learning_rate": 0.0001,
      "loss": 3.9015,
      "loss/crossentropy": 2.2566596269607544,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19805442541837692,
      "step": 28908
    },
    {
      "epoch": 0.5782,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.0152252197265625,
      "learning_rate": 0.0001,
      "loss": 3.8767,
      "loss/crossentropy": 1.8642338514328003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19232972711324692,
      "step": 28910
    },
    {
      "epoch": 0.57824,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.015119425455729167,
      "learning_rate": 0.0001,
      "loss": 4.1759,
      "loss/crossentropy": 1.9054778218269348,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1713176593184471,
      "step": 28912
    },
    {
      "epoch": 0.57828,
      "grad_norm": 1.734375,
      "grad_norm_var": 0.0199859619140625,
      "learning_rate": 0.0001,
      "loss": 3.6261,
      "loss/crossentropy": 1.7423803210258484,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16640456765890121,
      "step": 28914
    },
    {
      "epoch": 0.57832,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.02054443359375,
      "learning_rate": 0.0001,
      "loss": 3.9985,
      "loss/crossentropy": 2.229828178882599,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20995531231164932,
      "step": 28916
    },
    {
      "epoch": 0.57836,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.0248443603515625,
      "learning_rate": 0.0001,
      "loss": 3.9723,
      "loss/crossentropy": 1.483105719089508,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15799641609191895,
      "step": 28918
    },
    {
      "epoch": 0.5784,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.024395497639973958,
      "learning_rate": 0.0001,
      "loss": 4.2127,
      "loss/crossentropy": 2.2207179069519043,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1969587355852127,
      "step": 28920
    },
    {
      "epoch": 0.57844,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.012259674072265626,
      "learning_rate": 0.0001,
      "loss": 3.9546,
      "loss/crossentropy": 2.3045194149017334,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19103257358074188,
      "step": 28922
    },
    {
      "epoch": 0.57848,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.014289347330729167,
      "learning_rate": 0.0001,
      "loss": 4.1317,
      "loss/crossentropy": 2.0717945098876953,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20137913525104523,
      "step": 28924
    },
    {
      "epoch": 0.57852,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.014926910400390625,
      "learning_rate": 0.0001,
      "loss": 4.0922,
      "loss/crossentropy": 2.2904374599456787,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2105882465839386,
      "step": 28926
    },
    {
      "epoch": 0.57856,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.015818023681640626,
      "learning_rate": 0.0001,
      "loss": 3.9496,
      "loss/crossentropy": 2.3007075786590576,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22365766763687134,
      "step": 28928
    },
    {
      "epoch": 0.5786,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.012276204427083333,
      "learning_rate": 0.0001,
      "loss": 4.082,
      "loss/crossentropy": 2.0107430815696716,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18182440847158432,
      "step": 28930
    },
    {
      "epoch": 0.57864,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.013106028238932291,
      "learning_rate": 0.0001,
      "loss": 3.9859,
      "loss/crossentropy": 2.1066025495529175,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19574615359306335,
      "step": 28932
    },
    {
      "epoch": 0.57868,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.007995351155598959,
      "learning_rate": 0.0001,
      "loss": 3.9903,
      "loss/crossentropy": 2.1549283266067505,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20238593965768814,
      "step": 28934
    },
    {
      "epoch": 0.57872,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.010734049479166667,
      "learning_rate": 0.0001,
      "loss": 3.683,
      "loss/crossentropy": 2.0073814392089844,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.185798779129982,
      "step": 28936
    },
    {
      "epoch": 0.57876,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0111724853515625,
      "learning_rate": 0.0001,
      "loss": 3.8882,
      "loss/crossentropy": 1.7613274455070496,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1848607361316681,
      "step": 28938
    },
    {
      "epoch": 0.5788,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.08815078735351563,
      "learning_rate": 0.0001,
      "loss": 4.0142,
      "loss/crossentropy": 2.322823405265808,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20245341211557388,
      "step": 28940
    },
    {
      "epoch": 0.57884,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0876129150390625,
      "learning_rate": 0.0001,
      "loss": 3.8313,
      "loss/crossentropy": 1.7385436296463013,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17173148691654205,
      "step": 28942
    },
    {
      "epoch": 0.57888,
      "grad_norm": 1.7421875,
      "grad_norm_var": 0.08886286417643229,
      "learning_rate": 0.0001,
      "loss": 3.9017,
      "loss/crossentropy": 1.9626798033714294,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17217369377613068,
      "step": 28944
    },
    {
      "epoch": 0.57892,
      "grad_norm": 1.6640625,
      "grad_norm_var": 0.09463297526041667,
      "learning_rate": 0.0001,
      "loss": 3.7427,
      "loss/crossentropy": 2.4835134744644165,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2073291689157486,
      "step": 28946
    },
    {
      "epoch": 0.57896,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.09482192993164062,
      "learning_rate": 0.0001,
      "loss": 3.9099,
      "loss/crossentropy": 2.0418132543563843,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23464705049991608,
      "step": 28948
    },
    {
      "epoch": 0.579,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.09468968709309895,
      "learning_rate": 0.0001,
      "loss": 4.2291,
      "loss/crossentropy": 2.2382947206497192,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21655261516571045,
      "step": 28950
    },
    {
      "epoch": 0.57904,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.09024632771809896,
      "learning_rate": 0.0001,
      "loss": 4.0013,
      "loss/crossentropy": 2.0737303495407104,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19475168734788895,
      "step": 28952
    },
    {
      "epoch": 0.57908,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.09391988118489583,
      "learning_rate": 0.0001,
      "loss": 3.9371,
      "loss/crossentropy": 2.088689923286438,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19861633330583572,
      "step": 28954
    },
    {
      "epoch": 0.57912,
      "grad_norm": 2.0,
      "grad_norm_var": 0.011472320556640625,
      "learning_rate": 0.0001,
      "loss": 3.6761,
      "loss/crossentropy": 1.8260209560394287,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16850142180919647,
      "step": 28956
    },
    {
      "epoch": 0.57916,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.010263824462890625,
      "learning_rate": 0.0001,
      "loss": 3.8753,
      "loss/crossentropy": 1.834153652191162,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16283375024795532,
      "step": 28958
    },
    {
      "epoch": 0.5792,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.009569295247395833,
      "learning_rate": 0.0001,
      "loss": 3.7873,
      "loss/crossentropy": 1.9701993465423584,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18593734502792358,
      "step": 28960
    },
    {
      "epoch": 0.57924,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.006965128580729166,
      "learning_rate": 0.0001,
      "loss": 3.7081,
      "loss/crossentropy": 2.026169538497925,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18718570470809937,
      "step": 28962
    },
    {
      "epoch": 0.57928,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.006400553385416666,
      "learning_rate": 0.0001,
      "loss": 4.1803,
      "loss/crossentropy": 1.7088202238082886,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1926863193511963,
      "step": 28964
    },
    {
      "epoch": 0.57932,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.006498209635416667,
      "learning_rate": 0.0001,
      "loss": 4.1078,
      "loss/crossentropy": 2.008699059486389,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17876552045345306,
      "step": 28966
    },
    {
      "epoch": 0.57936,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.007279459635416667,
      "learning_rate": 0.0001,
      "loss": 3.8845,
      "loss/crossentropy": 1.7667925953865051,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16738490760326385,
      "step": 28968
    },
    {
      "epoch": 0.5794,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.010239410400390624,
      "learning_rate": 0.0001,
      "loss": 4.0384,
      "loss/crossentropy": 1.7718802094459534,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15908128768205643,
      "step": 28970
    },
    {
      "epoch": 0.57944,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.007437896728515625,
      "learning_rate": 0.0001,
      "loss": 4.0281,
      "loss/crossentropy": 2.086379885673523,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19513976573944092,
      "step": 28972
    },
    {
      "epoch": 0.57948,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.0073486328125,
      "learning_rate": 0.0001,
      "loss": 4.0432,
      "loss/crossentropy": 2.157579243183136,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2022448629140854,
      "step": 28974
    },
    {
      "epoch": 0.57952,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.007689412434895833,
      "learning_rate": 0.0001,
      "loss": 4.0684,
      "loss/crossentropy": 1.9351604580879211,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1752564013004303,
      "step": 28976
    },
    {
      "epoch": 0.57956,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.0084625244140625,
      "learning_rate": 0.0001,
      "loss": 3.9584,
      "loss/crossentropy": 1.8642477989196777,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1806860715150833,
      "step": 28978
    },
    {
      "epoch": 0.5796,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.008420562744140625,
      "learning_rate": 0.0001,
      "loss": 3.9329,
      "loss/crossentropy": 2.0846880078315735,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18332630395889282,
      "step": 28980
    },
    {
      "epoch": 0.57964,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.008595530192057292,
      "learning_rate": 0.0001,
      "loss": 3.993,
      "loss/crossentropy": 2.019991397857666,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1845393031835556,
      "step": 28982
    },
    {
      "epoch": 0.57968,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.008250935872395834,
      "learning_rate": 0.0001,
      "loss": 4.0237,
      "loss/crossentropy": 2.148271918296814,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20690763741731644,
      "step": 28984
    },
    {
      "epoch": 0.57972,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.0036740620930989583,
      "learning_rate": 0.0001,
      "loss": 4.1667,
      "loss/crossentropy": 2.243967056274414,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21643538773059845,
      "step": 28986
    },
    {
      "epoch": 0.57976,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.004388173421223958,
      "learning_rate": 0.0001,
      "loss": 4.1014,
      "loss/crossentropy": 2.094264507293701,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19375698268413544,
      "step": 28988
    },
    {
      "epoch": 0.5798,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.005425771077473958,
      "learning_rate": 0.0001,
      "loss": 3.6987,
      "loss/crossentropy": 2.0419931411743164,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1727323830127716,
      "step": 28990
    },
    {
      "epoch": 0.57984,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0053708394368489586,
      "learning_rate": 0.0001,
      "loss": 3.887,
      "loss/crossentropy": 2.1932610273361206,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20193511247634888,
      "step": 28992
    },
    {
      "epoch": 0.57988,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.006646474202473958,
      "learning_rate": 0.0001,
      "loss": 4.144,
      "loss/crossentropy": 1.9849725365638733,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20237861573696136,
      "step": 28994
    },
    {
      "epoch": 0.57992,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.006420644124348959,
      "learning_rate": 0.0001,
      "loss": 3.8113,
      "loss/crossentropy": 1.801219642162323,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1687634363770485,
      "step": 28996
    },
    {
      "epoch": 0.57996,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.005686187744140625,
      "learning_rate": 0.0001,
      "loss": 4.0288,
      "loss/crossentropy": 2.018694579601288,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17765481770038605,
      "step": 28998
    },
    {
      "epoch": 0.58,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.005598958333333333,
      "learning_rate": 0.0001,
      "loss": 4.1113,
      "loss/crossentropy": 1.8309300541877747,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19343653321266174,
      "step": 29000
    },
    {
      "epoch": 0.58004,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.005100250244140625,
      "learning_rate": 0.0001,
      "loss": 4.2134,
      "loss/crossentropy": 2.264810800552368,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20135793089866638,
      "step": 29002
    },
    {
      "epoch": 0.58008,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.004903157552083333,
      "learning_rate": 0.0001,
      "loss": 3.9869,
      "loss/crossentropy": 2.0125122666358948,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1977495700120926,
      "step": 29004
    },
    {
      "epoch": 0.58012,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.0034739176432291665,
      "learning_rate": 0.0001,
      "loss": 3.7476,
      "loss/crossentropy": 1.7581869959831238,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17235949635505676,
      "step": 29006
    },
    {
      "epoch": 0.58016,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0043609619140625,
      "learning_rate": 0.0001,
      "loss": 3.7818,
      "loss/crossentropy": 2.0644874572753906,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20059636235237122,
      "step": 29008
    },
    {
      "epoch": 0.5802,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.0029296875,
      "learning_rate": 0.0001,
      "loss": 4.0232,
      "loss/crossentropy": 2.2566142082214355,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18922609090805054,
      "step": 29010
    },
    {
      "epoch": 0.58024,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0065996805826822914,
      "learning_rate": 0.0001,
      "loss": 4.2414,
      "loss/crossentropy": 2.3137046098709106,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21420790255069733,
      "step": 29012
    },
    {
      "epoch": 0.58028,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.0064280192057291664,
      "learning_rate": 0.0001,
      "loss": 3.9192,
      "loss/crossentropy": 2.1429378986358643,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20075450092554092,
      "step": 29014
    },
    {
      "epoch": 0.58032,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.007567342122395833,
      "learning_rate": 0.0001,
      "loss": 4.0905,
      "loss/crossentropy": 2.5844578742980957,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21986322849988937,
      "step": 29016
    },
    {
      "epoch": 0.58036,
      "grad_norm": 1.875,
      "grad_norm_var": 0.0093902587890625,
      "learning_rate": 0.0001,
      "loss": 3.7977,
      "loss/crossentropy": 1.847869634628296,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16430244594812393,
      "step": 29018
    },
    {
      "epoch": 0.5804,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.010544586181640624,
      "learning_rate": 0.0001,
      "loss": 3.8369,
      "loss/crossentropy": 1.895844042301178,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18856079131364822,
      "step": 29020
    },
    {
      "epoch": 0.58044,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.009113566080729166,
      "learning_rate": 0.0001,
      "loss": 3.9936,
      "loss/crossentropy": 2.1094332933425903,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2025027722120285,
      "step": 29022
    },
    {
      "epoch": 0.58048,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.012172190348307292,
      "learning_rate": 0.0001,
      "loss": 4.1463,
      "loss/crossentropy": 1.820436716079712,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1872788816690445,
      "step": 29024
    },
    {
      "epoch": 0.58052,
      "grad_norm": 2.0,
      "grad_norm_var": 0.011641184488932291,
      "learning_rate": 0.0001,
      "loss": 4.1155,
      "loss/crossentropy": 2.371734142303467,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20477186143398285,
      "step": 29026
    },
    {
      "epoch": 0.58056,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.010273996988932292,
      "learning_rate": 0.0001,
      "loss": 4.3274,
      "loss/crossentropy": 2.4618901014328003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20902744680643082,
      "step": 29028
    },
    {
      "epoch": 0.5806,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.010044097900390625,
      "learning_rate": 0.0001,
      "loss": 4.0957,
      "loss/crossentropy": 1.9462957382202148,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18320611119270325,
      "step": 29030
    },
    {
      "epoch": 0.58064,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.010589345296223959,
      "learning_rate": 0.0001,
      "loss": 3.8647,
      "loss/crossentropy": 2.07541561126709,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19628040492534637,
      "step": 29032
    },
    {
      "epoch": 0.58068,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.009269205729166667,
      "learning_rate": 0.0001,
      "loss": 3.9063,
      "loss/crossentropy": 2.144878387451172,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21301092207431793,
      "step": 29034
    },
    {
      "epoch": 0.58072,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.009490712483723959,
      "learning_rate": 0.0001,
      "loss": 3.8692,
      "loss/crossentropy": 1.9113212823867798,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18449342995882034,
      "step": 29036
    },
    {
      "epoch": 0.58076,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0095611572265625,
      "learning_rate": 0.0001,
      "loss": 4.1034,
      "loss/crossentropy": 2.211828351020813,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19539960473775864,
      "step": 29038
    },
    {
      "epoch": 0.5808,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.005179595947265625,
      "learning_rate": 0.0001,
      "loss": 4.0377,
      "loss/crossentropy": 2.1476653814315796,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1979052647948265,
      "step": 29040
    },
    {
      "epoch": 0.58084,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.004239654541015625,
      "learning_rate": 0.0001,
      "loss": 3.8152,
      "loss/crossentropy": 1.5791842937469482,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17846909165382385,
      "step": 29042
    },
    {
      "epoch": 0.58088,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.002249908447265625,
      "learning_rate": 0.0001,
      "loss": 3.7192,
      "loss/crossentropy": 1.77503103017807,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1774892508983612,
      "step": 29044
    },
    {
      "epoch": 0.58092,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0028523763020833334,
      "learning_rate": 0.0001,
      "loss": 4.0933,
      "loss/crossentropy": 1.9995554089546204,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18692871183156967,
      "step": 29046
    },
    {
      "epoch": 0.58096,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.002750396728515625,
      "learning_rate": 0.0001,
      "loss": 3.9813,
      "loss/crossentropy": 2.057563364505768,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19816753268241882,
      "step": 29048
    },
    {
      "epoch": 0.581,
      "grad_norm": 1.7265625,
      "grad_norm_var": 0.010599517822265625,
      "learning_rate": 0.0001,
      "loss": 3.7361,
      "loss/crossentropy": 2.1810996532440186,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18617865443229675,
      "step": 29050
    },
    {
      "epoch": 0.58104,
      "grad_norm": 1.875,
      "grad_norm_var": 0.010038248697916667,
      "learning_rate": 0.0001,
      "loss": 4.0803,
      "loss/crossentropy": 2.212439775466919,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19946376979351044,
      "step": 29052
    },
    {
      "epoch": 0.58108,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.0109130859375,
      "learning_rate": 0.0001,
      "loss": 3.7017,
      "loss/crossentropy": 1.4078394174575806,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.14310228824615479,
      "step": 29054
    },
    {
      "epoch": 0.58112,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0114013671875,
      "learning_rate": 0.0001,
      "loss": 3.9539,
      "loss/crossentropy": 2.0167892575263977,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.206185445189476,
      "step": 29056
    },
    {
      "epoch": 0.58116,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.013312784830729167,
      "learning_rate": 0.0001,
      "loss": 4.0975,
      "loss/crossentropy": 2.0703752040863037,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20013286918401718,
      "step": 29058
    },
    {
      "epoch": 0.5812,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.015047200520833333,
      "learning_rate": 0.0001,
      "loss": 3.7723,
      "loss/crossentropy": 1.528024673461914,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16279471665620804,
      "step": 29060
    },
    {
      "epoch": 0.58124,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.016190338134765624,
      "learning_rate": 0.0001,
      "loss": 4.1933,
      "loss/crossentropy": 2.235237717628479,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22028259187936783,
      "step": 29062
    },
    {
      "epoch": 0.58128,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.017293294270833332,
      "learning_rate": 0.0001,
      "loss": 3.7427,
      "loss/crossentropy": 1.8778278231620789,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16768044978380203,
      "step": 29064
    },
    {
      "epoch": 0.58132,
      "grad_norm": 1.7265625,
      "grad_norm_var": 0.011102040608723959,
      "learning_rate": 0.0001,
      "loss": 3.5901,
      "loss/crossentropy": 1.8109752535820007,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16292864084243774,
      "step": 29066
    },
    {
      "epoch": 0.58136,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.023502604166666666,
      "learning_rate": 0.0001,
      "loss": 3.8298,
      "loss/crossentropy": 1.9164445996284485,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17200321704149246,
      "step": 29068
    },
    {
      "epoch": 0.5814,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.024103800455729168,
      "learning_rate": 0.0001,
      "loss": 4.1952,
      "loss/crossentropy": 2.1318963766098022,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20111309736967087,
      "step": 29070
    },
    {
      "epoch": 0.58144,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.023606109619140624,
      "learning_rate": 0.0001,
      "loss": 4.1548,
      "loss/crossentropy": 1.8939481377601624,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2058345079421997,
      "step": 29072
    },
    {
      "epoch": 0.58148,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0228424072265625,
      "learning_rate": 0.0001,
      "loss": 4.1685,
      "loss/crossentropy": 2.0977996587753296,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19700003415346146,
      "step": 29074
    },
    {
      "epoch": 0.58152,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.0199615478515625,
      "learning_rate": 0.0001,
      "loss": 4.0528,
      "loss/crossentropy": 2.163521647453308,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20974216610193253,
      "step": 29076
    },
    {
      "epoch": 0.58156,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.019358062744140626,
      "learning_rate": 0.0001,
      "loss": 4.0584,
      "loss/crossentropy": 2.2900352478027344,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20909854024648666,
      "step": 29078
    },
    {
      "epoch": 0.5816,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.018733723958333334,
      "learning_rate": 0.0001,
      "loss": 4.0277,
      "loss/crossentropy": 2.0664632320404053,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1772863119840622,
      "step": 29080
    },
    {
      "epoch": 0.58164,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.018436431884765625,
      "learning_rate": 0.0001,
      "loss": 3.7484,
      "loss/crossentropy": 1.9867297410964966,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18980251252651215,
      "step": 29082
    },
    {
      "epoch": 0.58168,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.00830078125,
      "learning_rate": 0.0001,
      "loss": 4.1918,
      "loss/crossentropy": 2.318376064300537,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22295164316892624,
      "step": 29084
    },
    {
      "epoch": 0.58172,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.006156412760416666,
      "learning_rate": 0.0001,
      "loss": 3.8932,
      "loss/crossentropy": 2.0880175828933716,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18823902308940887,
      "step": 29086
    },
    {
      "epoch": 0.58176,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.007039133707682292,
      "learning_rate": 0.0001,
      "loss": 4.1964,
      "loss/crossentropy": 2.5453847646713257,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24233754724264145,
      "step": 29088
    },
    {
      "epoch": 0.5818,
      "grad_norm": 1.734375,
      "grad_norm_var": 0.00999755859375,
      "learning_rate": 0.0001,
      "loss": 3.8693,
      "loss/crossentropy": 2.0354779958724976,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2055264338850975,
      "step": 29090
    },
    {
      "epoch": 0.58184,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.009938303629557292,
      "learning_rate": 0.0001,
      "loss": 4.1731,
      "loss/crossentropy": 2.3989592790603638,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18171776831150055,
      "step": 29092
    },
    {
      "epoch": 0.58188,
      "grad_norm": 2.125,
      "grad_norm_var": 0.013114166259765626,
      "learning_rate": 0.0001,
      "loss": 4.0719,
      "loss/crossentropy": 2.0136520862579346,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18895280361175537,
      "step": 29094
    },
    {
      "epoch": 0.58192,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.013071441650390625,
      "learning_rate": 0.0001,
      "loss": 3.9095,
      "loss/crossentropy": 2.057899534702301,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17919254302978516,
      "step": 29096
    },
    {
      "epoch": 0.58196,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.013002268473307292,
      "learning_rate": 0.0001,
      "loss": 3.8561,
      "loss/crossentropy": 1.9806170463562012,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2131671980023384,
      "step": 29098
    },
    {
      "epoch": 0.582,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.013734690348307292,
      "learning_rate": 0.0001,
      "loss": 3.747,
      "loss/crossentropy": 2.0952848196029663,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2057284563779831,
      "step": 29100
    },
    {
      "epoch": 0.58204,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.013346354166666666,
      "learning_rate": 0.0001,
      "loss": 4.2041,
      "loss/crossentropy": 2.2287687063217163,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18772760778665543,
      "step": 29102
    },
    {
      "epoch": 0.58208,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.012221018473307291,
      "learning_rate": 0.0001,
      "loss": 3.9164,
      "loss/crossentropy": 2.2227126359939575,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2002912387251854,
      "step": 29104
    },
    {
      "epoch": 0.58212,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.010042317708333333,
      "learning_rate": 0.0001,
      "loss": 3.724,
      "loss/crossentropy": 1.9384364485740662,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18695514649152756,
      "step": 29106
    },
    {
      "epoch": 0.58216,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.010481516520182291,
      "learning_rate": 0.0001,
      "loss": 3.8985,
      "loss/crossentropy": 1.7653113007545471,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19013341516256332,
      "step": 29108
    },
    {
      "epoch": 0.5822,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.008128865559895834,
      "learning_rate": 0.0001,
      "loss": 3.8348,
      "loss/crossentropy": 1.7308897972106934,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1701432168483734,
      "step": 29110
    },
    {
      "epoch": 0.58224,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.017435709635416668,
      "learning_rate": 0.0001,
      "loss": 4.3871,
      "loss/crossentropy": 2.286720633506775,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2024293765425682,
      "step": 29112
    },
    {
      "epoch": 0.58228,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.0161529541015625,
      "learning_rate": 0.0001,
      "loss": 3.744,
      "loss/crossentropy": 1.8618659377098083,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18028806149959564,
      "step": 29114
    },
    {
      "epoch": 0.58232,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0154449462890625,
      "learning_rate": 0.0001,
      "loss": 4.0077,
      "loss/crossentropy": 2.128199815750122,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2121022716164589,
      "step": 29116
    },
    {
      "epoch": 0.58236,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.0156890869140625,
      "learning_rate": 0.0001,
      "loss": 3.731,
      "loss/crossentropy": 2.0003921389579773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1782400906085968,
      "step": 29118
    },
    {
      "epoch": 0.5824,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.01651611328125,
      "learning_rate": 0.0001,
      "loss": 3.7362,
      "loss/crossentropy": 1.653384268283844,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1676875352859497,
      "step": 29120
    },
    {
      "epoch": 0.58244,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.016261545817057292,
      "learning_rate": 0.0001,
      "loss": 4.1866,
      "loss/crossentropy": 2.3602925539016724,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22520510107278824,
      "step": 29122
    },
    {
      "epoch": 0.58248,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.017814127604166667,
      "learning_rate": 0.0001,
      "loss": 4.0382,
      "loss/crossentropy": 1.839131772518158,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19381030648946762,
      "step": 29124
    },
    {
      "epoch": 0.58252,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.017878977457682292,
      "learning_rate": 0.0001,
      "loss": 4.178,
      "loss/crossentropy": 2.312969148159027,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22048135846853256,
      "step": 29126
    },
    {
      "epoch": 0.58256,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.010846964518229167,
      "learning_rate": 0.0001,
      "loss": 4.0541,
      "loss/crossentropy": 2.153268814086914,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2186889946460724,
      "step": 29128
    },
    {
      "epoch": 0.5826,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.011527252197265626,
      "learning_rate": 0.0001,
      "loss": 3.9336,
      "loss/crossentropy": 2.1008894443511963,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1893206238746643,
      "step": 29130
    },
    {
      "epoch": 0.58264,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.011331939697265625,
      "learning_rate": 0.0001,
      "loss": 4.0611,
      "loss/crossentropy": 2.0414587259292603,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1952778548002243,
      "step": 29132
    },
    {
      "epoch": 0.58268,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.013085683186848959,
      "learning_rate": 0.0001,
      "loss": 4.1025,
      "loss/crossentropy": 2.440865635871887,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21561746299266815,
      "step": 29134
    },
    {
      "epoch": 0.58272,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.012325032552083334,
      "learning_rate": 0.0001,
      "loss": 4.1542,
      "loss/crossentropy": 2.309555768966675,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22722266614437103,
      "step": 29136
    },
    {
      "epoch": 0.58276,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.012910715738932292,
      "learning_rate": 0.0001,
      "loss": 4.0456,
      "loss/crossentropy": 2.1777420043945312,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1994665190577507,
      "step": 29138
    },
    {
      "epoch": 0.5828,
      "grad_norm": 2.515625,
      "grad_norm_var": 0.03129781087239583,
      "learning_rate": 0.0001,
      "loss": 3.9863,
      "loss/crossentropy": 2.174704909324646,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19741538912057877,
      "step": 29140
    },
    {
      "epoch": 0.58284,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.03214111328125,
      "learning_rate": 0.0001,
      "loss": 3.889,
      "loss/crossentropy": 2.0467293858528137,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19651174545288086,
      "step": 29142
    },
    {
      "epoch": 0.58288,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.030418904622395833,
      "learning_rate": 0.0001,
      "loss": 4.0606,
      "loss/crossentropy": 2.050451397895813,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18523690849542618,
      "step": 29144
    },
    {
      "epoch": 0.58292,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.027787272135416666,
      "learning_rate": 0.0001,
      "loss": 4.1576,
      "loss/crossentropy": 2.2801162004470825,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20639225095510483,
      "step": 29146
    },
    {
      "epoch": 0.58296,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.03065973917643229,
      "learning_rate": 0.0001,
      "loss": 3.7002,
      "loss/crossentropy": 1.7782122492790222,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15972331911325455,
      "step": 29148
    },
    {
      "epoch": 0.583,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.029813639322916665,
      "learning_rate": 0.0001,
      "loss": 3.9282,
      "loss/crossentropy": 2.1129753589630127,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19876645505428314,
      "step": 29150
    },
    {
      "epoch": 0.58304,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.029259999593098957,
      "learning_rate": 0.0001,
      "loss": 4.0128,
      "loss/crossentropy": 1.7885400652885437,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17748355120420456,
      "step": 29152
    },
    {
      "epoch": 0.58308,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.029117584228515625,
      "learning_rate": 0.0001,
      "loss": 3.9049,
      "loss/crossentropy": 1.9744374752044678,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20229245722293854,
      "step": 29154
    },
    {
      "epoch": 0.58312,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.007785797119140625,
      "learning_rate": 0.0001,
      "loss": 3.8136,
      "loss/crossentropy": 1.7132557034492493,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17987029254436493,
      "step": 29156
    },
    {
      "epoch": 0.58316,
      "grad_norm": 1.875,
      "grad_norm_var": 0.007389068603515625,
      "learning_rate": 0.0001,
      "loss": 3.9586,
      "loss/crossentropy": 2.1243802309036255,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19921009242534637,
      "step": 29158
    },
    {
      "epoch": 0.5832,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.0078277587890625,
      "learning_rate": 0.0001,
      "loss": 3.8449,
      "loss/crossentropy": 1.9450802206993103,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19191914796829224,
      "step": 29160
    },
    {
      "epoch": 0.58324,
      "grad_norm": 1.875,
      "grad_norm_var": 0.007372792561848958,
      "learning_rate": 0.0001,
      "loss": 3.9835,
      "loss/crossentropy": 1.936660647392273,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18147709965705872,
      "step": 29162
    },
    {
      "epoch": 0.58328,
      "grad_norm": 2.125,
      "grad_norm_var": 0.009403228759765625,
      "learning_rate": 0.0001,
      "loss": 4.1735,
      "loss/crossentropy": 2.4342548847198486,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22083742171525955,
      "step": 29164
    },
    {
      "epoch": 0.58332,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0081298828125,
      "learning_rate": 0.0001,
      "loss": 3.7471,
      "loss/crossentropy": 1.8538039922714233,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17534375935792923,
      "step": 29166
    },
    {
      "epoch": 0.58336,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.006959788004557292,
      "learning_rate": 0.0001,
      "loss": 4.0749,
      "loss/crossentropy": 2.2351138591766357,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20727767050266266,
      "step": 29168
    },
    {
      "epoch": 0.5834,
      "grad_norm": 1.75,
      "grad_norm_var": 0.008055623372395833,
      "learning_rate": 0.0001,
      "loss": 3.7464,
      "loss/crossentropy": 1.655074954032898,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17760378867387772,
      "step": 29170
    },
    {
      "epoch": 0.58344,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.007902018229166667,
      "learning_rate": 0.0001,
      "loss": 3.9026,
      "loss/crossentropy": 1.7590230703353882,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17171693593263626,
      "step": 29172
    },
    {
      "epoch": 0.58348,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0090484619140625,
      "learning_rate": 0.0001,
      "loss": 3.8981,
      "loss/crossentropy": 1.7671796083450317,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16575640812516212,
      "step": 29174
    },
    {
      "epoch": 0.58352,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.012686920166015626,
      "learning_rate": 0.0001,
      "loss": 4.0845,
      "loss/crossentropy": 1.8754821419715881,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17734245210886002,
      "step": 29176
    },
    {
      "epoch": 0.58356,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.013014475504557291,
      "learning_rate": 0.0001,
      "loss": 3.9861,
      "loss/crossentropy": 2.152361035346985,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.24318483471870422,
      "step": 29178
    },
    {
      "epoch": 0.5836,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.010158030192057292,
      "learning_rate": 0.0001,
      "loss": 4.0301,
      "loss/crossentropy": 2.091741681098938,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20466475933790207,
      "step": 29180
    },
    {
      "epoch": 0.58364,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.011757151285807291,
      "learning_rate": 0.0001,
      "loss": 4.003,
      "loss/crossentropy": 2.493666410446167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20663270354270935,
      "step": 29182
    },
    {
      "epoch": 0.58368,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.012247467041015625,
      "learning_rate": 0.0001,
      "loss": 3.9938,
      "loss/crossentropy": 2.311075210571289,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2115970104932785,
      "step": 29184
    },
    {
      "epoch": 0.58372,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.010789998372395833,
      "learning_rate": 0.0001,
      "loss": 4.0891,
      "loss/crossentropy": 2.2983113527297974,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2209111899137497,
      "step": 29186
    },
    {
      "epoch": 0.58376,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.009992472330729167,
      "learning_rate": 0.0001,
      "loss": 4.1642,
      "loss/crossentropy": 2.035028040409088,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19093219190835953,
      "step": 29188
    },
    {
      "epoch": 0.5838,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.008333333333333333,
      "learning_rate": 0.0001,
      "loss": 3.9368,
      "loss/crossentropy": 2.1713255047798157,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1836071014404297,
      "step": 29190
    },
    {
      "epoch": 0.58384,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.005033111572265625,
      "learning_rate": 0.0001,
      "loss": 3.9997,
      "loss/crossentropy": 1.898379623889923,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19957400858402252,
      "step": 29192
    },
    {
      "epoch": 0.58388,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.004624176025390625,
      "learning_rate": 0.0001,
      "loss": 4.1757,
      "loss/crossentropy": 1.930584728717804,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1872590333223343,
      "step": 29194
    },
    {
      "epoch": 0.58392,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0045654296875,
      "learning_rate": 0.0001,
      "loss": 3.9956,
      "loss/crossentropy": 2.2888232469558716,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20355024188756943,
      "step": 29196
    },
    {
      "epoch": 0.58396,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.005961100260416667,
      "learning_rate": 0.0001,
      "loss": 4.0838,
      "loss/crossentropy": 2.242384433746338,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2161145657300949,
      "step": 29198
    },
    {
      "epoch": 0.584,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.0058024088541666664,
      "learning_rate": 0.0001,
      "loss": 3.9119,
      "loss/crossentropy": 1.7642216086387634,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17429838329553604,
      "step": 29200
    },
    {
      "epoch": 0.58404,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.005785878499348958,
      "learning_rate": 0.0001,
      "loss": 4.0763,
      "loss/crossentropy": 2.2808165550231934,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21707262098789215,
      "step": 29202
    },
    {
      "epoch": 0.58408,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.008158365885416666,
      "learning_rate": 0.0001,
      "loss": 4.2149,
      "loss/crossentropy": 2.2542625665664673,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21408841013908386,
      "step": 29204
    },
    {
      "epoch": 0.58412,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.0079010009765625,
      "learning_rate": 0.0001,
      "loss": 3.8883,
      "loss/crossentropy": 2.084090828895569,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18555748462677002,
      "step": 29206
    },
    {
      "epoch": 0.58416,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.01376953125,
      "learning_rate": 0.0001,
      "loss": 4.1465,
      "loss/crossentropy": 2.3946259021759033,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20357777923345566,
      "step": 29208
    },
    {
      "epoch": 0.5842,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.014697011311848958,
      "learning_rate": 0.0001,
      "loss": 4.1172,
      "loss/crossentropy": 2.2105261087417603,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2044535130262375,
      "step": 29210
    },
    {
      "epoch": 0.58424,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.018904368082682293,
      "learning_rate": 0.0001,
      "loss": 3.639,
      "loss/crossentropy": 1.5793652534484863,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1763077676296234,
      "step": 29212
    },
    {
      "epoch": 0.58428,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.016340128580729165,
      "learning_rate": 0.0001,
      "loss": 4.2916,
      "loss/crossentropy": 2.352652430534363,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22064512968063354,
      "step": 29214
    },
    {
      "epoch": 0.58432,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.017374674479166668,
      "learning_rate": 0.0001,
      "loss": 3.6412,
      "loss/crossentropy": 1.9069225192070007,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17932289838790894,
      "step": 29216
    },
    {
      "epoch": 0.58436,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.017899576822916666,
      "learning_rate": 0.0001,
      "loss": 3.8841,
      "loss/crossentropy": 1.7844374179840088,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17104727029800415,
      "step": 29218
    },
    {
      "epoch": 0.5844,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.01678441365559896,
      "learning_rate": 0.0001,
      "loss": 3.8258,
      "loss/crossentropy": 1.917112410068512,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17463716119527817,
      "step": 29220
    },
    {
      "epoch": 0.58444,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.017103830973307293,
      "learning_rate": 0.0001,
      "loss": 4.0424,
      "loss/crossentropy": 1.9697306156158447,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1797391176223755,
      "step": 29222
    },
    {
      "epoch": 0.58448,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.012092081705729167,
      "learning_rate": 0.0001,
      "loss": 4.0616,
      "loss/crossentropy": 2.1226717829704285,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1818629801273346,
      "step": 29224
    },
    {
      "epoch": 0.58452,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.010567220052083333,
      "learning_rate": 0.0001,
      "loss": 3.9699,
      "loss/crossentropy": 2.406674861907959,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20577572286128998,
      "step": 29226
    },
    {
      "epoch": 0.58456,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.008235677083333334,
      "learning_rate": 0.0001,
      "loss": 3.915,
      "loss/crossentropy": 2.14900803565979,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19237977266311646,
      "step": 29228
    },
    {
      "epoch": 0.5846,
      "grad_norm": 1.875,
      "grad_norm_var": 0.006803385416666667,
      "learning_rate": 0.0001,
      "loss": 3.9717,
      "loss/crossentropy": 1.7613251209259033,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15341567248106003,
      "step": 29230
    },
    {
      "epoch": 0.58464,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.0053403218587239586,
      "learning_rate": 0.0001,
      "loss": 3.9471,
      "loss/crossentropy": 2.0507588386535645,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1979541778564453,
      "step": 29232
    },
    {
      "epoch": 0.58468,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.005985260009765625,
      "learning_rate": 0.0001,
      "loss": 3.9946,
      "loss/crossentropy": 1.9034594297409058,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17682257294654846,
      "step": 29234
    },
    {
      "epoch": 0.58472,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0058258056640625,
      "learning_rate": 0.0001,
      "loss": 3.9959,
      "loss/crossentropy": 2.317684054374695,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21268346905708313,
      "step": 29236
    },
    {
      "epoch": 0.58476,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.0064656575520833336,
      "learning_rate": 0.0001,
      "loss": 4.2448,
      "loss/crossentropy": 2.15400493144989,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20123393833637238,
      "step": 29238
    },
    {
      "epoch": 0.5848,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.0051513671875,
      "learning_rate": 0.0001,
      "loss": 4.0108,
      "loss/crossentropy": 1.8610196709632874,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16698840260505676,
      "step": 29240
    },
    {
      "epoch": 0.58484,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.008213043212890625,
      "learning_rate": 0.0001,
      "loss": 3.9196,
      "loss/crossentropy": 1.9535472989082336,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17425117641687393,
      "step": 29242
    },
    {
      "epoch": 0.58488,
      "grad_norm": 1.875,
      "grad_norm_var": 0.008185831705729167,
      "learning_rate": 0.0001,
      "loss": 3.9185,
      "loss/crossentropy": 1.966432809829712,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19216462224721909,
      "step": 29244
    },
    {
      "epoch": 0.58492,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.009163157145182291,
      "learning_rate": 0.0001,
      "loss": 3.7805,
      "loss/crossentropy": 2.0415098071098328,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1909058690071106,
      "step": 29246
    },
    {
      "epoch": 0.58496,
      "grad_norm": 1.875,
      "grad_norm_var": 0.010025787353515624,
      "learning_rate": 0.0001,
      "loss": 4.0979,
      "loss/crossentropy": 2.1297216415405273,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19929852336645126,
      "step": 29248
    },
    {
      "epoch": 0.585,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.009417470296223958,
      "learning_rate": 0.0001,
      "loss": 4.0817,
      "loss/crossentropy": 2.1042102575302124,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20568375289440155,
      "step": 29250
    },
    {
      "epoch": 0.58504,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.009403483072916666,
      "learning_rate": 0.0001,
      "loss": 3.8923,
      "loss/crossentropy": 2.1890978813171387,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2029515951871872,
      "step": 29252
    },
    {
      "epoch": 0.58508,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009261067708333333,
      "learning_rate": 0.0001,
      "loss": 3.989,
      "loss/crossentropy": 2.061577618122101,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19042493402957916,
      "step": 29254
    },
    {
      "epoch": 0.58512,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.011366526285807291,
      "learning_rate": 0.0001,
      "loss": 4.0084,
      "loss/crossentropy": 2.0101218819618225,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19894610345363617,
      "step": 29256
    },
    {
      "epoch": 0.58516,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008332316080729167,
      "learning_rate": 0.0001,
      "loss": 4.0886,
      "loss/crossentropy": 1.7634521126747131,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18951942026615143,
      "step": 29258
    },
    {
      "epoch": 0.5852,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.06482518513997396,
      "learning_rate": 0.0001,
      "loss": 3.9788,
      "loss/crossentropy": 2.1544201374053955,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19801289588212967,
      "step": 29260
    },
    {
      "epoch": 0.58524,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.062154134114583336,
      "learning_rate": 0.0001,
      "loss": 3.9742,
      "loss/crossentropy": 2.2371232509613037,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19622696936130524,
      "step": 29262
    },
    {
      "epoch": 0.58528,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.06226781209309896,
      "learning_rate": 0.0001,
      "loss": 3.8049,
      "loss/crossentropy": 1.9571356773376465,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19151850044727325,
      "step": 29264
    },
    {
      "epoch": 0.58532,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.06256103515625,
      "learning_rate": 0.0001,
      "loss": 4.1047,
      "loss/crossentropy": 2.149628520011902,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19668041914701462,
      "step": 29266
    },
    {
      "epoch": 0.58536,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.06358006795247396,
      "learning_rate": 0.0001,
      "loss": 4.186,
      "loss/crossentropy": 2.01633083820343,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19449809193611145,
      "step": 29268
    },
    {
      "epoch": 0.5854,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.06281636555989584,
      "learning_rate": 0.0001,
      "loss": 4.0058,
      "loss/crossentropy": 2.144730567932129,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20602761209011078,
      "step": 29270
    },
    {
      "epoch": 0.58544,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.06256917317708334,
      "learning_rate": 0.0001,
      "loss": 3.9861,
      "loss/crossentropy": 1.7255420684814453,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16974858194589615,
      "step": 29272
    },
    {
      "epoch": 0.58548,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.06447931925455729,
      "learning_rate": 0.0001,
      "loss": 4.1475,
      "loss/crossentropy": 2.3286606073379517,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2005838081240654,
      "step": 29274
    },
    {
      "epoch": 0.58552,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0079498291015625,
      "learning_rate": 0.0001,
      "loss": 4.1118,
      "loss/crossentropy": 2.085566818714142,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19351552426815033,
      "step": 29276
    },
    {
      "epoch": 0.58556,
      "grad_norm": 1.7421875,
      "grad_norm_var": 0.010188547770182292,
      "learning_rate": 0.0001,
      "loss": 3.8693,
      "loss/crossentropy": 1.9797600507736206,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2020813450217247,
      "step": 29278
    },
    {
      "epoch": 0.5856,
      "grad_norm": 2.0,
      "grad_norm_var": 0.011328125,
      "learning_rate": 0.0001,
      "loss": 4.0625,
      "loss/crossentropy": 2.2091132402420044,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1922968104481697,
      "step": 29280
    },
    {
      "epoch": 0.58564,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.011665852864583333,
      "learning_rate": 0.0001,
      "loss": 4.0093,
      "loss/crossentropy": 2.220807909965515,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21127308905124664,
      "step": 29282
    },
    {
      "epoch": 0.58568,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.013206990559895833,
      "learning_rate": 0.0001,
      "loss": 3.9675,
      "loss/crossentropy": 1.9829080700874329,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18882058560848236,
      "step": 29284
    },
    {
      "epoch": 0.58572,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.011864217122395833,
      "learning_rate": 0.0001,
      "loss": 4.1373,
      "loss/crossentropy": 1.9328150153160095,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1966244801878929,
      "step": 29286
    },
    {
      "epoch": 0.58576,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0115631103515625,
      "learning_rate": 0.0001,
      "loss": 4.0272,
      "loss/crossentropy": 2.109495759010315,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19433388113975525,
      "step": 29288
    },
    {
      "epoch": 0.5858,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.009913889567057292,
      "learning_rate": 0.0001,
      "loss": 4.1732,
      "loss/crossentropy": 2.3193604946136475,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2086661458015442,
      "step": 29290
    },
    {
      "epoch": 0.58584,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.010422515869140624,
      "learning_rate": 0.0001,
      "loss": 3.9131,
      "loss/crossentropy": 2.015924036502838,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1846477910876274,
      "step": 29292
    },
    {
      "epoch": 0.58588,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.007323201497395833,
      "learning_rate": 0.0001,
      "loss": 4.1275,
      "loss/crossentropy": 2.2601329684257507,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19521460682153702,
      "step": 29294
    },
    {
      "epoch": 0.58592,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.005890909830729167,
      "learning_rate": 0.0001,
      "loss": 3.7702,
      "loss/crossentropy": 1.8056439757347107,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19355197250843048,
      "step": 29296
    },
    {
      "epoch": 0.58596,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.005564117431640625,
      "learning_rate": 0.0001,
      "loss": 4.2989,
      "loss/crossentropy": 2.071826934814453,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19889701902866364,
      "step": 29298
    },
    {
      "epoch": 0.586,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0033111572265625,
      "learning_rate": 0.0001,
      "loss": 4.0176,
      "loss/crossentropy": 2.2320820093154907,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21529612690210342,
      "step": 29300
    },
    {
      "epoch": 0.58604,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0034912109375,
      "learning_rate": 0.0001,
      "loss": 4.2805,
      "loss/crossentropy": 2.1989121437072754,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20347003638744354,
      "step": 29302
    },
    {
      "epoch": 0.58608,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.0037679036458333334,
      "learning_rate": 0.0001,
      "loss": 3.8216,
      "loss/crossentropy": 2.128119111061096,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1872687190771103,
      "step": 29304
    },
    {
      "epoch": 0.58612,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.010396067301432292,
      "learning_rate": 0.0001,
      "loss": 4.0913,
      "loss/crossentropy": 1.9635827541351318,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1851392686367035,
      "step": 29306
    },
    {
      "epoch": 0.58616,
      "grad_norm": 2.1875,
      "grad_norm_var": 0.0121002197265625,
      "learning_rate": 0.0001,
      "loss": 4.2218,
      "loss/crossentropy": 2.2543106079101562,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21720967441797256,
      "step": 29308
    },
    {
      "epoch": 0.5862,
      "grad_norm": 2.0,
      "grad_norm_var": 0.013387044270833334,
      "learning_rate": 0.0001,
      "loss": 3.8456,
      "loss/crossentropy": 2.3444888591766357,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2140304371714592,
      "step": 29310
    },
    {
      "epoch": 0.58624,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.013182576497395833,
      "learning_rate": 0.0001,
      "loss": 4.1098,
      "loss/crossentropy": 2.0403956174850464,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20437155663967133,
      "step": 29312
    },
    {
      "epoch": 0.58628,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.01605199178059896,
      "learning_rate": 0.0001,
      "loss": 4.0612,
      "loss/crossentropy": 1.9070860743522644,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18735133856534958,
      "step": 29314
    },
    {
      "epoch": 0.58632,
      "grad_norm": 1.875,
      "grad_norm_var": 0.018308258056640624,
      "learning_rate": 0.0001,
      "loss": 3.917,
      "loss/crossentropy": 1.9588303565979004,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16786205023527145,
      "step": 29316
    },
    {
      "epoch": 0.58636,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.0215240478515625,
      "learning_rate": 0.0001,
      "loss": 3.8377,
      "loss/crossentropy": 1.8938056230545044,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18695908039808273,
      "step": 29318
    },
    {
      "epoch": 0.5864,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.022867584228515626,
      "learning_rate": 0.0001,
      "loss": 3.7435,
      "loss/crossentropy": 2.254639148712158,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19976592808961868,
      "step": 29320
    },
    {
      "epoch": 0.58644,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.01688232421875,
      "learning_rate": 0.0001,
      "loss": 3.7988,
      "loss/crossentropy": 2.2075421810150146,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18852433562278748,
      "step": 29322
    },
    {
      "epoch": 0.58648,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.010860188802083334,
      "learning_rate": 0.0001,
      "loss": 3.9432,
      "loss/crossentropy": 1.6794416904449463,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18429157137870789,
      "step": 29324
    },
    {
      "epoch": 0.58652,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.010111236572265625,
      "learning_rate": 0.0001,
      "loss": 4.0925,
      "loss/crossentropy": 2.3620080947875977,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21314998716115952,
      "step": 29326
    },
    {
      "epoch": 0.58656,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0092681884765625,
      "learning_rate": 0.0001,
      "loss": 4.1378,
      "loss/crossentropy": 2.3381006717681885,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20351716130971909,
      "step": 29328
    },
    {
      "epoch": 0.5866,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.009765370686848959,
      "learning_rate": 0.0001,
      "loss": 3.9563,
      "loss/crossentropy": 2.0491183400154114,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19619758427143097,
      "step": 29330
    },
    {
      "epoch": 0.58664,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.009855143229166667,
      "learning_rate": 0.0001,
      "loss": 3.8267,
      "loss/crossentropy": 1.7091345191001892,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15915849804878235,
      "step": 29332
    },
    {
      "epoch": 0.58668,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.005589803059895833,
      "learning_rate": 0.0001,
      "loss": 4.0168,
      "loss/crossentropy": 2.1986491680145264,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1997922584414482,
      "step": 29334
    },
    {
      "epoch": 0.58672,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.006086985270182292,
      "learning_rate": 0.0001,
      "loss": 3.8094,
      "loss/crossentropy": 2.051343023777008,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19083235412836075,
      "step": 29336
    },
    {
      "epoch": 0.58676,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.008845774332682292,
      "learning_rate": 0.0001,
      "loss": 3.7667,
      "loss/crossentropy": 2.0580617785453796,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19386407732963562,
      "step": 29338
    },
    {
      "epoch": 0.5868,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.007983144124348958,
      "learning_rate": 0.0001,
      "loss": 3.8437,
      "loss/crossentropy": 1.988048791885376,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16248874366283417,
      "step": 29340
    },
    {
      "epoch": 0.58684,
      "grad_norm": 2.359375,
      "grad_norm_var": 0.022215779622395834,
      "learning_rate": 0.0001,
      "loss": 3.7604,
      "loss/crossentropy": 1.9649037718772888,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17722021788358688,
      "step": 29342
    },
    {
      "epoch": 0.58688,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.021801503499348958,
      "learning_rate": 0.0001,
      "loss": 4.1983,
      "loss/crossentropy": 2.014201283454895,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19323624670505524,
      "step": 29344
    },
    {
      "epoch": 0.58692,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.021170806884765626,
      "learning_rate": 0.0001,
      "loss": 3.8541,
      "loss/crossentropy": 1.9666126370429993,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17525266110897064,
      "step": 29346
    },
    {
      "epoch": 0.58696,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.021923573811848958,
      "learning_rate": 0.0001,
      "loss": 3.7258,
      "loss/crossentropy": 2.10392427444458,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1954725682735443,
      "step": 29348
    },
    {
      "epoch": 0.587,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.0308013916015625,
      "learning_rate": 0.0001,
      "loss": 4.0668,
      "loss/crossentropy": 1.6613619327545166,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21818602830171585,
      "step": 29350
    },
    {
      "epoch": 0.58704,
      "grad_norm": 1.875,
      "grad_norm_var": 0.030460611979166666,
      "learning_rate": 0.0001,
      "loss": 4.0683,
      "loss/crossentropy": 2.4305994510650635,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2147374525666237,
      "step": 29352
    },
    {
      "epoch": 0.58708,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.026944986979166665,
      "learning_rate": 0.0001,
      "loss": 4.2551,
      "loss/crossentropy": 2.1946258544921875,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22057125717401505,
      "step": 29354
    },
    {
      "epoch": 0.58712,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.025852203369140625,
      "learning_rate": 0.0001,
      "loss": 4.1854,
      "loss/crossentropy": 2.3252480030059814,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20392987877130508,
      "step": 29356
    },
    {
      "epoch": 0.58716,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.015999348958333333,
      "learning_rate": 0.0001,
      "loss": 3.9164,
      "loss/crossentropy": 1.9041990041732788,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1759115606546402,
      "step": 29358
    },
    {
      "epoch": 0.5872,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.0171783447265625,
      "learning_rate": 0.0001,
      "loss": 3.9968,
      "loss/crossentropy": 1.9955511689186096,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1914498507976532,
      "step": 29360
    },
    {
      "epoch": 0.58724,
      "grad_norm": 2.0,
      "grad_norm_var": 0.0184814453125,
      "learning_rate": 0.0001,
      "loss": 3.7173,
      "loss/crossentropy": 1.9557825922966003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19295810908079147,
      "step": 29362
    },
    {
      "epoch": 0.58728,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.01850153605143229,
      "learning_rate": 0.0001,
      "loss": 3.6251,
      "loss/crossentropy": 1.4499501585960388,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.143144428730011,
      "step": 29364
    },
    {
      "epoch": 0.58732,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.010536448160807291,
      "learning_rate": 0.0001,
      "loss": 3.8048,
      "loss/crossentropy": 2.1275508999824524,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19290220737457275,
      "step": 29366
    },
    {
      "epoch": 0.58736,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.011177317301432291,
      "learning_rate": 0.0001,
      "loss": 3.7983,
      "loss/crossentropy": 2.320215940475464,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19288549572229385,
      "step": 29368
    },
    {
      "epoch": 0.5874,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.009248606363932292,
      "learning_rate": 0.0001,
      "loss": 4.0161,
      "loss/crossentropy": 1.9968576431274414,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18840061873197556,
      "step": 29370
    },
    {
      "epoch": 0.58744,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.005671946207682291,
      "learning_rate": 0.0001,
      "loss": 4.2287,
      "loss/crossentropy": 2.2592599391937256,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21203969419002533,
      "step": 29372
    },
    {
      "epoch": 0.58748,
      "grad_norm": 2.0,
      "grad_norm_var": 0.007051595052083333,
      "learning_rate": 0.0001,
      "loss": 4.1003,
      "loss/crossentropy": 1.9274957180023193,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17894938588142395,
      "step": 29374
    },
    {
      "epoch": 0.58752,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.006982421875,
      "learning_rate": 0.0001,
      "loss": 3.7913,
      "loss/crossentropy": 2.252523422241211,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21168971061706543,
      "step": 29376
    },
    {
      "epoch": 0.58756,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.010188802083333334,
      "learning_rate": 0.0001,
      "loss": 3.95,
      "loss/crossentropy": 2.110221028327942,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2013455107808113,
      "step": 29378
    },
    {
      "epoch": 0.5876,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.008750152587890626,
      "learning_rate": 0.0001,
      "loss": 3.8604,
      "loss/crossentropy": 1.9283993244171143,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17572121322155,
      "step": 29380
    },
    {
      "epoch": 0.58764,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.007950592041015624,
      "learning_rate": 0.0001,
      "loss": 3.988,
      "loss/crossentropy": 2.23159658908844,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21318458020687103,
      "step": 29382
    },
    {
      "epoch": 0.58768,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.0097076416015625,
      "learning_rate": 0.0001,
      "loss": 3.9589,
      "loss/crossentropy": 2.2170876264572144,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22383015602827072,
      "step": 29384
    },
    {
      "epoch": 0.58772,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.011173248291015625,
      "learning_rate": 0.0001,
      "loss": 4.0071,
      "loss/crossentropy": 2.2335199117660522,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19595736265182495,
      "step": 29386
    },
    {
      "epoch": 0.58776,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0118072509765625,
      "learning_rate": 0.0001,
      "loss": 3.9657,
      "loss/crossentropy": 2.165065050125122,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20269524306058884,
      "step": 29388
    },
    {
      "epoch": 0.5878,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.012400054931640625,
      "learning_rate": 0.0001,
      "loss": 3.5373,
      "loss/crossentropy": 1.8111371397972107,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17589127272367477,
      "step": 29390
    },
    {
      "epoch": 0.58784,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.015110015869140625,
      "learning_rate": 0.0001,
      "loss": 3.8698,
      "loss/crossentropy": 2.083921492099762,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19300700724124908,
      "step": 29392
    },
    {
      "epoch": 0.58788,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.010929361979166666,
      "learning_rate": 0.0001,
      "loss": 4.168,
      "loss/crossentropy": 1.916080355644226,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18377907574176788,
      "step": 29394
    },
    {
      "epoch": 0.58792,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.011842600504557292,
      "learning_rate": 0.0001,
      "loss": 4.2627,
      "loss/crossentropy": 2.2706268429756165,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2135249674320221,
      "step": 29396
    },
    {
      "epoch": 0.58796,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.0125396728515625,
      "learning_rate": 0.0001,
      "loss": 3.649,
      "loss/crossentropy": 1.99881511926651,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20278233289718628,
      "step": 29398
    },
    {
      "epoch": 0.588,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.009087880452473959,
      "learning_rate": 0.0001,
      "loss": 3.9936,
      "loss/crossentropy": 1.9932519793510437,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18421290814876556,
      "step": 29400
    },
    {
      "epoch": 0.58804,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.008893839518229167,
      "learning_rate": 0.0001,
      "loss": 4.0858,
      "loss/crossentropy": 2.287453770637512,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20635423064231873,
      "step": 29402
    },
    {
      "epoch": 0.58808,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.008544921875,
      "learning_rate": 0.0001,
      "loss": 4.3109,
      "loss/crossentropy": 2.2095491886138916,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1934846043586731,
      "step": 29404
    },
    {
      "epoch": 0.58812,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.0108795166015625,
      "learning_rate": 0.0001,
      "loss": 4.1356,
      "loss/crossentropy": 2.0484838485717773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19288712739944458,
      "step": 29406
    },
    {
      "epoch": 0.58816,
      "grad_norm": 2.0,
      "grad_norm_var": 0.009110514322916667,
      "learning_rate": 0.0001,
      "loss": 3.9269,
      "loss/crossentropy": 2.431947708129883,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21592531353235245,
      "step": 29408
    },
    {
      "epoch": 0.5882,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.009049479166666667,
      "learning_rate": 0.0001,
      "loss": 4.0748,
      "loss/crossentropy": 2.0007177591323853,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1799403429031372,
      "step": 29410
    },
    {
      "epoch": 0.58824,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.009224446614583333,
      "learning_rate": 0.0001,
      "loss": 4.1588,
      "loss/crossentropy": 2.168426990509033,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2031354159116745,
      "step": 29412
    },
    {
      "epoch": 0.58828,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.007917277018229167,
      "learning_rate": 0.0001,
      "loss": 4.0162,
      "loss/crossentropy": 2.1504631638526917,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19324814528226852,
      "step": 29414
    },
    {
      "epoch": 0.58832,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.011515299479166666,
      "learning_rate": 0.0001,
      "loss": 3.9259,
      "loss/crossentropy": 2.013205111026764,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18360282480716705,
      "step": 29416
    },
    {
      "epoch": 0.58836,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.010776519775390625,
      "learning_rate": 0.0001,
      "loss": 4.0384,
      "loss/crossentropy": 2.155342757701874,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2049562856554985,
      "step": 29418
    },
    {
      "epoch": 0.5884,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.012041982014973958,
      "learning_rate": 0.0001,
      "loss": 4.0741,
      "loss/crossentropy": 2.117911696434021,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18994345515966415,
      "step": 29420
    },
    {
      "epoch": 0.58844,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.009422810872395833,
      "learning_rate": 0.0001,
      "loss": 4.0517,
      "loss/crossentropy": 1.980055332183838,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1862584576010704,
      "step": 29422
    },
    {
      "epoch": 0.58848,
      "grad_norm": 2.078125,
      "grad_norm_var": 0.009364573160807292,
      "learning_rate": 0.0001,
      "loss": 4.1284,
      "loss/crossentropy": 2.2244694232940674,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21067889779806137,
      "step": 29424
    },
    {
      "epoch": 0.58852,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.008876291910807292,
      "learning_rate": 0.0001,
      "loss": 4.1993,
      "loss/crossentropy": 2.213523805141449,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20092912763357162,
      "step": 29426
    },
    {
      "epoch": 0.58856,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.0081695556640625,
      "learning_rate": 0.0001,
      "loss": 4.1802,
      "loss/crossentropy": 1.9292373657226562,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17156004905700684,
      "step": 29428
    },
    {
      "epoch": 0.5886,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.011537424723307292,
      "learning_rate": 0.0001,
      "loss": 4.0741,
      "loss/crossentropy": 2.4728565216064453,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22577238827943802,
      "step": 29430
    },
    {
      "epoch": 0.58864,
      "grad_norm": 1.78125,
      "grad_norm_var": 0.011639149983723958,
      "learning_rate": 0.0001,
      "loss": 3.9782,
      "loss/crossentropy": 1.757906973361969,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19193939119577408,
      "step": 29432
    },
    {
      "epoch": 0.58868,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.011351521809895833,
      "learning_rate": 0.0001,
      "loss": 4.019,
      "loss/crossentropy": 2.120426893234253,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19062675535678864,
      "step": 29434
    },
    {
      "epoch": 0.58872,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.01217041015625,
      "learning_rate": 0.0001,
      "loss": 3.8409,
      "loss/crossentropy": 1.7993061542510986,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1687781661748886,
      "step": 29436
    },
    {
      "epoch": 0.58876,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.01608250935872396,
      "learning_rate": 0.0001,
      "loss": 3.775,
      "loss/crossentropy": 1.995628297328949,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18463349342346191,
      "step": 29438
    },
    {
      "epoch": 0.5888,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.015329742431640625,
      "learning_rate": 0.0001,
      "loss": 3.8311,
      "loss/crossentropy": 2.411523938179016,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2045421302318573,
      "step": 29440
    },
    {
      "epoch": 0.58884,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.017288970947265624,
      "learning_rate": 0.0001,
      "loss": 3.7145,
      "loss/crossentropy": 1.9225006103515625,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17880843579769135,
      "step": 29442
    },
    {
      "epoch": 0.58888,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.017594401041666666,
      "learning_rate": 0.0001,
      "loss": 3.9736,
      "loss/crossentropy": 2.1082658171653748,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19336317479610443,
      "step": 29444
    },
    {
      "epoch": 0.58892,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.013362630208333334,
      "learning_rate": 0.0001,
      "loss": 4.0781,
      "loss/crossentropy": 2.350805640220642,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19736752659082413,
      "step": 29446
    },
    {
      "epoch": 0.58896,
      "grad_norm": 1.875,
      "grad_norm_var": 0.009712727864583333,
      "learning_rate": 0.0001,
      "loss": 3.9396,
      "loss/crossentropy": 2.1030595302581787,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1941584125161171,
      "step": 29448
    },
    {
      "epoch": 0.589,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.009700266520182292,
      "learning_rate": 0.0001,
      "loss": 3.6338,
      "loss/crossentropy": 2.173567295074463,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19058758020401,
      "step": 29450
    },
    {
      "epoch": 0.58904,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0044514973958333336,
      "learning_rate": 0.0001,
      "loss": 4.1794,
      "loss/crossentropy": 2.113981068134308,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20183458924293518,
      "step": 29452
    },
    {
      "epoch": 0.58908,
      "grad_norm": 1.6796875,
      "grad_norm_var": 0.0050537109375,
      "learning_rate": 0.0001,
      "loss": 3.8333,
      "loss/crossentropy": 1.6564378142356873,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16441450640559196,
      "step": 29454
    },
    {
      "epoch": 0.58912,
      "grad_norm": 1.671875,
      "grad_norm_var": 0.007103474934895834,
      "learning_rate": 0.0001,
      "loss": 3.591,
      "loss/crossentropy": 1.7756662368774414,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1805124208331108,
      "step": 29456
    },
    {
      "epoch": 0.58916,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.006990305582682292,
      "learning_rate": 0.0001,
      "loss": 3.8145,
      "loss/crossentropy": 1.825185477733612,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1628718078136444,
      "step": 29458
    },
    {
      "epoch": 0.5892,
      "grad_norm": 1.875,
      "grad_norm_var": 0.006001790364583333,
      "learning_rate": 0.0001,
      "loss": 3.9426,
      "loss/crossentropy": 2.0170122385025024,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19748982787132263,
      "step": 29460
    },
    {
      "epoch": 0.58924,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.006783040364583334,
      "learning_rate": 0.0001,
      "loss": 3.8533,
      "loss/crossentropy": 1.8711951971054077,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20168818533420563,
      "step": 29462
    },
    {
      "epoch": 0.58928,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.006648508707682291,
      "learning_rate": 0.0001,
      "loss": 3.7884,
      "loss/crossentropy": 1.668130338191986,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1706988662481308,
      "step": 29464
    },
    {
      "epoch": 0.58932,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.006552886962890625,
      "learning_rate": 0.0001,
      "loss": 3.8763,
      "loss/crossentropy": 1.8372064232826233,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17468038201332092,
      "step": 29466
    },
    {
      "epoch": 0.58936,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.006811269124348958,
      "learning_rate": 0.0001,
      "loss": 4.235,
      "loss/crossentropy": 2.234304904937744,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1845388039946556,
      "step": 29468
    },
    {
      "epoch": 0.5894,
      "grad_norm": 1.7421875,
      "grad_norm_var": 0.00662841796875,
      "learning_rate": 0.0001,
      "loss": 3.5877,
      "loss/crossentropy": 1.9101244807243347,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17081473767757416,
      "step": 29470
    },
    {
      "epoch": 0.58944,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.004353586832682292,
      "learning_rate": 0.0001,
      "loss": 4.0295,
      "loss/crossentropy": 2.2079320549964905,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1952483355998993,
      "step": 29472
    },
    {
      "epoch": 0.58948,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.00458984375,
      "learning_rate": 0.0001,
      "loss": 3.9308,
      "loss/crossentropy": 1.8550963997840881,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18943903595209122,
      "step": 29474
    },
    {
      "epoch": 0.58952,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.004865519205729167,
      "learning_rate": 0.0001,
      "loss": 3.7654,
      "loss/crossentropy": 1.7661077976226807,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19584594666957855,
      "step": 29476
    },
    {
      "epoch": 0.58956,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.005029042561848958,
      "learning_rate": 0.0001,
      "loss": 4.2088,
      "loss/crossentropy": 2.338898777961731,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21608547866344452,
      "step": 29478
    },
    {
      "epoch": 0.5896,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.005242665608723958,
      "learning_rate": 0.0001,
      "loss": 4.1212,
      "loss/crossentropy": 2.3376599550247192,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20884696394205093,
      "step": 29480
    },
    {
      "epoch": 0.58964,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.005964152018229167,
      "learning_rate": 0.0001,
      "loss": 4.16,
      "loss/crossentropy": 2.241575002670288,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2200220599770546,
      "step": 29482
    },
    {
      "epoch": 0.58968,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.017805735270182293,
      "learning_rate": 0.0001,
      "loss": 4.0612,
      "loss/crossentropy": 2.010516047477722,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22505514323711395,
      "step": 29484
    },
    {
      "epoch": 0.58972,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.017545318603515624,
      "learning_rate": 0.0001,
      "loss": 3.9692,
      "loss/crossentropy": 2.0848931670188904,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21239446848630905,
      "step": 29486
    },
    {
      "epoch": 0.58976,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.016315714518229166,
      "learning_rate": 0.0001,
      "loss": 4.1173,
      "loss/crossentropy": 2.0518574118614197,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19289197772741318,
      "step": 29488
    },
    {
      "epoch": 0.5898,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.016460927327473958,
      "learning_rate": 0.0001,
      "loss": 4.2193,
      "loss/crossentropy": 2.1830244064331055,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21049733459949493,
      "step": 29490
    },
    {
      "epoch": 0.58984,
      "grad_norm": 2.0,
      "grad_norm_var": 0.018833160400390625,
      "learning_rate": 0.0001,
      "loss": 4.416,
      "loss/crossentropy": 2.1932766437530518,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1886359080672264,
      "step": 29492
    },
    {
      "epoch": 0.58988,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.018595123291015626,
      "learning_rate": 0.0001,
      "loss": 4.1936,
      "loss/crossentropy": 2.0677687525749207,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19729864597320557,
      "step": 29494
    },
    {
      "epoch": 0.58992,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0149169921875,
      "learning_rate": 0.0001,
      "loss": 4.1461,
      "loss/crossentropy": 2.181239366531372,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2087482586503029,
      "step": 29496
    },
    {
      "epoch": 0.58996,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.013132476806640625,
      "learning_rate": 0.0001,
      "loss": 4.0357,
      "loss/crossentropy": 2.2734841108322144,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22230970114469528,
      "step": 29498
    },
    {
      "epoch": 0.59,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.007454427083333334,
      "learning_rate": 0.0001,
      "loss": 4.1009,
      "loss/crossentropy": 2.1078773736953735,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18602187931537628,
      "step": 29500
    },
    {
      "epoch": 0.59004,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.009635416666666667,
      "learning_rate": 0.0001,
      "loss": 3.7355,
      "loss/crossentropy": 1.856580138206482,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.186849907040596,
      "step": 29502
    },
    {
      "epoch": 0.59008,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.011842600504557292,
      "learning_rate": 0.0001,
      "loss": 3.6756,
      "loss/crossentropy": 1.90110582113266,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18654846400022507,
      "step": 29504
    },
    {
      "epoch": 0.59012,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0134918212890625,
      "learning_rate": 0.0001,
      "loss": 4.0382,
      "loss/crossentropy": 1.874162197113037,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18833528459072113,
      "step": 29506
    },
    {
      "epoch": 0.59016,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.009694163004557292,
      "learning_rate": 0.0001,
      "loss": 4.0473,
      "loss/crossentropy": 2.0299471020698547,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19723627716302872,
      "step": 29508
    },
    {
      "epoch": 0.5902,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.01085205078125,
      "learning_rate": 0.0001,
      "loss": 3.9377,
      "loss/crossentropy": 1.8558722138404846,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18302107602357864,
      "step": 29510
    },
    {
      "epoch": 0.59024,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.010359446207682291,
      "learning_rate": 0.0001,
      "loss": 4.0567,
      "loss/crossentropy": 1.8753638863563538,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2061474472284317,
      "step": 29512
    },
    {
      "epoch": 0.59028,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.011954498291015626,
      "learning_rate": 0.0001,
      "loss": 3.8181,
      "loss/crossentropy": 1.9646863341331482,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18984466791152954,
      "step": 29514
    },
    {
      "epoch": 0.59032,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.0124755859375,
      "learning_rate": 0.0001,
      "loss": 3.7632,
      "loss/crossentropy": 1.9403189420700073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17063701152801514,
      "step": 29516
    },
    {
      "epoch": 0.59036,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.010774739583333333,
      "learning_rate": 0.0001,
      "loss": 3.7807,
      "loss/crossentropy": 1.9278589487075806,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1793733760714531,
      "step": 29518
    },
    {
      "epoch": 0.5904,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.01877009073893229,
      "learning_rate": 0.0001,
      "loss": 4.109,
      "loss/crossentropy": 2.2447856664657593,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21776193380355835,
      "step": 29520
    },
    {
      "epoch": 0.59044,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.015572102864583333,
      "learning_rate": 0.0001,
      "loss": 4.0259,
      "loss/crossentropy": 2.3484312891960144,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2081959918141365,
      "step": 29522
    },
    {
      "epoch": 0.59048,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.016434478759765624,
      "learning_rate": 0.0001,
      "loss": 3.9571,
      "loss/crossentropy": 2.121490001678467,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2033299133181572,
      "step": 29524
    },
    {
      "epoch": 0.59052,
      "grad_norm": 2.140625,
      "grad_norm_var": 0.017758941650390624,
      "learning_rate": 0.0001,
      "loss": 3.9742,
      "loss/crossentropy": 2.3825392723083496,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22553616762161255,
      "step": 29526
    },
    {
      "epoch": 0.59056,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.01761652628580729,
      "learning_rate": 0.0001,
      "loss": 3.9717,
      "loss/crossentropy": 2.027899742126465,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18623250722885132,
      "step": 29528
    },
    {
      "epoch": 0.5906,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.017732747395833335,
      "learning_rate": 0.0001,
      "loss": 4.3458,
      "loss/crossentropy": 2.231070041656494,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22132613509893417,
      "step": 29530
    },
    {
      "epoch": 0.59064,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.01613133748372396,
      "learning_rate": 0.0001,
      "loss": 4.1003,
      "loss/crossentropy": 2.16735976934433,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1951517015695572,
      "step": 29532
    },
    {
      "epoch": 0.59068,
      "grad_norm": 2.234375,
      "grad_norm_var": 0.019188435872395833,
      "learning_rate": 0.0001,
      "loss": 3.8917,
      "loss/crossentropy": 1.8973720073699951,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18189363181591034,
      "step": 29534
    },
    {
      "epoch": 0.59072,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.01687800089518229,
      "learning_rate": 0.0001,
      "loss": 3.7249,
      "loss/crossentropy": 2.0840883255004883,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19777310639619827,
      "step": 29536
    },
    {
      "epoch": 0.59076,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.016291300455729168,
      "learning_rate": 0.0001,
      "loss": 3.8481,
      "loss/crossentropy": 1.7272367477416992,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1722969114780426,
      "step": 29538
    },
    {
      "epoch": 0.5908,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.015024566650390625,
      "learning_rate": 0.0001,
      "loss": 3.9883,
      "loss/crossentropy": 2.2268466353416443,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19153434038162231,
      "step": 29540
    },
    {
      "epoch": 0.59084,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.012739817301432291,
      "learning_rate": 0.0001,
      "loss": 4.0964,
      "loss/crossentropy": 2.4328715801239014,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21078508347272873,
      "step": 29542
    },
    {
      "epoch": 0.59088,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.013325754801432292,
      "learning_rate": 0.0001,
      "loss": 4.0033,
      "loss/crossentropy": 1.9592428803443909,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19666817784309387,
      "step": 29544
    },
    {
      "epoch": 0.59092,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.011620076497395833,
      "learning_rate": 0.0001,
      "loss": 4.203,
      "loss/crossentropy": 1.8915135860443115,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18332552164793015,
      "step": 29546
    },
    {
      "epoch": 0.59096,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.011834462483723959,
      "learning_rate": 0.0001,
      "loss": 3.884,
      "loss/crossentropy": 1.9134865403175354,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2012096270918846,
      "step": 29548
    },
    {
      "epoch": 0.591,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0061948140462239586,
      "learning_rate": 0.0001,
      "loss": 4.2232,
      "loss/crossentropy": 2.280379056930542,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19985026121139526,
      "step": 29550
    },
    {
      "epoch": 0.59104,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.00355224609375,
      "learning_rate": 0.0001,
      "loss": 4.144,
      "loss/crossentropy": 2.000941574573517,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18684983253479004,
      "step": 29552
    },
    {
      "epoch": 0.59108,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.005020904541015625,
      "learning_rate": 0.0001,
      "loss": 3.7859,
      "loss/crossentropy": 1.9749475121498108,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18268440663814545,
      "step": 29554
    },
    {
      "epoch": 0.59112,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.0055653889973958336,
      "learning_rate": 0.0001,
      "loss": 4.0483,
      "loss/crossentropy": 2.169970750808716,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20792870223522186,
      "step": 29556
    },
    {
      "epoch": 0.59116,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.005712890625,
      "learning_rate": 0.0001,
      "loss": 4.0875,
      "loss/crossentropy": 2.059566855430603,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1943632885813713,
      "step": 29558
    },
    {
      "epoch": 0.5912,
      "grad_norm": 2.53125,
      "grad_norm_var": 0.0299713134765625,
      "learning_rate": 0.0001,
      "loss": 4.0545,
      "loss/crossentropy": 1.80315762758255,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15433073043823242,
      "step": 29560
    },
    {
      "epoch": 0.59124,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0330078125,
      "learning_rate": 0.0001,
      "loss": 3.9351,
      "loss/crossentropy": 2.1045247316360474,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19684398919343948,
      "step": 29562
    },
    {
      "epoch": 0.59128,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.03242162068684896,
      "learning_rate": 0.0001,
      "loss": 4.1244,
      "loss/crossentropy": 2.2456302642822266,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1998264193534851,
      "step": 29564
    },
    {
      "epoch": 0.59132,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.032389322916666664,
      "learning_rate": 0.0001,
      "loss": 3.8945,
      "loss/crossentropy": 2.105454444885254,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18675578385591507,
      "step": 29566
    },
    {
      "epoch": 0.59136,
      "grad_norm": 2.0,
      "grad_norm_var": 0.03316624959309896,
      "learning_rate": 0.0001,
      "loss": 4.0291,
      "loss/crossentropy": 1.8126618266105652,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19125473499298096,
      "step": 29568
    },
    {
      "epoch": 0.5914,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.032246907552083336,
      "learning_rate": 0.0001,
      "loss": 3.873,
      "loss/crossentropy": 1.8100510239601135,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18192324042320251,
      "step": 29570
    },
    {
      "epoch": 0.59144,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.031160481770833335,
      "learning_rate": 0.0001,
      "loss": 3.6655,
      "loss/crossentropy": 1.7124195098876953,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16115359216928482,
      "step": 29572
    },
    {
      "epoch": 0.59148,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.03163960774739583,
      "learning_rate": 0.0001,
      "loss": 3.9359,
      "loss/crossentropy": 2.2159979939460754,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20147743076086044,
      "step": 29574
    },
    {
      "epoch": 0.59152,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.009722646077473958,
      "learning_rate": 0.0001,
      "loss": 4.2278,
      "loss/crossentropy": 2.195053517818451,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20202681422233582,
      "step": 29576
    },
    {
      "epoch": 0.59156,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.007806142171223958,
      "learning_rate": 0.0001,
      "loss": 4.1261,
      "loss/crossentropy": 1.992495834827423,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18599994480609894,
      "step": 29578
    },
    {
      "epoch": 0.5916,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008147939046223959,
      "learning_rate": 0.0001,
      "loss": 4.0063,
      "loss/crossentropy": 2.1876922845840454,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.197817362844944,
      "step": 29580
    },
    {
      "epoch": 0.59164,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.008455403645833333,
      "learning_rate": 0.0001,
      "loss": 4.1484,
      "loss/crossentropy": 2.1962021589279175,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2175462618470192,
      "step": 29582
    },
    {
      "epoch": 0.59168,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.011242421468098958,
      "learning_rate": 0.0001,
      "loss": 3.6774,
      "loss/crossentropy": 1.9215996265411377,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18740177899599075,
      "step": 29584
    },
    {
      "epoch": 0.59172,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.007787068684895833,
      "learning_rate": 0.0001,
      "loss": 3.862,
      "loss/crossentropy": 1.6820173263549805,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1574452817440033,
      "step": 29586
    },
    {
      "epoch": 0.59176,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.007877349853515625,
      "learning_rate": 0.0001,
      "loss": 4.0942,
      "loss/crossentropy": 2.1028271913528442,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.187875434756279,
      "step": 29588
    },
    {
      "epoch": 0.5918,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.007865142822265626,
      "learning_rate": 0.0001,
      "loss": 3.7429,
      "loss/crossentropy": 1.9593722224235535,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1871783211827278,
      "step": 29590
    },
    {
      "epoch": 0.59184,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.007901763916015625,
      "learning_rate": 0.0001,
      "loss": 4.1708,
      "loss/crossentropy": 2.1829755902290344,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22405223548412323,
      "step": 29592
    },
    {
      "epoch": 0.59188,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.007330067952473958,
      "learning_rate": 0.0001,
      "loss": 3.9746,
      "loss/crossentropy": 1.9692201018333435,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18905997276306152,
      "step": 29594
    },
    {
      "epoch": 0.59192,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.008931223551432292,
      "learning_rate": 0.0001,
      "loss": 4.1957,
      "loss/crossentropy": 2.341002106666565,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21306651830673218,
      "step": 29596
    },
    {
      "epoch": 0.59196,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.010453287760416667,
      "learning_rate": 0.0001,
      "loss": 4.1617,
      "loss/crossentropy": 1.9179102778434753,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1874692291021347,
      "step": 29598
    },
    {
      "epoch": 0.592,
      "grad_norm": 2.328125,
      "grad_norm_var": 0.015421295166015625,
      "learning_rate": 0.0001,
      "loss": 4.036,
      "loss/crossentropy": 1.8973752856254578,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19447319954633713,
      "step": 29600
    },
    {
      "epoch": 0.59204,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.016501617431640626,
      "learning_rate": 0.0001,
      "loss": 3.8856,
      "loss/crossentropy": 1.8788208365440369,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16276411712169647,
      "step": 29602
    },
    {
      "epoch": 0.59208,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.01578547159830729,
      "learning_rate": 0.0001,
      "loss": 3.9585,
      "loss/crossentropy": 1.5552323460578918,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1751468926668167,
      "step": 29604
    },
    {
      "epoch": 0.59212,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.014273834228515626,
      "learning_rate": 0.0001,
      "loss": 4.0705,
      "loss/crossentropy": 1.906869649887085,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18109184503555298,
      "step": 29606
    },
    {
      "epoch": 0.59216,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.014764149983723959,
      "learning_rate": 0.0001,
      "loss": 4.1151,
      "loss/crossentropy": 2.23270046710968,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.23064442723989487,
      "step": 29608
    },
    {
      "epoch": 0.5922,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.01395263671875,
      "learning_rate": 0.0001,
      "loss": 4.0758,
      "loss/crossentropy": 2.187607169151306,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1993623673915863,
      "step": 29610
    },
    {
      "epoch": 0.59224,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.013425445556640625,
      "learning_rate": 0.0001,
      "loss": 4.1025,
      "loss/crossentropy": 2.1429349184036255,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19814343750476837,
      "step": 29612
    },
    {
      "epoch": 0.59228,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.014899698893229167,
      "learning_rate": 0.0001,
      "loss": 3.6302,
      "loss/crossentropy": 1.9003735780715942,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17693869769573212,
      "step": 29614
    },
    {
      "epoch": 0.59232,
      "grad_norm": 2.265625,
      "grad_norm_var": 0.0130523681640625,
      "learning_rate": 0.0001,
      "loss": 3.991,
      "loss/crossentropy": 2.2013859152793884,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19231925159692764,
      "step": 29616
    },
    {
      "epoch": 0.59236,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.01234130859375,
      "learning_rate": 0.0001,
      "loss": 3.9533,
      "loss/crossentropy": 1.3149051070213318,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1432316154241562,
      "step": 29618
    },
    {
      "epoch": 0.5924,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.013728841145833334,
      "learning_rate": 0.0001,
      "loss": 4.0781,
      "loss/crossentropy": 2.099441707134247,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19838576018810272,
      "step": 29620
    },
    {
      "epoch": 0.59244,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.015970865885416668,
      "learning_rate": 0.0001,
      "loss": 3.9236,
      "loss/crossentropy": 1.9029900431632996,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2029390037059784,
      "step": 29622
    },
    {
      "epoch": 0.59248,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.01672337849934896,
      "learning_rate": 0.0001,
      "loss": 4.0291,
      "loss/crossentropy": 2.159297227859497,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1991179883480072,
      "step": 29624
    },
    {
      "epoch": 0.59252,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.014964803059895834,
      "learning_rate": 0.0001,
      "loss": 4.1908,
      "loss/crossentropy": 2.0404341220855713,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1894593983888626,
      "step": 29626
    },
    {
      "epoch": 0.59256,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.012286122639973958,
      "learning_rate": 0.0001,
      "loss": 4.0989,
      "loss/crossentropy": 2.2824409008026123,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19680628180503845,
      "step": 29628
    },
    {
      "epoch": 0.5926,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.011905924479166666,
      "learning_rate": 0.0001,
      "loss": 4.131,
      "loss/crossentropy": 2.2231918573379517,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1871175393462181,
      "step": 29630
    },
    {
      "epoch": 0.59264,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.0030263264973958335,
      "learning_rate": 0.0001,
      "loss": 3.9817,
      "loss/crossentropy": 1.688015341758728,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1651618853211403,
      "step": 29632
    },
    {
      "epoch": 0.59268,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.003173828125,
      "learning_rate": 0.0001,
      "loss": 3.915,
      "loss/crossentropy": 1.8363903760910034,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1848619431257248,
      "step": 29634
    },
    {
      "epoch": 0.59272,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0029937744140625,
      "learning_rate": 0.0001,
      "loss": 3.7981,
      "loss/crossentropy": 2.2170976400375366,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.199843168258667,
      "step": 29636
    },
    {
      "epoch": 0.59276,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.0033770243326822916,
      "learning_rate": 0.0001,
      "loss": 4.0177,
      "loss/crossentropy": 1.707352876663208,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17033107578754425,
      "step": 29638
    },
    {
      "epoch": 0.5928,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.003295644124348958,
      "learning_rate": 0.0001,
      "loss": 4.1408,
      "loss/crossentropy": 2.418645143508911,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20665095001459122,
      "step": 29640
    },
    {
      "epoch": 0.59284,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.00450439453125,
      "learning_rate": 0.0001,
      "loss": 4.0686,
      "loss/crossentropy": 2.0012764930725098,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17807253450155258,
      "step": 29642
    },
    {
      "epoch": 0.59288,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.0054013570149739586,
      "learning_rate": 0.0001,
      "loss": 4.0358,
      "loss/crossentropy": 2.4282820224761963,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22419389337301254,
      "step": 29644
    },
    {
      "epoch": 0.59292,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.005669911702473958,
      "learning_rate": 0.0001,
      "loss": 3.9593,
      "loss/crossentropy": 1.9802210927009583,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18829302489757538,
      "step": 29646
    },
    {
      "epoch": 0.59296,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.004959869384765625,
      "learning_rate": 0.0001,
      "loss": 4.1733,
      "loss/crossentropy": 2.1611366271972656,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20142409205436707,
      "step": 29648
    },
    {
      "epoch": 0.593,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.006009674072265625,
      "learning_rate": 0.0001,
      "loss": 3.7293,
      "loss/crossentropy": 2.0067127346992493,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1892910748720169,
      "step": 29650
    },
    {
      "epoch": 0.59304,
      "grad_norm": 1.875,
      "grad_norm_var": 0.006591796875,
      "learning_rate": 0.0001,
      "loss": 3.9099,
      "loss/crossentropy": 1.8918231129646301,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17595022916793823,
      "step": 29652
    },
    {
      "epoch": 0.59308,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.00543212890625,
      "learning_rate": 0.0001,
      "loss": 4.0838,
      "loss/crossentropy": 2.310000419616699,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20750866830348969,
      "step": 29654
    },
    {
      "epoch": 0.59312,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.005448404947916667,
      "learning_rate": 0.0001,
      "loss": 3.7038,
      "loss/crossentropy": 2.0080875158309937,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18810434639453888,
      "step": 29656
    },
    {
      "epoch": 0.59316,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.0039670308430989586,
      "learning_rate": 0.0001,
      "loss": 3.8419,
      "loss/crossentropy": 1.9152463674545288,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17956481873989105,
      "step": 29658
    },
    {
      "epoch": 0.5932,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.002982330322265625,
      "learning_rate": 0.0001,
      "loss": 3.9465,
      "loss/crossentropy": 2.019734025001526,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1654059886932373,
      "step": 29660
    },
    {
      "epoch": 0.59324,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.003242746988932292,
      "learning_rate": 0.0001,
      "loss": 3.9672,
      "loss/crossentropy": 1.9415631294250488,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18466275930404663,
      "step": 29662
    },
    {
      "epoch": 0.59328,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.0037737528483072916,
      "learning_rate": 0.0001,
      "loss": 4.0275,
      "loss/crossentropy": 2.246294617652893,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20313822478055954,
      "step": 29664
    },
    {
      "epoch": 0.59332,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.005242665608723958,
      "learning_rate": 0.0001,
      "loss": 4.1661,
      "loss/crossentropy": 1.8673067688941956,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1858188509941101,
      "step": 29666
    },
    {
      "epoch": 0.59336,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.00509033203125,
      "learning_rate": 0.0001,
      "loss": 4.1556,
      "loss/crossentropy": 1.9696037769317627,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1936260536313057,
      "step": 29668
    },
    {
      "epoch": 0.5934,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.10892512003580729,
      "learning_rate": 0.0001,
      "loss": 4.1505,
      "loss/crossentropy": 2.2883976697921753,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22077778726816177,
      "step": 29670
    },
    {
      "epoch": 0.59344,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.1094378153483073,
      "learning_rate": 0.0001,
      "loss": 3.9617,
      "loss/crossentropy": 2.11427104473114,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1877901256084442,
      "step": 29672
    },
    {
      "epoch": 0.59348,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.1084307352701823,
      "learning_rate": 0.0001,
      "loss": 4.02,
      "loss/crossentropy": 2.256463050842285,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1980907917022705,
      "step": 29674
    },
    {
      "epoch": 0.59352,
      "grad_norm": 2.125,
      "grad_norm_var": 0.10796890258789063,
      "learning_rate": 0.0001,
      "loss": 3.926,
      "loss/crossentropy": 2.0554774403572083,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19413713365793228,
      "step": 29676
    },
    {
      "epoch": 0.59356,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.10651626586914062,
      "learning_rate": 0.0001,
      "loss": 3.9095,
      "loss/crossentropy": 2.0619908571243286,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20501630008220673,
      "step": 29678
    },
    {
      "epoch": 0.5936,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.10798314412434896,
      "learning_rate": 0.0001,
      "loss": 3.8964,
      "loss/crossentropy": 2.0001166462898254,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1859305128455162,
      "step": 29680
    },
    {
      "epoch": 0.59364,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.11613667805989583,
      "learning_rate": 0.0001,
      "loss": 3.7957,
      "loss/crossentropy": 2.0025678277015686,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18820206075906754,
      "step": 29682
    },
    {
      "epoch": 0.59368,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.11686604817708333,
      "learning_rate": 0.0001,
      "loss": 4.0391,
      "loss/crossentropy": 2.0470725297927856,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1986207813024521,
      "step": 29684
    },
    {
      "epoch": 0.59372,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010619862874348959,
      "learning_rate": 0.0001,
      "loss": 3.7557,
      "loss/crossentropy": 1.9196028113365173,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19124136865139008,
      "step": 29686
    },
    {
      "epoch": 0.59376,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.013315582275390625,
      "learning_rate": 0.0001,
      "loss": 3.6694,
      "loss/crossentropy": 1.624564528465271,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15498200803995132,
      "step": 29688
    },
    {
      "epoch": 0.5938,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.012060546875,
      "learning_rate": 0.0001,
      "loss": 4.0355,
      "loss/crossentropy": 2.147800922393799,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19418001919984818,
      "step": 29690
    },
    {
      "epoch": 0.59384,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.008719635009765626,
      "learning_rate": 0.0001,
      "loss": 4.0047,
      "loss/crossentropy": 2.1515848636627197,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17781265825033188,
      "step": 29692
    },
    {
      "epoch": 0.59388,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.0083740234375,
      "learning_rate": 0.0001,
      "loss": 3.9553,
      "loss/crossentropy": 2.034223735332489,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19139179587364197,
      "step": 29694
    },
    {
      "epoch": 0.59392,
      "grad_norm": 2.0,
      "grad_norm_var": 0.009226226806640625,
      "learning_rate": 0.0001,
      "loss": 3.9402,
      "loss/crossentropy": 1.8839016556739807,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1837429627776146,
      "step": 29696
    },
    {
      "epoch": 0.59396,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.006878407796223959,
      "learning_rate": 0.0001,
      "loss": 4.2001,
      "loss/crossentropy": 2.367957592010498,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21486172825098038,
      "step": 29698
    },
    {
      "epoch": 0.594,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.008062489827473958,
      "learning_rate": 0.0001,
      "loss": 3.8494,
      "loss/crossentropy": 2.02916818857193,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17551806569099426,
      "step": 29700
    },
    {
      "epoch": 0.59404,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.009384918212890624,
      "learning_rate": 0.0001,
      "loss": 4.1648,
      "loss/crossentropy": 1.8610569834709167,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17259076982736588,
      "step": 29702
    },
    {
      "epoch": 0.59408,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.007420857747395833,
      "learning_rate": 0.0001,
      "loss": 4.0775,
      "loss/crossentropy": 1.8655198812484741,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17619968950748444,
      "step": 29704
    },
    {
      "epoch": 0.59412,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.007775624593098958,
      "learning_rate": 0.0001,
      "loss": 4.0338,
      "loss/crossentropy": 1.7603416442871094,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.15166223049163818,
      "step": 29706
    },
    {
      "epoch": 0.59416,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.007726796468098958,
      "learning_rate": 0.0001,
      "loss": 3.8432,
      "loss/crossentropy": 1.9816042184829712,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1975991204380989,
      "step": 29708
    },
    {
      "epoch": 0.5942,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.0070574442545572914,
      "learning_rate": 0.0001,
      "loss": 4.1009,
      "loss/crossentropy": 2.2220958471298218,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20013637095689774,
      "step": 29710
    },
    {
      "epoch": 0.59424,
      "grad_norm": 2.203125,
      "grad_norm_var": 0.012719472249348959,
      "learning_rate": 0.0001,
      "loss": 4.0315,
      "loss/crossentropy": 1.947887122631073,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19075323641300201,
      "step": 29712
    },
    {
      "epoch": 0.59428,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.014286295572916666,
      "learning_rate": 0.0001,
      "loss": 4.0852,
      "loss/crossentropy": 2.3358702659606934,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21242079883813858,
      "step": 29714
    },
    {
      "epoch": 0.59432,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.01927464803059896,
      "learning_rate": 0.0001,
      "loss": 4.1379,
      "loss/crossentropy": 1.746841311454773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1843656674027443,
      "step": 29716
    },
    {
      "epoch": 0.59436,
      "grad_norm": 1.875,
      "grad_norm_var": 0.0187255859375,
      "learning_rate": 0.0001,
      "loss": 4.1713,
      "loss/crossentropy": 2.283482313156128,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2116861194372177,
      "step": 29718
    },
    {
      "epoch": 0.5944,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.016927083333333332,
      "learning_rate": 0.0001,
      "loss": 4.0167,
      "loss/crossentropy": 1.927943468093872,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18120041489601135,
      "step": 29720
    },
    {
      "epoch": 0.59444,
      "grad_norm": 1.8046875,
      "grad_norm_var": 0.021219889322916668,
      "learning_rate": 0.0001,
      "loss": 3.6467,
      "loss/crossentropy": 1.8562633395195007,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1756514087319374,
      "step": 29722
    },
    {
      "epoch": 0.59448,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.021862538655598958,
      "learning_rate": 0.0001,
      "loss": 3.7351,
      "loss/crossentropy": 2.0243565440177917,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1839814931154251,
      "step": 29724
    },
    {
      "epoch": 0.59452,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.023227691650390625,
      "learning_rate": 0.0001,
      "loss": 4.0603,
      "loss/crossentropy": 2.136290669441223,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19461742788553238,
      "step": 29726
    },
    {
      "epoch": 0.59456,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.017308553059895832,
      "learning_rate": 0.0001,
      "loss": 4.1118,
      "loss/crossentropy": 2.1178980469703674,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19608290493488312,
      "step": 29728
    },
    {
      "epoch": 0.5946,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.016209920247395832,
      "learning_rate": 0.0001,
      "loss": 4.1603,
      "loss/crossentropy": 2.1932308673858643,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19451629370450974,
      "step": 29730
    },
    {
      "epoch": 0.59464,
      "grad_norm": 2.21875,
      "grad_norm_var": 0.014890289306640625,
      "learning_rate": 0.0001,
      "loss": 4.1111,
      "loss/crossentropy": 2.140673279762268,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19651110470294952,
      "step": 29732
    },
    {
      "epoch": 0.59468,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.013423411051432292,
      "learning_rate": 0.0001,
      "loss": 4.0438,
      "loss/crossentropy": 2.174296498298645,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2079356089234352,
      "step": 29734
    },
    {
      "epoch": 0.59472,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.015730539957682293,
      "learning_rate": 0.0001,
      "loss": 4.0483,
      "loss/crossentropy": 1.7323570847511292,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18022552132606506,
      "step": 29736
    },
    {
      "epoch": 0.59476,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.011229451497395833,
      "learning_rate": 0.0001,
      "loss": 3.9672,
      "loss/crossentropy": 2.0798590779304504,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19130367040634155,
      "step": 29738
    },
    {
      "epoch": 0.5948,
      "grad_norm": 1.9921875,
      "grad_norm_var": 0.009837849934895834,
      "learning_rate": 0.0001,
      "loss": 3.9728,
      "loss/crossentropy": 2.1414352655410767,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19515492022037506,
      "step": 29740
    },
    {
      "epoch": 0.59484,
      "grad_norm": 1.875,
      "grad_norm_var": 0.01082763671875,
      "learning_rate": 0.0001,
      "loss": 3.8085,
      "loss/crossentropy": 2.0827420949935913,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1836787909269333,
      "step": 29742
    },
    {
      "epoch": 0.59488,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.010960896809895834,
      "learning_rate": 0.0001,
      "loss": 4.1457,
      "loss/crossentropy": 2.2884727716445923,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20683631300926208,
      "step": 29744
    },
    {
      "epoch": 0.59492,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.010838826497395834,
      "learning_rate": 0.0001,
      "loss": 3.9837,
      "loss/crossentropy": 1.6755734086036682,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17120809853076935,
      "step": 29746
    },
    {
      "epoch": 0.59496,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.0085205078125,
      "learning_rate": 0.0001,
      "loss": 4.1012,
      "loss/crossentropy": 1.8576670289039612,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19007987529039383,
      "step": 29748
    },
    {
      "epoch": 0.595,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.009227498372395834,
      "learning_rate": 0.0001,
      "loss": 4.0576,
      "loss/crossentropy": 1.9877051711082458,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18861526250839233,
      "step": 29750
    },
    {
      "epoch": 0.59504,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0072591145833333336,
      "learning_rate": 0.0001,
      "loss": 3.9335,
      "loss/crossentropy": 2.4191300868988037,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2063479870557785,
      "step": 29752
    },
    {
      "epoch": 0.59508,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.007279459635416667,
      "learning_rate": 0.0001,
      "loss": 4.1469,
      "loss/crossentropy": 2.145687997341156,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19573777168989182,
      "step": 29754
    },
    {
      "epoch": 0.59512,
      "grad_norm": 1.6640625,
      "grad_norm_var": 0.010823313395182292,
      "learning_rate": 0.0001,
      "loss": 3.7671,
      "loss/crossentropy": 1.956217110157013,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17622292786836624,
      "step": 29756
    },
    {
      "epoch": 0.59516,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.013716634114583333,
      "learning_rate": 0.0001,
      "loss": 3.976,
      "loss/crossentropy": 2.0639008283615112,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18797758221626282,
      "step": 29758
    },
    {
      "epoch": 0.5952,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.013073476155598958,
      "learning_rate": 0.0001,
      "loss": 3.9458,
      "loss/crossentropy": 2.0404372811317444,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18267738819122314,
      "step": 29760
    },
    {
      "epoch": 0.59524,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.015543619791666666,
      "learning_rate": 0.0001,
      "loss": 3.9625,
      "loss/crossentropy": 2.0870869159698486,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18522879481315613,
      "step": 29762
    },
    {
      "epoch": 0.59528,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.013602701822916667,
      "learning_rate": 0.0001,
      "loss": 4.0244,
      "loss/crossentropy": 2.0854042172431946,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.172662615776062,
      "step": 29764
    },
    {
      "epoch": 0.59532,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.014050038655598958,
      "learning_rate": 0.0001,
      "loss": 3.9533,
      "loss/crossentropy": 2.012739658355713,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20193946361541748,
      "step": 29766
    },
    {
      "epoch": 0.59536,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.013834635416666666,
      "learning_rate": 0.0001,
      "loss": 3.7903,
      "loss/crossentropy": 1.8651488423347473,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19326791167259216,
      "step": 29768
    },
    {
      "epoch": 0.5954,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.015258534749348959,
      "learning_rate": 0.0001,
      "loss": 3.9921,
      "loss/crossentropy": 1.8395410776138306,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17881877720355988,
      "step": 29770
    },
    {
      "epoch": 0.59544,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.010518137613932292,
      "learning_rate": 0.0001,
      "loss": 3.7816,
      "loss/crossentropy": 1.7089250087738037,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19955646991729736,
      "step": 29772
    },
    {
      "epoch": 0.59548,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.007122548421223959,
      "learning_rate": 0.0001,
      "loss": 4.0978,
      "loss/crossentropy": 2.233728766441345,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20136447995901108,
      "step": 29774
    },
    {
      "epoch": 0.59552,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.008324178059895833,
      "learning_rate": 0.0001,
      "loss": 4.0012,
      "loss/crossentropy": 2.104355573654175,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19211959093809128,
      "step": 29776
    },
    {
      "epoch": 0.59556,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0061075846354166664,
      "learning_rate": 0.0001,
      "loss": 3.931,
      "loss/crossentropy": 1.9304287433624268,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1887596696615219,
      "step": 29778
    },
    {
      "epoch": 0.5956,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.004878489176432291,
      "learning_rate": 0.0001,
      "loss": 3.9997,
      "loss/crossentropy": 1.9446918368339539,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17917344719171524,
      "step": 29780
    },
    {
      "epoch": 0.59564,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.004196929931640625,
      "learning_rate": 0.0001,
      "loss": 4.0013,
      "loss/crossentropy": 2.2256661653518677,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21204601228237152,
      "step": 29782
    },
    {
      "epoch": 0.59568,
      "grad_norm": 2.109375,
      "grad_norm_var": 0.0061431884765625,
      "learning_rate": 0.0001,
      "loss": 4.0934,
      "loss/crossentropy": 2.1072696447372437,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2179524451494217,
      "step": 29784
    },
    {
      "epoch": 0.59572,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.0048411051432291664,
      "learning_rate": 0.0001,
      "loss": 4.0207,
      "loss/crossentropy": 2.2444673776626587,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19245705753564835,
      "step": 29786
    },
    {
      "epoch": 0.59576,
      "grad_norm": 2.09375,
      "grad_norm_var": 0.006833648681640625,
      "learning_rate": 0.0001,
      "loss": 4.2724,
      "loss/crossentropy": 2.5239516496658325,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22635143250226974,
      "step": 29788
    },
    {
      "epoch": 0.5958,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.006883748372395833,
      "learning_rate": 0.0001,
      "loss": 3.9709,
      "loss/crossentropy": 1.7270027995109558,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18143896013498306,
      "step": 29790
    },
    {
      "epoch": 0.59584,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.006170399983723958,
      "learning_rate": 0.0001,
      "loss": 3.9258,
      "loss/crossentropy": 1.9974133372306824,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19476915150880814,
      "step": 29792
    },
    {
      "epoch": 0.59588,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.006601715087890625,
      "learning_rate": 0.0001,
      "loss": 3.999,
      "loss/crossentropy": 1.9523651003837585,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17940453439950943,
      "step": 29794
    },
    {
      "epoch": 0.59592,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.0069580078125,
      "learning_rate": 0.0001,
      "loss": 3.7075,
      "loss/crossentropy": 1.8218488097190857,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17136383056640625,
      "step": 29796
    },
    {
      "epoch": 0.59596,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.007098134358723958,
      "learning_rate": 0.0001,
      "loss": 4.0404,
      "loss/crossentropy": 2.1148566603660583,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21685151010751724,
      "step": 29798
    },
    {
      "epoch": 0.596,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.004833984375,
      "learning_rate": 0.0001,
      "loss": 4.0628,
      "loss/crossentropy": 2.4410229921340942,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20404956489801407,
      "step": 29800
    },
    {
      "epoch": 0.59604,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.005378214518229166,
      "learning_rate": 0.0001,
      "loss": 3.8664,
      "loss/crossentropy": 1.9751408100128174,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18364927917718887,
      "step": 29802
    },
    {
      "epoch": 0.59608,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.0031402587890625,
      "learning_rate": 0.0001,
      "loss": 3.861,
      "loss/crossentropy": 2.2535789012908936,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2096666395664215,
      "step": 29804
    },
    {
      "epoch": 0.59612,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.004142252604166666,
      "learning_rate": 0.0001,
      "loss": 3.817,
      "loss/crossentropy": 2.2020750641822815,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21073434501886368,
      "step": 29806
    },
    {
      "epoch": 0.59616,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.0040934244791666664,
      "learning_rate": 0.0001,
      "loss": 4.0591,
      "loss/crossentropy": 2.0761327147483826,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1929018497467041,
      "step": 29808
    },
    {
      "epoch": 0.5962,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.004064687093098958,
      "learning_rate": 0.0001,
      "loss": 4.1519,
      "loss/crossentropy": 2.014641046524048,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1975310891866684,
      "step": 29810
    },
    {
      "epoch": 0.59624,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.005182902018229167,
      "learning_rate": 0.0001,
      "loss": 3.8542,
      "loss/crossentropy": 1.9420717358589172,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1935308501124382,
      "step": 29812
    },
    {
      "epoch": 0.59628,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.005399576822916667,
      "learning_rate": 0.0001,
      "loss": 3.829,
      "loss/crossentropy": 1.7394835352897644,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16872025281190872,
      "step": 29814
    },
    {
      "epoch": 0.59632,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.008980305989583333,
      "learning_rate": 0.0001,
      "loss": 4.012,
      "loss/crossentropy": 2.0532928705215454,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20437174290418625,
      "step": 29816
    },
    {
      "epoch": 0.59636,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.008292388916015626,
      "learning_rate": 0.0001,
      "loss": 3.8792,
      "loss/crossentropy": 1.7072120904922485,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16668207198381424,
      "step": 29818
    },
    {
      "epoch": 0.5964,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.008063761393229167,
      "learning_rate": 0.0001,
      "loss": 4.2173,
      "loss/crossentropy": 2.362375855445862,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20404937118291855,
      "step": 29820
    },
    {
      "epoch": 0.59644,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.006894683837890625,
      "learning_rate": 0.0001,
      "loss": 3.9963,
      "loss/crossentropy": 2.334542155265808,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21750033646821976,
      "step": 29822
    },
    {
      "epoch": 0.59648,
      "grad_norm": 1.9765625,
      "grad_norm_var": 0.006463368733723958,
      "learning_rate": 0.0001,
      "loss": 4.0611,
      "loss/crossentropy": 2.1138614416122437,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2092488631606102,
      "step": 29824
    },
    {
      "epoch": 0.59652,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.006184895833333333,
      "learning_rate": 0.0001,
      "loss": 3.8785,
      "loss/crossentropy": 1.7644943594932556,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18817196786403656,
      "step": 29826
    },
    {
      "epoch": 0.59656,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.005407460530598958,
      "learning_rate": 0.0001,
      "loss": 4.1779,
      "loss/crossentropy": 2.1820908784866333,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20322736352682114,
      "step": 29828
    },
    {
      "epoch": 0.5966,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.0041656494140625,
      "learning_rate": 0.0001,
      "loss": 3.8968,
      "loss/crossentropy": 2.146401524543762,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19272416085004807,
      "step": 29830
    },
    {
      "epoch": 0.59664,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.0023943583170572915,
      "learning_rate": 0.0001,
      "loss": 3.9237,
      "loss/crossentropy": 1.8045800924301147,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1904245838522911,
      "step": 29832
    },
    {
      "epoch": 0.59668,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.00426025390625,
      "learning_rate": 0.0001,
      "loss": 3.9487,
      "loss/crossentropy": 2.107872247695923,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18807457387447357,
      "step": 29834
    },
    {
      "epoch": 0.59672,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.006048329671223958,
      "learning_rate": 0.0001,
      "loss": 3.9153,
      "loss/crossentropy": 2.2815098762512207,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2112163081765175,
      "step": 29836
    },
    {
      "epoch": 0.59676,
      "grad_norm": 2.0,
      "grad_norm_var": 0.006145985921223959,
      "learning_rate": 0.0001,
      "loss": 4.1667,
      "loss/crossentropy": 2.242028594017029,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2006349116563797,
      "step": 29838
    },
    {
      "epoch": 0.5968,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.006029256184895833,
      "learning_rate": 0.0001,
      "loss": 3.9325,
      "loss/crossentropy": 2.2297927141189575,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2074139192700386,
      "step": 29840
    },
    {
      "epoch": 0.59684,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.007071940104166666,
      "learning_rate": 0.0001,
      "loss": 4.1403,
      "loss/crossentropy": 2.2018691301345825,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21267356723546982,
      "step": 29842
    },
    {
      "epoch": 0.59688,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.007819620768229167,
      "learning_rate": 0.0001,
      "loss": 3.8575,
      "loss/crossentropy": 2.0878920555114746,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18239793926477432,
      "step": 29844
    },
    {
      "epoch": 0.59692,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.007917277018229167,
      "learning_rate": 0.0001,
      "loss": 4.0925,
      "loss/crossentropy": 2.0679028034210205,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20400486141443253,
      "step": 29846
    },
    {
      "epoch": 0.59696,
      "grad_norm": 1.6328125,
      "grad_norm_var": 0.010227203369140625,
      "learning_rate": 0.0001,
      "loss": 3.781,
      "loss/crossentropy": 2.0496456623077393,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2035546898841858,
      "step": 29848
    },
    {
      "epoch": 0.597,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.009325917561848958,
      "learning_rate": 0.0001,
      "loss": 4.0158,
      "loss/crossentropy": 2.0204665660858154,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1830250397324562,
      "step": 29850
    },
    {
      "epoch": 0.59704,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.009325154622395833,
      "learning_rate": 0.0001,
      "loss": 3.6884,
      "loss/crossentropy": 2.0521149039268494,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18589533865451813,
      "step": 29852
    },
    {
      "epoch": 0.59708,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.009506988525390624,
      "learning_rate": 0.0001,
      "loss": 3.6348,
      "loss/crossentropy": 1.8169047832489014,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17271429300308228,
      "step": 29854
    },
    {
      "epoch": 0.59712,
      "grad_norm": 1.84375,
      "grad_norm_var": 0.008514149983723959,
      "learning_rate": 0.0001,
      "loss": 4.0186,
      "loss/crossentropy": 2.029230833053589,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18239567428827286,
      "step": 29856
    },
    {
      "epoch": 0.59716,
      "grad_norm": 1.71875,
      "grad_norm_var": 0.0059893290201822914,
      "learning_rate": 0.0001,
      "loss": 3.97,
      "loss/crossentropy": 2.0296765565872192,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1822112277150154,
      "step": 29858
    },
    {
      "epoch": 0.5972,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.00657958984375,
      "learning_rate": 0.0001,
      "loss": 3.9411,
      "loss/crossentropy": 1.9933258891105652,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1953386887907982,
      "step": 29860
    },
    {
      "epoch": 0.59724,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.014229329427083333,
      "learning_rate": 0.0001,
      "loss": 3.9819,
      "loss/crossentropy": 1.9453927278518677,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19596701860427856,
      "step": 29862
    },
    {
      "epoch": 0.59728,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.011185455322265624,
      "learning_rate": 0.0001,
      "loss": 3.8056,
      "loss/crossentropy": 1.714016318321228,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17374874651432037,
      "step": 29864
    },
    {
      "epoch": 0.59732,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.01087646484375,
      "learning_rate": 0.0001,
      "loss": 3.6872,
      "loss/crossentropy": 2.0186676383018494,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19331404566764832,
      "step": 29866
    },
    {
      "epoch": 0.59736,
      "grad_norm": 1.71875,
      "grad_norm_var": 0.011922200520833334,
      "learning_rate": 0.0001,
      "loss": 3.8642,
      "loss/crossentropy": 1.9068130850791931,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18482722342014313,
      "step": 29868
    },
    {
      "epoch": 0.5974,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.014092763264973959,
      "learning_rate": 0.0001,
      "loss": 4.0994,
      "loss/crossentropy": 2.192137658596039,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19123639166355133,
      "step": 29870
    },
    {
      "epoch": 0.59744,
      "grad_norm": 1.7734375,
      "grad_norm_var": 0.014788564046223958,
      "learning_rate": 0.0001,
      "loss": 3.5734,
      "loss/crossentropy": 1.859050452709198,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19843916594982147,
      "step": 29872
    },
    {
      "epoch": 0.59748,
      "grad_norm": 2.28125,
      "grad_norm_var": 0.021469879150390624,
      "learning_rate": 0.0001,
      "loss": 4.1619,
      "loss/crossentropy": 1.5558834075927734,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17953873425722122,
      "step": 29874
    },
    {
      "epoch": 0.59752,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.021274566650390625,
      "learning_rate": 0.0001,
      "loss": 4.2012,
      "loss/crossentropy": 2.1445621252059937,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20609040558338165,
      "step": 29876
    },
    {
      "epoch": 0.59756,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.01730931599934896,
      "learning_rate": 0.0001,
      "loss": 4.0528,
      "loss/crossentropy": 1.8768232464790344,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16527344286441803,
      "step": 29878
    },
    {
      "epoch": 0.5976,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.018024698893229166,
      "learning_rate": 0.0001,
      "loss": 3.8604,
      "loss/crossentropy": 1.830302894115448,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17598530650138855,
      "step": 29880
    },
    {
      "epoch": 0.59764,
      "grad_norm": 1.875,
      "grad_norm_var": 0.018115234375,
      "learning_rate": 0.0001,
      "loss": 3.9474,
      "loss/crossentropy": 2.0607933402061462,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18962115049362183,
      "step": 29882
    },
    {
      "epoch": 0.59768,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.015657552083333335,
      "learning_rate": 0.0001,
      "loss": 3.8516,
      "loss/crossentropy": 2.2717671394348145,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21351712197065353,
      "step": 29884
    },
    {
      "epoch": 0.59772,
      "grad_norm": 1.765625,
      "grad_norm_var": 0.014383951822916666,
      "learning_rate": 0.0001,
      "loss": 3.8209,
      "loss/crossentropy": 2.0096693634986877,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1812792867422104,
      "step": 29886
    },
    {
      "epoch": 0.59776,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.0130035400390625,
      "learning_rate": 0.0001,
      "loss": 3.8955,
      "loss/crossentropy": 1.9364767670631409,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19597072899341583,
      "step": 29888
    },
    {
      "epoch": 0.5978,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.003348541259765625,
      "learning_rate": 0.0001,
      "loss": 4.1403,
      "loss/crossentropy": 2.1214417219161987,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1933349445462227,
      "step": 29890
    },
    {
      "epoch": 0.59784,
      "grad_norm": 1.8984375,
      "grad_norm_var": 0.007013956705729167,
      "learning_rate": 0.0001,
      "loss": 4.2015,
      "loss/crossentropy": 1.9339659214019775,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1958373337984085,
      "step": 29892
    },
    {
      "epoch": 0.59788,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.009913889567057292,
      "learning_rate": 0.0001,
      "loss": 4.3786,
      "loss/crossentropy": 1.940473735332489,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18802516907453537,
      "step": 29894
    },
    {
      "epoch": 0.59792,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.009230295817057291,
      "learning_rate": 0.0001,
      "loss": 3.8581,
      "loss/crossentropy": 2.048341751098633,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18541985005140305,
      "step": 29896
    },
    {
      "epoch": 0.59796,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.008780924479166667,
      "learning_rate": 0.0001,
      "loss": 4.1022,
      "loss/crossentropy": 1.9213348031044006,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17794149369001389,
      "step": 29898
    },
    {
      "epoch": 0.598,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.008807118733723958,
      "learning_rate": 0.0001,
      "loss": 4.0137,
      "loss/crossentropy": 2.160265564918518,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19693263620138168,
      "step": 29900
    },
    {
      "epoch": 0.59804,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.007718658447265625,
      "learning_rate": 0.0001,
      "loss": 4.1329,
      "loss/crossentropy": 2.3574010133743286,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20197532325983047,
      "step": 29902
    },
    {
      "epoch": 0.59808,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.007505035400390625,
      "learning_rate": 0.0001,
      "loss": 4.0925,
      "loss/crossentropy": 2.2992867827415466,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21149636805057526,
      "step": 29904
    },
    {
      "epoch": 0.59812,
      "grad_norm": 2.015625,
      "grad_norm_var": 0.007328033447265625,
      "learning_rate": 0.0001,
      "loss": 4.0631,
      "loss/crossentropy": 2.1905906200408936,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21494387090206146,
      "step": 29906
    },
    {
      "epoch": 0.59816,
      "grad_norm": 1.9140625,
      "grad_norm_var": 0.007808176676432291,
      "learning_rate": 0.0001,
      "loss": 3.6926,
      "loss/crossentropy": 2.0027430057525635,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19078665226697922,
      "step": 29908
    },
    {
      "epoch": 0.5982,
      "grad_norm": 2.046875,
      "grad_norm_var": 0.007346343994140625,
      "learning_rate": 0.0001,
      "loss": 4.0699,
      "loss/crossentropy": 2.5675058364868164,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21340946853160858,
      "step": 29910
    },
    {
      "epoch": 0.59824,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.007746378580729167,
      "learning_rate": 0.0001,
      "loss": 4.0047,
      "loss/crossentropy": 1.6819973587989807,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17643968760967255,
      "step": 29912
    },
    {
      "epoch": 0.59828,
      "grad_norm": 1.953125,
      "grad_norm_var": 0.008499908447265624,
      "learning_rate": 0.0001,
      "loss": 4.0427,
      "loss/crossentropy": 1.9997480511665344,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1763923391699791,
      "step": 29914
    },
    {
      "epoch": 0.59832,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.009580230712890625,
      "learning_rate": 0.0001,
      "loss": 3.9184,
      "loss/crossentropy": 2.02975070476532,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19244793057441711,
      "step": 29916
    },
    {
      "epoch": 0.59836,
      "grad_norm": 1.984375,
      "grad_norm_var": 0.009405263264973958,
      "learning_rate": 0.0001,
      "loss": 3.903,
      "loss/crossentropy": 1.9599812030792236,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20076382905244827,
      "step": 29918
    },
    {
      "epoch": 0.5984,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.010119374593098958,
      "learning_rate": 0.0001,
      "loss": 4.0101,
      "loss/crossentropy": 2.106861114501953,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18403494358062744,
      "step": 29920
    },
    {
      "epoch": 0.59844,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.009830474853515625,
      "learning_rate": 0.0001,
      "loss": 3.9163,
      "loss/crossentropy": 2.0991535782814026,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18648552894592285,
      "step": 29922
    },
    {
      "epoch": 0.59848,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.0095123291015625,
      "learning_rate": 0.0001,
      "loss": 4.0917,
      "loss/crossentropy": 2.205238461494446,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.192887581884861,
      "step": 29924
    },
    {
      "epoch": 0.59852,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.006807200113932292,
      "learning_rate": 0.0001,
      "loss": 4.1419,
      "loss/crossentropy": 1.8452889919281006,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17111057043075562,
      "step": 29926
    },
    {
      "epoch": 0.59856,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.0066640218098958336,
      "learning_rate": 0.0001,
      "loss": 3.8361,
      "loss/crossentropy": 2.1874170303344727,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20274078845977783,
      "step": 29928
    },
    {
      "epoch": 0.5986,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.009671783447265625,
      "learning_rate": 0.0001,
      "loss": 4.0676,
      "loss/crossentropy": 2.0940569639205933,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19653251022100449,
      "step": 29930
    },
    {
      "epoch": 0.59864,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.009440104166666666,
      "learning_rate": 0.0001,
      "loss": 4.0597,
      "loss/crossentropy": 2.1147854328155518,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19174247980117798,
      "step": 29932
    },
    {
      "epoch": 0.59868,
      "grad_norm": 1.8125,
      "grad_norm_var": 0.009279123942057292,
      "learning_rate": 0.0001,
      "loss": 3.8874,
      "loss/crossentropy": 1.9983789324760437,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19001173973083496,
      "step": 29934
    },
    {
      "epoch": 0.59872,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.007264963785807292,
      "learning_rate": 0.0001,
      "loss": 3.8884,
      "loss/crossentropy": 2.0145105123519897,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1948821321129799,
      "step": 29936
    },
    {
      "epoch": 0.59876,
      "grad_norm": 1.7578125,
      "grad_norm_var": 0.0087890625,
      "learning_rate": 0.0001,
      "loss": 3.9149,
      "loss/crossentropy": 2.0878870487213135,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19328270852565765,
      "step": 29938
    },
    {
      "epoch": 0.5988,
      "grad_norm": 2.0,
      "grad_norm_var": 0.008177693684895833,
      "learning_rate": 0.0001,
      "loss": 3.8649,
      "loss/crossentropy": 2.0433109998703003,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1775122806429863,
      "step": 29940
    },
    {
      "epoch": 0.59884,
      "grad_norm": 1.8671875,
      "grad_norm_var": 0.016060384114583333,
      "learning_rate": 0.0001,
      "loss": 4.339,
      "loss/crossentropy": 2.45761239528656,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20690447837114334,
      "step": 29942
    },
    {
      "epoch": 0.59888,
      "grad_norm": 2.171875,
      "grad_norm_var": 0.019052886962890626,
      "learning_rate": 0.0001,
      "loss": 4.2211,
      "loss/crossentropy": 2.149196147918701,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.22804293036460876,
      "step": 29944
    },
    {
      "epoch": 0.59892,
      "grad_norm": 1.8203125,
      "grad_norm_var": 0.0178131103515625,
      "learning_rate": 0.0001,
      "loss": 3.668,
      "loss/crossentropy": 2.208665132522583,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1886969953775406,
      "step": 29946
    },
    {
      "epoch": 0.59896,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.017479451497395833,
      "learning_rate": 0.0001,
      "loss": 3.9187,
      "loss/crossentropy": 2.1634607315063477,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2016885057091713,
      "step": 29948
    },
    {
      "epoch": 0.599,
      "grad_norm": 1.7890625,
      "grad_norm_var": 0.018651326497395832,
      "learning_rate": 0.0001,
      "loss": 4.0225,
      "loss/crossentropy": 2.289479374885559,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.209372416138649,
      "step": 29950
    },
    {
      "epoch": 0.59904,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.019535064697265625,
      "learning_rate": 0.0001,
      "loss": 3.9492,
      "loss/crossentropy": 2.1859498023986816,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2030099779367447,
      "step": 29952
    },
    {
      "epoch": 0.59908,
      "grad_norm": 1.8515625,
      "grad_norm_var": 0.017496744791666668,
      "learning_rate": 0.0001,
      "loss": 3.8509,
      "loss/crossentropy": 2.1713619232177734,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18735715746879578,
      "step": 29954
    },
    {
      "epoch": 0.59912,
      "grad_norm": 1.9609375,
      "grad_norm_var": 0.0174468994140625,
      "learning_rate": 0.0001,
      "loss": 4.1315,
      "loss/crossentropy": 2.0735312700271606,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20961667597293854,
      "step": 29956
    },
    {
      "epoch": 0.59916,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.009969075520833334,
      "learning_rate": 0.0001,
      "loss": 4.0316,
      "loss/crossentropy": 1.6638267636299133,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16829392313957214,
      "step": 29958
    },
    {
      "epoch": 0.5992,
      "grad_norm": 1.8828125,
      "grad_norm_var": 0.005635325113932292,
      "learning_rate": 0.0001,
      "loss": 4.0428,
      "loss/crossentropy": 1.924504578113556,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1855188086628914,
      "step": 29960
    },
    {
      "epoch": 0.59924,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.0067942301432291664,
      "learning_rate": 0.0001,
      "loss": 4.0467,
      "loss/crossentropy": 2.2729889154434204,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21448159217834473,
      "step": 29962
    },
    {
      "epoch": 0.59928,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.007609049479166667,
      "learning_rate": 0.0001,
      "loss": 3.8208,
      "loss/crossentropy": 1.9595959186553955,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21665186434984207,
      "step": 29964
    },
    {
      "epoch": 0.59932,
      "grad_norm": 2.0625,
      "grad_norm_var": 0.007002512613932292,
      "learning_rate": 0.0001,
      "loss": 3.9934,
      "loss/crossentropy": 2.208333909511566,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20341281592845917,
      "step": 29966
    },
    {
      "epoch": 0.59936,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.00604248046875,
      "learning_rate": 0.0001,
      "loss": 3.9484,
      "loss/crossentropy": 2.0344181060791016,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19173699617385864,
      "step": 29968
    },
    {
      "epoch": 0.5994,
      "grad_norm": 1.7421875,
      "grad_norm_var": 0.008063761393229167,
      "learning_rate": 0.0001,
      "loss": 4.0233,
      "loss/crossentropy": 1.9575786590576172,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16551142930984497,
      "step": 29970
    },
    {
      "epoch": 0.59944,
      "grad_norm": 1.828125,
      "grad_norm_var": 0.008174387613932292,
      "learning_rate": 0.0001,
      "loss": 3.7948,
      "loss/crossentropy": 1.7442238330841064,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.16575023531913757,
      "step": 29972
    },
    {
      "epoch": 0.59948,
      "grad_norm": 1.8359375,
      "grad_norm_var": 0.00819091796875,
      "learning_rate": 0.0001,
      "loss": 3.9174,
      "loss/crossentropy": 2.3819799423217773,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21309742331504822,
      "step": 29974
    },
    {
      "epoch": 0.59952,
      "grad_norm": 1.9296875,
      "grad_norm_var": 0.009165191650390625,
      "learning_rate": 0.0001,
      "loss": 3.8633,
      "loss/crossentropy": 2.097830832004547,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17393425852060318,
      "step": 29976
    },
    {
      "epoch": 0.59956,
      "grad_norm": 1.9375,
      "grad_norm_var": 0.007382965087890625,
      "learning_rate": 0.0001,
      "loss": 3.7568,
      "loss/crossentropy": 1.771178424358368,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18638163059949875,
      "step": 29978
    },
    {
      "epoch": 0.5996,
      "grad_norm": 1.96875,
      "grad_norm_var": 0.007289377848307291,
      "learning_rate": 0.0001,
      "loss": 4.2791,
      "loss/crossentropy": 2.419429302215576,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20359767973423004,
      "step": 29980
    },
    {
      "epoch": 0.59964,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.005881500244140625,
      "learning_rate": 0.0001,
      "loss": 3.7571,
      "loss/crossentropy": 2.0326388478279114,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20969726145267487,
      "step": 29982
    },
    {
      "epoch": 0.59968,
      "grad_norm": 1.9453125,
      "grad_norm_var": 0.005983225504557292,
      "learning_rate": 0.0001,
      "loss": 4.0842,
      "loss/crossentropy": 1.9649672508239746,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18691973388195038,
      "step": 29984
    },
    {
      "epoch": 0.59972,
      "grad_norm": 2.03125,
      "grad_norm_var": 0.00643310546875,
      "learning_rate": 0.0001,
      "loss": 3.8939,
      "loss/crossentropy": 1.9001884460449219,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.18768402934074402,
      "step": 29986
    },
    {
      "epoch": 0.59976,
      "grad_norm": 1.859375,
      "grad_norm_var": 0.006208292643229167,
      "learning_rate": 0.0001,
      "loss": 3.8919,
      "loss/crossentropy": 2.0225003361701965,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.2066223919391632,
      "step": 29988
    },
    {
      "epoch": 0.5998,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.006451161702473959,
      "learning_rate": 0.0001,
      "loss": 3.9285,
      "loss/crossentropy": 2.175786852836609,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.20807426422834396,
      "step": 29990
    },
    {
      "epoch": 0.59984,
      "grad_norm": 1.71875,
      "grad_norm_var": 0.007228342692057291,
      "learning_rate": 0.0001,
      "loss": 3.7598,
      "loss/crossentropy": 1.842478632926941,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.17608553171157837,
      "step": 29992
    },
    {
      "epoch": 0.59988,
      "grad_norm": 1.796875,
      "grad_norm_var": 0.007746378580729167,
      "learning_rate": 0.0001,
      "loss": 3.8669,
      "loss/crossentropy": 2.1033515334129333,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.19302091002464294,
      "step": 29994
    },
    {
      "epoch": 0.59992,
      "grad_norm": 1.921875,
      "grad_norm_var": 0.006990559895833333,
      "learning_rate": 0.0001,
      "loss": 4.0286,
      "loss/crossentropy": 2.3058249950408936,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1858331635594368,
      "step": 29996
    },
    {
      "epoch": 0.59996,
      "grad_norm": 1.890625,
      "grad_norm_var": 0.006200917561848958,
      "learning_rate": 0.0001,
      "loss": 4.1295,
      "loss/crossentropy": 2.3059096336364746,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.21149063110351562,
      "step": 29998
    },
    {
      "epoch": 0.6,
      "grad_norm": 1.90625,
      "grad_norm_var": 0.006029256184895833,
      "learning_rate": 0.0001,
      "loss": 3.9016,
      "loss/crossentropy": 2.0354926586151123,
      "loss/hidden": 0.0,
      "loss/jsd": 0.0,
      "loss/logits": 0.1987614408135414,
      "step": 30000
    }
  ],
  "logging_steps": 2,
  "max_steps": 50000,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 9223372036854775807,
  "save_steps": 10000,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": false
      },
      "attributes": {}
    }
  },
  "total_flos": 3.12442965983232e+19,
  "train_batch_size": 8,
  "trial_name": null,
  "trial_params": null
}