No module named 'train_class'

naszilla / nas-encodings

Encodings for neural architecture search

Apache License 2.0

29 stars 5 forks source link

Closed LiuJia68 closed 4 years ago

LiuJia68 commented 4 years ago

When search space=='darts', from train_class import Train ModuleNotFoundError: No module named 'train_class'

crwhite14 commented 4 years ago

Thanks for pointing out this mistake. We meant to add to the readme a bit more steps to run experiments on the DARTS search space.

Download our fork of the DARTS repo: https://github.com/naszilla/darts
If you don't put the repo in your home directory, i.e., ~/darts, then you might need to update line 7 of nas-encodings/darts/arch.py and line 8 of nas-encodings/train_arch_runner.py with the correct path.

I will update the readme to explain these steps.