yo

mosaicml · Dec 3, 2024 · fad0230 · fad0230
1 parent 790e25e
commit fad0230
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 9 deletions.
diff --git a/composer/callbacks/checkpoint_saver.py b/composer/callbacks/checkpoint_saver.py
@@ -125,9 +125,7 @@ class CheckpointSaver(Callback):  # noqa: D101
             *   The current epoch count is ``1``.
             *   The current batch count is ``42``.
 
-            The rank zero process will save the checkpoint to
-            ``"awesome-training-run/checkpoints/ep1-ba42-rank0"``.
-                ...
+            The rank zero process will save the checkpoint to ``"awesome-training-run/checkpoints/ep1-ba42-rank0"``.
 
         remote_file_name (str, optional): Format string for the checkpoint's remote file name.
             Default: ``"{{run_name}}/checkpoints/ep{{epoch}}-ba{{batch}}-rank{{rank}}"``.
@@ -160,11 +158,9 @@ class CheckpointSaver(Callback):  # noqa: D101
             *   The current epoch count is ``1``.
             *   The current batch count is ``42``.
 
-            The rank zero process will save the checkpoint to
-            ``'awesome-training-run/checkpoints/ep1-ba42-rank0'``,
-            and a symlink will be created at
-            ``'awesome-training-run/checkpoints/latest-rank0' -> 'awesome-training-run/checkpoints/ep1-ba42-rank0'``
-                ...
+            The rank zero process will save the checkpoint to ``'awesome-training-run/checkpoints/ep1-ba42-rank0'``,
+            and a symlink will be created at ``'awesome-training-run/checkpoints/latest-rank0' ->
+            'awesome-training-run/checkpoints/ep1-ba42-rank0'``
 
             Corresponding symlinks will be created at::
 

diff --git a/composer/utils/checkpoint.py b/composer/utils/checkpoint.py
@@ -1227,7 +1227,6 @@ def save_checkpoint(
         *   The current batch count is ``42``.
 
         The rank zero process will save the checkpoint to ``'ep1-ba42-rank0'``.
-            ...
 
     weights_only (bool, optional): If ``True``, save only the model weights instead of the entire training state.
         (default: ``False``)