diff options
author | Volpeon <git@volpeon.ink> | 2023-01-06 18:10:19 +0100 |
---|---|---|
committer | Volpeon <git@volpeon.ink> | 2023-01-06 18:10:19 +0100 |
commit | 2c3dcbb1bc3982b507c3b30b5deed5aedd2160b2 (patch) | |
tree | c32f2f4ded306ca46ec431908400416c9569e1c5 | |
parent | Update (diff) | |
download | textual-inversion-diff-2c3dcbb1bc3982b507c3b30b5deed5aedd2160b2.tar.gz textual-inversion-diff-2c3dcbb1bc3982b507c3b30b5deed5aedd2160b2.tar.bz2 textual-inversion-diff-2c3dcbb1bc3982b507c3b30b5deed5aedd2160b2.zip |
Relaxed ema_power for TI
-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 9aab00c..1b8c597 100644 --- a/train_ti.py +++ b/train_ti.py | |||
@@ -280,7 +280,7 @@ def parse_args(): | |||
280 | parser.add_argument( | 280 | parser.add_argument( |
281 | "--ema_power", | 281 | "--ema_power", |
282 | type=float, | 282 | type=float, |
283 | default=7/8 | 283 | default=4/5 |
284 | ) | 284 | ) |
285 | parser.add_argument( | 285 | parser.add_argument( |
286 | "--ema_max_decay", | 286 | "--ema_max_decay", |