forked from OSSInnovation/mindspore
!6753 modify funcion name of test case
Merge pull request !6753 from hwjiaorui/master
This commit is contained in:
commit
4238e34982
|
@ -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()
|
||||
|
|
|
@ -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…
Reference in New Issue