We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 58e6028 commit af23135Copy full SHA for af23135
torchft/manager.py
@@ -534,11 +534,11 @@ def _apply_pending_state_dict(self) -> None:
534
535
self._logger.info("applying pending state dict")
536
537
+ assert self._pending_state_dict is not None, "checkpoint was not staged"
538
assert (
539
self._load_state_dict is not None
540
), "user load_state_dict is not initialized."
541
self._load_state_dict(self._pending_state_dict["user"])
- assert self._pending_state_dict is not None, "checkpoint was not staged"
542
self._pending_state_dict = None
543
self._logger.info("Loaded state dict.")
544
0 commit comments