From 810046e3df4910c8f5c1a90e4b53aca45b3397e8 Mon Sep 17 00:00:00 2001
From: wuhongsheng <664116298@qq.com>
Date: 星期一, 01 七月 2024 10:42:58 +0800
Subject: [PATCH] 优化merge segments 参数，解决新闻联播男女主持人“晚上好”合并一个speakid问题 (#1861)
---
funasr/models/seaco_paraformer/export_meta.py | 9 ++++-----
1 files changed, 4 insertions(+), 5 deletions(-)
diff --git a/funasr/models/seaco_paraformer/export_meta.py b/funasr/models/seaco_paraformer/export_meta.py
index a246a29..947f4d9 100644
--- a/funasr/models/seaco_paraformer/export_meta.py
+++ b/funasr/models/seaco_paraformer/export_meta.py
@@ -53,7 +53,7 @@
0: "num_hotwords",
},
"hw_embed": {
- 0: "num_hotwords",
+ 1: "num_hotwords",
},
}
@@ -109,7 +109,9 @@
backbone_model.export_dynamic_axes = types.MethodType(
export_backbone_dynamic_axes, backbone_model
)
- backbone_model.export_name = types.MethodType(export_backbone_name, backbone_model)
+
+ embedder_model.export_name = "model_eb"
+ backbone_model.export_name = "model"
return backbone_model, embedder_model
@@ -198,6 +200,3 @@
"us_cif_peak": {0: "batch_size", 1: "alphas_length"},
}
-
-def export_backbone_name(self):
- return "model.onnx"
--
Gitblit v1.9.1