Browse Source

pynative-add-lenet

tags/v0.3.0-alpha
lvliang chang zherui 5 years ago
parent
commit
244a82b244
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      tests/st/pynative/test_ascend_lenet.py

+ 2
- 1
tests/st/pynative/test_ascend_lenet.py View File

@@ -150,6 +150,7 @@ def test_ascend_pynative_lenet():
end_time = time.time() end_time = time.time()
cost_time = end_time - start_time cost_time = end_time - start_time
total_time = total_time + cost_time total_time = total_time + cost_time
print("======epoch: ", epoch, " loss: ", loss_output.asnumpy(), " cost time: ", cost_time)

assert(total_time < 20.0) assert(total_time < 20.0)
assert(loss_output.asnumpy() < 0.01) assert(loss_output.asnumpy() < 0.01)

Loading…
Cancel
Save