ts to "silu") — The activation function to use.


latent_channels (int, optional, defaults to 4) — Number of channels in the latent space.


sample_size (int, optional, defaults to 32) — TODO



Variational Autoencoder (VAE) model with KL loss from the paper Auto-Encoding Variational Bayes by Diederik P. Kingma
and Max Welling.
This model inherits from ModelMixin. Check the superclass documentation for the generic methods the library
implements