amjadfqs's picture
End of training
563c0dd
{
"best_metric": 0.9842906234658811,
"best_model_checkpoint": "swin-base-patch4-window7-224-in22k-finetuned-brain-tumor-skullStrippded_04/checkpoint-123",
"epoch": 9.75609756097561,
"global_step": 200,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.62,
"learning_rate": 6.25e-05,
"loss": 1.2662,
"step": 10
},
{
"epoch": 1.0,
"eval_accuracy": 0.8370152184585175,
"eval_f1_score": 0.8309274634035486,
"eval_loss": 0.44239601492881775,
"eval_precision": 0.8463813017583761,
"eval_runtime": 208.875,
"eval_samples_per_second": 9.752,
"eval_steps_per_second": 0.077,
"step": 16
},
{
"epoch": 0.98,
"learning_rate": 0.0001,
"loss": 0.3778,
"step": 20
},
{
"epoch": 0.98,
"eval_accuracy": 0.9062346588119784,
"eval_f1_score": 0.9066525088672521,
"eval_loss": 0.2700486481189728,
"eval_precision": 0.9071937892005743,
"eval_runtime": 204.4765,
"eval_samples_per_second": 9.962,
"eval_steps_per_second": 0.103,
"step": 20
},
{
"epoch": 1.46,
"learning_rate": 9.444444444444444e-05,
"loss": 0.2951,
"step": 30
},
{
"epoch": 1.95,
"learning_rate": 8.888888888888889e-05,
"loss": 0.2377,
"step": 40
},
{
"epoch": 2.0,
"eval_accuracy": 0.9229258713794797,
"eval_f1_score": 0.9233847000973788,
"eval_loss": 0.20353713631629944,
"eval_precision": 0.9268537243090427,
"eval_runtime": 208.169,
"eval_samples_per_second": 9.785,
"eval_steps_per_second": 0.101,
"step": 41
},
{
"epoch": 2.44,
"learning_rate": 8.333333333333334e-05,
"loss": 0.1676,
"step": 50
},
{
"epoch": 2.93,
"learning_rate": 7.777777777777778e-05,
"loss": 0.1201,
"step": 60
},
{
"epoch": 2.98,
"eval_accuracy": 0.9464899361806578,
"eval_f1_score": 0.9466624955018323,
"eval_loss": 0.13448551297187805,
"eval_precision": 0.9512180953289231,
"eval_runtime": 206.9703,
"eval_samples_per_second": 9.842,
"eval_steps_per_second": 0.101,
"step": 61
},
{
"epoch": 3.41,
"learning_rate": 7.222222222222222e-05,
"loss": 0.1025,
"step": 70
},
{
"epoch": 3.9,
"learning_rate": 6.666666666666667e-05,
"loss": 0.0774,
"step": 80
},
{
"epoch": 4.0,
"eval_accuracy": 0.9612174766813942,
"eval_f1_score": 0.9618098339374384,
"eval_loss": 0.1229424849152565,
"eval_precision": 0.9643105703851083,
"eval_runtime": 205.1391,
"eval_samples_per_second": 9.93,
"eval_steps_per_second": 0.102,
"step": 82
},
{
"epoch": 4.39,
"learning_rate": 6.111111111111112e-05,
"loss": 0.058,
"step": 90
},
{
"epoch": 4.88,
"learning_rate": 5.555555555555556e-05,
"loss": 0.0495,
"step": 100
},
{
"epoch": 4.98,
"eval_accuracy": 0.9813451153657339,
"eval_f1_score": 0.9815016901025415,
"eval_loss": 0.05617663636803627,
"eval_precision": 0.9816047800258326,
"eval_runtime": 201.6332,
"eval_samples_per_second": 10.103,
"eval_steps_per_second": 0.104,
"step": 102
},
{
"epoch": 5.37,
"learning_rate": 5e-05,
"loss": 0.035,
"step": 110
},
{
"epoch": 5.85,
"learning_rate": 4.4444444444444447e-05,
"loss": 0.0358,
"step": 120
},
{
"epoch": 6.0,
"eval_accuracy": 0.9842906234658811,
"eval_f1_score": 0.9844072801091768,
"eval_loss": 0.04700547456741333,
"eval_precision": 0.9845888529063952,
"eval_runtime": 201.9642,
"eval_samples_per_second": 10.086,
"eval_steps_per_second": 0.104,
"step": 123
},
{
"epoch": 6.34,
"learning_rate": 3.888888888888889e-05,
"loss": 0.0301,
"step": 130
},
{
"epoch": 6.83,
"learning_rate": 3.3333333333333335e-05,
"loss": 0.0228,
"step": 140
},
{
"epoch": 6.98,
"eval_accuracy": 0.9833087874324987,
"eval_f1_score": 0.9834387946272402,
"eval_loss": 0.04467308893799782,
"eval_precision": 0.9836051007366963,
"eval_runtime": 201.6017,
"eval_samples_per_second": 10.104,
"eval_steps_per_second": 0.104,
"step": 143
},
{
"epoch": 7.32,
"learning_rate": 2.777777777777778e-05,
"loss": 0.021,
"step": 150
},
{
"epoch": 7.8,
"learning_rate": 2.2222222222222223e-05,
"loss": 0.0181,
"step": 160
},
{
"epoch": 8.0,
"eval_accuracy": 0.9828178694158075,
"eval_f1_score": 0.982990812967336,
"eval_loss": 0.04651310294866562,
"eval_precision": 0.9830987734058751,
"eval_runtime": 200.6389,
"eval_samples_per_second": 10.153,
"eval_steps_per_second": 0.105,
"step": 164
},
{
"epoch": 8.29,
"learning_rate": 1.6666666666666667e-05,
"loss": 0.0128,
"step": 170
},
{
"epoch": 8.78,
"learning_rate": 1.1111111111111112e-05,
"loss": 0.0132,
"step": 180
},
{
"epoch": 8.98,
"eval_accuracy": 0.9833087874324987,
"eval_f1_score": 0.9834792106943269,
"eval_loss": 0.043627530336380005,
"eval_precision": 0.9835721710721711,
"eval_runtime": 197.7113,
"eval_samples_per_second": 10.303,
"eval_steps_per_second": 0.106,
"step": 184
},
{
"epoch": 9.27,
"learning_rate": 5.555555555555556e-06,
"loss": 0.0097,
"step": 190
},
{
"epoch": 9.76,
"learning_rate": 0.0,
"loss": 0.0126,
"step": 200
},
{
"epoch": 9.76,
"eval_accuracy": 0.9837997054491899,
"eval_f1_score": 0.983950241371011,
"eval_loss": 0.04606052488088608,
"eval_precision": 0.9840362340362341,
"eval_runtime": 197.1193,
"eval_samples_per_second": 10.334,
"eval_steps_per_second": 0.107,
"step": 200
},
{
"epoch": 9.76,
"step": 200,
"total_flos": 6.366510302768824e+18,
"train_loss": 0.0735103191435337,
"train_runtime": 27763.2532,
"train_samples_per_second": 2.934,
"train_steps_per_second": 0.007
}
],
"max_steps": 200,
"num_train_epochs": 10,
"total_flos": 6.366510302768824e+18,
"trial_name": null,
"trial_params": null
}