Revert always_save_context to False (#10667)

Signed-off-by: Abhishree <[email protected]>
NVIDIA · Oct 4, 2024 · 771e0b6 · 771e0b6
1 parent 81a0ddd
commit 771e0b6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,
     ):