From 45f872c27da3d52dee592552435a46f7ae2cd374 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Fri, 1 Apr 2022 19:33:20 +0800 Subject: [PATCH] Remove final dropout --- egs/librispeech/ASR/pruned_transducer_stateless2/conformer.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless2/conformer.py b/egs/librispeech/ASR/pruned_transducer_stateless2/conformer.py index 83de82056..7573addaa 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless2/conformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless2/conformer.py @@ -80,7 +80,6 @@ class Conformer(EncoderInterface): ) self.encoder = ConformerEncoder(encoder_layer, num_encoder_layers) - self.final_dropout = nn.Dropout(p=dropout) if output_dim == d_model: self.encoder_output_layer = nn.Identity() else: @@ -119,7 +118,6 @@ class Conformer(EncoderInterface): x = self.encoder(x, pos_emb, src_key_padding_mask=mask, warmup=warmup) # (T, N, C) - x = self.final_dropout(x) x = self.encoder_output_layer(x) x = x.permute(1, 0, 2) # (T, N, C) ->(N, T, C)