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 bug RE self.training

Browse Source
This commit is contained in:
Daniel Povey 2022-10-05 15:34:39 +08:00
parent 81542832bf
commit 61f62837fa
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/conformer.py
View File

@ -348,7 +348,7 @@ class ConformerEncoder(nn.Module):
def get_random_mask(): def get_random_mask():
# 1.0 means don't drop the layer, 0.0 means drop the layer # 1.0 means don't drop the layer, 0.0 means drop the layer
mask = torch.ones(num_layers, device='cpu') mask = torch.ones(num_layers, device='cpu')
if self.training: if not self.training:
return mask return mask
r = rng.random() r = rng.random()
if r < 0.1: if r < 0.1: