speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

joint network bug fix

Browse Source
This commit is contained in:
aky15 2023-05-17 19:09:44 +08:00
parent 429995f4d0
commit 6f10b7bc41
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
funasr/tasks/asr.py
View File

@ -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.",