做注册任务网站源码律师网站开发

之后的两篇文章是对前段时间工作的一个总结。
一、环境配置
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