Skip to content

Commit

Permalink
docstring
Browse files Browse the repository at this point in the history
  • Loading branch information
awaelchli committed Jun 4, 2024
1 parent cae58a3 commit c205a4b
Showing 1 changed file with 1 addition and 6 deletions.
7 changes: 1 addition & 6 deletions src/lightning/fabric/strategies/model_parallel.py
Original file line number Diff line number Diff line change
Expand Up @@ -275,12 +275,7 @@ def load_checkpoint(
state: Optional[Union[Module, Optimizer, Dict[str, Union[Module, Optimizer, Any]]]] = None,
strict: bool = True,
) -> Dict[str, Any]:
"""Load the contents from a checkpoint and restore the state of the given objects.
Currently does not support loading the optimizer state if the model is distributed but the checkpoint is a full,
non-distributed checkpoint.
"""
"""Load the contents from a checkpoint and restore the state of the given objects."""
if not state:
raise ValueError(
f"Got {type(self).__name__}.load_checkpoint(..., state={state!r}) but a state with at least "
Expand Down

0 comments on commit c205a4b

Please sign in to comment.