mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Bug fix in scripting mode

Browse Source
This commit is contained in:
Daniel Povey 2022-12-02 20:28:17 +08:00
parent 9a2a58e20d
commit 84f51ab1b1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py
View File

@ -1272,8 +1272,8 @@ class Swoosh(torch.nn.Module):
"""Return tan-swish activation function which is tanh(x) sigmoid(x-1)n
"""
if torch.jit.is_scripting():
one = torch.tensor(1.0, dtype=x.dtype, device=x.device)
return torch.logaddexp(one, x - 1.125) - 0.08 * x - 0.3
zero = torch.tensor(0.0, dtype=x.dtype, device=x.device)
return torch.logaddexp(zero, x - 1.125) - 0.08 * x - 0.3
return SwooshFunction.apply(x)