From 947f0614c972e96bb211a0399f0328027229fa2b Mon Sep 17 00:00:00 2001 From: Fangjun Kuang Date: Thu, 15 Jun 2023 12:25:15 +0800 Subject: [PATCH] Fix running exported model on GPU. (#1131) --- egs/librispeech/ASR/zipformer/zipformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/zipformer/zipformer.py b/egs/librispeech/ASR/zipformer/zipformer.py index 612356a50..7d98dbeb1 100644 --- a/egs/librispeech/ASR/zipformer/zipformer.py +++ b/egs/librispeech/ASR/zipformer/zipformer.py @@ -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)