fix the bug

f6af37b4 · SunAhong1993 · 8fd11dda · f6af37b4
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

x2paddle/core/program.py x2paddle/core/program.py +1 -1

未找到文件。
--- a/x2paddle/core/program.py
+++ b/x2paddle/core/program.py
@@ -322,7 +322,7 @@ class PaddleGraph(object):
        if self.source_type == "caffe":
            custom_import = "from x2paddle.op_mapper.static.caffe2paddle " + \
                             "import caffe_custom_layer as x2paddle_nn"
-        if self.source_type == "onnx":
+        elif self.source_type == "onnx":
            custom_import = "from x2paddle.op_mapper.static.onnx2paddle " + \
                             "import onnx_custom_layer as x2paddle_nn"
        else: