提交 1c0ec36e 编写于 作者: Z zhangxuefei

Fix the bug that can't update the parameter use_gpu

上级 4bf70378
......@@ -17,6 +17,7 @@ from __future__ import division
from __future__ import print_function
import argparse
import ast
import distutils.util
......@@ -36,7 +37,10 @@ def parse_args():
"--data_path", type=str, help="all the data for train,valid,test")
parser.add_argument('--para_init', action='store_true')
parser.add_argument(
'--use_gpu', type=bool, default=False, help='whether using gpu')
'--use_gpu',
type=ast.literal_eval,
default=False,
help='whether using gpu')
parser.add_argument(
'--log_path',
help='path of the log file. If not set, logs are printed to console')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册