提交 de5c4e1b 编写于 作者: B Bobholamovic 提交者: cuicheng01

Change vdl dir

上级 b4f10436
......@@ -94,7 +94,7 @@ class Engine(object):
self.vdl_writer = None
if self.config['Global'][
'use_visualdl'] and mode == "train" and dist.get_rank() == 0:
vdl_writer_path = os.path.join(self.output_dir, "vdl")
vdl_writer_path = self.output_dir
if not os.path.exists(vdl_writer_path):
os.makedirs(vdl_writer_path)
self.vdl_writer = LogWriter(logdir=vdl_writer_path)
......
......@@ -110,7 +110,7 @@ def main(args):
# visualDL
vdl_writer = None
if global_config["use_visualdl"]:
vdl_dir = os.path.join(global_config["output_dir"], "vdl")
vdl_dir = global_config["output_dir"]
vdl_writer = LogWriter(vdl_dir)
# build dataloader
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册