speech_asr
2023-04-20 10bce3e6326afb1e1e9df088946c2af00fe126f5
update
1个文件已修改
13 ■■■■■ 已修改文件
funasr/build_utils/build_args.py 13 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/build_utils/build_args.py
@@ -74,6 +74,19 @@
            # e.g. --encoder and --encoder_conf
            class_choices.add_arguments(parser)
    elif args.task_name == "vad":
        from funasr.build_utils.build_vad_model import class_choices_list
        for class_choices in class_choices_list:
            # Append --<name> and --<name>_conf.
            # e.g. --encoder and --encoder_conf
            class_choices.add_arguments(parser)
        parser.add_argument(
            "--input_size",
            type=int_or_none,
            default=None,
            help="The number of input dimension of the feature",
        )
    else:
        raise NotImplementedError("Not supported task: {}".format(args.task_name))