Only log "changing autoresume" when actually changing (#653)

mosaicml · Oct 8, 2023 · df945fa · df945fa
1 parent 7fb084a
commit df945fa
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,