update

嘉渊

2023-04-24 abde816a6cd393697eac6ffea300061ec0ed8aae

update

 funasr/bin/train.py

@@ -19,7 +19,6 @@
from funasr.torch_utils.model_summary import model_summary
from funasr.torch_utils.pytorch_version import pytorch_cudnn_version
from funasr.torch_utils.set_all_random_seed import set_all_random_seed
from funasr.utils import config_argparse
from funasr.utils.prepare_data import prepare_data
from funasr.utils.types import str2bool
from funasr.utils.types import str_or_none
@@ -27,7 +26,7 @@


def get_parser():
    parser = config_argparse.ArgumentParser(
    parser = argparse.ArgumentParser(
        description="FunASR Common Training Parser",
    )


 funasr/build_utils/build_args.py

@@ -1,6 +1,5 @@
import argparse

from funasr.models.ctc import CTC
from funasr.utils import config_argparse
from funasr.utils.get_default_kwargs import get_default_kwargs
from funasr.utils.nested_dict_action import NestedDictAction
from funasr.utils.types import int_or_none
@@ -9,8 +8,8 @@


def build_args(args, extra_task_params):
    parser = argparse.ArgumentParser("Task related config")
    # if args.task_name == "asr":
    parser = config_argparse.ArgumentParser("Task related config")
    if args.task_name == "asr":
    from funasr.build_utils.build_asr_model import class_choices_list
    for class_choices in class_choices_list:
        class_choices.add_arguments(parser)
@@ -45,45 +44,45 @@
        help="The file path of noise scp file.",
    )

    # elif args.task_name == "pretrain":
    #     from funasr.build_utils.build_pretrain_model import class_choices_list
    #     for class_choices in class_choices_list:
    #         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",
    #     )
    #
    # elif args.task_name == "lm":
    #     from funasr.build_utils.build_lm_model import class_choices_list
    #     for class_choices in class_choices_list:
    #         class_choices.add_arguments(parser)
    #
    # elif args.task_name == "punc":
    #     from funasr.build_utils.build_punc_model import class_choices_list
    #     for class_choices in class_choices_list:
    #         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:
    #         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",
    #     )
    #
    # elif args.task_name == "diar":
    #     from funasr.build_utils.build_diar_model import class_choices_list
    #     for class_choices in class_choices_list:
    #         class_choices.add_arguments(parser)
    #
    # else:
    #     raise NotImplementedError("Not supported task: {}".format(args.task_name))
    elif args.task_name == "pretrain":
        from funasr.build_utils.build_pretrain_model import class_choices_list
        for class_choices in class_choices_list:
            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",
        )

    elif args.task_name == "lm":
        from funasr.build_utils.build_lm_model import class_choices_list
        for class_choices in class_choices_list:
            class_choices.add_arguments(parser)

    elif args.task_name == "punc":
        from funasr.build_utils.build_punc_model import class_choices_list
        for class_choices in class_choices_list:
            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:
            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",
        )

    elif args.task_name == "diar":
        from funasr.build_utils.build_diar_model import class_choices_list
        for class_choices in class_choices_list:
            class_choices.add_arguments(parser)

    else:
        raise NotImplementedError("Not supported task: {}".format(args.task_name))

    task_args = parser.parse_args(extra_task_params)
    return task_args