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

Revert whitening_limit from 1.1 to 2.2.

Browse Source
This commit is contained in:
Daniel Povey 2022-10-16 11:31:08 +08:00
parent 1135669e93
commit ef4650bc8e
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -839,12 +839,12 @@ class RelPositionMultiheadAttention(nn.Module):
# self.whiten_values is applied on the values in forward() # self.whiten_values is applied on the values in forward()
self.whiten_values = Whiten(num_groups=num_heads, self.whiten_values = Whiten(num_groups=num_heads,
whitening_limit=1.1, whitening_limit=2.0,
prob=(0.025, 0.25), prob=(0.025, 0.25),
grad_scale=0.025) grad_scale=0.025)
# self.whiten_keys is applied on the keys in forward() # self.whiten_keys is applied on the keys in forward()
self.whiten_keys = Whiten(num_groups=num_heads, self.whiten_keys = Whiten(num_groups=num_heads,
whitening_limit=1.1, whitening_limit=2.0,
prob=(0.025, 0.25), prob=(0.025, 0.25),
grad_scale=0.025) grad_scale=0.025)
@ -860,7 +860,7 @@ class RelPositionMultiheadAttention(nn.Module):
initial_scale=0.05) initial_scale=0.05)
# self.whiten_values2 is applied on the values in forward2() # self.whiten_values2 is applied on the values in forward2()
self.whiten_values2 = Whiten(num_groups=num_heads, self.whiten_values2 = Whiten(num_groups=num_heads,
whitening_limit=1.1, whitening_limit=2.0,
prob=(0.025, 0.25), prob=(0.025, 0.25),
grad_scale=0.025) grad_scale=0.025)