modelzoo.vision.pytorch.dit.layers.vae.VAEModel#

Classes

AutoencoderKL

Variational Autoencoder (VAE) model with KL loss from the paper Auto-Encoding Variational Bayes by Diederik P.

AutoencoderKLOutput

Output of AutoencoderKL encoding method.

DecoderOutput

Output of decoding method.