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

class modelzoo.vision.pytorch.dit.layers.vae.VAEEncoder.Encoder[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, down_block_types=('DownEncoderBlock2D',), block_out_channels=(64,), layers_per_block=2, norm_num_groups=32, act_fn='silu', double_z=True)[source]#

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

previous

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

next

modelzoo.vision.pytorch.dit.layers.vae.VAEModel