{
  "best_metric": 2.827199935913086,
  "best_model_checkpoint": "cdetr-mist1-brain-gt-tumors-8ah-6l/checkpoint-4830",
  "epoch": 50.0,
  "eval_steps": 500,
  "global_step": 5750,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 1.0,
      "learning_rate": 9.624347826086956e-06,
      "loss": 5.4149,
      "step": 115
    },
    {
      "epoch": 1.0,
      "eval_loss": 4.397365093231201,
      "eval_runtime": 5.8932,
      "eval_samples_per_second": 6.787,
      "eval_steps_per_second": 0.848,
      "step": 115
    },
    {
      "epoch": 2.0,
      "learning_rate": 9.231304347826087e-06,
      "loss": 3.9453,
      "step": 230
    },
    {
      "epoch": 2.0,
      "eval_loss": 3.651952028274536,
      "eval_runtime": 5.9289,
      "eval_samples_per_second": 6.747,
      "eval_steps_per_second": 0.843,
      "step": 230
    },
    {
      "epoch": 3.0,
      "learning_rate": 8.831304347826088e-06,
      "loss": 3.7269,
      "step": 345
    },
    {
      "epoch": 3.0,
      "eval_loss": 3.7602438926696777,
      "eval_runtime": 5.8799,
      "eval_samples_per_second": 6.803,
      "eval_steps_per_second": 0.85,
      "step": 345
    },
    {
      "epoch": 4.0,
      "learning_rate": 8.434782608695653e-06,
      "loss": 3.5898,
      "step": 460
    },
    {
      "epoch": 4.0,
      "eval_loss": 3.5670828819274902,
      "eval_runtime": 5.9645,
      "eval_samples_per_second": 6.706,
      "eval_steps_per_second": 0.838,
      "step": 460
    },
    {
      "epoch": 5.0,
      "learning_rate": 8.034782608695652e-06,
      "loss": 3.486,
      "step": 575
    },
    {
      "epoch": 5.0,
      "eval_loss": 3.4911675453186035,
      "eval_runtime": 5.9241,
      "eval_samples_per_second": 6.752,
      "eval_steps_per_second": 0.844,
      "step": 575
    },
    {
      "epoch": 6.0,
      "learning_rate": 7.634782608695653e-06,
      "loss": 3.4073,
      "step": 690
    },
    {
      "epoch": 6.0,
      "eval_loss": 3.409463405609131,
      "eval_runtime": 5.8302,
      "eval_samples_per_second": 6.861,
      "eval_steps_per_second": 0.858,
      "step": 690
    },
    {
      "epoch": 7.0,
      "learning_rate": 7.234782608695653e-06,
      "loss": 3.4181,
      "step": 805
    },
    {
      "epoch": 7.0,
      "eval_loss": 3.3183083534240723,
      "eval_runtime": 5.8894,
      "eval_samples_per_second": 6.792,
      "eval_steps_per_second": 0.849,
      "step": 805
    },
    {
      "epoch": 8.0,
      "learning_rate": 6.834782608695653e-06,
      "loss": 3.3603,
      "step": 920
    },
    {
      "epoch": 8.0,
      "eval_loss": 3.1111273765563965,
      "eval_runtime": 5.9175,
      "eval_samples_per_second": 6.76,
      "eval_steps_per_second": 0.845,
      "step": 920
    },
    {
      "epoch": 9.0,
      "learning_rate": 6.434782608695652e-06,
      "loss": 3.2777,
      "step": 1035
    },
    {
      "epoch": 9.0,
      "eval_loss": 3.1992499828338623,
      "eval_runtime": 5.9275,
      "eval_samples_per_second": 6.748,
      "eval_steps_per_second": 0.844,
      "step": 1035
    },
    {
      "epoch": 10.0,
      "learning_rate": 6.034782608695653e-06,
      "loss": 3.2851,
      "step": 1150
    },
    {
      "epoch": 10.0,
      "eval_loss": 3.3997130393981934,
      "eval_runtime": 5.8939,
      "eval_samples_per_second": 6.787,
      "eval_steps_per_second": 0.848,
      "step": 1150
    },
    {
      "epoch": 11.0,
      "learning_rate": 5.634782608695652e-06,
      "loss": 3.266,
      "step": 1265
    },
    {
      "epoch": 11.0,
      "eval_loss": 3.286081314086914,
      "eval_runtime": 6.1287,
      "eval_samples_per_second": 6.527,
      "eval_steps_per_second": 0.816,
      "step": 1265
    },
    {
      "epoch": 12.0,
      "learning_rate": 5.234782608695652e-06,
      "loss": 3.2803,
      "step": 1380
    },
    {
      "epoch": 12.0,
      "eval_loss": 3.18131947517395,
      "eval_runtime": 6.1123,
      "eval_samples_per_second": 6.544,
      "eval_steps_per_second": 0.818,
      "step": 1380
    },
    {
      "epoch": 13.0,
      "learning_rate": 4.834782608695652e-06,
      "loss": 3.1733,
      "step": 1495
    },
    {
      "epoch": 13.0,
      "eval_loss": 2.9837594032287598,
      "eval_runtime": 5.925,
      "eval_samples_per_second": 6.751,
      "eval_steps_per_second": 0.844,
      "step": 1495
    },
    {
      "epoch": 14.0,
      "learning_rate": 4.434782608695653e-06,
      "loss": 3.2094,
      "step": 1610
    },
    {
      "epoch": 14.0,
      "eval_loss": 3.1175174713134766,
      "eval_runtime": 5.9137,
      "eval_samples_per_second": 6.764,
      "eval_steps_per_second": 0.845,
      "step": 1610
    },
    {
      "epoch": 15.0,
      "learning_rate": 4.034782608695652e-06,
      "loss": 3.1718,
      "step": 1725
    },
    {
      "epoch": 15.0,
      "eval_loss": 3.0063586235046387,
      "eval_runtime": 5.9077,
      "eval_samples_per_second": 6.771,
      "eval_steps_per_second": 0.846,
      "step": 1725
    },
    {
      "epoch": 16.0,
      "learning_rate": 3.6347826086956524e-06,
      "loss": 3.1303,
      "step": 1840
    },
    {
      "epoch": 16.0,
      "eval_loss": 3.086920738220215,
      "eval_runtime": 5.9305,
      "eval_samples_per_second": 6.745,
      "eval_steps_per_second": 0.843,
      "step": 1840
    },
    {
      "epoch": 17.0,
      "learning_rate": 3.2347826086956524e-06,
      "loss": 3.0897,
      "step": 1955
    },
    {
      "epoch": 17.0,
      "eval_loss": 3.0305533409118652,
      "eval_runtime": 6.0975,
      "eval_samples_per_second": 6.56,
      "eval_steps_per_second": 0.82,
      "step": 1955
    },
    {
      "epoch": 18.0,
      "learning_rate": 2.8347826086956525e-06,
      "loss": 3.0233,
      "step": 2070
    },
    {
      "epoch": 18.0,
      "eval_loss": 2.9478588104248047,
      "eval_runtime": 5.9801,
      "eval_samples_per_second": 6.689,
      "eval_steps_per_second": 0.836,
      "step": 2070
    },
    {
      "epoch": 19.0,
      "learning_rate": 2.4347826086956525e-06,
      "loss": 3.0156,
      "step": 2185
    },
    {
      "epoch": 19.0,
      "eval_loss": 2.914536952972412,
      "eval_runtime": 6.0155,
      "eval_samples_per_second": 6.649,
      "eval_steps_per_second": 0.831,
      "step": 2185
    },
    {
      "epoch": 20.0,
      "learning_rate": 2.0347826086956525e-06,
      "loss": 3.0277,
      "step": 2300
    },
    {
      "epoch": 20.0,
      "eval_loss": 2.891925811767578,
      "eval_runtime": 6.058,
      "eval_samples_per_second": 6.603,
      "eval_steps_per_second": 0.825,
      "step": 2300
    },
    {
      "epoch": 21.0,
      "learning_rate": 1.6382608695652176e-06,
      "loss": 3.0847,
      "step": 2415
    },
    {
      "epoch": 21.0,
      "eval_loss": 2.9321396350860596,
      "eval_runtime": 6.1412,
      "eval_samples_per_second": 6.513,
      "eval_steps_per_second": 0.814,
      "step": 2415
    },
    {
      "epoch": 22.0,
      "learning_rate": 1.2382608695652176e-06,
      "loss": 3.0333,
      "step": 2530
    },
    {
      "epoch": 22.0,
      "eval_loss": 2.91283917427063,
      "eval_runtime": 6.139,
      "eval_samples_per_second": 6.516,
      "eval_steps_per_second": 0.814,
      "step": 2530
    },
    {
      "epoch": 23.0,
      "learning_rate": 8.382608695652175e-07,
      "loss": 3.0126,
      "step": 2645
    },
    {
      "epoch": 23.0,
      "eval_loss": 2.8627383708953857,
      "eval_runtime": 6.0565,
      "eval_samples_per_second": 6.604,
      "eval_steps_per_second": 0.826,
      "step": 2645
    },
    {
      "epoch": 24.0,
      "learning_rate": 5.234782608695652e-06,
      "loss": 2.9968,
      "step": 2760
    },
    {
      "epoch": 24.0,
      "eval_loss": 3.0186123847961426,
      "eval_runtime": 5.7749,
      "eval_samples_per_second": 6.926,
      "eval_steps_per_second": 0.866,
      "step": 2760
    },
    {
      "epoch": 25.0,
      "learning_rate": 5.036521739130436e-06,
      "loss": 3.0295,
      "step": 2875
    },
    {
      "epoch": 25.0,
      "eval_loss": 3.0147616863250732,
      "eval_runtime": 5.799,
      "eval_samples_per_second": 6.898,
      "eval_steps_per_second": 0.862,
      "step": 2875
    },
    {
      "epoch": 26.0,
      "learning_rate": 4.836521739130435e-06,
      "loss": 3.0294,
      "step": 2990
    },
    {
      "epoch": 26.0,
      "eval_loss": 3.034066915512085,
      "eval_runtime": 5.8474,
      "eval_samples_per_second": 6.841,
      "eval_steps_per_second": 0.855,
      "step": 2990
    },
    {
      "epoch": 27.0,
      "learning_rate": 4.636521739130435e-06,
      "loss": 3.0395,
      "step": 3105
    },
    {
      "epoch": 27.0,
      "eval_loss": 2.99969220161438,
      "eval_runtime": 5.7579,
      "eval_samples_per_second": 6.947,
      "eval_steps_per_second": 0.868,
      "step": 3105
    },
    {
      "epoch": 28.0,
      "learning_rate": 4.436521739130435e-06,
      "loss": 3.0445,
      "step": 3220
    },
    {
      "epoch": 28.0,
      "eval_loss": 3.0574753284454346,
      "eval_runtime": 5.6888,
      "eval_samples_per_second": 7.031,
      "eval_steps_per_second": 0.879,
      "step": 3220
    },
    {
      "epoch": 29.0,
      "learning_rate": 4.236521739130436e-06,
      "loss": 2.9761,
      "step": 3335
    },
    {
      "epoch": 29.0,
      "eval_loss": 2.97066068649292,
      "eval_runtime": 5.7458,
      "eval_samples_per_second": 6.962,
      "eval_steps_per_second": 0.87,
      "step": 3335
    },
    {
      "epoch": 30.0,
      "learning_rate": 4.038260869565218e-06,
      "loss": 3.0075,
      "step": 3450
    },
    {
      "epoch": 30.0,
      "eval_loss": 2.939246416091919,
      "eval_runtime": 5.722,
      "eval_samples_per_second": 6.991,
      "eval_steps_per_second": 0.874,
      "step": 3450
    },
    {
      "epoch": 31.0,
      "learning_rate": 3.838260869565218e-06,
      "loss": 3.0198,
      "step": 3565
    },
    {
      "epoch": 31.0,
      "eval_loss": 2.9121549129486084,
      "eval_runtime": 5.8674,
      "eval_samples_per_second": 6.817,
      "eval_steps_per_second": 0.852,
      "step": 3565
    },
    {
      "epoch": 32.0,
      "learning_rate": 3.6382608695652177e-06,
      "loss": 2.9782,
      "step": 3680
    },
    {
      "epoch": 32.0,
      "eval_loss": 2.947079658508301,
      "eval_runtime": 5.7346,
      "eval_samples_per_second": 6.975,
      "eval_steps_per_second": 0.872,
      "step": 3680
    },
    {
      "epoch": 33.0,
      "learning_rate": 3.4382608695652175e-06,
      "loss": 2.9773,
      "step": 3795
    },
    {
      "epoch": 33.0,
      "eval_loss": 3.030561923980713,
      "eval_runtime": 5.7197,
      "eval_samples_per_second": 6.993,
      "eval_steps_per_second": 0.874,
      "step": 3795
    },
    {
      "epoch": 34.0,
      "learning_rate": 3.2382608695652177e-06,
      "loss": 2.9528,
      "step": 3910
    },
    {
      "epoch": 34.0,
      "eval_loss": 2.8513386249542236,
      "eval_runtime": 5.6599,
      "eval_samples_per_second": 7.067,
      "eval_steps_per_second": 0.883,
      "step": 3910
    },
    {
      "epoch": 35.0,
      "learning_rate": 3.038260869565218e-06,
      "loss": 2.9228,
      "step": 4025
    },
    {
      "epoch": 35.0,
      "eval_loss": 2.899709701538086,
      "eval_runtime": 5.6812,
      "eval_samples_per_second": 7.041,
      "eval_steps_per_second": 0.88,
      "step": 4025
    },
    {
      "epoch": 36.0,
      "learning_rate": 2.8382608695652177e-06,
      "loss": 2.9221,
      "step": 4140
    },
    {
      "epoch": 36.0,
      "eval_loss": 2.8646340370178223,
      "eval_runtime": 5.6561,
      "eval_samples_per_second": 7.072,
      "eval_steps_per_second": 0.884,
      "step": 4140
    },
    {
      "epoch": 37.0,
      "learning_rate": 2.638260869565218e-06,
      "loss": 2.8933,
      "step": 4255
    },
    {
      "epoch": 37.0,
      "eval_loss": 2.8870697021484375,
      "eval_runtime": 5.7007,
      "eval_samples_per_second": 7.017,
      "eval_steps_per_second": 0.877,
      "step": 4255
    },
    {
      "epoch": 38.0,
      "learning_rate": 2.4382608695652177e-06,
      "loss": 2.8925,
      "step": 4370
    },
    {
      "epoch": 38.0,
      "eval_loss": 2.94067120552063,
      "eval_runtime": 5.6567,
      "eval_samples_per_second": 7.071,
      "eval_steps_per_second": 0.884,
      "step": 4370
    },
    {
      "epoch": 39.0,
      "learning_rate": 2.2382608695652175e-06,
      "loss": 2.9069,
      "step": 4485
    },
    {
      "epoch": 39.0,
      "eval_loss": 2.9625325202941895,
      "eval_runtime": 5.6615,
      "eval_samples_per_second": 7.065,
      "eval_steps_per_second": 0.883,
      "step": 4485
    },
    {
      "epoch": 40.0,
      "learning_rate": 2.0382608695652177e-06,
      "loss": 2.9246,
      "step": 4600
    },
    {
      "epoch": 40.0,
      "eval_loss": 2.994633197784424,
      "eval_runtime": 5.6894,
      "eval_samples_per_second": 7.031,
      "eval_steps_per_second": 0.879,
      "step": 4600
    },
    {
      "epoch": 41.0,
      "learning_rate": 1.8382608695652178e-06,
      "loss": 2.9089,
      "step": 4715
    },
    {
      "epoch": 41.0,
      "eval_loss": 2.8935651779174805,
      "eval_runtime": 5.7226,
      "eval_samples_per_second": 6.99,
      "eval_steps_per_second": 0.874,
      "step": 4715
    },
    {
      "epoch": 42.0,
      "learning_rate": 1.6382608695652176e-06,
      "loss": 2.8573,
      "step": 4830
    },
    {
      "epoch": 42.0,
      "eval_loss": 2.827199935913086,
      "eval_runtime": 5.6411,
      "eval_samples_per_second": 7.091,
      "eval_steps_per_second": 0.886,
      "step": 4830
    },
    {
      "epoch": 43.0,
      "learning_rate": 1.4382608695652176e-06,
      "loss": 2.8378,
      "step": 4945
    },
    {
      "epoch": 43.0,
      "eval_loss": 2.8543009757995605,
      "eval_runtime": 5.6643,
      "eval_samples_per_second": 7.062,
      "eval_steps_per_second": 0.883,
      "step": 4945
    },
    {
      "epoch": 44.0,
      "learning_rate": 1.2382608695652176e-06,
      "loss": 2.8957,
      "step": 5060
    },
    {
      "epoch": 44.0,
      "eval_loss": 2.859046459197998,
      "eval_runtime": 5.687,
      "eval_samples_per_second": 7.034,
      "eval_steps_per_second": 0.879,
      "step": 5060
    },
    {
      "epoch": 45.0,
      "learning_rate": 1.0382608695652174e-06,
      "loss": 2.8454,
      "step": 5175
    },
    {
      "epoch": 45.0,
      "eval_loss": 2.8524787425994873,
      "eval_runtime": 5.6623,
      "eval_samples_per_second": 7.064,
      "eval_steps_per_second": 0.883,
      "step": 5175
    },
    {
      "epoch": 46.0,
      "learning_rate": 8.382608695652175e-07,
      "loss": 2.7964,
      "step": 5290
    },
    {
      "epoch": 46.0,
      "eval_loss": 2.8657748699188232,
      "eval_runtime": 5.6625,
      "eval_samples_per_second": 7.064,
      "eval_steps_per_second": 0.883,
      "step": 5290
    },
    {
      "epoch": 47.0,
      "learning_rate": 6.382608695652175e-07,
      "loss": 2.8172,
      "step": 5405
    },
    {
      "epoch": 47.0,
      "eval_loss": 2.8872039318084717,
      "eval_runtime": 5.6798,
      "eval_samples_per_second": 7.043,
      "eval_steps_per_second": 0.88,
      "step": 5405
    },
    {
      "epoch": 48.0,
      "learning_rate": 4.382608695652174e-07,
      "loss": 2.8472,
      "step": 5520
    },
    {
      "epoch": 48.0,
      "eval_loss": 2.8653695583343506,
      "eval_runtime": 5.6595,
      "eval_samples_per_second": 7.068,
      "eval_steps_per_second": 0.883,
      "step": 5520
    },
    {
      "epoch": 49.0,
      "learning_rate": 2.3826086956521743e-07,
      "loss": 2.811,
      "step": 5635
    },
    {
      "epoch": 49.0,
      "eval_loss": 2.8571643829345703,
      "eval_runtime": 5.7272,
      "eval_samples_per_second": 6.984,
      "eval_steps_per_second": 0.873,
      "step": 5635
    },
    {
      "epoch": 50.0,
      "learning_rate": 3.82608695652174e-08,
      "loss": 2.801,
      "step": 5750
    },
    {
      "epoch": 50.0,
      "eval_loss": 2.830277919769287,
      "eval_runtime": 5.6412,
      "eval_samples_per_second": 7.091,
      "eval_steps_per_second": 0.886,
      "step": 5750
    },
    {
      "epoch": 50.0,
      "step": 5750,
      "total_flos": 1.147766207616e+19,
      "train_loss": 1.5786318518597147,
      "train_runtime": 2642.836,
      "train_samples_per_second": 8.703,
      "train_steps_per_second": 2.176
    }
  ],
  "logging_steps": 500,
  "max_steps": 5750,
  "num_train_epochs": 50,
  "save_steps": 500,
  "total_flos": 1.147766207616e+19,
  "trial_name": null,
  "trial_params": null
}