提交 507e6db3 编写于 作者: X Xinghai Sun 提交者: GitHub

Merge pull request #223 from xinghai-sun/bug

Fix a bug in use_gru argument parsing.
...@@ -68,8 +68,8 @@ parser.add_argument( ...@@ -68,8 +68,8 @@ parser.add_argument(
help="RNN layer cell number. (default: %(default)s)") help="RNN layer cell number. (default: %(default)s)")
parser.add_argument( parser.add_argument(
"--use_gru", "--use_gru",
default=True, default=False,
type=bool, type=distutils.util.strtobool,
help="Use GRU or simple RNN. (default: %(default)s)") help="Use GRU or simple RNN. (default: %(default)s)")
parser.add_argument( parser.add_argument(
"--use_gpu", "--use_gpu",
......
...@@ -40,8 +40,8 @@ parser.add_argument( ...@@ -40,8 +40,8 @@ parser.add_argument(
help="RNN layer cell number. (default: %(default)s)") help="RNN layer cell number. (default: %(default)s)")
parser.add_argument( parser.add_argument(
"--use_gru", "--use_gru",
default=True, default=False,
type=bool, type=distutils.util.strtobool,
help="Use GRU or simple RNN. (default: %(default)s)") help="Use GRU or simple RNN. (default: %(default)s)")
parser.add_argument( parser.add_argument(
"--use_gpu", "--use_gpu",
......
...@@ -35,8 +35,8 @@ parser.add_argument( ...@@ -35,8 +35,8 @@ parser.add_argument(
help="RNN layer cell number. (default: %(default)s)") help="RNN layer cell number. (default: %(default)s)")
parser.add_argument( parser.add_argument(
"--use_gru", "--use_gru",
default=True, default=False,
type=bool, type=distutils.util.strtobool,
help="Use GRU or simple RNN. (default: %(default)s)") help="Use GRU or simple RNN. (default: %(default)s)")
parser.add_argument( parser.add_argument(
"--use_gpu", "--use_gpu",
......
...@@ -43,7 +43,7 @@ parser.add_argument( ...@@ -43,7 +43,7 @@ parser.add_argument(
parser.add_argument( parser.add_argument(
"--use_gru", "--use_gru",
default=False, default=False,
type=bool, type=distutils.util.strtobool,
help="Use GRU or simple RNN. (default: %(default)s)") help="Use GRU or simple RNN. (default: %(default)s)")
parser.add_argument( parser.add_argument(
"--adam_learning_rate", "--adam_learning_rate",
......
...@@ -36,8 +36,8 @@ parser.add_argument( ...@@ -36,8 +36,8 @@ parser.add_argument(
help="RNN layer cell number. (default: %(default)s)") help="RNN layer cell number. (default: %(default)s)")
parser.add_argument( parser.add_argument(
"--use_gru", "--use_gru",
default=True, default=False,
type=bool, type=distutils.util.strtobool,
help="Use GRU or simple RNN. (default: %(default)s)") help="Use GRU or simple RNN. (default: %(default)s)")
parser.add_argument( parser.add_argument(
"--use_gpu", "--use_gpu",
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册