recover test_bert_tdt_lossscale testcase and update sklearn
This commit is contained in:
parent
9a81a0e969
commit
c9fc8b0747
|
@ -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
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue