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

use the ConvOp.verbose flag in GpuConv.

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