Cleanup TorchVision related functions (#18790)

2025-01-21 18:41:05 +08:00 · 2025-01-21 18:41:05 +08:00 · 5306a8cc1f
commit 5306a8cc1f
parent 066c5443f5
4 changed files with 8 additions and 13 deletions
--- a/ultralytics/nn/modules/conv.py
+++ b/ultralytics/nn/modules/conv.py
@ -336,7 +336,7 @@ class Concat(nn.Module):
 class Index(nn.Module):
    """Returns a particular index of the input."""

-    def __init__(self, c1, c2, index=0):
+    def __init__(self, index=0):
        """Returns a particular index of the input."""
        super().__init__()
        self.index = index