|
{ |
|
"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 |
|
} |
|
|