From 663e3ab58ee21d24b2f6d28f5d1050fba84be088 Mon Sep 17 00:00:00 2001
From: tianhao zhang <15600919271@163.com>
Date: Fri, 9 Sep 2022 09:52:49 +0000
Subject: [PATCH] fix dp init

---
 paddlespeech/s2t/training/trainer.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/paddlespeech/s2t/training/trainer.py b/paddlespeech/s2t/training/trainer.py
index 1093e4a12..4a69d78a4 100644
--- 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