mmcv.cnn.is_norm

mmcv.cnn.is_norm(layer: torch.nn.modules.module.Module, exclude: Optional[Union[type, tuple]] = None) → bool

Check if a layer is a normalization layer.

参数

• layer (nn.Module) – The layer to be checked.

• exclude (type | tuple[type]) – Types to be excluded.

返回

Whether the layer is a norm layer.

返回类型

bool