b0c0eb7b1f2ac9a983c550ee971cea275463d8fc,src/sdk/pynni/nni/compression/torch/builtin_pruners.py,L2FilterPruner,_get_mask,#L2FilterPruner#Any#Any#Any#,362

Before Change

        threshold = torch.topk(w_l2_norm.view(-1), num_prune, largest=False)[0].max()
        mask = torch.gt(w_l2_norm, threshold)[:, None, None, None].expand_as(weight).type_as(weight)

        return mask


class FPGMPruner(RankFilterPruner):
    

After Change

        mask_weight = torch.gt(w_l2_norm, threshold)[:, None, None, None].expand_as(weight).type_as(weight)
        mask_bias = torch.gt(w_l2_norm, threshold).type_as(weight)

        return {"weight": mask_weight.detach(), "bias": mask_bias.detach()}


class FPGMPruner(WeightRankFilterPruner):
    

Instances