amjadfqs's picture
Model save
9db0a2e
{
"best_metric": 0.9799685781618225,
"best_model_checkpoint": "swin-base-patch4-window7-224-in22k-finetuned-brain-tumor-final_11/checkpoint-134",
"epoch": 9.87012987012987,
"global_step": 190,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.52,
"learning_rate": 5.2631578947368424e-05,
"loss": 1.2367,
"step": 10
},
{
"epoch": 0.99,
"eval_accuracy": 0.8648860958366065,
"eval_f1_score": 0.8629101056918216,
"eval_loss": 0.35601529479026794,
"eval_precision": 0.8649012039898544,
"eval_runtime": 141.8367,
"eval_samples_per_second": 17.95,
"eval_sensitivity": 0.8629212850751643,
"eval_specificity": 0.9647885911185192,
"eval_steps_per_second": 0.183,
"step": 19
},
{
"epoch": 1.04,
"learning_rate": 9.941520467836257e-05,
"loss": 0.551,
"step": 20
},
{
"epoch": 1.56,
"learning_rate": 9.35672514619883e-05,
"loss": 0.2911,
"step": 30
},
{
"epoch": 1.97,
"eval_accuracy": 0.9296936370777691,
"eval_f1_score": 0.9289731644434347,
"eval_loss": 0.20870551466941833,
"eval_precision": 0.9335105762764986,
"eval_runtime": 142.1997,
"eval_samples_per_second": 17.904,
"eval_sensitivity": 0.9277077502089683,
"eval_specificity": 0.9821821858941076,
"eval_steps_per_second": 0.183,
"step": 38
},
{
"epoch": 2.08,
"learning_rate": 8.771929824561403e-05,
"loss": 0.2185,
"step": 40
},
{
"epoch": 2.6,
"learning_rate": 8.187134502923976e-05,
"loss": 0.1681,
"step": 50
},
{
"epoch": 2.96,
"eval_accuracy": 0.9564021995286724,
"eval_f1_score": 0.9557514024509197,
"eval_loss": 0.13926906883716583,
"eval_precision": 0.9557781221855981,
"eval_runtime": 143.2364,
"eval_samples_per_second": 17.775,
"eval_sensitivity": 0.9561759465584563,
"eval_specificity": 0.98897388236281,
"eval_steps_per_second": 0.182,
"step": 57
},
{
"epoch": 3.12,
"learning_rate": 7.602339181286549e-05,
"loss": 0.1165,
"step": 60
},
{
"epoch": 3.64,
"learning_rate": 7.017543859649122e-05,
"loss": 0.0923,
"step": 70
},
{
"epoch": 4.0,
"eval_accuracy": 0.9642576590730558,
"eval_f1_score": 0.9639088702338641,
"eval_loss": 0.11055397987365723,
"eval_precision": 0.9637231665579402,
"eval_runtime": 143.252,
"eval_samples_per_second": 17.773,
"eval_sensitivity": 0.9646712281479681,
"eval_specificity": 0.9909880590481308,
"eval_steps_per_second": 0.181,
"step": 77
},
{
"epoch": 4.16,
"learning_rate": 6.432748538011695e-05,
"loss": 0.0542,
"step": 80
},
{
"epoch": 4.68,
"learning_rate": 5.847953216374269e-05,
"loss": 0.0602,
"step": 90
},
{
"epoch": 4.99,
"eval_accuracy": 0.950510604870385,
"eval_f1_score": 0.9493780293983679,
"eval_loss": 0.1510021686553955,
"eval_precision": 0.9511714907126886,
"eval_runtime": 148.5403,
"eval_samples_per_second": 17.14,
"eval_sensitivity": 0.9504186601773192,
"eval_specificity": 0.9875279599551886,
"eval_steps_per_second": 0.175,
"step": 96
},
{
"epoch": 5.19,
"learning_rate": 5.2631578947368424e-05,
"loss": 0.0668,
"step": 100
},
{
"epoch": 5.71,
"learning_rate": 4.678362573099415e-05,
"loss": 0.0388,
"step": 110
},
{
"epoch": 5.97,
"eval_accuracy": 0.9666142969363708,
"eval_f1_score": 0.9666648974137618,
"eval_loss": 0.1145075261592865,
"eval_precision": 0.9670209795949075,
"eval_runtime": 144.463,
"eval_samples_per_second": 17.624,
"eval_sensitivity": 0.9672481450663073,
"eval_specificity": 0.9916021588463255,
"eval_steps_per_second": 0.18,
"step": 115
},
{
"epoch": 6.23,
"learning_rate": 4.093567251461988e-05,
"loss": 0.0304,
"step": 120
},
{
"epoch": 6.75,
"learning_rate": 3.508771929824561e-05,
"loss": 0.0197,
"step": 130
},
{
"epoch": 6.96,
"eval_accuracy": 0.9799685781618225,
"eval_f1_score": 0.9796169238566794,
"eval_loss": 0.07831881195306778,
"eval_precision": 0.9796898193485958,
"eval_runtime": 142.3671,
"eval_samples_per_second": 17.883,
"eval_sensitivity": 0.9796353926896586,
"eval_specificity": 0.9949621175641323,
"eval_steps_per_second": 0.183,
"step": 134
},
{
"epoch": 7.27,
"learning_rate": 2.9239766081871346e-05,
"loss": 0.0222,
"step": 140
},
{
"epoch": 7.79,
"learning_rate": 2.3391812865497074e-05,
"loss": 0.0172,
"step": 150
},
{
"epoch": 8.0,
"eval_accuracy": 0.9713275726630007,
"eval_f1_score": 0.9712887784402073,
"eval_loss": 0.10316599160432816,
"eval_precision": 0.9714643584340791,
"eval_runtime": 142.3047,
"eval_samples_per_second": 17.891,
"eval_sensitivity": 0.9717717854874542,
"eval_specificity": 0.992796790353424,
"eval_steps_per_second": 0.183,
"step": 154
},
{
"epoch": 8.31,
"learning_rate": 1.7543859649122806e-05,
"loss": 0.013,
"step": 160
},
{
"epoch": 8.83,
"learning_rate": 1.1695906432748537e-05,
"loss": 0.0169,
"step": 170
},
{
"epoch": 8.99,
"eval_accuracy": 0.9776119402985075,
"eval_f1_score": 0.977208698367329,
"eval_loss": 0.08539657294750214,
"eval_precision": 0.9771017028074661,
"eval_runtime": 147.2284,
"eval_samples_per_second": 17.293,
"eval_sensitivity": 0.9774414536608387,
"eval_specificity": 0.9943698997404483,
"eval_steps_per_second": 0.177,
"step": 173
},
{
"epoch": 9.35,
"learning_rate": 5.8479532163742686e-06,
"loss": 0.0128,
"step": 180
},
{
"epoch": 9.87,
"learning_rate": 0.0,
"loss": 0.0104,
"step": 190
},
{
"epoch": 9.87,
"eval_accuracy": 0.9752553024351924,
"eval_f1_score": 0.9749649987709283,
"eval_loss": 0.094758041203022,
"eval_precision": 0.9748580935041002,
"eval_runtime": 144.0279,
"eval_samples_per_second": 17.677,
"eval_sensitivity": 0.9753290664582137,
"eval_specificity": 0.9937797036439378,
"eval_steps_per_second": 0.181,
"step": 190
},
{
"epoch": 9.87,
"step": 190,
"total_flos": 5.909979961466413e+18,
"train_loss": 0.15983055751574668,
"train_runtime": 18885.9487,
"train_samples_per_second": 4.044,
"train_steps_per_second": 0.01
}
],
"max_steps": 190,
"num_train_epochs": 10,
"total_flos": 5.909979961466413e+18,
"trial_name": null,
"trial_params": null
}