Merge branch 'master' of github.com:modelscope/modelscope into feat/llm_riddles

This commit is contained in:
wenmeng.zwm
2023-11-09 21:00:04 +08:00

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'])