eriklindernoren / PyTorch-GAN

PyTorch implementations of Generative Adversarial Networks.
MIT License
16.17k stars 4.05k forks source link

Query: DCGAN Saving Weights AND Parallelization #189

Open KomputerMaster64 opened 1 year ago

KomputerMaster64 commented 1 year 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