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

Apply reverse_cutoff with param_pow, to make it not too strong.

Browse Source
This commit is contained in:
Daniel Povey 2022-06-26 11:07:08 +08:00
parent 0aa5a334d6
commit 0b811546f3
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/optim.py
View File

@ -98,7 +98,7 @@ class NeutralGradient(Optimizer):
param_eps=1.0e-06,
param_rel_eps=1.0e-04,
param_rel_max=1.0,
param_reverse_cutoff=16.0,
param_reverse_cutoff=4.0,
param_max_rms=2.0,
param_min_rms=1.0e-05,
max_fullcov_size=1023,
@ -863,8 +863,8 @@ class NeutralGradient(Optimizer):
# 2nd factor below becomes a 1/param_diag type of function when
# param_diag >> param_reverse_cutoff.
ans = (ans ** param_pow) * (param_reverse_cutoff / (param_diag + param_reverse_cutoff))
return ans
ans = ans * (param_reverse_cutoff / (param_diag + param_reverse_cutoff))
return ans ** param_pow
def _estimate_proj(self,