Fix init_cfg in resnet backbone (#697)

pull/594/head
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

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')