提交 4b6c9fda authored 作者: Frederic Bastien's avatar Frederic Bastien

lower the precission of the test of the gradient of ConvOp only when dx or dy !=1.

上级 07061384
......@@ -550,8 +550,8 @@ class TestConvOp(unittest.TestCase):
print mode, imshp, kshp, un_b, un_k, ss
#TODO the tolerance needed to pass is very high for float32(0.17). Is this acceptable? Expected?
tol = None
if typ=="float32":
tol = 0.17
if typ=="float32" and (ss[0]!=1 or ss[1]!=1):
tol = 0.1
utt.verify_grad(test_i, [imgvals],
cast_to_output_type=True,
tol=tol)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论