From 395f6741c8767017111c0d6e264e27178be01964 Mon Sep 17 00:00:00 2001
From: zhifu gao <zhifu.gzf@alibaba-inc.com>
Date: 星期四, 02 三月 2023 17:07:29 +0800
Subject: [PATCH] Merge pull request #172 from alibaba-damo-academy/dev_wjm
---
funasr/train/distributed_utils.py | 4 ++--
1 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/funasr/train/distributed_utils.py b/funasr/train/distributed_utils.py
index c897930..13f5744 100644
--- a/funasr/train/distributed_utils.py
+++ b/funasr/train/distributed_utils.py
@@ -53,7 +53,7 @@
# https://pytorch.org/docs/stable/distributed.html#torch.distributed.init_process_group
os.environ.setdefault("NCCL_BLOCKING_WAIT", "1")
- torch.distributed.init_process_group(backend='nccl',
+ torch.distributed.init_process_group(backend=self.dist_backend,
init_method=self.dist_init_method,
world_size=args.dist_world_size,
rank=args.dist_rank)
@@ -113,7 +113,7 @@
# https://pytorch.org/docs/stable/distributed.html#torch.distributed.init_process_group
os.environ.setdefault("NCCL_BLOCKING_WAIT", "1")
- torch.distributed.init_process_group(backend='nccl', init_method='env://')
+ torch.distributed.init_process_group(backend=self.dist_backend, init_method='env://')
self.dist_rank = torch.distributed.get_rank()
self.dist_world_size = torch.distributed.get_world_size()
self.local_rank = args.local_rank
--
Gitblit v1.9.1