From f87c9da9ccecabec549ee378a26618e26126edf4 Mon Sep 17 00:00:00 2001 From: Irene Dea Date: Fri, 4 Oct 2024 18:20:00 -0700 Subject: [PATCH] Do not consider run_name when auto-detecting autoresume (#1571) --- llmfoundry/command_utils/train.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/llmfoundry/command_utils/train.py b/llmfoundry/command_utils/train.py index 9a5931ddba..2168cfefd4 100644 --- a/llmfoundry/command_utils/train.py +++ b/llmfoundry/command_utils/train.py @@ -318,8 +318,7 @@ def train(cfg: DictConfig) -> Trainer: # Enable autoresume from model checkpoints if possible autoresume_default: bool = False - if logged_cfg.get('run_name', None) is not None \ - and train_cfg.save_folder is not None \ + if train_cfg.save_folder is not None \ and not train_cfg.save_overwrite \ and not train_cfg.save_weights_only: autoresume_default = True