Browse Source

!3737 Enlarge the threshold of resnet50 performance st in pynative

Merge pull request !3737 from JoyLvliang/enlarge-the-threshold-of-resnet50-performance-st-in-pynative
tags/v0.7.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
31e889881b
1 changed files with 5 additions and 3 deletions
  1. +5
    -3
      tests/st/pynative/test_pynative_resnet50.py

+ 5
- 3
tests/st/pynative/test_pynative_resnet50.py View File

@@ -413,6 +413,7 @@ def test_pynative_resnet50():

step = 0
max_step = 20
exceed_num = 0
data_set = create_dataset(repeat_num=1, training=True, batch_size=batch_size)
for element in data_set.create_dict_iterator():
step = step + 1
@@ -427,6 +428,7 @@ def test_pynative_resnet50():
end_time = time.time()
cost_time = end_time - start_time
print("======step: ", step, " loss: ", loss_output.asnumpy(), " cost time: ", cost_time)
if step > 1:
assert cost_time < 0.31
if step > 1 and cost_time > 0.32:
exceed_num = exceed_num + 1
assert exceed_num < 10

Loading…
Cancel
Save