Imported from https://github.com/richardrl/gan-pytorch