未验证 提交 49e243c9 编写于 作者: F Feng Xing 提交者: GitHub

test=document_fix (#35655)

上级 a4b67f78
...@@ -142,10 +142,7 @@ class TestParallelMarginSoftmaxCrossEntropyOp(unittest.TestCase): ...@@ -142,10 +142,7 @@ class TestParallelMarginSoftmaxCrossEntropyOp(unittest.TestCase):
return_softmax=True) return_softmax=True)
np.testing.assert_allclose( np.testing.assert_allclose(
loss_a.numpy(), loss_a.numpy(), loss_b.numpy(), rtol=1e-5)
loss_b.numpy(),
rtol=1e-5,
atol=1e-7)
integral_prob = np.zeros( integral_prob = np.zeros(
(batch_size, num_class), dtype=dtype) (batch_size, num_class), dtype=dtype)
...@@ -184,8 +181,7 @@ class TestParallelMarginSoftmaxCrossEntropyOp(unittest.TestCase): ...@@ -184,8 +181,7 @@ class TestParallelMarginSoftmaxCrossEntropyOp(unittest.TestCase):
np.testing.assert_allclose( np.testing.assert_allclose(
integral_data.grad.numpy(), integral_data.grad.numpy(),
integral_grad.numpy(), integral_grad.numpy(),
rtol=1e-5, rtol=1e-5)
atol=1e-7)
if __name__ == '__main__': if __name__ == '__main__':
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册