mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-09 01:52:41 +00:00
Change max-keep-prob to 0.95
This commit is contained in:
parent
262388134d
commit
2cde99509f
@ -231,7 +231,7 @@ class ConformerEncoderLayer(nn.Module):
|
||||
# compensate for the small scale by just producing larger output.
|
||||
warmup = max(warmup, 0.1)
|
||||
if self.training:
|
||||
warmup = min(warmup, 0.99) # effectively, layer-drop with 1-in-100 prob.
|
||||
warmup = min(warmup, 0.95) # effectively, layer-drop with 1-in-20 prob.
|
||||
alpha = 1.0 if torch.rand(()).item() <= warmup else 0.1
|
||||
|
||||
# macaron style feed forward module
|
||||
|
Loading…
x
Reference in New Issue
Block a user