diff --git a/x2paddle/op_mapper/caffe2paddle/caffe_op_mapper.py b/x2paddle/op_mapper/caffe2paddle/caffe_op_mapper.py index 8037308146c1752dc3b3270f0646a24e09e07124..f543122747c044df95bb663c85f84329cf7c2880 100644 --- a/x2paddle/op_mapper/caffe2paddle/caffe_op_mapper.py +++ b/x2paddle/op_mapper/caffe2paddle/caffe_op_mapper.py @@ -1209,10 +1209,10 @@ class CaffeOpMapper(): input = self.graph.get_input_node(node, idx=0, copy=True) params = node.layer.shuffle_channel_param self.paddle_graph.add_layer( - "paddle.fluid.layers.shuffle_channel", + "paddle.nn.functional.channel_shuffle", inputs={"x": input.name}, outputs=[node.layer_name], - group=params.group) + groups=params.group) def Upsample(self, node): assert len(