Machine-Learning-Collection icon indicating copy to clipboard operation
Machine-Learning-Collection copied to clipboard

Weight Init Update DCGANs PyTorch

Open Priyanshu-hawk opened this issue 2 years ago • 0 comments

Hello Sir, My name is Priyanshu Patel. I was learning GANs From your yt videos, when i was going through the code of DCGANs implementation i saw that you comment out BatchNorm2d, It works fine with MNIST handWritten dataset, but for celeb_dataset Discriminator converges to 0. To fix this issue i made few changes 'initialize_weights' function, this fix will prevent discriminator form convergence.

Thankyou [email protected]

Priyanshu-hawk avatar Mar 29 '23 15:03 Priyanshu-hawk