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

add bpemodel in build_trainer

Browse Source
This commit is contained in:
北念 2023-10-30 16:15:03 +08:00
parent 8a84ed6a4a
commit 63b980b030
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
funasr/bin/build_trainer.py
View File

@ -548,6 +548,7 @@ def build_trainer(modelscope_dict,
init_param = modelscope_dict['init_model'] init_param = modelscope_dict['init_model']
cmvn_file = modelscope_dict['cmvn_file'] cmvn_file = modelscope_dict['cmvn_file']
seg_dict_file = modelscope_dict['seg_dict'] seg_dict_file = modelscope_dict['seg_dict']
bpemodel = modelscope_dict['bpemodel']
# overwrite parameters # overwrite parameters
with open(config) as f: with open(config) as f:
@ -581,6 +582,10 @@ def build_trainer(modelscope_dict,
args.seg_dict_file = seg_dict_file args.seg_dict_file = seg_dict_file
else: else:
args.seg_dict_file = None args.seg_dict_file = None
if os.path.exists(bpemodel):
args.bpemodel = bpemodel
else:
args.bpemodel = None
args.data_dir = data_dir args.data_dir = data_dir
args.train_set = train_set args.train_set = train_set
args.dev_set = dev_set args.dev_set = dev_set