Pruning of image classification model - sensitivity¶
该教程以图像分类模型MobileNetV1为例,说明如何快速使用PaddleSlim的敏感度分析接口。 该示例包含以下步骤:
- 导入依赖
- 构建模型
- 定义输入数据
- 定义模型评估方法
- 训练模型
- 获取待分析卷积参数名称
- 分析敏感度
- 剪裁模型
以下章节依次次介绍每个步骤的内容。
1. 导入依赖¶
PaddleSlim依赖Paddle1.7版本,请确认已正确安装Paddle,然后按以下方式导入Paddle和PaddleSlim:
import paddle
import paddle.fluid as fluid
import paddleslim as slim