From 79c367471f25f2b8703195e572587e81ff8e229d Mon Sep 17 00:00:00 2001
From: root <root@bjyz-sys-gpu-kongming2.bjyz.baidu.com>
Date: Mon, 2 Dec 2019 07:46:07 +0000
Subject: [PATCH] rm onnx_infer.py

---
 setup.py                         | 9 +++------
 x2paddle/decoder/onnx_decoder.py | 6 ++++--
 2 files changed, 7 insertions(+), 8 deletions(-)

diff --git a/setup.py b/setup.py
index 5aa9148..5170820 100644
--- a/setup.py
+++ b/setup.py
@@ -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',
+    ]})
diff --git a/x2paddle/decoder/onnx_decoder.py b/x2paddle/decoder/onnx_decoder.py
index b916085..5673d22 100644
--- a/x2paddle/decoder/onnx_decoder.py
+++ b/x2paddle/decoder/onnx_decoder.py
@@ -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 = {
-- 
GitLab