ResBlock3D

class aixd.mlmodel.architecture.blocks.ResBlock3D(*args: Any, **kwargs: Any)[source]

Module implementing a 3D convolutional residual block.

Parameters:

in_channels (int) – The number of input channels.
out_channels (int) – The number of output channels.
scaling (str, optional, default=None) – The scaling mode to use. One of ‘down’ (reduce dimensionality), ‘up’ (increase dimensionality), or None (no change in dimensionality).
activation (Union[nn.Module, str], optional, default=’leaky_relu’) – The activation function to use
up_padding (Tuple[int], optional, default=(2, 2, 2, 2, 2, 2)) – The padding to use for upsampling.

Inherited Methods