From c9fc8b0747f79553c328cb42f0d880e09a1d102d Mon Sep 17 00:00:00 2001 From: anzhengqi Date: Wed, 22 Feb 2023 09:38:56 +0800 Subject: [PATCH] recover test_bert_tdt_lossscale testcase and update sklearn --- requirements.txt | 2 +- .../models/bert/bert_performance/test_bert_tdt_lossscale.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/requirements.txt b/requirements.txt index 3cdc03376df..f3b2435cac7 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,7 +6,7 @@ scipy >= 1.5.4 decorator >= 4.4.0 matplotlib >= 3.1.3 # for ut test opencv-python >= 4.1.2.30 # for ut test -sklearn >= 0.0 # for st test +scikit-learn >= 0.14 # for st test pandas >= 1.0.2 # for ut test packaging >= 20.0 pycocotools >= 2.0.2 # for st test diff --git a/tests/st/networks/models/bert/bert_performance/test_bert_tdt_lossscale.py b/tests/st/networks/models/bert/bert_performance/test_bert_tdt_lossscale.py index b5d058fa979..2efee342174 100644 --- a/tests/st/networks/models/bert/bert_performance/test_bert_tdt_lossscale.py +++ b/tests/st/networks/models/bert/bert_performance/test_bert_tdt_lossscale.py @@ -174,7 +174,7 @@ class TimeMonitor(Callback): self.per_step_mseconds_list.append(epoch_mseconds / self.data_size) -@pytest.mark.level1 +@pytest.mark.level0 @pytest.mark.platform_arm_ascend_training @pytest.mark.platform_x86_ascend_training @pytest.mark.env_onecard