amjadfqs's picture
Model save
a206e5c
{
"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
}