Ver Fonte

fix conficts

jiangjiajun há 5 anos atrás
pai
commit
1d266c904d
1 ficheiros alterados com 1 adições e 2 exclusões
  1. 1 2
      paddlex/convertor.py

+ 1 - 2
paddlex/convertor.py

@@ -46,11 +46,10 @@ def export_onnx_model(model, save_dir):
         from fluid_onnx.variables import paddle_variable_to_onnx_tensor, paddle_onnx_weight
         from debug.model_check import debug_model, Tracker
     except Exception as e:
-        logging.error(e)
         logging.error(
             "Import Module Failed! Please install paddle2onnx. Related requirements see https://github.com/PaddlePaddle/paddle2onnx."
         )
-        sys.exit(-1)
+        raise e
     place = fluid.CPUPlace()
     exe = fluid.Executor(place)
     inference_scope = fluid.global_scope()