Skip to content

Commit

Permalink
update checkpoint manager docstring (#866)
Browse files Browse the repository at this point in the history
Summary:
Pull Request resolved: #866

fixes docstring reference to updated function name

Reviewed By: diego-urgell

Differential Revision: D59928251

fbshipit-source-id: 69d3f6a98c40b09333f021fd56ef7c21bf606701
  • Loading branch information
JKSenthil authored and facebook-github-bot committed Jul 18, 2024
1 parent f42e302 commit 3a0bf42
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torchtnt/utils/checkpoint.py
Original file line number Diff line number Diff line change
Expand Up @@ -473,7 +473,7 @@ def should_save_checkpoint(self, checkpoint: CheckpointPath) -> bool:
def append_checkpoint(self, ckpt: CheckpointPath) -> None:
"""
This will update the internal state to keep track of the checkpoint. This function should only be called
when a checkpoint whose path was returned from `maybe_get_next_checkpoint_path` was successfully created.
when a checkpoint whose path was returned from `generate_checkpoint_path` was successfully created.
If a previous checkpoint should be removed to honor `keep_last_n_checkpoint`, it will be deleted on rank 0.
Args:
Expand Down

0 comments on commit 3a0bf42

Please sign in to comment.