From c0b36f0e919ad3bb8c68dc9f7fe480da324a2393 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=8C=97=E5=BF=B5?= Date: Thu, 9 Nov 2023 11:10:21 +0800 Subject: [PATCH] fix asr paraformer finetune bug --- modelscope/trainers/audio/asr_trainer.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/modelscope/trainers/audio/asr_trainer.py b/modelscope/trainers/audio/asr_trainer.py index 1162d6ff..5d9197c8 100644 --- a/modelscope/trainers/audio/asr_trainer.py +++ b/modelscope/trainers/audio/asr_trainer.py @@ -126,7 +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'])