diff --git a/examples/asr/conf/asr_adapters/asr_adaptation.yaml b/examples/asr/conf/asr_adapters/asr_adaptation.yaml index fad6fa981a1a..34f2b4293438 100644 --- a/examples/asr/conf/asr_adapters/asr_adaptation.yaml +++ b/examples/asr/conf/asr_adapters/asr_adaptation.yaml @@ -190,6 +190,7 @@ exp_manager: name: null project: null entity: null + save_dir: null resume_if_exists: false resume_ignore_no_checkpoint: false diff --git a/nemo/utils/exp_manager.py b/nemo/utils/exp_manager.py index 011c29c8317c..cf653bb34c87 100644 --- a/nemo/utils/exp_manager.py +++ b/nemo/utils/exp_manager.py @@ -665,7 +665,12 @@ def configure_loggers( wandb_kwargs = {} if "name" not in wandb_kwargs and "project" not in wandb_kwargs: raise ValueError("name and project are required for wandb_logger") - wandb_logger = WandbLogger(save_dir=exp_dir, version=version, **wandb_kwargs) + + # Update the wandb save_dir + if wandb_kwargs.get('save_dir', None) is None: + wandb_kwargs['save_dir'] = exp_dir + os.makedirs(wandb_kwargs['save_dir'], exist_ok=True) + wandb_logger = WandbLogger(version=version, **wandb_kwargs) logger_list.append(wandb_logger) logging.info("WandBLogger has been set up")