fix: Fixed W&B config log

mindee · Nov 29, 2021 · 9d9937d · 9d9937d
1 parent 9e5b3a7
commit 9d9937d
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/references/classification/train_tensorflow.py b/references/classification/train_tensorflow.py
@@ -185,14 +185,14 @@ def main(args):
             config={
                 "learning_rate": args.lr,
                 "epochs": args.epochs,
-                "weight_decay": args.weight_decay,
+                "weight_decay": 0.,
                 "batch_size": args.batch_size,
                 "architecture": args.arch,
                 "input_size": args.input_size,
                 "optimizer": "adam",
                 "framework": "tensorflow",
                 "vocab": args.vocab,
-                "scheduler": args.sched,
+                "scheduler": "exp_decay",
                 "pretrained": args.pretrained,
             }
         )

diff --git a/references/detection/train_tensorflow.py b/references/detection/train_tensorflow.py
@@ -169,13 +169,13 @@ def main(args):
             config={
                 "learning_rate": args.lr,
                 "epochs": args.epochs,
-                "weight_decay": args.weight_decay,
+                "weight_decay": 0.,
                 "batch_size": args.batch_size,
                 "architecture": args.arch,
                 "input_size": args.input_size,
                 "optimizer": "adam",
                 "framework": "tensorflow",
-                "scheduler": args.sched,
+                "scheduler": "exp_decay",
                 "train_hash": train_hash,
                 "val_hash": val_hash,
                 "pretrained": args.pretrained,

diff --git a/references/recognition/train_tensorflow.py b/references/recognition/train_tensorflow.py
@@ -179,13 +179,13 @@ def main(args):
             config={
                 "learning_rate": args.lr,
                 "epochs": args.epochs,
-                "weight_decay": args.weight_decay,
+                "weight_decay": 0.,
                 "batch_size": args.batch_size,
                 "architecture": args.arch,
                 "input_size": args.input_size,
                 "optimizer": "adam",
                 "framework": "tensorflow",
-                "scheduler": args.sched,
+                "scheduler": "exp_decay",
                 "vocab": args.vocab,
                 "train_hash": train_hash,
                 "val_hash": val_hash,