Update e2e_asr_transducer.py (#684)

aky15

2023-06-28 15847d83a3d368b299999734e587ec2089fa950d

Update e2e_asr_transducer.py (#684)

add missing modules

 funasr/models/e2e_asr_transducer.py

@@ -7,7 +7,9 @@
import torch
from packaging.version import parse as V
from typeguard import check_argument_types

from funasr.losses.label_smoothing_loss import (
    LabelSmoothingLoss,  # noqa: H301
)
from funasr.models.frontend.abs_frontend import AbsFrontend
from funasr.models.specaug.abs_specaug import AbsSpecAug
from funasr.models.decoder.rnnt_decoder import RNNTDecoder
@@ -15,6 +17,8 @@
from funasr.models.encoder.abs_encoder import AbsEncoder
from funasr.models.joint_net.joint_network import JointNetwork
from funasr.modules.nets_utils import get_transducer_task_io
from funasr.modules.nets_utils import th_accuracy
from funasr.modules.add_sos_eos import add_sos_eos
from funasr.layers.abs_normalize import AbsNormalize
from funasr.torch_utils.device_funcs import force_gatherable
from funasr.models.base_model import FunASRModel