Skip to content

Commit

Permalink
Change default for always_save_context to True
Browse files Browse the repository at this point in the history
Signed-off-by: Abhishree <abhishreetm@gmail.com>
  • Loading branch information
athitten committed Sep 27, 2024
1 parent e29e2d2 commit 607e404
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion nemo/lightning/pytorch/callbacks/model_checkpoint.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
):
Expand Down

0 comments on commit 607e404

Please sign in to comment.