From 95fc10961fa66b66ed92cdedc961801bb2330030 Mon Sep 17 00:00:00 2001 From: 雾聪 <wucong.lyb@alibaba-inc.com> Date: 星期四, 29 六月 2023 14:31:44 +0800 Subject: [PATCH] Merge branch 'main' of https://github.com/alibaba-damo-academy/FunASR into main --- 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