From 2363c63f957728f6721254291845d6c0eaa99f25 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:58:58 +0800
Subject: [PATCH] update

---
 funasr/bin/train.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/funasr/bin/train.py b/funasr/bin/train.py
index f684f3b6f..d5e7089c9 100755
--- a/funasr/bin/train.py
+++ b/funasr/bin/train.py
@@ -86,6 +86,22 @@ def get_parser():
         help="The master port for distributed training"
              "This value is used when dist_init_method == 'env://'",
     )
+    parser.add_argument(
+        "--dist_launcher",
+        default=None,
+        type=str_or_none,
+        choices=["slurm", "mpi", None],
+        help="The launcher type for distributed training",
+    )
+    parser.add_argument(
+        "--multiprocessing_distributed",
+        default=True,
+        type=str2bool,
+        help="Use multi-processing distributed training to launch "
+             "N processes per node, which has N GPUs. This is the "
+             "fastest way to use PyTorch for either single node or "
+             "multi node data parallel training",
+    )
     parser.add_argument(
         "--unused_parameters",
         type=str2bool,