提交 c89d9d6f authored 作者: sentient07's avatar sentient07

Simplification

上级 c749e68e
...@@ -26,7 +26,7 @@ class TestDnnConv2d(test_abstract_conv.BaseTestConv2d): ...@@ -26,7 +26,7 @@ class TestDnnConv2d(test_abstract_conv.BaseTestConv2d):
raise SkipTest(dnn_available.msg) raise SkipTest(dnn_available.msg)
mode = mode_with_gpu mode = mode_with_gpu
o = self.get_output_shape(i, f, s, b, (1,1)) o = self.get_output_shape(i, f, s, b, (1, 1))
self.run_fwd(inputs_shape=i, filters_shape=f, subsample=s, self.run_fwd(inputs_shape=i, filters_shape=f, subsample=s,
verify_grad=True, mode=mode, verify_grad=True, mode=mode,
provide_shape=provide_shape, border_mode=b, provide_shape=provide_shape, border_mode=b,
......
...@@ -67,7 +67,7 @@ def get_conv_output_shape(image_shape, kernel_shape, ...@@ -67,7 +67,7 @@ def get_conv_output_shape(image_shape, kernel_shape,
bsize, imshp = image_shape[0], image_shape[2:] bsize, imshp = image_shape[0], image_shape[2:]
nkern, kshp = kernel_shape[0], kernel_shape[2:] nkern, kshp = kernel_shape[0], kernel_shape[2:]
if not filter_dilation: if not filter_dilation:
filter_dilation = numpy.ones(numpy.asarray(subsample).shape, dtype='int') filter_dilation = numpy.ones(len(subsample), dtype='int')
if isinstance(border_mode, tuple): if isinstance(border_mode, tuple):
out_shp = tuple(get_conv_shape_1axis( out_shp = tuple(get_conv_shape_1axis(
imshp[i], kshp[i], border_mode[i], imshp[i], kshp[i], border_mode[i],
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论