fix x2paddle import (#1364)

paddleslim/common/load_model.py

@@ -115,24 +115,14 @@ def load_onnx_model(model_path, disable_feedback=False):
         return val_program, feed_target_names, fetch_targets
     else:
         # onnx to paddle inference model.
+        assert os.path.exists(
+            model_path), 'Not found `{}`, please check model path.'.format(
+                model_path)
         try:
             pkg.require('x2paddle')
         except:
             from pip._internal import main
             main(['install', 'x2paddle'])
-        try:
-            from x2paddle.decoder.onnx_decoder import ONNXDecoder
-            from x2paddle.op_mapper.onnx2paddle.onnx_op_mapper import ONNXOpMapper
-            from x2paddle.optimizer.optimizer import GraphOptimizer
-            from x2paddle.utils import ConverterCheck
-        except:
-            _logger.error(
-                "x2paddle is not installed, please use \"pip install x2paddle\"."
-            )
-        time_info = int(time.time())
-        if not disable_feedback:
-            ConverterCheck(
-                task="ONNX", time_info=time_info, convert_state="Start").start()
         # check onnx installation and version
         try:
             pkg.require('onnx')
@@ -147,6 +137,14 @@ def load_onnx_model(model_path, disable_feedback=False):
             from pip._internal import main
             main(['install', 'onnx==1.12.0'])
 
+        from x2paddle.decoder.onnx_decoder import ONNXDecoder
+        from x2paddle.op_mapper.onnx2paddle.onnx_op_mapper import ONNXOpMapper
+        from x2paddle.optimizer.optimizer import GraphOptimizer
+        from x2paddle.utils import ConverterCheck
+        time_info = int(time.time())
+        if not disable_feedback:
+            ConverterCheck(
+                task="ONNX", time_info=time_info, convert_state="Start").start()
         # support distributed convert model
         model_idx = paddle.distributed.get_rank(
         ) if paddle.distributed.get_world_size() > 1 else 0
@@ -166,9 +164,10 @@ def load_onnx_model(model_path, disable_feedback=False):
                 ConverterCheck(
                     task="ONNX", time_info=time_info,
                     convert_state="Success").start()
-        except:
-            _logger.info(
-                "[ERROR] x2paddle threw an exception, you can ask for help at: https://github.com/PaddlePaddle/X2Paddle/issues"
+        except Exception as e:
+            _logger.warning(e)
+            _logger.error(
+                "x2paddle threw an exception, you can ask for help at: https://github.com/PaddlePaddle/X2Paddle/issues"
             )
             sys.exit(1)