xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!18196 Fix the ST probabilistic failure in daily version<test_transformer> 

Merge pull request !18196 from casgj/master_ci
This commit is contained in:
i-robot 2021-06-11 16:09:47 +08:00 committed by Gitee
parent 032b03b4b0 dd9aa8bab5
commit 1ccb1f5946
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