Skip to content

Commit c44a058

Browse files
Fix kwargs parameter missing error in two_stage (open-mmlab#6256)
1 parent 7f9facc commit c44a058

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

Diff for: mmdet/models/detectors/two_stage.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -138,7 +138,8 @@ def forward_train(self,
138138
gt_bboxes,
139139
gt_labels=None,
140140
gt_bboxes_ignore=gt_bboxes_ignore,
141-
proposal_cfg=proposal_cfg)
141+
proposal_cfg=proposal_cfg,
142+
**kwargs)
142143
losses.update(rpn_losses)
143144
else:
144145
proposal_list = proposals

Diff for: mmdet/models/roi_heads/standard_roi_head.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,8 @@ def forward_train(self,
5858
gt_bboxes,
5959
gt_labels,
6060
gt_bboxes_ignore=None,
61-
gt_masks=None):
61+
gt_masks=None,
62+
**kwargs):
6263
"""
6364
Args:
6465
x (list[Tensor]): list of multi-level img features.

0 commit comments

Comments
 (0)