Skip to content

Commit ea29a1e

Browse files
authored
[Fix]: add yolof init_cfg (#8243)
1 parent 4b02faf commit ea29a1e

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

mmdet/models/detectors/yolof.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@ def __init__(self,
1414
bbox_head,
1515
train_cfg=None,
1616
test_cfg=None,
17-
pretrained=None):
17+
pretrained=None,
18+
init_cfg=None):
1819
super(YOLOF, self).__init__(backbone, neck, bbox_head, train_cfg,
19-
test_cfg, pretrained)
20+
test_cfg, pretrained, init_cfg)

0 commit comments

Comments
 (0)