|
{ |
|
"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 |
|
} |
|
|