提交 2c61796f authored 作者: João Victor Risso's avatar João Victor Risso

Use separate error tolerances for float32 and float16 in spatial transformer grad test

上级 92e772f9
......@@ -2511,8 +2511,10 @@ def test_dnn_spatialtf_grad():
return out
atol, rtol = None, None
if theano.config.floatX == 'float16' or theano.config.floatX == 'float32':
atol, rtol = 5e-2, 5e-2
if theano.config.floatX == 'float32':
rtol = 5e-2
elif theano.config.floatX == 'float16':
rtol = 1e-0
utt.verify_grad(grad_functor, [inputs_val, theta_val], mode=mode_with_gpu,
abs_tol=atol, rel_tol=rtol)
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论