diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-10-21 10:34:48 +0000 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-10-21 10:35:01 +0000 |
| commit | 7d6b388d71e5854c48847c09b2dfed683b377027 (patch) | |
| tree | 35d1ef42a993387f995955ce36dba7b86e081370 /modules/textual_inversion/textual_inversion.py | |
| parent | bf30673f5132c8f28357b31224c54331e788d3e7 (diff) | |
| parent | 2362d5f00eeecbe6a6f67fe6085da01a3d78e407 (diff) | |
| download | stable-diffusion-webui-gfx803-7d6b388d71e5854c48847c09b2dfed683b377027.tar.gz stable-diffusion-webui-gfx803-7d6b388d71e5854c48847c09b2dfed683b377027.tar.bz2 stable-diffusion-webui-gfx803-7d6b388d71e5854c48847c09b2dfed683b377027.zip | |
Merge branch 'ae'
Diffstat (limited to 'modules/textual_inversion/textual_inversion.py')
| -rw-r--r-- | modules/textual_inversion/textual_inversion.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/modules/textual_inversion/textual_inversion.py b/modules/textual_inversion/textual_inversion.py index 5776778b..529ed3e2 100644 --- a/modules/textual_inversion/textual_inversion.py +++ b/modules/textual_inversion/textual_inversion.py @@ -276,6 +276,7 @@ def train_embedding(embedding_name, learn_rate, batch_size, data_root, log_direc loss.backward()
optimizer.step()
+
epoch_num = embedding.step // len(ds)
epoch_step = embedding.step - (epoch_num * len(ds)) + 1
|