From ac10acf01fe0168eac250ff23892d255797f461d Mon Sep 17 00:00:00 2001 From: whs Date: Thu, 23 Apr 2020 17:01:43 +0800 Subject: [PATCH] Skip empty group while pruning. (#236) --- paddleslim/prune/pruner.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/paddleslim/prune/pruner.py b/paddleslim/prune/pruner.py index 90ff075e..52f388ac 100644 --- a/paddleslim/prune/pruner.py +++ b/paddleslim/prune/pruner.py @@ -91,6 +91,8 @@ class Pruner(): 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) -- GitLab