提交 b94dc4a5 authored 作者: Sina Honari's avatar Sina Honari

updating the st != ds case

上级 4fe11a7a
......@@ -236,7 +236,7 @@ class DownsampleFactorMax(Op):
gz, = grads
maxout = self(x)
if self.st != self.ds:
return theano.gradient.grad_not_implemented
return theano.gradient.grad_not_implemented()
return [DownsampleFactorMaxGrad(self.ds,
ignore_border=self.ignore_border,
st=self.st)(
......@@ -387,7 +387,7 @@ class DownsampleFactorMaxGrad(Op):
x, maxout, gz = inp
ggx, = grads
if self.st != self.ds:
return theano.gradient.grad_not_implemented
return theano.gradient.grad_not_implemented()
return [theano.tensor.zeros_like(x),
theano.tensor.zeros_like(maxout),
DownsampleFactorMaxGradGrad(
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论