mirrors/mmsegmentation
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmsegmentation.git synced 2025-06-03 22:03:48 +08:00
Code Issues Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Wencheng Wu 2022-08-30 20:43:10 +08:00 committed by GitHub
parent acff83909f
commit 74e13cfa61
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mmseg/models/decode_heads/decode_head.py
View File

@ -200,7 +200,7 @@ class BaseDecodeHead(BaseModule, metaclass=ABCMeta):
Returns: Returns:
dict[str, Tensor]: a dictionary of loss components 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) losses = self.losses(seg_logits, gt_semantic_seg)
return losses return losses