Update scripts/train/train.py

Co-authored-by: Daniel King <[email protected]>
mosaicml · Mar 2, 2024 · 6d71ef8 · 6d71ef8
1 parent 551d9f7
commit 6d71ef8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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',