out_arch_branch

out_arch_branch(in_ch, num_output_channels=2, activation='softmax')[source]

Group5 branch for MicroNet.

This branch defines architecture for auxiliary and the main output.

Parameters:

  • in_ch (int) – Number of input channels.

  • num_output_channels (int) – Number of output channels. default=2.

  • activation (str) – Activation function, default=”softmax”.

Returns:

An output of type torch.nn.Sequential

Return type:

torch.nn.Sequential