Zero Grad and Generator error missing in GAN

[abhi-glitchhg]

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

[someshsingh22]

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