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

change condition order

Browse Source
This commit is contained in:
yaozengwei 2024-10-12 19:05:23 +08:00
parent d186216025
commit 27198a32ac
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() or not self.training: 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":