tiny_diff.models.vae.model

Classes

ConvVAE([encoder_cls, decoder_cls])	VAE model that uses convolutional layers.
VAE([vae_loss, encoder_cls, decoder_cls, ...])	Abstract class for VAE models.