From bba454a34903ab6f3eb92ea3a113d2de75c54570 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Sat, 29 Oct 2022 17:08:19 +0800 Subject: [PATCH] Make decoder group size equal to 4. --- egs/librispeech/ASR/pruned_transducer_stateless7/decoder.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/decoder.py b/egs/librispeech/ASR/pruned_transducer_stateless7/decoder.py index 3875ab9d6..712dc8ce1 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/decoder.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/decoder.py @@ -69,7 +69,7 @@ class Decoder(nn.Module): out_channels=decoder_dim, kernel_size=context_size, padding=0, - groups=decoder_dim, + groups=decoder_dim//4, # group size == 4 bias=False, )