{ "best_metric": 0.8333333333333334, "best_model_checkpoint": "swin-base-patch4-window7-224-in22k-cons_Classification_base_V10/checkpoint-2", "epoch": 7.0, "eval_steps": 500, "global_step": 7, "is_hyper_param_search": false, "is_local_process_zero": true, "is_world_process_zero": true, "log_history": [ { "epoch": 1.0, "step": 1, "train_accuracy": 0.26666666666666666, "train_auc_class_0": 0.32000000000000006, "train_auc_class_1": 0.6200000000000001, "train_auc_class_2": 0.39999999999999997, "train_auc_overall": 0.4466666666666667, "train_loss": 1.1479226350784302, "train_runtime": 1.6122, "train_samples_per_second": 9.304, "train_steps_per_second": 0.62 }, { "epoch": 1.0, "eval_accuracy": 0.5, "eval_auc_class_0": 0.875, "eval_auc_class_1": 0.5, "eval_auc_class_2": 0.625, "eval_auc_overall": 0.6666666666666666, "eval_loss": 0.9469685554504395, "eval_runtime": 0.6769, "eval_samples_per_second": 8.863, "eval_steps_per_second": 1.477, "step": 1 }, { "epoch": 2.0, "step": 2, "train_accuracy": 0.6, "train_auc_class_0": 0.82, "train_auc_class_1": 0.96, "train_auc_class_2": 0.9, "train_auc_overall": 0.8933333333333332, "train_loss": 0.814429521560669, "train_runtime": 1.6517, "train_samples_per_second": 9.081, "train_steps_per_second": 0.605 }, { "epoch": 2.0, "eval_accuracy": 0.8333333333333334, "eval_auc_class_0": 1.0, "eval_auc_class_1": 1.0, "eval_auc_class_2": 1.0, "eval_auc_overall": 1.0, "eval_loss": 0.8139967918395996, "eval_runtime": 0.7014, "eval_samples_per_second": 8.555, "eval_steps_per_second": 1.426, "step": 2 }, { "epoch": 3.0, "step": 3, "train_accuracy": 0.8666666666666667, "train_auc_class_0": 1.0, "train_auc_class_1": 1.0, "train_auc_class_2": 0.9600000000000001, "train_auc_overall": 0.9866666666666667, "train_loss": 0.6608889698982239, "train_runtime": 1.7063, "train_samples_per_second": 8.791, "train_steps_per_second": 0.586 }, { "epoch": 3.0, "eval_accuracy": 0.8333333333333334, "eval_auc_class_0": 1.0, "eval_auc_class_1": 1.0, "eval_auc_class_2": 1.0, "eval_auc_overall": 1.0, "eval_loss": 0.7299937605857849, "eval_runtime": 0.7538, "eval_samples_per_second": 7.959, "eval_steps_per_second": 1.327, "step": 3 }, { "epoch": 4.0, "step": 4, "train_accuracy": 1.0, "train_auc_class_0": 1.0, "train_auc_class_1": 1.0, "train_auc_class_2": 1.0, "train_auc_overall": 1.0, "train_loss": 0.42420145869255066, "train_runtime": 1.6223, "train_samples_per_second": 9.246, "train_steps_per_second": 0.616 }, { "epoch": 4.0, "eval_accuracy": 0.8333333333333334, "eval_auc_class_0": 1.0, "eval_auc_class_1": 1.0, "eval_auc_class_2": 1.0, "eval_auc_overall": 1.0, "eval_loss": 0.6729159355163574, "eval_runtime": 0.772, "eval_samples_per_second": 7.772, "eval_steps_per_second": 1.295, "step": 4 }, { "epoch": 5.0, "step": 5, "train_accuracy": 0.9333333333333333, "train_auc_class_0": 1.0, "train_auc_class_1": 1.0, "train_auc_class_2": 1.0, "train_auc_overall": 1.0, "train_loss": 0.42396074533462524, "train_runtime": 1.6518, "train_samples_per_second": 9.081, "train_steps_per_second": 0.605 }, { "epoch": 5.0, "eval_accuracy": 0.8333333333333334, "eval_auc_class_0": 1.0, "eval_auc_class_1": 1.0, "eval_auc_class_2": 1.0, "eval_auc_overall": 1.0, "eval_loss": 0.6391382813453674, "eval_runtime": 0.6917, "eval_samples_per_second": 8.675, "eval_steps_per_second": 1.446, "step": 5 }, { "epoch": 6.0, "step": 6, "train_accuracy": 1.0, "train_auc_class_0": 1.0, "train_auc_class_1": 1.0, "train_auc_class_2": 1.0, "train_auc_overall": 1.0, "train_loss": 0.29362058639526367, "train_runtime": 1.6474, "train_samples_per_second": 9.105, "train_steps_per_second": 0.607 }, { "epoch": 6.0, "eval_accuracy": 0.8333333333333334, "eval_auc_class_0": 1.0, "eval_auc_class_1": 1.0, "eval_auc_class_2": 1.0, "eval_auc_overall": 1.0, "eval_loss": 0.6181725859642029, "eval_runtime": 0.6858, "eval_samples_per_second": 8.748, "eval_steps_per_second": 1.458, "step": 6 }, { "epoch": 7.0, "step": 7, "train_accuracy": 1.0, "train_auc_class_0": 1.0, "train_auc_class_1": 1.0, "train_auc_class_2": 1.0, "train_auc_overall": 1.0, "train_loss": 0.35040372610092163, "train_runtime": 1.7724, "train_samples_per_second": 8.463, "train_steps_per_second": 0.564 }, { "epoch": 7.0, "eval_accuracy": 0.8333333333333334, "eval_auc_class_0": 1.0, "eval_auc_class_1": 1.0, "eval_auc_class_2": 1.0, "eval_auc_overall": 1.0, "eval_loss": 0.6054350733757019, "eval_runtime": 0.7011, "eval_samples_per_second": 8.558, "eval_steps_per_second": 1.426, "step": 7 }, { "epoch": 7.0, "step": 7, "total_flos": 8226380544399360.0, "train_loss": 0.17954584530421666, "train_runtime": 78.6151, "train_samples_per_second": 1.336, "train_steps_per_second": 0.089 } ], "logging_steps": 10, "max_steps": 7, "num_input_tokens_seen": 0, "num_train_epochs": 7, "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": 8226380544399360.0, "train_batch_size": 128, "trial_name": null, "trial_params": null }