提交 be0f062f 编写于 作者: C channingss

add check torch version

上级 7670fee9
...@@ -461,6 +461,15 @@ class ONNXDecoder(object): ...@@ -461,6 +461,15 @@ class ONNXDecoder(object):
""" """
get dynamic shape from caffe2.backend get dynamic shape from caffe2.backend
""" """
try:
import torch
version = torch.__version__
if '1.1.0' not in version:
print("torch==1.1.0 is required")
return
except:
print("onnx is not installed, use \"pip install torch==1.1.0\".")
return
from caffe2.python.onnx.backend import prepare from caffe2.python.onnx.backend import prepare
shape = input_shapes[0] shape = input_shapes[0]
np_images = np.random.rand(shape[0], shape[1], shape[2], np_images = np.random.rand(shape[0], shape[1], shape[2],
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册