EVE icon indicating copy to clipboard operation
EVE copied to clipboard

Published 20 hours ago • verified publisher icon swook

train with multiple gpus

Open chaeyoun opened this issue 3 years ago • 1 comments

Did anyone train refinenet with multiple GPUs? I tried to use Dataparallel but seems like it doesn't work.

chaeyoun avatar Apr 06 '21 01:04 chaeyoun

Thanks for opening this issue. I myself did not previously have the chance to train EVE on multiple GPUs... so am unaware of which parts make it incompatible for nn.DataParallel. I am very much open to contributions in this direction, so anyone is very welcome to open a pull request for this purpose.

Thanks again.

swook avatar Apr 06 '21 01:04 swook