mirrors
/
mmcv
mirror of https://github.com/open-mmlab/mmcv.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[Enhancement]: refactor init cfg (#958)

pull/971/head
ZhangShilong 2021-04-18 22:40:13 +08:00 committed by GitHub
parent 5504b5fbcf
commit de4f14e9cd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mmcv/runner/base_module.py
View File

@ -22,8 +22,7 @@ class BaseModule(nn.Module, metaclass=ABCMeta):
# define default value of init_cfg instead of hard code
# in init_weigt() function
self._is_init = False
if init_cfg is not None:
self.init_cfg = init_cfg
self.init_cfg = init_cfg
# Backward compatibility in derived classes
# if pretrained is not None:
@ -40,7 +39,7 @@ class BaseModule(nn.Module, metaclass=ABCMeta):
from ..cnn import initialize
if not self._is_init:
if hasattr(self, 'init_cfg'):
if self.init_cfg:
initialize(self, self.init_cfg)
for m in self.children():
if hasattr(m, 'init_weight'):
@ -52,7 +51,7 @@ class BaseModule(nn.Module, metaclass=ABCMeta):
def __repr__(self):
s = super().__repr__()
if hasattr(self, 'init_cfg'):
if self.init_cfg:
s += f'\ninit_cfg={self.init_cfg}'
return s