提交 1af69154 authored 作者: Frederic Bastien's avatar Frederic Bastien

make test faster in debug mode.

上级 4062b0e1
...@@ -1631,7 +1631,7 @@ class T_subtensor(unittest.TestCase): ...@@ -1631,7 +1631,7 @@ class T_subtensor(unittest.TestCase):
self.failUnless(numpy.allclose(gshape, data.shape)) self.failUnless(numpy.allclose(gshape, data.shape))
def fct(t): def fct(t):
return sum(exp(t[idx_])) return sum(t[idx_])
utt.verify_grad(fct, [data]) utt.verify_grad(fct, [data])
if idx is idxs[0]: if idx is idxs[0]:
...@@ -1649,7 +1649,7 @@ class T_subtensor(unittest.TestCase): ...@@ -1649,7 +1649,7 @@ class T_subtensor(unittest.TestCase):
debug_mode = isinstance(theano.compile.mode.get_default_mode(), debug_mode = isinstance(theano.compile.mode.get_default_mode(),
theano.compile.DebugMode) theano.compile.DebugMode)
for i in range(data.shape[0]): for i in range(data.shape[0]):
for j in range(data.shape[0]): for j in range(0,data.shape[0],2):
idxs.append([i,j,(i+1)%data.shape[0]]) idxs.append([i,j,(i+1)%data.shape[0]])
self.grad_list_(idxs, data) self.grad_list_(idxs, data)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论