update

funasr/bin/train.py

@@ -10,12 +10,12 @@ 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.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/build_args.py

@@ -11,7 +11,7 @@ from funasr.utils.types import str_or_none
 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 @@ def build_args(args):
             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_lm_model.py

@@ -20,7 +20,7 @@ class_choices_list = [
 ]
 
 
-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

@@ -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):