From 74e13cfa61050a5a0752fb79ffc9db9dc927bd8a Mon Sep 17 00:00:00 2001 From: Wencheng Wu <41542251+274869388@users.noreply.github.com> Date: Tue, 30 Aug 2022 20:43:10 +0800 Subject: [PATCH] [Fix] Fix decode head forward_train error. (#1997) --- mmseg/models/decode_heads/decode_head.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmseg/models/decode_heads/decode_head.py b/mmseg/models/decode_heads/decode_head.py index d08b1d0b6..8ffc683bf 100644 --- 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