Zero Grad and Generator error missing in GAN

Adversarial-Deep-Learning / code-soup

This is a collection of algorithms and approaches used in the book adversarial deep learning

MIT License

18 stars 18 forks source link

Closed abhi-glitchhg closed 2 years ago

abhi-glitchhg commented 2 years ago

add self.discriminator.zero_grad() also return an additional argument errG from step method of GAN class

someshsingh22 commented 2 years ago

Thanks @abhi-glitchhg you can make the necessary changes in the GAN implementation