From df1c97af800f52e5a57ed804dfe8f95e84477e3e Mon Sep 17 00:00:00 2001 From: LDOUBLEV Date: Tue, 2 Feb 2021 21:27:48 +0800 Subject: [PATCH] fix bug --- tools/program.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tools/program.py b/tools/program.py index 99a37432..e91a70e0 100755 --- 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) -- GitLab