Imported from https://github.com/ANLGBOY/RealNVP-with-PyTorch