Imported from https://github.com/Mxbonn/INQ-pytorch