未验证 提交 c47bfe98 编写于 作者: Y Yang Zhang 提交者: GitHub

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
上级 c545b9b6
...@@ -50,9 +50,9 @@ class TestSyncBatchNormOpTraining(unittest.TestCase): ...@@ -50,9 +50,9 @@ class TestSyncBatchNormOpTraining(unittest.TestCase):
"""Setup.""" """Setup."""
#self.dtype = np.float32 #self.dtype = np.float32
self.dtype = np.float64 self.dtype = np.float64
self.N = 32 self.N = 8
self.C = 16 self.C = 16
self.H = 64 self.H = 32
self.W = 32 self.W = 32
self.dshape = [self.N, self.C, self.H, self.W] self.dshape = [self.N, self.C, self.H, self.W]
self.atol = 1e-3 self.atol = 1e-3
...@@ -196,9 +196,9 @@ class TestFP16SyncBatchNormOpTraining(TestSyncBatchNormOpTraining): ...@@ -196,9 +196,9 @@ class TestFP16SyncBatchNormOpTraining(TestSyncBatchNormOpTraining):
def setUp(self): def setUp(self):
"""Setup.""" """Setup."""
self.dtype = np.float16 self.dtype = np.float16
self.N = 32 self.N = 8
self.C = 16 self.C = 16
self.H = 64 self.H = 32
self.W = 32 self.W = 32
self.dshape = [self.N, self.C, self.H, self.W] self.dshape = [self.N, self.C, self.H, self.W]
self.atol = 1e-2 self.atol = 1e-2
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册