提交 5b971be5 authored 作者: Frederic Bastien's avatar Frederic Bastien

make test faster in debug mode

上级 7ceab2bc
......@@ -124,9 +124,11 @@ class TestDownsampleFactorMax(unittest.TestCase):
g = tensor.grad(c, images)
g_val = function([images], [g.shape, tensor.min(tensor.min(tensor.min(g))), tensor.max(tensor.max(tensor.max(g)))])(imval)
def mp(input):
return max_pool2D(input, maxpoolshp, ignore_border)
utt.verify_grad(mp, [imval], rng=rng)
#removed as already tested in test_max_pool2D_2D and test_max_pool2D_6D
#This make test in debug mode too slow.
# def mp(input):
# return max_pool2D(input, maxpoolshp, ignore_border)
# utt.verify_grad(mp, [imval], rng=rng)
def test_max_pool2D_6D(self):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论