未验证 提交 432c3a15 编写于 作者: C channings 提交者: GitHub

Update convert.py

上级 44e2f4b3
...@@ -114,11 +114,11 @@ def onnx2paddle(model_path, save_dir): ...@@ -114,11 +114,11 @@ def onnx2paddle(model_path, save_dir):
import torch import torch
version = torch.__version__ version = torch.__version__
if '1.1.0' not in version: if '1.1.0' not in version:
print("your model have dynamic graph, torch==1.1.0 is required") print("torch==1.1.0 is required")
return return
except: except:
print( print(
"your model have dynamic graph, we use caffe2 to inference graph, please use \"pip install torch==1.1.0\"." "we use caffe2 to inference graph, please use \"pip install torch==1.1.0\"."
) )
return return
print("Now translating model from onnx to paddle.") print("Now translating model from onnx to paddle.")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册