From 10bee0b3c05bee0503658c385e8ca57f39030c8a Mon Sep 17 00:00:00 2001 From: Siju Samuel Date: Mon, 13 Apr 2020 12:10:19 +0530 Subject: [PATCH] typo bug in qnn test --- tests/python/frontend/pytorch/qnn_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/python/frontend/pytorch/qnn_test.py b/tests/python/frontend/pytorch/qnn_test.py index 82e3393a4a3d2..bbc3ad325a6f0 100644 --- a/tests/python/frontend/pytorch/qnn_test.py +++ b/tests/python/frontend/pytorch/qnn_test.py @@ -395,7 +395,7 @@ def get_imagenet_input(): max_abs_diff = np.max(np.abs(tvm_result - pt_result)) mean_abs_diff = np.mean(np.abs(tvm_result - pt_result)) num_identical = np.sum(tvm_result == pt_result) - pt_top3_labels = np.argsort(pt_result)[::-1][:3] + pt_top3_labels = np.argsort(tvm_result)[::-1][:3] tvm_top3_labels = np.argsort(pt_result)[::-1][:3] print("\nModel name: %s" % model_name)