update README

This commit is contained in:
D-X-Y
2019-10-04 23:32:19 +10:00
parent f8f3f382e0
commit 1dee8c014a
4 changed files with 8 additions and 62 deletions

View File

@@ -1,6 +1,6 @@
# Nueral Architecture Search
This project contains the following neural architecture search algorithms, implemented in [PyTorch](http://pytorch.org).
This project contains the following neural architecture search algorithms, implemented in [PyTorch](http://pytorch.org). More NAS resources can be found in [Awesome-NAS](https://github.com/D-X-Y/Awesome-NAS).
- Network Pruning via Transformable Architecture Search, NeurIPS 2019
- One-Shot Neural Architecture Search via Self-Evaluated Template Network, ICCV 2019
@@ -20,6 +20,7 @@ In this paper, we proposed a differentiable searching strategy for transformable
<img src="https://d-x-y.github.com/resources/paper-icon/NIPS-2019-TAS.png" width="700">
### Usage
Use `bash ./scripts/prepare.sh` to prepare data splits for `CIFAR-10`, `CIFARR-100`, and `ILSVRC2012`.
@@ -50,6 +51,7 @@ Highlight: we equip one-shot NAS with an architecture sampler and train network
<img src="https://d-x-y.github.com/resources/paper-icon/ICCV-2019-SETN.png" width="450">
### Usage
Please use the following scripts to train the searched SETN-searched CNN on CIFAR-10, CIFAR-100, and ImageNet.
```
CUDA_VISIBLE_DEVICES=0 bash ./scripts/nas-infer-train.sh cifar10 SETN 96 -1
@@ -81,6 +83,7 @@ Searching codes come soon!
# Citation
If you find that this project helps your research, please consider citing some of the following papers:
```
@inproceedings{dong2019tas,