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

Use lower tolerance values for float16 in spatial transformer test

上级 150cb11d
...@@ -2469,7 +2469,10 @@ def test_dnn_spatialtf(): ...@@ -2469,7 +2469,10 @@ def test_dnn_spatialtf():
st_cpu_func = theano.function([t_img, t_theta], st_cpu, mode=mode_without_gpu) st_cpu_func = theano.function([t_img, t_theta], st_cpu, mode=mode_without_gpu)
img_out_cpu = st_cpu_func(img, theta) img_out_cpu = st_cpu_func(img, theta)
utt.assert_allclose(img_out_cpu, img_out_gpu) atol, rtol = None, None
if theano.config.floatX == 'float16':
atol, rtol = 1e-0, 1e-2
utt.assert_allclose(img_out_cpu, img_out_gpu, atol=atol, rtol=rtol)
def test_dnn_spatialtf_grad(): def test_dnn_spatialtf_grad():
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论