diff --git a/python/paddle/fluid/tests/unittests/test_sync_batch_norm_op.py b/python/paddle/fluid/tests/unittests/test_sync_batch_norm_op.py index bfd22dbe1cee628d4549f69cfedc09409b6ef953..9a380c886e9154690e420abda5011ca3aee9b6bb 100644 --- a/python/paddle/fluid/tests/unittests/test_sync_batch_norm_op.py +++ b/python/paddle/fluid/tests/unittests/test_sync_batch_norm_op.py @@ -50,9 +50,9 @@ class TestSyncBatchNormOpTraining(unittest.TestCase): """Setup.""" #self.dtype = np.float32 self.dtype = np.float64 - self.N = 32 + self.N = 8 self.C = 16 - self.H = 64 + self.H = 32 self.W = 32 self.dshape = [self.N, self.C, self.H, self.W] self.atol = 1e-3 @@ -196,9 +196,9 @@ class TestFP16SyncBatchNormOpTraining(TestSyncBatchNormOpTraining): def setUp(self): """Setup.""" self.dtype = np.float16 - self.N = 32 + self.N = 8 self.C = 16 - self.H = 64 + self.H = 32 self.W = 32 self.dshape = [self.N, self.C, self.H, self.W] self.atol = 1e-2