Training the MNIST cGAN
Learn about training the conditional GAN.
Training loop
The main GAN training loop is also updated to pass a label tensor to the discriminator and generator. The following shows only the code inside the epoch loop.
Access this course and 1400+ top-rated courses and projects.