|
@@ -41,7 +41,7 @@ def build_args(args, parser, extra_task_params):
|
|
|
"--cmvn_file",
|
|
"--cmvn_file",
|
|
|
type=str_or_none,
|
|
type=str_or_none,
|
|
|
default=None,
|
|
default=None,
|
|
|
- help="The file path of noise scp file.",
|
|
|
|
|
|
|
+ help="The path of cmvn file.",
|
|
|
)
|
|
)
|
|
|
|
|
|
|
|
elif args.task_name == "pretrain":
|
|
elif args.task_name == "pretrain":
|
|
@@ -75,6 +75,12 @@ def build_args(args, parser, extra_task_params):
|
|
|
default=None,
|
|
default=None,
|
|
|
help="The number of input dimension of the feature",
|
|
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":
|
|
elif args.task_name == "diar":
|
|
|
from funasr.build_utils.build_diar_model import class_choices_list
|
|
from funasr.build_utils.build_diar_model import class_choices_list
|