提交 8a4a330a 编写于 作者: C channingss

add prompt

上级 212c3d90
......@@ -464,10 +464,12 @@ class ONNXDecoder(object):
import torch
version = torch.__version__
if '1.1.0' not in version:
print("torch==1.1.0 is required")
print("your model have dynamic graph, torch==1.1.0 is required")
return
except:
print("onnx is not installed, use \"pip install torch==1.1.0\".")
print(
"your model have dynamic graph, we use caff2 to inference graph, please use \"pip install torch==1.1.0\"."
)
return
from caffe2.python.onnx.backend import prepare
shape = input_shapes[0]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册