Training GAN models You can find the code to train 2D model from WGAN.py and 3D model from train_new_tensor.py