/vae-annealing

A simple pytorch implementation for calculating VAE loss components and annealing KLD loss while training VAEs, especially RNN-based

Primary LanguagePythonMIT LicenseMIT

Watchers