From 3122637266a7e7e134b15af37a3faee6fe0fd7a3 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Sat, 17 Sep 2022 13:18:58 +0800 Subject: [PATCH] Use ScaledLinear where I previously had StructuredLinear --- egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py b/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py index cf2f05999..63adfa792 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py @@ -1020,7 +1020,7 @@ class Conv2dSubsampling(nn.Module): DoubleSwish(), ) out_height = (((in_channels - 1) // 2 - 1) // 2) - self.out = nn.Linear(out_height * layer3_channels, out_channels) + self.out = ScaledLinear(out_height * layer3_channels, out_channels) # set learn_eps=False because out_norm is preceded by `out`, and `out` # itself has learned scale, so the extra degree of freedom is not # needed.