hubertrybka / vae-annealing

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

Related projects

Alternatives and complementary repositories for vae-annealing