Fix pruner to support repeat pruning. (#36)

6cde8f01 · whs · GitHub · d7e074c2 · 6cde8f01
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

paddleslim/prune/pruner.py paddleslim/prune/pruner.py +1 -0

未找到文件。
--- a/paddleslim/prune/pruner.py
+++ b/paddleslim/prune/pruner.py
@@ -125,6 +125,7 @@ class Pruner():
                    param_t.set(pruned_param, place)
        graph.update_groups_of_conv()
+        graph.infer_shape()
        return graph.program, param_backup, param_shape_backup
    def _cal_pruned_idx(self, param, ratio, axis):