未验证 提交 f17bd178 编写于 作者: L lijianshe02 提交者: GitHub

add decorator skip_check_grad_ci for test_group_norm (#21910)

* add decorator skip_check_grad_ci for test_group_norm test=develop
上级 feabed13
...@@ -19,7 +19,7 @@ import numpy as np ...@@ -19,7 +19,7 @@ import numpy as np
from operator import mul from operator import mul
import paddle.fluid.core as core import paddle.fluid.core as core
import paddle.fluid as fluid import paddle.fluid as fluid
from op_test import OpTest from op_test import OpTest, skip_check_grad_ci
from testsuite import create_op from testsuite import create_op
...@@ -151,6 +151,10 @@ class TestGroupNormOpBigEps3(TestGroupNormOp): ...@@ -151,6 +151,10 @@ class TestGroupNormOpBigEps3(TestGroupNormOp):
self.attrs['epsilon'] = 0.5 self.attrs['epsilon'] = 0.5
@skip_check_grad_ci(
reason='''This test case is used to ensure whether the gradient checking results between CPU and GPU
are consistent when using the same inputs, thus, it doesn't need to call check_grad.'''
)
class TestGroupNormOpLargeData(TestGroupNormOp): class TestGroupNormOpLargeData(TestGroupNormOp):
def init_test_case(self): def init_test_case(self):
self.shape = (2, 32, 64, 64) self.shape = (2, 32, 64, 64)
...@@ -190,6 +194,10 @@ class TestGroupNormOpBigEps3_With_NHWC(TestGroupNormOp): ...@@ -190,6 +194,10 @@ class TestGroupNormOpBigEps3_With_NHWC(TestGroupNormOp):
self.data_format = "NHWC" self.data_format = "NHWC"
@skip_check_grad_ci(
reason='''This test case is used to ensure whether the gradient checking results between CPU and GPU
are consistent when using the same inputs, thus, it doesn't need to call check_grad.'''
)
class TestGroupNormOpLargeData_With_NHWC(TestGroupNormOp): class TestGroupNormOpLargeData_With_NHWC(TestGroupNormOp):
def init_test_case(self): def init_test_case(self):
self.shape = (2, 64, 32, 32) # NCHW self.shape = (2, 64, 32, 32) # NCHW
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册