From 0f50861fed137c70313c4e37f5fa6b232e3de004 Mon Sep 17 00:00:00 2001 From: Abhishree Date: Fri, 27 Sep 2024 13:13:09 -0700 Subject: [PATCH] Revert always_save_context to False Signed-off-by: Abhishree --- 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 6aee365a3f60..2df4ca56d1a0 100644 --- a/nemo/lightning/pytorch/callbacks/model_checkpoint.py +++ b/nemo/lightning/pytorch/callbacks/model_checkpoint.py @@ -73,7 +73,7 @@ def __init__( train_time_interval: Optional[timedelta] = None, save_on_train_epoch_end: Optional[bool] = False, # Save after training, not after validation save_optim_on_train_end: Optional[bool] = False, - always_save_context: bool = True, + always_save_context: bool = False, save_context_on_train_end: bool = True, **kwargs, ):