From 6d71ef84fcb73dc3080bf81678af54a3140b3884 Mon Sep 17 00:00:00 2001 From: Milo Cress Date: Sat, 2 Mar 2024 15:49:26 -0500 Subject: [PATCH] Update scripts/train/train.py Co-authored-by: Daniel King <43149077+dakinggg@users.noreply.github.com> --- scripts/train/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/train/train.py b/scripts/train/train.py index 02f18e764a..61fbc8c8e1 100644 --- a/scripts/train/train.py +++ b/scripts/train/train.py @@ -269,7 +269,7 @@ def main(cfg: DictConfig) -> Trainer: cfg, 'save_latest_filename', must_exist=False, - default_value='latest-rank{rank}_sharded_magic_keyword.pt' + default_value='latest-sharded-rank{rank}' if is_state_dict_sharded else 'latest-rank{rank}.pt') save_overwrite: bool = pop_config(cfg, 'save_overwrite',