Merge ad1b264 into 967d9d5

open-mmlab · Mar 18, 2021 · 292cdc5 · 292cdc5
2 parents 967d9d5 + ad1b264
commit 292cdc5
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/mmcv/runner/base_module.py b/mmcv/runner/base_module.py
@@ -40,14 +40,12 @@ def init_weight(self):
         from ..cnn import initialize
 
         if not self._is_init:
-
             if hasattr(self, 'init_cfg'):
                 initialize(self, self.init_cfg)
-                self._is_init = True
             for module in self.children():
                 if 'init_weight' in dir(module):
                     module.init_weight()
-
+            self._is_init = True
         else:
             warnings.warn('This module has bee initialized, \
                 please call initialize(module, init_cfg) to reinitialize it')