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

modify name test

This commit is contained in:
hwjiaorui 2020-09-23 09:44:40 +08:00
parent 9dad633ca8
commit 2e1e36f8af
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_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()

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_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()