fix dp init

3 years ago · 663e3ab58e
parent 08b9c45811
commit 663e3ab58e
1 changed files with 3 additions and 1 deletions
--- a/paddlespeech/s2t/training/trainer.py
+++ b/paddlespeech/s2t/training/trainer.py
@ -19,7 +19,9 @@ from pathlib import Path
 import paddle
 from paddle import distributed as dist
-dist.init_parallel_env()
+world_size = dist.get_world_size()
 if world_size > 1:
    dist.init_parallel_env()
 from visualdl import LogWriter