From 786776ef435942b231ad8803fd747707eb1a78fb Mon Sep 17 00:00:00 2001 From: huchunmei Date: Wed, 21 Jul 2021 14:03:01 +0800 Subject: [PATCH] clould --- tests/st/model_zoo_tests/transformer/test_transformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/st/model_zoo_tests/transformer/test_transformer.py b/tests/st/model_zoo_tests/transformer/test_transformer.py index 7ee8769944a..fa4b4f9c8fc 100644 --- a/tests/st/model_zoo_tests/transformer/test_transformer.py +++ b/tests/st/model_zoo_tests/transformer/test_transformer.py @@ -171,7 +171,7 @@ def test_transformer(): # assertion occurs while the loss value, overflow state or loss_scale value is wrong loss_value = np.array(callback.loss_list) - assert np.allclose(loss_value[0], 11.241606, 0, 0.000005) + assert np.allclose(loss_value[0], 11.241601, 0, 0.000005) expect_loss_value = [11.241606, 11.243232, 11.217459, 11.204157, 11.213804, 11.215373, 11.190564, 11.150393, 11.191823, 11.160045]