|
|
|
@@ -412,6 +412,6 @@ def test_pynative_resnet50(): |
|
|
|
end_time = time.time() |
|
|
|
cost_time = end_time - start_time |
|
|
|
print("======step: ", step, " loss: ", loss_output.asnumpy(), " cost time: ", cost_time) |
|
|
|
if step > 1 and cost_time > 0.15: |
|
|
|
if step > 1 and cost_time > 0.18: |
|
|
|
exceed_num = exceed_num + 1 |
|
|
|
assert exceed_num < 18 |
|
|
|
assert exceed_num < 20 |