提交 150cb11d authored 作者: João Victor Risso's avatar João Victor Risso

Raise error tolerances to 1e-3 in spatial transformer grad test

上级 90e31adb
......@@ -2509,4 +2509,5 @@ def test_dnn_spatialtf_grad():
out = dnn.dnn_spatialtf(inputs, theta)
return out
utt.verify_grad(grad_functor, [inputs_val, theta_val], mode=mode_with_gpu)
utt.verify_grad(grad_functor, [inputs_val, theta_val], mode=mode_with_gpu,
abs_tol=1e-3, rel_tol=1e-3)
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论