neural_compressor.experimental.pruner_legacy.gradient_sensitivity

Gradient sensitivity pruner.

Classes

GradientSensitivityPruner

Gradient sensitivity pruner class.

Module Contents

class neural_compressor.experimental.pruner_legacy.gradient_sensitivity.GradientSensitivityPruner(model, local_config, global_config)[source]

Gradient sensitivity pruner class.

Parameters:

  • model (object) – The original model (currently PyTorchModel instance).

  • local_config (Conf) – configs specific for this pruning instance.

  • global_config (Conf) – global configs which may be overwritten by local_config.