Skip to content

P
Paddle

PaddlePaddle / Paddle
大约 1 年 前同步成功

通知 2299
Star 20931
Fork 5422
P
Paddle
提交 08a7b1de 编写于 作者: C chengduoZH
浏览文件
操作

fix unit test

上级 91db457f
显示空白变更内容
内联 并排
Showing with 3 addition and 2 deletion
python/paddle/v2/framework/tests/test_conv3d_op.py
浏览文件 @ 08a7b1de
...@@ -65,6 +65,7 @@ class TestConv3dOp(OpTest): ...@@ -65,6 +65,7 @@ class TestConv3dOp(OpTest):
self.check_grad( self.check_grad(
set(['Input', 'Filter']), 'Output', max_relative_error=0.05) set(['Input', 'Filter']), 'Output', max_relative_error=0.05)
def test_check_grad_no_filter(self):
self.check_grad( self.check_grad(
['Input'], ['Input'],
'Output', 'Output',
...@@ -81,7 +82,7 @@ class TestConv3dOp(OpTest): ...@@ -81,7 +82,7 @@ class TestConv3dOp(OpTest):
def init_test_case(self): def init_test_case(self):
self.pad = [0, 0, 0] self.pad = [0, 0, 0]
self.stride = [1, 1, 1] self.stride = [1, 1, 1]
self.input_size = [2, 3, 5, 5, 5] # NCDHW self.input_size = [2, 3, 4, 4, 4] # NCDHW
assert np.mod(self.input_size[1], self.groups) == 0 assert np.mod(self.input_size[1], self.groups) == 0
f_c = self.input_size[1] / self.groups f_c = self.input_size[1] / self.groups
self.filter_size = [6, f_c, 3, 3, 3] self.filter_size = [6, f_c, 3, 3, 3]
...@@ -97,7 +98,7 @@ class TestCase1(TestConv3dOp): ...@@ -97,7 +98,7 @@ class TestCase1(TestConv3dOp):
def init_test_case(self): def init_test_case(self):
self.pad = [1, 1, 1] self.pad = [1, 1, 1]
self.stride = [1, 1, 1] self.stride = [1, 1, 1]
self.input_size = [2, 3, 5, 5, 5] # NCDHW self.input_size = [2, 3, 4, 4, 4] # NCDHW
assert np.mod(self.input_size[1], self.groups) == 0 assert np.mod(self.input_size[1], self.groups) == 0
f_c = self.input_size[1] / self.groups f_c = self.input_size[1] / self.groups
self.filter_size = [6, f_c, 3, 3, 3] self.filter_size = [6, f_c, 3, 3, 3]
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或