From b1f4cd1f4f67758baed22fbb8d9d55b685bd7cd1 Mon Sep 17 00:00:00 2001
From: littletomatodonkey <dazhiningsibuqu@163.com>
Date: Wed, 19 Jan 2022 17:29:39 +0800
Subject: [PATCH] fix vdl makedir (#5288)

---
 tools/program.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tools/program.py b/tools/program.py
index 743ace09..10299940 100755
--- a/tools/program.py
+++ b/tools/program.py
@@ -511,7 +511,7 @@ def preprocess(is_train=False):
 
     config['Global']['distributed'] = dist.get_world_size() != 1
 
-    if config['Global']['use_visualdl']:
+    if config['Global']['use_visualdl'] and dist.get_rank() == 0:
         from visualdl import LogWriter
         save_model_dir = config['Global']['save_model_dir']
         vdl_writer_path = '{}/vdl/'.format(save_model_dir)
-- 
GitLab