diff options
Diffstat (limited to 'training')
| -rw-r--r-- | training/functional.py | 2 |
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: |