提交 8069fff5 编写于 作者: W wjj19950828

rm fluid for channel_shuffle

上级 ff6c6996
......@@ -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(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册