From c2dee5e3c29eba79e591d9e9caebaef15ea4e56b Mon Sep 17 00:00:00 2001 From: hnluo <haoneng.lhn@alibaba-inc.com> Date: 星期四, 29 六月 2023 11:09:28 +0800 Subject: [PATCH] Merge pull request #687 from alibaba-damo-academy/dev_lhn --- funasr/models/e2e_asr_transducer.py | 6 +++++- 1 files changed, 5 insertions(+), 1 deletions(-) diff --git a/funasr/models/e2e_asr_transducer.py b/funasr/models/e2e_asr_transducer.py index 3f9f31c..4e33bd6 100644 --- a/funasr/models/e2e_asr_transducer.py +++ b/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 -- Gitblit v1.9.1