Imported from https://github.com/mcclow12/wgan-gp-pytorch