未验证 提交 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):
"""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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册