From 8069fff5e6d19f248ebf062d58307fb019ee9ab9 Mon Sep 17 00:00:00 2001
From: wjj19950828 <wjjisloser@163.com>
Date: Tue, 7 Jun 2022 15:11:02 +0800
Subject: [PATCH] rm fluid for channel_shuffle

---
 x2paddle/op_mapper/caffe2paddle/caffe_op_mapper.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/x2paddle/op_mapper/caffe2paddle/caffe_op_mapper.py b/x2paddle/op_mapper/caffe2paddle/caffe_op_mapper.py
index 8037308..f543122 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(
-- 
GitLab