diff options
| author | Volpeon <git@volpeon.ink> | 2023-01-04 22:06:05 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-01-04 22:06:05 +0100 |
| commit | a5e45e2c0dab95589e5fbaa4fe87d18484fbbe68 (patch) | |
| tree | 8bd97a745e1113b1035c504ec484e099f878aed0 /train_ti.py | |
| parent | Various updates (diff) | |
| download | textual-inversion-diff-a5e45e2c0dab95589e5fbaa4fe87d18484fbbe68.tar.gz textual-inversion-diff-a5e45e2c0dab95589e5fbaa4fe87d18484fbbe68.tar.bz2 textual-inversion-diff-a5e45e2c0dab95589e5fbaa4fe87d18484fbbe68.zip | |
Update
Diffstat (limited to 'train_ti.py')
| -rw-r--r-- | train_ti.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/train_ti.py b/train_ti.py index 1b60f64..8ada98c 100644 --- a/train_ti.py +++ b/train_ti.py | |||
| @@ -889,7 +889,7 @@ def main(): | |||
| 889 | on_train=on_train, | 889 | on_train=on_train, |
| 890 | on_eval=on_eval, | 890 | on_eval=on_eval, |
| 891 | ) | 891 | ) |
| 892 | lr_finder.run(end_lr=1e3) | 892 | lr_finder.run(num_epochs=200, end_lr=1e3) |
| 893 | 893 | ||
| 894 | plt.savefig(basepath.joinpath("lr.png"), dpi=300) | 894 | plt.savefig(basepath.joinpath("lr.png"), dpi=300) |
| 895 | plt.close() | 895 | plt.close() |