From 98f8f94a845a6c6aab756ff1de552abc56811df8 Mon Sep 17 00:00:00 2001
From: jamesking76 <james.a.king76@gmail.com>
Date: Mon, 25 Nov 2024 10:56:19 +0000
Subject: [PATCH] Fix bug: moved handling of the case where imp is none before
 len is called

---
 torch_pruning/pruner/algorithms/metapruner.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torch_pruning/pruner/algorithms/metapruner.py b/torch_pruning/pruner/algorithms/metapruner.py
index 7e92bda..464e121 100644
--- a/torch_pruning/pruner/algorithms/metapruner.py
+++ b/torch_pruning/pruner/algorithms/metapruner.py
@@ -418,8 +418,8 @@ def _prune(self) -> typing.Generator:
                     if group is None: continue
                 ch_groups = self._get_channel_groups(group)
                 imp = self.estimate_importance(group) # raw importance score
-                group_size = len(imp) // ch_groups
                 if imp is None: continue 
+                group_size = len(imp) // ch_groups
                 if ch_groups > 1: # layers with dimension grouping, such as GroupConv, GroupNorm, Multi-head attention, etc.
                     # We average importance across groups here. For example:
                     # imp = [1, 2, 3, 4, 5, 6] with ch_groups=2.