提交 d6651735 authored 作者: Frederic's avatar Frederic

Make test run on the GPU as it should.

上级 87a5ccc0
...@@ -869,5 +869,5 @@ def test_stack_rows_segfault_070312(): ...@@ -869,5 +869,5 @@ def test_stack_rows_segfault_070312():
out = theano.shared(numpy.random.rand(1, 2, 2, 3).astype('float32')) out = theano.shared(numpy.random.rand(1, 2, 2, 3).astype('float32'))
op = theano.tensor.nnet.conv.ConvOp(imshp=(80, 96, 96), kshp=(9, 9), op = theano.tensor.nnet.conv.ConvOp(imshp=(80, 96, 96), kshp=(9, 9),
nkern=1, bsize=1) nkern=1, bsize=1)
f = theano.function([], [], updates=[(out, op(img, kern))]) f = theano.function([], [], updates=[(out, op(img, kern))], mode=theano_mode)
f() f()
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论