mirror of
https://github.com/k2-fsa/icefall.git
synced 2025-08-26 18:24:18 +00:00
Fix errors.
This commit is contained in:
parent
cafd06e909
commit
e47fab29a5
@ -716,7 +716,7 @@ def test_layernorm_gru_forward(device="cpu"):
|
||||
|
||||
s_state_sum.backward()
|
||||
t_state_sum.backward()
|
||||
assert_allclose(x.grad, x_clone.grad, atol=1e-4)
|
||||
assert_allclose(x.grad, x_clone.grad, atol=1e-2)
|
||||
|
||||
|
||||
def _test_lstm(device):
|
||||
|
Loading…
x
Reference in New Issue
Block a user