Multi GPU setup

[aaron851113]

Thanks for your code !

I got an error with running original code : Traceback (most recent call last): File "train_search.py", line 203, in <module> main() File "train_search.py", line 82, in main arch_params = list(map(id, model.module.arch_parameters())) File "/opt/conda/envs/darts-gpus/lib/python3.5/site-packages/torch/nn/modules/module.py", line 535, in __getattr__ type(self).__name__, name)) AttributeError: 'Network' object has no attribute 'module'

so that I modify train_search.py line 64 as following , and it could run . gpus = [int(i) for i in args.gpu.split(',')] as gpus = [0,1,2,3] (I have 4 V100s)