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 init_cfg in resnet backbone (#697)

Browse Source
This commit is contained in:
Ivan Rubachev 2021-07-15 06:45:14 +03:00 committed by GitHub
parent 00defd6b7f
commit 55085a85c3
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/backbones/resnet.py
View File

@ -400,7 +400,7 @@ class ResNet(BaseModule):
zero_init_residual=True,
pretrained=None,
init_cfg=None):
super(ResNet, self).__init__()
super(ResNet, self).__init__(init_cfg)
if depth not in self.arch_settings:
raise KeyError(f'invalid depth {depth} for resnet')