diff --git a/ultralytics/nn/modules/block.py b/ultralytics/nn/modules/block.py index 53a6e91de..97d140af1 100644 --- a/ultralytics/nn/modules/block.py +++ b/ultralytics/nn/modules/block.py @@ -537,7 +537,6 @@ class BNContrastiveHead(nn.Module): Args: embed_dims (int): Embed dimensions of text and image features. - norm_cfg (dict): Normalization parameters. """ def __init__(self, embed_dims: int):