importance_hooks

Modules

modelopt.torch.prune.importance_hooks.base_hooks

Forward hooks for activation-based importance estimation.

modelopt.torch.prune.importance_hooks.base_hooks_analysis

Analysis tools for evaluating importance scores from hooks.

modelopt.torch.prune.importance_hooks.compare_module_outputs

Compare module output tensors from different model variants.

modelopt.torch.prune.importance_hooks.plugins

Plugin-specific hooks for importance estimation.

Forward hooks for estimating importance scores for pruning.