Fix a bug in use_gru argument parsing.

7 years ago · 177af05953
parent 8f89a9bdd4
commit 177af05953
5 changed files with 9 additions and 9 deletions
--- a/demo_server.py
+++ b/demo_server.py
@ -68,8 +68,8 @@ parser.add_argument(
    help="RNN layer cell number. (default: %(default)s)")
 parser.add_argument(
    "--use_gru",
-    default=True,
-    type=bool,
+    default=False,
+    type=distutils.util.strtobool,
    help="Use GRU or simple RNN. (default: %(default)s)")
 parser.add_argument(
    "--use_gpu",
--- a/evaluate.py
+++ b/evaluate.py
@ -40,8 +40,8 @@ parser.add_argument(
    help="RNN layer cell number. (default: %(default)s)")
 parser.add_argument(
    "--use_gru",
-    default=True,
-    type=bool,
+    default=False,
+    type=distutils.util.strtobool,
    help="Use GRU or simple RNN. (default: %(default)s)")
 parser.add_argument(
    "--use_gpu",
--- a/infer.py
+++ b/infer.py
@ -35,8 +35,8 @@ parser.add_argument(
    help="RNN layer cell number. (default: %(default)s)")
 parser.add_argument(
    "--use_gru",
-    default=True,
-    type=bool,
+    default=False,
+    type=distutils.util.strtobool,
    help="Use GRU or simple RNN. (default: %(default)s)")
 parser.add_argument(
    "--use_gpu",
--- a/train.py
+++ b/train.py
@ -43,7 +43,7 @@ parser.add_argument(
 parser.add_argument(
    "--use_gru",
    default=False,
-    type=bool,
+    type=distutils.util.strtobool,
    help="Use GRU or simple RNN. (default: %(default)s)")
 parser.add_argument(
    "--adam_learning_rate",
--- a/tune.py
+++ b/tune.py
@ -36,8 +36,8 @@ parser.add_argument(
    help="RNN layer cell number. (default: %(default)s)")
 parser.add_argument(
    "--use_gru",
-    default=True,
-    type=bool,
+    default=False,
+    type=distutils.util.strtobool,
    help="Use GRU or simple RNN. (default: %(default)s)")
 parser.add_argument(
    "--use_gpu",