openvinotoolkit/stable-diffusion-webui
1
1
Fork 0
mirror of https://github.com/openvinotoolkit/stable-diffusion-webui.git synced 2024-12-15 15:13:45 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fixed progress bar output for epoch

Browse Source
This commit is contained in:
alg-wiki 2022-10-10 17:31:33 +09:00
parent 3110f895b2
commit 4ee7519fc2
No known key found for this signature in database
GPG Key ID: 9D27E57D1F43E750
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/textual_inversion/textual_inversion.py
View File

@ -228,7 +228,7 @@ def train_embedding(embedding_name, learn_rate, data_root, log_directory, traini
optimizer.step() optimizer.step()
epoch_num = math.floor(embedding.step / epoch_len) 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}") pbar.set_description(f"[Epoch {epoch_num}: {epoch_step}/{epoch_len}]loss: {losses.mean():.7f}")