fix vdl makedir (#5288)

This commit is contained in:
littletomatodonkey
2022-01-19 17:29:39 +08:00
committed by GitHub
parent 63d5f7859f
commit b1f4cd1f4f
+1 -1
View File
@@ -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)