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

Fix issue with eval mode in ActivationDropoutLinear (#1770)

Browse Source 
* Fix issue with eval mode in ActivationDropoutLinear

---------

Co-authored-by: Daniel Povey <dpovey@gmail.com>
This commit is contained in:
Zengwei Yao 2024-10-12 19:09:05 +08:00 committed by GitHub
parent d9844d847f
commit fbba712887
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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/scaling.py
View File

@ -1635,7 +1635,7 @@ class ActivationDropoutAndLinear(torch.nn.Module):
self.dropout_shared_dim = dropout_shared_dim self.dropout_shared_dim = dropout_shared_dim
def forward(self, x: Tensor): def forward(self, x: Tensor):
if torch.jit.is_scripting() or torch.jit.is_tracing(): if not self.training or torch.jit.is_scripting() or torch.jit.is_tracing():
if self.activation == "SwooshL": if self.activation == "SwooshL":
x = SwooshLForward(x) x = SwooshLForward(x)
elif self.activation == "SwooshR": elif self.activation == "SwooshR":