From 0bf8edca37d384eac6d0c345edd6b1a2b88f97ae Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期一, 30 九月 2024 15:43:13 +0800
Subject: [PATCH] find_unused_parameters
---
funasr/bin/train_ds.py | 3 ---
1 files changed, 0 insertions(+), 3 deletions(-)
diff --git a/funasr/bin/train_ds.py b/funasr/bin/train_ds.py
index 69c2712..5b1eeaa 100644
--- a/funasr/bin/train_ds.py
+++ b/funasr/bin/train_ds.py
@@ -84,9 +84,6 @@
dist.init_process_group(
backend=kwargs.get("backend", "nccl"),
init_method="env://",
- find_unused_parameters=kwargs.get("train_conf", {}).get(
- "find_unused_parameters", False
- ),
)
torch.cuda.set_device(local_rank)
--
Gitblit v1.9.1