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

fix a test in FAST_COMPILE mode.

上级 4b1424ec
...@@ -418,7 +418,7 @@ def test_neibs2images_crash_on_grad(): ...@@ -418,7 +418,7 @@ def test_neibs2images_crash_on_grad():
neibs_val = numpy.random.rand(600,4) neibs_val = numpy.random.rand(600,4)
to_images = T.sum(neibs2images(neibs, (2,2), (2,3,20,20))) to_images = T.sum(neibs2images(neibs, (2,2), (2,3,20,20)))
g = T.grad(to_images, neibs) g = T.grad(to_images, neibs)
fn = theano.function([neibs], to_images) fn = theano.function([neibs], to_images, mode=mode_without_gpu)
print "Compiled" print "Compiled"
fn(neibs_val) fn(neibs_val)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论