From 67cc4bdf7ca8c863dee1a4e949942eb71cf2223f Mon Sep 17 00:00:00 2001 From: s444415 Date: Wed, 14 Dec 2022 20:43:43 +0100 Subject: [PATCH] push to hub callback edit --- donut-train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/donut-train.py b/donut-train.py index ff4c3de..8eaa167 100644 --- a/donut-train.py +++ b/donut-train.py @@ -328,7 +328,7 @@ class PushToHubCallback(Callback): print(f"Pushing model to the hub, epoch {trainer.current_epoch}") pl_module.model.push_to_hub(OUTPUT_MODEL_PATH, commit_message=f"Training in progress, epoch {trainer.current_epoch}") - pl_module.processor.push_to_hub(OUTPUT_MODEL_PATH, commit_message=f"Training in progress, epoch {trainer.current_epoch}") + # pl_module.processor.push_to_hub(OUTPUT_MODEL_PATH, commit_message=f"Training in progress, epoch {trainer.current_epoch}") def on_train_end(self, trainer, pl_module): print(f"Pushing model to the hub after training")