From ef6966154952f79cc08f9a9e80cf17e0c72c9f35 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Sun, 19 Sep 2021 21:18:00 +0800 Subject: [PATCH] Change some defaults.. --- egs/librispeech/ASR/conformer_ctc_bn_2d/conformer.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/egs/librispeech/ASR/conformer_ctc_bn_2d/conformer.py b/egs/librispeech/ASR/conformer_ctc_bn_2d/conformer.py index 3a667ce15..448447007 100644 --- a/egs/librispeech/ASR/conformer_ctc_bn_2d/conformer.py +++ b/egs/librispeech/ASR/conformer_ctc_bn_2d/conformer.py @@ -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: