mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
Decrease random_prob from 0.5 to 0.333
This commit is contained in:
parent
461ad3655a
commit
d398f0ed70
@ -311,7 +311,7 @@ class ConformerEncoder(nn.Module):
|
||||
self.combiner = AttentionCombine(
|
||||
num_channels=encoder_layer.d_model,
|
||||
num_inputs=len(self.aux_layers),
|
||||
random_prob=0.5,
|
||||
random_prob=0.333,
|
||||
)
|
||||
|
||||
def forward(
|
||||
@ -1039,7 +1039,7 @@ class AttentionCombine(nn.Module):
|
||||
self,
|
||||
num_channels: int,
|
||||
num_inputs: int,
|
||||
random_prob: float = 0.5,
|
||||
random_prob: float = 0.333,
|
||||
) -> None:
|
||||
"""
|
||||
Args:
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user