Imported from https://github.com/dev-chauhan/PQG-pytorch