pynative-add-lenet

5 years ago · 244a82b244
parent 3e68560023
commit 244a82b244
1 changed files with 2 additions and 1 deletions
--- 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)
-