/TorchPruner

On-the-fly Structured Pruning for PyTorch models. This library implements several attributions metrics and structured pruning utils for neural networks in PyTorch.

Primary LanguageJupyter NotebookMIT LicenseMIT

Watchers