From 0e8468d9fd2c38e973879b13e9e9dce2417177ea Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BC=A0=E6=98=A5=E4=B9=94?= <83450930+Liyulingyue@users.noreply.github.com> Date: Wed, 6 Nov 2024 19:20:51 +0800 Subject: [PATCH] Update wav2vec2_ASR.py --- .../s2t/models/wav2vec2/wav2vec2_ASR.py | 18 ------------------ 1 file changed, 18 deletions(-) diff --git a/paddlespeech/s2t/models/wav2vec2/wav2vec2_ASR.py b/paddlespeech/s2t/models/wav2vec2/wav2vec2_ASR.py index 65a5f8e59..64195defc 100755 --- a/paddlespeech/s2t/models/wav2vec2/wav2vec2_ASR.py +++ b/paddlespeech/s2t/models/wav2vec2/wav2vec2_ASR.py @@ -80,24 +80,6 @@ class Wav2vec2ASR(nn.Layer): return ctc_loss - @paddle.no_grad() - def extract_features(self, wav): - if self.normalize_wav: - wav = F.layer_norm(wav, wav.shape[-1]) - - # Extract wav2vec output - out = self.wav2vec2(wav)[0] - # We normalize the output if required - if self.output_norm: - out = F.layer_norm(out, out.shape[-1]) - - if self.training and hasattr(self.config, 'spec_augment'): - feats = self.spec_augment(out) - else: - feats = out - - return feats - @paddle.no_grad() def decode(self, feats: paddle.Tensor,