| | |
|---|
| | | from funasr.tasks.asr import ASRTaskParaformer as ASRTask |
|---|
| | | elif mode == "uniasr": |
|---|
| | | from funasr.tasks.asr import ASRTaskUniASR as ASRTask |
|---|
| | | elif mode == "mfcca": |
|---|
| | | from funasr.tasks.asr import ASRTaskMFCCA as ASRTask |
|---|
| | | elif mode == "tp": |
|---|
| | | from funasr.tasks.asr import ASRTaskAligner as ASRTask |
|---|
| | | else: |
|---|
| | | raise ValueError("Unknown mode: {}".format(mode)) |
|---|
| | | parser = ASRTask.get_parser() |
|---|
| | |
|---|
| | | finetune_configs = yaml.safe_load(f) |
|---|
| | | # set data_types |
|---|
| | | if dataset_type == "large": |
|---|
| | | finetune_configs["dataset_conf"]["data_types"] = "sound,text" |
|---|
| | | if 'data_types' not in finetune_configs['dataset_conf']: |
|---|
| | | finetune_configs["dataset_conf"]["data_types"] = "sound,text" |
|---|
| | | finetune_configs = update_dct(configs, finetune_configs) |
|---|
| | | for key, value in finetune_configs.items(): |
|---|
| | | if hasattr(args, key): |
|---|
