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

rm onnx_infer.py

上级 bc736d9c
...@@ -23,9 +23,6 @@ setuptools.setup( ...@@ -23,9 +23,6 @@ setuptools.setup(
"Operating System :: OS Independent", "Operating System :: OS Independent",
], ],
license='Apache 2.0', license='Apache 2.0',
entry_points={ entry_points={'console_scripts': [
'console_scripts': [
'x2paddle=x2paddle.convert:main', 'x2paddle=x2paddle.convert:main',
# 'onnx_infer=x2paddle.onnx_infer:main' ]})
]
})
...@@ -490,8 +490,10 @@ class ONNXDecoder(object): ...@@ -490,8 +490,10 @@ class ONNXDecoder(object):
model = onnx.load(model_path) model = onnx.load(model_path)
model = onnx.shape_inference.infer_shapes(model) model = onnx.shape_inference.infer_shapes(model)
if len(model.graph.value_info) < len(model.graph.node) - 1: if len(model.graph.value_info) < len(model.graph.node) - 1:
print( _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" '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: try:
datatype_map = { datatype_map = {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册