amjadfqs's picture
Model save
ff789c0
{
"best_metric": 0.9536527886881383,
"best_model_checkpoint": "swin-base-patch4-window7-224-in22k-finetuned-brain-tumor-final_08/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": 2.6315789473684212e-05,
"loss": 1.3618,
"step": 10
},
{
"epoch": 0.99,
"eval_accuracy": 0.7541241162608012,
"eval_f1_score": 0.7431413920390226,
"eval_loss": 0.6237759590148926,
"eval_precision": 0.7821438740378028,
"eval_runtime": 145.202,
"eval_samples_per_second": 17.534,
"eval_steps_per_second": 0.179,
"step": 19
},
{
"epoch": 1.04,
"learning_rate": 4.970760233918128e-05,
"loss": 0.7616,
"step": 20
},
{
"epoch": 1.56,
"learning_rate": 4.678362573099415e-05,
"loss": 0.3833,
"step": 30
},
{
"epoch": 1.97,
"eval_accuracy": 0.8864886095836606,
"eval_f1_score": 0.8884029543399138,
"eval_loss": 0.3097341060638428,
"eval_precision": 0.8970411183921092,
"eval_runtime": 148.0889,
"eval_samples_per_second": 17.192,
"eval_steps_per_second": 0.176,
"step": 38
},
{
"epoch": 2.08,
"learning_rate": 4.3859649122807014e-05,
"loss": 0.2463,
"step": 40
},
{
"epoch": 2.6,
"learning_rate": 4.093567251461988e-05,
"loss": 0.2011,
"step": 50
},
{
"epoch": 2.96,
"eval_accuracy": 0.9053417124901807,
"eval_f1_score": 0.9077657506236181,
"eval_loss": 0.2599542438983917,
"eval_precision": 0.917053808216762,
"eval_runtime": 147.6542,
"eval_samples_per_second": 17.243,
"eval_steps_per_second": 0.176,
"step": 57
},
{
"epoch": 3.12,
"learning_rate": 3.8011695906432746e-05,
"loss": 0.1248,
"step": 60
},
{
"epoch": 3.64,
"learning_rate": 3.508771929824561e-05,
"loss": 0.1124,
"step": 70
},
{
"epoch": 4.0,
"eval_accuracy": 0.9328358208955224,
"eval_f1_score": 0.934236013338499,
"eval_loss": 0.17926859855651855,
"eval_precision": 0.9380812862955372,
"eval_runtime": 143.4334,
"eval_samples_per_second": 17.75,
"eval_steps_per_second": 0.181,
"step": 77
},
{
"epoch": 4.16,
"learning_rate": 3.216374269005848e-05,
"loss": 0.0832,
"step": 80
},
{
"epoch": 4.68,
"learning_rate": 2.9239766081871346e-05,
"loss": 0.0711,
"step": 90
},
{
"epoch": 4.99,
"eval_accuracy": 0.9497250589159466,
"eval_f1_score": 0.9509017188515625,
"eval_loss": 0.13851618766784668,
"eval_precision": 0.9522273063860834,
"eval_runtime": 148.4015,
"eval_samples_per_second": 17.156,
"eval_steps_per_second": 0.175,
"step": 96
},
{
"epoch": 5.19,
"learning_rate": 2.6315789473684212e-05,
"loss": 0.0612,
"step": 100
},
{
"epoch": 5.71,
"learning_rate": 2.3391812865497074e-05,
"loss": 0.0518,
"step": 110
},
{
"epoch": 5.97,
"eval_accuracy": 0.9485467399842891,
"eval_f1_score": 0.950064185369353,
"eval_loss": 0.15061290562152863,
"eval_precision": 0.95234085726758,
"eval_runtime": 142.4306,
"eval_samples_per_second": 17.875,
"eval_steps_per_second": 0.183,
"step": 115
},
{
"epoch": 6.23,
"learning_rate": 2.046783625730994e-05,
"loss": 0.0456,
"step": 120
},
{
"epoch": 6.75,
"learning_rate": 1.7543859649122806e-05,
"loss": 0.0393,
"step": 130
},
{
"epoch": 6.96,
"eval_accuracy": 0.9536527886881383,
"eval_f1_score": 0.9549266168938111,
"eval_loss": 0.14219628274440765,
"eval_precision": 0.9563791141223957,
"eval_runtime": 145.1506,
"eval_samples_per_second": 17.54,
"eval_steps_per_second": 0.179,
"step": 134
},
{
"epoch": 7.27,
"learning_rate": 1.4619883040935673e-05,
"loss": 0.0285,
"step": 140
},
{
"epoch": 7.79,
"learning_rate": 1.1695906432748537e-05,
"loss": 0.0361,
"step": 150
},
{
"epoch": 8.0,
"eval_accuracy": 0.9481539670070699,
"eval_f1_score": 0.9496892350749182,
"eval_loss": 0.1544964462518692,
"eval_precision": 0.9521837214200338,
"eval_runtime": 143.4209,
"eval_samples_per_second": 17.752,
"eval_steps_per_second": 0.181,
"step": 154
},
{
"epoch": 8.31,
"learning_rate": 8.771929824561403e-06,
"loss": 0.025,
"step": 160
},
{
"epoch": 8.83,
"learning_rate": 5.8479532163742686e-06,
"loss": 0.025,
"step": 170
},
{
"epoch": 8.99,
"eval_accuracy": 0.9501178318931658,
"eval_f1_score": 0.9516150235142697,
"eval_loss": 0.14823174476623535,
"eval_precision": 0.9540548356482033,
"eval_runtime": 148.466,
"eval_samples_per_second": 17.149,
"eval_steps_per_second": 0.175,
"step": 173
},
{
"epoch": 9.35,
"learning_rate": 2.9239766081871343e-06,
"loss": 0.0162,
"step": 180
},
{
"epoch": 9.87,
"learning_rate": 0.0,
"loss": 0.0204,
"step": 190
},
{
"epoch": 9.87,
"eval_accuracy": 0.9512961508248232,
"eval_f1_score": 0.952674470844708,
"eval_loss": 0.1474471539258957,
"eval_precision": 0.9549628106843154,
"eval_runtime": 143.2338,
"eval_samples_per_second": 17.775,
"eval_steps_per_second": 0.182,
"step": 190
},
{
"epoch": 9.87,
"step": 190,
"total_flos": 5.909979961466413e+18,
"train_loss": 0.19446192054372086,
"train_runtime": 18924.5884,
"train_samples_per_second": 4.035,
"train_steps_per_second": 0.01
}
],
"max_steps": 190,
"num_train_epochs": 10,
"total_flos": 5.909979961466413e+18,
"trial_name": null,
"trial_params": null
}