From 4e8a875b6a4dd2cf87ebd980e59eefb1bc038364 Mon Sep 17 00:00:00 2001 From: Sasha Doubov Date: Tue, 12 Mar 2024 17:39:00 -0700 Subject: [PATCH] Fix typo in monolithic chkpt callback docs (#1024) * Fix typo in monolithic chkpt callback docs * reorder to match function signature --- llmfoundry/callbacks/monolithic_ckpt_callback.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/llmfoundry/callbacks/monolithic_ckpt_callback.py b/llmfoundry/callbacks/monolithic_ckpt_callback.py index a71db9bd43..aaa68763f5 100644 --- a/llmfoundry/callbacks/monolithic_ckpt_callback.py +++ b/llmfoundry/callbacks/monolithic_ckpt_callback.py @@ -21,10 +21,10 @@ class MonolithicCheckpointSaver(Callback): Args: save_folder (str): Folder to save checkpoints to (can be a URI) - filename (str): Filename to save checkpoints to. batch_interval (int): Number of batches between checkpoints. + filename (str): Filename to save checkpoints to. overwrite (bool): Whether to overwrite previous checkpoints. - keep_optimizer(bool): Whether to save the optimizer state in the monolithic checkpoint. + keep_optimizers (bool): Whether to save the optimizer state in the monolithic checkpoint. """ def __init__(self,