From f87811e65c1f9cdace638122df5f29c150a50b60 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Wed, 30 Mar 2022 21:41:46 +0800 Subject: [PATCH] Fix RE identity --- .../ASR/pruned_transducer_stateless2/conformer.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless2/conformer.py b/egs/librispeech/ASR/pruned_transducer_stateless2/conformer.py index 03a47927f..528cc48f4 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless2/conformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless2/conformer.py @@ -83,7 +83,7 @@ class Conformer(EncoderInterface): aux_layers=list(range(0, num_encoder_layers-1, aux_layer_period))) if output_dim == d_model: - self.encoder_output_layer = Identity() + self.encoder_output_layer = nn.Identity() else: self.encoder_output_layer = ScaledLinear(d_model, output_dim, initial_speed=0.5) @@ -936,10 +936,6 @@ class ConvolutionModule(nn.Module): return x.permute(2, 0, 1) -class Identity(torch.nn.Module): - def forward(self, x: Tensor) -> Tensor: - return x - class Conv2dSubsampling(nn.Module): """Convolutional 2D subsampling (to 1/4 length).