Fix a bug in use_gru argument parsing.

177af059 · Xinghai Sun · 8f89a9bd · 177af059 · 177af059 · 177af059
隐藏空白更改
内联并排

Showing with 9 addition and 9 deletion

demo_server.py demo_server.py +2 -2

evaluate.py evaluate.py +2 -2

infer.py infer.py +2 -2

train.py train.py +1 -1

tune.py tune.py +2 -2

未找到文件。
--- 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",