README.md 1.1 KB
Newer Older
C
ceci3 已提交
1 2 3 4
# 网络结构搜索示例

本示例介绍如何使用网络结构搜索接口,搜索到一个更小或者精度更高的模型,该文档仅介绍paddleslim中SANAS的使用及如何利用SANAS得到模型结构,完整示例代码请参考sa_nas_mobilenetv2.py或者block_sa_nas_mobilenetv2.py。

C
ceci3 已提交
5 6
## 数据准备
本示例默认使用cifar10数据,cifar10数据会根据调用的paddle接口自动下载,无需额外准备。
C
ceci3 已提交
7

C
ceci3 已提交
8 9
## 接口介绍
请参考<a href='../../docs/docs/api/nas_api.md'>神经网络搜索API文档</a>
C
ceci3 已提交
10

C
ceci3 已提交
11 12 13
本示例为在MobileNetV2的搜索空间上搜索FLOPs更小的模型。
## 1 搜索空间配置
默认搜索空间为`MobileNetV2`,详细的搜索空间配置请参考<a href='../../docs/docs/search_space.md'>搜索空间配置文档</a>
C
ceci3 已提交
14

C
ceci3 已提交
15
## 2 启动训练
C
ceci3 已提交
16

C
ceci3 已提交
17 18 19
### 2.1 启动基于MobileNetV2初始模型结构构造搜索空间的实验
```shell
CUDA_VISIBLE_DEVICES=0 python sa_nas_mobilenetv2.py
C
ceci3 已提交
20 21 22
```


C
ceci3 已提交
23 24 25
### 2.2 启动基于MobileNetV2的block构造搜索空间的实验
```shell
CUDA_VISIBLE_DEVICES=0 python block_sa_nas_mobilenetv2.py
C
ceci3 已提交
26
```