From 71b7c453810834a3ae3c42f38153a2d01cdf692e Mon Sep 17 00:00:00 2001 From: itasli Date: Wed, 23 Aug 2023 10:06:42 +0200 Subject: [PATCH] fix undefined save_model_dir when using wandb (#10251) (#10709) * 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 511ee9dd..b01c2e43 100755 --- 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: -- GitLab