|
|
|
@@ -131,9 +131,11 @@ class GradWrap(nn.Cell): |
|
|
|
@pytest.mark.level0 |
|
|
|
@pytest.mark.platform_arm_ascend_training |
|
|
|
@pytest.mark.platform_x86_ascend_training |
|
|
|
@pytest.mark.platform_x86_cpu |
|
|
|
@pytest.mark.platform_x86_gpu_training |
|
|
|
@pytest.mark.env_onecard |
|
|
|
def test_ascend_pynative_lenet(): |
|
|
|
context.set_context(mode=context.PYNATIVE_MODE, device_target="Ascend") |
|
|
|
context.set_context(mode=context.PYNATIVE_MODE) |
|
|
|
|
|
|
|
epoch_size = 20 |
|
|
|
batch_size = 32 |
|
|
|
|