From 87da7393041a9afaa42dc8c4b0fd3bcff624c182 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=98=89=E6=B8=8A?= Date: Mon, 24 Apr 2023 23:09:33 +0800 Subject: [PATCH] update --- funasr/models/e2e_asr.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/funasr/models/e2e_asr.py b/funasr/models/e2e_asr.py index 950d69960..d3d5dfdc6 100644 --- a/funasr/models/e2e_asr.py +++ b/funasr/models/e2e_asr.py @@ -17,9 +17,8 @@ from funasr.losses.label_smoothing_loss import ( LabelSmoothingLoss, # noqa: H301 ) from funasr.models.ctc import CTC +from funasr.models.encoder.abs_encoder import AbsEncoder from funasr.models.decoder.abs_decoder import AbsDecoder -from funasr.models.postencoder.abs_postencoder import AbsPostEncoder -from funasr.models.preencoder.abs_preencoder import AbsPreEncoder from funasr.models.base_model import FunASRModel from funasr.modules.add_sos_eos import add_sos_eos from funasr.modules.e2e_asr_common import ErrorCalculator @@ -45,9 +44,7 @@ class ESPnetASRModel(FunASRModel): frontend: Optional[torch.nn.Module], specaug: Optional[torch.nn.Module], normalize: Optional[torch.nn.Module], - preencoder: Optional[AbsPreEncoder], - encoder: torch.nn.Module, - postencoder: Optional[AbsPostEncoder], + encoder: AbsEncoder, decoder: AbsDecoder, ctc: CTC, ctc_weight: float = 0.5,