mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
Increase feature_mask_dropout_prob to 0.125
This commit is contained in:
parent
c2e39bd488
commit
3dd25d6b2d
@ -306,7 +306,7 @@ class Zipformer2(EncoderInterface):
|
||||
# we divide the dropped-out feature dimensions into two equal groups;
|
||||
# the first group is dropped out with probability 0.1, the second
|
||||
# with probability approximately twice that.
|
||||
feature_mask_dropout_prob = 0.1
|
||||
feature_mask_dropout_prob = 0.125
|
||||
|
||||
# frame_mask_max1 shape: (num_frames_max, batch_size, 1)
|
||||
frame_mask_max1 = (torch.rand(num_frames_max, batch_size, 1,
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user