Fix: Fix the expression type error and changge in to float.

chairc · Sep 22, 2023 · 2ca7449 · 2ca7449
1 parent 6297617
commit 2ca7449
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/train.py b/tools/train.py
@@ -360,7 +360,7 @@ def main(args):
     # Option: gelu/silu/relu/relu6/lrelu
     parser.add_argument("--act", type=str, default="gelu")
     # Learning rate (needed)
-    parser.add_argument("--lr", type=int, default=3e-4)
+    parser.add_argument("--lr", type=float, default=3e-4)
     # Learning rate function (needed)
     # Option: linear/cosine/warmup_cosine
     parser.add_argument("--lr_func", type=str, default="")