提交 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,
filter_flip=filter_flip,
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):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论