Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-24 20:19:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #627 from Lizerui9926/master

Browse Source 
Fix asr paraformer finetune bug
This commit is contained in:
zhifu gao
2023-11-09 12:41:17 +08:00
committed by GitHub
parent 069b271436 9324fb4422
commit 29b7f16339
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -126,8 +126,11 @@ 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 'bpemodel' in config['model']['model_config']:
cfg_dict['bpemodel'] = os.path.join(
cur_dir, config['model']['model_config']['bpemodel'])
else:
cfg_dict['bpemodel'] = None
if 'init_model' in config['model']['model_config']:
cfg_dict['init_model'] = os.path.join(
cur_dir, config['model']['model_config']['init_model'])