diff --git a/python/paddle/v2/framework/tests/test_pool2d_op.py b/python/paddle/v2/framework/tests/test_pool2d_op.py index 549898ec94e1d7f81928e47ea6f651f1fc778033..cf327508f710096e0ee3bb20e3cfc7b201d08e88 100644 --- a/python/paddle/v2/framework/tests/test_pool2d_op.py +++ b/python/paddle/v2/framework/tests/test_pool2d_op.py @@ -62,7 +62,8 @@ class TestPool2d_Op(OpTest): self.check_output() def test_check_grad(self): - self.check_grad(set(['Input']), 'Output', max_relative_error=0.07) + if self.pool_type != "max": + self.check_grad(set(['Input']), 'Output', max_relative_error=0.07) def initTestCase(self): self.pool_type = "ave" @@ -84,15 +85,16 @@ class TestCase2(TestPool2d_Op): self.paddings = [1, 1] -# class TestCase1(TestPool2d_Op): -# def initTestCase(self): -# self.op_type = "pool2d" -# self.pool_type = "max" -# self.pool2D_forward_naive = max_pool2D_forward_naive -# self.shape = [2, 3, 5, 5] -# self.ksize = [3, 3] -# self.strides = [1, 1] -# self.paddings = [1, 1] +class TestCase1(TestPool2d_Op): + def initTestCase(self): + self.op_type = "pool2d" + self.pool_type = "max" + self.pool2D_forward_naive = max_pool2D_forward_naive + self.shape = [2, 3, 5, 5] + self.ksize = [3, 3] + self.strides = [1, 1] + self.paddings = [1, 1] + if __name__ == '__main__': unittest.main() diff --git a/python/paddle/v2/framework/tests/test_pool3d_op.py b/python/paddle/v2/framework/tests/test_pool3d_op.py index f8e9a768e03837b257b41f36b5f908bad6ec9357..cfd0ced150bd23a1e751c940c7f5c34a84e1fde6 100644 --- a/python/paddle/v2/framework/tests/test_pool3d_op.py +++ b/python/paddle/v2/framework/tests/test_pool3d_op.py @@ -72,7 +72,8 @@ class TestPool3d_Op(OpTest): self.check_output() def test_check_grad(self): - self.check_grad(set(['Input']), 'Output', max_relative_error=0.07) + if self.pool_type != "max": + self.check_grad(set(['Input']), 'Output', max_relative_error=0.07) def initTestCase(self): self.pool_type = "ave" @@ -94,15 +95,16 @@ class TestCase1(TestPool3d_Op): self.paddings = [1, 1, 1] -# class TestCase2(TestPool3d_Op): -# def initTestCase(self): -# self.op_type = "pool3d" -# self.pool_type = "max" -# self.pool3D_forward_naive = max_pool3D_forward_naive -# self.shape = [2, 3, 5, 5, 5] -# self.ksize = [3, 3, 3] -# self.strides = [1, 1, 1] -# self.paddings = [1, 1, 1] +class TestCase2(TestPool3d_Op): + def initTestCase(self): + self.op_type = "pool3d" + self.pool_type = "max" + self.pool3D_forward_naive = max_pool3D_forward_naive + self.shape = [2, 3, 5, 5, 5] + self.ksize = [3, 3, 3] + self.strides = [1, 1, 1] + self.paddings = [1, 1, 1] + if __name__ == '__main__': unittest.main()