提交 7090cf73 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

In pooling, compare the strides with the window size, not 1

上级 647e4a44
......@@ -1944,7 +1944,7 @@ def local_gpu_downsample_factor_max_grad_grad(node):
'padding', 'mode')
if (node.op.padding != (0, 0) or
node.op.mode != 'max' or
node.op.st != (1, 1)):
node.op.st != node.op.ds):
return
x, z, gx = node.inputs
if (x.owner and isinstance(x.owner.op, HostFromGpu)):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论