fix bug

2021-02-02 21:27:48 +08:00 · 2021-02-02 21:27:48 +08:00 · df1c97af80
parent adc83ca9ae
commit df1c97af80
1 changed files with 1 additions and 0 deletions
--- a/tools/program.py
+++ b/tools/program.py
@ -381,6 +381,7 @@ def preprocess(is_train=False):
    logger = get_logger(name='root', log_file=log_file)
    if config['Global']['use_visualdl']:
        from visualdl import LogWriter
+        save_model_dir = config['Global']['save_model_dir']
        vdl_writer_path = '{}/vdl/'.format(save_model_dir)
        os.makedirs(vdl_writer_path, exist_ok=True)
        vdl_writer = LogWriter(logdir=vdl_writer_path)