提交 b6febcb9 authored 作者: Frederic's avatar Frederic

Remove warning as it work now!

上级 03ed5920
......@@ -1288,11 +1288,10 @@ def local_conv_gemm(node):
if (isinstance(node.op, GpuConv) and
node.op.border_mode == 'valid' and
node.op.subsample == (1, 1)):
print "WARNING, YOU ARE USING BUGGED CODE!"
img, kern = node.inputs
img = gpu_contiguous(img)
kern = kern[:, :, ::-1, ::-1]
kern = gpu_contiguous(kern)
kern = gpu_contiguous(kern)
return [GpuConvMM(node.op.border_mode)(img, kern)]
gpu_optimizer.register("conv_gemm", local_conv_gemm)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论