group2_arch_branch

tiatoolbox . models . architecture . micronet . group2_arch_branch

group2_arch_branch(in_ch, out_ch)

Group2 branch for MicroNet.

Parameters

• in_ch (int) – Number of input channels.

• out_ch (int) – Number of output channels.

Returns

An output of type torch.nn.ModuleDict

Return type

torch.nn.ModuleDict