Change default for always_save_context to True (#11014)

Signed-off-by: Abhishree <[email protected]> Co-authored-by: Pablo Garay <[email protected]>
NVIDIA · Oct 25, 2024 · d2ba2c3 · d2ba2c3
1 parent 5f2c7fe
commit d2ba2c3
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 = False,
+        always_save_context: bool = True,
         save_context_on_train_end: bool = True,
         **kwargs,
     ):