How to trian with multiple gpu?

znxlwm / pytorch-generative-model-collections

Collection of generative models in Pytorch version.

2.6k stars 548 forks source link

How to trian with multiple gpu? #29

Open greenairy opened 3 years ago

greenairy commented 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.