sfqcyy
/
mindspore
forked from OSSInnovation/mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!6753 modify funcion name of test case 

Merge pull request !6753 from hwjiaorui/master
This commit is contained in:
mindspore-ci-bot 2020-09-23 14:18:03 +08:00 committed by Gitee
parent b993ea0288 2e1e36f8af
commit 4238e34982
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
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_arm_ascend_training
@pytest.mark.platform_x86_ascend_training @pytest.mark.platform_x86_ascend_training
@pytest.mark.env_onecard @pytest.mark.env_onecard
def train_on_ascend(): def test_mobilenetv2_quant():
set_seed(1) set_seed(1)
context.set_context(mode=context.GRAPH_MODE, device_target="Ascend") context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
config = config_ascend_quant config = config_ascend_quant
@ -120,4 +120,4 @@ def train_on_ascend():
if __name__ == '__main__': if __name__ == '__main__':
train_on_ascend() test_mobilenetv2_quant()

4
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_arm_ascend_training
@pytest.mark.platform_x86_ascend_training @pytest.mark.platform_x86_ascend_training
@pytest.mark.env_onecard @pytest.mark.env_onecard
def train_on_ascend(): def test_resnet50_quant():
set_seed(1) set_seed(1)
context.set_context(mode=context.GRAPH_MODE, device_target="Ascend") context.set_context(mode=context.GRAPH_MODE, device_target="Ascend")
config = config_quant config = config_quant
@ -128,4 +128,4 @@ def train_on_ascend():
if __name__ == '__main__': if __name__ == '__main__':
train_on_ascend() test_resnet50_quant()