starling.models.blocks

Functions

instance_norm

layer_norm

Classes

DownsampleBlock

LayerNorm

LayerNorm that supports two data formats: channels_last (default) or channels_first.

MinPool2d

ResBlockDecBasic

ResBlockDecBottleneck

ResBlockEncBasic

ResBlockEncBottleneck

ResizeConv2d

UpsampleBlock

vanilla_Decoder

vanilla_Encoder