{ "best_metric": 0.9760408483896308, "best_model_checkpoint": "swin-base-patch4-window7-224-in22k-finetuned-brain-tumor-final_12/checkpoint-270", "epoch": 10.0, "global_step": 300, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.33, "learning_rate": 3.3333333333333335e-05, "loss": 1.3016, "step": 10 }, { "epoch": 0.67, "learning_rate": 6.666666666666667e-05, "loss": 0.6828, "step": 20 }, { "epoch": 1.0, "learning_rate": 0.0001, "loss": 0.3644, "step": 30 }, { "epoch": 1.0, "eval_accuracy": 0.8955223880597015, "eval_f1_score": 0.8974001833670424, "eval_loss": 0.29182684421539307, "eval_precision": 0.907029413169219, "eval_runtime": 141.6635, "eval_samples_per_second": 17.972, "eval_sensitivity": 0.8957445034204233, "eval_specificity": 0.9733575562525472, "eval_steps_per_second": 0.282, "step": 30 }, { "epoch": 1.33, "learning_rate": 9.62962962962963e-05, "loss": 0.2491, "step": 40 }, { "epoch": 1.67, "learning_rate": 9.25925925925926e-05, "loss": 0.1871, "step": 50 }, { "epoch": 2.0, "learning_rate": 8.888888888888889e-05, "loss": 0.2177, "step": 60 }, { "epoch": 2.0, "eval_accuracy": 0.9151610369206599, "eval_f1_score": 0.9154763574988526, "eval_loss": 0.231949582695961, "eval_precision": 0.9237202260263041, "eval_runtime": 141.1838, "eval_samples_per_second": 18.033, "eval_sensitivity": 0.9156321134434031, "eval_specificity": 0.9785744694099139, "eval_steps_per_second": 0.283, "step": 60 }, { "epoch": 2.33, "learning_rate": 8.518518518518518e-05, "loss": 0.1231, "step": 70 }, { "epoch": 2.67, "learning_rate": 8.148148148148148e-05, "loss": 0.1208, "step": 80 }, { "epoch": 3.0, "learning_rate": 7.777777777777778e-05, "loss": 0.1171, "step": 90 }, { "epoch": 3.0, "eval_accuracy": 0.9489395129615082, "eval_f1_score": 0.9493719922653666, "eval_loss": 0.1653948426246643, "eval_precision": 0.9531974738275035, "eval_runtime": 141.9016, "eval_samples_per_second": 17.942, "eval_sensitivity": 0.9492288551886346, "eval_specificity": 0.9871567326622118, "eval_steps_per_second": 0.282, "step": 90 }, { "epoch": 3.33, "learning_rate": 7.407407407407407e-05, "loss": 0.0833, "step": 100 }, { "epoch": 3.67, "learning_rate": 7.037037037037038e-05, "loss": 0.0658, "step": 110 }, { "epoch": 4.0, "learning_rate": 6.666666666666667e-05, "loss": 0.068, "step": 120 }, { "epoch": 4.0, "eval_accuracy": 0.9450117831893166, "eval_f1_score": 0.9450733960859783, "eval_loss": 0.15997998416423798, "eval_precision": 0.9465681527498646, "eval_runtime": 141.3669, "eval_samples_per_second": 18.01, "eval_sensitivity": 0.9454632114511998, "eval_specificity": 0.986114525355863, "eval_steps_per_second": 0.283, "step": 120 }, { "epoch": 4.33, "learning_rate": 6.296296296296296e-05, "loss": 0.0632, "step": 130 }, { "epoch": 4.67, "learning_rate": 5.925925925925926e-05, "loss": 0.0664, "step": 140 }, { "epoch": 5.0, "learning_rate": 5.555555555555556e-05, "loss": 0.0499, "step": 150 }, { "epoch": 5.0, "eval_accuracy": 0.9654359780047133, "eval_f1_score": 0.9655558479515587, "eval_loss": 0.0947260707616806, "eval_precision": 0.9656214890103378, "eval_runtime": 140.2722, "eval_samples_per_second": 18.15, "eval_sensitivity": 0.9656782254257974, "eval_specificity": 0.9912799165215678, "eval_steps_per_second": 0.285, "step": 150 }, { "epoch": 5.33, "learning_rate": 5.185185185185185e-05, "loss": 0.0397, "step": 160 }, { "epoch": 5.67, "learning_rate": 4.814814814814815e-05, "loss": 0.0364, "step": 170 }, { "epoch": 6.0, "learning_rate": 4.4444444444444447e-05, "loss": 0.0302, "step": 180 }, { "epoch": 6.0, "eval_accuracy": 0.9713275726630007, "eval_f1_score": 0.9713961045425188, "eval_loss": 0.08822068572044373, "eval_precision": 0.9714994973822426, "eval_runtime": 140.5761, "eval_samples_per_second": 18.111, "eval_sensitivity": 0.9715333079033974, "eval_specificity": 0.9927834971929292, "eval_steps_per_second": 0.285, "step": 180 }, { "epoch": 6.33, "learning_rate": 4.074074074074074e-05, "loss": 0.0221, "step": 190 }, { "epoch": 6.67, "learning_rate": 3.7037037037037037e-05, "loss": 0.0299, "step": 200 }, { "epoch": 7.0, "learning_rate": 3.3333333333333335e-05, "loss": 0.0207, "step": 210 }, { "epoch": 7.0, "eval_accuracy": 0.9697564807541241, "eval_f1_score": 0.9698988819326013, "eval_loss": 0.10023057460784912, "eval_precision": 0.9707690938825231, "eval_runtime": 141.3669, "eval_samples_per_second": 18.01, "eval_sensitivity": 0.9698979472474506, "eval_specificity": 0.9924055150633387, "eval_steps_per_second": 0.283, "step": 210 }, { "epoch": 7.33, "learning_rate": 2.962962962962963e-05, "loss": 0.0238, "step": 220 }, { "epoch": 7.67, "learning_rate": 2.5925925925925925e-05, "loss": 0.0232, "step": 230 }, { "epoch": 8.0, "learning_rate": 2.2222222222222223e-05, "loss": 0.0205, "step": 240 }, { "epoch": 8.0, "eval_accuracy": 0.9524744697564808, "eval_f1_score": 0.9521142281383307, "eval_loss": 0.15503662824630737, "eval_precision": 0.9544453536467878, "eval_runtime": 139.2147, "eval_samples_per_second": 18.288, "eval_sensitivity": 0.9528977045128385, "eval_specificity": 0.9880677808840496, "eval_steps_per_second": 0.287, "step": 240 }, { "epoch": 8.33, "learning_rate": 1.8518518518518518e-05, "loss": 0.0166, "step": 250 }, { "epoch": 8.67, "learning_rate": 1.4814814814814815e-05, "loss": 0.0184, "step": 260 }, { "epoch": 9.0, "learning_rate": 1.1111111111111112e-05, "loss": 0.0163, "step": 270 }, { "epoch": 9.0, "eval_accuracy": 0.9760408483896308, "eval_f1_score": 0.9760955386577261, "eval_loss": 0.07892601191997528, "eval_precision": 0.9762470546227865, "eval_runtime": 142.9926, "eval_samples_per_second": 17.805, "eval_sensitivity": 0.976185641458774, "eval_specificity": 0.9939801768095587, "eval_steps_per_second": 0.28, "step": 270 }, { "epoch": 9.33, "learning_rate": 7.4074074074074075e-06, "loss": 0.0108, "step": 280 }, { "epoch": 9.67, "learning_rate": 3.7037037037037037e-06, "loss": 0.0097, "step": 290 }, { "epoch": 10.0, "learning_rate": 0.0, "loss": 0.0181, "step": 300 }, { "epoch": 10.0, "eval_accuracy": 0.9736842105263158, "eval_f1_score": 0.9737106615818243, "eval_loss": 0.09225168079137802, "eval_precision": 0.973953427126813, "eval_runtime": 141.4794, "eval_samples_per_second": 17.996, "eval_sensitivity": 0.9738430623943066, "eval_specificity": 0.9933886802049243, "eval_steps_per_second": 0.283, "step": 300 }, { "epoch": 10.0, "step": 300, "total_flos": 5.983391481940132e+18, "train_loss": 0.1365605974694093, "train_runtime": 14584.4296, "train_samples_per_second": 5.236, "train_steps_per_second": 0.021 } ], "max_steps": 300, "num_train_epochs": 10, "total_flos": 5.983391481940132e+18, "trial_name": null, "trial_params": null }