{ "best_metric": 0.7681704260651628, "best_model_checkpoint": "swin-tiny-patch4-window7-224-finetunedo/checkpoint-12", "epoch": 8.0, "eval_steps": 500, "global_step": 24, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 1.0, "eval_loss": 0.5081062316894531, "eval_roc_auc": 0.736842105263158, "eval_runtime": 0.5379, "eval_samples_per_second": 74.363, "eval_steps_per_second": 3.718, "step": 3 }, { "epoch": 2.0, "eval_loss": 0.43417197465896606, "eval_roc_auc": 0.7631578947368421, "eval_runtime": 0.5781, "eval_samples_per_second": 69.192, "eval_steps_per_second": 3.46, "step": 6 }, { "epoch": 3.0, "eval_loss": 0.43640345335006714, "eval_roc_auc": 0.7493734335839599, "eval_runtime": 0.3468, "eval_samples_per_second": 115.327, "eval_steps_per_second": 5.766, "step": 9 }, { "epoch": 3.33, "learning_rate": 3.3333333333333335e-05, "loss": 0.4067, "step": 10 }, { "epoch": 4.0, "eval_loss": 0.446413516998291, "eval_roc_auc": 0.7681704260651628, "eval_runtime": 0.2866, "eval_samples_per_second": 139.576, "eval_steps_per_second": 6.979, "step": 12 }, { "epoch": 5.0, "eval_loss": 0.46299201250076294, "eval_roc_auc": 0.7656641604010025, "eval_runtime": 0.2863, "eval_samples_per_second": 139.701, "eval_steps_per_second": 6.985, "step": 15 }, { "epoch": 6.0, "eval_loss": 0.45169758796691895, "eval_roc_auc": 0.744360902255639, "eval_runtime": 0.3942, "eval_samples_per_second": 101.474, "eval_steps_per_second": 5.074, "step": 18 }, { "epoch": 6.67, "learning_rate": 9.523809523809523e-06, "loss": 0.4211, "step": 20 }, { "epoch": 7.0, "eval_loss": 0.4539761543273926, "eval_roc_auc": 0.7180451127819549, "eval_runtime": 0.2978, "eval_samples_per_second": 134.324, "eval_steps_per_second": 6.716, "step": 21 }, { "epoch": 8.0, "eval_loss": 0.45407772064208984, "eval_roc_auc": 0.7180451127819549, "eval_runtime": 0.2973, "eval_samples_per_second": 134.55, "eval_steps_per_second": 6.728, "step": 24 }, { "epoch": 8.0, "step": 24, "total_flos": 7.078985663171789e+16, "train_loss": 0.4116581827402115, "train_runtime": 69.7528, "train_samples_per_second": 40.83, "train_steps_per_second": 0.344 } ], "logging_steps": 10, "max_steps": 24, "num_train_epochs": 8, "save_steps": 500, "total_flos": 7.078985663171789e+16, "trial_name": null, "trial_params": null }