提交 8f0bd5b5 编写于 作者: T Tingquan Gao 提交者: Tingquan Gao

fix: fix vdl makedir

上级 58dccfd0
......@@ -84,7 +84,8 @@ class Engine(object):
# for visualdl
self.vdl_writer = None
if self.config['Global']['use_visualdl'] and mode == "train":
if self.config['Global'][
'use_visualdl'] and mode == "train" and dist.get_rank() == 0:
vdl_writer_path = os.path.join(self.output_dir, "vdl")
if not os.path.exists(vdl_writer_path):
os.makedirs(vdl_writer_path)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册