add more detail of warning in basemodule

open-mmlab · Mar 18, 2021 · 0a7be2e · 0a7be2e
1 parent 00870b9
commit 0a7be2e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mmcv/runner/base_module.py b/mmcv/runner/base_module.py
@@ -47,8 +47,8 @@ def init_weight(self):
                     module.init_weight()
             self._is_init = True
         else:
-            warnings.warn('This module has bee initialized, \
-                please call initialize(module, init_cfg) to reinitialize it')
+            warnings.warn(f'init_weight of {self.__class__.__name__} has '
+                          f'been called more than once.')
 
     def __repr__(self):
         s = super().__repr__()