fix undefined save_model_dir when using wandb (#10251)

* fix undefined save_model_dir when using wand * Update program.py
PaddlePaddle · Jun 30, 2023 · 4b275f1 · 4b275f1
1 parent 8c9f43f
commit 4b275f1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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: