From 5a61d9886c040ce4eeab6a17af9d04debec9c0f4 Mon Sep 17 00:00:00 2001 From: zhaoting Date: Thu, 23 Jun 2022 11:31:47 +0800 Subject: [PATCH] update transformer base line --- 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 b87f27b0b36..34a2a34ce23 100644 --- a/tests/st/model_zoo_tests/transformer/test_transformer.py +++ b/tests/st/model_zoo_tests/transformer/test_transformer.py @@ -187,7 +187,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.241601, 0, 0.000005) + assert np.allclose(loss_value[0], 11.241601, 0, 0.00005) expect_loss_value = [11.241606, 11.243232, 11.217459, 11.204157, 11.213804, 11.215373, 11.190564, 11.150393, 11.191823, 11.160045]