Imported from https://github.com/apple2373/PyTorch-SmallGAN