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