2022年11月20日
pytorch中的剪枝操作
摘要： 深度学习技术依赖于过参数化模型，这是不利于部署的，相反，生物神经网络是使用高效的稀疏连接的。 通过减少模型中的参数数量来压缩模型的技术非常重要，为减少内存、电池和硬件的消耗，而牺牲准确性，实现在设备上部署轻量级模型。 在Pytorch中，主要通过torch.nn.utils.prune来进行剪枝，以 阅读全文
posted @ 2022-11-20 22:40 牛犁heart 阅读(1950) 评论(0) 推荐(0) 编辑