From fad02309ea4f33d77d1a9efbb155bbb3a3d490d7 Mon Sep 17 00:00:00 2001 From: Saaketh Date: Tue, 3 Dec 2024 11:46:09 -0500 Subject: [PATCH] yo --- composer/callbacks/checkpoint_saver.py | 12 ++++-------- composer/utils/checkpoint.py | 1 - 2 files changed, 4 insertions(+), 9 deletions(-) diff --git a/composer/callbacks/checkpoint_saver.py b/composer/callbacks/checkpoint_saver.py index 813dd2f9b3..1ef55d5dbc 100644 --- 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 index a724b2d351..95f4b47614 100644 --- 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``)