Skip to content

Commit efcd364

Browse files
authored
[Fix] Fix load_model_state_dict in BaseStrategy (#1447)
1 parent 504fa4f commit efcd364

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

mmengine/_strategy/base.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -799,7 +799,8 @@ def load_model_state_dict(
799799
else:
800800
model = self.model
801801

802-
_load_checkpoint_to_model(model, state_dict, strict, revise_keys)
802+
_load_checkpoint_to_model(
803+
model, state_dict, strict=strict, revise_keys=revise_keys)
803804

804805
def load_optim_state_dict(self, state_dict: dict) -> None:
805806
"""Load optimizer state from dict."""

0 commit comments

Comments
 (0)