diff options
| author | Volpeon <git@volpeon.ink> | 2023-01-13 09:17:44 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-01-13 09:17:44 +0100 |
| commit | 07e8111a98afbf912ba99a28e2d7ea647b2d29fc (patch) | |
| tree | 853aaae6f5ff477b73d5649f3f89303ff8654b70 /train_dreambooth.py | |
| parent | Code deduplication (diff) | |
| download | textual-inversion-diff-07e8111a98afbf912ba99a28e2d7ea647b2d29fc.tar.gz textual-inversion-diff-07e8111a98afbf912ba99a28e2d7ea647b2d29fc.tar.bz2 textual-inversion-diff-07e8111a98afbf912ba99a28e2d7ea647b2d29fc.zip | |
Added TI decay start offset
Diffstat (limited to 'train_dreambooth.py')
| -rw-r--r-- | train_dreambooth.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/train_dreambooth.py b/train_dreambooth.py index da3a075..0fe590f 100644 --- a/train_dreambooth.py +++ b/train_dreambooth.py | |||
| @@ -907,7 +907,7 @@ def main(): | |||
| 907 | on_before_optimize=on_before_optimize, | 907 | on_before_optimize=on_before_optimize, |
| 908 | on_after_optimize=on_after_optimize, | 908 | on_after_optimize=on_after_optimize, |
| 909 | ) | 909 | ) |
| 910 | lr_finder.run(num_epochs=100, end_lr=1e3) | 910 | lr_finder.run(num_epochs=100, end_lr=1e2) |
| 911 | 911 | ||
| 912 | plt.savefig(basepath.joinpath("lr.png"), dpi=300) | 912 | plt.savefig(basepath.joinpath("lr.png"), dpi=300) |
| 913 | plt.close() | 913 | plt.close() |