From 15847d83a3d368b299999734e587ec2089fa950d Mon Sep 17 00:00:00 2001 From: aky15 <ankeyuthu@gmail.com> Date: 星期三, 28 六月 2023 22:37:43 +0800 Subject: [PATCH] Update e2e_asr_transducer.py (#684) --- 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