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

Use default tolerances in assert when testing the spatial transformer

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