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

Bug fix in caching_eval

Browse Source
This commit is contained in:
Daniel Povey 2022-12-15 23:24:36 +08:00
parent d26ee2bf81
commit f5d4fb092d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/librispeech/ASR/pruned_transducer_stateless7/scaling.py
View File

@ -265,7 +265,7 @@ class CachingEvalFunction(torch.autograd.Function):
y2 = m(x)
assert torch.allclose(y, y2, atol=1.0e-02)
# this call to backward() should create grads in the module's parameters
y.backward(gradient=y_grad)
y2.backward(gradient=y_grad)
# restore the state from before we entered this function
random.setstate(random_state)