提交 f1610456 编写于 作者: M Manan Goel

Fixed bug to check if 'use_wandb' and 'use_visualdl' flags are addded or not

上级 87ba1c36
...@@ -560,12 +560,12 @@ def preprocess(is_train=False): ...@@ -560,12 +560,12 @@ def preprocess(is_train=False):
loggers = [] loggers = []
if config['Global']['use_visualdl']: if 'use_visualdl' in config['Global'] and config['Global']['use_visualdl']:
save_model_dir = config['Global']['save_model_dir'] save_model_dir = config['Global']['save_model_dir']
vdl_writer_path = '{}/vdl/'.format(save_model_dir) vdl_writer_path = '{}/vdl/'.format(save_model_dir)
log_writer = VDLLogger(save_model_dir) log_writer = VDLLogger(save_model_dir)
loggers.append(log_writer) loggers.append(log_writer)
if config['Global']['use_wandb'] or 'wandb' in config: if ('use_wandb' in config['Global'] and config['Global']['use_wandb']) or 'wandb' in config:
save_dir = config['Global']['save_model_dir'] save_dir = config['Global']['save_model_dir']
wandb_writer_path = "{}/wandb".format(save_dir) wandb_writer_path = "{}/wandb".format(save_dir)
if "wandb" in config: if "wandb" in config:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册