From 6cde8f0193a10cbe5ec177313c71241ef67f9e2d Mon Sep 17 00:00:00 2001
From: whs <wanghaoshuang@baidu.com>
Date: Tue, 14 Jan 2020 18:09:30 +0800
Subject: [PATCH] Fix pruner to support repeat pruning. (#36)

---
 paddleslim/prune/pruner.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/paddleslim/prune/pruner.py b/paddleslim/prune/pruner.py
index 8e656300..98abe950 100644
--- 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):
-- 
GitLab