未验证 提交 2bb6377f 编写于 作者: P parap1uie-s 提交者: GitHub

Update prune_walker.py (#357)

上级 e6cffba8
...@@ -542,7 +542,7 @@ class depthwise_conv2d(PruneWorker): ...@@ -542,7 +542,7 @@ class depthwise_conv2d(PruneWorker):
self._visit(filter_var, 0) self._visit(filter_var, 0)
new_groups = filter_var.shape()[0] - len(pruned_idx) 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(): for op in filter_var.outputs():
self._prune_op(op, filter_var, 0, pruned_idx) self._prune_op(op, filter_var, 0, pruned_idx)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册