未验证 提交 6cde8f01 编写于 作者: W whs 提交者: GitHub

Fix pruner to support repeat pruning. (#36)

上级 d7e074c2
...@@ -125,6 +125,7 @@ class Pruner(): ...@@ -125,6 +125,7 @@ class Pruner():
param_t.set(pruned_param, place) param_t.set(pruned_param, place)
graph.update_groups_of_conv() graph.update_groups_of_conv()
graph.infer_shape()
return graph.program, param_backup, param_shape_backup return graph.program, param_backup, param_shape_backup
def _cal_pruned_idx(self, param, ratio, axis): def _cal_pruned_idx(self, param, ratio, axis):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册