def is_norm(modules):
    """Check if is one of the norms."""
    if isinstance(modules, (GroupNorm, _BatchNorm)):
        return True
    return False
Check if is one of the norms.
tests/test_models/test_backbones/test_conformer.py