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

add prompt

上级 212c3d90
...@@ -464,10 +464,12 @@ class ONNXDecoder(object): ...@@ -464,10 +464,12 @@ class ONNXDecoder(object):
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("torch==1.1.0 is required") print("your model have dynamic graph, torch==1.1.0 is required")
return return
except: 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 return
from caffe2.python.onnx.backend import prepare from caffe2.python.onnx.backend import prepare
shape = input_shapes[0] shape = input_shapes[0]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册