fix undefined save_model_dir when using wandb (PaddlePaddle#10251) (P…

…addlePaddle#10709) * fix undefined save_model_dir when using wand * Update program.py
truongnn1106 · Aug 23, 2023 · 71b7c45 · 71b7c45
1 parent daa7bed
commit 71b7c45
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/program.py b/tools/program.py
@@ -697,7 +697,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: