Imported from https://github.com/ja3067/wgan-pytorch