diff options
| author | Volpeon <git@volpeon.ink> | 2023-04-03 14:17:49 +0200 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-04-03 14:17:49 +0200 |
| commit | ce3407636232a86be84dd7472bbfd0c0ac5ed81f (patch) | |
| tree | f331915568e4ca62427f25f6b49b0dd123b24678 /train_ti.py | |
| parent | Bring back Lion optimizer (diff) | |
| download | textual-inversion-diff-ce3407636232a86be84dd7472bbfd0c0ac5ed81f.tar.gz textual-inversion-diff-ce3407636232a86be84dd7472bbfd0c0ac5ed81f.tar.bz2 textual-inversion-diff-ce3407636232a86be84dd7472bbfd0c0ac5ed81f.zip | |
TI: No tag dropout by default
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 c242625..8dde1ba 100644 --- a/train_ti.py +++ b/train_ti.py | |||
| @@ -186,7 +186,7 @@ def parse_args(): | |||
| 186 | parser.add_argument( | 186 | parser.add_argument( |
| 187 | "--tag_dropout", | 187 | "--tag_dropout", |
| 188 | type=float, | 188 | type=float, |
| 189 | default=0.1, | 189 | default=0, |
| 190 | help="Tag dropout probability.", | 190 | help="Tag dropout probability.", |
| 191 | ) | 191 | ) |
| 192 | parser.add_argument( | 192 | parser.add_argument( |