mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-08 17:42:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix running exported model on GPU. (#1131)

Browse Source
This commit is contained in:
Fangjun Kuang 2023-06-15 12:25:15 +08:00 committed by GitHub
parent 0ad037d076
commit 947f0614c9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/librispeech/ASR/zipformer/zipformer.py
View File

@ -2190,7 +2190,7 @@ class ConvolutionModule(nn.Module):
x = self.in_proj(x) # (time, batch, 2*channels) 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) s = self.sigmoid(s)
x = x * s x = x * s
# (time, batch, channels) # (time, batch, channels)