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

Increase grad_scale and prob in score_balancer

Browse Source
This commit is contained in:
Daniel Povey 2023-05-29 13:20:07 +08:00
parent 42f3ad0a11
commit f05f1a6353
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
egs/libriheavy/LM/zipformer1/subformer.py
View File

@ -875,7 +875,10 @@ class LearnedDownsamplingModule(nn.Module):
min_positive=1/(2*downsampling_factor),
max_positive=0.6,
min_abs=1.0,
max_abs=4.0)
max_abs=4.0,
grad_scale=0.1,
prob=ScheduledFloat((0.0, 1.0), (8000.0, 0.25), default=0.0))
# below are for diagnostics.
self.copy_weights1 = nn.Identity()