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

Prevent crash due to error in C_diff

Browse Source
This commit is contained in:
Daniel Povey 2022-06-17 12:35:55 +08:00
parent 6c499dcd66
commit 2fe4af8c99
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
egs/librispeech/ASR/pruned_transducer_stateless7/optim.py
View File

@ -709,7 +709,9 @@ class NeutralGradient(Optimizer):
C_diff = C_check - C C_diff = C_check - C
# Roundoff can cause significant differences, so use a fairly large # Roundoff can cause significant differences, so use a fairly large
# threshold of 0.001. We may increase this later or even remove the check. # threshold of 0.001. We may increase this later or even remove the check.
assert C_diff.abs().mean() < 0.01 * C.diag().mean() if not C_diff.abs().mean() < 0.01 * C.diag().mean():
print("Warning: large C_diff: {C_diff.abs().mean()}, C diag mean: {C.diag().mean()}")
return P return P