mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-12-11 06:55:27 +00:00
Use ScaledLinear where I previously had StructuredLinear
This commit is contained in:
parent
4a2b940321
commit
3122637266
@ -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.
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user