# NasNet 2018
Most of the code in this repo is from https://github.com/MarSaKi/nasnet
This code is the reimplementation of "Learning Transferable Architectures for Scalable Image Recognition",
including the training process of controller. This code contains three algorithms to search model, Random Search,
Policy Gradient and PPO.
## Requirements
```
Python >= 3.6.7, PyTorch == 0.4.0
```
## Architecture Search
```
python train_search.py --cutout --algorithm RS #use random search
python train_search.py --cutout --algorithm PG #use policy gradient
python train_search.py --cutout --algorithm PPO #use PPO
```
Note the validation performance in this step does not indicate the final performance of the architecture. One must train the obtained genotype/architecture from scratch using full-sized models.
Also the default setting is training with 20 processes and 3 GPU. Change the processes to 10:
```
python train_search.py --cutout --episodes 10
```
or modify code in random_search.py, policy_gradient.py and PPO.py .
## Architecture Evaluation
Because of the limitation of time and computation resource, I didn't train the candidate genotypes/architectures from scratch.
## Results
```
python draw.py
```
<p align="center">
<img src="fig/search.png" alt="search_process" width="75%">
</p>
We can see RL search is better than random search, also PPO is more stable and faster.
没有合适的资源?快使用搜索试试~ 我知道了~
nasnet_cifar10:带有EarlyStopping的CIFAR10修改的NASNet
共16个文件
py:11个
md:1个
png:1个
5星 · 超过95%的资源 需积分: 9 1 下载量 136 浏览量
2021-04-03
21:18:58
上传
评论
收藏 85KB ZIP 举报
温馨提示
NasNet 2018年 此仓库中的大多数代码来自 该代码是“学习可伸缩图像识别的可传输体系结构”的重新实现,包括控制器的训练过程。 该代码包含三种用于搜索模型的算法:随机搜索,策略梯度和PPO。 要求 Python >= 3.6.7, PyTorch == 0.4.0 建筑搜索 python train_search.py --cutout --algorithm RS #use random search python train_search.py --cutout --algorithm PG #use policy gradient python train_search.py --cutout --algorithm PPO #use PPO 请注意,此步骤中的验证性能并不表示体系结构的最终性能。 必须使用全尺寸模型从头开始训练获得的基因型/架构。 同样,默认设置是使用2
资源详情
资源评论
资源推荐
收起资源包目录
nasnet_cifar10-main.zip (16个子文件)
nasnet_cifar10-main
operations.py 4KB
utils.py 3KB
genotypes.py 114B
train_search.py 3KB
PPO.py 7KB
policy_gradient.py 4KB
Worker.py 4KB
run.sh 274B
random_search.py 3KB
model.py 4KB
.gitignore 2KB
controller.py 7KB
draw.py 2KB
README.md 1KB
fig
search.png 60KB
search.pdf 18KB
共 16 条
- 1
zhuyurrr
- 粉丝: 29
- 资源: 4714
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
最新资源
- hrnet关键点检测onnx模型
- chromedriver-win64-132.0.6808.0
- chromedriver-win64-132.0.6807.0
- chromedriver-win64-132.0.6806.0
- 带有界面的多线程爬虫软件源码,使用python爬取漫画图片,可采用多线程爬取,可做毕业设计
- chromedriver-win64-132.0.6804.0
- 我电脑的zsh配置,oh my zsh + powerlevel10k + Hack Nerd Font
- pcie xdma xilinx 上位机应用
- chromedriver-win64-132.0.6803.1
- chromedriver-win64-132.0.6803.0
资源上传下载、课程学习等过程中有任何疑问或建议,欢迎提出宝贵意见哦~我们会及时处理!
点击此处反馈
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功
评论1