Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

recover test_bert_tdt_lossscale testcase and update sklearn

This commit is contained in:
anzhengqi 2023-02-22 09:38:56 +08:00
parent 9a81a0e969
commit c9fc8b0747
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
requirements.txt
View File

@ -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

2
tests/st/networks/models/bert/bert_performance/test_bert_tdt_lossscale.py
View File

@ -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