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

Fix test condition

Browse Source
This commit is contained in:
Daniel Povey 2022-12-16 12:24:54 +08:00
parent 53ab18a862
commit e84f525840
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/pruned_transducer_stateless7/zipformer.py
View File

@ -1730,7 +1730,7 @@ class ConvNeXt(nn.Module):
The returned value has the same shape as x.
"""
if (not self.training) or torch.jit.is_scripting() or random.random() < float(self.layerdrop_prob):
if torch.jit.is_scripting() or (self.training and random.random() < float(self.layerdrop_prob)):
return x
bypass = x