fix vdl makedir (#5288)

b1f4cd1f · littletomatodonkey · GitHub · 63d5f785 · b1f4cd1f
Unverified Commit b1f4cd1f authored Jan 19, 2022 by littletomatodonkey Committed by GitHub Jan 19, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

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

No files found.
--- 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)