提交 a1495e81 authored 作者: Frederic Bastien's avatar Frederic Bastien

use the ConvOp.verbose flag in GpuConv.

上级 31b44f44
......@@ -130,7 +130,8 @@ class GpuConv(Op):
subsample=(1,1),
logical_img_hw=None,
logical_kern_hw=None,
logical_kern_align_top=True):
logical_kern_align_top=True,
verbose=0):
self.border_mode = border_mode
self.subsample = subsample
if logical_img_hw is not None:
......@@ -146,6 +147,7 @@ class GpuConv(Op):
# grid
self.logical_kern_hw = tuple(logical_kern_hw)
self.logical_kern_align_top = logical_kern_align_top
self.verbose=verbose
def __eq__(self, other):
return type(self) == type(other) \
......@@ -188,7 +190,7 @@ class GpuConv(Op):
logical_img_shape=self.logical_img_hw,
logical_kern_shape=self.logical_kern_hw,
kern_align=self.logical_kern_align_top,
verbose=0)
verbose=self.verbose)
class GpuDownsampleFactorMax(Op):
def __init__(self, ds, ignore_border=False):
......
......@@ -301,7 +301,8 @@ def local_gpu_conv(node):
subsample=(op.dx, op.dy),
logical_img_hw=op.imshp_logical[1:3],
logical_kern_hw=op.kshp_logical,
logical_kern_align_top=op.kshp_logical_top_aligned
logical_kern_align_top=op.kshp_logical_top_aligned,
verbose=op.verbose
)
#HACK to print the number of MFlops in the profiler output.
if hasattr(op,'flops'):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论