Browse Source

annotate test case

tags/v1.1.0
wsq3 5 years ago
parent
commit
afc0b0b61a
1 changed files with 5 additions and 5 deletions
  1. +5
    -5
      tests/st/model_zoo_tests/DeepFM/test_deepfm.py

+ 5
- 5
tests/st/model_zoo_tests/DeepFM/test_deepfm.py View File

@@ -14,7 +14,7 @@
# ============================================================================
"""train_criteo."""
import os
import pytest
# import pytest

from mindspore import context
from mindspore.train.model import Model
@@ -27,10 +27,10 @@ from src.callback import EvalCallBack, LossCallBack, TimeMonitor

set_seed(1)

@pytest.mark.level0
@pytest.mark.platform_arm_ascend_training
@pytest.mark.platform_x86_ascend_training
@pytest.mark.env_onecard
# @pytest.mark.level0
# @pytest.mark.platform_arm_ascend_training
# @pytest.mark.platform_x86_ascend_training
# @pytest.mark.env_onecard
def test_deepfm():
data_config = DataConfig()
train_config = TrainConfig()


Loading…
Cancel
Save