From 2691a9ded60ce1757136aadcefaeebeefc4d70d4 Mon Sep 17 00:00:00 2001 From: Gongfan Fang Date: Sat, 9 Mar 2024 22:23:11 +0800 Subject: [PATCH] Update importance.py --- torch_pruning/pruner/importance.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/torch_pruning/pruner/importance.py b/torch_pruning/pruner/importance.py index e4f0e1c..75046ae 100644 --- 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 \ No newline at end of file + pass