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

Fix weight decay formula by adding 1/1-beta

Browse Source
This commit is contained in:
Daniel Povey 2022-04-05 00:31:28 +08:00
parent 234366e51c
commit 2b0727a355
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
egs/librispeech/ASR/pruned_transducer_stateless2/optim.py
View File

@ -156,9 +156,12 @@ class Eve(Optimizer):
# epsilon = 1.0 * 0.5 * (1.0e-04 / 0.1) = 1.0e-06. # epsilon = 1.0 * 0.5 * (1.0e-04 / 0.1) = 1.0e-06.
# Note that this is close to the "traditional" value used for weight # Note that this is close to the "traditional" value used for weight
# decay. # decay.
#
# this is the weight-decay amount... # this is the weight-decay amount...
weight_decay = (delta ** 2).mean() * (0.5 * (step_size / target_rms) ** 2) #
# Regarding the 1/1-beta factor below: this is to compensate for the deltas on successive
# frames being correlated. I have to figure out the justification.
weight_decay = (delta ** 2).mean() * (0.5 * (step_size / target_rms) ** 2 * (1.0 / (1.0 - beta)))
p.mul_(1 - weight_decay) p.mul_(1 - weight_decay)
p.add_(delta, alpha=-step_size) p.add_(delta, alpha=-step_size)