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

Change vdl dir

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