opensource/PaddleOCR
3
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2026-09-24 23:33:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Manan Goel
2022-04-15 09:44:42 +00:00
parent 87ba1c36bc
commit f161045675
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tools/program.py
+2 -2
View File
@@ -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: