fix unittest random failed test=develop (#19654)

sigmoid_bug
Aurelius84 5 years ago committed by GitHub
parent 5c4eb394e9
commit 25c0eb28cc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -505,8 +505,9 @@ class DistributionTest(unittest.TestCase):
feed={'logits': logits_np},
fetch_list=[entropy_np, kl_np])
np.testing.assert_allclose(
output_entropy_np, gt_entropy_np, rtol=tolerance)
np.testing.assert_allclose(output_kl_np, gt_kl_np, rtol=tolerance)
output_entropy_np, gt_entropy_np, rtol=tolerance, atol=tolerance)
np.testing.assert_allclose(
output_kl_np, gt_kl_np, rtol=tolerance, atol=tolerance)
def test_multivariateNormalDiag_distribution(self,
batch_size=2,
@ -568,8 +569,9 @@ class DistributionTest(unittest.TestCase):
},
fetch_list=[entropy_np, kl_np])
np.testing.assert_allclose(
output_entropy_np, gt_entropy_np, rtol=tolerance)
np.testing.assert_allclose(output_kl_np, gt_kl_np, rtol=tolerance)
output_entropy_np, gt_entropy_np, rtol=tolerance, atol=tolerance)
np.testing.assert_allclose(
output_kl_np, gt_kl_np, rtol=tolerance, atol=tolerance)
if __name__ == '__main__':

Loading…
Cancel
Save