summaryrefslogtreecommitdiffstats
path: root/training
diff options
context:
space:
mode:
Diffstat (limited to 'training')
-rw-r--r--training/functional.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/training/functional.py b/training/functional.py
index 71b2fe9..7d49782 100644
--- a/training/functional.py
+++ b/training/functional.py
@@ -606,8 +606,8 @@ def train_loop(
606 # Create the pipeline using using the trained modules and save it. 606 # Create the pipeline using using the trained modules and save it.
607 if accelerator.is_main_process: 607 if accelerator.is_main_process:
608 print("Finished!") 608 print("Finished!")
609 on_checkpoint(global_step + global_step_offset, "end")
610 on_sample(global_step + global_step_offset) 609 on_sample(global_step + global_step_offset)
610 on_checkpoint(global_step + global_step_offset, "end")
611 611
612 except KeyboardInterrupt: 612 except KeyboardInterrupt:
613 if accelerator.is_main_process: 613 if accelerator.is_main_process: