Imported from https://github.com/AlexHex7/SimGAN_pytorch