提交 1dc69e67 authored 作者: Frederic's avatar Frederic

Fix crash during the memory profiling of GpuConv.

上级 24e1117e
...@@ -627,7 +627,7 @@ class GpuConv(GpuOp): ...@@ -627,7 +627,7 @@ class GpuConv(GpuOp):
out, = outputs out, = outputs
assert images[1] == kerns[1] assert images[1] == kerns[1]
flops = 0 flops = 0
if self.out_mode == "valid": if self.border_mode == "valid":
# nb mul and add by output pixel # nb mul and add by output pixel
flops = kerns[2] * kerns[3] * 2 flops = kerns[2] * kerns[3] * 2
# nb flops by output image # nb flops by output image
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论