aky15
2023-05-17 6f10b7bc41544371a2985ca426de905176a0766d
joint network bug fix
1个文件已修改
11 ■■■■■ 已修改文件
funasr/tasks/asr.py 11 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
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.",