提交 65d17342 authored 作者: Gijs van Tulder's avatar Gijs van Tulder

Fix bug in conv3d_grad_wrt_weights.

上级 e685f292
...@@ -1173,7 +1173,7 @@ def conv3d_grad_wrt_weights(input, ...@@ -1173,7 +1173,7 @@ def conv3d_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[:-3]) return gradWeight_op(input, output_grad, filter_shape[-3:])
def bilinear_kernel_2D(ratio, normalize=True): def bilinear_kernel_2D(ratio, normalize=True):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论