fix vdl makedir (#5288)

b1f4cd1f · littletomatodonkey · GitHub · 63d5f785 · b1f4cd1f
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

tools/program.py tools/program.py +1 -1

未找到文件。
--- 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)