walkerning / aw_nas

aw_nas: A Modularized and Extensible NAS Framework
MIT License
246 stars 31 forks source link

Generate sample config error and FBNet support? #50

Open qianjiangcn opened 3 years ago

qianjiangcn commented 3 years ago

Hi, thank you for sharing this great work for NAS!

While running the following command for generating configs: awnas gen-sample-config -r nasbench-101 -d image ./sample_nb101.yaml I get error: Error: Invalid value for "-r" / "--rollout-type": invalid choice: nasbench-101. (choose from discrete, differentiable, mutation, dense_discrete, dense_mutation, ofa, ssd_ofa, compare, general, layer2, layer2-differentiable, macro-stagewise, macro-stagewise-diff, macro-sink-connect-diff, micro-dense, micro-dense-diff)

Seems like nasbench-101 is not supported as the -r argument, and I am wondering what would be the way to generate configs for other NAS approaches like nasbench101 and FBNet? Thank you!

TangChangcheng commented 3 years ago

Hi, I try to run the command: awnas gen-sample-config -r nasbench-101 -d image ./sample_nb101.yaml and it works.

Perhaps you need to check whether there is the warning at the head of the log about failed to import nasbench-101, which means you need to install nasbench-101 on your machine firstly.