未验证 提交 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
from operator import mul
import paddle.fluid.core as core
import paddle.fluid as fluid
from op_test import OpTest
from op_test import OpTest, skip_check_grad_ci
from testsuite import create_op
......@@ -151,6 +151,10 @@ class TestGroupNormOpBigEps3(TestGroupNormOp):
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):
def init_test_case(self):
self.shape = (2, 32, 64, 64)
......@@ -190,6 +194,10 @@ class TestGroupNormOpBigEps3_With_NHWC(TestGroupNormOp):
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):
def init_test_case(self):
self.shape = (2, 64, 32, 32) # NCHW
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册