diff --git a/modelscope/trainers/audio/asr_trainer.py b/modelscope/trainers/audio/asr_trainer.py index 04d57f51..5d9197c8 100644 --- a/modelscope/trainers/audio/asr_trainer.py +++ b/modelscope/trainers/audio/asr_trainer.py @@ -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'])