diff --git a/tests/st/quantization/mobilenetv2_quant/test_mobilenetv2_quant.py b/tests/st/quantization/mobilenetv2_quant/test_mobilenetv2_quant.py index acb9531b9c..e682f3d635 100644 --- a/tests/st/quantization/mobilenetv2_quant/test_mobilenetv2_quant.py +++ b/tests/st/quantization/mobilenetv2_quant/test_mobilenetv2_quant.py @@ -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() diff --git a/tests/st/quantization/resnet50_quant/test_resnet50_quant.py b/tests/st/quantization/resnet50_quant/test_resnet50_quant.py index 3bac6a13d5..37c03ea0e1 100755 --- a/tests/st/quantization/resnet50_quant/test_resnet50_quant.py +++ b/tests/st/quantization/resnet50_quant/test_resnet50_quant.py @@ -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()