update StackNFramesSubsampling (wenet-e2e#2551)

ZailiWang · Jun 4, 2024 · fcb4b98 · fcb4b98
1 parent f28c86f
commit fcb4b98
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/wenet/transformer/subsampling.py b/wenet/transformer/subsampling.py
@@ -385,6 +385,6 @@ def forward(
             new_mask = ~make_pad_mask(seq_len, max_len=s // self.stride)
             x = x.view(b, s // self.stride, self.idim * self.stride)
             _, pos_emb = self.pos_enc_class(x, offset)
-            x = self.norm(x)
-            x = self.out(x)
+        x = self.norm(x)
+        x = self.out(x)
         return x, pos_emb, new_mask.unsqueeze(1)