From 4370be7d67f4cec00c14c8a8b3e999d50b75585d Mon Sep 17 00:00:00 2001
From: Volpeon <git@volpeon.ink>
Date: Thu, 12 Jan 2023 07:18:21 +0100
Subject: Disable Adam weight decay

---
 train_ti.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'train_ti.py')

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():
     parser.add_argument(
         "--adam_weight_decay",
         type=float,
-        default=1e-2,
+        default=0,
         help="Weight decay to use."
     )
     parser.add_argument(
-- 
cgit v1.2.3-54-g00ecf