Imported from https://github.com/9310gaurav/ali-pytorch