pytorch模型剪枝学习笔记

pytorch代码仓库 pytorch在19年11月份的时候合入了这部分剪枝的代码。pytorch提供一些直接可用的api，用户只需要传入需要剪枝的module实例和需要剪枝的参数名字，系统自动帮助完成剪枝操作，看起来接口挺简单。比如 def random_structured(module, name, amount, dim) pytorch支持的几种类型的剪枝策略： 详细分析 pytorch

>>阅读原文<<