From 0eb3b1161b8c832a3a54831d6879b4923f23b265 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=98=89=E6=B8=8A?= <wangjiaming.wjm@alibaba-inc.com>
Date: Mon, 24 Apr 2023 19:52:25 +0800
Subject: [PATCH] update

---
 funasr/bin/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/funasr/bin/train.py b/funasr/bin/train.py
index 9c8f672e7..27d5c4ad6 100755
--- a/funasr/bin/train.py
+++ b/funasr/bin/train.py
@@ -444,7 +444,7 @@ if __name__ == '__main__':
 
     # ddp init
     os.environ['CUDA_VISIBLE_DEVICES'] = str(args.gpu_id)
-    args.distributed = args.dist_world_size > 1
+    args.distributed = args.ngpu > 1 or args.dist_world_size > 1
     distributed_option = build_distributed(args)
 
     # for logging