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

Classes

Downsample2D

A downsampling layer with an optional convolution.

ResnetBlock2D

A Resnet block.

Upsample2D

An upsampling layer with an optional convolution.

previous

modelzoo.vision.pytorch.dit.layers.vae.DownEncoderBlock2D.DownEncoderBlock2D

next

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