Swap random_prob and single_prob, to reduce prob of being randomized.

This commit is contained in:
Daniel Povey 2022-10-01 23:50:38 +08:00
parent 8d517a69e4
commit cf5f7e5dfd

View File

@ -1071,8 +1071,8 @@ class AttentionCombine(nn.Module):
self,
num_channels: int,
num_inputs: int,
random_prob: float = 0.333,
single_prob: float = 0.25,
random_prob: float = 0.25,
single_prob: float = 0.333,
) -> None:
"""
Args: