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 batch_bins configuration

Browse Source
This commit is contained in:
shixian.shi 2023-05-05 14:09:41 +08:00
parent 3c4898fe16
commit 651a945773
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"]: