提交 b101be92 authored 作者: James Bergstra's avatar James Bergstra

more testing of convgrad

上级 526044fa
......@@ -307,15 +307,15 @@ class TestConvOp(unittest.TestCase):
bsize = 2
imgs = T.dmatrix('imgs')
kerns = T.dmatrix('kerns')
kshps = [(3,3)]
for mode in 'valid', 'full':
kshps = [(3,3), (5,5), (12,12)]
# 'full' mode should support kernels bigger than the input
if mode == 'full':
kshps.append((12,12))
for mode in 'valid', 'full':
for imshp in (5,5),(2,5,5),(2,10,10): # (12,10), (3,12,11):
# 'full' mode should support kernels bigger than the input
if mode == 'valid' and (kshps[0] > imshp[1]):
continue
visdim = 1 if len(imshp)!=3 else imshp[0]
for kshp in kshps:
imgvals = N.random.random(N.hstack((bsize,imshp)))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论