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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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)