layers

Submodules for the layers.

mfai.pytorch.models.gan_dgmr.layers.get_conv_layer(conv_type='standard')

Return a conv layer based on the passed in string name.

Return type:

Union[type[Conv2d], type[CoordConv], type[Conv3d]]

Parameters:

conv_type (Literal['standard', 'coord', '3d'])

attention	Attention module.
conv_gru	Implementation of Conv GRU and cell module.
coord_conv	The coordinate convolution class.