Dubhe/dubhe-tadl/enas/README.md

• Efficient Neural Architecture Search (ENAS)
• • 1. Requirements
  • 2.Train
  • • Stage1: search an architecture
    • Stage2: select (deprecated)
    • stage3: retrain

Efficient Neural Architecture Search (ENAS)

1. Requirements

torch
torchvision
collections
argparser
pickle
pytest-shutil

2.Train

Stage1: search an architecture

• macro search

python trainer.py --trial_id=0 --search_for macro  --best_selected_space_path='./macro_selected_space.json' --result_path='./macro_result.json'

• micro search

python trainer.py --trial_id=0 --search_for micro  --best_selected_space_path='./micro_selected_space.json' --result_path='./micro_result.json'

Stage2: select (deprecated)

python selector.py

stage3: retrain

• macro search

python retrainer.py --search_for macro  --best_checkpoint_dir='./macro_checkpoint.pth' --best_selected_space_path=
'./macro_selected_space.json'  --result_path='./macro_result.json'

• micro search

python retrainer.py --search_for micro  --best_checkpoint_dir='./micro_checkpoint.pth' --best_selected_space_path=
'./micro_selected_space.json'  --result_path='./micro_result.json'