diff options
| author | alg-wiki <alg.4chan@gmail.com> | 2022-10-10 17:31:33 +0900 |
|---|---|---|
| committer | alg-wiki <alg.4chan@gmail.com> | 2022-10-10 17:31:33 +0900 |
| commit | 4ee7519fc2e459ce8eff1f61f1655afba393357c (patch) | |
| tree | 7d688a03a47563894b6f2bf74e4f453c1a08a156 | |
| parent | 3110f895b2718a3a25aae419fdf5c87c177ec9f4 (diff) | |
Fixed progress bar output for epoch
| -rw-r--r-- | modules/textual_inversion/textual_inversion.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/textual_inversion/textual_inversion.py b/modules/textual_inversion/textual_inversion.py index e34dc2e8..769682ea 100644 --- a/modules/textual_inversion/textual_inversion.py +++ b/modules/textual_inversion/textual_inversion.py @@ -228,7 +228,7 @@ def train_embedding(embedding_name, learn_rate, data_root, log_directory, traini optimizer.step()
epoch_num = math.floor(embedding.step / epoch_len)
- epoch_step = embedding.step - (epoch_num * epoch_len)
+ epoch_step = embedding.step - (epoch_num * epoch_len) + 1
pbar.set_description(f"[Epoch {epoch_num}: {epoch_step}/{epoch_len}]loss: {losses.mean():.7f}")
|