upBlockNoSkip

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

Bases: Module

Spatial upsampling and then convBlock

Parameters:

• in_channels (int) – number of input channels

• out_channels (int) – number of output channels

forward(x1)

Forward pass of the upBlock.

Parameters:

x1 (torch.Tensor) – Input tensor

Returns:

Output tensor

Return type:

torch.Tensor