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 e2ce2e2540
commit e610ed189c
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/backbones/resnet.py
View File

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