{ "best_metric": 2.8627383708953857, "best_model_checkpoint": "cdetr-mist1-brain-gt-tumors-8ah-6l/checkpoint-2645", "epoch": 25.0, "eval_steps": 500, "global_step": 2875, "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": 4.382608695652174e-07, "loss": 2.9701, "step": 2760 }, { "epoch": 24.0, "eval_loss": 2.868567705154419, "eval_runtime": 6.0333, "eval_samples_per_second": 6.63, "eval_steps_per_second": 0.829, "step": 2760 }, { "epoch": 25.0, "learning_rate": 3.82608695652174e-08, "loss": 2.9964, "step": 2875 }, { "epoch": 25.0, "eval_loss": 2.8917393684387207, "eval_runtime": 6.0377, "eval_samples_per_second": 6.625, "eval_steps_per_second": 0.828, "step": 2875 }, { "epoch": 25.0, "step": 2875, "total_flos": 5.73883103808e+18, "train_loss": 3.3358317977241847, "train_runtime": 2483.4302, "train_samples_per_second": 4.631, "train_steps_per_second": 1.158 } ], "logging_steps": 500, "max_steps": 2875, "num_train_epochs": 25, "save_steps": 500, "total_flos": 5.73883103808e+18, "trial_name": null, "trial_params": null }