synergyai-jaeung's picture
Training in progress, epoch 1
aa7fae6 verified
raw
history blame
6.08 kB
{
"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
}