downBlock

class colibri.models.custom_layers.downBlock(in_channels, out_channels)

Bases: Module

Spatial downsampling and then convBlock

Parameters:

• in_channels (int) – number of input channels

• out_channels (int) – number of output channels

Returns:

DownBlock model

Return type:

nn.Module