neural_compressor.compression.pruner.pruners.progressive

Progressive pruner.

Module Contents

Classes

PytorchProgressivePruner

Pruning Pruner.
class neural_compressor.compression.pruner.pruners.progressive.PytorchProgressivePruner(config, modules)[source]

Pruning Pruner.

A Pruner class derived from BasicPruner. In this pruner, mask interpolation will be applied. Mask interpolation is a fine-grained improvement for NxM structured pruning by adding interval

masks between masks of two pruning steps.

Parameters:

  • modules – A dict {“module_name”: Tensor} that stores the pruning modules’ weights.

  • config – A config dict object that contains the pruner information.

Inherit from parent class Pruner.