diffusion_models.models.vae.ResNetDecoderBlock¶

class diffusion_models.models.vae.ResNetDecoderBlock(in_channels, out_channels)[source]¶

Bases: Module

__init__(in_channels, out_channels)[source]¶

Constructor of ResNetDecoderBlock.

Inspired by ResNet18/ResNet34, using strided transpose convolutions.

Methods

__init__(in_channels, out_channels)

Constructor of ResNetDecoderBlock.

forward(x)

rtype:: Union[Tensor, 'batch channels height width'), Tensor, 'batch channels*2 height/2 width/2')]