diffusion_models.models.vae

Classes

ResNet18Decoder([hidden_dim, out_channels])
ResNet18Encoder(in_channels[, hidden_dim])	Class implementing the ResNet encoder.
ResNetBlock(in_channels, out_channels)	Class implementing the ResNet Basic Building Block, currently limited to usage in ResNet18 and ResNet34.
ResNetDecoderBlock(in_channels, out_channels)
SkipDownSampler(in_channels, out_channels)	Class SkipDownSampler.
SkipUpSampler(*args, **kwargs)
VariationalAutoencoder([in_channels, hidden_dim])	Class implementing a Variational Autoencoder.