Fix init_cfg in resnet backbone (#697)
parent
e2ce2e2540
commit
e610ed189c
|
@ -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')
|
||||
|
||||
|
|
Loading…
Reference in New Issue