From 4b275f1857dfd44524db0860c6aa9bfcb75a7a81 Mon Sep 17 00:00:00 2001
From: itasli <ilyas.tasli@outlook.fr>
Date: Fri, 30 Jun 2023 03:59:38 +0000
Subject: [PATCH] fix undefined save_model_dir when using wandb (#10251)
* fix undefined save_model_dir when using wand
* Update program.py
---
tools/program.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/tools/program.py b/tools/program.py
index a8373435c..209212fde 100755
--- 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: