Query: DCGAN Saving Weights AND Parallelization

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