mmcv.cnn.build_norm_layer

mmcv.cnn.build_norm_layer(cfg: Dict, num_features: int, postfix: Union[int, str] = '') → Tuple[str, torch.nn.modules.module.Module]

Build normalization layer.

参数

• cfg (dict) –

  The norm layer config, which should contain:

  • type (str): Layer type.

  • layer args: Args needed to instantiate a norm layer.

  • requires_grad (bool, optional): Whether stop gradient updates.

• num_features (int) – Number of input channels.

• postfix (int | str) – The postfix to be appended into norm abbreviation to create named layer.

返回

The first element is the layer name consisting of abbreviation and postfix, e.g., bn1, gn. The second element is the created norm layer.

返回类型

tuple[str, nn.Module]