diff --git a/x2paddle/convert.py b/x2paddle/convert.py index 2c3e157c94490ccc2a05d2a621166fb040cd2ac6..a7e4b6059da5dd9f0b91c643ce2fdad087b359fe 100644 --- a/x2paddle/convert.py +++ b/x2paddle/convert.py @@ -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('.')