fix undefined save_model_dir when using wandb (#10251)

* fix undefined save_model_dir when using wand * Update program.py
2023-06-30 03:59:38 +00:00 · 2023-06-30 03:59:38 +00:00 · 4b275f1857
parent 8c9f43f44f
commit 4b275f1857
1 changed files with 1 additions and 1 deletions
--- a/tools/program.py
+++ b/tools/program.py
@ -685,7 +685,7 @@ def preprocess(is_train=False):
            wandb_params = config['wandb']
        else:
            wandb_params = dict()
-        wandb_params.update({'save_dir': save_model_dir})
+        wandb_params.update({'save_dir': save_dir})
        log_writer = WandbLogger(**wandb_params, config=config)
        loggers.append(log_writer)
    else: