提交 598f485b authored 作者: Frederic's avatar Frederic

warn about bugged code.

上级 c5728f55
...@@ -1259,6 +1259,7 @@ gpu_optimizer.register("conv_fft_full", local_conv_fft_full) ...@@ -1259,6 +1259,7 @@ gpu_optimizer.register("conv_fft_full", local_conv_fft_full)
import theano.tensor.signal.downsample as downsample import theano.tensor.signal.downsample as downsample
@register_opt() @register_opt()
@local_optimizer([downsample.DownsampleFactorMax]) @local_optimizer([downsample.DownsampleFactorMax])
def local_gpu_downsample_factor_max(node): def local_gpu_downsample_factor_max(node):
...@@ -1287,7 +1288,9 @@ def local_conv_gemm(node): ...@@ -1287,7 +1288,9 @@ def local_conv_gemm(node):
if (isinstance(node.op, GpuConv) and if (isinstance(node.op, GpuConv) and
node.op.border_mode == 'valid' and node.op.border_mode == 'valid' and
node.op.subsample == (1, 1)): node.op.subsample == (1, 1)):
return [GpuConvMM(node.op.border_mode)(*node.inputs)] print "WARNING, YOU ARE USING BUGGED CODE!"
return [GpuConvMM(node.op.border_mode)(node.inputs[0],
node.inputs[1],)]
gpu_optimizer.register("conv_gemm", local_conv_gemm) gpu_optimizer.register("conv_gemm", local_conv_gemm)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论