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

!31304 fix bert preset precision 

Merge pull request !31304 from wanyiming/fix_st
This commit is contained in:
i-robot 2022-03-16 01:30:46 +00:00 committed by Gitee
parent d3f7e99126 b30217b161
commit 2fd52742bb
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/st/networks/models/bert/bert_precision/test_bert_tdt_lossscale.py
View File

@ -229,8 +229,8 @@ def test_bert_precision(enable_graph_kernel=False):
12.185522, 12.386192]
else:
assert np.allclose(loss_value[0], 12.2066, 0, 0.0005)
expect_loss_value = [12.206557, 11.940715, 11.930809, 11.937332, 11.931781, 12.555943, 12.130274, 12.783247,
12.359643, 12.578059]
expect_loss_value = [12.206557, 11.941042, 11.931373, 11.938049, 11.932574, 12.556715, 12.130486, 12.7836895,
12.360162, 12.578404]
print("loss value: {}".format(loss_value))
assert np.allclose(loss_value, expect_loss_value, 0, 0.0005)