提交 26fe6416 authored 作者: Frederic Bastien's avatar Frederic Bastien

make test of convop faster by removing redondent case.

上级 340b0e18
...@@ -437,8 +437,7 @@ class TestConvOp(unittest.TestCase): ...@@ -437,8 +437,7 @@ class TestConvOp(unittest.TestCase):
kshps = [(3,4)] kshps = [(3,4)]
imshps = [(2,8,7)] imshps = [(2,8,7)]
modes = ['valid', 'full'] modes = ['valid', 'full']
unroll_batch=[0,1,3] unroll = [(0,0),(1,1),(1,4),(3,1),(3,4)]
unroll_kern=[0,1,4]
ssizes = [(1,1),(2,2)] ssizes = [(1,1),(2,2)]
for typ in types: for typ in types:
...@@ -452,8 +451,7 @@ class TestConvOp(unittest.TestCase): ...@@ -452,8 +451,7 @@ class TestConvOp(unittest.TestCase):
# 'full' mode should support kernels bigger than the input # 'full' mode should support kernels bigger than the input
if mode == 'valid' and (t<0).any(): if mode == 'valid' and (t<0).any():
continue continue
for un_b in unroll_batch: for un_b,un_k in unroll:
for un_k in unroll_kern:
for ss in ssizes: for ss in ssizes:
imgvals = N.array(N.random.random(N.hstack((bsize,imshp))),dtype=imgs.dtype) imgvals = N.array(N.random.random(N.hstack((bsize,imshp))),dtype=imgs.dtype)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论