aky15
2023-04-17 b3b4c1bc5bb068c9aff99740e6257d12c6676ff7
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(