未验证 提交 71b7c453 编写于 作者: I itasli 提交者: GitHub

fix undefined save_model_dir when using wandb (#10251) (#10709)

* fix undefined save_model_dir when using wand

* Update program.py
上级 daa7bed5
...@@ -697,7 +697,7 @@ def preprocess(is_train=False): ...@@ -697,7 +697,7 @@ def preprocess(is_train=False):
wandb_params = config['wandb'] wandb_params = config['wandb']
else: else:
wandb_params = dict() 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) log_writer = WandbLogger(**wandb_params, config=config)
loggers.append(log_writer) loggers.append(log_writer)
else: else:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册