summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorVolpeon <git@volpeon.ink>2023-01-12 07:18:21 +0100
committerVolpeon <git@volpeon.ink>2023-01-12 07:18:21 +0100
commit4370be7d67f4cec00c14c8a8b3e999d50b75585d (patch)
tree5b85ab90389fb33dad8254144ed629c3ff508cab
parentFix (diff)
downloadtextual-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.py2
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(