Build GAN with PyTorch and Amazon SageMaker