modelzoo.vision.pytorch.dit.layers.vae.ResNetBlock2D.Upsample2D#

class modelzoo.vision.pytorch.dit.layers.vae.ResNetBlock2D.Upsample2D[source]#

Bases: torch.nn.Module

An upsampling layer with an optional convolution.

Parameters

  • channels – channels in the inputs and outputs.

  • use_conv – a bool determining if a convolution is applied.

  • use_conv_transpose –

  • out_channels –

Methods

forward

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

Call self as a function.

__init__(channels, use_conv=False, use_conv_transpose=False, out_channels=None, name='conv')[source]#
static __new__(cls, *args: Any, **kwargs: Any) → Any#

previous

modelzoo.vision.pytorch.dit.layers.vae.ResNetBlock2D.ResnetBlock2D

next

modelzoo.vision.pytorch.dit.layers.vae.UNetMidBlock2D