From 63b980b03027b7b0c56b32fcab6b2805fd64d79d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=8C=97=E5=BF=B5?= Date: Mon, 30 Oct 2023 16:15:03 +0800 Subject: [PATCH] add bpemodel in build_trainer --- funasr/bin/build_trainer.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/funasr/bin/build_trainer.py b/funasr/bin/build_trainer.py index 52aa5093e..61af7663d 100644 --- a/funasr/bin/build_trainer.py +++ b/funasr/bin/build_trainer.py @@ -548,6 +548,7 @@ def build_trainer(modelscope_dict, init_param = modelscope_dict['init_model'] cmvn_file = modelscope_dict['cmvn_file'] seg_dict_file = modelscope_dict['seg_dict'] + bpemodel = modelscope_dict['bpemodel'] # overwrite parameters with open(config) as f: @@ -581,6 +582,10 @@ def build_trainer(modelscope_dict, args.seg_dict_file = seg_dict_file else: args.seg_dict_file = None + if os.path.exists(bpemodel): + args.bpemodel = bpemodel + else: + args.bpemodel = None args.data_dir = data_dir args.train_set = train_set args.dev_set = dev_set