提交 63aaef9b 编写于 作者: M mamingjie-China

run pre-commit

上级 c990aed7
...@@ -193,7 +193,7 @@ def main(): ...@@ -193,7 +193,7 @@ def main():
assert args.framework is not None, "--framework is not defined(support tensorflow/caffe/onnx)" assert args.framework is not None, "--framework is not defined(support tensorflow/caffe/onnx)"
assert args.save_dir is not None, "--save_dir is not defined" assert args.save_dir is not None, "--save_dir is not defined"
if args.framework == "onnx": if args.framework == "onnx":
try: try:
import onnxruntime as rt import onnxruntime as rt
...@@ -202,8 +202,10 @@ def main(): ...@@ -202,8 +202,10 @@ def main():
print("onnxruntime==1.0.0 is required") print("onnxruntime==1.0.0 is required")
return return
except: except:
print("onnxruntime is not installed, use \"pip install onnxruntime==1.0.0\".") print(
"onnxruntime is not installed, use \"pip install onnxruntime==1.0.0\"."
)
try: try:
import paddle import paddle
v0, v1, v2 = paddle.__version__.split('.') v0, v1, v2 = paddle.__version__.split('.')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册