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