fix vdl makedir (#5288)

2022-01-19 17:29:39 +08:00 · 2022-01-19 17:29:39 +08:00 · b1f4cd1f4f
parent 63d5f7859f
commit b1f4cd1f4f
1 changed files with 1 additions and 1 deletions
--- 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)