提交 f6af37b4 编写于 作者: S SunAhong1993

fix the bug

上级 8fd11dda
...@@ -322,7 +322,7 @@ class PaddleGraph(object): ...@@ -322,7 +322,7 @@ class PaddleGraph(object):
if self.source_type == "caffe": if self.source_type == "caffe":
custom_import = "from x2paddle.op_mapper.static.caffe2paddle " + \ custom_import = "from x2paddle.op_mapper.static.caffe2paddle " + \
"import caffe_custom_layer as x2paddle_nn" "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 " + \ custom_import = "from x2paddle.op_mapper.static.onnx2paddle " + \
"import onnx_custom_layer as x2paddle_nn" "import onnx_custom_layer as x2paddle_nn"
else: else:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册