[Fix] Fix decode head forward_train error. (#1997)

2022-08-30 20:43:10 +08:00 · 2022-08-30 20:43:10 +08:00 · 74e13cfa61
parent acff83909f
commit 74e13cfa61
1 changed files with 1 additions and 1 deletions
--- a/mmseg/models/decode_heads/decode_head.py
+++ b/mmseg/models/decode_heads/decode_head.py
@ -200,7 +200,7 @@ class BaseDecodeHead(BaseModule, metaclass=ABCMeta):
        Returns:
            dict[str, Tensor]: a dictionary of loss components
        """
-        seg_logits = self.forward(inputs)
+        seg_logits = self(inputs)
        losses = self.losses(seg_logits, gt_semantic_seg)
        return losses