Imported from https://github.com/escuccim/vaegan-pytorch