提交 754f30fa authored 作者: Frederic Bastien's avatar Frederic Bastien

added a verbose flag to ConvOp that is used the GpuConv

上级 a53ccd27
......@@ -33,7 +33,8 @@ class ConvOp(Op):
unroll_kern=4,
imshp_logical=None,
kshp_logical=None,
kshp_logical_top_aligned=True):
kshp_logical_top_aligned=True,
verbose=0):
"""
......@@ -46,6 +47,7 @@ class ConvOp(Op):
out_mode - 'valid', 'full'
unroll_batch - c code generation option
unroll_kern - c code generation option
verbose - passed to GpuConv
The reason that this op does the summation over convolutions within the 'stack' is that
it allows us to be memory-efficient about how gradients are calculated. If, for
......@@ -76,6 +78,7 @@ class ConvOp(Op):
self.bsize=bsize
self.dx=dx
self.dy=dy
self.verbose=verbose
# a triple
self.imshp_logical = self.imshp if imshp_logical is None else tuple(imshp_logical)
assert len(self.imshp) == len(self.imshp_logical)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论