eriklindernoren / PyTorch-GAN

PyTorch implementations of Generative Adversarial Networks.
MIT License
16.51k stars 4.09k forks source link

Query: DCGAN Saving Weights AND Parallelization #189

Open KomputerMaster64 opened 2 years ago

KomputerMaster64 commented 2 years ago

Respected sir, Thank you for sharing the implementation of the essentail Generative Adversarial Networks. It would be great if you could share how to modify the code for say the DCGAN implementatino to save .pth files for generating samples for testing. I also wanted to ask you about how to modify the code for multiple GPU training. Regards Prabhav