{
  "best_metric": 0.996875,
  "best_model_checkpoint": "vit-msn-small-finetuned-alzheimers/checkpoint-765",
  "epoch": 48.888888888888886,
  "eval_steps": 500,
  "global_step": 1100,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.4444444444444444,
      "grad_norm": 8.449820518493652,
      "learning_rate": 4.5454545454545455e-06,
      "loss": 0.2587,
      "step": 10
    },
    {
      "epoch": 0.8888888888888888,
      "grad_norm": 13.680850982666016,
      "learning_rate": 9.090909090909091e-06,
      "loss": 0.2996,
      "step": 20
    },
    {
      "epoch": 0.9777777777777777,
      "eval_accuracy": 0.84375,
      "eval_loss": 0.38971763849258423,
      "eval_runtime": 3.5179,
      "eval_samples_per_second": 181.926,
      "eval_steps_per_second": 2.843,
      "step": 22
    },
    {
      "epoch": 1.3333333333333333,
      "grad_norm": 9.488574981689453,
      "learning_rate": 1.3636363636363637e-05,
      "loss": 0.4023,
      "step": 30
    },
    {
      "epoch": 1.7777777777777777,
      "grad_norm": 18.977561950683594,
      "learning_rate": 1.8181818181818182e-05,
      "loss": 0.3703,
      "step": 40
    },
    {
      "epoch": 2.0,
      "eval_accuracy": 0.859375,
      "eval_loss": 0.3594878911972046,
      "eval_runtime": 3.9024,
      "eval_samples_per_second": 164.001,
      "eval_steps_per_second": 2.563,
      "step": 45
    },
    {
      "epoch": 2.2222222222222223,
      "grad_norm": 11.33133602142334,
      "learning_rate": 2.272727272727273e-05,
      "loss": 0.3541,
      "step": 50
    },
    {
      "epoch": 2.6666666666666665,
      "grad_norm": 16.366662979125977,
      "learning_rate": 2.7272727272727273e-05,
      "loss": 0.3087,
      "step": 60
    },
    {
      "epoch": 2.977777777777778,
      "eval_accuracy": 0.8625,
      "eval_loss": 0.3777163326740265,
      "eval_runtime": 3.8599,
      "eval_samples_per_second": 165.808,
      "eval_steps_per_second": 2.591,
      "step": 67
    },
    {
      "epoch": 3.111111111111111,
      "grad_norm": 18.307331085205078,
      "learning_rate": 3.181818181818182e-05,
      "loss": 0.3195,
      "step": 70
    },
    {
      "epoch": 3.5555555555555554,
      "grad_norm": 42.80950164794922,
      "learning_rate": 3.6363636363636364e-05,
      "loss": 0.3483,
      "step": 80
    },
    {
      "epoch": 4.0,
      "grad_norm": 18.051124572753906,
      "learning_rate": 4.0909090909090915e-05,
      "loss": 0.486,
      "step": 90
    },
    {
      "epoch": 4.0,
      "eval_accuracy": 0.81875,
      "eval_loss": 0.4530211389064789,
      "eval_runtime": 3.6057,
      "eval_samples_per_second": 177.495,
      "eval_steps_per_second": 2.773,
      "step": 90
    },
    {
      "epoch": 4.444444444444445,
      "grad_norm": 15.844127655029297,
      "learning_rate": 4.545454545454546e-05,
      "loss": 0.3521,
      "step": 100
    },
    {
      "epoch": 4.888888888888889,
      "grad_norm": 11.87112808227539,
      "learning_rate": 5e-05,
      "loss": 0.3307,
      "step": 110
    },
    {
      "epoch": 4.977777777777778,
      "eval_accuracy": 0.8234375,
      "eval_loss": 0.45600825548171997,
      "eval_runtime": 3.572,
      "eval_samples_per_second": 179.171,
      "eval_steps_per_second": 2.8,
      "step": 112
    },
    {
      "epoch": 5.333333333333333,
      "grad_norm": 17.418668746948242,
      "learning_rate": 4.94949494949495e-05,
      "loss": 0.3404,
      "step": 120
    },
    {
      "epoch": 5.777777777777778,
      "grad_norm": 20.148906707763672,
      "learning_rate": 4.898989898989899e-05,
      "loss": 0.306,
      "step": 130
    },
    {
      "epoch": 6.0,
      "eval_accuracy": 0.8671875,
      "eval_loss": 0.3470742106437683,
      "eval_runtime": 3.5697,
      "eval_samples_per_second": 179.289,
      "eval_steps_per_second": 2.801,
      "step": 135
    },
    {
      "epoch": 6.222222222222222,
      "grad_norm": 11.451733589172363,
      "learning_rate": 4.848484848484849e-05,
      "loss": 0.2873,
      "step": 140
    },
    {
      "epoch": 6.666666666666667,
      "grad_norm": 15.43708610534668,
      "learning_rate": 4.797979797979798e-05,
      "loss": 0.3005,
      "step": 150
    },
    {
      "epoch": 6.977777777777778,
      "eval_accuracy": 0.8859375,
      "eval_loss": 0.3024638891220093,
      "eval_runtime": 3.8788,
      "eval_samples_per_second": 164.998,
      "eval_steps_per_second": 2.578,
      "step": 157
    },
    {
      "epoch": 7.111111111111111,
      "grad_norm": 20.696516036987305,
      "learning_rate": 4.7474747474747476e-05,
      "loss": 0.3206,
      "step": 160
    },
    {
      "epoch": 7.555555555555555,
      "grad_norm": 12.01241397857666,
      "learning_rate": 4.696969696969697e-05,
      "loss": 0.2851,
      "step": 170
    },
    {
      "epoch": 8.0,
      "grad_norm": 17.638214111328125,
      "learning_rate": 4.6464646464646464e-05,
      "loss": 0.319,
      "step": 180
    },
    {
      "epoch": 8.0,
      "eval_accuracy": 0.8984375,
      "eval_loss": 0.24509796500205994,
      "eval_runtime": 3.9737,
      "eval_samples_per_second": 161.061,
      "eval_steps_per_second": 2.517,
      "step": 180
    },
    {
      "epoch": 8.444444444444445,
      "grad_norm": 21.502132415771484,
      "learning_rate": 4.595959595959596e-05,
      "loss": 0.2968,
      "step": 190
    },
    {
      "epoch": 8.88888888888889,
      "grad_norm": 20.09746742248535,
      "learning_rate": 4.545454545454546e-05,
      "loss": 0.3489,
      "step": 200
    },
    {
      "epoch": 8.977777777777778,
      "eval_accuracy": 0.928125,
      "eval_loss": 0.18142804503440857,
      "eval_runtime": 3.7455,
      "eval_samples_per_second": 170.872,
      "eval_steps_per_second": 2.67,
      "step": 202
    },
    {
      "epoch": 9.333333333333334,
      "grad_norm": 16.205760955810547,
      "learning_rate": 4.494949494949495e-05,
      "loss": 0.2915,
      "step": 210
    },
    {
      "epoch": 9.777777777777779,
      "grad_norm": 27.511030197143555,
      "learning_rate": 4.4444444444444447e-05,
      "loss": 0.3251,
      "step": 220
    },
    {
      "epoch": 10.0,
      "eval_accuracy": 0.915625,
      "eval_loss": 0.24511559307575226,
      "eval_runtime": 3.6361,
      "eval_samples_per_second": 176.011,
      "eval_steps_per_second": 2.75,
      "step": 225
    },
    {
      "epoch": 10.222222222222221,
      "grad_norm": 11.171629905700684,
      "learning_rate": 4.3939393939393944e-05,
      "loss": 0.308,
      "step": 230
    },
    {
      "epoch": 10.666666666666666,
      "grad_norm": 12.315302848815918,
      "learning_rate": 4.343434343434344e-05,
      "loss": 0.3034,
      "step": 240
    },
    {
      "epoch": 10.977777777777778,
      "eval_accuracy": 0.940625,
      "eval_loss": 0.15658709406852722,
      "eval_runtime": 3.5876,
      "eval_samples_per_second": 178.392,
      "eval_steps_per_second": 2.787,
      "step": 247
    },
    {
      "epoch": 11.11111111111111,
      "grad_norm": 14.539319038391113,
      "learning_rate": 4.292929292929293e-05,
      "loss": 0.2847,
      "step": 250
    },
    {
      "epoch": 11.555555555555555,
      "grad_norm": 17.26177215576172,
      "learning_rate": 4.242424242424243e-05,
      "loss": 0.2754,
      "step": 260
    },
    {
      "epoch": 12.0,
      "grad_norm": 12.163039207458496,
      "learning_rate": 4.191919191919192e-05,
      "loss": 0.2746,
      "step": 270
    },
    {
      "epoch": 12.0,
      "eval_accuracy": 0.8921875,
      "eval_loss": 0.24925951659679413,
      "eval_runtime": 3.5808,
      "eval_samples_per_second": 178.73,
      "eval_steps_per_second": 2.793,
      "step": 270
    },
    {
      "epoch": 12.444444444444445,
      "grad_norm": 11.92519474029541,
      "learning_rate": 4.141414141414142e-05,
      "loss": 0.2889,
      "step": 280
    },
    {
      "epoch": 12.88888888888889,
      "grad_norm": 13.212408065795898,
      "learning_rate": 4.0909090909090915e-05,
      "loss": 0.2369,
      "step": 290
    },
    {
      "epoch": 12.977777777777778,
      "eval_accuracy": 0.9375,
      "eval_loss": 0.1622403860092163,
      "eval_runtime": 3.8343,
      "eval_samples_per_second": 166.914,
      "eval_steps_per_second": 2.608,
      "step": 292
    },
    {
      "epoch": 13.333333333333334,
      "grad_norm": 11.68896770477295,
      "learning_rate": 4.0404040404040405e-05,
      "loss": 0.2465,
      "step": 300
    },
    {
      "epoch": 13.777777777777779,
      "grad_norm": 14.610076904296875,
      "learning_rate": 3.98989898989899e-05,
      "loss": 0.2231,
      "step": 310
    },
    {
      "epoch": 14.0,
      "eval_accuracy": 0.9359375,
      "eval_loss": 0.17805945873260498,
      "eval_runtime": 3.9695,
      "eval_samples_per_second": 161.231,
      "eval_steps_per_second": 2.519,
      "step": 315
    },
    {
      "epoch": 14.222222222222221,
      "grad_norm": 12.407272338867188,
      "learning_rate": 3.939393939393939e-05,
      "loss": 0.2177,
      "step": 320
    },
    {
      "epoch": 14.666666666666666,
      "grad_norm": 7.3430256843566895,
      "learning_rate": 3.888888888888889e-05,
      "loss": 0.2281,
      "step": 330
    },
    {
      "epoch": 14.977777777777778,
      "eval_accuracy": 0.953125,
      "eval_loss": 0.12681424617767334,
      "eval_runtime": 3.6554,
      "eval_samples_per_second": 175.082,
      "eval_steps_per_second": 2.736,
      "step": 337
    },
    {
      "epoch": 15.11111111111111,
      "grad_norm": 10.262022018432617,
      "learning_rate": 3.838383838383838e-05,
      "loss": 0.209,
      "step": 340
    },
    {
      "epoch": 15.555555555555555,
      "grad_norm": 9.078124046325684,
      "learning_rate": 3.787878787878788e-05,
      "loss": 0.2134,
      "step": 350
    },
    {
      "epoch": 16.0,
      "grad_norm": 14.094355583190918,
      "learning_rate": 3.7373737373737376e-05,
      "loss": 0.2001,
      "step": 360
    },
    {
      "epoch": 16.0,
      "eval_accuracy": 0.9140625,
      "eval_loss": 0.24309130012989044,
      "eval_runtime": 3.5892,
      "eval_samples_per_second": 178.311,
      "eval_steps_per_second": 2.786,
      "step": 360
    },
    {
      "epoch": 16.444444444444443,
      "grad_norm": 12.868298530578613,
      "learning_rate": 3.686868686868687e-05,
      "loss": 0.2312,
      "step": 370
    },
    {
      "epoch": 16.88888888888889,
      "grad_norm": 7.863047122955322,
      "learning_rate": 3.6363636363636364e-05,
      "loss": 0.183,
      "step": 380
    },
    {
      "epoch": 16.977777777777778,
      "eval_accuracy": 0.9625,
      "eval_loss": 0.10167054831981659,
      "eval_runtime": 3.6007,
      "eval_samples_per_second": 177.743,
      "eval_steps_per_second": 2.777,
      "step": 382
    },
    {
      "epoch": 17.333333333333332,
      "grad_norm": 8.919840812683105,
      "learning_rate": 3.5858585858585855e-05,
      "loss": 0.1997,
      "step": 390
    },
    {
      "epoch": 17.77777777777778,
      "grad_norm": 11.91215705871582,
      "learning_rate": 3.535353535353535e-05,
      "loss": 0.1891,
      "step": 400
    },
    {
      "epoch": 18.0,
      "eval_accuracy": 0.9390625,
      "eval_loss": 0.180230051279068,
      "eval_runtime": 3.5491,
      "eval_samples_per_second": 180.325,
      "eval_steps_per_second": 2.818,
      "step": 405
    },
    {
      "epoch": 18.22222222222222,
      "grad_norm": 9.109786033630371,
      "learning_rate": 3.484848484848485e-05,
      "loss": 0.213,
      "step": 410
    },
    {
      "epoch": 18.666666666666668,
      "grad_norm": 5.232081413269043,
      "learning_rate": 3.434343434343435e-05,
      "loss": 0.1862,
      "step": 420
    },
    {
      "epoch": 18.977777777777778,
      "eval_accuracy": 0.9765625,
      "eval_loss": 0.08689282089471817,
      "eval_runtime": 3.9268,
      "eval_samples_per_second": 162.982,
      "eval_steps_per_second": 2.547,
      "step": 427
    },
    {
      "epoch": 19.11111111111111,
      "grad_norm": 13.340733528137207,
      "learning_rate": 3.3838383838383844e-05,
      "loss": 0.1664,
      "step": 430
    },
    {
      "epoch": 19.555555555555557,
      "grad_norm": 7.66475772857666,
      "learning_rate": 3.3333333333333335e-05,
      "loss": 0.1727,
      "step": 440
    },
    {
      "epoch": 20.0,
      "grad_norm": 13.612215995788574,
      "learning_rate": 3.282828282828283e-05,
      "loss": 0.1935,
      "step": 450
    },
    {
      "epoch": 20.0,
      "eval_accuracy": 0.96875,
      "eval_loss": 0.10791148245334625,
      "eval_runtime": 3.9472,
      "eval_samples_per_second": 162.141,
      "eval_steps_per_second": 2.533,
      "step": 450
    },
    {
      "epoch": 20.444444444444443,
      "grad_norm": 9.189305305480957,
      "learning_rate": 3.232323232323233e-05,
      "loss": 0.1945,
      "step": 460
    },
    {
      "epoch": 20.88888888888889,
      "grad_norm": 9.650483131408691,
      "learning_rate": 3.181818181818182e-05,
      "loss": 0.1797,
      "step": 470
    },
    {
      "epoch": 20.977777777777778,
      "eval_accuracy": 0.95625,
      "eval_loss": 0.12502644956111908,
      "eval_runtime": 3.6171,
      "eval_samples_per_second": 176.939,
      "eval_steps_per_second": 2.765,
      "step": 472
    },
    {
      "epoch": 21.333333333333332,
      "grad_norm": 7.25011682510376,
      "learning_rate": 3.131313131313132e-05,
      "loss": 0.1767,
      "step": 480
    },
    {
      "epoch": 21.77777777777778,
      "grad_norm": 9.036290168762207,
      "learning_rate": 3.080808080808081e-05,
      "loss": 0.1605,
      "step": 490
    },
    {
      "epoch": 22.0,
      "eval_accuracy": 0.971875,
      "eval_loss": 0.06545940786600113,
      "eval_runtime": 3.5923,
      "eval_samples_per_second": 178.16,
      "eval_steps_per_second": 2.784,
      "step": 495
    },
    {
      "epoch": 22.22222222222222,
      "grad_norm": 5.982744216918945,
      "learning_rate": 3.0303030303030306e-05,
      "loss": 0.1493,
      "step": 500
    },
    {
      "epoch": 22.666666666666668,
      "grad_norm": 13.987672805786133,
      "learning_rate": 2.9797979797979796e-05,
      "loss": 0.1848,
      "step": 510
    },
    {
      "epoch": 22.977777777777778,
      "eval_accuracy": 0.9765625,
      "eval_loss": 0.08063917607069016,
      "eval_runtime": 3.5801,
      "eval_samples_per_second": 178.764,
      "eval_steps_per_second": 2.793,
      "step": 517
    },
    {
      "epoch": 23.11111111111111,
      "grad_norm": 15.488668441772461,
      "learning_rate": 2.9292929292929294e-05,
      "loss": 0.1426,
      "step": 520
    },
    {
      "epoch": 23.555555555555557,
      "grad_norm": 11.646829605102539,
      "learning_rate": 2.878787878787879e-05,
      "loss": 0.1667,
      "step": 530
    },
    {
      "epoch": 24.0,
      "grad_norm": 6.619264602661133,
      "learning_rate": 2.8282828282828282e-05,
      "loss": 0.1498,
      "step": 540
    },
    {
      "epoch": 24.0,
      "eval_accuracy": 0.9578125,
      "eval_loss": 0.11159060150384903,
      "eval_runtime": 3.6016,
      "eval_samples_per_second": 177.701,
      "eval_steps_per_second": 2.777,
      "step": 540
    },
    {
      "epoch": 24.444444444444443,
      "grad_norm": 7.8661627769470215,
      "learning_rate": 2.777777777777778e-05,
      "loss": 0.1287,
      "step": 550
    },
    {
      "epoch": 24.88888888888889,
      "grad_norm": 7.934934616088867,
      "learning_rate": 2.7272727272727273e-05,
      "loss": 0.1394,
      "step": 560
    },
    {
      "epoch": 24.977777777777778,
      "eval_accuracy": 0.9671875,
      "eval_loss": 0.0806862860918045,
      "eval_runtime": 3.8749,
      "eval_samples_per_second": 165.167,
      "eval_steps_per_second": 2.581,
      "step": 562
    },
    {
      "epoch": 25.333333333333332,
      "grad_norm": 10.52723217010498,
      "learning_rate": 2.676767676767677e-05,
      "loss": 0.1524,
      "step": 570
    },
    {
      "epoch": 25.77777777777778,
      "grad_norm": 7.490493297576904,
      "learning_rate": 2.6262626262626268e-05,
      "loss": 0.1584,
      "step": 580
    },
    {
      "epoch": 26.0,
      "eval_accuracy": 0.9796875,
      "eval_loss": 0.05252554267644882,
      "eval_runtime": 3.9703,
      "eval_samples_per_second": 161.197,
      "eval_steps_per_second": 2.519,
      "step": 585
    },
    {
      "epoch": 26.22222222222222,
      "grad_norm": 7.178821563720703,
      "learning_rate": 2.575757575757576e-05,
      "loss": 0.153,
      "step": 590
    },
    {
      "epoch": 26.666666666666668,
      "grad_norm": 7.702730178833008,
      "learning_rate": 2.5252525252525256e-05,
      "loss": 0.1302,
      "step": 600
    },
    {
      "epoch": 26.977777777777778,
      "eval_accuracy": 0.9828125,
      "eval_loss": 0.05131406709551811,
      "eval_runtime": 3.7014,
      "eval_samples_per_second": 172.905,
      "eval_steps_per_second": 2.702,
      "step": 607
    },
    {
      "epoch": 27.11111111111111,
      "grad_norm": 9.482915878295898,
      "learning_rate": 2.474747474747475e-05,
      "loss": 0.1577,
      "step": 610
    },
    {
      "epoch": 27.555555555555557,
      "grad_norm": 10.196369171142578,
      "learning_rate": 2.4242424242424244e-05,
      "loss": 0.1543,
      "step": 620
    },
    {
      "epoch": 28.0,
      "grad_norm": 10.799006462097168,
      "learning_rate": 2.3737373737373738e-05,
      "loss": 0.1356,
      "step": 630
    },
    {
      "epoch": 28.0,
      "eval_accuracy": 0.9875,
      "eval_loss": 0.04204293340444565,
      "eval_runtime": 3.9774,
      "eval_samples_per_second": 160.91,
      "eval_steps_per_second": 2.514,
      "step": 630
    },
    {
      "epoch": 28.444444444444443,
      "grad_norm": 7.129752159118652,
      "learning_rate": 2.3232323232323232e-05,
      "loss": 0.1291,
      "step": 640
    },
    {
      "epoch": 28.88888888888889,
      "grad_norm": 6.642085552215576,
      "learning_rate": 2.272727272727273e-05,
      "loss": 0.1101,
      "step": 650
    },
    {
      "epoch": 28.977777777777778,
      "eval_accuracy": 0.9875,
      "eval_loss": 0.03539272025227547,
      "eval_runtime": 4.0166,
      "eval_samples_per_second": 159.34,
      "eval_steps_per_second": 2.49,
      "step": 652
    },
    {
      "epoch": 29.333333333333332,
      "grad_norm": 8.5753755569458,
      "learning_rate": 2.2222222222222223e-05,
      "loss": 0.1445,
      "step": 660
    },
    {
      "epoch": 29.77777777777778,
      "grad_norm": 13.183974266052246,
      "learning_rate": 2.171717171717172e-05,
      "loss": 0.1227,
      "step": 670
    },
    {
      "epoch": 30.0,
      "eval_accuracy": 0.9765625,
      "eval_loss": 0.0582769513130188,
      "eval_runtime": 3.9913,
      "eval_samples_per_second": 160.35,
      "eval_steps_per_second": 2.505,
      "step": 675
    },
    {
      "epoch": 30.22222222222222,
      "grad_norm": 9.026564598083496,
      "learning_rate": 2.1212121212121215e-05,
      "loss": 0.1209,
      "step": 680
    },
    {
      "epoch": 30.666666666666668,
      "grad_norm": 12.504347801208496,
      "learning_rate": 2.070707070707071e-05,
      "loss": 0.1158,
      "step": 690
    },
    {
      "epoch": 30.977777777777778,
      "eval_accuracy": 0.990625,
      "eval_loss": 0.025325458496809006,
      "eval_runtime": 3.9678,
      "eval_samples_per_second": 161.299,
      "eval_steps_per_second": 2.52,
      "step": 697
    },
    {
      "epoch": 31.11111111111111,
      "grad_norm": 5.276214599609375,
      "learning_rate": 2.0202020202020203e-05,
      "loss": 0.119,
      "step": 700
    },
    {
      "epoch": 31.555555555555557,
      "grad_norm": 8.732769012451172,
      "learning_rate": 1.9696969696969697e-05,
      "loss": 0.1156,
      "step": 710
    },
    {
      "epoch": 32.0,
      "grad_norm": 5.604591369628906,
      "learning_rate": 1.919191919191919e-05,
      "loss": 0.117,
      "step": 720
    },
    {
      "epoch": 32.0,
      "eval_accuracy": 0.990625,
      "eval_loss": 0.023098567500710487,
      "eval_runtime": 3.8505,
      "eval_samples_per_second": 166.213,
      "eval_steps_per_second": 2.597,
      "step": 720
    },
    {
      "epoch": 32.44444444444444,
      "grad_norm": 7.332610607147217,
      "learning_rate": 1.8686868686868688e-05,
      "loss": 0.1213,
      "step": 730
    },
    {
      "epoch": 32.888888888888886,
      "grad_norm": 12.890093803405762,
      "learning_rate": 1.8181818181818182e-05,
      "loss": 0.1022,
      "step": 740
    },
    {
      "epoch": 32.977777777777774,
      "eval_accuracy": 0.9796875,
      "eval_loss": 0.0725882276892662,
      "eval_runtime": 3.8065,
      "eval_samples_per_second": 168.135,
      "eval_steps_per_second": 2.627,
      "step": 742
    },
    {
      "epoch": 33.333333333333336,
      "grad_norm": 13.247682571411133,
      "learning_rate": 1.7676767676767676e-05,
      "loss": 0.1257,
      "step": 750
    },
    {
      "epoch": 33.77777777777778,
      "grad_norm": 6.758236885070801,
      "learning_rate": 1.7171717171717173e-05,
      "loss": 0.1221,
      "step": 760
    },
    {
      "epoch": 34.0,
      "eval_accuracy": 0.996875,
      "eval_loss": 0.015964530408382416,
      "eval_runtime": 3.7585,
      "eval_samples_per_second": 170.283,
      "eval_steps_per_second": 2.661,
      "step": 765
    },
    {
      "epoch": 34.22222222222222,
      "grad_norm": 8.521262168884277,
      "learning_rate": 1.6666666666666667e-05,
      "loss": 0.1014,
      "step": 770
    },
    {
      "epoch": 34.666666666666664,
      "grad_norm": 5.949100971221924,
      "learning_rate": 1.6161616161616165e-05,
      "loss": 0.0956,
      "step": 780
    },
    {
      "epoch": 34.977777777777774,
      "eval_accuracy": 0.984375,
      "eval_loss": 0.048214979469776154,
      "eval_runtime": 3.6909,
      "eval_samples_per_second": 173.399,
      "eval_steps_per_second": 2.709,
      "step": 787
    },
    {
      "epoch": 35.111111111111114,
      "grad_norm": 10.151766777038574,
      "learning_rate": 1.565656565656566e-05,
      "loss": 0.1135,
      "step": 790
    },
    {
      "epoch": 35.55555555555556,
      "grad_norm": 9.514137268066406,
      "learning_rate": 1.5151515151515153e-05,
      "loss": 0.1109,
      "step": 800
    },
    {
      "epoch": 36.0,
      "grad_norm": 4.4278717041015625,
      "learning_rate": 1.4646464646464647e-05,
      "loss": 0.0856,
      "step": 810
    },
    {
      "epoch": 36.0,
      "eval_accuracy": 0.9875,
      "eval_loss": 0.025567293167114258,
      "eval_runtime": 3.6537,
      "eval_samples_per_second": 175.165,
      "eval_steps_per_second": 2.737,
      "step": 810
    },
    {
      "epoch": 36.44444444444444,
      "grad_norm": 8.582184791564941,
      "learning_rate": 1.4141414141414141e-05,
      "loss": 0.0994,
      "step": 820
    },
    {
      "epoch": 36.888888888888886,
      "grad_norm": 9.628859519958496,
      "learning_rate": 1.3636363636363637e-05,
      "loss": 0.0996,
      "step": 830
    },
    {
      "epoch": 36.977777777777774,
      "eval_accuracy": 0.990625,
      "eval_loss": 0.021057253703475,
      "eval_runtime": 3.6772,
      "eval_samples_per_second": 174.046,
      "eval_steps_per_second": 2.719,
      "step": 832
    },
    {
      "epoch": 37.333333333333336,
      "grad_norm": 5.165952682495117,
      "learning_rate": 1.3131313131313134e-05,
      "loss": 0.0915,
      "step": 840
    },
    {
      "epoch": 37.77777777777778,
      "grad_norm": 5.831385135650635,
      "learning_rate": 1.2626262626262628e-05,
      "loss": 0.0848,
      "step": 850
    },
    {
      "epoch": 38.0,
      "eval_accuracy": 0.9796875,
      "eval_loss": 0.04457371309399605,
      "eval_runtime": 3.6584,
      "eval_samples_per_second": 174.938,
      "eval_steps_per_second": 2.733,
      "step": 855
    },
    {
      "epoch": 38.22222222222222,
      "grad_norm": 9.629181861877441,
      "learning_rate": 1.2121212121212122e-05,
      "loss": 0.0972,
      "step": 860
    },
    {
      "epoch": 38.666666666666664,
      "grad_norm": 6.214244365692139,
      "learning_rate": 1.1616161616161616e-05,
      "loss": 0.1001,
      "step": 870
    },
    {
      "epoch": 38.977777777777774,
      "eval_accuracy": 0.9875,
      "eval_loss": 0.02742326818406582,
      "eval_runtime": 3.668,
      "eval_samples_per_second": 174.481,
      "eval_steps_per_second": 2.726,
      "step": 877
    },
    {
      "epoch": 39.111111111111114,
      "grad_norm": 11.1734619140625,
      "learning_rate": 1.1111111111111112e-05,
      "loss": 0.0919,
      "step": 880
    },
    {
      "epoch": 39.55555555555556,
      "grad_norm": 6.438005447387695,
      "learning_rate": 1.0606060606060607e-05,
      "loss": 0.0988,
      "step": 890
    },
    {
      "epoch": 40.0,
      "grad_norm": 5.9803643226623535,
      "learning_rate": 1.0101010101010101e-05,
      "loss": 0.0976,
      "step": 900
    },
    {
      "epoch": 40.0,
      "eval_accuracy": 0.9921875,
      "eval_loss": 0.022529248148202896,
      "eval_runtime": 3.7092,
      "eval_samples_per_second": 172.543,
      "eval_steps_per_second": 2.696,
      "step": 900
    },
    {
      "epoch": 40.44444444444444,
      "grad_norm": 7.562661647796631,
      "learning_rate": 9.595959595959595e-06,
      "loss": 0.085,
      "step": 910
    },
    {
      "epoch": 40.888888888888886,
      "grad_norm": 7.695030212402344,
      "learning_rate": 9.090909090909091e-06,
      "loss": 0.0864,
      "step": 920
    },
    {
      "epoch": 40.977777777777774,
      "eval_accuracy": 0.9921875,
      "eval_loss": 0.0207191314548254,
      "eval_runtime": 4.0024,
      "eval_samples_per_second": 159.904,
      "eval_steps_per_second": 2.499,
      "step": 922
    },
    {
      "epoch": 41.333333333333336,
      "grad_norm": 8.4052734375,
      "learning_rate": 8.585858585858587e-06,
      "loss": 0.088,
      "step": 930
    },
    {
      "epoch": 41.77777777777778,
      "grad_norm": 8.705794334411621,
      "learning_rate": 8.080808080808082e-06,
      "loss": 0.0865,
      "step": 940
    },
    {
      "epoch": 42.0,
      "eval_accuracy": 0.996875,
      "eval_loss": 0.01933131366968155,
      "eval_runtime": 3.9909,
      "eval_samples_per_second": 160.365,
      "eval_steps_per_second": 2.506,
      "step": 945
    },
    {
      "epoch": 42.22222222222222,
      "grad_norm": 7.478874683380127,
      "learning_rate": 7.5757575757575764e-06,
      "loss": 0.0815,
      "step": 950
    },
    {
      "epoch": 42.666666666666664,
      "grad_norm": 5.25657320022583,
      "learning_rate": 7.0707070707070704e-06,
      "loss": 0.0773,
      "step": 960
    },
    {
      "epoch": 42.977777777777774,
      "eval_accuracy": 0.9921875,
      "eval_loss": 0.020288193598389626,
      "eval_runtime": 3.6594,
      "eval_samples_per_second": 174.89,
      "eval_steps_per_second": 2.733,
      "step": 967
    },
    {
      "epoch": 43.111111111111114,
      "grad_norm": 4.1972246170043945,
      "learning_rate": 6.565656565656567e-06,
      "loss": 0.0799,
      "step": 970
    },
    {
      "epoch": 43.55555555555556,
      "grad_norm": 6.9554972648620605,
      "learning_rate": 6.060606060606061e-06,
      "loss": 0.0772,
      "step": 980
    },
    {
      "epoch": 44.0,
      "grad_norm": 6.343081951141357,
      "learning_rate": 5.555555555555556e-06,
      "loss": 0.075,
      "step": 990
    },
    {
      "epoch": 44.0,
      "eval_accuracy": 0.996875,
      "eval_loss": 0.013058523647487164,
      "eval_runtime": 3.5774,
      "eval_samples_per_second": 178.899,
      "eval_steps_per_second": 2.795,
      "step": 990
    },
    {
      "epoch": 44.44444444444444,
      "grad_norm": 4.900812149047852,
      "learning_rate": 5.050505050505051e-06,
      "loss": 0.0736,
      "step": 1000
    },
    {
      "epoch": 44.888888888888886,
      "grad_norm": 5.955135345458984,
      "learning_rate": 4.5454545454545455e-06,
      "loss": 0.0761,
      "step": 1010
    },
    {
      "epoch": 44.977777777777774,
      "eval_accuracy": 0.99375,
      "eval_loss": 0.012860281392931938,
      "eval_runtime": 3.5897,
      "eval_samples_per_second": 178.288,
      "eval_steps_per_second": 2.786,
      "step": 1012
    },
    {
      "epoch": 45.333333333333336,
      "grad_norm": 4.250102996826172,
      "learning_rate": 4.040404040404041e-06,
      "loss": 0.0707,
      "step": 1020
    },
    {
      "epoch": 45.77777777777778,
      "grad_norm": 6.8997931480407715,
      "learning_rate": 3.5353535353535352e-06,
      "loss": 0.0624,
      "step": 1030
    },
    {
      "epoch": 46.0,
      "eval_accuracy": 0.996875,
      "eval_loss": 0.011364495381712914,
      "eval_runtime": 3.6128,
      "eval_samples_per_second": 177.146,
      "eval_steps_per_second": 2.768,
      "step": 1035
    },
    {
      "epoch": 46.22222222222222,
      "grad_norm": 10.210082054138184,
      "learning_rate": 3.0303030303030305e-06,
      "loss": 0.0762,
      "step": 1040
    },
    {
      "epoch": 46.666666666666664,
      "grad_norm": 4.9201788902282715,
      "learning_rate": 2.5252525252525253e-06,
      "loss": 0.0557,
      "step": 1050
    },
    {
      "epoch": 46.977777777777774,
      "eval_accuracy": 0.9953125,
      "eval_loss": 0.010208332911133766,
      "eval_runtime": 3.8474,
      "eval_samples_per_second": 166.347,
      "eval_steps_per_second": 2.599,
      "step": 1057
    },
    {
      "epoch": 47.111111111111114,
      "grad_norm": 3.725327491760254,
      "learning_rate": 2.0202020202020206e-06,
      "loss": 0.0613,
      "step": 1060
    },
    {
      "epoch": 47.55555555555556,
      "grad_norm": 3.1549530029296875,
      "learning_rate": 1.5151515151515152e-06,
      "loss": 0.0601,
      "step": 1070
    },
    {
      "epoch": 48.0,
      "grad_norm": 6.280518054962158,
      "learning_rate": 1.0101010101010103e-06,
      "loss": 0.0708,
      "step": 1080
    },
    {
      "epoch": 48.0,
      "eval_accuracy": 0.9953125,
      "eval_loss": 0.01160599384456873,
      "eval_runtime": 3.9922,
      "eval_samples_per_second": 160.314,
      "eval_steps_per_second": 2.505,
      "step": 1080
    },
    {
      "epoch": 48.44444444444444,
      "grad_norm": 6.1849260330200195,
      "learning_rate": 5.050505050505052e-07,
      "loss": 0.0699,
      "step": 1090
    },
    {
      "epoch": 48.888888888888886,
      "grad_norm": 7.637501239776611,
      "learning_rate": 0.0,
      "loss": 0.0667,
      "step": 1100
    },
    {
      "epoch": 48.888888888888886,
      "eval_accuracy": 0.9953125,
      "eval_loss": 0.013088616542518139,
      "eval_runtime": 3.601,
      "eval_samples_per_second": 177.727,
      "eval_steps_per_second": 2.777,
      "step": 1100
    },
    {
      "epoch": 48.888888888888886,
      "step": 1100,
      "total_flos": 5.510665685119795e+18,
      "train_loss": 0.17931611462072894,
      "train_runtime": 3528.3547,
      "train_samples_per_second": 81.624,
      "train_steps_per_second": 0.312
    }
  ],
  "logging_steps": 10,
  "max_steps": 1100,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 50,
  "save_steps": 500,
  "total_flos": 5.510665685119795e+18,
  "train_batch_size": 64,
  "trial_name": null,
  "trial_params": null
}