From a8f0aad81de964941493c57351925071f3a8b733 Mon Sep 17 00:00:00 2001
From: sugarcase <shi.pengteng@163.com>
Date: 星期五, 27 九月 2024 14:16:28 +0800
Subject: [PATCH] fsmn_kws_mt finetune and inference adapt to right modelscope hub (#2113)
---
examples/industrial_data_pretraining/fsmn_kws_mt/convert.py | 4 ----
1 files changed, 0 insertions(+), 4 deletions(-)
diff --git a/examples/industrial_data_pretraining/fsmn_kws_mt/convert.py b/examples/industrial_data_pretraining/fsmn_kws_mt/convert.py
index e63e689..a6ef0f8 100644
--- a/examples/industrial_data_pretraining/fsmn_kws_mt/convert.py
+++ b/examples/industrial_data_pretraining/fsmn_kws_mt/convert.py
@@ -49,8 +49,6 @@
copyfile(network_file, os.path.join(model_dir, 'origin.torch.pt'))
model = FsmnKWSMTConvert(
- vocab_size=configs['encoder_conf']['output_dim'],
- vocab_size2=configs['encoder_conf']['output_dim2'],
encoder='FSMNMTConvert',
encoder_conf=configs['encoder_conf'],
ctc_conf=configs['ctc_conf'],
@@ -82,8 +80,6 @@
model_name="convert.torch.pt"
):
model = FsmnKWSMTConvert(
- vocab_size=configs['encoder_conf']['output_dim'],
- vocab_size2=configs['encoder_conf']['output_dim2'],
encoder='FSMNMTConvert',
encoder_conf=configs['encoder_conf'],
ctc_conf=configs['ctc_conf'],
--
Gitblit v1.9.1