未验证 提交 64c41a03 编写于 作者: S SunAhong1993 提交者: GitHub

fix the convert.py args (#556)

上级 b8dcdf69
...@@ -41,7 +41,6 @@ def arg_parser(): ...@@ -41,7 +41,6 @@ def arg_parser():
parser.add_argument( parser.add_argument(
"--save_dir", "--save_dir",
"-s", "-s",
required=True,
type=_text_type, type=_text_type,
default=None, default=None,
help="path to save translated model") help="path to save translated model")
...@@ -221,6 +220,8 @@ def main(): ...@@ -221,6 +220,8 @@ def main():
x2paddle.__version__)) x2paddle.__version__))
return return
assert args.save_dir is not None, "--save_dir is not defined"
try: try:
import platform import platform
v0, v1, v2 = platform.python_version().split('.') v0, v1, v2 = platform.python_version().split('.')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册