{ "best_metric": 0.9265375854214123, "best_model_checkpoint": "swin-base-patch4-window7-224-in22k-finetuned-brain-tumor-final/checkpoint-186", "epoch": 9.638554216867469, "global_step": 200, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.48, "learning_rate": 5e-06, "loss": 1.3805, "step": 10 }, { "epoch": 0.96, "learning_rate": 1e-05, "loss": 1.2212, "step": 20 }, { "epoch": 0.96, "eval_accuracy": 0.6429384965831435, "eval_f1_score": 0.6225332030044981, "eval_loss": 1.140669822692871, "eval_precision": 0.6600729727071496, "eval_runtime": 97.1469, "eval_samples_per_second": 18.076, "eval_steps_per_second": 0.288, "step": 20 }, { "epoch": 1.45, "learning_rate": 9.444444444444445e-06, "loss": 0.8664, "step": 30 }, { "epoch": 1.93, "learning_rate": 8.888888888888888e-06, "loss": 0.565, "step": 40 }, { "epoch": 1.98, "eval_accuracy": 0.8325740318906606, "eval_f1_score": 0.8311373357165648, "eval_loss": 0.5161515474319458, "eval_precision": 0.8427982091540198, "eval_runtime": 98.3593, "eval_samples_per_second": 17.853, "eval_steps_per_second": 0.285, "step": 41 }, { "epoch": 2.41, "learning_rate": 8.333333333333334e-06, "loss": 0.4226, "step": 50 }, { "epoch": 2.89, "learning_rate": 7.77777777777778e-06, "loss": 0.3245, "step": 60 }, { "epoch": 2.99, "eval_accuracy": 0.8804100227790432, "eval_f1_score": 0.8784127627414283, "eval_loss": 0.3265204429626465, "eval_precision": 0.8843478014022281, "eval_runtime": 94.9107, "eval_samples_per_second": 18.502, "eval_steps_per_second": 0.295, "step": 62 }, { "epoch": 3.37, "learning_rate": 7.222222222222223e-06, "loss": 0.2761, "step": 70 }, { "epoch": 3.86, "learning_rate": 6.666666666666667e-06, "loss": 0.2618, "step": 80 }, { "epoch": 4.0, "eval_accuracy": 0.9066059225512528, "eval_f1_score": 0.9053659025451456, "eval_loss": 0.2712633013725281, "eval_precision": 0.9105145368026273, "eval_runtime": 95.6192, "eval_samples_per_second": 18.365, "eval_steps_per_second": 0.293, "step": 83 }, { "epoch": 4.34, "learning_rate": 6.111111111111112e-06, "loss": 0.225, "step": 90 }, { "epoch": 4.82, "learning_rate": 5.555555555555557e-06, "loss": 0.2164, "step": 100 }, { "epoch": 4.96, "eval_accuracy": 0.8946469248291572, "eval_f1_score": 0.8929116621992249, "eval_loss": 0.28123489022254944, "eval_precision": 0.899366339703112, "eval_runtime": 95.1864, "eval_samples_per_second": 18.448, "eval_steps_per_second": 0.294, "step": 103 }, { "epoch": 5.3, "learning_rate": 5e-06, "loss": 0.1898, "step": 110 }, { "epoch": 5.78, "learning_rate": 4.444444444444444e-06, "loss": 0.1814, "step": 120 }, { "epoch": 5.98, "eval_accuracy": 0.9060364464692483, "eval_f1_score": 0.9043444445100207, "eval_loss": 0.24107009172439575, "eval_precision": 0.9091049886409681, "eval_runtime": 96.0888, "eval_samples_per_second": 18.275, "eval_steps_per_second": 0.291, "step": 124 }, { "epoch": 6.27, "learning_rate": 3.88888888888889e-06, "loss": 0.1608, "step": 130 }, { "epoch": 6.75, "learning_rate": 3.3333333333333333e-06, "loss": 0.1481, "step": 140 }, { "epoch": 6.99, "eval_accuracy": 0.9100227790432802, "eval_f1_score": 0.9083931079494111, "eval_loss": 0.23449133336544037, "eval_precision": 0.9129716383829837, "eval_runtime": 94.2446, "eval_samples_per_second": 18.632, "eval_steps_per_second": 0.297, "step": 145 }, { "epoch": 7.23, "learning_rate": 2.7777777777777783e-06, "loss": 0.1487, "step": 150 }, { "epoch": 7.71, "learning_rate": 2.222222222222222e-06, "loss": 0.1468, "step": 160 }, { "epoch": 8.0, "eval_accuracy": 0.9071753986332574, "eval_f1_score": 0.9054886526694435, "eval_loss": 0.23397402465343475, "eval_precision": 0.9108051352873032, "eval_runtime": 94.4553, "eval_samples_per_second": 18.591, "eval_steps_per_second": 0.296, "step": 166 }, { "epoch": 8.19, "learning_rate": 1.6666666666666667e-06, "loss": 0.1592, "step": 170 }, { "epoch": 8.67, "learning_rate": 1.111111111111111e-06, "loss": 0.1336, "step": 180 }, { "epoch": 8.96, "eval_accuracy": 0.9265375854214123, "eval_f1_score": 0.9252257576617917, "eval_loss": 0.19247999787330627, "eval_precision": 0.9269521372101541, "eval_runtime": 95.6309, "eval_samples_per_second": 18.362, "eval_steps_per_second": 0.293, "step": 186 }, { "epoch": 9.16, "learning_rate": 5.555555555555555e-07, "loss": 0.1361, "step": 190 }, { "epoch": 9.64, "learning_rate": 0.0, "loss": 0.133, "step": 200 }, { "epoch": 9.64, "eval_accuracy": 0.9219817767653758, "eval_f1_score": 0.9207089542401595, "eval_loss": 0.20209115743637085, "eval_precision": 0.9235132299409764, "eval_runtime": 96.5665, "eval_samples_per_second": 18.184, "eval_steps_per_second": 0.29, "step": 200 }, { "epoch": 9.64, "step": 200, "total_flos": 3.9796565447839334e+18, "train_loss": 0.3648529052734375, "train_runtime": 9658.8821, "train_samples_per_second": 5.453, "train_steps_per_second": 0.021 } ], "max_steps": 200, "num_train_epochs": 10, "total_flos": 3.9796565447839334e+18, "trial_name": null, "trial_params": null }