Open chaeyoun opened 3 years ago
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.
Did anyone train refinenet with multiple GPUs? I tried to use Dataparallel but seems like it doesn't work.