Update importance.py

VainF · Mar 9, 2024 · 2691a9d · 2691a9d
1 parent 12392c8
commit 2691a9d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/torch_pruning/pruner/importance.py b/torch_pruning/pruner/importance.py
@@ -309,7 +309,7 @@ def __init__(self, p=2, group_reduction="mean", normalizer='lamp', bias=False):
         super().__init__(p=p, group_reduction=group_reduction, normalizer=normalizer, bias=bias)
 
 
-class FPGMImportance(MagnitudeImportance):
+class FPGMImportance(GroupNormImportance):
     """Filter Pruning via Geometric Median for Deep Convolutional Neural Networks Acceleration,
     http://openaccess.thecvf.com/content_CVPR_2019/papers/He_Filter_Pruning_via_Geometric_Median_for_Deep_Convolutional_Neural_Networks_CVPR_2019_paper.pdf
     """
@@ -533,7 +533,7 @@ def __call__(self, group):
         group_imp = self._normalize(group_imp, self.normalizer)
         return group_imp
 
-class OBDCImportance(MagnitudeImportance):
+class OBDCImportance(GroupNormImportance):
     """EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasis:
        http://proceedings.mlr.press/v97/wang19g/wang19g.pdf
     """
@@ -816,4 +816,4 @@ class TaylorImportance(GroupTaylorImportance):
     pass
 
 class HessianImportance(GroupHessianImportance):
-    pass
+    pass