Fix the random fail in test_image_classification.py (#14551)

test=develop
local_add_cudnn_lstm
qingqing01 7 years ago committed by GitHub
parent a7188d5bc7
commit c52f65e071
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -239,7 +239,7 @@ def infer(use_cuda, save_dirname=None):
assert len(results[0]) == len(transpiler_results[0])
for i in range(len(results[0])):
np.testing.assert_almost_equal(
results[0][i], transpiler_results[0][i], decimal=5)
results[0][i], transpiler_results[0][i], decimal=4)
print("infer results: ", results[0])

Loading…
Cancel
Save