prune
Modules
Module implementing |
|
Module implementing |
|
Handles prune plugins for third-party modules. |
|
High-level API to automatically prune and optimize your model with various algorithms. |
A simplified API for modelopt.torch.nas
for pruning algorithms.
This module provides a simplified API for pruning that is based on the NAS infrastructure but simplifies the overall workflow to accommodate for the simpler nature of pruning algorithms.