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