mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
Fix test condition
This commit is contained in:
parent
53ab18a862
commit
e84f525840
@ -1730,7 +1730,7 @@ class ConvNeXt(nn.Module):
|
|||||||
|
|
||||||
The returned value has the same shape as x.
|
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
|
return x
|
||||||
|
|
||||||
bypass = x
|
bypass = x
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user