hnluo
2023-08-15 3a71c7a227ebd8b61995e2fb8ee0c3c25a8c19c4
funasr/build_utils/build_args.py
@@ -54,6 +54,12 @@
            default=None,
            help="The number of input dimension of the feature",
        )
        task_parser.add_argument(
            "--cmvn_file",
            type=str_or_none,
            default=None,
            help="The path of cmvn file.",
        )
    elif args.task_name == "lm":
        from funasr.build_utils.build_lm_model import class_choices_list
@@ -86,6 +92,12 @@
        from funasr.build_utils.build_diar_model import class_choices_list
        for class_choices in class_choices_list:
            class_choices.add_arguments(task_parser)
        task_parser.add_argument(
            "--input_size",
            type=int_or_none,
            default=None,
            help="The number of input dimension of the feature",
        )
    elif args.task_name == "sv":
        from funasr.build_utils.build_sv_model import class_choices_list