From 6f10b7bc41544371a2985ca426de905176a0766d Mon Sep 17 00:00:00 2001
From: aky15 <ankeyu.aky@11.17.44.249>
Date: Wed, 17 May 2023 19:09:44 +0800
Subject: [PATCH] joint network bug fix

---
 funasr/tasks/asr.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/funasr/tasks/asr.py b/funasr/tasks/asr.py
index 0bb056365..5de475f19 100644
--- a/funasr/tasks/asr.py
+++ b/funasr/tasks/asr.py
@@ -224,6 +224,15 @@ rnnt_decoder_choices = ClassChoices(
     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 @@ class ASRTask(AbsTask):
             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.",