轻松实现结构化剪枝DepGraph(2)

文章目录

    • 1. 内容概要
    • 2. 权值剪枝器(MagnitudePruner)
      • 2.1 算法介绍
      • 2.2 基于Torch-Pruning实现MagnitudePruner
    • 3. Slimming剪枝器
      • 3.1 算法概要
      • 2.2 基于Torch-Pruning实现SlimmingPruner
    • 4. 总结

1. 内容概要

本文介绍如何基于Torch-Pruning库轻松实现两种经典剪枝算法。Torch-Pruning项目内置了多种剪枝器(High-level Pruner),可用于实现模型的整体结构化剪枝,从而提高网络的推理速度,并降低内存消耗。本文将着重介绍两种经典剪枝器,即MagnitudePruner

你可能感兴趣的:(模型轻量化,算法,剪枝,机器学习)