提交 d051a930 authored 作者: Vikram's avatar Vikram

Minor fixes

上级 b92d858b
...@@ -1648,8 +1648,7 @@ def local_abstractconv_gemm(node): ...@@ -1648,8 +1648,7 @@ def local_abstractconv_gemm(node):
if prod1 > prod2: if prod1 > prod2:
rval = GpuCorrMM_gradWeights(border_mode, rval = GpuCorrMM_gradWeights(border_mode,
subsample, subsample,
filter_dilation, filter_dilation)(
unshared)(
gpu_contiguous(img.dimshuffle(1, 0, 2, 3)), gpu_contiguous(img.dimshuffle(1, 0, 2, 3)),
gpu_contiguous(kern.dimshuffle(1, 0, 2, 3))) gpu_contiguous(kern.dimshuffle(1, 0, 2, 3)))
# (we need to wrap the result in as_gpuarray_variable, # (we need to wrap the result in as_gpuarray_variable,
......
...@@ -300,7 +300,7 @@ def get_conv_gradinputs_shape(kernel_shape, top_shape, ...@@ -300,7 +300,7 @@ def get_conv_gradinputs_shape(kernel_shape, top_shape,
""" """
bsize, topshp = top_shape[0], top_shape[2:] bsize, topshp = top_shape[0], top_shape[2:]
convdim = len(topshp) - 2 convdim = len(top_shape) - 2
nkern, kshp = kernel_shape[1], kernel_shape[-convdim:] nkern, kshp = kernel_shape[1], kernel_shape[-convdim:]
if filter_dilation is None: if filter_dilation is None:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论