|
{ |
|
"best_metric": 0.048169028013944626, |
|
"best_model_checkpoint": "google/vit-base-patch16-224-in21k_covid_19_ct_scans/checkpoint-432", |
|
"epoch": 12.0, |
|
"eval_steps": 500, |
|
"global_step": 648, |
|
"is_hyper_param_search": false, |
|
"is_local_process_zero": true, |
|
"is_world_process_zero": true, |
|
"log_history": [ |
|
{ |
|
"epoch": 0.018518518518518517, |
|
"grad_norm": 3.8348426818847656, |
|
"learning_rate": 0.00019969135802469138, |
|
"loss": 0.6804, |
|
"step": 1 |
|
}, |
|
{ |
|
"epoch": 1.0, |
|
"eval_AUC": 0.5, |
|
"eval_F1": 0.47277227722772275, |
|
"eval_Precision": 0.8967136150234741, |
|
"eval_Recall": 1.0, |
|
"eval_accuracy": 0.8967136150234741, |
|
"eval_loss": 0.38209885358810425, |
|
"eval_runtime": 4.2504, |
|
"eval_samples_per_second": 50.113, |
|
"eval_steps_per_second": 3.294, |
|
"step": 54 |
|
}, |
|
{ |
|
"epoch": 2.0, |
|
"eval_AUC": 0.5, |
|
"eval_F1": 0.47277227722772275, |
|
"eval_Precision": 0.8967136150234741, |
|
"eval_Recall": 1.0, |
|
"eval_accuracy": 0.8967136150234741, |
|
"eval_loss": 0.413409024477005, |
|
"eval_runtime": 4.3554, |
|
"eval_samples_per_second": 48.904, |
|
"eval_steps_per_second": 3.214, |
|
"step": 108 |
|
}, |
|
{ |
|
"epoch": 3.0, |
|
"eval_AUC": 0.5454545454545454, |
|
"eval_F1": 0.5584577114427861, |
|
"eval_Precision": 0.9052132701421801, |
|
"eval_Recall": 1.0, |
|
"eval_accuracy": 0.9061032863849765, |
|
"eval_loss": 0.27077189087867737, |
|
"eval_runtime": 4.2214, |
|
"eval_samples_per_second": 50.458, |
|
"eval_steps_per_second": 3.316, |
|
"step": 162 |
|
}, |
|
{ |
|
"epoch": 4.0, |
|
"eval_AUC": 0.7272727272727273, |
|
"eval_F1": 0.797271573604061, |
|
"eval_Precision": 0.9408866995073891, |
|
"eval_Recall": 1.0, |
|
"eval_accuracy": 0.9436619718309859, |
|
"eval_loss": 0.2405475229024887, |
|
"eval_runtime": 4.1493, |
|
"eval_samples_per_second": 51.335, |
|
"eval_steps_per_second": 3.374, |
|
"step": 216 |
|
}, |
|
{ |
|
"epoch": 5.0, |
|
"eval_AUC": 0.7272727272727273, |
|
"eval_F1": 0.797271573604061, |
|
"eval_Precision": 0.9408866995073891, |
|
"eval_Recall": 1.0, |
|
"eval_accuracy": 0.9436619718309859, |
|
"eval_loss": 0.21931496262550354, |
|
"eval_runtime": 4.3053, |
|
"eval_samples_per_second": 49.474, |
|
"eval_steps_per_second": 3.252, |
|
"step": 270 |
|
}, |
|
{ |
|
"epoch": 6.0, |
|
"eval_AUC": 0.9309852451213707, |
|
"eval_F1": 0.8774901960784314, |
|
"eval_Precision": 0.9891304347826086, |
|
"eval_Recall": 0.9528795811518325, |
|
"eval_accuracy": 0.9483568075117371, |
|
"eval_loss": 0.1719074547290802, |
|
"eval_runtime": 4.1609, |
|
"eval_samples_per_second": 51.191, |
|
"eval_steps_per_second": 3.365, |
|
"step": 324 |
|
}, |
|
{ |
|
"epoch": 7.0, |
|
"eval_AUC": 0.9519276534983341, |
|
"eval_F1": 0.9611998299836055, |
|
"eval_Precision": 0.9895833333333334, |
|
"eval_Recall": 0.9947643979057592, |
|
"eval_accuracy": 0.9859154929577465, |
|
"eval_loss": 0.05248570814728737, |
|
"eval_runtime": 4.1623, |
|
"eval_samples_per_second": 51.174, |
|
"eval_steps_per_second": 3.364, |
|
"step": 378 |
|
}, |
|
{ |
|
"epoch": 8.0, |
|
"eval_AUC": 0.9545454545454545, |
|
"eval_F1": 0.9735863095238095, |
|
"eval_Precision": 0.9896373056994818, |
|
"eval_Recall": 1.0, |
|
"eval_accuracy": 0.9906103286384976, |
|
"eval_loss": 0.048169028013944626, |
|
"eval_runtime": 4.2766, |
|
"eval_samples_per_second": 49.806, |
|
"eval_steps_per_second": 3.274, |
|
"step": 432 |
|
}, |
|
{ |
|
"epoch": 9.0, |
|
"eval_AUC": 0.8863636363636364, |
|
"eval_F1": 0.9294374875770225, |
|
"eval_Precision": 0.9744897959183674, |
|
"eval_Recall": 1.0, |
|
"eval_accuracy": 0.9765258215962441, |
|
"eval_loss": 0.09067811071872711, |
|
"eval_runtime": 4.2163, |
|
"eval_samples_per_second": 50.519, |
|
"eval_steps_per_second": 3.32, |
|
"step": 486 |
|
}, |
|
{ |
|
"epoch": 9.25925925925926, |
|
"grad_norm": 0.03957719728350639, |
|
"learning_rate": 4.567901234567901e-05, |
|
"loss": 0.1258, |
|
"step": 500 |
|
}, |
|
{ |
|
"epoch": 10.0, |
|
"eval_AUC": 0.8863636363636364, |
|
"eval_F1": 0.9294374875770225, |
|
"eval_Precision": 0.9744897959183674, |
|
"eval_Recall": 1.0, |
|
"eval_accuracy": 0.9765258215962441, |
|
"eval_loss": 0.10092553496360779, |
|
"eval_runtime": 4.2027, |
|
"eval_samples_per_second": 50.682, |
|
"eval_steps_per_second": 3.331, |
|
"step": 540 |
|
}, |
|
{ |
|
"epoch": 11.0, |
|
"eval_AUC": 0.8863636363636364, |
|
"eval_F1": 0.9294374875770225, |
|
"eval_Precision": 0.9744897959183674, |
|
"eval_Recall": 1.0, |
|
"eval_accuracy": 0.9765258215962441, |
|
"eval_loss": 0.10514290630817413, |
|
"eval_runtime": 4.2216, |
|
"eval_samples_per_second": 50.454, |
|
"eval_steps_per_second": 3.316, |
|
"step": 594 |
|
}, |
|
{ |
|
"epoch": 12.0, |
|
"eval_AUC": 0.8863636363636364, |
|
"eval_F1": 0.9294374875770225, |
|
"eval_Precision": 0.9744897959183674, |
|
"eval_Recall": 1.0, |
|
"eval_accuracy": 0.9765258215962441, |
|
"eval_loss": 0.10642894357442856, |
|
"eval_runtime": 4.2244, |
|
"eval_samples_per_second": 50.421, |
|
"eval_steps_per_second": 3.314, |
|
"step": 648 |
|
}, |
|
{ |
|
"epoch": 12.0, |
|
"step": 648, |
|
"total_flos": 7.894883901934633e+17, |
|
"train_loss": 0.09885871769101531, |
|
"train_runtime": 224.7245, |
|
"train_samples_per_second": 45.336, |
|
"train_steps_per_second": 2.884 |
|
} |
|
], |
|
"logging_steps": 500, |
|
"max_steps": 648, |
|
"num_input_tokens_seen": 0, |
|
"num_train_epochs": 12, |
|
"save_steps": 500, |
|
"stateful_callbacks": { |
|
"TrainerControl": { |
|
"args": { |
|
"should_epoch_stop": false, |
|
"should_evaluate": false, |
|
"should_log": false, |
|
"should_save": true, |
|
"should_training_stop": true |
|
}, |
|
"attributes": {} |
|
} |
|
}, |
|
"total_flos": 7.894883901934633e+17, |
|
"train_batch_size": 16, |
|
"trial_name": null, |
|
"trial_params": null |
|
} |
|
|