mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-12-11 06:55:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Bug fix

Browse Source
This commit is contained in:
Daniel Povey 2022-09-28 20:58:34 +08:00
parent df795912ed
commit e5666628bd
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py
View File

@ -474,7 +474,6 @@ class AttentionDownsample(torch.nn.Module):
Require out_channels > in_channels. Require out_channels > in_channels.
""" """
super(AttentionDownsample, self).__init__() super(AttentionDownsample, self).__init__()
assert out_channels > in_channels
self.query = nn.Parameter(torch.randn(in_channels) * (in_channels ** -0.5)) self.query = nn.Parameter(torch.randn(in_channels) * (in_channels ** -0.5))
# fill in the extra dimensions with a projection of the input # fill in the extra dimensions with a projection of the input