From 584d0bc0ebb0de360c8cc3c05c26e376b943fefa Mon Sep 17 00:00:00 2001
From: aky15 <ankeyuthu@gmail.com>
Date: 星期二, 23 五月 2023 17:49:01 +0800
Subject: [PATCH] Merge pull request #538 from alibaba-damo-academy/dev_aky2
---
funasr/tasks/asr.py | 9 ++-------
1 files changed, 2 insertions(+), 7 deletions(-)
diff --git a/funasr/tasks/asr.py b/funasr/tasks/asr.py
index 8e4f9cc..8244856 100644
--- a/funasr/tasks/asr.py
+++ b/funasr/tasks/asr.py
@@ -363,12 +363,6 @@
default=get_default_kwargs(CTC),
help="The keyword arguments for CTC class.",
)
- group.add_argument(
- "--joint_network_conf",
- action=NestedDictAction,
- default=None,
- help="The keyword arguments for joint network class.",
- )
group = parser.add_argument_group(description="Preprocess related")
group.add_argument(
@@ -1379,6 +1373,7 @@
num_optimizers: int = 1
class_choices_list = [
+ model_choices,
frontend_choices,
specaug_choices,
normalize_choices,
@@ -1476,7 +1471,7 @@
try:
model_class = model_choices.get_class(args.model)
except AttributeError:
- model_class = model_choices.get_class("asr")
+ model_class = model_choices.get_class("rnnt_unified")
model = model_class(
vocab_size=vocab_size,
--
Gitblit v1.9.1