Skip to content

GitLab

Commit df1c97af authored by LDOUBLEV's avatar LDOUBLEV
Browse files

fix bug

parent adc83ca9
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tools/program.py
View file @ df1c97af
...@@ -381,6 +381,7 @@ def preprocess(is_train=False): ...@@ -381,6 +381,7 @@ def preprocess(is_train=False):
logger = get_logger(name='root', log_file=log_file) logger = get_logger(name='root', log_file=log_file)
if config['Global']['use_visualdl']: if config['Global']['use_visualdl']:
from visualdl import LogWriter from visualdl import LogWriter
save_model_dir = config['Global']['save_model_dir']
vdl_writer_path = '{}/vdl/'.format(save_model_dir) vdl_writer_path = '{}/vdl/'.format(save_model_dir)
os.makedirs(vdl_writer_path, exist_ok=True) os.makedirs(vdl_writer_path, exist_ok=True)
vdl_writer = LogWriter(logdir=vdl_writer_path) vdl_writer = LogWriter(logdir=vdl_writer_path)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment