From 8f0bd5b58296d56d9cd230e104ce8bc395360701 Mon Sep 17 00:00:00 2001
From: Tingquan Gao <gaotingquan@baidu.com>
Date: Wed, 19 Jan 2022 05:57:21 +0000
Subject: [PATCH] fix: fix vdl makedir

---
 ppcls/engine/engine.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/ppcls/engine/engine.py b/ppcls/engine/engine.py
index a58ac1e9..f86b092e 100644
--- a/ppcls/engine/engine.py
+++ b/ppcls/engine/engine.py
@@ -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)
-- 
GitLab