503a3579511fcb40642a09958dd5377875647309,src/sdk/pynni/nni/compression/torch/builtin_pruners.py,L1FilterPruner,calc_mask,#L1FilterPruner#Any#Any#,308

Before Change

            if k == 0:
                return torch.ones(weight.shape).type_as(weight)
            w_abs_structured = w_abs.view(filters, -1).sum(dim=1)
            threshold = torch.topk(w_abs_structured.view(-1), k, largest=False).values.max()
            mask = torch.gt(w_abs_structured, threshold)[:, None, None, None].expand_as(weight).type_as(weight)
        finally:
            self.mask_dict.update({layer.name: mask})

After Change

            if k == 0:
                return torch.ones(weight.shape).type_as(weight)
            w_abs_structured = w_abs.view(filters, -1).sum(dim=1)
            threshold = torch.topk(w_abs_structured.view(-1), k, largest=False)[].max()
            mask = torch.gt(w_abs_structured, threshold)[:, None, None, None].expand_as(weight).type_as(weight)
        finally:
            self.mask_dict.update({layer.name: mask})

Instances