modelzoo.vision.pytorch.dit.layers.vae.UpDecoderBlock2D.UpDecoderBlock2D

class modelzoo.vision.pytorch.dit.layers.vae.UpDecoderBlock2D.UpDecoderBlock2D

Bases: torch.nn.Module

Methods

forward

__call__(*args: Any, **kwargs: Any) → Any

Call self as a function.

__init__(in_channels: int, out_channels: int, dropout: float = 0.0, num_layers: int = 1, resnet_eps: float = 1e-06, resnet_time_scale_shift: str = 'default', resnet_act_fn: str = 'swish', resnet_groups: int = 32, resnet_pre_norm: bool = True, output_scale_factor=1.0, add_upsample=True)

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

previous

modelzoo.vision.pytorch.dit.layers.vae.UpDecoderBlock2D

next

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