From 3fdfec1049ddbc5124c6b4a45c880ccb6089c5cd Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Wed, 11 Jan 2023 13:18:08 +0800 Subject: [PATCH] Replace dropout2 on Conv2dSubsampling with Dropout3, share time dim --- egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py b/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py index b2d99c83f..7a77e8cce 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py @@ -2013,7 +2013,7 @@ class Conv2dSubsampling(nn.Module): # max_log_eps=0.0 is to prevent both eps and the output of self.out from # getting large, there is an unnecessary degree of freedom. self.out_norm = BasicNorm(out_channels) - self.dropout = Dropout2(dropout) + self.dropout = Dropout3(dropout, shared_dim=1) def forward(self, x: torch.Tensor) -> torch.Tensor: