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

!49216 recover test_bert_tdt_lossscale testcase 

Merge pull request !49216 from anzhengqi/fix-testcase-bert-performance
This commit is contained in:
i-robot 2023-02-23 13:03:29 +00:00 committed by Gitee
parent 63dd755fc6 c9fc8b0747
commit b69a295fcc
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
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