提交 11e3b03a authored 作者: Frederic's avatar Frederic

Disable cudnn pooling as we have problems with borders.

上级 40f47bae
...@@ -1122,8 +1122,8 @@ if cuda_available: ...@@ -1122,8 +1122,8 @@ if cuda_available:
subsample = node.op.subsample subsample = node.op.subsample
return [dnn_conv(gpu_contiguous(img), gpu_contiguous(kern), return [dnn_conv(gpu_contiguous(img), gpu_contiguous(kern),
border_mode=border_mode, subsample=subsample)] border_mode=border_mode, subsample=subsample)]
# DISABLED as there is problems in the handling of borders
@register_opt('cudnn') # @register_opt('cudnn')
@local_optimizer([GpuDownsampleFactorMax]) @local_optimizer([GpuDownsampleFactorMax])
def local_pool_dnn(node): def local_pool_dnn(node):
if not dnn_available(): if not dnn_available():
...@@ -1135,7 +1135,8 @@ if cuda_available: ...@@ -1135,7 +1135,8 @@ if cuda_available:
ds = node.op.ds ds = node.op.ds
return [dnn_pool(gpu_contiguous(img), ds, ds)] return [dnn_pool(gpu_contiguous(img), ds, ds)]
@register_opt('cudnn') # DISABLED as there is problems in the handling of borders
# @register_opt('cudnn')
@local_optimizer([GpuDownsampleFactorMaxGrad]) @local_optimizer([GpuDownsampleFactorMaxGrad])
def local_pool_dnn_grad(node): def local_pool_dnn_grad(node):
if not dnn_available(): if not dnn_available():
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论