嘉渊
2023-07-26 8e53dd6b778a6df01854df725b0fa5b4ef6f8317
funasr/build_utils/build_args.py
@@ -41,7 +41,7 @@
            "--cmvn_file",
            type=str_or_none,
            default=None,
            help="The file path of noise scp file.",
            help="The path of cmvn file.",
        )
    elif args.task_name == "pretrain":
@@ -75,6 +75,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 == "diar":
        from funasr.build_utils.build_diar_model import class_choices_list