From 8936365c5c5d92c89ff26a0aa694d66e75c0f0f8 Mon Sep 17 00:00:00 2001 From: Yifan Yang Date: Thu, 15 Jun 2023 18:01:27 +0800 Subject: [PATCH] Fix --- egs/librispeech/ASR/zipformer/zipformer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/egs/librispeech/ASR/zipformer/zipformer.py b/egs/librispeech/ASR/zipformer/zipformer.py index 9c2614a30..7d98dbeb1 100644 --- a/egs/librispeech/ASR/zipformer/zipformer.py +++ b/egs/librispeech/ASR/zipformer/zipformer.py @@ -2138,7 +2138,7 @@ class ConvolutionModule(nn.Module): x = self.in_proj(x) # (time, batch, 2*channels) - x, s = x.chunk(2, dim=2) + x, s = x.chunk(2, dim=-1) s = self.balancer1(s) s = self.sigmoid(s) x = self.activation1(x) # identity. @@ -2190,7 +2190,7 @@ class ConvolutionModule(nn.Module): x = self.in_proj(x) # (time, batch, 2*channels) - x, s = x.chunk(2, dim=-1) + x, s = x.chunk(2, dim=2) s = self.sigmoid(s) x = x * s # (time, batch, channels)