ResBlockConv

class aixd.mlmodel.architecture.blocks.ResBlockConv(*args: Any, **kwargs: Any)

Bases: Module

Module implementing a convolutional residual block.

Parameters:

• dim (int) – The number of dimensions of the input data (1, 2, or 3).

• in_channels (int) – The number of input channels.

• out_channels (int) – The number of output channels.

• scaling (str) – The scaling mode to use. One of ‘down’ (reduce dimensionality), ‘up’ (increase dimensionality), or None (no change in dimensionality).

• activation (Union[nn.Module, str]) – The activation function to use.

• up_padding (Tuple[int]) – The padding to use for upsampling. If shape was uneven when encoding (downsampling), tensor must also be cropped to uneven when decoding (upsampling).

Methods

forward