Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
You can not select more than 25 topics
Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
费翔
c2cace2e4c
|
2 years ago | |
|---|---|---|
| configs | 2 years ago | |
| doc | 2 years ago | |
| search | 2 years ago | |
| tools | 2 years ago | |
| train | 2 years ago | |
| xnas | 2 years ago | |
| .gitignore | 2 years ago | |
| LICENSE | 2 years ago | |
| README.md | 2 years ago | |
| environment.yaml | 2 years ago | |
README.md
XNAS
XNAS is an effective neural architecture search codebase, written in PyTorch.
Installation
git clone https://github.com/MAC-AutoML/XNAS.git
cd XNAS
# set root path
export PYTHONPATH=$PYTHONPATH:/Path/to/XNAS
Usage
# [Optional] 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
