diff options
Diffstat (limited to 'training')
| -rw-r--r-- | training/functional.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/training/functional.py b/training/functional.py index f5c111e..1b6162b 100644 --- a/training/functional.py +++ b/training/functional.py | |||
| @@ -486,6 +486,7 @@ def train_loop( | |||
| 486 | if accelerator.is_main_process: | 486 | if accelerator.is_main_process: |
| 487 | print("Interrupted") | 487 | print("Interrupted") |
| 488 | on_checkpoint(global_step + global_step_offset, "end") | 488 | on_checkpoint(global_step + global_step_offset, "end") |
| 489 | raise KeyboardInterrupt | ||
| 489 | 490 | ||
| 490 | 491 | ||
| 491 | def train( | 492 | def train( |