From 244a82b2445cb86673fbec98e3879a303d7230fb Mon Sep 17 00:00:00 2001 From: lvliang Date: Wed, 1 Apr 2020 14:08:26 +0800 Subject: [PATCH] pynative-add-lenet --- tests/st/pynative/test_ascend_lenet.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/st/pynative/test_ascend_lenet.py b/tests/st/pynative/test_ascend_lenet.py index 7f69b4585f6..c03b2eee850 100644 --- a/tests/st/pynative/test_ascend_lenet.py +++ b/tests/st/pynative/test_ascend_lenet.py @@ -150,6 +150,7 @@ def test_ascend_pynative_lenet(): end_time = time.time() cost_time = end_time - start_time total_time = total_time + cost_time + print("======epoch: ", epoch, " loss: ", loss_output.asnumpy(), " cost time: ", cost_time) + assert(total_time < 20.0) assert(loss_output.asnumpy() < 0.01) - \ No newline at end of file