diff --git a/torch_pruning/pruner/algorithms/metapruner.py b/torch_pruning/pruner/algorithms/metapruner.py index 930ae67..987f0d6 100644 --- 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