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