Fixed a bug in global head pruning

VainF · Oct 16, 2023 · 3ab8214 · 3ab8214
1 parent 50ad632
commit 3ab8214
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/torch_pruning/pruner/algorithms/metapruner.py b/torch_pruning/pruner/algorithms/metapruner.py
@@ -132,12 +132,9 @@ def __init__(
             output_transform=output_transform,
             unwrapped_parameters=unwrapped_parameters,
             customized_pruners=customized_pruners,
-            ignored_layers=self.ignored_layers,
             ignored_params=self.ignored_params,
         )
 
-
-
         ###############################################
         # Iterative pruning
         # The pruner will prune the model iteratively for several steps to achieve the target pruning ratio