add comment for wholemodule

open-mmlab · Mar 24, 2021 · d4eef56 · d4eef56
1 parent 0254ee2
commit d4eef56
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/mmcv/cnn/utils/weight_init.py b/mmcv/cnn/utils/weight_init.py
@@ -344,6 +344,9 @@ def __call__(self, module):
 
 def _initialize(module, cfg, wholemodule=False):
     func = build_from_cfg(cfg, INITIALIZERS)
+    # wholemodule flag is for override mode, there is no layer key in override
+    # and initializer will give init values for the whole module with the name
+    # in override.
     func.wholemodule = wholemodule
     func(module)