提交 55655580 authored 作者: Frederic Bastien's avatar Frederic Bastien

Make test in python 32bit. sum, average cause small difference result ~1e-17

上级 5c022322
...@@ -663,9 +663,7 @@ class TestDownsampleFactorMax(utt.InferShapeTester): ...@@ -663,9 +663,7 @@ class TestDownsampleFactorMax(utt.InferShapeTester):
output = pool_2d(images, maxpoolshp, ignore_border, output = pool_2d(images, maxpoolshp, ignore_border,
mode=mode) mode=mode)
output_val = function([images], output)(imval) output_val = function([images], output)(imval)
assert numpy.all(output_val == numpy_output_val), ( utt.assert_allclose(output_val, numpy_output_val)
"output_val is %s, numpy_output_val is %s"
% (output_val, numpy_output_val))
def mp(input): def mp(input):
return pool_2d(input, maxpoolshp, ignore_border, return pool_2d(input, maxpoolshp, ignore_border,
...@@ -686,12 +684,7 @@ class TestDownsampleFactorMax(utt.InferShapeTester): ...@@ -686,12 +684,7 @@ class TestDownsampleFactorMax(utt.InferShapeTester):
patch_size = (2, 2) patch_size = (2, 2)
op = max_pool_2d_same_size(input, patch_size) op = max_pool_2d_same_size(input, patch_size)
op_output = function([input], op)(test_input_array) op_output = function([input], op)(test_input_array)
assert numpy.all(op_output == test_answer_array), ( utt.assert_allclose(op_output, test_answer_array)
"op_output is %s, test_answer_array is %s" % (
op_output, test_answer_array
)
)
def mp(input): def mp(input):
return max_pool_2d_same_size(input, patch_size) return max_pool_2d_same_size(input, patch_size)
...@@ -716,9 +709,7 @@ class TestDownsampleFactorMax(utt.InferShapeTester): ...@@ -716,9 +709,7 @@ class TestDownsampleFactorMax(utt.InferShapeTester):
output = pool_2d(images, maxpoolshp, ignore_border, output = pool_2d(images, maxpoolshp, ignore_border,
mode=mode) mode=mode)
output_val = function([images], output)(imval) output_val = function([images], output)(imval)
assert numpy.all(output_val == numpy_output_val), ( utt.assert_allclose(output_val, numpy_output_val)
"output_val is %s, numpy_output_val is %s"
% (output_val, numpy_output_val))
# removed as already tested in test_max_pool_2d_2D # removed as already tested in test_max_pool_2d_2D
# This make test in debug mode too slow. # This make test in debug mode too slow.
...@@ -745,7 +736,7 @@ class TestDownsampleFactorMax(utt.InferShapeTester): ...@@ -745,7 +736,7 @@ class TestDownsampleFactorMax(utt.InferShapeTester):
output = pool_2d(images, maxpoolshp, ignore_border, output = pool_2d(images, maxpoolshp, ignore_border,
mode=mode) mode=mode)
output_val = function([images], output)(imval) output_val = function([images], output)(imval)
assert numpy.all(output_val == numpy_output_val) utt.assert_allclose(output_val, numpy_output_val)
# removed as already tested in test_max_pool_2d_2D # removed as already tested in test_max_pool_2d_2D
# This make test in debug mode too slow. # This make test in debug mode too slow.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论