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

Fix test mode with random layer dropout

Browse Source
This commit is contained in:
Daniel Povey 2022-03-25 23:28:52 +08:00
parent d2ed3dfc90
commit 0e694739f2
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
egs/librispeech/ASR/pruned_transducer_stateless2/conformer.py
View File

@ -231,7 +231,8 @@ class ConformerEncoderLayer(nn.Module):
# period we sometimes use scale 1.0; this ensures that the modules do not
# compensate for the small scale by just producing larger output.
warmup = max(warmup, 0.1)
warmup = min(warmup, 0.95) # effectively, layer-drop.
if self.training:
warmup = min(warmup, 0.95) # effectively, layer-drop.
alpha = 1.0 if torch.rand(()).item() <= warmup else 0.1
# macaron style feed forward module