Fixed bug to check if 'use_wandb' and 'use_visualdl' flags are addded or not
@@ -560,12 +560,12 @@ def preprocess(is_train=False):
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']
vdl_writer_path = '{}/vdl/'.format(save_model_dir)
log_writer = VDLLogger(save_model_dir)
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']
wandb_writer_path = "{}/wandb".format(save_dir)
if "wandb" in config:
