From fae856e23d45fd27d5fd55fd036e8e3fc7b24915 Mon Sep 17 00:00:00 2001
From: 雾聪 <wucong.lyb@alibaba-inc.com>
Date: 星期五, 02 六月 2023 23:00:08 +0800
Subject: [PATCH] update funasr-onnx-offline
---
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