Training a GAN model