diff options
| author | Volpeon <git@volpeon.ink> | 2023-01-17 12:13:11 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-01-17 12:13:11 +0100 |
| commit | ddc4b6965db625c6d79d5ede60694e66ea18bbd0 (patch) | |
| tree | b02f825f3d47a0b0b77b1c8678e1d1156b1e1ba7 /train_ti.py | |
| parent | Fix (diff) | |
| download | textual-inversion-diff-ddc4b6965db625c6d79d5ede60694e66ea18bbd0.tar.gz textual-inversion-diff-ddc4b6965db625c6d79d5ede60694e66ea18bbd0.tar.bz2 textual-inversion-diff-ddc4b6965db625c6d79d5ede60694e66ea18bbd0.zip | |
Smaller emb decay
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 fc34d27..7aa4960 100644 --- a/train_ti.py +++ b/train_ti.py | |||
| @@ -407,7 +407,7 @@ def parse_args(): | |||
| 407 | ) | 407 | ) |
| 408 | parser.add_argument( | 408 | parser.add_argument( |
| 409 | "--emb_decay", | 409 | "--emb_decay", |
| 410 | default=1e-1, | 410 | default=1e-2, |
| 411 | type=float, | 411 | type=float, |
| 412 | help="Embedding decay factor." | 412 | help="Embedding decay factor." |
| 413 | ) | 413 | ) |