重庆网站建设有限公司,二手商品网站制作,今天国内重大新闻事件,闽清住房和城乡建设局网站之后的两篇文章是对前段时间工作的一个总结。 一、环境配置 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