From 1c38a90c371a47e09c1123ff7acad20e5de6a68b Mon Sep 17 00:00:00 2001 From: D-X-Y <280835372@qq.com> Date: Wed, 17 Mar 2021 09:33:40 +0000 Subject: [PATCH] Update README --- exps/NATS-algos/README.md | 21 +++++++++++++++++++++ exps/algos/README.md | 2 +- 2 files changed, 22 insertions(+), 1 deletion(-) create mode 100644 exps/NATS-algos/README.md diff --git a/exps/NATS-algos/README.md b/exps/NATS-algos/README.md new file mode 100644 index 0000000..3c367df --- /dev/null +++ b/exps/NATS-algos/README.md @@ -0,0 +1,21 @@ +# NAS Algorithms evaluated in [NATS-Bench](https://arxiv.org/abs/2009.00437) + +The Python files in this folder are used to re-produce the results in ``NATS-Bench: Benchmarking NAS Algorithms for Architecture Topology and Size''. + +- `[search-size.py](https://github.com/D-X-Y/AutoDL-Projects/blob/main/exps/NATS-algos/search-size.py)` contains codes for weight-sharing-based search on the size search space. +- `[search-cell.py](https://github.com/D-X-Y/AutoDL-Projects/blob/main/exps/NATS-algos/search-cell.py)` contains codes for weight-sharing-based search on the topology search space. +- `[bohb.py](https://github.com/D-X-Y/AutoDL-Projects/blob/main/exps/NATS-algos/bohb.py)` contains the BOHB algorithm for both size and topology search spaces. + +## Citation + +If you find that this project helps your research, please consider citing the related paper: +``` +@article{dong2021nats, + title = {{NATS-Bench}: Benchmarking NAS Algorithms for Architecture Topology and Size}, + author = {Dong, Xuanyi and Liu, Lu and Musial, Katarzyna and Gabrys, Bogdan}, + doi = {10.1109/TPAMI.2021.3054824}, + journal = {IEEE Transactions on Pattern Analysis and Machine Intelligence (TPAMI)}, + year = {2021}, + note = {\mbox{doi}:\url{10.1109/TPAMI.2021.3054824}} +} +``` diff --git a/exps/algos/README.md b/exps/algos/README.md index e929b83..91121ac 100644 --- a/exps/algos/README.md +++ b/exps/algos/README.md @@ -2,4 +2,4 @@ The Python files in this folder are used to re-produce the results in our NAS-Bench-201 paper. -We will upgrade the codes to be more general and extendable. The new codes are at [coming soon]. +We have upgraded the codes to be more general and extendable at [NATS-algos](https://github.com/D-X-Y/AutoDL-Projects/tree/main/exps/NATS-algos).