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

update tests/st/model_zoo_tests/transformer/test_transformer.py.

This commit is contained in:
casgj 2021-06-11 11:07:55 +08:00 committed by Gitee
parent f606951063
commit dd9aa8bab5
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/model_zoo_tests/transformer/test_transformer.py
View File

@ -195,12 +195,12 @@ def test_transformer():
epoch_mseconds = np.array(time_monitor_callback.epoch_mseconds_list)[2]
expect_epoch_mseconds = 2400
print("epoch mseconds: {}".format(epoch_mseconds))
assert epoch_mseconds <= expect_epoch_mseconds + 20
assert epoch_mseconds <= expect_epoch_mseconds + 100
per_step_mseconds = np.array(time_monitor_callback.per_step_mseconds_list)[2]
expect_per_step_mseconds = 240
print("per step mseconds: {}".format(per_step_mseconds))
assert per_step_mseconds <= expect_per_step_mseconds + 2
assert per_step_mseconds <= expect_per_step_mseconds + 10
@pytest.mark.level1