Imported from https://github.com/charlescheng0117/pytorch-WGAN-GP-TTUR-CelebA