update

speech_asr

2023-04-20 993fdd8ecf50e9260c2885c273a279186a68d1f3

update

 funasr/bin/train.py

@@ -10,12 +10,12 @@
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.build_args import build_args
from funasr.utils.build_dataloader import build_dataloader
from funasr.utils.build_distributed import build_distributed
from funasr.utils.build_model import build_model
from funasr.utils.build_optimizer import build_optimizer
from funasr.utils.build_scheduler import build_scheduler
from funasr.build_utils.build_args import build_args
from funasr.build_utils.build_dataloader import build_dataloader
from funasr.build_utils.build_distributed import build_distributed
from funasr.build_utils.build_model import build_model
from funasr.build_utils.build_optimizer import build_optimizer
from funasr.build_utils.build_scheduler import build_scheduler
from funasr.utils.prepare_data import prepare_data
from funasr.utils.types import str2bool
from funasr.utils.yaml_no_alias_safe_dump import yaml_no_alias_safe_dump

 funasr/build_utils/__init__.py

 funasr/build_utils/build_args.py

File was renamed from funasr/utils/build_args.py
@@ -11,7 +11,7 @@
def build_args(args):
    parser = argparse.ArgumentParser("Task related config")
    if args.task_name == "asr":
        from funasr.utils.build_asr_model import class_choices_list
        from funasr.build_utils.build_asr_model import class_choices_list
        for class_choices in class_choices_list:
            # Append --<name> and --<name>_conf.
            # e.g. --encoder and --encoder_conf
@@ -80,7 +80,7 @@
            help="The file path of noise scp file.",
        )
    elif args.task_name == "pretrain":
        from funasr.utils.build_pretrain_model import class_choices_list
        from funasr.build_utils.build_pretrain_model import class_choices_list
        for class_choices in class_choices_list:
            # Append --<name> and --<name>_conf.
            # e.g. --encoder and --encoder_conf

 funasr/build_utils/build_asr_model.py

 funasr/build_utils/build_dataloader.py

 funasr/build_utils/build_distributed.py

 funasr/build_utils/build_lm_model.py

File was renamed from funasr/utils/build_lm_model.py
@@ -20,7 +20,7 @@
]


def build_pretrain_model(args):
def build_lm_model(args):
    # token_list
    if args.token_list is not None:
        with open(args.token_list) as f:

 funasr/build_utils/build_model.py

File was renamed from funasr/utils/build_model.py
@@ -1,5 +1,6 @@
from funasr.utils.build_asr_model import build_asr_model
from funasr.utils.build_pretrain_model import build_pretrain_model
from funasr.build_utils.build_asr_model import build_asr_model
from funasr.build_utils.build_pretrain_model import build_pretrain_model
from funasr.build_utils.build_lm_model import build_lm_model


def build_model(args):

 funasr/build_utils/build_optimizer.py

 funasr/build_utils/build_pretrain_model.py

 funasr/build_utils/build_scheduler.py