Skip to content

Commit

Permalink
Change default parallel_save to False (#9633)
Browse files Browse the repository at this point in the history
Signed-off-by: Mikołaj Błaż <[email protected]>
  • Loading branch information
mikolajblaz authored Jul 8, 2024
1 parent 7c05678 commit 0661390
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions nemo/utils/callbacks/dist_ckpt_io.py
Original file line number Diff line number Diff line change
Expand Up @@ -205,7 +205,7 @@ def __init__(
async_save: bool = False,
torch_dist_multiproc: Optional[int] = None,
assume_constant_structure: bool = False,
parallel_save: bool = True,
parallel_save: bool = False,
parallel_load: bool = False,
):
super().__init__()
Expand Down Expand Up @@ -238,7 +238,7 @@ def from_config(cls, model_cfg: dict, async_save: bool = False):
load_directly_on_device=model_cfg.get('dist_ckpt_load_on_device', True),
async_save=async_save,
torch_dist_multiproc=model_cfg.get('dist_ckpt_torch_dist_multiproc', None),
parallel_save=model_cfg.get('dist_ckpt_parallel_save', True),
parallel_save=model_cfg.get('dist_ckpt_parallel_save', False),
parallel_load=model_cfg.get('dist_ckpt_parallel_load', False),
)

Expand Down

0 comments on commit 0661390

Please sign in to comment.