提交 c50b91c4 authored 作者: Nicolas Ballas's avatar Nicolas Ballas

fix conv2d_grad_wrt_weights bug

上级 35e54bb7
...@@ -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 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论