2de52a8976971da4836727ba9242fedcc7474878,src/sdk/pynni/nni/compression/torch/weight_rank_filter_pruners.py,WeightRankFilterPruner,__init__,#WeightRankFilterPruner#Any#Any#,18

Before Change

        

        super().__init__(model, config_list)
        self.mask_calculated_ops = set()  // operations whose mask has been calculated

    def get_mask(self, base_mask, weight, num_prune):
        raise NotImplementedError("{} get_mask is not implemented".format(self.__class__.__name__))

After Change

        

        super().__init__(model, config_list)
        self.register_buffer("if_calculated", torch.tensor(False)) // pylint: disable=not-callable

    def get_mask(self, base_mask, weight, num_prune):
        raise NotImplementedError("{} get_mask is not implemented".format(self.__class__.__name__))

Instances