Update prune_walker.py (#357)

2bb6377f · parap1uie-s · GitHub · e6cffba8 · 2bb6377f
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

paddleslim/prune/prune_walker.py paddleslim/prune/prune_walker.py +1 -1

未找到文件。
--- a/paddleslim/prune/prune_walker.py
+++ b/paddleslim/prune/prune_walker.py
@@ -542,7 +542,7 @@ class depthwise_conv2d(PruneWorker):
            self._visit(filter_var, 0)

            new_groups = filter_var.shape()[0] - len(pruned_idx)
-            op.set_attr("groups", new_groups)
+            self.op.set_attr("groups", new_groups)

            for op in filter_var.outputs():
                self._prune_op(op, filter_var, 0, pruned_idx)