Skip empty group while pruning. (PaddlePaddle#236)

lizexu123 · Apr 23, 2020 · ac10acf · ac10acf
1 parent f3fcd83
commit ac10acf
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/paddleslim/prune/pruner.py b/paddleslim/prune/pruner.py
@@ -91,6 +91,8 @@ def prune(self,
         pruned_params = []
         for param, ratio in zip(params, ratios):
             group = collect_convs([param], graph, visited)[0]  # [(name, axis)]
+            if group is None or len(group) == 0:
+                continue
             if only_graph and self.idx_selector.__name__ == "default_idx_selector":
 
                 param_v = graph.var(param)