[Fis] Set _is_init to True when pretrained is True (#555)

* set _is_init to True when pretrained is True * fix comments
open-mmlab · Oct 10, 2022 · 46add35 · 46add35
1 parent 6245621
commit 46add35
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/mmengine/hub/hub.py b/mmengine/hub/hub.py
@@ -80,4 +80,8 @@ def get_model(cfg_path: str, pretrained: bool = False, **kwargs):
         model = MODELS.build(cfg.model, default_args=kwargs)
         if pretrained:
             load_checkpoint(model, cfg.model_path)
+            # Hack to use pretrained weights.
+            # If we do not set _is_init here, Runner will call
+            # `model.init_weights()` to overwrite the pretrained model.
+            model._is_init = True
         return model