Shortcuts

mmengine.utils.dl_utils.is_norm

mmengine.utils.dl_utils.is_norm(layer, exclude=None)[源代码]

Check if a layer is a normalization layer.

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

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

返回:

Whether the layer is a norm layer.

返回类型:

bool