modelzoo.vision.pytorch.dit.layers.vae.VAEDecoder.Decoder#

class modelzoo.vision.pytorch.dit.layers.vae.VAEDecoder.Decoder[source]#

Bases: torch.nn.Module

Methods

forward

__call__(*args: Any, **kwargs: Any) → Any#: Call self as a function.

__init__(in_channels=3, out_channels=3, up_block_types=('UpDecoderBlock2D',), block_out_channels=(64,), layers_per_block=2, norm_num_groups=32, act_fn='silu')[source]#

static __new__(cls, *args: Any, **kwargs: Any) → Any#

previous

modelzoo.vision.pytorch.dit.layers.vae.VAEDecoder

next

modelzoo.vision.pytorch.dit.layers.vae.VAEEncoder