add decorator skip_check_grad_ci for test_group_norm (#21910)

* add decorator skip_check_grad_ci for test_group_norm test=develop
release/1.7
lijianshe02 5 years ago committed by GitHub
parent feabed1309
commit f17bd1786a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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

Loading…
Cancel
Save