summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-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(