mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-13 20:12:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Change some defaults..

Browse Source
This commit is contained in:
Daniel Povey 2021-09-19 21:18:00 +08:00
parent b0dd4215fe
commit ef69661549
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
egs/librispeech/ASR/conformer_ctc_bn_2d/conformer.py
View File

@ -186,11 +186,11 @@ class BidirectionalConformer(nn.Module):
nhead: int = 4, nhead: int = 4,
dim_feedforward: int = 2048, dim_feedforward: int = 2048,
num_trunk_encoder_layers: int = 12, num_trunk_encoder_layers: int = 12,
num_ctc_encoder_layers: int = 4, num_ctc_encoder_layers: int = 2,
num_decoder_layers: int = 6, num_decoder_layers: int = 6,
num_reverse_encoder_layers: int = 4, num_reverse_encoder_layers: int = 4,
num_reverse_decoder_layers: int = 4, num_reverse_decoder_layers: int = 4,
num_self_predictor_layers: int = 3, num_self_predictor_layers: int = 2,
dropout: float = 0.1, dropout: float = 0.1,
cnn_module_kernel: int = 31, cnn_module_kernel: int = 31,
is_bpe: bool = False, is_bpe: bool = False,
@ -537,7 +537,8 @@ class BidirectionalConformer(nn.Module):
""" """
This is the reverse decoder function, which returns the total probability of the This is the reverse decoder function, which returns the total probability of the
labels sampled in the discrete bottleneck layer, as predicted from the labels sampled in the discrete bottleneck layer, as predicted from the
supervision word-sequence. supervision word-sequence. Caution: it has the opposite sign from
the result of decoder_forward().
Args: Args:
positive_embed_shifted: positive_embed_shifted: