提交 675fe64f authored 作者: fay's avatar fay

pool opt

上级 1bb77277
...@@ -644,8 +644,8 @@ class PoolGrad(OpenMPOp): ...@@ -644,8 +644,8 @@ class PoolGrad(OpenMPOp):
class MaxPoolGrad(PoolGrad): class MaxPoolGrad(PoolGrad):
def __init__(self, ds, ignore_border, st=None, padding=(0, 0), openmp=None): def __init__(self, ds, ignore_border, st=None, padding=(0, 0), mode='max', openmp=None):
PoolGrad.__init__(self, ds, ignore_border, st, padding, mode='max', openmp=openmp) PoolGrad.__init__(self, ds, ignore_border, st, padding, mode, openmp)
def make_node(self, x, maxout, gz): def make_node(self, x, maxout, gz):
# make_node should only be called by the grad function of # make_node should only be called by the grad function of
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论