Imported from https://github.com/gcwl/pytorch-pix2pix