From e694a916fcaab4c03e8b9ade3c4208f105f81f00 Mon Sep 17 00:00:00 2001 From: LianLiguang Date: Wed, 16 Dec 2020 10:03:56 +0800 Subject: [PATCH] adjust-st-of-resnet50-in-ascend-pynative-to-gpu-pynative --- tests/st/pynative/test_pynative_resnet50_gpu.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/st/pynative/test_pynative_resnet50_gpu.py b/tests/st/pynative/test_pynative_resnet50_gpu.py index 0cb1c75dbc..9402e51853 100644 --- a/tests/st/pynative/test_pynative_resnet50_gpu.py +++ b/tests/st/pynative/test_pynative_resnet50_gpu.py @@ -412,6 +412,6 @@ 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 and cost_time > 0.15: + if step > 1 and cost_time > 0.18: exceed_num = exceed_num + 1 - assert exceed_num < 18 + assert exceed_num < 20