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