s444415/donut
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

push to hub callback edit

Browse Source
This commit is contained in:
s444415 2022-12-14 20:43:43 +01:00
parent 322a495d9c
commit 67cc4bdf7c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
donut-train.py
View File

@ -328,7 +328,7 @@ class PushToHubCallback(Callback):
print(f"Pushing model to the hub, epoch {trainer.current_epoch}") print(f"Pushing model to the hub, epoch {trainer.current_epoch}")
pl_module.model.push_to_hub(OUTPUT_MODEL_PATH, pl_module.model.push_to_hub(OUTPUT_MODEL_PATH,
commit_message=f"Training in progress, epoch {trainer.current_epoch}") 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): def on_train_end(self, trainer, pl_module):
print(f"Pushing model to the hub after training") print(f"Pushing model to the hub after training")