This repository is an experimental branch checked out from XNAS and has been deprecated.
For future updates, please check out: https://github.com/MAC-AutoML/XNAS.git
XNAS is an effective neural architecture search codebase, written in PyTorch.
git clone https://github.com/MAC-AutoML/XNAS.git
cd XNAS# set root path
export PYTHONPATH=$PYTHONPATH:/Path/to/XNAS
# set gpu devices
export CUDA_VISIBLE_DEVICES=0
# unit test example
python tools/test_func/sng_function_optimization.py
# train example
python train/DARTS_train.py --cfg configs/search/darts.yaml
# replace config example
python train/DARTS_train.py --cfg configs/search/darts.yaml OUT_DIR /username/project/XNAS/experiment/darts/test1