提交 1244f717 authored 作者: Frederic Bastien's avatar Frederic Bastien

Try to silence some failing test in corner case with big input.

上级 8a64516f
......@@ -1296,6 +1296,8 @@ def test_conv3d_bwd():
# Raise tolerance for float16
if theano.config.floatX == 'float16':
rtol = 5e-2
elif max(inputs_shape) > 1024 or max(filters_shape) > 1024:
rtol = 2e-5
utt.assert_allclose(res_ref[0], res[0], rtol=rtol)
utt.assert_allclose(res_ref[1], res[1], rtol=rtol)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论