Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-07-10 12:33:28 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #613 from Lizerui9926/master

Browse Source 
add bpemodel path in asr_trainer to support paraformer-16k-en finetune
This commit is contained in:
zhifu gao
2023-11-02 16:07:01 +08:00
committed by GitHub
parent eff11817a0 60155a0f7b
commit cfba0b0b7e
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
modelscope/trainers/audio/asr_trainer.py
View File

@@ -126,6 +126,7 @@ class ASRTrainer(BaseTrainer):
cfg_dict['cmvn_file'] = os.path.join(
cur_dir, config['model']['model_config']['mvn_file'])
cfg_dict['seg_dict'] = os.path.join(cur_dir, 'seg_dict')
cfg_dict['bpemodel'] = os.path.join(cur_dir, config['model']['model_config']['bpemodel'])
if 'init_model' in config['model']['model_config']:
cfg_dict['init_model'] = os.path.join(
cur_dir, config['model']['model_config']['init_model'])