diff options
| author | Volpeon <git@volpeon.ink> | 2023-04-16 19:17:27 +0200 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-04-16 19:17:27 +0200 |
| commit | 6c47c433f21d62b274d7aff82be767847b8090b1 (patch) | |
| tree | a93d327a929f971e537b8dd96921e03346f22171 /training | |
| parent | Improved automation caps (diff) | |
| download | textual-inversion-diff-6c47c433f21d62b274d7aff82be767847b8090b1.tar.gz textual-inversion-diff-6c47c433f21d62b274d7aff82be767847b8090b1.tar.bz2 textual-inversion-diff-6c47c433f21d62b274d7aff82be767847b8090b1.zip | |
Fix
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 ebc40de..c6ceb20 100644 --- a/training/functional.py +++ b/training/functional.py | |||
| @@ -519,7 +519,7 @@ def train_loop( | |||
| 519 | try: | 519 | try: |
| 520 | for epoch in range(num_epochs): | 520 | for epoch in range(num_epochs): |
| 521 | if accelerator.is_main_process: | 521 | if accelerator.is_main_process: |
| 522 | if epoch % sample_frequency == 0 and (cycle == 1 or epoch != 0): | 522 | if epoch % sample_frequency == 0 and (cycle == 0 or epoch != 0): |
| 523 | local_progress_bar.clear() | 523 | local_progress_bar.clear() |
| 524 | global_progress_bar.clear() | 524 | global_progress_bar.clear() |
| 525 | 525 | ||