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

Fix eps value

Browse Source
This commit is contained in:
Daniel Povey 2022-05-19 22:25:18 +08:00
parent 1edc0fa841
commit 8fd9e64fdf
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_stateless4/optim.py
View File

@ -221,7 +221,7 @@ class Abel(Optimizer):
params, params,
lr=1e-3, lr=1e-3,
betas=(0.9, 0.98), betas=(0.9, 0.98),
eps=1e-04, eps=1e-08,
target_rms=0.1, target_rms=0.1,
): ):
@ -691,9 +691,9 @@ class Cain(Optimizer):
def _must_store_stats(self, step: int) -> Tuple[bool, bool]: def _must_store_stats(self, step: int) -> Tuple[bool, bool]:
""" """
Returns (must_store_stats, must_zero_stats), where Returns (must_store_stats, must_zero_stats), where
must_store_stats: bool, is True for the 20 steps preceding must_store_stats: bool, is True for the 100 or 200 steps preceding
a step on which we will estimate the transform. a step on which we will estimate the transform.
must_zero_stats: bool, is True for only the 1st of the 20 must_zero_stats: bool, is True for only the 1st of the 100 or 200
steps mentioned above. steps mentioned above.
""" """
if step < 4000: if step < 4000: