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