Skip to content
Multinomial Distribution Learning for Effective Neural Architecture Search
Python Shell
Branch: master
Clone or download
Latest commit ee6c8fe Jun 24, 2019
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
data_providers first commit May 18, 2019
figs update figs May 21, 2019
mobile_config first commit May 18, 2019
models fix bugs and update README May 21, 2019
modules first commit May 18, 2019
utils fix bugs and update README May 21, 2019
.gitignore fix bugs and update README May 21, 2019
README.md Update README.md Jun 24, 2019
run_cpu_imagenet.sh first commit May 18, 2019
run_darts_cifar.sh first commit May 18, 2019
run_darts_imagenet.sh first commit May 18, 2019
run_exp.py fix bugs and update README May 21, 2019
run_gpu_imagenet.sh first commit May 18, 2019
run_manager.py fix bugs and update README May 21, 2019

README.md

Multinomial Distribution Learning for Effective Neural Architecture Search

Here we propose a method to extremely accelerate NAS, without reinforcement learning or gradient, just by sampling architectures from a distribution and comparing these architectures, estimating their relative performance rather than absolute performance, iteratively updating parameters of the distribution while training.

Our paper is available at 《Multinomial Distribution Learning for Effective Neural Architecture Search》

Here we provide our test codes and pretrained models, our code is based on DARTS and ProxylessNAS, pretrained models can be downloaded here

Requirements

  • PyTorch 1.0
  • DALI

Evaluate

You need to modified your path to dataset in data_providers/cifar10.py and data_providers/imagenet.pyconfig.sh is used to prepare your environment, you should write this file by yourself and here we use it to prepare dataset and packages

To evaluate the model in DARTS setting, just run

for cifar10 :

chmod +x run_darts_cifar.sh
./run_darts_cifar.sh

for imagenet :

chmod +x run_darts_imagenet.sh
./run_darts_imagenet.sh

To evaluate the model in Mobile setting, just run

for gpu :

chmod +x run_gpu_imagenet.sh
./run_gpu_imagenet.sh

for cpu :

chmod +x run_cpu_imagenet.sh
./run_cpu_imagenet.sh

Performance

You can’t perform that action at this time.