{ "best_metric": 0.33828961849212646, "best_model_checkpoint": "google/vit-base-patch16-224-in21k_covid_19_ct_scans/checkpoint-55", "epoch": 15.0, "eval_steps": 500, "global_step": 825, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.01818181818181818, "grad_norm": 3.793684244155884, "learning_rate": 0.00019975757575757576, "loss": 0.7218, "step": 1 }, { "epoch": 1.0, "eval_AUC": 0.5, "eval_F1": 0.47540983606557374, "eval_Precision": 0.90625, "eval_Recall": 1.0, "eval_accuracy": 0.90625, "eval_loss": 0.33828961849212646, "eval_runtime": 4.7492, "eval_samples_per_second": 80.855, "eval_steps_per_second": 5.053, "step": 55 }, { "epoch": 2.0, "eval_AUC": 0.5, "eval_F1": 0.47540983606557374, "eval_Precision": 0.90625, "eval_Recall": 1.0, "eval_accuracy": 0.90625, "eval_loss": 0.382261723279953, "eval_runtime": 4.831, "eval_samples_per_second": 79.486, "eval_steps_per_second": 4.968, "step": 110 }, { "epoch": 3.0, "eval_AUC": 0.5, "eval_F1": 0.47540983606557374, "eval_Precision": 0.90625, "eval_Recall": 1.0, "eval_accuracy": 0.90625, "eval_loss": 0.3956604301929474, "eval_runtime": 4.8897, "eval_samples_per_second": 78.532, "eval_steps_per_second": 4.908, "step": 165 }, { "epoch": 4.0, "eval_AUC": 0.5, "eval_F1": 0.47540983606557374, "eval_Precision": 0.90625, "eval_Recall": 1.0, "eval_accuracy": 0.90625, "eval_loss": 0.44851580262184143, "eval_runtime": 4.8632, "eval_samples_per_second": 78.96, "eval_steps_per_second": 4.935, "step": 220 }, { "epoch": 5.0, "eval_AUC": 0.4942528735632184, "eval_F1": 0.4725274725274725, "eval_Precision": 0.9052631578947369, "eval_Recall": 0.9885057471264368, "eval_accuracy": 0.8958333333333334, "eval_loss": 0.4786018133163452, "eval_runtime": 4.8854, "eval_samples_per_second": 78.601, "eval_steps_per_second": 4.913, "step": 275 }, { "epoch": 6.0, "eval_AUC": 0.49712643678160917, "eval_F1": 0.473972602739726, "eval_Precision": 0.9057591623036649, "eval_Recall": 0.9942528735632183, "eval_accuracy": 0.9010416666666666, "eval_loss": 0.5316488146781921, "eval_runtime": 5.3191, "eval_samples_per_second": 72.193, "eval_steps_per_second": 4.512, "step": 330 }, { "epoch": 7.0, "eval_AUC": 0.49712643678160917, "eval_F1": 0.473972602739726, "eval_Precision": 0.9057591623036649, "eval_Recall": 0.9942528735632183, "eval_accuracy": 0.9010416666666666, "eval_loss": 0.553883969783783, "eval_runtime": 4.8592, "eval_samples_per_second": 79.026, "eval_steps_per_second": 4.939, "step": 385 }, { "epoch": 8.0, "eval_AUC": 0.49712643678160917, "eval_F1": 0.473972602739726, "eval_Precision": 0.9057591623036649, "eval_Recall": 0.9942528735632183, "eval_accuracy": 0.9010416666666666, "eval_loss": 0.5799620747566223, "eval_runtime": 4.7649, "eval_samples_per_second": 80.589, "eval_steps_per_second": 5.037, "step": 440 }, { "epoch": 9.0, "eval_AUC": 0.49712643678160917, "eval_F1": 0.473972602739726, "eval_Precision": 0.9057591623036649, "eval_Recall": 0.9942528735632183, "eval_accuracy": 0.9010416666666666, "eval_loss": 0.5976574420928955, "eval_runtime": 4.7034, "eval_samples_per_second": 81.644, "eval_steps_per_second": 5.103, "step": 495 }, { "epoch": 9.090909090909092, "grad_norm": 0.018258079886436462, "learning_rate": 7.878787878787879e-05, "loss": 0.0987, "step": 500 }, { "epoch": 10.0, "eval_AUC": 0.49712643678160917, "eval_F1": 0.473972602739726, "eval_Precision": 0.9057591623036649, "eval_Recall": 0.9942528735632183, "eval_accuracy": 0.9010416666666666, "eval_loss": 0.6109693646430969, "eval_runtime": 4.7422, "eval_samples_per_second": 80.974, "eval_steps_per_second": 5.061, "step": 550 }, { "epoch": 11.0, "eval_AUC": 0.49712643678160917, "eval_F1": 0.473972602739726, "eval_Precision": 0.9057591623036649, "eval_Recall": 0.9942528735632183, "eval_accuracy": 0.9010416666666666, "eval_loss": 0.6211499571800232, "eval_runtime": 4.7723, "eval_samples_per_second": 80.464, "eval_steps_per_second": 5.029, "step": 605 }, { "epoch": 12.0, "eval_AUC": 0.49712643678160917, "eval_F1": 0.473972602739726, "eval_Precision": 0.9057591623036649, "eval_Recall": 0.9942528735632183, "eval_accuracy": 0.9010416666666666, "eval_loss": 0.6287771463394165, "eval_runtime": 4.9768, "eval_samples_per_second": 77.159, "eval_steps_per_second": 4.822, "step": 660 }, { "epoch": 13.0, "eval_AUC": 0.49712643678160917, "eval_F1": 0.473972602739726, "eval_Precision": 0.9057591623036649, "eval_Recall": 0.9942528735632183, "eval_accuracy": 0.9010416666666666, "eval_loss": 0.6341301798820496, "eval_runtime": 4.7422, "eval_samples_per_second": 80.974, "eval_steps_per_second": 5.061, "step": 715 }, { "epoch": 14.0, "eval_AUC": 0.49712643678160917, "eval_F1": 0.473972602739726, "eval_Precision": 0.9057591623036649, "eval_Recall": 0.9942528735632183, "eval_accuracy": 0.9010416666666666, "eval_loss": 0.6373991370201111, "eval_runtime": 4.8768, "eval_samples_per_second": 78.74, "eval_steps_per_second": 4.921, "step": 770 }, { "epoch": 15.0, "eval_AUC": 0.49712643678160917, "eval_F1": 0.473972602739726, "eval_Precision": 0.9057591623036649, "eval_Recall": 0.9942528735632183, "eval_accuracy": 0.9010416666666666, "eval_loss": 0.6385161280632019, "eval_runtime": 4.6939, "eval_samples_per_second": 81.808, "eval_steps_per_second": 5.113, "step": 825 }, { "epoch": 15.0, "step": 825, "total_flos": 1.0112704644704256e+18, "train_loss": 0.06119063453240828, "train_runtime": 257.3401, "train_samples_per_second": 50.711, "train_steps_per_second": 3.206 } ], "logging_steps": 500, "max_steps": 825, "num_input_tokens_seen": 0, "num_train_epochs": 15, "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": 1.0112704644704256e+18, "train_batch_size": 16, "trial_name": null, "trial_params": null }