From cc2c1d1d53dea5d2c45f858d1baa5bd279f47987 Mon Sep 17 00:00:00 2001
From: nichongjia-2007 <nichongjia@gmail.com>
Date: 星期三, 31 五月 2023 14:39:25 +0800
Subject: [PATCH] Merge branch 'main' of https://github.com/alibaba-damo-academy/FunASR
---
funasr/tasks/asr.py | 11 +++--------
1 files changed, 3 insertions(+), 8 deletions(-)
diff --git a/funasr/tasks/asr.py b/funasr/tasks/asr.py
index 8e4f9cc..92333ab 100644
--- a/funasr/tasks/asr.py
+++ b/funasr/tasks/asr.py
@@ -103,7 +103,7 @@
name="specaug",
classes=dict(
specaug=SpecAug,
- specaug_lfr=SpecAugLFR,
+ specaug_lfr=FSpecAugLR,
),
type_check=AbsSpecAug,
default=None,
@@ -362,12 +362,6 @@
action=NestedDictAction,
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")
@@ -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