Shortcuts

mmengine.utils.dl_utils.is_norm

mmengine.utils.dl_utils.is_norm(layer, exclude=None)[source]

Check if a layer is a normalization layer.

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

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

Returns:

Whether the layer is a norm layer.

Return type:

bool