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-07-26 19:11:19 +08:00
parent fcac51fbd7
commit 8e53dd6b77
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
funasr/bin/build_trainer.py
View File

@ -529,13 +529,14 @@ def build_trainer(modelscope_dict,
**kwargs):
parser = get_parser()
args, extra_task_params = parser.parse_known_args()
if extra_task_params:
args = build_args(args, parser, extra_task_params)
args = build_args(args, parser, extra_task_params)
if args.local_rank is not None:
args.distributed = True
args.simple_ddp = True
else:
args.distributed = False
args.ngpu = 1
args.local_rank = args.local_rank if args.local_rank is not None else 0
local_rank = args.local_rank
if "CUDA_VISIBLE_DEVICES" in os.environ.keys():
@ -588,6 +589,10 @@ def build_trainer(modelscope_dict,
args.output_dir = output_dir
args.gpu_id = args.local_rank
args.config = finetune_config
args.use_pai = False
args.batch_type = "length"
args.oss_bucket = None
args.input_size = None
if optim is not None:
args.optim = optim
if lr is not None: