mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Decrease random_prob from 0.5 to 0.333

Browse Source
This commit is contained in:
Daniel Povey 2022-09-29 13:55:33 +08:00
parent 461ad3655a
commit d398f0ed70
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py
View File

@ -311,7 +311,7 @@ class ConformerEncoder(nn.Module):
self.combiner = AttentionCombine( self.combiner = AttentionCombine(
num_channels=encoder_layer.d_model, num_channels=encoder_layer.d_model,
num_inputs=len(self.aux_layers), num_inputs=len(self.aux_layers),
random_prob=0.5, random_prob=0.333,
) )
def forward( def forward(
@ -1039,7 +1039,7 @@ class AttentionCombine(nn.Module):
self, self,
num_channels: int, num_channels: int,
num_inputs: int, num_inputs: int,
random_prob: float = 0.5, random_prob: float = 0.333,
) -> None: ) -> None:
""" """
Args: Args: