composer.models.unet.layers#

composer.models.unet.layers

Functions

get_conv

composer.models.unet.layers.get_conv

get_norm

composer.models.unet.layers.get_norm

get_output_padding

composer.models.unet.layers.get_output_padding

get_padding

composer.models.unet.layers.get_padding

get_transp_conv

composer.models.unet.layers.get_transp_conv

Classes

ConvBlock

composer.models.unet.layers.ConvBlock

ConvLayer

composer.models.unet.layers.ConvLayer

OutputBlock

composer.models.unet.layers.OutputBlock

ResidBlock

composer.models.unet.layers.ResidBlock

UpsampleBlock

composer.models.unet.layers.UpsampleBlock

Attributes

  • convolutions

  • normalizations