speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
嘉渊 2023-04-24 19:58:58 +08:00
parent 996b951365
commit 2363c63f95
1 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
funasr/bin/train.py
View File

@ -86,6 +86,22 @@ def get_parser():
help="The master port for distributed training" help="The master port for distributed training"
"This value is used when dist_init_method == 'env://'", "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( parser.add_argument(
"--unused_parameters", "--unused_parameters",
type=str2bool, type=str2bool,