| File was renamed from funasr/utils/build_args.py |
| | |
| | | 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 |
| | |
| | | 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 |