synergyai-jaeung's picture
Training in progress, epoch 1
c1ebc49 verified
raw
history blame
No virus
6.18 kB
{
"best_metric": 0.3346473276615143,
"best_model_checkpoint": "google/vit-base-patch16-224-in21k_covid_19_ct_scans/checkpoint-47",
"epoch": 12.0,
"eval_steps": 500,
"global_step": 564,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"epoch": 0.02127659574468085,
"grad_norm": 4.243431091308594,
"learning_rate": 0.000199645390070922,
"loss": 0.7767,
"step": 1
},
{
"epoch": 1.0,
"eval_AUC": 0.5,
"eval_F1": 0.47572815533980584,
"eval_Precision": 0.9074074074074074,
"eval_Recall": 1.0,
"eval_accuracy": 0.9074074074074074,
"eval_loss": 0.3346473276615143,
"eval_runtime": 5.5299,
"eval_samples_per_second": 58.591,
"eval_steps_per_second": 3.798,
"step": 47
},
{
"epoch": 2.0,
"eval_AUC": 0.6204081632653061,
"eval_F1": 0.5919028340080972,
"eval_Precision": 0.9311594202898551,
"eval_Recall": 0.8741496598639455,
"eval_accuracy": 0.8271604938271605,
"eval_loss": 0.5513330698013306,
"eval_runtime": 5.5902,
"eval_samples_per_second": 57.959,
"eval_steps_per_second": 3.757,
"step": 94
},
{
"epoch": 3.0,
"eval_AUC": 0.5,
"eval_F1": 0.47572815533980584,
"eval_Precision": 0.9074074074074074,
"eval_Recall": 1.0,
"eval_accuracy": 0.9074074074074074,
"eval_loss": 0.4290040135383606,
"eval_runtime": 5.491,
"eval_samples_per_second": 59.005,
"eval_steps_per_second": 3.824,
"step": 141
},
{
"epoch": 4.0,
"eval_AUC": 0.5965986394557823,
"eval_F1": 0.6365832614322692,
"eval_Precision": 0.9240506329113924,
"eval_Recall": 0.9931972789115646,
"eval_accuracy": 0.9197530864197531,
"eval_loss": 0.433300256729126,
"eval_runtime": 5.3718,
"eval_samples_per_second": 60.315,
"eval_steps_per_second": 3.909,
"step": 188
},
{
"epoch": 5.0,
"eval_AUC": 0.5897959183673469,
"eval_F1": 0.618104667609618,
"eval_Precision": 0.9230769230769231,
"eval_Recall": 0.9795918367346939,
"eval_accuracy": 0.9074074074074074,
"eval_loss": 0.5040868520736694,
"eval_runtime": 5.759,
"eval_samples_per_second": 56.26,
"eval_steps_per_second": 3.646,
"step": 235
},
{
"epoch": 6.0,
"eval_AUC": 0.5948979591836735,
"eval_F1": 0.6316786661614249,
"eval_Precision": 0.9238095238095239,
"eval_Recall": 0.9897959183673469,
"eval_accuracy": 0.9166666666666666,
"eval_loss": 0.4848092496395111,
"eval_runtime": 13.7532,
"eval_samples_per_second": 23.558,
"eval_steps_per_second": 1.527,
"step": 282
},
{
"epoch": 7.0,
"eval_AUC": 0.5965986394557823,
"eval_F1": 0.6365832614322692,
"eval_Precision": 0.9240506329113924,
"eval_Recall": 0.9931972789115646,
"eval_accuracy": 0.9197530864197531,
"eval_loss": 0.48772257566452026,
"eval_runtime": 5.5683,
"eval_samples_per_second": 58.186,
"eval_steps_per_second": 3.771,
"step": 329
},
{
"epoch": 8.0,
"eval_AUC": 0.5965986394557823,
"eval_F1": 0.6365832614322692,
"eval_Precision": 0.9240506329113924,
"eval_Recall": 0.9931972789115646,
"eval_accuracy": 0.9197530864197531,
"eval_loss": 0.49257075786590576,
"eval_runtime": 5.2567,
"eval_samples_per_second": 61.635,
"eval_steps_per_second": 3.995,
"step": 376
},
{
"epoch": 9.0,
"eval_AUC": 0.5965986394557823,
"eval_F1": 0.6365832614322692,
"eval_Precision": 0.9240506329113924,
"eval_Recall": 0.9931972789115646,
"eval_accuracy": 0.9197530864197531,
"eval_loss": 0.49581947922706604,
"eval_runtime": 5.6218,
"eval_samples_per_second": 57.633,
"eval_steps_per_second": 3.735,
"step": 423
},
{
"epoch": 10.0,
"eval_AUC": 0.5965986394557823,
"eval_F1": 0.6365832614322692,
"eval_Precision": 0.9240506329113924,
"eval_Recall": 0.9931972789115646,
"eval_accuracy": 0.9197530864197531,
"eval_loss": 0.4980551600456238,
"eval_runtime": 5.5112,
"eval_samples_per_second": 58.79,
"eval_steps_per_second": 3.81,
"step": 470
},
{
"epoch": 10.638297872340425,
"grad_norm": 0.012480547651648521,
"learning_rate": 2.269503546099291e-05,
"loss": 0.0381,
"step": 500
},
{
"epoch": 11.0,
"eval_AUC": 0.5965986394557823,
"eval_F1": 0.6365832614322692,
"eval_Precision": 0.9240506329113924,
"eval_Recall": 0.9931972789115646,
"eval_accuracy": 0.9197530864197531,
"eval_loss": 0.4996081292629242,
"eval_runtime": 5.5015,
"eval_samples_per_second": 58.893,
"eval_steps_per_second": 3.817,
"step": 517
},
{
"epoch": 12.0,
"eval_AUC": 0.5965986394557823,
"eval_F1": 0.6365832614322692,
"eval_Precision": 0.9240506329113924,
"eval_Recall": 0.9931972789115646,
"eval_accuracy": 0.9197530864197531,
"eval_loss": 0.5001835823059082,
"eval_runtime": 5.5423,
"eval_samples_per_second": 58.46,
"eval_steps_per_second": 3.789,
"step": 564
},
{
"epoch": 12.0,
"step": 564,
"total_flos": 6.862690600268268e+17,
"train_loss": 0.03526242822408676,
"train_runtime": 251.233,
"train_samples_per_second": 35.25,
"train_steps_per_second": 2.245
}
],
"logging_steps": 500,
"max_steps": 564,
"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": 6.862690600268268e+17,
"train_batch_size": 16,
"trial_name": null,
"trial_params": null
}