Browse Source

modify name test

tags/v1.1.0
hwjiaorui 5 years ago
parent
commit
2e1e36f8af
2 changed files with 4 additions and 4 deletions
  1. +2
    -2
      tests/st/quantization/mobilenetv2_quant/test_mobilenetv2_quant.py
  2. +2
    -2
      tests/st/quantization/resnet50_quant/test_resnet50_quant.py

+ 2
- 2
tests/st/quantization/mobilenetv2_quant/test_mobilenetv2_quant.py View File

@@ -60,7 +60,7 @@ dataset_path = "/dataset/workspace/mindspore_dataset/cifar-10-batches-bin/"
@pytest.mark.platform_arm_ascend_training
@pytest.mark.platform_x86_ascend_training
@pytest.mark.env_onecard
def train_on_ascend():
def test_mobilenetv2_quant():
set_seed(1)
context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
config = config_ascend_quant
@@ -120,4 +120,4 @@ def train_on_ascend():


if __name__ == '__main__':
train_on_ascend()
test_mobilenetv2_quant()

+ 2
- 2
tests/st/quantization/resnet50_quant/test_resnet50_quant.py View File

@@ -63,7 +63,7 @@ dataset_path = "/dataset/workspace/mindspore_dataset/cifar-10-batches-bin/"
@pytest.mark.platform_arm_ascend_training
@pytest.mark.platform_x86_ascend_training
@pytest.mark.env_onecard
def train_on_ascend():
def test_resnet50_quant():
set_seed(1)
context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
config = config_quant
@@ -128,4 +128,4 @@ def train_on_ascend():


if __name__ == '__main__':
train_on_ascend()
test_resnet50_quant()

Loading…
Cancel
Save