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

Merge pull request #457 from alibaba-damo-academy/dev_sx

Browse Source 
update batch_bins configuration
This commit is contained in:
zhifu gao 2023-05-05 14:43:15 +08:00 committed by GitHub
commit b39d744cd5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
funasr/bin/build_trainer.py
View File

@ -132,8 +132,7 @@ def build_trainer(modelscope_dict,
if args.dataset_type == "small":
args.batch_bins = batch_bins
elif args.dataset_type == "large":
if "batch_size" not in args.dataset_conf["batch_conf"]:
args.dataset_conf["batch_conf"]["batch_size"] = batch_bins
args.dataset_conf["batch_conf"]["batch_size"] = batch_bins
else:
raise ValueError(f"Not supported dataset_type={args.dataset_type}")
if args.normalize in ["null", "none", "None"]: