[Refactoring] Revise function of is_init flag in BaseModule (#898)

open-mmlab · Mar 18, 2021 · 00870b9 · 00870b9
1 parent 967d9d5
commit 00870b9
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')