D-X-Y / AutoDL-Projects

Automated deep learning algorithms implemented in PyTorch.
MIT License
1.57k stars 284 forks source link

Reproduce NAS-Bench-201 REA algo #108

Closed JackFram closed 3 years ago

JackFram commented 3 years ago

Appreciated for your great work!

Which Algorithm REA for NAS-Bench-201

Describe the Question when running bash ./scripts-search/NAS-Bench-201-algos/BOHB.sh cifar10 -1 the error

Traceback (most recent call last):
  File "./exps/NAS-Bench-201-algos/reinforce.py", line 290, in <module>
    save_dir, index = main(args, nas_bench)
  File "./exps/NAS-Bench-201-algos/reinforce.py", line 187, in main
    reward, cost_time = train_and_eval(arch, nas_bench, extra_info, dataname)
  File "/home/ubuntu/zzh/XAutoDL/exps/NAS-Bench-201-algos/R_EA.py", line 111, in train_and_eval
    raise ValueError("NOT IMPLEMENT YET")
ValueError: NOT IMPLEMENT YET

occurred, any workaround?

D-X-Y commented 3 years ago

It is because that you did not download the benchmark file. See here, you need to download NAS-Bench-201-v1_1-096897.pth and put it into ${TORCH_HOME}.

In addition, the NAS-Bench-201 has been upgraded to NATS-Bench, and it is suggested to use the BOHB script in NATS-Bench: https://github.com/D-X-Y/AutoDL-Projects/blob/main/exps/NATS-algos/bohb.py#L8 In short, three key difference for your target experiment are:

JackFram commented 3 years ago

Cool, my mistake, I ran REA instead of BOHB and there is a line in the code indicating not implemented yet. Anyway, during the traceback, it seems still to be a nas bench API location problem. Appreciated!

D-X-Y commented 3 years ago

No problem~ I should print more informative debug info.