{
  "best_metric": 0.07106676697731018,
  "best_model_checkpoint": "vit_epochs5_batch32_lr5e-05_size224_tiles4_seed3_q1\\checkpoint-469",
  "epoch": 5.0,
  "eval_steps": 500,
  "global_step": 2345,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.010660980810234541,
      "grad_norm": 0.008500133641064167,
      "learning_rate": 4.989339019189766e-05,
      "loss": 0.0014,
      "step": 5
    },
    {
      "epoch": 0.021321961620469083,
      "grad_norm": 0.008744274266064167,
      "learning_rate": 4.978678038379531e-05,
      "loss": 0.0416,
      "step": 10
    },
    {
      "epoch": 0.031982942430703626,
      "grad_norm": 7.240825653076172,
      "learning_rate": 4.9680170575692967e-05,
      "loss": 0.0704,
      "step": 15
    },
    {
      "epoch": 0.042643923240938165,
      "grad_norm": 0.051411617547273636,
      "learning_rate": 4.957356076759062e-05,
      "loss": 0.0341,
      "step": 20
    },
    {
      "epoch": 0.053304904051172705,
      "grad_norm": 10.886138916015625,
      "learning_rate": 4.9466950959488276e-05,
      "loss": 0.1281,
      "step": 25
    },
    {
      "epoch": 0.06396588486140725,
      "grad_norm": 4.6408185958862305,
      "learning_rate": 4.936034115138593e-05,
      "loss": 0.0365,
      "step": 30
    },
    {
      "epoch": 0.07462686567164178,
      "grad_norm": 2.041301727294922,
      "learning_rate": 4.9253731343283586e-05,
      "loss": 0.051,
      "step": 35
    },
    {
      "epoch": 0.08528784648187633,
      "grad_norm": 4.444990634918213,
      "learning_rate": 4.914712153518124e-05,
      "loss": 0.0229,
      "step": 40
    },
    {
      "epoch": 0.09594882729211088,
      "grad_norm": 0.863449215888977,
      "learning_rate": 4.904051172707889e-05,
      "loss": 0.0175,
      "step": 45
    },
    {
      "epoch": 0.10660980810234541,
      "grad_norm": 0.09464845806360245,
      "learning_rate": 4.893390191897655e-05,
      "loss": 0.0537,
      "step": 50
    },
    {
      "epoch": 0.11727078891257996,
      "grad_norm": 2.4883787631988525,
      "learning_rate": 4.88272921108742e-05,
      "loss": 0.0227,
      "step": 55
    },
    {
      "epoch": 0.1279317697228145,
      "grad_norm": 0.03964410722255707,
      "learning_rate": 4.872068230277186e-05,
      "loss": 0.0079,
      "step": 60
    },
    {
      "epoch": 0.13859275053304904,
      "grad_norm": 1.0171135663986206,
      "learning_rate": 4.861407249466951e-05,
      "loss": 0.0057,
      "step": 65
    },
    {
      "epoch": 0.14925373134328357,
      "grad_norm": 0.19365371763706207,
      "learning_rate": 4.850746268656717e-05,
      "loss": 0.0298,
      "step": 70
    },
    {
      "epoch": 0.15991471215351813,
      "grad_norm": 0.037564586848020554,
      "learning_rate": 4.840085287846482e-05,
      "loss": 0.0149,
      "step": 75
    },
    {
      "epoch": 0.17057569296375266,
      "grad_norm": 7.981247425079346,
      "learning_rate": 4.829424307036248e-05,
      "loss": 0.112,
      "step": 80
    },
    {
      "epoch": 0.1812366737739872,
      "grad_norm": 5.489224910736084,
      "learning_rate": 4.8187633262260126e-05,
      "loss": 0.0093,
      "step": 85
    },
    {
      "epoch": 0.19189765458422176,
      "grad_norm": 3.1869020462036133,
      "learning_rate": 4.808102345415779e-05,
      "loss": 0.0302,
      "step": 90
    },
    {
      "epoch": 0.2025586353944563,
      "grad_norm": 7.598812103271484,
      "learning_rate": 4.7974413646055436e-05,
      "loss": 0.0866,
      "step": 95
    },
    {
      "epoch": 0.21321961620469082,
      "grad_norm": 0.008008907549083233,
      "learning_rate": 4.78678038379531e-05,
      "loss": 0.0175,
      "step": 100
    },
    {
      "epoch": 0.22388059701492538,
      "grad_norm": 0.2593108117580414,
      "learning_rate": 4.7761194029850745e-05,
      "loss": 0.034,
      "step": 105
    },
    {
      "epoch": 0.2345415778251599,
      "grad_norm": 0.040492020547389984,
      "learning_rate": 4.765458422174841e-05,
      "loss": 0.0166,
      "step": 110
    },
    {
      "epoch": 0.24520255863539445,
      "grad_norm": 3.5951945781707764,
      "learning_rate": 4.7547974413646055e-05,
      "loss": 0.0773,
      "step": 115
    },
    {
      "epoch": 0.255863539445629,
      "grad_norm": 4.061870574951172,
      "learning_rate": 4.7441364605543716e-05,
      "loss": 0.0735,
      "step": 120
    },
    {
      "epoch": 0.26652452025586354,
      "grad_norm": 0.06774216145277023,
      "learning_rate": 4.7334754797441364e-05,
      "loss": 0.0238,
      "step": 125
    },
    {
      "epoch": 0.2771855010660981,
      "grad_norm": 3.8128223419189453,
      "learning_rate": 4.7228144989339026e-05,
      "loss": 0.0057,
      "step": 130
    },
    {
      "epoch": 0.2878464818763326,
      "grad_norm": 0.010711973533034325,
      "learning_rate": 4.7121535181236674e-05,
      "loss": 0.0025,
      "step": 135
    },
    {
      "epoch": 0.29850746268656714,
      "grad_norm": 9.293421745300293,
      "learning_rate": 4.7014925373134335e-05,
      "loss": 0.1074,
      "step": 140
    },
    {
      "epoch": 0.3091684434968017,
      "grad_norm": 0.1561048924922943,
      "learning_rate": 4.690831556503198e-05,
      "loss": 0.0209,
      "step": 145
    },
    {
      "epoch": 0.31982942430703626,
      "grad_norm": 7.143270969390869,
      "learning_rate": 4.6801705756929645e-05,
      "loss": 0.0233,
      "step": 150
    },
    {
      "epoch": 0.3304904051172708,
      "grad_norm": 1.846913456916809,
      "learning_rate": 4.669509594882729e-05,
      "loss": 0.0612,
      "step": 155
    },
    {
      "epoch": 0.3411513859275053,
      "grad_norm": 0.008405081927776337,
      "learning_rate": 4.658848614072495e-05,
      "loss": 0.0327,
      "step": 160
    },
    {
      "epoch": 0.35181236673773986,
      "grad_norm": 0.10778635740280151,
      "learning_rate": 4.64818763326226e-05,
      "loss": 0.0265,
      "step": 165
    },
    {
      "epoch": 0.3624733475479744,
      "grad_norm": 0.18631644546985626,
      "learning_rate": 4.637526652452026e-05,
      "loss": 0.04,
      "step": 170
    },
    {
      "epoch": 0.373134328358209,
      "grad_norm": 0.01041215006262064,
      "learning_rate": 4.626865671641791e-05,
      "loss": 0.0029,
      "step": 175
    },
    {
      "epoch": 0.3837953091684435,
      "grad_norm": 0.013913564383983612,
      "learning_rate": 4.6162046908315566e-05,
      "loss": 0.0159,
      "step": 180
    },
    {
      "epoch": 0.39445628997867804,
      "grad_norm": 3.8473548889160156,
      "learning_rate": 4.605543710021322e-05,
      "loss": 0.027,
      "step": 185
    },
    {
      "epoch": 0.4051172707889126,
      "grad_norm": 4.190244197845459,
      "learning_rate": 4.5948827292110876e-05,
      "loss": 0.0099,
      "step": 190
    },
    {
      "epoch": 0.4157782515991471,
      "grad_norm": 0.17146198451519012,
      "learning_rate": 4.584221748400853e-05,
      "loss": 0.0563,
      "step": 195
    },
    {
      "epoch": 0.42643923240938164,
      "grad_norm": 0.013823948800563812,
      "learning_rate": 4.5735607675906185e-05,
      "loss": 0.0209,
      "step": 200
    },
    {
      "epoch": 0.43710021321961623,
      "grad_norm": 0.031576912850141525,
      "learning_rate": 4.562899786780384e-05,
      "loss": 0.0101,
      "step": 205
    },
    {
      "epoch": 0.44776119402985076,
      "grad_norm": 0.06271517276763916,
      "learning_rate": 4.5522388059701495e-05,
      "loss": 0.0388,
      "step": 210
    },
    {
      "epoch": 0.4584221748400853,
      "grad_norm": 0.19199784100055695,
      "learning_rate": 4.541577825159915e-05,
      "loss": 0.0347,
      "step": 215
    },
    {
      "epoch": 0.4690831556503198,
      "grad_norm": 1.0773451328277588,
      "learning_rate": 4.5309168443496804e-05,
      "loss": 0.0022,
      "step": 220
    },
    {
      "epoch": 0.47974413646055436,
      "grad_norm": 0.02732914499938488,
      "learning_rate": 4.520255863539446e-05,
      "loss": 0.0189,
      "step": 225
    },
    {
      "epoch": 0.4904051172707889,
      "grad_norm": 8.48770809173584,
      "learning_rate": 4.5095948827292114e-05,
      "loss": 0.0119,
      "step": 230
    },
    {
      "epoch": 0.5010660980810234,
      "grad_norm": 0.013021047227084637,
      "learning_rate": 4.498933901918977e-05,
      "loss": 0.0064,
      "step": 235
    },
    {
      "epoch": 0.511727078891258,
      "grad_norm": 0.10618117451667786,
      "learning_rate": 4.488272921108742e-05,
      "loss": 0.0018,
      "step": 240
    },
    {
      "epoch": 0.5223880597014925,
      "grad_norm": 5.691826820373535,
      "learning_rate": 4.477611940298508e-05,
      "loss": 0.0243,
      "step": 245
    },
    {
      "epoch": 0.5330490405117271,
      "grad_norm": 0.7095174193382263,
      "learning_rate": 4.466950959488273e-05,
      "loss": 0.0366,
      "step": 250
    },
    {
      "epoch": 0.5437100213219617,
      "grad_norm": 0.005367923993617296,
      "learning_rate": 4.456289978678039e-05,
      "loss": 0.002,
      "step": 255
    },
    {
      "epoch": 0.5543710021321961,
      "grad_norm": 0.038158997893333435,
      "learning_rate": 4.445628997867804e-05,
      "loss": 0.0004,
      "step": 260
    },
    {
      "epoch": 0.5650319829424307,
      "grad_norm": 0.23171843588352203,
      "learning_rate": 4.43496801705757e-05,
      "loss": 0.0337,
      "step": 265
    },
    {
      "epoch": 0.5756929637526652,
      "grad_norm": 0.0788385346531868,
      "learning_rate": 4.424307036247335e-05,
      "loss": 0.0003,
      "step": 270
    },
    {
      "epoch": 0.5863539445628998,
      "grad_norm": 0.003817387158051133,
      "learning_rate": 4.4136460554371006e-05,
      "loss": 0.001,
      "step": 275
    },
    {
      "epoch": 0.5970149253731343,
      "grad_norm": 4.842153072357178,
      "learning_rate": 4.402985074626866e-05,
      "loss": 0.0035,
      "step": 280
    },
    {
      "epoch": 0.6076759061833689,
      "grad_norm": 0.011162374168634415,
      "learning_rate": 4.3923240938166316e-05,
      "loss": 0.0308,
      "step": 285
    },
    {
      "epoch": 0.6183368869936035,
      "grad_norm": 0.0014759559417143464,
      "learning_rate": 4.381663113006397e-05,
      "loss": 0.0047,
      "step": 290
    },
    {
      "epoch": 0.6289978678038379,
      "grad_norm": 0.004739244934171438,
      "learning_rate": 4.3710021321961625e-05,
      "loss": 0.0008,
      "step": 295
    },
    {
      "epoch": 0.6396588486140725,
      "grad_norm": 0.0018368859309703112,
      "learning_rate": 4.360341151385928e-05,
      "loss": 0.0152,
      "step": 300
    },
    {
      "epoch": 0.650319829424307,
      "grad_norm": 8.993382453918457,
      "learning_rate": 4.3496801705756935e-05,
      "loss": 0.0266,
      "step": 305
    },
    {
      "epoch": 0.6609808102345416,
      "grad_norm": 0.007308036554604769,
      "learning_rate": 4.339019189765459e-05,
      "loss": 0.0387,
      "step": 310
    },
    {
      "epoch": 0.6716417910447762,
      "grad_norm": 0.00822661817073822,
      "learning_rate": 4.328358208955224e-05,
      "loss": 0.022,
      "step": 315
    },
    {
      "epoch": 0.6823027718550106,
      "grad_norm": 0.001521592028439045,
      "learning_rate": 4.31769722814499e-05,
      "loss": 0.0216,
      "step": 320
    },
    {
      "epoch": 0.6929637526652452,
      "grad_norm": 0.0023254365660250187,
      "learning_rate": 4.307036247334755e-05,
      "loss": 0.0441,
      "step": 325
    },
    {
      "epoch": 0.7036247334754797,
      "grad_norm": 8.555769920349121,
      "learning_rate": 4.29637526652452e-05,
      "loss": 0.0529,
      "step": 330
    },
    {
      "epoch": 0.7142857142857143,
      "grad_norm": 6.5984416007995605,
      "learning_rate": 4.2857142857142856e-05,
      "loss": 0.0092,
      "step": 335
    },
    {
      "epoch": 0.7249466950959488,
      "grad_norm": 0.15340390801429749,
      "learning_rate": 4.275053304904051e-05,
      "loss": 0.0165,
      "step": 340
    },
    {
      "epoch": 0.7356076759061834,
      "grad_norm": 0.12547218799591064,
      "learning_rate": 4.2643923240938166e-05,
      "loss": 0.0326,
      "step": 345
    },
    {
      "epoch": 0.746268656716418,
      "grad_norm": 0.020376257598400116,
      "learning_rate": 4.253731343283582e-05,
      "loss": 0.013,
      "step": 350
    },
    {
      "epoch": 0.7569296375266524,
      "grad_norm": 5.508441925048828,
      "learning_rate": 4.2430703624733475e-05,
      "loss": 0.038,
      "step": 355
    },
    {
      "epoch": 0.767590618336887,
      "grad_norm": 8.317116737365723,
      "learning_rate": 4.232409381663113e-05,
      "loss": 0.0932,
      "step": 360
    },
    {
      "epoch": 0.7782515991471215,
      "grad_norm": 0.11259393393993378,
      "learning_rate": 4.2217484008528785e-05,
      "loss": 0.0846,
      "step": 365
    },
    {
      "epoch": 0.7889125799573561,
      "grad_norm": 0.002068769419565797,
      "learning_rate": 4.211087420042644e-05,
      "loss": 0.0139,
      "step": 370
    },
    {
      "epoch": 0.7995735607675906,
      "grad_norm": 8.80409049987793,
      "learning_rate": 4.2004264392324094e-05,
      "loss": 0.0588,
      "step": 375
    },
    {
      "epoch": 0.8102345415778252,
      "grad_norm": 0.07313705235719681,
      "learning_rate": 4.189765458422175e-05,
      "loss": 0.0011,
      "step": 380
    },
    {
      "epoch": 0.8208955223880597,
      "grad_norm": 2.5655972957611084,
      "learning_rate": 4.1791044776119404e-05,
      "loss": 0.0137,
      "step": 385
    },
    {
      "epoch": 0.8315565031982942,
      "grad_norm": 4.252593517303467,
      "learning_rate": 4.168443496801706e-05,
      "loss": 0.0071,
      "step": 390
    },
    {
      "epoch": 0.8422174840085288,
      "grad_norm": 3.0531997680664062,
      "learning_rate": 4.157782515991471e-05,
      "loss": 0.0038,
      "step": 395
    },
    {
      "epoch": 0.8528784648187633,
      "grad_norm": 0.5812060236930847,
      "learning_rate": 4.147121535181237e-05,
      "loss": 0.0065,
      "step": 400
    },
    {
      "epoch": 0.8635394456289979,
      "grad_norm": 0.005927251651883125,
      "learning_rate": 4.136460554371002e-05,
      "loss": 0.0747,
      "step": 405
    },
    {
      "epoch": 0.8742004264392325,
      "grad_norm": 4.556783676147461,
      "learning_rate": 4.125799573560768e-05,
      "loss": 0.0272,
      "step": 410
    },
    {
      "epoch": 0.8848614072494669,
      "grad_norm": 0.22763700783252716,
      "learning_rate": 4.115138592750533e-05,
      "loss": 0.0685,
      "step": 415
    },
    {
      "epoch": 0.8955223880597015,
      "grad_norm": 0.04457894712686539,
      "learning_rate": 4.104477611940299e-05,
      "loss": 0.0354,
      "step": 420
    },
    {
      "epoch": 0.906183368869936,
      "grad_norm": 3.034590482711792,
      "learning_rate": 4.093816631130064e-05,
      "loss": 0.0129,
      "step": 425
    },
    {
      "epoch": 0.9168443496801706,
      "grad_norm": 1.252897024154663,
      "learning_rate": 4.0831556503198296e-05,
      "loss": 0.0277,
      "step": 430
    },
    {
      "epoch": 0.9275053304904051,
      "grad_norm": 0.011865551583468914,
      "learning_rate": 4.072494669509595e-05,
      "loss": 0.0092,
      "step": 435
    },
    {
      "epoch": 0.9381663113006397,
      "grad_norm": 8.566657066345215,
      "learning_rate": 4.0618336886993606e-05,
      "loss": 0.0674,
      "step": 440
    },
    {
      "epoch": 0.9488272921108742,
      "grad_norm": 0.4269752502441406,
      "learning_rate": 4.051172707889126e-05,
      "loss": 0.0231,
      "step": 445
    },
    {
      "epoch": 0.9594882729211087,
      "grad_norm": 0.0633544996380806,
      "learning_rate": 4.0405117270788915e-05,
      "loss": 0.0068,
      "step": 450
    },
    {
      "epoch": 0.9701492537313433,
      "grad_norm": 0.01764209195971489,
      "learning_rate": 4.029850746268657e-05,
      "loss": 0.0026,
      "step": 455
    },
    {
      "epoch": 0.9808102345415778,
      "grad_norm": 0.38884612917900085,
      "learning_rate": 4.0191897654584225e-05,
      "loss": 0.0018,
      "step": 460
    },
    {
      "epoch": 0.9914712153518124,
      "grad_norm": 6.430479526519775,
      "learning_rate": 4.008528784648188e-05,
      "loss": 0.0164,
      "step": 465
    },
    {
      "epoch": 1.0,
      "eval_accuracy": 0.9818666666666667,
      "eval_loss": 0.07106676697731018,
      "eval_runtime": 17.9774,
      "eval_samples_per_second": 208.596,
      "eval_steps_per_second": 6.564,
      "step": 469
    },
    {
      "epoch": 1.0021321961620469,
      "grad_norm": 0.001062327646650374,
      "learning_rate": 3.997867803837953e-05,
      "loss": 0.0284,
      "step": 470
    },
    {
      "epoch": 1.0127931769722816,
      "grad_norm": 0.008273986168205738,
      "learning_rate": 3.987206823027719e-05,
      "loss": 0.001,
      "step": 475
    },
    {
      "epoch": 1.023454157782516,
      "grad_norm": 0.3330083191394806,
      "learning_rate": 3.976545842217484e-05,
      "loss": 0.0014,
      "step": 480
    },
    {
      "epoch": 1.0341151385927505,
      "grad_norm": 0.0020932976622134447,
      "learning_rate": 3.96588486140725e-05,
      "loss": 0.0187,
      "step": 485
    },
    {
      "epoch": 1.044776119402985,
      "grad_norm": 0.01144615188241005,
      "learning_rate": 3.9552238805970146e-05,
      "loss": 0.0107,
      "step": 490
    },
    {
      "epoch": 1.0554371002132197,
      "grad_norm": 0.008168315514922142,
      "learning_rate": 3.944562899786781e-05,
      "loss": 0.0023,
      "step": 495
    },
    {
      "epoch": 1.0660980810234542,
      "grad_norm": 0.008570204488933086,
      "learning_rate": 3.9339019189765456e-05,
      "loss": 0.0002,
      "step": 500
    },
    {
      "epoch": 1.0767590618336886,
      "grad_norm": 0.005483146291226149,
      "learning_rate": 3.923240938166312e-05,
      "loss": 0.0041,
      "step": 505
    },
    {
      "epoch": 1.0874200426439233,
      "grad_norm": 14.994518280029297,
      "learning_rate": 3.9125799573560765e-05,
      "loss": 0.0108,
      "step": 510
    },
    {
      "epoch": 1.0980810234541578,
      "grad_norm": 0.013237055391073227,
      "learning_rate": 3.901918976545843e-05,
      "loss": 0.0002,
      "step": 515
    },
    {
      "epoch": 1.1087420042643923,
      "grad_norm": 0.03122478350996971,
      "learning_rate": 3.8912579957356075e-05,
      "loss": 0.0002,
      "step": 520
    },
    {
      "epoch": 1.1194029850746268,
      "grad_norm": 0.4769461750984192,
      "learning_rate": 3.8805970149253736e-05,
      "loss": 0.0149,
      "step": 525
    },
    {
      "epoch": 1.1300639658848615,
      "grad_norm": 0.02249765768647194,
      "learning_rate": 3.8699360341151384e-05,
      "loss": 0.0003,
      "step": 530
    },
    {
      "epoch": 1.140724946695096,
      "grad_norm": 0.0008661285974085331,
      "learning_rate": 3.8592750533049046e-05,
      "loss": 0.0345,
      "step": 535
    },
    {
      "epoch": 1.1513859275053304,
      "grad_norm": 0.001317141461186111,
      "learning_rate": 3.8486140724946694e-05,
      "loss": 0.0004,
      "step": 540
    },
    {
      "epoch": 1.1620469083155651,
      "grad_norm": 0.3196207582950592,
      "learning_rate": 3.8379530916844355e-05,
      "loss": 0.003,
      "step": 545
    },
    {
      "epoch": 1.1727078891257996,
      "grad_norm": 0.0016030617989599705,
      "learning_rate": 3.8272921108742e-05,
      "loss": 0.0006,
      "step": 550
    },
    {
      "epoch": 1.183368869936034,
      "grad_norm": 0.004598675761371851,
      "learning_rate": 3.8166311300639665e-05,
      "loss": 0.0002,
      "step": 555
    },
    {
      "epoch": 1.1940298507462686,
      "grad_norm": 0.0012001367285847664,
      "learning_rate": 3.805970149253731e-05,
      "loss": 0.0001,
      "step": 560
    },
    {
      "epoch": 1.2046908315565032,
      "grad_norm": 6.540524482727051,
      "learning_rate": 3.7953091684434974e-05,
      "loss": 0.0036,
      "step": 565
    },
    {
      "epoch": 1.2153518123667377,
      "grad_norm": 0.013638881966471672,
      "learning_rate": 3.784648187633262e-05,
      "loss": 0.004,
      "step": 570
    },
    {
      "epoch": 1.2260127931769722,
      "grad_norm": 0.001845084480009973,
      "learning_rate": 3.7739872068230284e-05,
      "loss": 0.0178,
      "step": 575
    },
    {
      "epoch": 1.236673773987207,
      "grad_norm": 0.0015873904339969158,
      "learning_rate": 3.763326226012793e-05,
      "loss": 0.0088,
      "step": 580
    },
    {
      "epoch": 1.2473347547974414,
      "grad_norm": 0.5405803322792053,
      "learning_rate": 3.752665245202559e-05,
      "loss": 0.0006,
      "step": 585
    },
    {
      "epoch": 1.2579957356076759,
      "grad_norm": 0.012557074427604675,
      "learning_rate": 3.742004264392324e-05,
      "loss": 0.0001,
      "step": 590
    },
    {
      "epoch": 1.2686567164179103,
      "grad_norm": 0.001125634997151792,
      "learning_rate": 3.73134328358209e-05,
      "loss": 0.0001,
      "step": 595
    },
    {
      "epoch": 1.279317697228145,
      "grad_norm": 7.692508697509766,
      "learning_rate": 3.720682302771855e-05,
      "loss": 0.0254,
      "step": 600
    },
    {
      "epoch": 1.2899786780383795,
      "grad_norm": 0.07067856937646866,
      "learning_rate": 3.710021321961621e-05,
      "loss": 0.0198,
      "step": 605
    },
    {
      "epoch": 1.3006396588486142,
      "grad_norm": 0.022861022502183914,
      "learning_rate": 3.699360341151386e-05,
      "loss": 0.0002,
      "step": 610
    },
    {
      "epoch": 1.3113006396588487,
      "grad_norm": 0.017372073605656624,
      "learning_rate": 3.6886993603411515e-05,
      "loss": 0.0068,
      "step": 615
    },
    {
      "epoch": 1.3219616204690832,
      "grad_norm": 0.17841441929340363,
      "learning_rate": 3.678038379530917e-05,
      "loss": 0.0003,
      "step": 620
    },
    {
      "epoch": 1.3326226012793176,
      "grad_norm": 0.0014389833668246865,
      "learning_rate": 3.6673773987206824e-05,
      "loss": 0.0004,
      "step": 625
    },
    {
      "epoch": 1.3432835820895521,
      "grad_norm": 0.0018343289848417044,
      "learning_rate": 3.656716417910448e-05,
      "loss": 0.001,
      "step": 630
    },
    {
      "epoch": 1.3539445628997868,
      "grad_norm": 0.008533881977200508,
      "learning_rate": 3.6460554371002134e-05,
      "loss": 0.0139,
      "step": 635
    },
    {
      "epoch": 1.3646055437100213,
      "grad_norm": 0.08136972784996033,
      "learning_rate": 3.635394456289979e-05,
      "loss": 0.0003,
      "step": 640
    },
    {
      "epoch": 1.375266524520256,
      "grad_norm": 0.007784366607666016,
      "learning_rate": 3.624733475479744e-05,
      "loss": 0.0001,
      "step": 645
    },
    {
      "epoch": 1.3859275053304905,
      "grad_norm": 0.044875115156173706,
      "learning_rate": 3.61407249466951e-05,
      "loss": 0.0002,
      "step": 650
    },
    {
      "epoch": 1.396588486140725,
      "grad_norm": 0.004530559293925762,
      "learning_rate": 3.603411513859275e-05,
      "loss": 0.0429,
      "step": 655
    },
    {
      "epoch": 1.4072494669509594,
      "grad_norm": 0.10593441128730774,
      "learning_rate": 3.592750533049041e-05,
      "loss": 0.0005,
      "step": 660
    },
    {
      "epoch": 1.417910447761194,
      "grad_norm": 0.01833045668900013,
      "learning_rate": 3.582089552238806e-05,
      "loss": 0.0002,
      "step": 665
    },
    {
      "epoch": 1.4285714285714286,
      "grad_norm": 0.004971946123987436,
      "learning_rate": 3.571428571428572e-05,
      "loss": 0.0318,
      "step": 670
    },
    {
      "epoch": 1.439232409381663,
      "grad_norm": 0.004689859226346016,
      "learning_rate": 3.560767590618337e-05,
      "loss": 0.027,
      "step": 675
    },
    {
      "epoch": 1.4498933901918978,
      "grad_norm": 0.009141859598457813,
      "learning_rate": 3.5501066098081026e-05,
      "loss": 0.0002,
      "step": 680
    },
    {
      "epoch": 1.4605543710021323,
      "grad_norm": 0.008217399008572102,
      "learning_rate": 3.539445628997868e-05,
      "loss": 0.0029,
      "step": 685
    },
    {
      "epoch": 1.4712153518123667,
      "grad_norm": 4.9714436531066895,
      "learning_rate": 3.5287846481876336e-05,
      "loss": 0.0191,
      "step": 690
    },
    {
      "epoch": 1.4818763326226012,
      "grad_norm": 6.374760150909424,
      "learning_rate": 3.518123667377399e-05,
      "loss": 0.0272,
      "step": 695
    },
    {
      "epoch": 1.4925373134328357,
      "grad_norm": 0.0025179891381412745,
      "learning_rate": 3.5074626865671645e-05,
      "loss": 0.0002,
      "step": 700
    },
    {
      "epoch": 1.5031982942430704,
      "grad_norm": 0.0022709995973855257,
      "learning_rate": 3.496801705756929e-05,
      "loss": 0.0456,
      "step": 705
    },
    {
      "epoch": 1.5138592750533049,
      "grad_norm": 0.017147591337561607,
      "learning_rate": 3.4861407249466955e-05,
      "loss": 0.0001,
      "step": 710
    },
    {
      "epoch": 1.5245202558635396,
      "grad_norm": 0.016144555062055588,
      "learning_rate": 3.47547974413646e-05,
      "loss": 0.0002,
      "step": 715
    },
    {
      "epoch": 1.535181236673774,
      "grad_norm": 0.8192442059516907,
      "learning_rate": 3.4648187633262264e-05,
      "loss": 0.0008,
      "step": 720
    },
    {
      "epoch": 1.5458422174840085,
      "grad_norm": 4.73784065246582,
      "learning_rate": 3.454157782515991e-05,
      "loss": 0.0386,
      "step": 725
    },
    {
      "epoch": 1.556503198294243,
      "grad_norm": 0.023815130814909935,
      "learning_rate": 3.4434968017057574e-05,
      "loss": 0.0012,
      "step": 730
    },
    {
      "epoch": 1.5671641791044775,
      "grad_norm": 0.09039128571748734,
      "learning_rate": 3.432835820895522e-05,
      "loss": 0.0012,
      "step": 735
    },
    {
      "epoch": 1.5778251599147122,
      "grad_norm": 0.003255978226661682,
      "learning_rate": 3.422174840085288e-05,
      "loss": 0.028,
      "step": 740
    },
    {
      "epoch": 1.5884861407249466,
      "grad_norm": 0.0013892430579289794,
      "learning_rate": 3.411513859275053e-05,
      "loss": 0.0008,
      "step": 745
    },
    {
      "epoch": 1.5991471215351813,
      "grad_norm": 0.013135437853634357,
      "learning_rate": 3.400852878464819e-05,
      "loss": 0.0144,
      "step": 750
    },
    {
      "epoch": 1.6098081023454158,
      "grad_norm": 0.046194031834602356,
      "learning_rate": 3.390191897654584e-05,
      "loss": 0.0002,
      "step": 755
    },
    {
      "epoch": 1.6204690831556503,
      "grad_norm": 0.001058321911841631,
      "learning_rate": 3.37953091684435e-05,
      "loss": 0.0001,
      "step": 760
    },
    {
      "epoch": 1.6311300639658848,
      "grad_norm": 0.002939931582659483,
      "learning_rate": 3.368869936034115e-05,
      "loss": 0.0006,
      "step": 765
    },
    {
      "epoch": 1.6417910447761193,
      "grad_norm": 0.013005989603698254,
      "learning_rate": 3.358208955223881e-05,
      "loss": 0.0004,
      "step": 770
    },
    {
      "epoch": 1.652452025586354,
      "grad_norm": 0.0013527501141652465,
      "learning_rate": 3.347547974413646e-05,
      "loss": 0.0002,
      "step": 775
    },
    {
      "epoch": 1.6631130063965884,
      "grad_norm": 0.0027004110161215067,
      "learning_rate": 3.336886993603412e-05,
      "loss": 0.0001,
      "step": 780
    },
    {
      "epoch": 1.6737739872068231,
      "grad_norm": 0.0015707183629274368,
      "learning_rate": 3.326226012793177e-05,
      "loss": 0.0002,
      "step": 785
    },
    {
      "epoch": 1.6844349680170576,
      "grad_norm": 11.830730438232422,
      "learning_rate": 3.3155650319829424e-05,
      "loss": 0.0164,
      "step": 790
    },
    {
      "epoch": 1.695095948827292,
      "grad_norm": 0.0006959062884561718,
      "learning_rate": 3.304904051172708e-05,
      "loss": 0.0003,
      "step": 795
    },
    {
      "epoch": 1.7057569296375266,
      "grad_norm": 0.1154661774635315,
      "learning_rate": 3.294243070362473e-05,
      "loss": 0.0006,
      "step": 800
    },
    {
      "epoch": 1.716417910447761,
      "grad_norm": 0.0029258604627102613,
      "learning_rate": 3.283582089552239e-05,
      "loss": 0.0001,
      "step": 805
    },
    {
      "epoch": 1.7270788912579957,
      "grad_norm": 0.010158653371036053,
      "learning_rate": 3.272921108742004e-05,
      "loss": 0.0072,
      "step": 810
    },
    {
      "epoch": 1.7377398720682304,
      "grad_norm": 0.023314760997891426,
      "learning_rate": 3.26226012793177e-05,
      "loss": 0.0003,
      "step": 815
    },
    {
      "epoch": 1.748400852878465,
      "grad_norm": 0.005231409799307585,
      "learning_rate": 3.251599147121535e-05,
      "loss": 0.0551,
      "step": 820
    },
    {
      "epoch": 1.7590618336886994,
      "grad_norm": 0.0467340461909771,
      "learning_rate": 3.240938166311301e-05,
      "loss": 0.0004,
      "step": 825
    },
    {
      "epoch": 1.7697228144989339,
      "grad_norm": 0.02259242720901966,
      "learning_rate": 3.230277185501066e-05,
      "loss": 0.0033,
      "step": 830
    },
    {
      "epoch": 1.7803837953091683,
      "grad_norm": 0.12362359464168549,
      "learning_rate": 3.2196162046908317e-05,
      "loss": 0.001,
      "step": 835
    },
    {
      "epoch": 1.7910447761194028,
      "grad_norm": 0.0032769027166068554,
      "learning_rate": 3.208955223880597e-05,
      "loss": 0.0002,
      "step": 840
    },
    {
      "epoch": 1.8017057569296375,
      "grad_norm": 0.32296222448349,
      "learning_rate": 3.1982942430703626e-05,
      "loss": 0.0273,
      "step": 845
    },
    {
      "epoch": 1.8123667377398722,
      "grad_norm": 6.43861722946167,
      "learning_rate": 3.187633262260128e-05,
      "loss": 0.0025,
      "step": 850
    },
    {
      "epoch": 1.8230277185501067,
      "grad_norm": 0.0008607152267359197,
      "learning_rate": 3.1769722814498935e-05,
      "loss": 0.0001,
      "step": 855
    },
    {
      "epoch": 1.8336886993603412,
      "grad_norm": 0.008054632693529129,
      "learning_rate": 3.166311300639659e-05,
      "loss": 0.0002,
      "step": 860
    },
    {
      "epoch": 1.8443496801705757,
      "grad_norm": 0.0036459804978221655,
      "learning_rate": 3.1556503198294245e-05,
      "loss": 0.0001,
      "step": 865
    },
    {
      "epoch": 1.8550106609808101,
      "grad_norm": 0.1348894238471985,
      "learning_rate": 3.14498933901919e-05,
      "loss": 0.0016,
      "step": 870
    },
    {
      "epoch": 1.8656716417910446,
      "grad_norm": 13.615499496459961,
      "learning_rate": 3.1343283582089554e-05,
      "loss": 0.0086,
      "step": 875
    },
    {
      "epoch": 1.8763326226012793,
      "grad_norm": 0.007840185426175594,
      "learning_rate": 3.123667377398721e-05,
      "loss": 0.0004,
      "step": 880
    },
    {
      "epoch": 1.886993603411514,
      "grad_norm": 0.01908479817211628,
      "learning_rate": 3.1130063965884864e-05,
      "loss": 0.0002,
      "step": 885
    },
    {
      "epoch": 1.8976545842217485,
      "grad_norm": 0.09402158111333847,
      "learning_rate": 3.102345415778252e-05,
      "loss": 0.0006,
      "step": 890
    },
    {
      "epoch": 1.908315565031983,
      "grad_norm": 0.008091027848422527,
      "learning_rate": 3.0916844349680173e-05,
      "loss": 0.0009,
      "step": 895
    },
    {
      "epoch": 1.9189765458422174,
      "grad_norm": 0.0007408737437799573,
      "learning_rate": 3.081023454157783e-05,
      "loss": 0.0046,
      "step": 900
    },
    {
      "epoch": 1.929637526652452,
      "grad_norm": 0.00704607367515564,
      "learning_rate": 3.070362473347548e-05,
      "loss": 0.0001,
      "step": 905
    },
    {
      "epoch": 1.9402985074626866,
      "grad_norm": 0.006379020866006613,
      "learning_rate": 3.059701492537314e-05,
      "loss": 0.0006,
      "step": 910
    },
    {
      "epoch": 1.950959488272921,
      "grad_norm": 0.004528722260147333,
      "learning_rate": 3.0490405117270792e-05,
      "loss": 0.0002,
      "step": 915
    },
    {
      "epoch": 1.9616204690831558,
      "grad_norm": 0.0007978453068062663,
      "learning_rate": 3.0383795309168444e-05,
      "loss": 0.0001,
      "step": 920
    },
    {
      "epoch": 1.9722814498933903,
      "grad_norm": 0.0015598306199535728,
      "learning_rate": 3.0277185501066102e-05,
      "loss": 0.0023,
      "step": 925
    },
    {
      "epoch": 1.9829424307036247,
      "grad_norm": 0.0013448342215269804,
      "learning_rate": 3.0170575692963753e-05,
      "loss": 0.0001,
      "step": 930
    },
    {
      "epoch": 1.9936034115138592,
      "grad_norm": 0.02841823734343052,
      "learning_rate": 3.006396588486141e-05,
      "loss": 0.0048,
      "step": 935
    },
    {
      "epoch": 2.0,
      "eval_accuracy": 0.9824,
      "eval_loss": 0.07852894067764282,
      "eval_runtime": 17.0526,
      "eval_samples_per_second": 219.908,
      "eval_steps_per_second": 6.92,
      "step": 938
    },
    {
      "epoch": 2.0042643923240937,
      "grad_norm": 0.0028412993997335434,
      "learning_rate": 2.9957356076759063e-05,
      "loss": 0.0001,
      "step": 940
    },
    {
      "epoch": 2.014925373134328,
      "grad_norm": 0.000876828795298934,
      "learning_rate": 2.9850746268656714e-05,
      "loss": 0.0001,
      "step": 945
    },
    {
      "epoch": 2.025586353944563,
      "grad_norm": 0.0006893366225995123,
      "learning_rate": 2.9744136460554372e-05,
      "loss": 0.0001,
      "step": 950
    },
    {
      "epoch": 2.0362473347547976,
      "grad_norm": 0.0008075434598140419,
      "learning_rate": 2.9637526652452023e-05,
      "loss": 0.0001,
      "step": 955
    },
    {
      "epoch": 2.046908315565032,
      "grad_norm": 0.0038414648734033108,
      "learning_rate": 2.953091684434968e-05,
      "loss": 0.0001,
      "step": 960
    },
    {
      "epoch": 2.0575692963752665,
      "grad_norm": 0.0021969624795019627,
      "learning_rate": 2.9424307036247333e-05,
      "loss": 0.0125,
      "step": 965
    },
    {
      "epoch": 2.068230277185501,
      "grad_norm": 0.0006368861068040133,
      "learning_rate": 2.931769722814499e-05,
      "loss": 0.0008,
      "step": 970
    },
    {
      "epoch": 2.0788912579957355,
      "grad_norm": 0.0008505034493282437,
      "learning_rate": 2.9211087420042642e-05,
      "loss": 0.0002,
      "step": 975
    },
    {
      "epoch": 2.08955223880597,
      "grad_norm": 0.0005651818937622011,
      "learning_rate": 2.91044776119403e-05,
      "loss": 0.0001,
      "step": 980
    },
    {
      "epoch": 2.100213219616205,
      "grad_norm": 0.0007468526600860059,
      "learning_rate": 2.8997867803837952e-05,
      "loss": 0.0001,
      "step": 985
    },
    {
      "epoch": 2.1108742004264394,
      "grad_norm": 0.0006225996767170727,
      "learning_rate": 2.889125799573561e-05,
      "loss": 0.0001,
      "step": 990
    },
    {
      "epoch": 2.121535181236674,
      "grad_norm": 0.0006689952570013702,
      "learning_rate": 2.878464818763326e-05,
      "loss": 0.0001,
      "step": 995
    },
    {
      "epoch": 2.1321961620469083,
      "grad_norm": 0.0006338393432088196,
      "learning_rate": 2.867803837953092e-05,
      "loss": 0.0003,
      "step": 1000
    },
    {
      "epoch": 2.142857142857143,
      "grad_norm": 0.0007292141672223806,
      "learning_rate": 2.857142857142857e-05,
      "loss": 0.0001,
      "step": 1005
    },
    {
      "epoch": 2.1535181236673773,
      "grad_norm": 12.55198860168457,
      "learning_rate": 2.846481876332623e-05,
      "loss": 0.0114,
      "step": 1010
    },
    {
      "epoch": 2.1641791044776117,
      "grad_norm": 0.0006730745080858469,
      "learning_rate": 2.835820895522388e-05,
      "loss": 0.0001,
      "step": 1015
    },
    {
      "epoch": 2.1748400852878467,
      "grad_norm": 0.007554246578365564,
      "learning_rate": 2.825159914712154e-05,
      "loss": 0.0001,
      "step": 1020
    },
    {
      "epoch": 2.185501066098081,
      "grad_norm": 0.0007020197808742523,
      "learning_rate": 2.814498933901919e-05,
      "loss": 0.0001,
      "step": 1025
    },
    {
      "epoch": 2.1961620469083156,
      "grad_norm": 0.001348011428490281,
      "learning_rate": 2.8038379530916848e-05,
      "loss": 0.0001,
      "step": 1030
    },
    {
      "epoch": 2.20682302771855,
      "grad_norm": 0.0015917104901745915,
      "learning_rate": 2.79317697228145e-05,
      "loss": 0.0001,
      "step": 1035
    },
    {
      "epoch": 2.2174840085287846,
      "grad_norm": 0.0009162525529973209,
      "learning_rate": 2.7825159914712157e-05,
      "loss": 0.0002,
      "step": 1040
    },
    {
      "epoch": 2.228144989339019,
      "grad_norm": 0.0025192436296492815,
      "learning_rate": 2.771855010660981e-05,
      "loss": 0.0001,
      "step": 1045
    },
    {
      "epoch": 2.2388059701492535,
      "grad_norm": 0.0014659248990938067,
      "learning_rate": 2.7611940298507467e-05,
      "loss": 0.0002,
      "step": 1050
    },
    {
      "epoch": 2.2494669509594885,
      "grad_norm": 0.0009001969010569155,
      "learning_rate": 2.7505330490405118e-05,
      "loss": 0.0001,
      "step": 1055
    },
    {
      "epoch": 2.260127931769723,
      "grad_norm": 0.0017667623469606042,
      "learning_rate": 2.7398720682302776e-05,
      "loss": 0.0046,
      "step": 1060
    },
    {
      "epoch": 2.2707889125799574,
      "grad_norm": 0.0006441583391278982,
      "learning_rate": 2.7292110874200428e-05,
      "loss": 0.0001,
      "step": 1065
    },
    {
      "epoch": 2.281449893390192,
      "grad_norm": 0.0020542279817163944,
      "learning_rate": 2.7185501066098086e-05,
      "loss": 0.0001,
      "step": 1070
    },
    {
      "epoch": 2.2921108742004264,
      "grad_norm": 0.020054394379258156,
      "learning_rate": 2.7078891257995737e-05,
      "loss": 0.0001,
      "step": 1075
    },
    {
      "epoch": 2.302771855010661,
      "grad_norm": 0.007889393717050552,
      "learning_rate": 2.6972281449893395e-05,
      "loss": 0.0002,
      "step": 1080
    },
    {
      "epoch": 2.3134328358208958,
      "grad_norm": 0.0017209519864991307,
      "learning_rate": 2.6865671641791047e-05,
      "loss": 0.0001,
      "step": 1085
    },
    {
      "epoch": 2.3240938166311302,
      "grad_norm": 0.03484933450818062,
      "learning_rate": 2.6759061833688705e-05,
      "loss": 0.0001,
      "step": 1090
    },
    {
      "epoch": 2.3347547974413647,
      "grad_norm": 0.0012376690283417702,
      "learning_rate": 2.6652452025586356e-05,
      "loss": 0.0001,
      "step": 1095
    },
    {
      "epoch": 2.345415778251599,
      "grad_norm": 0.0011016852222383022,
      "learning_rate": 2.6545842217484007e-05,
      "loss": 0.0002,
      "step": 1100
    },
    {
      "epoch": 2.3560767590618337,
      "grad_norm": 0.0012145757209509611,
      "learning_rate": 2.6439232409381666e-05,
      "loss": 0.0001,
      "step": 1105
    },
    {
      "epoch": 2.366737739872068,
      "grad_norm": 0.00464074220508337,
      "learning_rate": 2.6332622601279317e-05,
      "loss": 0.0071,
      "step": 1110
    },
    {
      "epoch": 2.3773987206823026,
      "grad_norm": 0.004989962093532085,
      "learning_rate": 2.6226012793176975e-05,
      "loss": 0.0002,
      "step": 1115
    },
    {
      "epoch": 2.388059701492537,
      "grad_norm": 0.001367892837151885,
      "learning_rate": 2.6119402985074626e-05,
      "loss": 0.0001,
      "step": 1120
    },
    {
      "epoch": 2.398720682302772,
      "grad_norm": 0.0006096764700487256,
      "learning_rate": 2.6012793176972285e-05,
      "loss": 0.0001,
      "step": 1125
    },
    {
      "epoch": 2.4093816631130065,
      "grad_norm": 0.7724946737289429,
      "learning_rate": 2.5906183368869936e-05,
      "loss": 0.0006,
      "step": 1130
    },
    {
      "epoch": 2.420042643923241,
      "grad_norm": 0.0005675335414707661,
      "learning_rate": 2.5799573560767594e-05,
      "loss": 0.0001,
      "step": 1135
    },
    {
      "epoch": 2.4307036247334755,
      "grad_norm": 0.007536550983786583,
      "learning_rate": 2.5692963752665245e-05,
      "loss": 0.0001,
      "step": 1140
    },
    {
      "epoch": 2.44136460554371,
      "grad_norm": 0.0018686552066355944,
      "learning_rate": 2.5586353944562904e-05,
      "loss": 0.0001,
      "step": 1145
    },
    {
      "epoch": 2.4520255863539444,
      "grad_norm": 0.0014288212405517697,
      "learning_rate": 2.5479744136460555e-05,
      "loss": 0.0001,
      "step": 1150
    },
    {
      "epoch": 2.4626865671641793,
      "grad_norm": 0.0007582632824778557,
      "learning_rate": 2.537313432835821e-05,
      "loss": 0.0005,
      "step": 1155
    },
    {
      "epoch": 2.473347547974414,
      "grad_norm": 0.0008698916644789279,
      "learning_rate": 2.5266524520255864e-05,
      "loss": 0.0001,
      "step": 1160
    },
    {
      "epoch": 2.4840085287846483,
      "grad_norm": 0.0008244248456321657,
      "learning_rate": 2.515991471215352e-05,
      "loss": 0.0001,
      "step": 1165
    },
    {
      "epoch": 2.4946695095948828,
      "grad_norm": 0.0006152259884402156,
      "learning_rate": 2.5053304904051174e-05,
      "loss": 0.0001,
      "step": 1170
    },
    {
      "epoch": 2.5053304904051172,
      "grad_norm": 0.011618640273809433,
      "learning_rate": 2.494669509594883e-05,
      "loss": 0.0001,
      "step": 1175
    },
    {
      "epoch": 2.5159914712153517,
      "grad_norm": 0.0004514007596299052,
      "learning_rate": 2.4840085287846483e-05,
      "loss": 0.0001,
      "step": 1180
    },
    {
      "epoch": 2.526652452025586,
      "grad_norm": 0.0015603859210386872,
      "learning_rate": 2.4733475479744138e-05,
      "loss": 0.0016,
      "step": 1185
    },
    {
      "epoch": 2.5373134328358207,
      "grad_norm": 0.0025411758106201887,
      "learning_rate": 2.4626865671641793e-05,
      "loss": 0.0001,
      "step": 1190
    },
    {
      "epoch": 2.5479744136460556,
      "grad_norm": 0.0005884451093152165,
      "learning_rate": 2.4520255863539444e-05,
      "loss": 0.0001,
      "step": 1195
    },
    {
      "epoch": 2.55863539445629,
      "grad_norm": 0.0005032878252677619,
      "learning_rate": 2.44136460554371e-05,
      "loss": 0.0001,
      "step": 1200
    },
    {
      "epoch": 2.5692963752665245,
      "grad_norm": 0.003127189353108406,
      "learning_rate": 2.4307036247334754e-05,
      "loss": 0.0001,
      "step": 1205
    },
    {
      "epoch": 2.579957356076759,
      "grad_norm": 0.03236782178282738,
      "learning_rate": 2.420042643923241e-05,
      "loss": 0.0001,
      "step": 1210
    },
    {
      "epoch": 2.5906183368869935,
      "grad_norm": 0.0005310071865096688,
      "learning_rate": 2.4093816631130063e-05,
      "loss": 0.0001,
      "step": 1215
    },
    {
      "epoch": 2.6012793176972284,
      "grad_norm": 0.03421960026025772,
      "learning_rate": 2.3987206823027718e-05,
      "loss": 0.0006,
      "step": 1220
    },
    {
      "epoch": 2.611940298507463,
      "grad_norm": 0.0005128477932885289,
      "learning_rate": 2.3880597014925373e-05,
      "loss": 0.0001,
      "step": 1225
    },
    {
      "epoch": 2.6226012793176974,
      "grad_norm": 0.0007355711422860622,
      "learning_rate": 2.3773987206823027e-05,
      "loss": 0.0001,
      "step": 1230
    },
    {
      "epoch": 2.633262260127932,
      "grad_norm": 0.00048258117749355733,
      "learning_rate": 2.3667377398720682e-05,
      "loss": 0.0001,
      "step": 1235
    },
    {
      "epoch": 2.6439232409381663,
      "grad_norm": 0.0016943534137681127,
      "learning_rate": 2.3560767590618337e-05,
      "loss": 0.0001,
      "step": 1240
    },
    {
      "epoch": 2.654584221748401,
      "grad_norm": 0.0006591964629478753,
      "learning_rate": 2.345415778251599e-05,
      "loss": 0.004,
      "step": 1245
    },
    {
      "epoch": 2.6652452025586353,
      "grad_norm": 0.054243750870227814,
      "learning_rate": 2.3347547974413646e-05,
      "loss": 0.0002,
      "step": 1250
    },
    {
      "epoch": 2.6759061833688698,
      "grad_norm": 0.0005023294361308217,
      "learning_rate": 2.32409381663113e-05,
      "loss": 0.0001,
      "step": 1255
    },
    {
      "epoch": 2.6865671641791042,
      "grad_norm": 0.0005061823758296669,
      "learning_rate": 2.3134328358208956e-05,
      "loss": 0.0001,
      "step": 1260
    },
    {
      "epoch": 2.697228144989339,
      "grad_norm": 0.0005865807761438191,
      "learning_rate": 2.302771855010661e-05,
      "loss": 0.0001,
      "step": 1265
    },
    {
      "epoch": 2.7078891257995736,
      "grad_norm": 0.002101427875459194,
      "learning_rate": 2.2921108742004265e-05,
      "loss": 0.0001,
      "step": 1270
    },
    {
      "epoch": 2.718550106609808,
      "grad_norm": 0.0006000754656270146,
      "learning_rate": 2.281449893390192e-05,
      "loss": 0.0018,
      "step": 1275
    },
    {
      "epoch": 2.7292110874200426,
      "grad_norm": 0.0004894935409538448,
      "learning_rate": 2.2707889125799575e-05,
      "loss": 0.0001,
      "step": 1280
    },
    {
      "epoch": 2.739872068230277,
      "grad_norm": 0.0008587398333474994,
      "learning_rate": 2.260127931769723e-05,
      "loss": 0.0001,
      "step": 1285
    },
    {
      "epoch": 2.750533049040512,
      "grad_norm": 0.0010474597802385688,
      "learning_rate": 2.2494669509594884e-05,
      "loss": 0.0001,
      "step": 1290
    },
    {
      "epoch": 2.7611940298507465,
      "grad_norm": 0.011952316388487816,
      "learning_rate": 2.238805970149254e-05,
      "loss": 0.0001,
      "step": 1295
    },
    {
      "epoch": 2.771855010660981,
      "grad_norm": 0.0004125136765651405,
      "learning_rate": 2.2281449893390194e-05,
      "loss": 0.0017,
      "step": 1300
    },
    {
      "epoch": 2.7825159914712154,
      "grad_norm": 0.0005646991194225848,
      "learning_rate": 2.217484008528785e-05,
      "loss": 0.0001,
      "step": 1305
    },
    {
      "epoch": 2.79317697228145,
      "grad_norm": 0.0030557531863451004,
      "learning_rate": 2.2068230277185503e-05,
      "loss": 0.0001,
      "step": 1310
    },
    {
      "epoch": 2.8038379530916844,
      "grad_norm": 0.018145538866519928,
      "learning_rate": 2.1961620469083158e-05,
      "loss": 0.0001,
      "step": 1315
    },
    {
      "epoch": 2.814498933901919,
      "grad_norm": 0.00047241643187589943,
      "learning_rate": 2.1855010660980813e-05,
      "loss": 0.0001,
      "step": 1320
    },
    {
      "epoch": 2.8251599147121533,
      "grad_norm": 0.0017407148843631148,
      "learning_rate": 2.1748400852878467e-05,
      "loss": 0.0001,
      "step": 1325
    },
    {
      "epoch": 2.835820895522388,
      "grad_norm": 0.0005856688367202878,
      "learning_rate": 2.164179104477612e-05,
      "loss": 0.0,
      "step": 1330
    },
    {
      "epoch": 2.8464818763326227,
      "grad_norm": 0.17735929787158966,
      "learning_rate": 2.1535181236673773e-05,
      "loss": 0.0001,
      "step": 1335
    },
    {
      "epoch": 2.857142857142857,
      "grad_norm": 0.0005312514258548617,
      "learning_rate": 2.1428571428571428e-05,
      "loss": 0.0002,
      "step": 1340
    },
    {
      "epoch": 2.8678038379530917,
      "grad_norm": 0.0004337857535574585,
      "learning_rate": 2.1321961620469083e-05,
      "loss": 0.0002,
      "step": 1345
    },
    {
      "epoch": 2.878464818763326,
      "grad_norm": 0.0004275246465113014,
      "learning_rate": 2.1215351812366738e-05,
      "loss": 0.0001,
      "step": 1350
    },
    {
      "epoch": 2.8891257995735606,
      "grad_norm": 0.0005103625589981675,
      "learning_rate": 2.1108742004264392e-05,
      "loss": 0.0001,
      "step": 1355
    },
    {
      "epoch": 2.8997867803837956,
      "grad_norm": 0.00044624603469856083,
      "learning_rate": 2.1002132196162047e-05,
      "loss": 0.01,
      "step": 1360
    },
    {
      "epoch": 2.91044776119403,
      "grad_norm": 0.0007846157532185316,
      "learning_rate": 2.0895522388059702e-05,
      "loss": 0.0001,
      "step": 1365
    },
    {
      "epoch": 2.9211087420042645,
      "grad_norm": 0.00048564866301603615,
      "learning_rate": 2.0788912579957357e-05,
      "loss": 0.0001,
      "step": 1370
    },
    {
      "epoch": 2.931769722814499,
      "grad_norm": 0.0004862641217187047,
      "learning_rate": 2.068230277185501e-05,
      "loss": 0.0002,
      "step": 1375
    },
    {
      "epoch": 2.9424307036247335,
      "grad_norm": 0.00044588197488337755,
      "learning_rate": 2.0575692963752666e-05,
      "loss": 0.0,
      "step": 1380
    },
    {
      "epoch": 2.953091684434968,
      "grad_norm": 0.0007043131627142429,
      "learning_rate": 2.046908315565032e-05,
      "loss": 0.0001,
      "step": 1385
    },
    {
      "epoch": 2.9637526652452024,
      "grad_norm": 0.0005226381472311914,
      "learning_rate": 2.0362473347547976e-05,
      "loss": 0.0001,
      "step": 1390
    },
    {
      "epoch": 2.974413646055437,
      "grad_norm": 0.0008309752447530627,
      "learning_rate": 2.025586353944563e-05,
      "loss": 0.0001,
      "step": 1395
    },
    {
      "epoch": 2.9850746268656714,
      "grad_norm": 0.0011215871199965477,
      "learning_rate": 2.0149253731343285e-05,
      "loss": 0.0001,
      "step": 1400
    },
    {
      "epoch": 2.9957356076759063,
      "grad_norm": 0.024289239197969437,
      "learning_rate": 2.004264392324094e-05,
      "loss": 0.0001,
      "step": 1405
    },
    {
      "epoch": 3.0,
      "eval_accuracy": 0.9826666666666667,
      "eval_loss": 0.08699585497379303,
      "eval_runtime": 16.9165,
      "eval_samples_per_second": 221.678,
      "eval_steps_per_second": 6.975,
      "step": 1407
    },
    {
      "epoch": 3.0063965884861408,
      "grad_norm": 0.001167010748758912,
      "learning_rate": 1.9936034115138594e-05,
      "loss": 0.0001,
      "step": 1410
    },
    {
      "epoch": 3.0170575692963753,
      "grad_norm": 0.000434078712714836,
      "learning_rate": 1.982942430703625e-05,
      "loss": 0.0,
      "step": 1415
    },
    {
      "epoch": 3.0277185501066097,
      "grad_norm": 0.0004404344072099775,
      "learning_rate": 1.9722814498933904e-05,
      "loss": 0.0001,
      "step": 1420
    },
    {
      "epoch": 3.038379530916844,
      "grad_norm": 0.0010413752170279622,
      "learning_rate": 1.961620469083156e-05,
      "loss": 0.0001,
      "step": 1425
    },
    {
      "epoch": 3.0490405117270787,
      "grad_norm": 0.00230170926079154,
      "learning_rate": 1.9509594882729213e-05,
      "loss": 0.0001,
      "step": 1430
    },
    {
      "epoch": 3.0597014925373136,
      "grad_norm": 0.0011793397134169936,
      "learning_rate": 1.9402985074626868e-05,
      "loss": 0.0001,
      "step": 1435
    },
    {
      "epoch": 3.070362473347548,
      "grad_norm": 0.000488606165163219,
      "learning_rate": 1.9296375266524523e-05,
      "loss": 0.0,
      "step": 1440
    },
    {
      "epoch": 3.0810234541577826,
      "grad_norm": 0.00046734214993193746,
      "learning_rate": 1.9189765458422178e-05,
      "loss": 0.0001,
      "step": 1445
    },
    {
      "epoch": 3.091684434968017,
      "grad_norm": 0.00043289692257530987,
      "learning_rate": 1.9083155650319832e-05,
      "loss": 0.0001,
      "step": 1450
    },
    {
      "epoch": 3.1023454157782515,
      "grad_norm": 0.00043649112922139466,
      "learning_rate": 1.8976545842217487e-05,
      "loss": 0.0001,
      "step": 1455
    },
    {
      "epoch": 3.113006396588486,
      "grad_norm": 0.0006658025085926056,
      "learning_rate": 1.8869936034115142e-05,
      "loss": 0.0001,
      "step": 1460
    },
    {
      "epoch": 3.1236673773987205,
      "grad_norm": 0.000517833570484072,
      "learning_rate": 1.8763326226012797e-05,
      "loss": 0.0,
      "step": 1465
    },
    {
      "epoch": 3.1343283582089554,
      "grad_norm": 0.0004990474553778768,
      "learning_rate": 1.865671641791045e-05,
      "loss": 0.0,
      "step": 1470
    },
    {
      "epoch": 3.14498933901919,
      "grad_norm": 0.00046692078467458487,
      "learning_rate": 1.8550106609808106e-05,
      "loss": 0.0001,
      "step": 1475
    },
    {
      "epoch": 3.1556503198294243,
      "grad_norm": 0.05067034438252449,
      "learning_rate": 1.8443496801705757e-05,
      "loss": 0.0021,
      "step": 1480
    },
    {
      "epoch": 3.166311300639659,
      "grad_norm": 0.000601542298682034,
      "learning_rate": 1.8336886993603412e-05,
      "loss": 0.0001,
      "step": 1485
    },
    {
      "epoch": 3.1769722814498933,
      "grad_norm": 0.0008663756307214499,
      "learning_rate": 1.8230277185501067e-05,
      "loss": 0.0001,
      "step": 1490
    },
    {
      "epoch": 3.1876332622601278,
      "grad_norm": 0.002149822423234582,
      "learning_rate": 1.812366737739872e-05,
      "loss": 0.0001,
      "step": 1495
    },
    {
      "epoch": 3.1982942430703627,
      "grad_norm": 0.0017691924003884196,
      "learning_rate": 1.8017057569296376e-05,
      "loss": 0.0001,
      "step": 1500
    },
    {
      "epoch": 3.208955223880597,
      "grad_norm": 0.0004495820903684944,
      "learning_rate": 1.791044776119403e-05,
      "loss": 0.0001,
      "step": 1505
    },
    {
      "epoch": 3.2196162046908317,
      "grad_norm": 0.0004403364146128297,
      "learning_rate": 1.7803837953091686e-05,
      "loss": 0.0,
      "step": 1510
    },
    {
      "epoch": 3.230277185501066,
      "grad_norm": 0.00045741343637928367,
      "learning_rate": 1.769722814498934e-05,
      "loss": 0.0,
      "step": 1515
    },
    {
      "epoch": 3.2409381663113006,
      "grad_norm": 0.000488805235363543,
      "learning_rate": 1.7590618336886995e-05,
      "loss": 0.0,
      "step": 1520
    },
    {
      "epoch": 3.251599147121535,
      "grad_norm": 0.00039901811396703124,
      "learning_rate": 1.7484008528784647e-05,
      "loss": 0.0001,
      "step": 1525
    },
    {
      "epoch": 3.2622601279317696,
      "grad_norm": 0.000695625611115247,
      "learning_rate": 1.73773987206823e-05,
      "loss": 0.0,
      "step": 1530
    },
    {
      "epoch": 3.272921108742004,
      "grad_norm": 0.0007208099705167115,
      "learning_rate": 1.7270788912579956e-05,
      "loss": 0.0,
      "step": 1535
    },
    {
      "epoch": 3.283582089552239,
      "grad_norm": 0.0008213461842387915,
      "learning_rate": 1.716417910447761e-05,
      "loss": 0.0001,
      "step": 1540
    },
    {
      "epoch": 3.2942430703624734,
      "grad_norm": 0.0012995878932997584,
      "learning_rate": 1.7057569296375266e-05,
      "loss": 0.0001,
      "step": 1545
    },
    {
      "epoch": 3.304904051172708,
      "grad_norm": 0.002836803672835231,
      "learning_rate": 1.695095948827292e-05,
      "loss": 0.0001,
      "step": 1550
    },
    {
      "epoch": 3.3155650319829424,
      "grad_norm": 0.000432594126323238,
      "learning_rate": 1.6844349680170575e-05,
      "loss": 0.0001,
      "step": 1555
    },
    {
      "epoch": 3.326226012793177,
      "grad_norm": 0.00044748326763510704,
      "learning_rate": 1.673773987206823e-05,
      "loss": 0.0,
      "step": 1560
    },
    {
      "epoch": 3.3368869936034113,
      "grad_norm": 0.0004975043702870607,
      "learning_rate": 1.6631130063965885e-05,
      "loss": 0.0,
      "step": 1565
    },
    {
      "epoch": 3.3475479744136463,
      "grad_norm": 0.00048893375787884,
      "learning_rate": 1.652452025586354e-05,
      "loss": 0.0,
      "step": 1570
    },
    {
      "epoch": 3.3582089552238807,
      "grad_norm": 0.000968233565799892,
      "learning_rate": 1.6417910447761194e-05,
      "loss": 0.0001,
      "step": 1575
    },
    {
      "epoch": 3.368869936034115,
      "grad_norm": 0.0003993576974608004,
      "learning_rate": 1.631130063965885e-05,
      "loss": 0.0,
      "step": 1580
    },
    {
      "epoch": 3.3795309168443497,
      "grad_norm": 0.0005020779790356755,
      "learning_rate": 1.6204690831556504e-05,
      "loss": 0.0001,
      "step": 1585
    },
    {
      "epoch": 3.390191897654584,
      "grad_norm": 0.0005774927558377385,
      "learning_rate": 1.6098081023454158e-05,
      "loss": 0.0,
      "step": 1590
    },
    {
      "epoch": 3.4008528784648187,
      "grad_norm": 0.0005787923000752926,
      "learning_rate": 1.5991471215351813e-05,
      "loss": 0.0002,
      "step": 1595
    },
    {
      "epoch": 3.411513859275053,
      "grad_norm": 0.0008096958626993,
      "learning_rate": 1.5884861407249468e-05,
      "loss": 0.0001,
      "step": 1600
    },
    {
      "epoch": 3.4221748400852876,
      "grad_norm": 0.00038943535764701664,
      "learning_rate": 1.5778251599147122e-05,
      "loss": 0.0001,
      "step": 1605
    },
    {
      "epoch": 3.4328358208955225,
      "grad_norm": 0.0005638746661134064,
      "learning_rate": 1.5671641791044777e-05,
      "loss": 0.0,
      "step": 1610
    },
    {
      "epoch": 3.443496801705757,
      "grad_norm": 0.00037838646676391363,
      "learning_rate": 1.5565031982942432e-05,
      "loss": 0.0001,
      "step": 1615
    },
    {
      "epoch": 3.4541577825159915,
      "grad_norm": 0.0005717452149838209,
      "learning_rate": 1.5458422174840087e-05,
      "loss": 0.0001,
      "step": 1620
    },
    {
      "epoch": 3.464818763326226,
      "grad_norm": 0.0006832830840721726,
      "learning_rate": 1.535181236673774e-05,
      "loss": 0.0001,
      "step": 1625
    },
    {
      "epoch": 3.4754797441364604,
      "grad_norm": 0.0011972938664257526,
      "learning_rate": 1.5245202558635396e-05,
      "loss": 0.0,
      "step": 1630
    },
    {
      "epoch": 3.486140724946695,
      "grad_norm": 0.0004888517432846129,
      "learning_rate": 1.5138592750533051e-05,
      "loss": 0.0,
      "step": 1635
    },
    {
      "epoch": 3.49680170575693,
      "grad_norm": 0.0013256361708045006,
      "learning_rate": 1.5031982942430706e-05,
      "loss": 0.0001,
      "step": 1640
    },
    {
      "epoch": 3.5074626865671643,
      "grad_norm": 0.0010635104263201356,
      "learning_rate": 1.4925373134328357e-05,
      "loss": 0.0001,
      "step": 1645
    },
    {
      "epoch": 3.518123667377399,
      "grad_norm": 0.0019707968458533287,
      "learning_rate": 1.4818763326226012e-05,
      "loss": 0.0001,
      "step": 1650
    },
    {
      "epoch": 3.5287846481876333,
      "grad_norm": 0.0007115269545465708,
      "learning_rate": 1.4712153518123666e-05,
      "loss": 0.0,
      "step": 1655
    },
    {
      "epoch": 3.5394456289978677,
      "grad_norm": 0.00043455700506456196,
      "learning_rate": 1.4605543710021321e-05,
      "loss": 0.0,
      "step": 1660
    },
    {
      "epoch": 3.550106609808102,
      "grad_norm": 0.00057604955509305,
      "learning_rate": 1.4498933901918976e-05,
      "loss": 0.0001,
      "step": 1665
    },
    {
      "epoch": 3.5607675906183367,
      "grad_norm": 0.00047078431816771626,
      "learning_rate": 1.439232409381663e-05,
      "loss": 0.0001,
      "step": 1670
    },
    {
      "epoch": 3.571428571428571,
      "grad_norm": 0.0009252798627130687,
      "learning_rate": 1.4285714285714285e-05,
      "loss": 0.0001,
      "step": 1675
    },
    {
      "epoch": 3.582089552238806,
      "grad_norm": 0.0003781556442845613,
      "learning_rate": 1.417910447761194e-05,
      "loss": 0.0,
      "step": 1680
    },
    {
      "epoch": 3.5927505330490406,
      "grad_norm": 0.0014854730106890202,
      "learning_rate": 1.4072494669509595e-05,
      "loss": 0.0001,
      "step": 1685
    },
    {
      "epoch": 3.603411513859275,
      "grad_norm": 0.0004669760528486222,
      "learning_rate": 1.396588486140725e-05,
      "loss": 0.0,
      "step": 1690
    },
    {
      "epoch": 3.6140724946695095,
      "grad_norm": 0.00039115696563385427,
      "learning_rate": 1.3859275053304904e-05,
      "loss": 0.0,
      "step": 1695
    },
    {
      "epoch": 3.624733475479744,
      "grad_norm": 0.0019996201153844595,
      "learning_rate": 1.3752665245202559e-05,
      "loss": 0.0,
      "step": 1700
    },
    {
      "epoch": 3.635394456289979,
      "grad_norm": 0.002407630905508995,
      "learning_rate": 1.3646055437100214e-05,
      "loss": 0.0,
      "step": 1705
    },
    {
      "epoch": 3.6460554371002134,
      "grad_norm": 0.00037330054328776896,
      "learning_rate": 1.3539445628997869e-05,
      "loss": 0.0001,
      "step": 1710
    },
    {
      "epoch": 3.656716417910448,
      "grad_norm": 0.0032050053123384714,
      "learning_rate": 1.3432835820895523e-05,
      "loss": 0.0001,
      "step": 1715
    },
    {
      "epoch": 3.6673773987206824,
      "grad_norm": 0.0003663256939034909,
      "learning_rate": 1.3326226012793178e-05,
      "loss": 0.0001,
      "step": 1720
    },
    {
      "epoch": 3.678038379530917,
      "grad_norm": 0.0006334013887681067,
      "learning_rate": 1.3219616204690833e-05,
      "loss": 0.0,
      "step": 1725
    },
    {
      "epoch": 3.6886993603411513,
      "grad_norm": 0.0009968049125745893,
      "learning_rate": 1.3113006396588488e-05,
      "loss": 0.0,
      "step": 1730
    },
    {
      "epoch": 3.699360341151386,
      "grad_norm": 0.000671451271045953,
      "learning_rate": 1.3006396588486142e-05,
      "loss": 0.0081,
      "step": 1735
    },
    {
      "epoch": 3.7100213219616203,
      "grad_norm": 0.0004945637192577124,
      "learning_rate": 1.2899786780383797e-05,
      "loss": 0.0,
      "step": 1740
    },
    {
      "epoch": 3.7206823027718547,
      "grad_norm": 0.0004414471914060414,
      "learning_rate": 1.2793176972281452e-05,
      "loss": 0.0001,
      "step": 1745
    },
    {
      "epoch": 3.7313432835820897,
      "grad_norm": 0.0008040583925321698,
      "learning_rate": 1.2686567164179105e-05,
      "loss": 0.0,
      "step": 1750
    },
    {
      "epoch": 3.742004264392324,
      "grad_norm": 0.0006331412587314844,
      "learning_rate": 1.257995735607676e-05,
      "loss": 0.0,
      "step": 1755
    },
    {
      "epoch": 3.7526652452025586,
      "grad_norm": 0.0004544199618976563,
      "learning_rate": 1.2473347547974414e-05,
      "loss": 0.0001,
      "step": 1760
    },
    {
      "epoch": 3.763326226012793,
      "grad_norm": 0.0005459372187033296,
      "learning_rate": 1.2366737739872069e-05,
      "loss": 0.0,
      "step": 1765
    },
    {
      "epoch": 3.7739872068230276,
      "grad_norm": 0.00038835889426991343,
      "learning_rate": 1.2260127931769722e-05,
      "loss": 0.0,
      "step": 1770
    },
    {
      "epoch": 3.7846481876332625,
      "grad_norm": 0.0005388223216868937,
      "learning_rate": 1.2153518123667377e-05,
      "loss": 0.0001,
      "step": 1775
    },
    {
      "epoch": 3.795309168443497,
      "grad_norm": 0.0003705797716975212,
      "learning_rate": 1.2046908315565032e-05,
      "loss": 0.0001,
      "step": 1780
    },
    {
      "epoch": 3.8059701492537314,
      "grad_norm": 0.0003858578857034445,
      "learning_rate": 1.1940298507462686e-05,
      "loss": 0.0,
      "step": 1785
    },
    {
      "epoch": 3.816631130063966,
      "grad_norm": 0.0025148168206214905,
      "learning_rate": 1.1833688699360341e-05,
      "loss": 0.0,
      "step": 1790
    },
    {
      "epoch": 3.8272921108742004,
      "grad_norm": 0.00044439485645852983,
      "learning_rate": 1.1727078891257996e-05,
      "loss": 0.0,
      "step": 1795
    },
    {
      "epoch": 3.837953091684435,
      "grad_norm": 0.002464845310896635,
      "learning_rate": 1.162046908315565e-05,
      "loss": 0.0,
      "step": 1800
    },
    {
      "epoch": 3.8486140724946694,
      "grad_norm": 0.0005794114549644291,
      "learning_rate": 1.1513859275053305e-05,
      "loss": 0.0022,
      "step": 1805
    },
    {
      "epoch": 3.859275053304904,
      "grad_norm": 0.0005555557436309755,
      "learning_rate": 1.140724946695096e-05,
      "loss": 0.0,
      "step": 1810
    },
    {
      "epoch": 3.8699360341151388,
      "grad_norm": 0.000977195450104773,
      "learning_rate": 1.1300639658848615e-05,
      "loss": 0.0001,
      "step": 1815
    },
    {
      "epoch": 3.8805970149253732,
      "grad_norm": 0.0018520756857469678,
      "learning_rate": 1.119402985074627e-05,
      "loss": 0.0001,
      "step": 1820
    },
    {
      "epoch": 3.8912579957356077,
      "grad_norm": 0.0008402444655075669,
      "learning_rate": 1.1087420042643924e-05,
      "loss": 0.0001,
      "step": 1825
    },
    {
      "epoch": 3.901918976545842,
      "grad_norm": 0.0008242878247983754,
      "learning_rate": 1.0980810234541579e-05,
      "loss": 0.0,
      "step": 1830
    },
    {
      "epoch": 3.9125799573560767,
      "grad_norm": 0.003950777929276228,
      "learning_rate": 1.0874200426439234e-05,
      "loss": 0.0001,
      "step": 1835
    },
    {
      "epoch": 3.923240938166311,
      "grad_norm": 0.0004298113053664565,
      "learning_rate": 1.0767590618336887e-05,
      "loss": 0.0001,
      "step": 1840
    },
    {
      "epoch": 3.933901918976546,
      "grad_norm": 0.0006574545404873788,
      "learning_rate": 1.0660980810234541e-05,
      "loss": 0.0,
      "step": 1845
    },
    {
      "epoch": 3.9445628997867805,
      "grad_norm": 0.00034868618240579963,
      "learning_rate": 1.0554371002132196e-05,
      "loss": 0.0,
      "step": 1850
    },
    {
      "epoch": 3.955223880597015,
      "grad_norm": 0.002188840415328741,
      "learning_rate": 1.0447761194029851e-05,
      "loss": 0.0,
      "step": 1855
    },
    {
      "epoch": 3.9658848614072495,
      "grad_norm": 0.0004396145523060113,
      "learning_rate": 1.0341151385927506e-05,
      "loss": 0.0,
      "step": 1860
    },
    {
      "epoch": 3.976545842217484,
      "grad_norm": 0.0005642068572342396,
      "learning_rate": 1.023454157782516e-05,
      "loss": 0.0,
      "step": 1865
    },
    {
      "epoch": 3.9872068230277184,
      "grad_norm": 0.0003875910770148039,
      "learning_rate": 1.0127931769722815e-05,
      "loss": 0.0,
      "step": 1870
    },
    {
      "epoch": 3.997867803837953,
      "grad_norm": 0.00042787310667335987,
      "learning_rate": 1.002132196162047e-05,
      "loss": 0.0,
      "step": 1875
    },
    {
      "epoch": 4.0,
      "eval_accuracy": 0.9845333333333334,
      "eval_loss": 0.08250279724597931,
      "eval_runtime": 16.8698,
      "eval_samples_per_second": 222.29,
      "eval_steps_per_second": 6.995,
      "step": 1876
    },
    {
      "epoch": 4.008528784648187,
      "grad_norm": 0.00038766584475524724,
      "learning_rate": 9.914712153518125e-06,
      "loss": 0.0,
      "step": 1880
    },
    {
      "epoch": 4.019189765458422,
      "grad_norm": 0.00038398653850890696,
      "learning_rate": 9.80810234541578e-06,
      "loss": 0.0,
      "step": 1885
    },
    {
      "epoch": 4.029850746268656,
      "grad_norm": 0.00045901568955741823,
      "learning_rate": 9.701492537313434e-06,
      "loss": 0.0,
      "step": 1890
    },
    {
      "epoch": 4.040511727078891,
      "grad_norm": 0.0007242396823130548,
      "learning_rate": 9.594882729211089e-06,
      "loss": 0.0,
      "step": 1895
    },
    {
      "epoch": 4.051172707889126,
      "grad_norm": 0.0006220832001417875,
      "learning_rate": 9.488272921108744e-06,
      "loss": 0.0001,
      "step": 1900
    },
    {
      "epoch": 4.061833688699361,
      "grad_norm": 0.00045091615174897015,
      "learning_rate": 9.381663113006398e-06,
      "loss": 0.0,
      "step": 1905
    },
    {
      "epoch": 4.072494669509595,
      "grad_norm": 0.00039596817805431783,
      "learning_rate": 9.275053304904053e-06,
      "loss": 0.0,
      "step": 1910
    },
    {
      "epoch": 4.08315565031983,
      "grad_norm": 0.002194940112531185,
      "learning_rate": 9.168443496801706e-06,
      "loss": 0.0001,
      "step": 1915
    },
    {
      "epoch": 4.093816631130064,
      "grad_norm": 0.0003560879558790475,
      "learning_rate": 9.06183368869936e-06,
      "loss": 0.0,
      "step": 1920
    },
    {
      "epoch": 4.104477611940299,
      "grad_norm": 0.0004938474157825112,
      "learning_rate": 8.955223880597016e-06,
      "loss": 0.0,
      "step": 1925
    },
    {
      "epoch": 4.115138592750533,
      "grad_norm": 0.00036476305103860795,
      "learning_rate": 8.84861407249467e-06,
      "loss": 0.0,
      "step": 1930
    },
    {
      "epoch": 4.1257995735607675,
      "grad_norm": 0.0008207400096580386,
      "learning_rate": 8.742004264392323e-06,
      "loss": 0.0,
      "step": 1935
    },
    {
      "epoch": 4.136460554371002,
      "grad_norm": 0.00038536402280442417,
      "learning_rate": 8.635394456289978e-06,
      "loss": 0.0,
      "step": 1940
    },
    {
      "epoch": 4.1471215351812365,
      "grad_norm": 0.0004361090250313282,
      "learning_rate": 8.528784648187633e-06,
      "loss": 0.0,
      "step": 1945
    },
    {
      "epoch": 4.157782515991471,
      "grad_norm": 0.0003583457146305591,
      "learning_rate": 8.422174840085288e-06,
      "loss": 0.0,
      "step": 1950
    },
    {
      "epoch": 4.1684434968017055,
      "grad_norm": 0.00043614330934360623,
      "learning_rate": 8.315565031982942e-06,
      "loss": 0.0001,
      "step": 1955
    },
    {
      "epoch": 4.17910447761194,
      "grad_norm": 0.0005300487973727286,
      "learning_rate": 8.208955223880597e-06,
      "loss": 0.0,
      "step": 1960
    },
    {
      "epoch": 4.189765458422174,
      "grad_norm": 0.0003390677156858146,
      "learning_rate": 8.102345415778252e-06,
      "loss": 0.0,
      "step": 1965
    },
    {
      "epoch": 4.20042643923241,
      "grad_norm": 0.004341974854469299,
      "learning_rate": 7.995735607675907e-06,
      "loss": 0.0,
      "step": 1970
    },
    {
      "epoch": 4.211087420042644,
      "grad_norm": 0.0005244372296147048,
      "learning_rate": 7.889125799573561e-06,
      "loss": 0.0,
      "step": 1975
    },
    {
      "epoch": 4.221748400852879,
      "grad_norm": 0.00035956804640591145,
      "learning_rate": 7.782515991471216e-06,
      "loss": 0.0,
      "step": 1980
    },
    {
      "epoch": 4.232409381663113,
      "grad_norm": 0.002688586013391614,
      "learning_rate": 7.67590618336887e-06,
      "loss": 0.0001,
      "step": 1985
    },
    {
      "epoch": 4.243070362473348,
      "grad_norm": 0.0004000146291218698,
      "learning_rate": 7.5692963752665255e-06,
      "loss": 0.0001,
      "step": 1990
    },
    {
      "epoch": 4.253731343283582,
      "grad_norm": 0.0003765385481528938,
      "learning_rate": 7.4626865671641785e-06,
      "loss": 0.0,
      "step": 1995
    },
    {
      "epoch": 4.264392324093817,
      "grad_norm": 0.000515621155500412,
      "learning_rate": 7.356076759061833e-06,
      "loss": 0.0,
      "step": 2000
    },
    {
      "epoch": 4.275053304904051,
      "grad_norm": 0.0009000042919069529,
      "learning_rate": 7.249466950959488e-06,
      "loss": 0.0,
      "step": 2005
    },
    {
      "epoch": 4.285714285714286,
      "grad_norm": 0.0013124350225552917,
      "learning_rate": 7.142857142857143e-06,
      "loss": 0.0,
      "step": 2010
    },
    {
      "epoch": 4.29637526652452,
      "grad_norm": 0.0018238076008856297,
      "learning_rate": 7.0362473347547975e-06,
      "loss": 0.0001,
      "step": 2015
    },
    {
      "epoch": 4.3070362473347545,
      "grad_norm": 0.0013175200438126922,
      "learning_rate": 6.929637526652452e-06,
      "loss": 0.0,
      "step": 2020
    },
    {
      "epoch": 4.317697228144989,
      "grad_norm": 0.053842801600694656,
      "learning_rate": 6.823027718550107e-06,
      "loss": 0.002,
      "step": 2025
    },
    {
      "epoch": 4.3283582089552235,
      "grad_norm": 0.0003529018722474575,
      "learning_rate": 6.716417910447762e-06,
      "loss": 0.0,
      "step": 2030
    },
    {
      "epoch": 4.339019189765459,
      "grad_norm": 0.001788545399904251,
      "learning_rate": 6.609808102345416e-06,
      "loss": 0.0,
      "step": 2035
    },
    {
      "epoch": 4.349680170575693,
      "grad_norm": 0.0003504894848447293,
      "learning_rate": 6.503198294243071e-06,
      "loss": 0.0,
      "step": 2040
    },
    {
      "epoch": 4.360341151385928,
      "grad_norm": 0.00036186777288094163,
      "learning_rate": 6.396588486140726e-06,
      "loss": 0.0,
      "step": 2045
    },
    {
      "epoch": 4.371002132196162,
      "grad_norm": 0.0006138259777799249,
      "learning_rate": 6.28997867803838e-06,
      "loss": 0.0,
      "step": 2050
    },
    {
      "epoch": 4.381663113006397,
      "grad_norm": 0.0005072400090284646,
      "learning_rate": 6.1833688699360345e-06,
      "loss": 0.0,
      "step": 2055
    },
    {
      "epoch": 4.392324093816631,
      "grad_norm": 0.0008174916147254407,
      "learning_rate": 6.076759061833688e-06,
      "loss": 0.0,
      "step": 2060
    },
    {
      "epoch": 4.402985074626866,
      "grad_norm": 0.00040691232425160706,
      "learning_rate": 5.970149253731343e-06,
      "loss": 0.0,
      "step": 2065
    },
    {
      "epoch": 4.4136460554371,
      "grad_norm": 0.0010525727411732078,
      "learning_rate": 5.863539445628998e-06,
      "loss": 0.0,
      "step": 2070
    },
    {
      "epoch": 4.424307036247335,
      "grad_norm": 0.0006079195300117135,
      "learning_rate": 5.756929637526653e-06,
      "loss": 0.0,
      "step": 2075
    },
    {
      "epoch": 4.434968017057569,
      "grad_norm": 0.00035886032856069505,
      "learning_rate": 5.650319829424307e-06,
      "loss": 0.0,
      "step": 2080
    },
    {
      "epoch": 4.445628997867804,
      "grad_norm": 0.0007788026123307645,
      "learning_rate": 5.543710021321962e-06,
      "loss": 0.0,
      "step": 2085
    },
    {
      "epoch": 4.456289978678038,
      "grad_norm": 0.0007285819156095386,
      "learning_rate": 5.437100213219617e-06,
      "loss": 0.0,
      "step": 2090
    },
    {
      "epoch": 4.466950959488273,
      "grad_norm": 0.0004360245366115123,
      "learning_rate": 5.330490405117271e-06,
      "loss": 0.0,
      "step": 2095
    },
    {
      "epoch": 4.477611940298507,
      "grad_norm": 0.0007146059651859105,
      "learning_rate": 5.2238805970149255e-06,
      "loss": 0.0,
      "step": 2100
    },
    {
      "epoch": 4.4882729211087415,
      "grad_norm": 0.00039442122215405107,
      "learning_rate": 5.11727078891258e-06,
      "loss": 0.0,
      "step": 2105
    },
    {
      "epoch": 4.498933901918977,
      "grad_norm": 0.00037271485780365765,
      "learning_rate": 5.010660980810235e-06,
      "loss": 0.0,
      "step": 2110
    },
    {
      "epoch": 4.509594882729211,
      "grad_norm": 0.0004216345550958067,
      "learning_rate": 4.90405117270789e-06,
      "loss": 0.0,
      "step": 2115
    },
    {
      "epoch": 4.520255863539446,
      "grad_norm": 0.0010674895020201802,
      "learning_rate": 4.797441364605544e-06,
      "loss": 0.0,
      "step": 2120
    },
    {
      "epoch": 4.53091684434968,
      "grad_norm": 0.00036971975350752473,
      "learning_rate": 4.690831556503199e-06,
      "loss": 0.0,
      "step": 2125
    },
    {
      "epoch": 4.541577825159915,
      "grad_norm": 0.00036792727769352496,
      "learning_rate": 4.584221748400853e-06,
      "loss": 0.0001,
      "step": 2130
    },
    {
      "epoch": 4.552238805970149,
      "grad_norm": 0.0004503585514612496,
      "learning_rate": 4.477611940298508e-06,
      "loss": 0.0,
      "step": 2135
    },
    {
      "epoch": 4.562899786780384,
      "grad_norm": 0.0008292795973829925,
      "learning_rate": 4.371002132196162e-06,
      "loss": 0.0,
      "step": 2140
    },
    {
      "epoch": 4.573560767590618,
      "grad_norm": 0.0024220841005444527,
      "learning_rate": 4.264392324093816e-06,
      "loss": 0.0001,
      "step": 2145
    },
    {
      "epoch": 4.584221748400853,
      "grad_norm": 0.00035092837060801685,
      "learning_rate": 4.157782515991471e-06,
      "loss": 0.0,
      "step": 2150
    },
    {
      "epoch": 4.594882729211087,
      "grad_norm": 0.0006493948749266565,
      "learning_rate": 4.051172707889126e-06,
      "loss": 0.0,
      "step": 2155
    },
    {
      "epoch": 4.605543710021322,
      "grad_norm": 0.0008510401821695268,
      "learning_rate": 3.944562899786781e-06,
      "loss": 0.0,
      "step": 2160
    },
    {
      "epoch": 4.616204690831556,
      "grad_norm": 0.00036558014107868075,
      "learning_rate": 3.837953091684435e-06,
      "loss": 0.0001,
      "step": 2165
    },
    {
      "epoch": 4.6268656716417915,
      "grad_norm": 0.0018046260811388493,
      "learning_rate": 3.7313432835820893e-06,
      "loss": 0.0,
      "step": 2170
    },
    {
      "epoch": 4.637526652452026,
      "grad_norm": 0.0003468076465651393,
      "learning_rate": 3.624733475479744e-06,
      "loss": 0.0,
      "step": 2175
    },
    {
      "epoch": 4.6481876332622605,
      "grad_norm": 0.0004059917700942606,
      "learning_rate": 3.5181236673773987e-06,
      "loss": 0.0,
      "step": 2180
    },
    {
      "epoch": 4.658848614072495,
      "grad_norm": 0.0007510407012887299,
      "learning_rate": 3.4115138592750535e-06,
      "loss": 0.0,
      "step": 2185
    },
    {
      "epoch": 4.669509594882729,
      "grad_norm": 0.0005453999037854373,
      "learning_rate": 3.304904051172708e-06,
      "loss": 0.0,
      "step": 2190
    },
    {
      "epoch": 4.680170575692964,
      "grad_norm": 0.00035293426481075585,
      "learning_rate": 3.198294243070363e-06,
      "loss": 0.0001,
      "step": 2195
    },
    {
      "epoch": 4.690831556503198,
      "grad_norm": 0.0004789367667399347,
      "learning_rate": 3.0916844349680173e-06,
      "loss": 0.0,
      "step": 2200
    },
    {
      "epoch": 4.701492537313433,
      "grad_norm": 0.0017639091238379478,
      "learning_rate": 2.9850746268656716e-06,
      "loss": 0.0001,
      "step": 2205
    },
    {
      "epoch": 4.712153518123667,
      "grad_norm": 0.0003756133664865047,
      "learning_rate": 2.8784648187633263e-06,
      "loss": 0.0,
      "step": 2210
    },
    {
      "epoch": 4.722814498933902,
      "grad_norm": 0.16620169579982758,
      "learning_rate": 2.771855010660981e-06,
      "loss": 0.0072,
      "step": 2215
    },
    {
      "epoch": 4.733475479744136,
      "grad_norm": 0.0007001258782111108,
      "learning_rate": 2.6652452025586354e-06,
      "loss": 0.0,
      "step": 2220
    },
    {
      "epoch": 4.744136460554371,
      "grad_norm": 0.0006105925422161818,
      "learning_rate": 2.55863539445629e-06,
      "loss": 0.0,
      "step": 2225
    },
    {
      "epoch": 4.754797441364605,
      "grad_norm": 0.000489444297272712,
      "learning_rate": 2.452025586353945e-06,
      "loss": 0.0,
      "step": 2230
    },
    {
      "epoch": 4.76545842217484,
      "grad_norm": 0.00037245533894747496,
      "learning_rate": 2.3454157782515996e-06,
      "loss": 0.0,
      "step": 2235
    },
    {
      "epoch": 4.776119402985074,
      "grad_norm": 0.0018702890956774354,
      "learning_rate": 2.238805970149254e-06,
      "loss": 0.0,
      "step": 2240
    },
    {
      "epoch": 4.786780383795309,
      "grad_norm": 0.00040366360917687416,
      "learning_rate": 2.132196162046908e-06,
      "loss": 0.0,
      "step": 2245
    },
    {
      "epoch": 4.797441364605544,
      "grad_norm": 0.0018535067792981863,
      "learning_rate": 2.025586353944563e-06,
      "loss": 0.0,
      "step": 2250
    },
    {
      "epoch": 4.8081023454157785,
      "grad_norm": 0.0004247617907822132,
      "learning_rate": 1.9189765458422177e-06,
      "loss": 0.0,
      "step": 2255
    },
    {
      "epoch": 4.818763326226013,
      "grad_norm": 0.0003945778007619083,
      "learning_rate": 1.812366737739872e-06,
      "loss": 0.0,
      "step": 2260
    },
    {
      "epoch": 4.8294243070362475,
      "grad_norm": 0.00040564127266407013,
      "learning_rate": 1.7057569296375267e-06,
      "loss": 0.0,
      "step": 2265
    },
    {
      "epoch": 4.840085287846482,
      "grad_norm": 0.00043385071330703795,
      "learning_rate": 1.5991471215351815e-06,
      "loss": 0.0,
      "step": 2270
    },
    {
      "epoch": 4.850746268656716,
      "grad_norm": 0.000637573073618114,
      "learning_rate": 1.4925373134328358e-06,
      "loss": 0.0,
      "step": 2275
    },
    {
      "epoch": 4.861407249466951,
      "grad_norm": 0.00034186383709311485,
      "learning_rate": 1.3859275053304905e-06,
      "loss": 0.0,
      "step": 2280
    },
    {
      "epoch": 4.872068230277185,
      "grad_norm": 0.00036258844193071127,
      "learning_rate": 1.279317697228145e-06,
      "loss": 0.0,
      "step": 2285
    },
    {
      "epoch": 4.88272921108742,
      "grad_norm": 0.0010802382603287697,
      "learning_rate": 1.1727078891257998e-06,
      "loss": 0.0,
      "step": 2290
    },
    {
      "epoch": 4.893390191897654,
      "grad_norm": 0.00040743834688328207,
      "learning_rate": 1.066098081023454e-06,
      "loss": 0.002,
      "step": 2295
    },
    {
      "epoch": 4.904051172707889,
      "grad_norm": 0.00040871876990422606,
      "learning_rate": 9.594882729211088e-07,
      "loss": 0.0,
      "step": 2300
    },
    {
      "epoch": 4.914712153518123,
      "grad_norm": 0.0005127965705469251,
      "learning_rate": 8.528784648187634e-07,
      "loss": 0.0,
      "step": 2305
    },
    {
      "epoch": 4.925373134328359,
      "grad_norm": 0.0003736602084245533,
      "learning_rate": 7.462686567164179e-07,
      "loss": 0.0,
      "step": 2310
    },
    {
      "epoch": 4.936034115138593,
      "grad_norm": 0.001036849687807262,
      "learning_rate": 6.396588486140725e-07,
      "loss": 0.0,
      "step": 2315
    },
    {
      "epoch": 4.946695095948828,
      "grad_norm": 0.000461318384623155,
      "learning_rate": 5.33049040511727e-07,
      "loss": 0.0,
      "step": 2320
    },
    {
      "epoch": 4.957356076759062,
      "grad_norm": 0.0003375077503733337,
      "learning_rate": 4.264392324093817e-07,
      "loss": 0.0,
      "step": 2325
    },
    {
      "epoch": 4.968017057569297,
      "grad_norm": 0.0007256953394971788,
      "learning_rate": 3.1982942430703626e-07,
      "loss": 0.0,
      "step": 2330
    },
    {
      "epoch": 4.978678038379531,
      "grad_norm": 0.0012693832395598292,
      "learning_rate": 2.1321961620469084e-07,
      "loss": 0.0,
      "step": 2335
    },
    {
      "epoch": 4.9893390191897655,
      "grad_norm": 0.00045820162631571293,
      "learning_rate": 1.0660980810234542e-07,
      "loss": 0.0001,
      "step": 2340
    },
    {
      "epoch": 5.0,
      "grad_norm": 0.0009981872281059623,
      "learning_rate": 0.0,
      "loss": 0.0,
      "step": 2345
    },
    {
      "epoch": 5.0,
      "eval_accuracy": 0.9845333333333334,
      "eval_loss": 0.08336344361305237,
      "eval_runtime": 16.6771,
      "eval_samples_per_second": 224.859,
      "eval_steps_per_second": 7.076,
      "step": 2345
    },
    {
      "epoch": 5.0,
      "step": 2345,
      "total_flos": 5.8118992210944e+18,
      "train_loss": 0.007418321467136025,
      "train_runtime": 781.1111,
      "train_samples_per_second": 96.017,
      "train_steps_per_second": 3.002
    }
  ],
  "logging_steps": 5,
  "max_steps": 2345,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 5,
  "save_steps": 500,
  "stateful_callbacks": {
    "TrainerControl": {
      "args": {
        "should_epoch_stop": false,
        "should_evaluate": false,
        "should_log": false,
        "should_save": true,
        "should_training_stop": true
      },
      "attributes": {}
    }
  },
  "total_flos": 5.8118992210944e+18,
  "train_batch_size": 32,
  "trial_name": null,
  "trial_params": null
}