modelzoo.vision.pytorch.dit.layers.vae.DownEncoderBlock2D.DownEncoderBlock2D#
- class modelzoo.vision.pytorch.dit.layers.vae.DownEncoderBlock2D.DownEncoderBlock2D[source]#
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_downsample=True, downsample_padding=1)[source]#
- static __new__(cls, *args: Any, **kwargs: Any) Any #