提交 497cb173 编写于 作者: R root

rm onnx_infer.py

上级 bc736d9c
......@@ -23,9 +23,6 @@ setuptools.setup(
"Operating System :: OS Independent",
],
license='Apache 2.0',
entry_points={
'console_scripts': [
'x2paddle=x2paddle.convert:main',
# 'onnx_infer=x2paddle.onnx_infer:main'
]
})
entry_points={'console_scripts': [
'x2paddle=x2paddle.convert:main',
]})
......@@ -490,8 +490,10 @@ class ONNXDecoder(object):
model = onnx.load(model_path)
model = onnx.shape_inference.infer_shapes(model)
if len(model.graph.value_info) < len(model.graph.node) - 1:
print(
"shape inference for some operators failed, those operators will be assignd node.out_shape==None, refer to https://github.com/onnx/onnx/blob/master/docs/ShapeInference.md"
_logger.warning(
'shape inference for some operators failed, '
'those operators will be assignd node.out_shape==None, '
'refer to https://github.com/onnx/onnx/blob/master/docs/ShapeInference.md'
)
try:
datatype_map = {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册