[Fix] fix load_model_state_dict in BaseStrategy

open-mmlab · Nov 30, 2023 · b6a4f00 · b6a4f00
1 parent 85c0976
commit b6a4f00
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mmengine/_strategy/base.py b/mmengine/_strategy/base.py
@@ -799,7 +799,8 @@ def load_model_state_dict(
         else:
             model = self.model
 
-        _load_checkpoint_to_model(model, state_dict, strict, revise_keys)
+        _load_checkpoint_to_model(
+            model, state_dict, strict=strict, revise_keys=revise_keys)
 
     def load_optim_state_dict(self, state_dict: dict) -> None:
         """Load optimizer state from dict."""