diff options
| author | Volpeon <git@volpeon.ink> | 2023-01-12 07:18:21 +0100 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-01-12 07:18:21 +0100 |
| commit | 4370be7d67f4cec00c14c8a8b3e999d50b75585d (patch) | |
| tree | 5b85ab90389fb33dad8254144ed629c3ff508cab | |
| parent | Fix (diff) | |
| download | textual-inversion-diff-4370be7d67f4cec00c14c8a8b3e999d50b75585d.tar.gz textual-inversion-diff-4370be7d67f4cec00c14c8a8b3e999d50b75585d.tar.bz2 textual-inversion-diff-4370be7d67f4cec00c14c8a8b3e999d50b75585d.zip | |
Disable Adam weight decay
| -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 951f3dd..2c5037f 100644 --- a/train_ti.py +++ b/train_ti.py | |||
| @@ -324,7 +324,7 @@ def parse_args(): | |||
| 324 | parser.add_argument( | 324 | parser.add_argument( |
| 325 | "--adam_weight_decay", | 325 | "--adam_weight_decay", |
| 326 | type=float, | 326 | type=float, |
| 327 | default=1e-2, | 327 | default=0, |
| 328 | help="Weight decay to use." | 328 | help="Weight decay to use." |
| 329 | ) | 329 | ) |
| 330 | parser.add_argument( | 330 | parser.add_argument( |