Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

clould

This commit is contained in:
huchunmei 2021-07-21 14:03:01 +08:00
parent 5c40ca1f23
commit 786776ef43
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/st/model_zoo_tests/transformer/test_transformer.py
View File

@ -171,7 +171,7 @@ def test_transformer():
# assertion occurs while the loss value, overflow state or loss_scale value is wrong # assertion occurs while the loss value, overflow state or loss_scale value is wrong
loss_value = np.array(callback.loss_list) 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, expect_loss_value = [11.241606, 11.243232, 11.217459, 11.204157, 11.213804,
11.215373, 11.190564, 11.150393, 11.191823, 11.160045] 11.215373, 11.190564, 11.150393, 11.191823, 11.160045]