mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-12 19:42:19 +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,
dim_feedforward: int = 2048,
num_trunk_encoder_layers: int = 12,
num_ctc_encoder_layers: int = 4,
num_ctc_encoder_layers: int = 2,
num_decoder_layers: int = 6,
num_reverse_encoder_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,
cnn_module_kernel: int = 31,
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
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:
positive_embed_shifted: