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

fix ci deepfm and data sink st

This commit is contained in:
jiaorui 2023-02-21 16:20:39 +08:00
parent 6e1aca2f3c
commit f983745f46
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/st/model_zoo_tests/DeepFM/test_deepfm.py
View File

@ -27,7 +27,8 @@ from src.callback import EvalCallBack, LossCallBack, TimeMonitor
set_seed(1)
@pytest.mark.level1
@pytest.mark.level0
@pytest.mark.platform_arm_ascend_training
@pytest.mark.platform_x86_ascend_training
@pytest.mark.env_onecard

2
tests/st/train/test_data_sink.py
View File

@ -98,7 +98,7 @@ def _train_func_sink(model, dataset, loss_fn, opt, input_signature=None):
print("loss: ", loss)
@pytest.mark.level1
@pytest.mark.level0
@pytest.mark.platform_arm_ascend_training
@pytest.mark.platform_x86_ascend_training
@pytest.mark.platform_x86_gpu_training