From 6f10b7bc41544371a2985ca426de905176a0766d Mon Sep 17 00:00:00 2001
From: aky15 <ankeyu.aky@11.17.44.249>
Date: 星期三, 17 五月 2023 19:09:44 +0800
Subject: [PATCH] joint network bug fix
---
funasr/tasks/asr.py | 11 ++++++++++-
1 files changed, 10 insertions(+), 1 deletions(-)
diff --git a/funasr/tasks/asr.py b/funasr/tasks/asr.py
index 0bb0563..5de475f 100644
--- a/funasr/tasks/asr.py
+++ b/funasr/tasks/asr.py
@@ -224,6 +224,15 @@
default="rnnt",
)
+joint_network_choices = ClassChoices(
+ name="joint_network",
+ classes=dict(
+ joint_network=JointNetwork,
+ ),
+ default="joint_network",
+ optional=True,
+)
+
predictor_choices = ClassChoices(
name="predictor",
classes=dict(
@@ -353,7 +362,7 @@
help="The keyword arguments for CTC class.",
)
group.add_argument(
- "--joint_net_conf",
+ "--joint_network_conf",
action=NestedDictAction,
default=None,
help="The keyword arguments for joint network class.",
--
Gitblit v1.9.1