diff options
| author | Volpeon <git@volpeon.ink> | 2023-01-03 13:32:37 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-01-03 13:32:37 +0100 |
| commit | c55920681ccd15c183f168619d9b332cd837163c (patch) | |
| tree | 9b86ba07b89a282153276577174820bcdb3f0387 /train_ti.py | |
| parent | Added vector dropout (diff) | |
| download | textual-inversion-diff-c55920681ccd15c183f168619d9b332cd837163c.tar.gz textual-inversion-diff-c55920681ccd15c183f168619d9b332cd837163c.tar.bz2 textual-inversion-diff-c55920681ccd15c183f168619d9b332cd837163c.zip | |
Don't use vector_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 cacbbc7..1685dc4 100644 --- a/train_ti.py +++ b/train_ti.py | |||
| @@ -157,7 +157,7 @@ def parse_args(): | |||
| 157 | parser.add_argument( | 157 | parser.add_argument( |
| 158 | "--vector_dropout", | 158 | "--vector_dropout", |
| 159 | type=int, | 159 | type=int, |
| 160 | default=0.1, | 160 | default=0, |
| 161 | help="Vector dropout probability.", | 161 | help="Vector dropout probability.", |
| 162 | ) | 162 | ) |
| 163 | parser.add_argument( | 163 | parser.add_argument( |