Contrastive Neural Architecture Search with Neural Architecture Comparators [pdf]
Pytorch Implementation for "Contrastive Neural Architecture Search with Neural Architecture Comparators".
Please install all the requirements in requirements.txt
.
First, we need to download the architecture-accuracy pairs data.
wget https://github.com/chenyaofo/CTNAS/releases/download/data/nas_bench.json -O ctnas/data/nas_bench.json
Train on NAS-Bench-101
python ctnas/train.py --space nasbench --data ctnas/data/nas_bench.json --train_batch_size 256 --output output/nasbench_search
If you use any part of our code in your research, please cite our paper:
@InProceedings{chen2021contrastive,
title = {Contrastive Neural Architecture Search with Neural Architecture Comparators},
author = {Yaofo Chen and Yong Guo and Qi Chen and Minli Li and Yaowei Wang and Wei Zeng and Mingkui Tan},
booktitle = {The IEEE Conference on Computer Vision and Pattern Recognition},
year = {2021}
}