提交 574753c0 authored 作者: Frederic's avatar Frederic

Fix test following fix to dnn pool opt

上级 5f92bf25
......@@ -178,7 +178,7 @@ def test_dnn_tag():
try:
f = theano.function(
[x],
max_pool_2d(x, ds=(2, 2)),
max_pool_2d(x, ds=(2, 2), ignore_border=True),
mode=mode_with_gpu.including("cudnn"))
except (AssertionError, RuntimeError), e:
assert not cuda.dnn.dnn_available()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论