Merge branch 'main' into fix-mpt

mosaicml · Oct 9, 2023 · f8b3606 · f8b3606
2 parents 2ae1c94 + df945fa
commit f8b3606
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/scripts/train/train.py b/scripts/train/train.py
@@ -393,9 +393,12 @@ def main(cfg: DictConfig) -> Trainer:
         and save_folder is not None \
         and not save_overwrite \
         and not save_weights_only:
+        autoresume_default = True
+
+    if cfg.get('autoresume') is None and autoresume_default:
         print('As run_name, save_folder, and save_latest_filename are set, \
                 changing autoresume default to True...')
-        autoresume_default = True
+
     autoresume: bool = pop_config(cfg,
                                   'autoresume',
                                   must_exist=False,