group3_arch_branch

tiatoolbox . models . architecture . micronet . group3_arch_branch

group3_arch_branch(in_ch, skip, out_ch)

Group3 branch for MicroNet.

Parameters

• in_ch (int) – Number of input channels.

• skip (int) – Number of channels for the skip connection.

• out_ch (int) – Number of output channels.

Returns

An output of type torch.nn.ModuleDict

Return type

torch.nn.ModuleDict