aladdinpersson / Machine-Learning-Collection

A resource for learning about Machine learning & Deep Learning
https://www.youtube.com/c/AladdinPersson
MIT License
7.69k stars 2.7k forks source link

Weight Init Update DCGANs PyTorch #149

Open Priyanshu-hawk opened 1 year ago

Priyanshu-hawk commented 1 year ago

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 priyanshupatel.hawk@gmail.com