Training the MNIST cGAN

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.