提交 b6a9d268 authored 作者: Kashif Rasul's avatar Kashif Rasul

fix flake8 warnings

上级 af7ca865
...@@ -454,8 +454,7 @@ def test_pooling_opt(): ...@@ -454,8 +454,7 @@ def test_pooling_opt():
f = theano.function( f = theano.function(
[x], [x],
max_pool_2d(x, ds=(2, 2), mode='average_inc_pad', max_pool_2d(x, ds=(2, 2), mode='average_inc_pad', ignore_border=True),
ignore_border=True),
mode=mode_with_gpu) mode=mode_with_gpu)
assert any([isinstance(n.op, cuda.dnn.GpuDnnPool) assert any([isinstance(n.op, cuda.dnn.GpuDnnPool)
......
...@@ -294,7 +294,8 @@ def test_pooling_opt(): ...@@ -294,7 +294,8 @@ def test_pooling_opt():
f = theano.function( f = theano.function(
[x], [x],
T.grad(max_pool_2d(x, ds=(2, 2), mode='average_inc_pad', T.grad(max_pool_2d(x, ds=(2, 2), mode='average_inc_pad',
ignore_border=True).sum(), x), ignore_border=True).sum(),
x),
mode=mode_with_gpu.including("cudnn")) mode=mode_with_gpu.including("cudnn"))
assert any([isinstance(n.op, dnn.GpuDnnPoolGrad) assert any([isinstance(n.op, dnn.GpuDnnPoolGrad)
...@@ -302,6 +303,7 @@ def test_pooling_opt(): ...@@ -302,6 +303,7 @@ def test_pooling_opt():
f(numpy.zeros((10, 10), dtype='float32')) f(numpy.zeros((10, 10), dtype='float32'))
def test_dnn_tag(): def test_dnn_tag():
""" """
Test that if cudnn isn't avail we crash and that if it is avail, we use it. Test that if cudnn isn't avail we crash and that if it is avail, we use it.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论