Liu, Hanxiao, Karen Simonyan, and Yiming Yang. "Darts: Differentiable architecture search." arXiv preprint arXiv:1806.09055 (2018). [arxiv]
- python 3
- pytorch >= 0.4
- graphviz
- First install using
apt install
and thenpip install
. - or conda install may make it work.
- First install using
- numpy
- tensorboardX
Dataset | Final validation acc | Best validation acc |
---|---|---|
MNIST | 99.75% | 99.80% |
Fashion-MNIST | 99.20% | 99.31% |
CIFAR-10 | 97.17% | 97.23% |
97.17%, final validation accuracy in CIFAR-10, is the same number as the paper.
Search-training phase of Fashion-MNIST
Augment-validation phase of CIFAR-10 and Fashion-MNIST
https://github.com/quark0/darts (official implementation)
- Supporting pytorch >= 0.4
- Code that is easy to read and commented.
- Implemenation of architect
- Original implementation is very slow in pytorch >= 0.4.
- Various dataset
- Tensorboard
- No RNN
and so on.