fix type hints in neck (#76)

* a little change * a little change
2022-09-23 15:06:37 +08:00 · 2022-09-23 15:06:37 +08:00 · 5eb6094565
parent 59f3d30fd2
commit 5eb6094565
1 changed files with 1 additions and 1 deletions
--- a/mmyolo/models/necks/base_yolo_neck.py
+++ b/mmyolo/models/necks/base_yolo_neck.py
@ -117,7 +117,7 @@ class BaseYOLONeck(BaseModule, metaclass=ABCMeta):
        if self.freeze_all:
            self._freeze_all()

-    def forward(self, inputs: torch.Tensor) -> tuple:
+    def forward(self, inputs: List[torch.Tensor]) -> tuple:
        """Forward function."""
        assert len(inputs) == len(self.in_channels)
        # reduce layers