From c47bfe98cf7a3a3d6fba81e2b5e3eabc226abe5e Mon Sep 17 00:00:00 2001 From: Yang Zhang Date: Fri, 13 Nov 2020 13:24:20 +0800 Subject: [PATCH] Reduce input data size for syncbn unit test by 8x (#28571) this is a preemptive measure, the unit test was failing occasionally with cudnn internal error --- .../fluid/tests/unittests/test_sync_batch_norm_op.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) 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 bfd22dbe1ce..9a380c886e9 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 -- GitLab