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

clean-up

上级 417e6fb7
...@@ -47,7 +47,7 @@ class TestDnnConv2d(test_abstract_conv.BaseTestConv2d): ...@@ -47,7 +47,7 @@ class TestDnnConv2d(test_abstract_conv.BaseTestConv2d):
filter_flip=flip, target_op=GpuDnnConvGradI) filter_flip=flip, target_op=GpuDnnConvGradI)
class TestDnnConv2d(test_abstract_conv.BaseTestConv2d): class TestDebugMode(test_abstract_conv.BaseTestConv2d):
def setUp(self): def setUp(self):
super(TestDnnConv2d, self).setUp() super(TestDnnConv2d, self).setUp()
# provide_shape is not used by the CuDNN impementation # provide_shape is not used by the CuDNN impementation
......
...@@ -638,7 +638,6 @@ class AbstractConv2d_gradWeights(BaseAbstractConv2d): ...@@ -638,7 +638,6 @@ class AbstractConv2d_gradWeights(BaseAbstractConv2d):
new_img = numpy.zeros((img.shape[0], img.shape[1], new_img = numpy.zeros((img.shape[0], img.shape[1],
img.shape[2] + 2 *pad_h, img.shape[2] + 2 *pad_h,
img.shape[3] + 2 * pad_w), dtype=img.dtype) img.shape[3] + 2 * pad_w), dtype=img.dtype)
#import pdb; pdb.set_trace()
new_img[:, :, pad_h:img.shape[2]+pad_h, pad_w:img.shape[3]+pad_w] = img new_img[:, :, pad_h:img.shape[2]+pad_h, pad_w:img.shape[3]+pad_w] = img
img = new_img img = new_img
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论