From b3b4c1bc5bb068c9aff99740e6257d12c6676ff7 Mon Sep 17 00:00:00 2001
From: aky15 <ankeyu.aky@11.17.44.249>
Date: 星期一, 17 四月 2023 11:19:14 +0800
Subject: [PATCH] rename some functions
---
funasr/tasks/asr_transducer.py | 18 +++++++++---------
1 files changed, 9 insertions(+), 9 deletions(-)
diff --git a/funasr/tasks/asr_transducer.py b/funasr/tasks/asr_transducer.py
index cae18c1..d4136d0 100644
--- a/funasr/tasks/asr_transducer.py
+++ b/funasr/tasks/asr_transducer.py
@@ -21,13 +21,13 @@
LightweightConvolutionTransformerDecoder,
TransformerDecoder,
)
-from funasr.models.rnnt_decoder.abs_decoder import AbsDecoder
-from funasr.models.rnnt_decoder.rnn_decoder import RNNDecoder
-from funasr.models.rnnt_decoder.stateless_decoder import StatelessDecoder
-from funasr.models.encoder.chunk_encoder import ChunkEncoder as Encoder
-from funasr.models.e2e_transducer import TransducerModel
-from funasr.models.e2e_transducer_unified import UnifiedTransducerModel
-from funasr.models.joint_network import JointNetwork
+from funasr.models.rnnt_predictor.abs_decoder import AbsDecoder
+from funasr.models.rnnt_predictor.rnn_decoder import RNNDecoder
+from funasr.models.rnnt_predictor.stateless_decoder import StatelessDecoder
+from funasr.models.encoder.conformer_encoder import ConformerChunkEncoder
+from funasr.models.e2e_asr_transducer import TransducerModel
+from funasr.models.e2e_asr_transducer_unified import UnifiedTransducerModel
+from funasr.models.joint_net.joint_network import JointNetwork
from funasr.layers.abs_normalize import AbsNormalize
from funasr.layers.global_mvn import GlobalMVN
from funasr.layers.utterance_mvn import UtteranceMVN
@@ -72,9 +72,9 @@
encoder_choices = ClassChoices(
"encoder",
classes=dict(
- encoder=Encoder,
+ chunk_conformer=ConformerChunkEncoder,
),
- default="encoder",
+ default="chunk_conformer",
)
decoder_choices = ClassChoices(
--
Gitblit v1.9.1