提交 1a8edc31 authored 作者: Frédéric Bastien's avatar Frédéric Bastien 提交者: GitHub

Merge pull request #4911 from ballasn/gradweightbug

fix conv2d_grad_wrt_weights bug
...@@ -422,7 +422,7 @@ def conv2d_grad_wrt_weights(input, ...@@ -422,7 +422,7 @@ def conv2d_grad_wrt_weights(input,
filter_flip=filter_flip, filter_flip=filter_flip,
filter_dilation=filter_dilation) filter_dilation=filter_dilation)
return gradWeight_op(input, output_grad, filter_shape[:-2]) return gradWeight_op(input, output_grad, filter_shape[-2:])
def bilinear_kernel_2D(ratio, normalize=True): def bilinear_kernel_2D(ratio, normalize=True):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论