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

!36390 update transformer base line 

Merge pull request !36390 from zhaoting/master
This commit is contained in:
i-robot 2022-06-25 06:48:51 +00:00 committed by Gitee
parent 1893fb997a 5a61d9886c
commit 091e00fc38
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
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

@ -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]