amjadfqs's picture
Model save
5b0230a
{
"best_metric": 0.9265375854214123,
"best_model_checkpoint": "swin-base-patch4-window7-224-in22k-finetuned-brain-tumor-final/checkpoint-186",
"epoch": 9.638554216867469,
"global_step": 200,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.48,
"learning_rate": 5e-06,
"loss": 1.3805,
"step": 10
},
{
"epoch": 0.96,
"learning_rate": 1e-05,
"loss": 1.2212,
"step": 20
},
{
"epoch": 0.96,
"eval_accuracy": 0.6429384965831435,
"eval_f1_score": 0.6225332030044981,
"eval_loss": 1.140669822692871,
"eval_precision": 0.6600729727071496,
"eval_runtime": 97.1469,
"eval_samples_per_second": 18.076,
"eval_steps_per_second": 0.288,
"step": 20
},
{
"epoch": 1.45,
"learning_rate": 9.444444444444445e-06,
"loss": 0.8664,
"step": 30
},
{
"epoch": 1.93,
"learning_rate": 8.888888888888888e-06,
"loss": 0.565,
"step": 40
},
{
"epoch": 1.98,
"eval_accuracy": 0.8325740318906606,
"eval_f1_score": 0.8311373357165648,
"eval_loss": 0.5161515474319458,
"eval_precision": 0.8427982091540198,
"eval_runtime": 98.3593,
"eval_samples_per_second": 17.853,
"eval_steps_per_second": 0.285,
"step": 41
},
{
"epoch": 2.41,
"learning_rate": 8.333333333333334e-06,
"loss": 0.4226,
"step": 50
},
{
"epoch": 2.89,
"learning_rate": 7.77777777777778e-06,
"loss": 0.3245,
"step": 60
},
{
"epoch": 2.99,
"eval_accuracy": 0.8804100227790432,
"eval_f1_score": 0.8784127627414283,
"eval_loss": 0.3265204429626465,
"eval_precision": 0.8843478014022281,
"eval_runtime": 94.9107,
"eval_samples_per_second": 18.502,
"eval_steps_per_second": 0.295,
"step": 62
},
{
"epoch": 3.37,
"learning_rate": 7.222222222222223e-06,
"loss": 0.2761,
"step": 70
},
{
"epoch": 3.86,
"learning_rate": 6.666666666666667e-06,
"loss": 0.2618,
"step": 80
},
{
"epoch": 4.0,
"eval_accuracy": 0.9066059225512528,
"eval_f1_score": 0.9053659025451456,
"eval_loss": 0.2712633013725281,
"eval_precision": 0.9105145368026273,
"eval_runtime": 95.6192,
"eval_samples_per_second": 18.365,
"eval_steps_per_second": 0.293,
"step": 83
},
{
"epoch": 4.34,
"learning_rate": 6.111111111111112e-06,
"loss": 0.225,
"step": 90
},
{
"epoch": 4.82,
"learning_rate": 5.555555555555557e-06,
"loss": 0.2164,
"step": 100
},
{
"epoch": 4.96,
"eval_accuracy": 0.8946469248291572,
"eval_f1_score": 0.8929116621992249,
"eval_loss": 0.28123489022254944,
"eval_precision": 0.899366339703112,
"eval_runtime": 95.1864,
"eval_samples_per_second": 18.448,
"eval_steps_per_second": 0.294,
"step": 103
},
{
"epoch": 5.3,
"learning_rate": 5e-06,
"loss": 0.1898,
"step": 110
},
{
"epoch": 5.78,
"learning_rate": 4.444444444444444e-06,
"loss": 0.1814,
"step": 120
},
{
"epoch": 5.98,
"eval_accuracy": 0.9060364464692483,
"eval_f1_score": 0.9043444445100207,
"eval_loss": 0.24107009172439575,
"eval_precision": 0.9091049886409681,
"eval_runtime": 96.0888,
"eval_samples_per_second": 18.275,
"eval_steps_per_second": 0.291,
"step": 124
},
{
"epoch": 6.27,
"learning_rate": 3.88888888888889e-06,
"loss": 0.1608,
"step": 130
},
{
"epoch": 6.75,
"learning_rate": 3.3333333333333333e-06,
"loss": 0.1481,
"step": 140
},
{
"epoch": 6.99,
"eval_accuracy": 0.9100227790432802,
"eval_f1_score": 0.9083931079494111,
"eval_loss": 0.23449133336544037,
"eval_precision": 0.9129716383829837,
"eval_runtime": 94.2446,
"eval_samples_per_second": 18.632,
"eval_steps_per_second": 0.297,
"step": 145
},
{
"epoch": 7.23,
"learning_rate": 2.7777777777777783e-06,
"loss": 0.1487,
"step": 150
},
{
"epoch": 7.71,
"learning_rate": 2.222222222222222e-06,
"loss": 0.1468,
"step": 160
},
{
"epoch": 8.0,
"eval_accuracy": 0.9071753986332574,
"eval_f1_score": 0.9054886526694435,
"eval_loss": 0.23397402465343475,
"eval_precision": 0.9108051352873032,
"eval_runtime": 94.4553,
"eval_samples_per_second": 18.591,
"eval_steps_per_second": 0.296,
"step": 166
},
{
"epoch": 8.19,
"learning_rate": 1.6666666666666667e-06,
"loss": 0.1592,
"step": 170
},
{
"epoch": 8.67,
"learning_rate": 1.111111111111111e-06,
"loss": 0.1336,
"step": 180
},
{
"epoch": 8.96,
"eval_accuracy": 0.9265375854214123,
"eval_f1_score": 0.9252257576617917,
"eval_loss": 0.19247999787330627,
"eval_precision": 0.9269521372101541,
"eval_runtime": 95.6309,
"eval_samples_per_second": 18.362,
"eval_steps_per_second": 0.293,
"step": 186
},
{
"epoch": 9.16,
"learning_rate": 5.555555555555555e-07,
"loss": 0.1361,
"step": 190
},
{
"epoch": 9.64,
"learning_rate": 0.0,
"loss": 0.133,
"step": 200
},
{
"epoch": 9.64,
"eval_accuracy": 0.9219817767653758,
"eval_f1_score": 0.9207089542401595,
"eval_loss": 0.20209115743637085,
"eval_precision": 0.9235132299409764,
"eval_runtime": 96.5665,
"eval_samples_per_second": 18.184,
"eval_steps_per_second": 0.29,
"step": 200
},
{
"epoch": 9.64,
"step": 200,
"total_flos": 3.9796565447839334e+18,
"train_loss": 0.3648529052734375,
"train_runtime": 9658.8821,
"train_samples_per_second": 5.453,
"train_steps_per_second": 0.021
}
],
"max_steps": 200,
"num_train_epochs": 10,
"total_flos": 3.9796565447839334e+18,
"trial_name": null,
"trial_params": null
}