[Fix] Fix load_model_state_dict in BaseStrategy (#1447)

open-mmlab · Dec 23, 2023 · efcd364 · efcd364
1 parent 504fa4f
commit efcd364
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."""