From 771e0b6ca46f1de5e9b3942f3d0c13841f51eacd Mon Sep 17 00:00:00 2001 From: Abhishree Thittenamane <47577437+athitten@users.noreply.github.com> Date: Fri, 4 Oct 2024 09:11:38 -0700 Subject: [PATCH] Revert always_save_context to False (#10667) 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 486c9a4fe79c..0433b80b4b45 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, ):