未验证 提交 f34fb0a2 编写于 作者: Z zhupengyang 提交者: GitHub

normal: enlarge threshold to reduce random error (#26763)

上级 b1f9ed60
...@@ -25,7 +25,7 @@ class TestNormalAPI(unittest.TestCase): ...@@ -25,7 +25,7 @@ class TestNormalAPI(unittest.TestCase):
self.mean = 1.0 self.mean = 1.0
self.std = 0.0 self.std = 0.0
self.shape = None self.shape = None
self.repeat_num = 1000 self.repeat_num = 2000
self.set_attrs() self.set_attrs()
self.dtype = self.get_dtype() self.dtype = self.get_dtype()
self.place=paddle.CUDAPlace(0) \ self.place=paddle.CUDAPlace(0) \
...@@ -134,8 +134,8 @@ class TestNormalAPI(unittest.TestCase): ...@@ -134,8 +134,8 @@ class TestNormalAPI(unittest.TestCase):
if isinstance(self.mean, np.ndarray) else self.mean if isinstance(self.mean, np.ndarray) else self.mean
std_ref=self.std.reshape([1, -1]) \ std_ref=self.std.reshape([1, -1]) \
if isinstance(self.std, np.ndarray) else self.std if isinstance(self.std, np.ndarray) else self.std
self.assertTrue(np.allclose(mean_ref, mean, 0.1, 0.1)) self.assertTrue(np.allclose(mean_ref, mean, 0.2, 0.2))
self.assertTrue(np.allclose(std_ref, std, 0.1, 0.1)) self.assertTrue(np.allclose(std_ref, std, 0.2, 0.2))
class TestNormalAPI_mean_is_tensor(TestNormalAPI): class TestNormalAPI_mean_is_tensor(TestNormalAPI):
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册