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

run pre-commit

上级 c990aed7
......@@ -193,7 +193,7 @@ def main():
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"
if args.framework == "onnx":
try:
import onnxruntime as rt
......@@ -202,8 +202,10 @@ def main():
print("onnxruntime==1.0.0 is required")
return
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:
import paddle
v0, v1, v2 = paddle.__version__.split('.')
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册