Imported from https://github.com/chwneo7/VAE-code