{ "best_metric": 0.2538217604160309, "best_model_checkpoint": "google/vit-base-patch16-224-in21k_GI_diagnosis/checkpoint-400", "epoch": 3.0, "global_step": 600, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 0.01, "learning_rate": 0.00019966666666666668, "loss": 1.3805, "step": 1 }, { "epoch": 1.0, "eval_Macro F1": 0.853134632463581, "eval_Macro Precision": 0.911093881243135, "eval_Macro Recall": 0.86375, "eval_Micro F1": 0.86375, "eval_Micro Precision": 0.86375, "eval_Micro Recall": 0.86375, "eval_Weighted F1": 0.8531346324635811, "eval_Weighted Precision": 0.911093881243135, "eval_Weighted Recall": 0.86375, "eval_accuracy": 0.86375, "eval_loss": 0.5006434917449951, "eval_runtime": 724.2269, "eval_samples_per_second": 1.105, "eval_steps_per_second": 0.138, "step": 200 }, { "epoch": 2.0, "eval_Macro F1": 0.9364724676515609, "eval_Macro Precision": 0.9454970963639796, "eval_Macro Recall": 0.9375, "eval_Micro F1": 0.9375, "eval_Micro Precision": 0.9375, "eval_Micro Recall": 0.9375, "eval_Weighted F1": 0.936472467651561, "eval_Weighted Precision": 0.9454970963639796, "eval_Weighted Recall": 0.9375, "eval_accuracy": 0.9375, "eval_loss": 0.2538217604160309, "eval_runtime": 724.1098, "eval_samples_per_second": 1.105, "eval_steps_per_second": 0.138, "step": 400 }, { "epoch": 2.5, "learning_rate": 3.3333333333333335e-05, "loss": 0.0628, "step": 500 }, { "epoch": 3.0, "eval_Macro F1": 0.8740311301580649, "eval_Macro Precision": 0.9157051711944594, "eval_Macro Recall": 0.8812500000000001, "eval_Micro F1": 0.88125, "eval_Micro Precision": 0.88125, "eval_Micro Recall": 0.88125, "eval_Weighted F1": 0.8740311301580648, "eval_Weighted Precision": 0.9157051711944594, "eval_Weighted Recall": 0.88125, "eval_accuracy": 0.88125, "eval_loss": 0.5797314643859863, "eval_runtime": 724.2187, "eval_samples_per_second": 1.105, "eval_steps_per_second": 0.138, "step": 600 }, { "epoch": 3.0, "step": 600, "total_flos": 7.439364353949696e+17, "train_loss": 0.05536448404192924, "train_runtime": 27748.8376, "train_samples_per_second": 0.346, "train_steps_per_second": 0.022 } ], "max_steps": 600, "num_train_epochs": 3, "total_flos": 7.439364353949696e+17, "trial_name": null, "trial_params": null }