修剪

模块

modelopt.torch.prune.config	修剪模式的默认配置。
modelopt.torch.prune.fastnas	实现用于搜索的fasnas剪枝算法的模块。
modelopt.torch.prune.gradnas	实现用于搜索的gradnas剪枝算法的模块。
modelopt.torch.prune.mcore_gpt_minitron	模块实现了用于NVIDIA Megatron-Core / NeMo模型的mcore_gpt_minitron剪枝算法。
modelopt.torch.prune.mode	模块实现并描述了在NAS转换过程中可以使用的模式。
modelopt.torch.prune.plugins	处理第三方模块的修剪插件。
modelopt.torch.prune.pruning	高级API，用于使用各种算法自动修剪和优化您的模型。

为剪枝算法提供的简化API，适用于modelopt.torch.nas。

该模块提供了一个简化的API用于剪枝，该API基于NAS基础设施，但简化了整个工作流程以适应剪枝算法的简单性质。