A surrogate benchmark for neural architecture search
Project description
NAS-Bench-301
This repository containts code for the paper: "NAS-Bench-301 and the Case for Surrogate Benchmarks for Neural Architecture Search".
The surrogate models can be downloaded on figshare. This includes the models for v0.9 and v1.0.
To install all requirements (this may take a few minutes), run
$ cat requirements.txt | xargs -n 1 -L 1 pip install
$ pip install nasbench301
To run a quick example, adapt the model paths in 'nasbench301/example.py' and from the base directory run
$ export PYTHONPATH=$PWD
$ python3 nasbench301/example.py
NOTE: This codebase is still subject to changes. Upcoming updates include improved versions of the surrogate models and code for all experiments from the paper. The API may still be subject to changes.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
nasbench301-0.2.tar.gz
(4.6 kB
view hashes)
Built Distribution
Close
Hashes for nasbench301-0.2-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 27123a6f8b865496ac9d78a7ddd46c8b0965bbcabe02ad88f4163888cf067166 |
|
MD5 | 596e30172ea3b674f9f3fd600fdd74dd |
|
BLAKE2b-256 | 0dabf5d15e2bf7c13f3e59ec2110f575c567d70fdd70ba63b5f66193cf8fc7c5 |