From e40b115282bea0e6ec481ef2d28acd2f1abd99a9 Mon Sep 17 00:00:00 2001 From: Yuxuan Lu Date: Thu, 25 Sep 2025 10:51:24 -0700 Subject: [PATCH] Fix model checkpoint async_save attribute Signed-off-by: Yuxuan Lu --- nemo/lightning/pytorch/callbacks/model_checkpoint.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nemo/lightning/pytorch/callbacks/model_checkpoint.py b/nemo/lightning/pytorch/callbacks/model_checkpoint.py index a0e271aa884d..b7a94eefd595 100644 --- a/nemo/lightning/pytorch/callbacks/model_checkpoint.py +++ b/nemo/lightning/pytorch/callbacks/model_checkpoint.py @@ -57,7 +57,6 @@ class ModelCheckpoint(PTLModelCheckpoint): model, trainer, and dataloader to allow for reproducibility of experiments. save_context_on_train_end: Whether to dump the artifacts on_train_end regardless of whether ``always_save_context`` is ``True``. - async_save: Whether to enable asynchronous checkpointing. Attributes: UNFINISHED_CHECKPOINT_SUFFIX (str): Suffix for unfinished checkpoint files. @@ -69,6 +68,7 @@ class ModelCheckpoint(PTLModelCheckpoint): best_model_score (float): Score of the best checkpoint. best_model_path (str): Path to the best checkpoint. kth_best_model_path (str): Path to the kth best checkpoint. + async_save: Whether to enable asynchronous checkpointing. """ UNFINISHED_CHECKPOINT_SUFFIX = "-unfinished"