摘要:
pytorch 实现剪枝的思路是 生成一个掩码,然后同时保存 原参数、mask、新参数,如下图 pytorch 剪枝分为 局部剪枝、全局剪枝、自定义剪枝; 局部剪枝 是对 模型内 的部分模块 的 部分参数 进行剪枝,全局剪枝是对 整个模型进行剪枝; 本文旨在记录 pytorch 剪枝模块的用法,首先 阅读全文
posted @ 2022-08-16 19:11
努力的孔子
阅读(1585)
评论(0)
推荐(1)
浙公网安备 33010602011771号