diffusion_models.models.vae.ResNet18Decoder¶

class diffusion_models.models.vae.ResNet18Decoder(hidden_dim=256, out_channels=3)[source]¶

Bases: Module

__init__(hidden_dim=256, out_channels=3)[source]¶

Methods

__init__([hidden_dim, out_channels])

forward(x)

rtype:: Tensor, 'batch channels height width')