Imported from https://github.com/eriklindernoren/PyTorch-GAN