diff options
| author | Melan <alexleander91@gmail.com> | 2022-10-13 10:37:58 +0000 |
|---|---|---|
| committer | Melan <alexleander91@gmail.com> | 2022-10-13 10:37:58 +0000 |
| commit | 8636b50aea83f9c743f005722d9f3f8ee9303e00 (patch) | |
| tree | 01afcd9e7cecdc6bac22af40da01c4ef2d49aa82 /modules/textual_inversion/textual_inversion.py | |
| parent | 1cfc2a18981ee56bdb69a2de7b463a11ad05e329 (diff) | |
| download | stable-diffusion-webui-gfx803-8636b50aea83f9c743f005722d9f3f8ee9303e00.tar.gz stable-diffusion-webui-gfx803-8636b50aea83f9c743f005722d9f3f8ee9303e00.tar.bz2 stable-diffusion-webui-gfx803-8636b50aea83f9c743f005722d9f3f8ee9303e00.zip | |
Add learn_rate to csv and removed a left-over debug statement
Diffstat (limited to 'modules/textual_inversion/textual_inversion.py')
| -rw-r--r-- | modules/textual_inversion/textual_inversion.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/modules/textual_inversion/textual_inversion.py b/modules/textual_inversion/textual_inversion.py index 25038a89..b83df079 100644 --- a/modules/textual_inversion/textual_inversion.py +++ b/modules/textual_inversion/textual_inversion.py @@ -262,14 +262,15 @@ def train_embedding(embedding_name, learn_rate, data_root, log_directory, traini with open(os.path.join(log_directory, "textual_inversion_loss.csv"), "a+") as fout:
- csv_writer = csv.DictWriter(fout, fieldnames=["epoch", "epoch_step", "loss"])
+ csv_writer = csv.DictWriter(fout, fieldnames=["epoch", "epoch_step", "loss", "learn_rate"])
if write_csv_header:
csv_writer.writeheader()
csv_writer.writerow({"epoch": epoch_num + 1,
"epoch_step": epoch_step - 1,
- "loss": f"{losses.mean():.7f}"})
+ "loss": f"{losses.mean():.7f}",
+ "learn_rate": scheduler.learn_rate})
if embedding.step > 0 and images_dir is not None and embedding.step % create_image_every == 0:
last_saved_image = os.path.join(images_dir, f'{embedding_name}-{embedding.step}.png')
|