Imported from https://github.com/One-sixth/gan-qp-mod-pytorch