search

VDIGPKU / DADA

[ECCV 2020] DADA: Differentiable Automatic Data Augmentation
MIT License
189 stars 29 forks source link

Does it support Data Parallel and multiple GPU? #2

Open bchao1 opened 4 years ago

bchao1 commented 4 years ago

As stated in the problem, I tried using data parallel and commenting out set_device for multi-gpu, but seems to not work.

The script keeps showing RuntimeError: Caught RuntimeError in DataLoader worker process 0. RuntimeError: CUDA error: initialization error

Thanks!

latstars commented 4 years ago

Only the training and inference script support multi-gpu, and the searching script do not support multi-gpu.