mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-09 18:12:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Replace ExpScaleRelu with DoubleSwish()

Browse Source
This commit is contained in:
Daniel Povey 2022-03-12 19:00:48 +08:00
parent db7a3b6eea
commit be0a79cbca
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/librispeech/ASR/conformer_ctc/subsampling.py
View File

@ -49,13 +49,13 @@ class Conv2dSubsampling(nn.Module):
), ),
DerivBalancer(channel_dim=1, threshold=0.05, DerivBalancer(channel_dim=1, threshold=0.05,
max_factor=0.01), max_factor=0.01),
ExpScaleRelu(odim, 1, 1, speed=20.0), DoubleSwish(),
ScaledConv2d( ScaledConv2d(
in_channels=odim, out_channels=odim, kernel_size=3, stride=2 in_channels=odim, out_channels=odim, kernel_size=3, stride=2
), ),
DerivBalancer(channel_dim=1, threshold=0.05, DerivBalancer(channel_dim=1, threshold=0.05,
max_factor=0.01), max_factor=0.01),
ExpScaleRelu(odim, 1, 1, speed=20.0), DoubleSwish(),
) )
self.out = ScaledLinear(odim * (((idim - 1) // 2 - 1) // 2), odim) self.out = ScaledLinear(odim * (((idim - 1) // 2 - 1) // 2), odim)
self.out_norm = BasicNorm(odim) self.out_norm = BasicNorm(odim)