做分销网站,单页网站在线制作,设计开发输入清单,织梦者网站模板之后的两篇文章是对前段时间工作的一个总结。 一、环境配置 1.1、文章以b导的代码为模板，环境配置比较简单（第二篇蒸馏篇结束后会放置剪枝蒸馏配置好的百度网盘链接），其他算法自行配置，在剪枝之前，需要保证算法能够在本地跑通。 B导链接： https://github.com/bubbliiiin…之后的两篇文章是对前段时间工作的一个总结。 一、环境配置 1.1、文章以b导的代码为模板，环境配置比较简单（第二篇蒸馏篇结束后会放置剪枝蒸馏配置好的百度网盘链接），其他算法自行配置，在剪枝之前，需要保证算法能够在本地跑通。 B导链接： https://github.com/bubbliiiing/deeplabv3-plus-pytorch 1.2、安装torch_pruning 在利用剪枝功能前，需要安装一下剪枝的库。需要安装0.2.7版本，剪枝时的一些log信息会自动保存在logs文件夹下。
pip install torch_pruning==0.2.7二、剪枝过程 2.1、读取模型结构 此处分两种情况，第一种为torch.save(model,…)保存的权重，第二种为torch.save(model.state_dict(),…)保存的权重，如下适用于第二种情况下，第一种情况本身包含模型结构信息。
def
