search

vlievin / biva-pytorch

Official PyTorch BIVA implementation (BIVA: A Very Deep Hierarchy of Latent Variables for Generative Modeling)
MIT License
81 stars 11 forks source link

Use correct stages when creating a VAE or BIVA model. #1

Closed martinengelcke closed 4 years ago

martinengelcke commented 4 years ago

It looks like the stages for creating a VAE or a BIVA model have been accidentally mixed up in biva/model/deepvae.py.

vlievin commented 4 years ago

Thanks for the fix!