From ed1e4f7ae0b8c0b76c8b3a66fb3f4e13ed40bdd1 Mon Sep 17 00:00:00 2001 From: mwang Date: Wed, 17 Mar 2021 19:18:22 +0800 Subject: [PATCH] adjust performance of bert thor --- .../networks/models/bert/bert_performance/test_bert_thor.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/st/networks/models/bert/bert_performance/test_bert_thor.py b/tests/st/networks/models/bert/bert_performance/test_bert_thor.py index 8f9f752a8a..80b55ae4dd 100644 --- a/tests/st/networks/models/bert/bert_performance/test_bert_thor.py +++ b/tests/st/networks/models/bert/bert_performance/test_bert_thor.py @@ -193,7 +193,7 @@ def train_process_bert_thor(q, device_id, epoch_size, device_num): @pytest.mark.platform_arm_ascend_training @pytest.mark.platform_x86_ascend_training @pytest.mark.env_single -def test_bert_thor_mlperf_8p(): +def test_bert_thor_8p(): """test bert thor mlperf 8p""" q = Queue() device_num = 8 @@ -234,12 +234,12 @@ def test_bert_thor_mlperf_8p(): os.system("rm -rf " + str(i)) print("End training...") - assert mean_cost < 78 + assert mean_cost < 66 assert mean_loss < 8.125 if __name__ == '__main__': begin = time.time() - test_bert_thor_mlperf_8p() + test_bert_thor_8p() end = time.time() print("time span is", end - begin, flush=True)