Open greenairy opened 3 years ago
Hello @znxlwm,
I have noticed that the GAN models are all defined with model.cuda() that use only one device. Is there any way to train them with multiple gpus so that the input size can be increased without decreasing teh batch size? Thank you.
model.cuda()
Hello @znxlwm,
I have noticed that the GAN models are all defined with
model.cuda()
that use only one device. Is there any way to train them with multiple gpus so that the input size can be increased without decreasing teh batch size? Thank you.